ogghst commited on
Commit
f7a2733
·
verified ·
1 Parent(s): df38b42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -4,6 +4,8 @@ import requests
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
 
 
7
 
8
  from Gradio_UI import GradioUI
9
 
@@ -44,8 +46,10 @@ def get_current_time_in_timezone(timezone: str) -> str:
44
  except Exception as e:
45
  return f"Error fetching time for timezone '{timezone}': {str(e)}"
46
 
 
 
 
47
 
48
- #final_answer = FinalAnswerTool()
49
 
50
  # 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:
51
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
@@ -58,7 +62,6 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may
58
  custom_role_conversions=None,
59
  )
60
 
61
- final_answer = FinalAnswerTool()
62
 
63
  # Import tool from Hub
64
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
@@ -68,7 +71,7 @@ with open("prompts.yaml", 'r') as stream:
68
 
69
  agent = CodeAgent(
70
  model=model,
71
- tools=[final_answer, confuse_me, image_generation_tool], ## add your tools here (don't remove final answer)
72
  max_steps=6,
73
  verbosity_level=1,
74
  grammar=None,
 
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
7
+ from tools.web_search import DuckDuckGoSearchTool
8
+ from tools.visit_webpage import VisitWebpageTool
9
 
10
  from Gradio_UI import GradioUI
11
 
 
46
  except Exception as e:
47
  return f"Error fetching time for timezone '{timezone}': {str(e)}"
48
 
49
+ final_answer = FinalAnswerTool()
50
+ web_search_tool = DuckDuckGoSearchTool()
51
+ visit_webpage_tool = VisitWebpageTool()
52
 
 
53
 
54
  # 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:
55
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
 
62
  custom_role_conversions=None,
63
  )
64
 
 
65
 
66
  # Import tool from Hub
67
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
71
 
72
  agent = CodeAgent(
73
  model=model,
74
+ tools=[final_answer, confuse_me, image_generation_tool, web_search_tool, visit_webpage_tool], ## add your tools here (don't remove final answer)
75
  max_steps=6,
76
  verbosity_level=1,
77
  grammar=None,