Update tools.py
Browse files
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 |
-
#
|
| 52 |
-
TOOLS =
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
}
|