Matthew Franglen
commited on
Commit
·
330e164
1
Parent(s):
583c32a
Rename weather tool to match repeated failed invocations
Browse files- src/agent.py +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
|
9 |
|
10 |
|
11 |
def get_agent() -> CodeAgent:
|
@@ -27,7 +27,7 @@ def get_agent() -> CodeAgent:
|
|
27 |
agent = CodeAgent(
|
28 |
model=model,
|
29 |
tools=[
|
30 |
-
|
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
|
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:
|