Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,9 @@ import requests
|
|
4 |
import pytz
|
5 |
import yaml
|
6 |
from tools.final_answer import FinalAnswerTool
|
|
|
|
|
|
|
7 |
|
8 |
from Gradio_UI import GradioUI
|
9 |
|
@@ -309,7 +312,11 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
|
|
309 |
|
310 |
with open("prompts.yaml", 'r') as stream:
|
311 |
prompt_templates = yaml.safe_load(stream)
|
312 |
-
|
|
|
|
|
|
|
|
|
313 |
agent = CodeAgent(
|
314 |
model=model,
|
315 |
tools=[
|
@@ -317,7 +324,9 @@ agent = CodeAgent(
|
|
317 |
task_manager,
|
318 |
timer_tool,
|
319 |
inspiration_generator,
|
320 |
-
get_current_time_in_timezone
|
|
|
|
|
321 |
], ## add your tools here (don't remove final answer)
|
322 |
max_steps=6,
|
323 |
verbosity_level=1,
|
|
|
4 |
import pytz
|
5 |
import yaml
|
6 |
from tools.final_answer import FinalAnswerTool
|
7 |
+
# Import vos outils personnalisés
|
8 |
+
from tools.visit_webpage import VisitWebpageTool
|
9 |
+
from tools.web_search import DuckDuckGoSearchTool
|
10 |
|
11 |
from Gradio_UI import GradioUI
|
12 |
|
|
|
312 |
|
313 |
with open("prompts.yaml", 'r') as stream:
|
314 |
prompt_templates = yaml.safe_load(stream)
|
315 |
+
|
316 |
+
|
317 |
+
web_search_tool = DuckDuckGoSearchTool(max_results=5) # Vous pouvez ajuster max_results selon vos besoins
|
318 |
+
visit_webpage_tool = VisitWebpageTool()
|
319 |
+
|
320 |
agent = CodeAgent(
|
321 |
model=model,
|
322 |
tools=[
|
|
|
324 |
task_manager,
|
325 |
timer_tool,
|
326 |
inspiration_generator,
|
327 |
+
get_current_time_in_timezone,
|
328 |
+
web_search_tool,
|
329 |
+
visit_webpage_tool
|
330 |
], ## add your tools here (don't remove final answer)
|
331 |
max_steps=6,
|
332 |
verbosity_level=1,
|