Lim search and subtask breaker

pipeline.py

@@ -21,7 +21,8 @@ from supervisor import (
     gemini_supervisor_rag_brainstorm, execute_medswin_task,
     gemini_supervisor_synthesize, gemini_supervisor_challenge,
     gemini_supervisor_enhance_answer, gemini_supervisor_check_clarity,
-    gemini_clinical_intake_triage, gemini_summarize_clinical_insights
+    gemini_clinical_intake_triage, gemini_summarize_clinical_insights,
+    MAX_SEARCH_STRATEGIES
 )
 
 MAX_CLINICAL_QA_ROUNDS = 5
@@ -463,7 +464,7 @@ def stream_chat(
         
         all_search_results = []
         strategy_jobs = []
-        for strategy in search_strategies.get("search_strategies", [])[:4]:
+        for strategy in search_strategies.get("search_strategies", [])[:MAX_SEARCH_STRATEGIES]:
             search_query = strategy.get("strategy", message)
             target_sources = strategy.get("target_sources", 2)
             strategy_jobs.append({

supervisor.py

@@ -8,6 +8,11 @@ from client import MCP_AVAILABLE, call_agent
 from config import GEMINI_MODEL, GEMINI_MODEL_LITE
 from utils import format_prompt_manually
 
+# Maximum number of subtasks for query breakdown
+MAX_SUBTASKS = 5
+# Maximum number of search strategies
+MAX_SEARCH_STRATEGIES = 3
+
 try:
     import nest_asyncio
 except ImportError:
@@ -28,7 +33,7 @@ async def gemini_supervisor_breakdown_async(query: str, use_rag: bool, use_web_s
     
     estimated_time_per_task = 8
     max_topics_by_time = max(2, int((remaining_time - 20) / estimated_time_per_task))
-    max_topics = min(max_topics_by_time, 10)
+    max_topics = min(max_topics_by_time, MAX_SUBTASKS)
     
     prompt = f"""You are a supervisor agent coordinating with a MedSwin medical specialist model.
 Break the following medical query into focused sub-topics that MedSwin can answer sequentially.
@@ -110,9 +115,9 @@ Guidelines:
 
 
 async def gemini_supervisor_search_strategies_async(query: str, time_elapsed: float) -> dict:
-    """Gemini Supervisor: In search mode, break query into 1-4 searching strategies"""
+    """Gemini Supervisor: In search mode, break query into searching strategies"""
     prompt = f"""You are supervising web search for a medical query.
-Break this query into 1-4 focused search strategies (each targeting 1-2 sources).
+Break this query into 1-{MAX_SEARCH_STRATEGIES} focused search strategies (each targeting 1-2 sources).
 
 Query: "{query}"
 
@@ -127,7 +132,7 @@ Return ONLY valid JSON:
     }},
     ...
   ],
-  "max_strategies": 4
+  "max_strategies": {MAX_SEARCH_STRATEGIES}
 }}
 
 Keep strategies focused and avoid overlap."""