abtsousa commited on
Commit
823995c
·
1 Parent(s): eb3f029

Add wiki_parse_html to tools

Browse files
Files changed (3) hide show
  1. agent/nodes.py +1 -1
  2. tools/__init__.py +2 -1
  3. 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")