Olympics-AI-Agent / trip_agents.py
menikev's picture
Upload 3 files
08a9a1d verified
raw
history blame contribute delete
No virus
1.73 kB
from crewai import Agent
from langchain.llms import OpenAI
from tools.browser_tools import BrowserTools
from tools.calculator_tools import CalculatorTools
from tools.search_tools import SearchTools
class TripAgents():
def city_selection_agent(self):
return Agent(
role='City Selection Expert',
goal='Select the best city based on weather, season, and prices',
backstory=
'An expert in analyzing travel data to pick ideal destinations',
# tools=[
# SearchTools.search_internet,
# BrowserTools.scrape_and_summarize_website,
# ],
verbose=True,
allow_delegation=False)
def local_expert(self):
return 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=[
# SearchTools.search_internet,
# BrowserTools.scrape_and_summarize_website,
# ],
verbose=True,
allow_delegation=False)
def travel_concierge(self):
return 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=[
# # SearchTools.search_internet,
# # BrowserTools.scrape_and_summarize_website,
# # CalculatorTools.calculate,
# ],
verbose=True,
allow_delegation=False)