arlindkadra commited on
Commit
b993331
·
1 Parent(s): eb97e64

Adding tavily instead of duckduckgosearch

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -0
  2. 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
- search_tool = DuckDuckGoSearchRun()
 
 
 
 
 
 
 
 
 
 
 
 
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: