Update chat/rag_engine.py

chat/rag_engine.py

@@ -336,6 +336,17 @@ class GuidelineDetector:
             'coverage_percentage': round(coverage_percentage, 1) if guidelines_to_check else 0
         }
 
+# ADD THIS AT THE VERY TOP OF rag_engine.py (before any other imports)
+import sys
+import os
+
+# Add the project root to Python path
+current_dir = os.path.dirname(os.path.abspath(__file__))
+parent_dir = os.path.dirname(current_dir)  # This goes from chat/ to MedSearchPro/
+
+if parent_dir not in sys.path:
+    sys.path.insert(0, parent_dir)
+
 
 # Import existing modules
 try:
@@ -345,7 +356,7 @@ try:
     from chat.comparator import CrossPaperComparator
     from chat.gap_analyzer import ResearchGapAnalyzer
     from lib.memory_manager import ConversationMemory
-    from llm.llm_provider import XAIGrokProvider
+    from llm.llm_provider import XAIGrokProvider, GrokLLM
 except ImportError as e:
     print(f"⚠️  Some dependencies not found - using simplified mode: {e}")