skills
/
deerflow2
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

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, 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: