fixed indentation

app.py

@@ -6,6 +6,7 @@ from tools.find_flight import FlightsFinderTool
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 import yaml
+from tool.mock_tools import MoodToNeedTool, NeedToDestinationTool, WeatherTool, FlightsFinderTool, FinalAnswerTool
 
 # Initialize Claude model via Hugging Face
 model = HfApiModel(
@@ -19,14 +20,28 @@ with open("prompts.yaml", "r") as f:
     prompt_templates = yaml.safe_load(f)
 
 # Define the agent with all tools
+# agent = CodeAgent(
+#     model=model,
+#     tools=[
+#         MoodToNeedTool(),          # Step 1: Mood → Need
+#         NeedToDestinationTool(),   # Step 2: Need → Destination
+#         WeatherTool(),             # Step 3: Weather for destination
+#         FlightsFinderTool(),       # Step 4: Destination → Flights           # Step 5: Claude wrap
+#         FinalAnswerTool()       # Required final output
+#     ],
+#     max_steps=6,
+#     verbosity_level=1,
+#     prompt_templates=prompt_templates
+# )
+
 agent = CodeAgent(
     model=model,
     tools=[
         MoodToNeedTool(),          # Step 1: Mood → Need
         NeedToDestinationTool(),   # Step 2: Need → Destination
         WeatherTool(),             # Step 3: Weather for destination
-        FlightsFinderTool(),       # Step 4: Destination → Flights           # Step 5: Claude wrap
-        FinalAnswerTool()       # Required final output
+        FlightsFinderTool(),       # Step 4: Destination → Flights
+        FinalAnswerTool()          # Step 5: Claude wrap
     ],
     max_steps=6,
     verbosity_level=1,

prompts.yaml

@@ -67,7 +67,6 @@ planning:
     8. Call final_answer() with all outputs.
     <end_plan>
 
-
   update_facts_pre_messages: |-
     ### 1. Facts given in the task
     ### 2. Facts that we have learned

tools/mock_tools.py

@@ -0,0 +1,81 @@
+from smolagents.tools import tool
+
+@tool
+def mood_to_need(mood: str) -> str:
+    """
+    Map a mood to a need.
+
+    Args:
+        mood (str): The current emotional state of the user.
+
+    Returns:
+        str: A description of what the user needs based on the mood.
+    """
+    return "You need relaxation and nature."
+
+@tool
+def need_to_destination(need: str) -> str:
+    """
+    Map a need to a destination.
+
+    Args:
+        need (str): The user's identified need (e.g., relaxation, adventure).
+
+    Returns:
+        str: A suggested destination that fulfills the need.
+    """
+    return "Bali, Indonesia"
+
+@tool
+def get_weather(dest: str) -> str:
+    """
+    Get weather forecast for a destination.
+
+    Args:
+        dest (str): The destination location.
+
+    Returns:
+        str: A weather forecast for the given destination.
+    """
+    return "Sunny and 28°C"
+
+@tool
+def get_flights(dest: str) -> str:
+    """
+    Get flight options for a destination.
+
+    Args:
+        dest (str): The destination location.
+
+    Returns:
+        str: A list of flight options for the destination.
+    """
+    return "Flight from Paris to Bali: €600 roundtrip"
+
+@tool
+def final_wrap(info: str) -> str:
+    """
+    Create a final wrap-up message.
+
+    Args:
+        info (str): Summary information about the destination and travel.
+
+    Returns:
+        str: A personalized wrap-up message.
+    """
+    return f"Bali sounds like a perfect place for relaxation with great weather and affordable flights!"
+
+@tool
+def final_answer_tool(answer: str) -> str:
+    """
+    Provides a final answer to the user.
+
+    Args:
+        answer (str): The final recommendation or conclusion.
+
+    Returns:
+        str: The same final answer.
+    """
+    return answer
+    
+