bassommma commited on
Commit
a39aaa4
·
verified ·
1 Parent(s): a4ba279

Update tools.py

Browse files
Files changed (1) hide show
  1. tools.py +3 -1
tools.py CHANGED
@@ -2,11 +2,13 @@
2
  from typing import Annotated
3
  from valetax_rag import ValetaxRAG
4
  from typing import List, Optional
5
-
6
  from config import llm,memory
7
 
8
  rag = ValetaxRAG()
9
 
 
 
10
 
11
  async def rag_tool(query: Annotated[str, "The user’s question about the documents"]) -> str:
12
  """Answer user questions using the async RAG pipeline."""
 
2
  from typing import Annotated
3
  from valetax_rag import ValetaxRAG
4
  from typing import List, Optional
5
+ from langchain_community.tools import DuckDuckGoSearchRun
6
  from config import llm,memory
7
 
8
  rag = ValetaxRAG()
9
 
10
+ search = DuckDuckGoSearchRun()
11
+
12
 
13
  async def rag_tool(query: Annotated[str, "The user’s question about the documents"]) -> str:
14
  """Answer user questions using the async RAG pipeline."""