Matthew Franglen commited on
Commit
330e164
·
1 Parent(s): 583c32a

Rename weather tool to match repeated failed invocations

Browse files
Files changed (2) hide show
  1. src/agent.py +2 -2
  2. src/tools/weather.py +1 -1
src/agent.py CHANGED
@@ -5,7 +5,7 @@ from src.paths import PROMPT_FILE
5
  from src.tools.final_answer import FinalAnswerTool
6
  from src.tools.image_generation import image_generation_tool
7
  from src.tools.summarizer import prepare_picture_description
8
- from src.tools.weather import get_current_weather_in_location
9
 
10
 
11
  def get_agent() -> CodeAgent:
@@ -27,7 +27,7 @@ def get_agent() -> CodeAgent:
27
  agent = CodeAgent(
28
  model=model,
29
  tools=[
30
- get_current_weather_in_location,
31
  prepare_picture_description,
32
  image_generation_tool,
33
  final_answer,
 
5
  from src.tools.final_answer import FinalAnswerTool
6
  from src.tools.image_generation import image_generation_tool
7
  from src.tools.summarizer import prepare_picture_description
8
+ from src.tools.weather import get_current_weather
9
 
10
 
11
  def get_agent() -> CodeAgent:
 
27
  agent = CodeAgent(
28
  model=model,
29
  tools=[
30
+ get_current_weather,
31
  prepare_picture_description,
32
  image_generation_tool,
33
  final_answer,
src/tools/weather.py CHANGED
@@ -7,7 +7,7 @@ from src.tools.models import GeocodedLocation, LocationWeather
7
 
8
 
9
  @tool
10
- def get_current_weather_in_location(location: str) -> str:
11
  """A tool that fetches the current weather for a specified location.
12
  You should use this to make your picture of that location more accurate by including the current weather.
13
  Args:
 
7
 
8
 
9
  @tool
10
+ def get_current_weather(location: str) -> str:
11
  """A tool that fetches the current weather for a specified location.
12
  You should use this to make your picture of that location more accurate by including the current weather.
13
  Args: