mzhKU commited on
Commit
dedcbf6
·
1 Parent(s): ae7a494
Files changed (3) hide show
  1. Gradio_UI.py +1 -1
  2. app.py +15 -36
  3. tools/visit_webpage.py +1 -0
Gradio_UI.py CHANGED
@@ -283,7 +283,7 @@ class GradioUI:
283
  [upload_file, file_uploads_log],
284
  [upload_status, file_uploads_log],
285
  )
286
- text_input = gr.Textbox(lines=1, label="Chat Message")
287
  text_input.submit(
288
  self.log_user_message,
289
  [text_input, file_uploads_log],
 
283
  [upload_file, file_uploads_log],
284
  [upload_status, file_uploads_log],
285
  )
286
+ text_input = gr.Textbox(lines=1, label="Chat Message", value="List the management of Google")
287
  text_input.submit(
288
  self.log_user_message,
289
  [text_input, file_uploads_log],
app.py CHANGED
@@ -1,61 +1,40 @@
 
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: #it's import to specify the return type
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
18
- """
19
- return "What magic will you build ?"
20
-
21
- @tool
22
- def get_current_time_in_timezone(timezone: str) -> str:
23
- """A tool that fetches the current local time in a specified timezone.
24
- Args:
25
- timezone: A string representing a valid timezone (e.g., 'America/New_York').
26
- """
27
- try:
28
- # Create timezone object
29
- tz = pytz.timezone(timezone)
30
- # Get current time in that timezone
31
- local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
32
- return f"The current local time in {timezone} is: {local_time}"
33
- except Exception as e:
34
- return f"Error fetching time for timezone '{timezone}': {str(e)}"
35
-
36
 
 
 
37
  final_answer = FinalAnswerTool()
38
-
39
- # 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:
40
- # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
41
 
42
  model = HfApiModel(
43
- max_tokens=2096,
44
- temperature=0.5,
45
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
46
- custom_role_conversions=None,
47
  )
48
 
49
-
50
  # Import tool from Hub
51
- image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
52
 
53
  with open("prompts.yaml", 'r') as stream:
54
  prompt_templates = yaml.safe_load(stream)
55
 
56
  agent = CodeAgent(
57
  model=model,
58
- tools=[final_answer], ## add your tools here (don't remove final answer)
59
  max_steps=6,
60
  verbosity_level=1,
61
  grammar=None,
 
1
+ import os
2
  from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
3
  import datetime
4
  import requests
5
  import pytz
6
  import yaml
7
  from tools.final_answer import FinalAnswerTool
8
+ from tools.visit_webpage import VisitWebpageTool
9
+ from tools.web_search import DuckDuckGoSearchTool
10
 
11
  from Gradio_UI import GradioUI
12
 
13
+ f = open("hf_token_read.txt")
14
+ d = f.read()
15
+ f.close()
16
+ os.environ["HF_TOKEN"] = d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
 
18
+ visit_webpage_tool = VisitWebpageTool()
19
+ websearch_tool = DuckDuckGoSearchTool()
20
  final_answer = FinalAnswerTool()
21
+ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 
22
 
23
  model = HfApiModel(
24
+ max_tokens=1000,
25
+ temperature=0.5,
26
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
27
+ custom_role_conversions=None,
28
  )
29
 
 
30
  # Import tool from Hub
 
31
 
32
  with open("prompts.yaml", 'r') as stream:
33
  prompt_templates = yaml.safe_load(stream)
34
 
35
  agent = CodeAgent(
36
  model=model,
37
+ tools=[final_answer, websearch_tool, visit_webpage_tool], ## add your tools here (don't remove final answer)
38
  max_steps=6,
39
  verbosity_level=1,
40
  grammar=None,
tools/visit_webpage.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from typing import Any, Optional
2
  from smolagents.tools import Tool
3
  import requests
 
1
+ import re
2
  from typing import Any, Optional
3
  from smolagents.tools import Tool
4
  import requests