Spaces:
Sleeping
Sleeping
| from pymongo.mongo_client import MongoClient | |
| from pymongo.server_api import ServerApi | |
| from qdrant_client import QdrantClient | |
| from qdrant_client import QdrantClient | |
| import os | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| # MongoDB connection | |
| MONGODB_CONNECTION_STRING = os.getenv("MONGODB_CONNECTION_STRING") | |
| DB_NAME = "chatbot_demo" | |
| # Demo chatbot defaults | |
| DEMO_MAX_MESSAGES = 20 | |
| DEMO_EXPIRY_DAYS = 7 | |
| # Create a new client and connect to the server | |
| client = MongoClient(MONGODB_CONNECTION_STRING, server_api=ServerApi('1')) | |
| # Send a ping to confirm a successful connection | |
| try: | |
| client.admin.command('ping') | |
| print("Pinged your deployment. You successfully connected to MongoDB!") | |
| demo_database = client["Demo"] | |
| chatbot_database = client['Chatbot'] | |
| logs_databse = client["Logs"] | |
| demo_form_submissions = demo_database["demo_form_submissions"] | |
| demo_chatbot_configs = demo_database["demo_chatbot_config"] | |
| prod_chatbot_configs = chatbot_database['rag_configs'] | |
| admin_database = client["AdminLogs"] | |
| chat_logs = logs_databse["ChatLogs"] | |
| print("Connected to MongoDB collection successfully!") | |
| except Exception as e: | |
| print(e) | |
| #Qdrant | |
| QDRANT_URL = os.getenv("QDRANT_URL") | |
| QDRANT_API_KEY = os.getenv("QDRANT_API_KEY") | |
| qdrant_client = QdrantClient(url=QDRANT_URL, api_key=QDRANT_API_KEY) | |
| embedding_model = "intfloat/e5-base-v2" |