mirror of
https://github.com/danielmiessler/fabric
synced 2024-11-10 07:10:31 +00:00
46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
|
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)
|