github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-11 07:25:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

improve duck duck go tool (#13165)

Browse Source
This commit is contained in:
Harrison Chase 2023-11-09 20:49:39 -08:00 committed by GitHub
parent 850336bcf1
commit 0a2b1c7471
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
libs/langchain/langchain/tools/ddg_search/tool.py
View File

@ -1,14 +1,18 @@
"""Tool for the DuckDuckGo search API.""" """Tool for the DuckDuckGo search API."""
import warnings import warnings
from typing import Any, Optional from typing import Any, Optional, Type
from langchain.callbacks.manager import CallbackManagerForToolRun from langchain.callbacks.manager import CallbackManagerForToolRun
from langchain.pydantic_v1 import Field from langchain.pydantic_v1 import BaseModel, Field
from langchain.tools.base import BaseTool from langchain.tools.base import BaseTool
from langchain.utilities.duckduckgo_search import DuckDuckGoSearchAPIWrapper from langchain.utilities.duckduckgo_search import DuckDuckGoSearchAPIWrapper
class DDGInput(BaseModel):
query: str = Field(description="search query to look up")
class DuckDuckGoSearchRun(BaseTool): class DuckDuckGoSearchRun(BaseTool):
"""Tool that queries the DuckDuckGo search API.""" """Tool that queries the DuckDuckGo search API."""
@ -21,6 +25,7 @@ class DuckDuckGoSearchRun(BaseTool):
api_wrapper: DuckDuckGoSearchAPIWrapper = Field( api_wrapper: DuckDuckGoSearchAPIWrapper = Field(
default_factory=DuckDuckGoSearchAPIWrapper default_factory=DuckDuckGoSearchAPIWrapper
) )
args_schema: Type[BaseModel] = DDGInput
def _run( def _run(
self, self,
@ -45,6 +50,7 @@ class DuckDuckGoSearchResults(BaseTool):
default_factory=DuckDuckGoSearchAPIWrapper default_factory=DuckDuckGoSearchAPIWrapper
) )
backend: str = "api" backend: str = "api"
args_schema: Type[BaseModel] = DDGInput
def _run( def _run(
self, self,