fix: background investigator node support more search engine (#75)

Change-Id: I030a2b9218dfbda2dd2383b7a73266dd7de589c7
This commit is contained in:
Zhao Longjie
2025-05-12 20:15:47 +08:00
committed by GitHub
parent 98e0d7cbb6
commit 9266201fe5
3 changed files with 23 additions and 16 deletions

View File

@@ -14,11 +14,10 @@ from src.tools.tavily_search.tavily_search_results_with_images import (
TavilySearchResultsWithImages,
)
from .decorators import create_logged_tool
from src.tools.decorators import create_logged_tool
logger = logging.getLogger(__name__)
LoggedTavilySearch = create_logged_tool(TavilySearchResultsWithImages)
tavily_search_tool = LoggedTavilySearch(
name="web_search",
@@ -53,5 +52,7 @@ arxiv_search_tool = LoggedArxivSearch(
)
if __name__ == "__main__":
results = tavily_search_tool.invoke("cute panda")
results = LoggedDuckDuckGoSearch(
name="web_search", max_results=SEARCH_MAX_RESULTS, output_format="list"
).invoke("cute panda")
print(json.dumps(results, indent=2, ensure_ascii=False))