Spaces:
Sleeping
Sleeping
Update app.py
Browse filesupdated debugging and weatherinfo
app.py
CHANGED
|
@@ -1,17 +1,16 @@
|
|
| 1 |
-
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
|
| 2 |
import datetime
|
| 3 |
import requests
|
| 4 |
import pytz
|
| 5 |
import yaml
|
| 6 |
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:
|
| 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
|
|
@@ -35,9 +34,14 @@ def get_current_time_in_timezone(timezone: str) -> str:
|
|
| 35 |
|
| 36 |
@tool
|
| 37 |
def get_weather(location: str) -> str:
|
| 38 |
-
"""
|
|
|
|
|
|
|
| 39 |
Args:
|
| 40 |
location: A string representing a city name, coordinates, or location (e.g., 'London', 'New York, NY', '40.7128,-74.0060').
|
|
|
|
|
|
|
|
|
|
| 41 |
"""
|
| 42 |
try:
|
| 43 |
# Using wttr.in API which provides weather data in a simple format
|
|
@@ -83,29 +87,58 @@ Wind: {wind_speed_kmph} km/h ({wind_speed_mph} mph) {wind_dir}"""
|
|
| 83 |
except Exception as e:
|
| 84 |
return f"Error fetching weather for '{location}': {str(e)}"
|
| 85 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 86 |
|
| 87 |
final_answer = FinalAnswerTool()
|
| 88 |
|
| 89 |
# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
|
| 90 |
-
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
|
| 91 |
-
|
| 92 |
model = HfApiModel(
|
| 93 |
-
max_tokens=2096,
|
| 94 |
-
temperature=0.5,
|
| 95 |
-
model_id='Qwen/Qwen2.5-Coder-32B-Instruct'
|
| 96 |
-
custom_role_conversions=None,
|
| 97 |
)
|
| 98 |
|
| 99 |
-
|
| 100 |
# Import tool from Hub
|
| 101 |
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
|
| 102 |
|
| 103 |
with open("prompts.yaml", 'r') as stream:
|
| 104 |
prompt_templates = yaml.safe_load(stream)
|
| 105 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 106 |
agent = CodeAgent(
|
| 107 |
model=model,
|
| 108 |
-
tools=
|
| 109 |
max_steps=6,
|
| 110 |
verbosity_level=1,
|
| 111 |
grammar=None,
|
|
@@ -115,5 +148,4 @@ agent = CodeAgent(
|
|
| 115 |
prompt_templates=prompt_templates
|
| 116 |
)
|
| 117 |
|
| 118 |
-
|
| 119 |
GradioUI(agent).launch()
|
|
|
|
| 1 |
+
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool, Tool
|
| 2 |
import datetime
|
| 3 |
import requests
|
| 4 |
import pytz
|
| 5 |
import yaml
|
| 6 |
from tools.final_answer import FinalAnswerTool
|
|
|
|
| 7 |
from Gradio_UI import GradioUI
|
| 8 |
|
| 9 |
# Below is an example of a tool that does nothing. Amaze us with your creativity !
|
| 10 |
@tool
|
| 11 |
+
def my_custom_tool(arg1: str, arg2: int) -> str: # it's import to specify the return type
|
| 12 |
+
# Keep this format for the description / args / args description but feel free to modify the tool
|
| 13 |
+
"""A tool that does nothing yet
|
| 14 |
Args:
|
| 15 |
arg1: the first argument
|
| 16 |
arg2: the second argument
|
|
|
|
| 34 |
|
| 35 |
@tool
|
| 36 |
def get_weather(location: str) -> str:
|
| 37 |
+
"""
|
| 38 |
+
Get current weather conditions for any location. Use this tool when asked about weather, temperature, or climate conditions.
|
| 39 |
+
|
| 40 |
Args:
|
| 41 |
location: A string representing a city name, coordinates, or location (e.g., 'London', 'New York, NY', '40.7128,-74.0060').
|
| 42 |
+
|
| 43 |
+
Returns:
|
| 44 |
+
str: A formatted string containing the current weather information including temperature, conditions, humidity, and wind.
|
| 45 |
"""
|
| 46 |
try:
|
| 47 |
# Using wttr.in API which provides weather data in a simple format
|
|
|
|
| 87 |
except Exception as e:
|
| 88 |
return f"Error fetching weather for '{location}': {str(e)}"
|
| 89 |
|
| 90 |
+
# Alternative: Create a wrapper tool with a more explicit name
|
| 91 |
+
@tool
|
| 92 |
+
def check_current_weather(location: str) -> str:
|
| 93 |
+
"""
|
| 94 |
+
Check the current weather for a location. This is the primary tool for weather information.
|
| 95 |
+
|
| 96 |
+
Args:
|
| 97 |
+
location: City name or location (e.g., 'Paris', 'Tokyo, Japan')
|
| 98 |
+
|
| 99 |
+
Returns:
|
| 100 |
+
str: Current weather conditions with temperature, humidity, and wind information
|
| 101 |
+
"""
|
| 102 |
+
return get_weather(location)
|
| 103 |
|
| 104 |
final_answer = FinalAnswerTool()
|
| 105 |
|
| 106 |
# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
|
| 107 |
+
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
|
|
|
|
| 108 |
model = HfApiModel(
|
| 109 |
+
max_tokens=2096,
|
| 110 |
+
temperature=0.5,
|
| 111 |
+
model_id='Qwen/Qwen2.5-Coder-32B-Instruct', # it is possible that this model may be overloaded
|
| 112 |
+
custom_role_conversions=None,
|
| 113 |
)
|
| 114 |
|
|
|
|
| 115 |
# Import tool from Hub
|
| 116 |
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
|
| 117 |
|
| 118 |
with open("prompts.yaml", 'r') as stream:
|
| 119 |
prompt_templates = yaml.safe_load(stream)
|
| 120 |
+
|
| 121 |
+
# Create tools list with all tools
|
| 122 |
+
tools_list = [
|
| 123 |
+
final_answer,
|
| 124 |
+
get_current_time_in_timezone,
|
| 125 |
+
get_weather,
|
| 126 |
+
check_current_weather, # Adding the alternative weather tool
|
| 127 |
+
my_custom_tool,
|
| 128 |
+
image_generation_tool
|
| 129 |
+
]
|
| 130 |
+
|
| 131 |
+
# Debug: Print available tools
|
| 132 |
+
print("Available tools:")
|
| 133 |
+
for tool in tools_list:
|
| 134 |
+
if hasattr(tool, 'name'):
|
| 135 |
+
print(f" - {tool.name}: {tool.description if hasattr(tool, 'description') else 'No description'}")
|
| 136 |
+
else:
|
| 137 |
+
print(f" - {tool}")
|
| 138 |
+
|
| 139 |
agent = CodeAgent(
|
| 140 |
model=model,
|
| 141 |
+
tools=tools_list, # Use the tools list
|
| 142 |
max_steps=6,
|
| 143 |
verbosity_level=1,
|
| 144 |
grammar=None,
|
|
|
|
| 148 |
prompt_templates=prompt_templates
|
| 149 |
)
|
| 150 |
|
|
|
|
| 151 |
GradioUI(agent).launch()
|