black

Author: piyushroshan

services/chatbot/src/chatbot/retrieverutils.py

@@ -46,7 +46,7 @@ async def get_chroma_collection(api_key):
         name="chats",
         embedding_function=OpenAIEmbeddingFunction(
             api_key=api_key,
-            model="text-embedding-3-large",
+            model_name="text-embedding-3-large",
         ),
     )
     return collection

services/chatbot/src/mcpserver/config.py

@@ -3,11 +3,12 @@
 
 load_dotenv()
 
+
 class Config:
     TLS_ENABLED = os.getenv("TLS_ENABLED", "false").lower() in ("true", "1", "yes")
     WEB_SERVICE = os.getenv("WEB_SERVICE", "crapi-web")
     IDENTITY_SERVICE = os.getenv("IDENTITY_SERVICE", "crapi-identity:8080")
     CHROMA_PERSIST_DIRECTORY = os.getenv("CHROMA_PERSIST_DIRECTORY", "/app/vectorstore")
     OPENAPI_SPEC = os.getenv("OPENAPI_SPEC", "/app/resources/crapi-openapi-spec.json")
     API_USER = os.getenv("API_USER", "admin@example.com")
-    API_PASSWORD = os.getenv("API_PASSWORD", "Admin!123")
+    API_PASSWORD = os.getenv("API_PASSWORD", "Admin!123")

services/chatbot/src/mcpserver/server.py

@@ -17,7 +17,9 @@
 logger.setLevel(logging.DEBUG)
 
 BASE_URL = f"{'https' if Config.TLS_ENABLED else 'http'}://{Config.WEB_SERVICE}"
-BASE_IDENTITY_URL = f"{'https' if Config.TLS_ENABLED else 'http'}://{Config.IDENTITY_SERVICE}"
+BASE_IDENTITY_URL = (
+    f"{'https' if Config.TLS_ENABLED else 'http'}://{Config.IDENTITY_SERVICE}"
+)
 API_KEY = None
 
 
@@ -57,6 +59,7 @@ def get_api_key():
                 return API_KEY
     return API_KEY
 
+
 # Async HTTP client for API calls
 def get_http_client():
     """Create and configure the HTTP client with appropriate authentication."""
@@ -68,7 +71,8 @@ def get_http_client():
         headers=headers,
     )
 
-# Load your OpenAPI spec 
+
+# Load your OpenAPI spec
 with open(Config.OPENAPI_SPEC, "r") as f:
     openapi_spec = json.load(f)
 

services/chatbot/src/mcpserver/tool_helpers.py

@@ -7,6 +7,7 @@
 from langchain.chains import RetrievalQA
 from langchain_openai import ChatOpenAI
 
+
 async def get_any_api_key():
     if os.environ.get("CHATBOT_OPENAI_API_KEY"):
         return os.environ.get("CHATBOT_OPENAI_API_KEY")