ChaosBot commited on
Commit
2ce3bdc
·
verified ·
1 Parent(s): ae7a494

add some tools

Browse files
Files changed (1) hide show
  1. app.py +17 -8
app.py CHANGED
@@ -7,16 +7,25 @@ from tools.final_answer import FinalAnswerTool
7
 
8
  from Gradio_UI import GradioUI
9
 
10
- # Below is an example of a tool that does nothing. Amaze us with your creativity !
11
  @tool
12
- def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
13
- #Keep this format for the description / args / args description but feel free to modify the tool
14
- """A tool that does nothing yet
15
  Args:
16
- arg1: the first argument
17
- arg2: the second argument
 
 
18
  """
19
- return "What magic will you build ?"
 
 
 
 
 
 
 
 
20
 
21
  @tool
22
  def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +64,7 @@ with open("prompts.yaml", 'r') as stream:
55
 
56
  agent = CodeAgent(
57
  model=model,
58
- tools=[final_answer], ## add your tools here (don't remove final answer)
59
  max_steps=6,
60
  verbosity_level=1,
61
  grammar=None,
 
7
 
8
  from Gradio_UI import GradioUI
9
 
 
10
  @tool
11
+ def get_weather(city: str) -> str:
12
+ """A tool that fetches the current weather information for a specified city.
13
+
14
  Args:
15
+ city (str): The name of the city to fetch the weather for (e.g., 'New York').
16
+
17
+ Returns:
18
+ str: A string containing the weather information or an error message.
19
  """
20
+ api_url = f"https://api.weather.com/v1/location/{city}?apiKey=5c241d89f91274015a577e3e17d43370"
21
+ response = requests.get(api_url)
22
+
23
+ if response.status_code == 200:
24
+ data = response.json()
25
+ return data.get("weather", "No weather information available")
26
+ else:
27
+ return "Error: Unable to fetch weather data."
28
+
29
 
30
  @tool
31
  def get_current_time_in_timezone(timezone: str) -> str:
 
64
 
65
  agent = CodeAgent(
66
  model=model,
67
+ tools=[final_answer, image_generation_tool, get_current_time_in_timezone, get_weather], ## add your tools here (don't remove final answer)
68
  max_steps=6,
69
  verbosity_level=1,
70
  grammar=None,