Update app.py
Browse files
app.py
CHANGED
@@ -191,14 +191,14 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
|
|
191 |
# (1) Create the LLM wrapper
|
192 |
llm_model = BasicModel(model_id=MODEL_ID, hf_token=hf_token)
|
193 |
# (2) Create the tools
|
194 |
-
web_structure_analyzer_tool = WebpageStructureAnalyzerTool()
|
195 |
wiki_tool = CachedWikiTool()
|
196 |
search_tool = CachedWebSearchTool()
|
197 |
python_tool = PythonInterpreterTool()
|
198 |
html_parse_tool = VisitWebpageTool()
|
199 |
#(3) Create the CodeAgent, passing the LLM wrapper and tools
|
200 |
agent = CodeAgent(model=llm_model,
|
201 |
-
tools=[
|
202 |
max_steps=5,
|
203 |
add_base_tools=True,
|
204 |
additional_authorized_imports=["dateparser", "bs4", "regex"])
|
|
|
191 |
# (1) Create the LLM wrapper
|
192 |
llm_model = BasicModel(model_id=MODEL_ID, hf_token=hf_token)
|
193 |
# (2) Create the tools
|
194 |
+
# web_structure_analyzer_tool = WebpageStructureAnalyzerTool()
|
195 |
wiki_tool = CachedWikiTool()
|
196 |
search_tool = CachedWebSearchTool()
|
197 |
python_tool = PythonInterpreterTool()
|
198 |
html_parse_tool = VisitWebpageTool()
|
199 |
#(3) Create the CodeAgent, passing the LLM wrapper and tools
|
200 |
agent = CodeAgent(model=llm_model,
|
201 |
+
tools=[wiki_tool, search_tool, python_tool, html_parse_tool], # <- web_structure_analyzer_tool has some bugs, don't include here
|
202 |
max_steps=5,
|
203 |
add_base_tools=True,
|
204 |
additional_authorized_imports=["dateparser", "bs4", "regex"])
|