wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-19 12:24:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: update TavilySearchWithImages to inherit from TavilySearchResults (#522)

Browse Source
This commit is contained in:
CHANGXUBO
2025-08-21 09:52:12 +08:00
committed by GitHub
parent 270d8c3712
commit db6c1bf7cb
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/tools/tavily_search/tavily_search_results_with_images.py
View File

@@ -9,7 +9,9 @@ from langchain.callbacks.manager import (
AsyncCallbackManagerForToolRun, AsyncCallbackManagerForToolRun,
CallbackManagerForToolRun, CallbackManagerForToolRun,
) )
from langchain_tavily.tavily_search import TavilySearch
# from langchain_tavily.tavily_search import TavilySearch
from langchain_community.tools.tavily_search.tool import TavilySearchResults
from pydantic import Field from pydantic import Field
from src.tools.tavily_search.tavily_search_api_wrapper import ( from src.tools.tavily_search.tavily_search_api_wrapper import (
@@ -19,7 +21,7 @@ from src.tools.tavily_search.tavily_search_api_wrapper import (
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class TavilySearchWithImages(TavilySearch): # type: ignore[override, override] class TavilySearchWithImages(TavilySearchResults): # type: ignore[override, override]
"""Tool that queries the Tavily Search API and gets back json. """Tool that queries the Tavily Search API and gets back json.
Setup: Setup: