Spaces:
Sleeping
Sleeping
| import os | |
| from dotenv import load_dotenv | |
| from utilities.llm.LlmManager import LlmManager | |
| from utilities.mongodb.MongoDBManager import MongoDBManager | |
| credentials = {} | |
| def check_user(username, password): | |
| if username in credentials and credentials[username] == password: | |
| print("π Login successful!") | |
| return True | |
| else: | |
| print("β Incorrect username or password") | |
| return False | |
| def initialize(): | |
| global credentials | |
| try: | |
| load_dotenv() | |
| mongo_uri = os.getenv("MONGO_URI") | |
| db_name = os.getenv("DB_NAME") | |
| llm_manager = LlmManager() | |
| mongo_db_manager = MongoDBManager(uri=mongo_uri, db_name=db_name) | |
| credentials[os.getenv("USERNAME")] = os.getenv("PASSWORD") | |
| i = 1 | |
| while os.getenv(f"USERNAME_{i}") and os.getenv(f"PASSWORD_{i}"): | |
| credentials[os.getenv(f"USERNAME_{i}")] = os.getenv(f"PASSWORD_{i}") | |
| i += 1 | |
| return llm_manager,mongo_db_manager | |
| except Exception as e: | |
| print("Error initializing configuration:", e, flush=True) | |
| return None,None | |
| # Qdrant | |
| # try: | |
| # qdrant_manager = QdrantLangchainManager( | |
| # qdrant_url=os.getenv("QDRANT_URL"), | |
| # qdrant_api_key=os.getenv("QDRANT_API_KEY"), | |
| # llm=ChatOpenAI(model="gpt-4o-mini", streaming=True), | |
| # crossencoder_model=crossencoder_model | |
| # ) | |
| # config['qdrant_manager'] = qdrant_manager | |
| # config['qdrant_connected'] = True | |
| # except Exception as e: | |
| # config['qdrant_connected'] = False | |
| # print(f"Error connecting to Qdrant: {e}", flush=True) | |
| return config |