Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add expected_output to trip_tasks.py #71

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 18 additions & 12 deletions trip_planner/trip_tasks.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,16 +14,18 @@ def identify_task(self, agent, origin, cities, interests, range):
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.
{self.__tip_section()}

Traveling from: {origin}
City Options: {cities}
Trip Date: {range}
Traveler Interests: {interests}
"""),
expected_output=dedent("""
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.
"""),
agent=agent)

Expand All @@ -41,14 +43,16 @@ def gather_task(self, agent, origin, interests, range):
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.
{self.__tip_section()}

Trip Date: {range}
Traveling from: {origin}
Traveler Interests: {interests}
"""),
expected_output=dedent("""
The final answer must be a comprehensive city guide,
rich in cultural insights and practical tips,
tailored to enhance the travel experience.
"""),
agent=agent)

Expand All @@ -65,19 +69,21 @@ def plan_task(self, agent, origin, interests, range):
This itinerary should cover all aspects of the trip,
from arrival to departure, integrating the city guide
information with practical travel logistics.


Trip Date: {range}
Traveling from: {origin}
Traveler Interests: {interests}
"""),
expected_output=dedent(f"""
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
# up each place, what make them special! {self.__tip_section()}

Trip Date: {range}
Traveling from: {origin}
Traveler Interests: {interests}
"""),
agent=agent)


def __tip_section(self):
return "If you do your BEST WORK, I'll tip you $100!"