borisyich commited on
Commit
3483ac6
·
verified ·
1 Parent(s): d7ae409

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +27 -27
agent.py CHANGED
@@ -11,33 +11,33 @@ model = TransformersModel(
11
  #token=os.getenv('HF_TOKEN')
12
  )
13
 
14
- web_agent = ToolCallingAgent(
15
- tools=[
16
- DuckDuckGoSearchTool(),
17
- VisitWebpageTool(),
18
- WikipediaSearchTool()
19
- ],
20
- model=model,
21
- name="search_agent",
22
- description="Runs web searches for you. Give it your query as an argument.",
23
- )
24
- python_agent = CodeAgent(
25
- tools=[
26
- PythonInterpreterTool()
27
- ],
28
- model=model,
29
- name='python_agent',
30
- description='Use additional_authorized_imports for you. You need to do actions and help to answer the questions with python code',
31
- additional_authorized_imports=[
32
- "json",
33
- "pandas",
34
- "numpy",
35
- "requests",
36
- "time",
37
- "datetime",
38
- ],
39
- add_base_tools=True,
40
- )
41
 
42
  class BasicAgent:
43
  """An agent who is able to answer questions."""
 
11
  #token=os.getenv('HF_TOKEN')
12
  )
13
 
14
+ #web_agent = ToolCallingAgent(
15
+ # tools=[
16
+ # DuckDuckGoSearchTool(),
17
+ # VisitWebpageTool(),
18
+ # WikipediaSearchTool()
19
+ # ],
20
+ # model=model,
21
+ # name="search_agent",
22
+ # description="Runs web searches for you. Give it your query as an argument.",
23
+ # )
24
+ #python_agent = CodeAgent(
25
+ # tools=[
26
+ # PythonInterpreterTool()
27
+ # ],
28
+ # model=model,
29
+ # name='python_agent',
30
+ # description='Use additional_authorized_imports for you. You need to do actions and help to answer the questions with python code',
31
+ # additional_authorized_imports=[
32
+ # "json",
33
+ # "pandas",
34
+ # "numpy",
35
+ # "requests",
36
+ # "time",
37
+ # "datetime",
38
+ # ],
39
+ # add_base_tools=True,
40
+ # )
41
 
42
  class BasicAgent:
43
  """An agent who is able to answer questions."""