fix vector db error

Author: Md-Emon-Hasan

.github/workflows/ci-cd.yml

@@ -2,9 +2,9 @@ name: CI/CD Pipeline
 
 on:
   push:
-    branches: [ main ]
+    branches: [ master ]
   pull_request:
-    branches: [ main ]
+    branches: [ master ]
 
 jobs:
   backend-test:

.gitignore

@@ -3,4 +3,4 @@ frontend/dist/
 node_modules/
 __pycache__/
 *.pyc
-.pytest_cache/
+.pytest_cache/

backend/.coverage

@@ -40,13 +40,21 @@ def ExecutorAgent(state: AgentState) -> AgentState:
             f"Medical Information:\n{content}\n\n"
             "Provide a clear, caring response in 2-4 sentences. Be professional and reassuring."
         )
-        response = llm.invoke(prompt)
-        answer = (
-            response.content.strip()
-            if hasattr(response, "content")
-            else str(response).strip()
-        )
-        logger.info("Executor: Generated response from documents")
+        try:
+            response = llm.invoke(prompt)
+            answer = (
+                response.content.strip()
+                if hasattr(response, "content")
+                else str(response).strip()
+            )
+            logger.info("Executor: Generated response from documents")
+        except Exception as e:
+            logger.error("Executor: LLM generation failed: %s", str(e))
+            answer = (
+                "I understand your concern about your symptoms. For accurate medical advice, "
+                "please consult with a healthcare professional who can properly evaluate your condition."
+            )
+            source_info = "System Message"
 
     elif state.get("llm_success") and state.get("generation"):
         answer = state["generation"]

backend/app/agents/executor.py

@@ -14,6 +14,7 @@ def LLMAgent(state: AgentState) -> AgentState:
     if not llm:
         state["llm_success"] = False
         state["llm_attempted"] = True
+        state["generation"] = "Medical AI service is temporarily unavailable."
         return state
 
     # Build conversation context

backend/app/agents/llm_agent.py

@@ -20,7 +20,11 @@ def TavilyAgent(state: AgentState) -> AgentState:
         return state
 
     search_query = f"{state['question']} medical health treatment symptoms"
-    results = tavily.invoke(search_query)
+    try:
+        results = tavily.invoke(search_query)
+    except Exception as e:
+        logger.error("Tavily: Search failed: %s", str(e))
+        results = []
 
     valid_results = [
         r

backend/app/agents/tavily.py

@@ -5,6 +5,7 @@
 
 import logging
 import os
+import sys
 from datetime import datetime
 from logging.handlers import RotatingFileHandler
 
@@ -13,21 +14,32 @@
 
 def setup_logging(log_dir: str = LOG_DIR) -> logging.Logger:
     """Setup rotating file + console logging. Returns the 'medigenius' logger."""
-    if not os.path.exists(log_dir):
-        os.makedirs(log_dir)
-
     _logger = logging.getLogger("medigenius")
-    _logger.setLevel(logging.INFO)
 
     # Avoid duplicate handlers on re-import
     if _logger.handlers:
         return _logger
 
-    log_file = os.path.join(
-        log_dir, f"medigenius_{datetime.now().strftime('%Y%m%d')}.log"
-    )
-    file_handler = RotatingFileHandler(log_file, maxBytes=10 * 1024 * 1024, backupCount=5)
-    file_handler.setLevel(logging.INFO)
+    # Skip file logging if in test environment
+    is_testing = "pytest" in sys.modules or os.getenv("TESTING") == "1"
+
+    if is_testing:
+        _logger.setLevel(logging.DEBUG)
+    else:
+        _logger.setLevel(logging.INFO)
+        if not os.path.exists(log_dir):
+            os.makedirs(log_dir)
+
+        log_file = os.path.join(
+            log_dir, f"medigenius_{datetime.now().strftime('%Y%m%d')}.log"
+        )
+        file_handler = RotatingFileHandler(log_file, maxBytes=10 * 1024 * 1024, backupCount=5)
+        file_handler.setLevel(logging.INFO)
+        file_handler.setFormatter(logging.Formatter(
+            "%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+            datefmt="%Y-%m-%d %H:%M:%S",
+        ))
+        _logger.addHandler(file_handler)
 
     console_handler = logging.StreamHandler()
     console_handler.setLevel(logging.INFO)
@@ -36,11 +48,9 @@ def setup_logging(log_dir: str = LOG_DIR) -> logging.Logger:
         "%(asctime)s - %(name)s - %(levelname)s - %(message)s",
         datefmt="%Y-%m-%d %H:%M:%S",
     )
-    file_handler.setFormatter(formatter)
     console_handler.setFormatter(formatter)
-
-    _logger.addHandler(file_handler)
     _logger.addHandler(console_handler)
+
     return _logger
 
 

backend/app/core/logging_config.py

@@ -4,7 +4,7 @@
 """
 
 from datetime import datetime
-from typing import Dict, Optional
+from typing import Dict
 
 from sqlalchemy import Column, DateTime, Integer, String, Text
 from sqlalchemy.orm import declarative_base

backend/app/models/message.py

@@ -6,7 +6,11 @@
 from app.tools.llm_client import get_llm
 from app.tools.pdf_loader import load_pdf, process_pdf, split_documents
 from app.tools.tavily_search import get_tavily_search
-from app.tools.vector_store import get_embeddings, get_or_create_vectorstore, get_retriever
+from app.tools.vector_store import (
+    get_embeddings,
+    get_or_create_vectorstore,
+    get_retriever,
+)
 from app.tools.wikipedia_search import get_wikipedia_wrapper
 
 __all__ = [