test

app.py

@@ -19,7 +19,7 @@ model = LiteLLMModel(
 )
 
 # Load prompt templates
-with open("prompts_new.yaml", "r") as f:
+with open("prompts_simple.yaml", "r") as f:
     prompt_templates = yaml.safe_load(f)
 
 # Define the agent with all tools

prompts_simple.yaml

@@ -0,0 +1,52 @@
+system_prompt: |-
+  You are WanderMind, a travel assistant. Follow these steps EXACTLY:
+
+  1. First, check if user provided: mood, origin city, travel dates
+  2. If missing info, ask for it and STOP
+  3. If all info provided, follow this EXACT sequence:
+
+  Step A: need = MoodToNeed(mood="user's exact mood")
+  Step B: destinations = NeedToDestination(need=need)  
+  Step C: weather = weather_forecast(location="first destination from list")
+  Step D: safety = country_info(country="country name", info_type="security")
+  Step E: flights = flights_finder(departure_airport="CDG", arrival_airport="destination airport", outbound_date="2025-06-15", return_date="2025-06-22")
+  Step F: final_answer("Complete travel recommendation with all details")
+
+  🚨 CRITICAL: 
+  - Use tools ONE AT A TIME
+  - Print each result before moving to next step
+  - Don't write complex Python logic
+  - Don't assume data formats
+  - Just call tools and show results
+
+  Example:
+  ```py
+  # Step 1: Get need
+  need = MoodToNeed(mood="happy")
+  print(f"Emotional need: {need}")
+  ```
+
+planning:
+  initial_plan: |-
+    1. Check user inputs (mood, origin, dates)
+    2. Call MoodToNeed(mood)
+    3. Call NeedToDestination(need)
+    4. Call weather_forecast(location)
+    5. Call country_info(country)
+    6. Call flights_finder(airports and dates)
+    7. Call final_answer(complete plan)
+
+managed_agent:
+  task: |-
+    You are a helpful sub-agent named '{{name}}'.
+    Your manager gives you this task: {{task}}
+    Return your result with final_answer().
+
+  report: |-
+    Final answer from agent '{{name}}': {{final_answer}}
+
+final_answer:
+  pre_messages: |-
+    Summarizing your travel plan:
+  post_messages: |-
+    Enjoy your journey! 

test_tools.py

@@ -0,0 +1,59 @@
+#!/usr/bin/env python3
+
+from tools.mood_to_need import MoodToNeedTool, claude_mood_to_need_model
+from tools.need_to_destination import NeedToDestinationTool, claude_need_to_destination_model
+from tools.weather_tool import WeatherTool
+from tools.country_info_tool import CountryInfoTool
+from tools.find_flight import FlightsFinderTool
+
+def test_tools():
+    print("🧪 Testing individual tools...\n")
+    
+    # Test 1: MoodToNeed
+    print("1️⃣ Testing MoodToNeed...")
+    mood_tool = MoodToNeedTool(model=claude_mood_to_need_model)
+    need = mood_tool.forward(mood="happy")
+    print(f"   Input: 'happy' → Output: '{need}'\n")
+    
+    # Test 2: NeedToDestination
+    print("2️⃣ Testing NeedToDestination...")
+    destination_tool = NeedToDestinationTool(model=claude_need_to_destination_model)
+    destinations = destination_tool.forward(need=need)
+    print(f"   Input: '{need}' → Output: {destinations}\n")
+    
+    # Test 3: WeatherTool
+    print("3️⃣ Testing WeatherTool...")
+    weather_tool = WeatherTool()
+    if destinations and len(destinations) > 0:
+        first_dest = destinations[0]["destination"]
+        weather = weather_tool.forward(location=first_dest)
+        print(f"   Input: '{first_dest}' → Output: {weather[:200]}...\n")
+    
+    # Test 4: CountryInfoTool
+    print("4️⃣ Testing CountryInfoTool...")
+    country_tool = CountryInfoTool()
+    if destinations and len(destinations) > 0:
+        # Extract country from destination
+        dest_parts = destinations[0]["destination"].split(", ")
+        if len(dest_parts) > 1:
+            country = dest_parts[1]
+            safety = country_tool.forward(country=country, info_type="security")
+            print(f"   Input: '{country}' → Output: {safety[:200]}...\n")
+    
+    # Test 5: FlightsFinderTool
+    print("5️⃣ Testing FlightsFinderTool...")
+    flights_tool = FlightsFinderTool()
+    if destinations and len(destinations) > 0:
+        dest_airport = destinations[0]["departure"]["to_airport"]
+        flights = flights_tool.forward(
+            departure_airport="CDG",
+            arrival_airport=dest_airport,
+            outbound_date="2025-06-15",
+            return_date="2025-06-22"
+        )
+        print(f"   Input: CDG → {dest_airport} → Output: {flights[:200]}...\n")
+    
+    print("✅ All tools tested!")
+
+if __name__ == "__main__":
+    test_tools()