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
+from tools import route_tool, cost_tool, traffic_tool, weather_tool
 
 route_agent = SubAgent(
     name="RealRouteAgent",
@@ -59,6 +59,22 @@ traffic_agent = SubAgent(
     tools=[traffic_tool],
 )
 
+weather_agent  = SubAgent(
+    name="RealWeatherAnalyzer",
+    description="Analyzes weather when both origin and destination are provided",
+    system_prompt="""
+    You are a weather analysis expert.
+
+    CRITICAL RULES:
+    - ONLY use 'real_weather_analyzer' if user provides BOTH origin AND destination
+    - 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.
+    """
+)
+
 coordinator = SubAgent(
     name="Coordinator",
     description="Synthesizes information and handles user queries intelligently",
@@ -66,11 +82,11 @@ coordinator = SubAgent(
     You are the delivery coordinator. Your job is to understand what the user wants and respond appropriately.
     
     CRITICAL DECISION LOGIC:
-    
+
     1. IF user asks for ROUTE planning:
        - Check if BOTH origin AND destination are provided
        - If missing, ask for them
-       - If provided, use route agent
+       - If provided, only use route agent
     
     2. IF user asks for COST calculation with specific numbers (distance, weight, duration):
        - Use cost agent DIRECTLY
@@ -82,7 +98,12 @@ coordinator = SubAgent(
        - If missing, ask for them
        - If provided, use traffic agent
     
-    4. IF user asks for COMPLETE delivery planning (route + cost + traffic):
+    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):
        - Check if ALL required info is provided (origin, destination, weight)
        - If anything is missing, ask for it
        - If all provided, coordinate all agents
@@ -92,13 +113,14 @@ coordinator = SubAgent(
     2. Route Summary (distance, duration)
     3. Cost Breakdown
     4. Traffic Conditions
-    5. Final Recommendations
+    5. Weather Conditions
+    6. Final Recommendations
     
     MAP LINK FORMATTING:
     - When you receive a map URL like "/view-map/route_X_Y.html", format it as:
-      "📍 VIEW INTERACTIVE MAP: [CLICK HERE](/view-map/route_X_Y.html)"
+      "VIEW INTERACTIVE MAP: [CLICK HERE](/view-map/route_X_Y.html)"
     - Place this at the TOP of your response
     
     DO NOT make assumptions. If information is missing, ask the user.
     """,
-)
+)