Spaces:
Runtime error
Runtime error
Commit ·
b993331
1
Parent(s): eb97e64
Adding tavily instead of duckduckgosearch
Browse files- requirements.txt +1 -0
- tools.py +14 -1
requirements.txt
CHANGED
|
@@ -7,6 +7,7 @@ langchain>=0.3.27
|
|
| 7 |
langchain-community>=0.3.27
|
| 8 |
langchain-core>=0.3.74
|
| 9 |
langchain-huggingface>=0.3.1
|
|
|
|
| 10 |
langgraph>=0.6.6
|
| 11 |
rank-bm25>=0.2.2
|
| 12 |
sentence-transformers>=5.1.0
|
|
|
|
| 7 |
langchain-community>=0.3.27
|
| 8 |
langchain-core>=0.3.74
|
| 9 |
langchain-huggingface>=0.3.1
|
| 10 |
+
langchain-tavily>=0.2.11
|
| 11 |
langgraph>=0.6.6
|
| 12 |
rank-bm25>=0.2.2
|
| 13 |
sentence-transformers>=5.1.0
|
tools.py
CHANGED
|
@@ -5,10 +5,23 @@ from langchain_community.retrievers import WikipediaRetriever
|
|
| 5 |
from langchain_community.document_loaders import ArxivLoader, WikipediaLoader
|
| 6 |
from langchain_core.documents import Document
|
| 7 |
from langchain_core.tools import tool
|
|
|
|
| 8 |
from sentence_transformers import SentenceTransformer
|
| 9 |
|
| 10 |
|
| 11 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
@tool
|
| 14 |
def wikipedia_search(query: str) -> str:
|
|
|
|
| 5 |
from langchain_community.document_loaders import ArxivLoader, WikipediaLoader
|
| 6 |
from langchain_core.documents import Document
|
| 7 |
from langchain_core.tools import tool
|
| 8 |
+
from langchain_tavily import TavilySearch
|
| 9 |
from sentence_transformers import SentenceTransformer
|
| 10 |
|
| 11 |
|
| 12 |
+
|
| 13 |
+
search_tool = TavilySearch(
|
| 14 |
+
max_results=5,
|
| 15 |
+
topic="general",
|
| 16 |
+
# include_answer=False,
|
| 17 |
+
# include_raw_content=False,
|
| 18 |
+
# include_images=False,
|
| 19 |
+
# include_image_descriptions=False,
|
| 20 |
+
# search_depth="basic",
|
| 21 |
+
# time_range="day",
|
| 22 |
+
# include_domains=None,
|
| 23 |
+
# exclude_domains=None
|
| 24 |
+
)
|
| 25 |
|
| 26 |
@tool
|
| 27 |
def wikipedia_search(query: str) -> str:
|