| from core.services.supabase.user_management.user_service import UserManagement | |
| from core.services.supabase.user_management.chatbot_management import SupabaseChatoBotManagement | |
| from core.services.supabase.chat_ai_setup.chataiusermanagement import ChatAIUserDBSetup | |
| from core.services.supabase.user_management.chat_history import get_chat_history | |
| from supabase.client import create_client | |
| from qdrant_client import QdrantClient | |
| import os | |
| SUPABASE_URL = os.getenv("SUPABASE_URL") | |
| SUPABASE_KEY = os.getenv("SUPABASE_KEY") | |
| QDRANT_URL = os.getenv("QDRANT_URL") | |
| QDRANT_API_KEY = os.getenv("QDRANT_API_KEY") | |
| class SupabaseUserManagementPipeline: | |
| def __init__(self): | |
| self.supabase_client = create_client(SUPABASE_URL, SUPABASE_KEY) | |
| self.qdrant_client = QdrantClient(url=QDRANT_URL, api_key=QDRANT_API_KEY) | |
| self.user_management = UserManagement(supabase_client=self.supabase_client) | |
| self.chatbot_management = SupabaseChatoBotManagement(supabase_client=self.supabase_client, | |
| qdrant_client=self.qdrant_client) | |
| self.user_db_setup = ChatAIUserDBSetup(supabase_client=self.supabase_client) | |
| def user_signup_(self, username: str,password: str,email:str) -> dict: | |
| return self.user_management.user_signup(username=username, password=password, email=email) | |
| def user_signin_(self, username: str, password: str,email:str) -> dict: | |
| return self.user_management.user_signin(username=username, password=password,email=email) | |
| def check_session_(self) -> dict: | |
| return self.user_management.check_session() | |
| def get_user_data_(self, access_token: str) -> dict: | |
| return self.user_management.get_user_data(access_token=access_token) | |
| def refresh_session__(self, refresh_token: str) -> dict: | |
| return self.user_management.refresh_session_(refresh_token=refresh_token) | |
| def login_with_access_token_(self, access_token: str, refresh_token: str) -> dict: | |
| return self.user_management.login_with_access_token(access_token=access_token, refresh_token=refresh_token) | |
| def username_creation_oauth_(self, username: str, user_id: str): | |
| return self.user_management.user_name_creation_oauth(user_id=user_id, username=username) | |
| def set_session_data_(self, access_token: str, refresh_token: str, user_id: str): | |
| return self.user_management.set_session_data(access_token=access_token, refresh_token=refresh_token, | |
| user_id=user_id) | |
| def sign_out_(self): | |
| return self.user_management.sign_out_() | |
| def oauth_signin_(self) -> dict: | |
| return self.user_management.oauth() | |
| def new_chatbot_(self, chatbot_name: str, username: str): | |
| return self.chatbot_management.new_chatbot(chatbot_name=chatbot_name, username=username) | |
| def get_chat_history_(self, vectorstore: str): | |
| return get_chat_history(vectorstore=vectorstore, supabase_client=self.supabase_client) | |
| def delete_table(self, table_name: str): | |
| return self.chatbot_management.delete_table(table_name=table_name) | |
| def list_tables(self, username: str): | |
| return self.chatbot_management.list_tables(username=username) | |
| def create_data_source_name(self, source_name: str, username: str): | |
| return self.chatbot_management.create_data_source_name(source_name=source_name, username=username) | |
| def delete_qdrant_cluster(self, vectorstorename): | |
| self.qdrant_client.delete_collection(collection_name=vectorstorename) | |