github/deer-flow
1
0
Fork 0
mirror of https://github.com/bytedance/deer-flow.git synced 2026-04-25 11:18:22 +00: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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats 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,
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 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__)
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.
Setup: