Spaces:
Runtime error
Runtime error
abtsousa
commited on
Commit
·
823995c
1
Parent(s):
eb3f029
Add wiki_parse_html to tools
Browse files- agent/nodes.py +1 -1
- tools/__init__.py +2 -1
- tools/search.py +0 -1
agent/nodes.py
CHANGED
@@ -57,7 +57,7 @@ def _get_model() -> BaseChatModel:
|
|
57 |
return ChatDeepSeek(
|
58 |
model="deepseek-chat",
|
59 |
temperature=MODEL_TEMPERATURE if MODEL_TEMPERATURE else 0.0,
|
60 |
-
max_retries=2
|
61 |
)
|
62 |
|
63 |
def _get_tools() -> list[BaseTool]:
|
|
|
57 |
return ChatDeepSeek(
|
58 |
model="deepseek-chat",
|
59 |
temperature=MODEL_TEMPERATURE if MODEL_TEMPERATURE else 0.0,
|
60 |
+
max_retries=2
|
61 |
)
|
62 |
|
63 |
def _get_tools() -> list[BaseTool]:
|
tools/__init__.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
from .wikipedia import wiki_search
|
2 |
from .search import web_search
|
3 |
from .code_interpreter import execute_code_multilang
|
4 |
from .files import file_management_toolkit
|
@@ -13,6 +13,7 @@ def get_all_tools() -> list[BaseTool]:
|
|
13 |
"""
|
14 |
tools = [
|
15 |
wiki_search,
|
|
|
16 |
web_search,
|
17 |
execute_code_multilang
|
18 |
]
|
|
|
1 |
+
from .wikipedia import wiki_search, wiki_parse_html
|
2 |
from .search import web_search
|
3 |
from .code_interpreter import execute_code_multilang
|
4 |
from .files import file_management_toolkit
|
|
|
13 |
"""
|
14 |
tools = [
|
15 |
wiki_search,
|
16 |
+
wiki_parse_html,
|
17 |
web_search,
|
18 |
execute_code_multilang
|
19 |
]
|
tools/search.py
CHANGED
@@ -1,4 +1,3 @@
|
|
1 |
-
from langchain_core.tools import tool
|
2 |
from langchain_tavily import TavilySearch
|
3 |
|
4 |
web_search = TavilySearch(max_results=3, topic="general")
|
|
|
|
|
1 |
from langchain_tavily import TavilySearch
|
2 |
|
3 |
web_search = TavilySearch(max_results=3, topic="general")
|