Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[simple] add duckduckgo to load_tools.py #3376

Merged
merged 6 commits into from
Apr 25, 2023
Merged

[simple] add duckduckgo to load_tools.py #3376

Changes from 2 commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
48144bb
add duckduckgo to loadtools
skcoirz Apr 23, 2023
3c5b2aa
Merge branch 'hwchase17:master' into master
skcoirz Apr 23, 2023
4a9c1a0
Merge branch 'hwchase17:master' into master
skcoirz Apr 24, 2023
022f790
fix lint error in agent/load_tools.py
skcoirz Apr 24, 2023
a5412a0
Merge branch 'hwchase17:master' into master
skcoirz Apr 24, 2023
36de043
Merge branch 'hwchase17:master' into master
skcoirz Apr 25, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions langchain/agents/load_tools.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
from langchain.tools.arxiv.tool import ArxivQueryRun
from langchain.tools.base import BaseTool
from langchain.tools.bing_search.tool import BingSearchRun
from langchain.tools.ddg_search.tool import DuckDuckGoSearchTool
from langchain.tools.google_search.tool import GoogleSearchResults, GoogleSearchRun
from langchain.tools.human.tool import HumanInputRun
from langchain.tools.python.tool import PythonREPLTool
Expand All @@ -31,6 +32,7 @@
from langchain.utilities.apify import ApifyWrapper
from langchain.utilities.bash import BashProcess
from langchain.utilities.bing_search import BingSearchAPIWrapper
from langchain.utilities.duckduckgo_search import DuckDuckGoSearchAPIWrapper
from langchain.utilities.google_search import GoogleSearchAPIWrapper
from langchain.utilities.google_serper import GoogleSerperAPIWrapper
from langchain.utilities.searx_search import SearxSearchWrapper
Expand Down Expand Up @@ -214,6 +216,8 @@ def _get_searx_search_results_json(**kwargs: Any) -> BaseTool:
def _get_bing_search(**kwargs: Any) -> BaseTool:
return BingSearchRun(api_wrapper=BingSearchAPIWrapper(**kwargs))

def _get_ddg_search(**kwargs: Any) -> BaseTool:
return DuckDuckGoSearchTool(api_wrapper=DuckDuckGoSearchAPIWrapper(**kwargs))

def _get_human_tool(**kwargs: Any) -> BaseTool:
return HumanInputRun(**kwargs)
Expand All @@ -237,6 +241,7 @@ def _get_human_tool(**kwargs: Any) -> BaseTool:
["searx_host", "engines", "num_results", "aiosession"],
),
"bing-search": (_get_bing_search, ["bing_subscription_key", "bing_search_url"]),
"ddg-search": (_get_ddg_search, []),
"google-serper": (_get_google_serper, ["serper_api_key"]),
"serpapi": (_get_serpapi, ["serpapi_api_key", "aiosession"]),
"searx-search": (_get_searx_search, ["searx_host", "engines", "aiosession"]),
Expand Down