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 660a93338a
commit 2cc9463474
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: