add tyty files

tools/mood_to_need.py

@@ -1,6 +1,6 @@
-"""Tool to map a user's mood to a vacation need."""
-
 from smolagents.tools import Tool
+from anthropic import Anthropic, HUMAN_PROMPT, AI_PROMPT
+import os
 
 class MoodToNeedTool(Tool):
     """
@@ -22,6 +22,7 @@ class MoodToNeedTool(Tool):
         Args:
             model: A callable language model with a __call__(str) -> str interface.
         """
+        super().__init__()
         self.model = model
 
     def forward(self, mood: str) -> str:
@@ -44,4 +45,17 @@ class MoodToNeedTool(Tool):
             f'Need:'
         )
         response = self.model(prompt)
-        return response.strip()
+        return response.strip()
+
+client = Anthropic(api_key=os.environ["ANTHROPIC_KEY"])
+
+def claude_mood_to_need_model(prompt: str) -> str:
+    message = client.messages.create(
+        model="claude-3-opus-20240229",
+        max_tokens=1024,
+        temperature=0.7,
+        messages=[
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return message.content[0].text

tools/need_to_destination.py

@@ -1,4 +1,6 @@
 from smolagents.tools import Tool
+from anthropic import Anthropic
+import os
 import json
 
 class NeedToDestinationTool(Tool):
@@ -16,32 +18,32 @@ class NeedToDestinationTool(Tool):
 
     def forward(self, need: str) -> list[dict]:
         prompt = f"""
-You are a travel agent AI.
+            You are a travel agent AI.
 
-Based on the user's need: "{need}",
-suggest 2-3 travel destinations with round-trip flight information.
+            Based on the user's need: "{need}",
+            suggest 2-3 travel destinations with round-trip flight information.
 
-Return the output as valid JSON in the following format:
+            Return the output as valid JSON in the following format:
 
-[
-  {{
-    "destination": "DestinationName",
-    "departure": {{
-      "date": "YYYY-MM-DD",
-      "from_airport": "{self.departure_airport}",
-      "to_airport": "XXX"
-    }},
-    "return": {{
-      "date": "YYYY-MM-DD",
-      "from_airport": "XXX",
-      "to_airport": "{self.departure_airport}"
-    }}
-  }},
-  ...
-]
+            [
+            {{
+                "destination": "DestinationName",
+                "departure": {{
+                "date": "YYYY-MM-DD",
+                "from_airport": "{self.departure_airport}",
+                "to_airport": "XXX"
+                }},
+                "return": {{
+                "date": "YYYY-MM-DD",
+                "from_airport": "XXX",
+                "to_airport": "{self.departure_airport}"
+                }}
+            }},
+            ...
+            ]
 
-DO NOT add explanations, only return raw JSON.
-"""
+            DO NOT add explanations, only return raw JSON.
+            """
         result = self.model(prompt)
         try:
             destinations = json.loads(result.strip())
@@ -50,42 +52,15 @@ DO NOT add explanations, only return raw JSON.
 
         return destinations
     
-def mock_model(prompt: str) -> str:
-    return """
-    [
-      {
-        "destination": "Nice",
-        "departure": {
-          "date": "2025-07-01",
-          "from_airport": "CDG",
-          "to_airport": "NCE"
-        },
-        "return": {
-          "date": "2025-07-07",
-          "from_airport": "NCE",
-          "to_airport": "CDG"
-        }
-      },
-      {
-        "destination": "Barcelona",
-        "departure": {
-          "date": "2025-07-02",
-          "from_airport": "CDG",
-          "to_airport": "BCN"
-        },
-        "return": {
-          "date": "2025-07-08",
-          "from_airport": "BCN",
-          "to_airport": "CDG"
-        }
-      }
-    ]
-    """
+client = Anthropic(api_key=os.environ["ANTHROPIC_KEY"])
 
-if __name__ == "__main__":
-    tool = NeedToDestinationTool(model=mock_model)
-    test_need = "A relaxing beach holiday"
-    destinations = tool(need=test_need)
-    print("Destinations returned by the tool:")
-    for dest in destinations:
-        print(dest)
+def claude_need_to_destination_model(prompt: str) -> str:
+    message = client.messages.create(
+        model="claude-3-opus-20240229",
+        max_tokens=1024,
+        temperature=0.7,
+        messages=[
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return message.content[0].text

tools/test_mood_to_destination.py

@@ -0,0 +1,21 @@
+from tools.mood_to_need import MoodToNeedTool, claude_mood_to_need_model
+from tools.need_to_destination import NeedToDestinationTool, claude_need_to_destination_model
+import json
+
+mood_tool = MoodToNeedTool(model=claude_mood_to_need_model)
+mood = "I'm feeling stuck in a routine and need change"
+# mood = "I feel overwhelmed and burned out."
+# mood = "I just got out of a break up"
+
+need = mood_tool(mood=mood)
+print(f"Mood: {mood}")
+print("Need:", need)
+
+destination_tool = NeedToDestinationTool(model=claude_need_to_destination_model, departure_airport="CDG")
+try:
+    destinations = destination_tool(need=need)
+    print("\n→ Suggested Destinations:")
+    for dest in destinations:
+        print(json.dumps(dest, indent=2))
+except ValueError as e:
+    print(f"Error parsing Claude output: {e}")