Aakash010 commited on
Commit
95dc30f
·
verified ·
1 Parent(s): 5dbbfb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -4
app.py CHANGED
@@ -35,7 +35,13 @@ def get_current_time_in_timezone(timezone: str) -> str:
35
  final_answer = FinalAnswerTool()
36
 
37
  # ✅ web_search_tool is defined here
38
- web_search = DuckDuckGoSearchTool(max_results=5)
 
 
 
 
 
 
39
 
40
  model = HfApiModel(
41
  max_tokens=2096,
@@ -44,7 +50,12 @@ model = HfApiModel(
44
  custom_role_conversions=None,
45
  )
46
 
47
- image_generator = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 
 
 
 
48
 
49
  with open("prompts.yaml", 'r') as stream:
50
  prompt_templates = yaml.safe_load(stream)
@@ -52,9 +63,9 @@ with open("prompts.yaml", 'r') as stream:
52
  agent = CodeAgent(
53
  model=model,
54
  # ✅ web_search_tool is NOW included in the tools list
55
- tools=[final_answer, my_custom_tool, get_current_time_in_timezone, web_search, image_generator],
56
  max_steps=6,
57
- verbosity_level=1,
58
  grammar=None,
59
  planning_interval=None,
60
  name=None,
 
35
  final_answer = FinalAnswerTool()
36
 
37
  # ✅ web_search_tool is defined here
38
+ web_search_tool = DuckDuckGoSearchTool(max_results=5)
39
+ web_search_tool.name = "web_search" # ← force a clean, obvious name
40
+ web_search_tool.description = """Searches the web using DuckDuckGo and returns results.
41
+ Args:
42
+ query: The search query string to look up on the web.
43
+ """
44
+
45
 
46
  model = HfApiModel(
47
  max_tokens=2096,
 
50
  custom_role_conversions=None,
51
  )
52
 
53
+ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
54
+ image_generation_tool.name = "generate_image"
55
+ image_generation_tool.description = """Generates an image from a text prompt.
56
+ Args:
57
+ prompt: A detailed text description of the image to generate.
58
+ """
59
 
60
  with open("prompts.yaml", 'r') as stream:
61
  prompt_templates = yaml.safe_load(stream)
 
63
  agent = CodeAgent(
64
  model=model,
65
  # ✅ web_search_tool is NOW included in the tools list
66
+ tools=[final_answer, my_custom_tool, get_current_time_in_timezone, web_search_tool, image_generation_tool],
67
  max_steps=6,
68
+ verbosity_level=2,
69
  grammar=None,
70
  planning_interval=None,
71
  name=None,