fix: mongo db database name in settings

Author: dartpain

application/agents/base.py

@@ -10,6 +10,7 @@
 from application.llm.llm_creator import LLMCreator
 from application.logging import build_stack_data, log_activity, LogContext
 from application.retriever.base import BaseRetriever
+from application.core.settings import settings
 from bson.objectid import ObjectId
 
 
@@ -61,7 +62,7 @@ def _gen_inner(
 
     def _get_tools(self, api_key: str = None) -> Dict[str, Dict]:
         mongo = MongoDB.get_client()
-        db = mongo["docsgpt"]
+        db = mongo[settings.MONGO_DB_NAME]
         agents_collection = db["agents"]
         tools_collection = db["user_tools"]
 
@@ -82,7 +83,7 @@ def _get_tools(self, api_key: str = None) -> Dict[str, Dict]:
 
     def _get_user_tools(self, user="local"):
         mongo = MongoDB.get_client()
-        db = mongo["docsgpt"]
+        db = mongo[settings.MONGO_DB_NAME]
         user_tools_collection = db["user_tools"]
         user_tools = user_tools_collection.find({"user": user, "status": True})
         user_tools = list(user_tools)

application/api/answer/routes.py

@@ -23,7 +23,7 @@
 logger = logging.getLogger(__name__)
 
 mongo = MongoDB.get_client()
-db = mongo["docsgpt"]
+db = mongo[settings.MONGO_DB_NAME]
 conversations_collection = db["conversations"]
 sources_collection = db["sources"]
 prompts_collection = db["prompts"]

application/api/internal/routes.py

@@ -11,7 +11,7 @@
 
 logger = logging.getLogger(__name__)
 mongo = MongoDB.get_client()
-db = mongo["docsgpt"]
+db = mongo[settings.MONGO_DB_NAME]
 conversations_collection = db["conversations"]
 sources_collection = db["sources"]
 

application/api/user/routes.py

@@ -23,7 +23,7 @@
 from application.vectorstore.vector_creator import VectorCreator
 
 mongo = MongoDB.get_client()
-db = mongo["docsgpt"]
+db = mongo[settings.MONGO_DB_NAME]
 conversations_collection = db["conversations"]
 sources_collection = db["sources"]
 prompts_collection = db["prompts"]

application/core/settings.py

@@ -19,6 +19,7 @@ class Settings(BaseSettings):
     CELERY_BROKER_URL: str = "redis://localhost:6379/0"
     CELERY_RESULT_BACKEND: str = "redis://localhost:6379/1"
     MONGO_URI: str = "mongodb://localhost:27017/docsgpt"
+    MONGO_DB_NAME: str = "docsgpt"
     MODEL_PATH: str = os.path.join(current_dir, "models/docsgpt-7b-f16.gguf")
     DEFAULT_MAX_HISTORY: int = 150
     MODEL_TOKEN_LIMITS: dict = {

application/llm/google_ai.py

@@ -5,6 +5,7 @@
 
 from application.llm.base import BaseLLM
 from application.storage.storage_creator import StorageCreator
+from application.core.settings import settings
 
 
 class GoogleLLM(BaseLLM):
@@ -120,7 +121,7 @@ def _upload_file_to_google(self, attachment):
 
             from application.core.mongo_db import MongoDB
             mongo = MongoDB.get_client()
-            db = mongo["docsgpt"]
+            db = mongo[settings.MONGO_DB_NAME]
             attachments_collection = db["attachments"]
             if '_id' in attachment:
                 attachments_collection.update_one(

application/llm/openai.py

@@ -291,7 +291,7 @@ def _upload_file_to_openai(self, attachment):
 
             from application.core.mongo_db import MongoDB
             mongo = MongoDB.get_client()
-            db = mongo["docsgpt"]
+            db = mongo[settings.MONGO_DB_NAME]
             attachments_collection = db["attachments"]
             if '_id' in attachment:
                 attachments_collection.update_one(

application/logging.py

@@ -7,6 +7,7 @@
 from typing import Any, Callable, Dict, Generator, List
 
 from application.core.mongo_db import MongoDB
+from application.core.settings import settings
 
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
@@ -131,7 +132,7 @@ def _log_to_mongodb(
 ) -> None:
     try:
         mongo = MongoDB.get_client()
-        db = mongo["docsgpt"]
+        db = mongo[settings.MONGO_DB_NAME]
         user_logs_collection = db["stack_logs"]
 
         log_entry = {

application/usage.py

@@ -2,10 +2,11 @@
 from datetime import datetime
 
 from application.core.mongo_db import MongoDB
+from application.core.settings import settings
 from application.utils import num_tokens_from_object_or_list, num_tokens_from_string
 
 mongo = MongoDB.get_client()
-db = mongo["docsgpt"]
+db = mongo[settings.MONGO_DB_NAME]
 usage_collection = db["token_usage"]
 
 

application/worker.py

@@ -26,7 +26,7 @@
 from application.utils import count_tokens_docs
 
 mongo = MongoDB.get_client()
-db = mongo["docsgpt"]
+db = mongo[settings.MONGO_DB_NAME]
 sources_collection = db["sources"]
 
 # Constants
@@ -356,7 +356,7 @@ def attachment_worker(self, file_info, user):
     """
 
     mongo = MongoDB.get_client()
-    db = mongo["docsgpt"]
+    db = mongo[settings.MONGO_DB_NAME]
     attachments_collection = db["attachments"]
 
     filename = file_info["filename"]