ELI / config.py
StefanoDUrso's picture
adding login
f6d4ec3
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