Fix: chat_history addition

agents/router_agent.py

@@ -1,5 +1,3 @@
-# agents/router_agent.py
-
 import re
 from .academic_agent import AcademicAgent
 from .drug_info_agent import DrugInfoAgent
@@ -9,9 +7,6 @@ from .viva_agent import VivaAgent
 
 class RouterAgent:
     def __init__(self, gemini_model=None):
-        """
-        Initializes the router and all specialist agents.
-        """
         self.academic_agent = AcademicAgent(gemini_model)
         self.drug_info_agent = DrugInfoAgent(gemini_model)
         self.mnemonic_agent = MnemonicAgent(gemini_model)
@@ -20,14 +15,13 @@ class RouterAgent:
 
     def route_query(self, query: str, file_context: str, viva_state: dict, chat_history: list):
         """
-        Determines the user's intent and routes the query with all necessary
-        context to the correct specialist agent.
+        Determines user intent and correctly routes the query with all
+        necessary context (file_context, chat_history, etc.) to the
+        correct specialist agent.
         """
         query_lower = query.lower()
 
-        # --- Intent Detection Logic ---
-
-        # 1. Viva Agent (High priority for session commands)
+        # 1. Viva Agent (High priority)
         if viva_state and viva_state.get('active'):
             return self.viva_agent.process_query(query, file_context, viva_state)
         if any(cmd in query_lower for cmd in ["viva", "interview", "start viva"]):
@@ -42,11 +36,8 @@ class RouterAgent:
             return self.quiz_agent.process_query(query, file_context, chat_history)
 
         # 4. Drug Info Agent
-        drug_keywords = ["drug", "medicine", "medication", "side effect", "dosage"]
-        specific_drugs = r'\b(paracetamol|ibuprofen|metformin|aspirin|amoxicillin)\b'
-        
-        if any(cmd in query_lower for cmd in drug_keywords) or re.search(specific_drugs, query_lower):
-             return self.drug_info_agent.process_query(query, file_context, chat_history)
+        if any(cmd in query_lower for cmd in ["drug", "medicine", "medication", "side effect", "dosage"]):
+            return self.drug_info_agent.process_query(query, file_context, chat_history)
 
-        # 5. Default to Academic Agent for general queries
+        # 5. Default to Academic Agent
         return self.academic_agent.process_query(query, file_context, chat_history)