Merge branch 'main' of https://huggingface.co/spaces/AIAgentsTravel/AI_Agent_Hackathon

app.py

@@ -9,6 +9,7 @@ from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 from smolagents import MultiStepAgent, ActionStep, AgentText, AgentImage, AgentAudio, handle_agent_output_types
 from Gradio_UI import GradioUI
 import yaml
+# from tools.mock_tools import MoodToNeedTool, NeedToDestinationTool, WeatherTool, FlightsFinderTool, FinalAnswerTool
 
 # Initialize Claude model via Hugging Face
 model = HfApiModel(
@@ -22,6 +23,20 @@ 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=[

prompts.yaml

@@ -93,15 +93,6 @@ planning:
     8. Call final_answer() with all outputs.
     <end_plan>
 
-evaluate_destination_weather_and_country:
-  system: |-
-    You are a travel advisor who evaluates whether the destination is suitable for the user based on the weather and country information. Consider the weather forecast, the suggested activity, and relevant country details (e.g., safety, accessibility, cultural factors).
-  prompt: |-
-    The destination is {city}, {country}, and the suggested activity is: {activity}.
-    The weather forecast is: {weather}.
-    Relevant country information: {country_info}.
-    Based on this information, is this destination a good idea for the activity? Answer with "Yes" or "No" and provide a short explanation.
-
   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
+    
+