Spaces:
Sleeping
Sleeping
File size: 1,213 Bytes
eee5f58 |
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 |
import os, datetime
from langchain.tools import tool
@tool
def save_plan(plan: str, location: str)-> str:
"""
Save the travel plan to a file with a timestamp.
Args:
plan (str): The travel plan to save.
location (str): The location for which the plan is created.
Returns:
str: Confirmation message with the filename.
"""
os.makedirs("plans", exist_ok=True)
timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
filename = f"plans/{location.replace(' ', '_').lower()}_{timestamp}.txt"
with open(filename, "w", encoding='utf-8') as file:
file.write(plan)
return f'Plan saved as {filename}'
@tool
def suggest_trip(city:str):
"""
Suggest a trip itinerary for a specific city.
Args:
city (str): The city for which to suggest a trip.
Returns:
str: A brief itinerary suggestion.
"""
return f"""
Top 5 things to do in {city}:
1. Visit historical sites
2. Try traditional cuisine
3. Explore museums or art districts
4. Go hiking or sightseeing
5. Experience local nightlife or music
"""
tools = [
save_plan,
suggest_trip,
] |