github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-24 07:35:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: added option to change how duckduckgosearchresults tool converts api outputs into string (#22580)

Browse Source 
Co-authored-by: Bagatur <baskaryan@gmail.com>
This commit is contained in:
Isaac Francisco 2024-09-02 15:42:19 -07:00 committed by GitHub
parent 78ff51ce83
commit 4833375200
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
libs/community/langchain_community/tools/ddg_search/tool.py
View File

@ -1,7 +1,7 @@
"""Tool for the DuckDuckGo search API."""
import warnings
from typing import Any, Optional, Type
from typing import Any, List, Optional, Type
from langchain_core.callbacks import CallbackManagerForToolRun
from langchain_core.pydantic_v1 import BaseModel, Field
@ -89,6 +89,10 @@ class DuckDuckGoSearchResults(BaseTool):
)
backend: str = "text"
args_schema: Type[BaseModel] = DDGInput
keys_to_include: Optional[List[str]] = None
"""Which keys from each result to include. If None all keys are included."""
results_separator: str = ", "
"""Character for separating results."""
def _run(
self,
@ -97,8 +101,17 @@ class DuckDuckGoSearchResults(BaseTool):
) -> str:
"""Use the tool."""
res = self.api_wrapper.results(query, self.max_results, source=self.backend)
res_strs = [", ".join([f"{k}: {v}" for k, v in d.items()]) for d in res]
return ", ".join([f"[{rs}]" for rs in res_strs])
res_strs = [
", ".join(
[
f"{k}: {v}"
for k, v in d.items()
if not self.keys_to_include or k in self.keys_to_include
]
)
for d in res
]
return self.results_separator.join(res_strs)
def DuckDuckGoSearchTool(*args: Any, **kwargs: Any) -> DuckDuckGoSearchRun: