Update app.py
Browse filesAdded base tools
app.py
CHANGED
|
@@ -1,4 +1,3 @@
|
|
| 1 |
-
|
| 2 |
from smolagents import CodeAgent, HfApiModel, load_tool, tool
|
| 3 |
from tools.final_answer import FinalAnswerTool
|
| 4 |
from typing import List
|
|
@@ -27,6 +26,7 @@ def search_wikipedia_pages(search_string: str) -> List[str]:
|
|
| 27 |
print (f"An error occured {s_error}")
|
| 28 |
|
| 29 |
@tool
|
|
|
|
| 30 |
def get_wikipedia_page_content(page_name: str, p_summary: bool = True) -> str:
|
| 31 |
"""This tool helps to get the content of wikipedia pages by their name.
|
| 32 |
If you use this tool based on the output of search_wikipedia_pages
|
|
@@ -71,7 +71,8 @@ agent = CodeAgent(
|
|
| 71 |
planning_interval=None,
|
| 72 |
name=None,
|
| 73 |
description=None,
|
| 74 |
-
prompt_templates=prompt_templates
|
|
|
|
| 75 |
)
|
| 76 |
|
| 77 |
GradioUI(agent).launch()
|
|
|
|
|
|
|
| 1 |
from smolagents import CodeAgent, HfApiModel, load_tool, tool
|
| 2 |
from tools.final_answer import FinalAnswerTool
|
| 3 |
from typing import List
|
|
|
|
| 26 |
print (f"An error occured {s_error}")
|
| 27 |
|
| 28 |
@tool
|
| 29 |
+
|
| 30 |
def get_wikipedia_page_content(page_name: str, p_summary: bool = True) -> str:
|
| 31 |
"""This tool helps to get the content of wikipedia pages by their name.
|
| 32 |
If you use this tool based on the output of search_wikipedia_pages
|
|
|
|
| 71 |
planning_interval=None,
|
| 72 |
name=None,
|
| 73 |
description=None,
|
| 74 |
+
prompt_templates=prompt_templates,
|
| 75 |
+
add_base_tools=True
|
| 76 |
)
|
| 77 |
|
| 78 |
GradioUI(agent).launch()
|