IamRulo commited on
Commit
6f13b80
·
verified ·
1 Parent(s): 4dfd7f1

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -13,7 +13,7 @@ from langchain_core.messages import SystemMessage, HumanMessage
13
  from langchain_core.tools import tool
14
  from langchain.tools.retriever import create_retriever_tool
15
 
16
- from duckduckgo_search import ddg
17
 
18
  load_dotenv()
19
 
@@ -120,10 +120,10 @@ def web_search(query: str) -> str:
120
  for doc in search_docs
121
  ])
122
  """
123
- search_docs = ddg(query, max_results=3)
124
  formatted_search_docs = "\n\n---\n\n".join(
125
  [
126
- f'<Document source="{doc["href"]}" page=""/>\n{doc["body"]}\n</Document>'
127
  for doc in search_docs
128
  ])
129
 
 
13
  from langchain_core.tools import tool
14
  from langchain.tools.retriever import create_retriever_tool
15
 
16
+ from duckduckgo_search import DDGS
17
 
18
  load_dotenv()
19
 
 
120
  for doc in search_docs
121
  ])
122
  """
123
+ search_docs = DDGS().text(query, max_results=3)
124
  formatted_search_docs = "\n\n---\n\n".join(
125
  [
126
+ f'<Document source="{doc["href"]}"/>\n{doc["body"]}\n</Document>'
127
  for doc in search_docs
128
  ])
129