Update agent.py
Browse files
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
|
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 =
|
124 |
formatted_search_docs = "\n\n---\n\n".join(
|
125 |
[
|
126 |
-
f'<Document source="{doc["href"]}"
|
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 |
|