File size: 7,222 Bytes
a9923c5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 |
from crewai import Agent, Task, Crew, Process
from textwrap import dedent
from datetime import date
from config.TripPlanner import browser_tools, calculator_tools, search_tools
import os
#Model options
llm_models = [
"gemini/gemini-1.5-flash",
"gemini/gemini-1.5-pro",
"gemini/gemini-pro"
]
selected_model = llm_models[0]
def configure_api_keys(gemini_api_key):
if not gemini_api_key:
raise ValueError("Gemini API key is required")
os.environ['GEMINI_API_KEY'] = gemini_api_key
def set_model(selected_model_name):
global selected_model
selected_model = selected_model_name
def run_crew_tp(gemini_api_key, origin, cities, trip_date, interests):
try:
# Configure API keys (no search tool used)
configure_api_keys(gemini_api_key)
# Agents
city_selection_agent = Agent(
role='City Selection Expert',
goal='Select the best city based on weather, season, and prices',
backstory='An expert in analyzing travel data to pic idea; destinations',
tools=[
search_tools.SearchTools.search_internet, browser_tools.
BrowserTools.scrape_and_summarize_website
],
llm=selected_model,
verbose=True)
local_expert = Agent(
role='Local Expert at this city',
goal='Provide the BEST insights about the selected city',
backstory="""A knowledgeable local guide with extensive information
about the city, it's attractions and customs""",
tools=[
search_tools.SearchTools.search_internet,
browser_tools.BrowserTools.scrape_and_summarize_website,
],
llm=selected_model,
verbose=True)
travel_concierge = Agent(
role='Amazing Travel Concierge',
goal="""Create the most amazing travel itineraries with budget and
packing suggestions for the city""",
backstory="""Specialist in travel planning and logistics with
decades of experience""",
tools=[
search_tools.SearchTools.search_internet,
browser_tools.BrowserTools.scrape_and_summarize_website,
calculator_tools.CalculatorTools.calculate,
],
llm=selected_model,
verbose=True)
#Tasks
identity_task = Task(
description=dedent(f"""
Analyze and select the best city for the trip based
on specific criteria such as weather patterns, seasonal
events, and travel costs. This task involves comparing
multiple cities, considering factors like current weather
conditions, upcoming cultural or seasonal events, and
overall travel expenses.
Your final answer must be a detailed
report on the chosen city, and everything you found out
about it, including the actual flight costs, weather
forecast and attractions.
Traveling from: {origin}
City Options: {cities}
Trip Date: {trip_date}
Traveler Interests: {interests}
"""),
agent=city_selection_agent,
expected_output="Detailed report on the chosen city including flight costs, weather forecast, and attractions"
)
gather_task = Task(
description=dedent(f"""
As a local expert on this city you must compile an
in-depth guide for someone traveling there and wanting
to have THE BEST trip ever!
Gather information about key attractions, local customs,
special events, and daily activity recommendations.
Find the best spots to go to, the kind of place only a
local would know.
This guide should provide a thorough overview of what
the city has to offer, including hidden gems, cultural
hotspots, must-visit landmarks, weather forecasts, and
high level costs.
The final answer must be a comprehensive city guide,
rich in cultural insights and practical tips,
tailored to enhance the travel experience.
Trip Date: {trip_date}
Traveling from: {origin}
Traveler Interests: {interests}
"""),
agent=local_expert,
expected_output="Comprehensive city guide including hidden gems, cultural hotspots, and practical travel tips"
)
plan_task = Task(
description=dedent(f"""
Expand this guide into a full 7-day travel
itinerary with detailed per-day plans, including
weather forecasts, places to eat, packing suggestions,
and a budget breakdown.
You MUST suggest actual places to visit, actual hotels
to stay and actual restaurants to go to.
This itinerary should cover all aspects of the trip,
from arrival to departure, integrating the city guide
information with practical travel logistics.
Your final answer MUST be a complete expanded travel plan,
formatted as markdown, encompassing a daily schedule,
anticipated weather conditions, recommended clothing and
items to pack, and a detailed budget, ensuring THE BEST
TRIP EVER. Be specific and give it a reason why you picked
each place, what makes them special!
Trip Date: {trip_date}
Traveling from: {origin}
Traveler Interests: {interests}
"""),
agent=travel_concierge,
expected_output="Complete expanded travel plan with daily schedule, weather conditions, packing suggestions, and budget breakdown",
output_file = 'trip_planner.md'
)
crew = Crew(
agents=[city_selection_agent, local_expert, travel_concierge],
tasks=[identity_task, gather_task, plan_task],
process=Process.sequential,
output_log_file=True,
verbose=True
)
crew.kickoff(inputs={
'origin': origin,
'cities': cities,
'trip_date': trip_date,
'interests': interests
})
with open("trip_planner.md", "r", encoding='utf-8') as f:
plan = f.read()
with open("logs.txt", 'r', encoding='utf-8') as f:
logs = f.read()
with open("logs.txt", 'w', encoding='utf-8') as f:
f.truncate(0)
return plan, logs
except Exception as e:
return f"Error: {str(e)}", str(e)
|