rsobieski commited on
Commit
95f4932
·
verified ·
1 Parent(s): 9922f58

Update tools.py

Browse files
Files changed (1) hide show
  1. tools.py +11 -3
tools.py CHANGED
@@ -1,4 +1,3 @@
1
- """Tool definitions for LangGraph agent."""
2
  from langchain_core.tools import tool
3
  from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
4
  from langchain_community.tools.tavily_search import TavilySearchResults
@@ -48,5 +47,14 @@ def web_search(query: str) -> str:
48
  docs = TavilySearchResults(max_results=3).invoke(query=query)
49
  return "\n\n".join(doc.page_content for doc in docs)
50
 
51
- # Export list of all tools
52
- TOOLS = [multiply, add, subtract, divide, modulus, wiki_search, arxiv_search, web_search]
 
 
 
 
 
 
 
 
 
 
 
1
  from langchain_core.tools import tool
2
  from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
3
  from langchain_community.tools.tavily_search import TavilySearchResults
 
47
  docs = TavilySearchResults(max_results=3).invoke(query=query)
48
  return "\n\n".join(doc.page_content for doc in docs)
49
 
50
+ # Tools as a dict for manual agent usage
51
+ TOOLS = {
52
+ "multiply": multiply,
53
+ "add": add,
54
+ "subtract": subtract,
55
+ "divide": divide,
56
+ "modulus": modulus,
57
+ "wiki_search": wiki_search,
58
+ "arxiv_search": arxiv_search,
59
+ "web_search": web_search,
60
+ }