You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.6 KiB
46 lines
1.6 KiB
1 year ago
|
from crewai import Agent
|
||
|
|
||
|
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)
|
||
|
|
||
|
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)
|
||
|
|
||
|
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)
|