From e395cb7f4b36e46763e0a98cb0039587629c6d3d Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 3 Oct 2024 14:59:57 -0300 Subject: [PATCH] chore: Add type annotation for 'status' and format multiline string in TavilyAISearch tool (#4011) * Format multiline string in TavilySearchToolComponent description * Add type annotation for 'status' and import 'Any' in TavilyAISearch tool --- .../base/langflow/components/tools/TavilyAISearch.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/components/tools/TavilyAISearch.py b/src/backend/base/langflow/components/tools/TavilyAISearch.py index 3a3e95f18..577e16d65 100644 --- a/src/backend/base/langflow/components/tools/TavilyAISearch.py +++ b/src/backend/base/langflow/components/tools/TavilyAISearch.py @@ -1,3 +1,5 @@ +from typing import Any + import httpx from langchain.tools import StructuredTool from pydantic import BaseModel, Field @@ -10,7 +12,8 @@ from langflow.schema import Data class TavilySearchToolComponent(LCToolComponent): display_name = "Tavily AI Search" - description = """**Tavily AI** is a search engine optimized for LLMs and RAG, aimed at efficient, quick, and persistent search results. It can be used independently or as an agent tool. + description = """**Tavily AI** is a search engine optimized for LLMs and RAG, \ + aimed at efficient, quick, and persistent search results. It can be used independently or as an agent tool. Note: Check 'Advanced' for all options. """ @@ -144,7 +147,7 @@ Note: Check 'Advanced' for all options. if include_images and search_results.get("images"): data_results.append(Data(data={"images": search_results["images"]})) - self.status = data_results + self.status: Any = data_results return data_results except httpx.HTTPStatusError as e: