Update sub_agents.py

sub_agents.py

@@ -1,5 +1,5 @@
 from deepagents.middleware.subagents import SubAgent
-from tools import route_tool, cost_tool, traffic_tool, weather_tool
+from tools import route_tool, cost_tool, traffic_tool, weather_tool,forecast_weather_tool, multi_route_tool
 
 route_agent = SubAgent(
     name="RealRouteAgent",
@@ -20,6 +20,29 @@ route_agent = SubAgent(
     tools=[route_tool],
 )
 
+multi_route_agent = SubAgent(
+    name="MultiRouteAgent",
+    description="Plans optimal route visiting multiple destinations when origin and list of destinations are provided",
+    system_prompt="""
+    You are a multi-destination route optimization expert.
+    
+    CRITICAL RULES:
+    - ONLY use 'multi_route_planner' if user provides:
+      * ONE origin location
+      * MULTIPLE destination locations (2 or more)
+    - If locations are missing, respond: "I need an origin and at least 2 destinations to plan a multi-stop route."
+    - DO NOT use this for single destination - use regular route planning instead
+    - Valid queries: "Route from Delhi to Bhopal, Chandigarh, and Rohtak"
+    
+    When all locations ARE provided:
+    - Find the optimal visiting order that minimizes total travel time
+    - Show total distance and duration
+    - Present step-by-step route segments
+    """,
+    tools=[multi_route_tool],
+)
+
+
 cost_agent = SubAgent(
     name="RealCostAgent",
     description="Calculates costs when origin, destination, distance, weight, and duration are provided",
@@ -59,22 +82,28 @@ traffic_agent = SubAgent(
     tools=[traffic_tool],
 )
 
-weather_agent  = SubAgent(
+weather_agent = SubAgent(
     name="RealWeatherAnalyzer",
-    description="Analyzes weather when both origin and destination are provided",
+    description="Analyzes current and forecasted weather when both origin and destination are provided",
     system_prompt="""
-    You are a weather analysis expert.
+    You are a weather analysis expert with forecasting capabilities.
 
     CRITICAL RULES:
     - ONLY use 'real_weather_analyzer' if user provides BOTH origin AND destination
+    - Use 'forecast_weather' when user asks for future weather predictions at a location
     - If locations are missing, respond: "I need both origin and destination to analyze weather. Please provide both locations."
     - DO NOT assume locations
     - DO NOT analyze weather if user only asks for cost calculations
     
-    When locations ARE provided, give insights on current weather conditions.
-    """
+    When locations ARE provided:
+    - Give insights on CURRENT weather conditions using real_weather_analyzer
+    - If user asks about future weather, also use forecast_weather for each location
+    - Combine current conditions with forecasted trends for better insights
+    """,
+    tools=[weather_tool, forecast_weather_tool]  
 )
 
+
 coordinator = SubAgent(
     name="Coordinator",
     description="Synthesizes information and handles user queries intelligently",
@@ -101,9 +130,20 @@ coordinator = SubAgent(
     4. IF user asks for WEATHER analysis:
        - Check if BOTH origin AND destination are provided
        - If missing, ask for them
-       - If provided, use weather agent
-    
-    5. IF user asks for COMPLETE delivery planning (route + cost + traffic + weather):
+       - If provided, use weather agent for CURRENT conditions
+       - If user asks about FUTURE weather (e.g., "will it rain tomorrow?", "forecast", "next 24 hours"):
+         * Also call forecast_weather for origin and destination
+         * Combine current conditions with forecast trends
+       - Present comprehensive weather analysis
+    
+    5. IF user asks for MULTI-DESTINATION route planning:
+        - Check if origin AND multiple destinations (2+) are provided
+        - If missing, ask for them
+        - If provided, use multi_route_agent
+        - Present optimal visiting order with time/distance breakdown
+        - Format response as shown in RESPONSE FORMATS section
+    
+    6. IF user asks for COMPLETE delivery planning (route + cost + traffic + weather):
        - Check if ALL required info is provided (origin, destination, weight)
        - If anything is missing, ask for it
        - If all provided, coordinate all agents