Harrison/ddg (#3206)

Co-authored-by: itai <itai.marks@gmail.com>
Co-authored-by: Itai Marks <itaim@users.noreply.github.com>
Co-authored-by: Tianyi Pan <60060750+tipani86@users.noreply.github.com>
Co-authored-by: Tianyi Pan <tianyi.pan@clobotics.com>
Co-authored-by: Adilzhan Ismailov <13088690+aismlv@users.noreply.github.com>
Co-authored-by: Justin Flick <Justinjayflick@gmail.com>
Co-authored-by: Justin Flick <jflick@homesite.com>
This commit is contained in:
Harrison Chase
2023-04-19 21:32:26 -07:00
committed by GitHub
parent 36c10f8a52
commit d2520a5f1e
8 changed files with 261 additions and 4 deletions

View File

@@ -0,0 +1,22 @@
import pytest
from langchain.tools.ddg_search.tool import DuckDuckGoSearchTool
def ddg_installed() -> bool:
try:
from duckduckgo_search import ddg # noqa: F401
return True
except Exception as e:
print(f"duckduckgo not installed, skipping test {e}")
return False
@pytest.mark.skipif(not ddg_installed(), reason="requires duckduckgo-search package")
def test_ddg_search_tool() -> None:
keywords = "Bella Ciao"
tool = DuckDuckGoSearchTool()
result = tool(keywords)
print(result)
assert len(result.split()) > 20