wehub-resource-sync
35c9fb2445
CI Pipeline / code-quality (push) Waiting to run
CI Pipeline / test (macos-latest, 3.10) (push) Blocked by required conditions
CI Pipeline / test (macos-latest, 3.11) (push) Blocked by required conditions
CI Pipeline / test (macos-latest, 3.12) (push) Blocked by required conditions
CI Pipeline / test (macos-latest, 3.13) (push) Blocked by required conditions
CI Pipeline / test (ubuntu-latest, 3.10) (push) Blocked by required conditions
CI Pipeline / test (ubuntu-latest, 3.11) (push) Blocked by required conditions
CI Pipeline / test (ubuntu-latest, 3.12) (push) Blocked by required conditions
CI Pipeline / test (ubuntu-latest, 3.13) (push) Blocked by required conditions
CI Pipeline / test (windows-latest, 3.10) (push) Blocked by required conditions
CI Pipeline / test (windows-latest, 3.11) (push) Blocked by required conditions
CI Pipeline / test (windows-latest, 3.12) (push) Blocked by required conditions
CI Pipeline / test (windows-latest, 3.13) (push) Blocked by required conditions
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
from tools import (
|
|
llm_call,
|
|
weather_tool,
|
|
currency_converter_tool,
|
|
flight_price_estimator_tool,
|
|
)
|
|
from ragaai_catalyst import trace_agent, current_span
|
|
|
|
class ItineraryAgent:
|
|
def __init__(self, persona="Itinerary Agent"):
|
|
self.persona = persona
|
|
|
|
@trace_agent(name="plan_itinerary", agent_type="travel_planner", version="1.0.0")
|
|
def plan_itinerary(self, user_preferences, duration=3):
|
|
# Add metrics for the planning process
|
|
current_span().add_metrics(
|
|
name="itinerary_planning",
|
|
score=0.8,
|
|
reasoning="Planning comprehensive travel itinerary",
|
|
cost=0.01,
|
|
latency=0.5,
|
|
)
|
|
|
|
# Get weather information
|
|
weather = weather_tool(user_preferences["destination"])
|
|
|
|
# Get currency conversion if needed
|
|
if "budget_currency" in user_preferences and user_preferences["budget_currency"] != "USD":
|
|
budget = currency_converter_tool(
|
|
user_preferences["budget"], user_preferences["budget_currency"], "USD"
|
|
)
|
|
else:
|
|
budget = user_preferences["budget"]
|
|
|
|
# Get flight price estimation
|
|
flight_price = flight_price_estimator_tool(
|
|
user_preferences["origin"], user_preferences["destination"]
|
|
)
|
|
|
|
# Prepare prompt for the LLM
|
|
prompt = f"""As a {self.persona}, create a {duration}-day itinerary for a trip to {user_preferences['destination']}.
|
|
Weather: {weather}
|
|
Budget: ${budget}
|
|
Flight Price: {flight_price}
|
|
Preferences: {user_preferences.get('preferences', 'No specific preferences')}
|
|
|
|
Please provide a detailed day-by-day itinerary."""
|
|
|
|
# Generate itinerary using LLM
|
|
return llm_call(prompt)
|