Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import requests
|
|
4 |
import inspect
|
5 |
import pandas as pd
|
6 |
#import smolagents #to test
|
7 |
-
from smolagents import CodeAgent, InferenceClientModel,
|
8 |
from huggingface_hub import InferenceClient
|
9 |
import json
|
10 |
from final_answer import FinalAnswerTool
|
@@ -58,7 +58,7 @@ custom_role_conversions=None,
|
|
58 |
|
59 |
#Tools
|
60 |
final_answer = FinalAnswerTool()
|
61 |
-
duckDuckGoSearch = DuckDuckGoSearchTool() #smolagent version
|
62 |
visitWebpage = VisitWebpageTool()
|
63 |
wikipediaLookup = WikipediaLookupTool()
|
64 |
webSearch = web_search_DuckDuckGoSearchTool()
|
@@ -66,7 +66,7 @@ webSearch = web_search_DuckDuckGoSearchTool()
|
|
66 |
#Agent
|
67 |
agent_codeagent = CodeAgent(
|
68 |
model=model,
|
69 |
-
tools=[final_answer,
|
70 |
max_steps=3,
|
71 |
verbosity_level=1,
|
72 |
grammar=None,
|
|
|
4 |
import inspect
|
5 |
import pandas as pd
|
6 |
#import smolagents #to test
|
7 |
+
from smolagents import CodeAgent, InferenceClientModel, load_tool, tool #DuckDuckGoSearchTool,
|
8 |
from huggingface_hub import InferenceClient
|
9 |
import json
|
10 |
from final_answer import FinalAnswerTool
|
|
|
58 |
|
59 |
#Tools
|
60 |
final_answer = FinalAnswerTool()
|
61 |
+
#duckDuckGoSearch = DuckDuckGoSearchTool() #smolagent version
|
62 |
visitWebpage = VisitWebpageTool()
|
63 |
wikipediaLookup = WikipediaLookupTool()
|
64 |
webSearch = web_search_DuckDuckGoSearchTool()
|
|
|
66 |
#Agent
|
67 |
agent_codeagent = CodeAgent(
|
68 |
model=model,
|
69 |
+
tools=[final_answer, wikipediaLookup, visitWebpage, webSearch], ## add your tools here (don't remove final answer) duckDuckGoSearch,
|
70 |
max_steps=3,
|
71 |
verbosity_level=1,
|
72 |
grammar=None,
|