langchain/tests/integration_tests/utilities/test_duckduckdgo_search_api.py

23 lines
573 B
Python
Raw Normal View History

import pytest
from langchain.tools.ddg_search.tool import DuckDuckGoSearchRun
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 = DuckDuckGoSearchRun()
result = tool(keywords)
print(result)
assert len(result.split()) > 20