From cf4f49c6e9617f5927092495cd634e89b5f74dd4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Thu, 7 Mar 2024 20:32:18 -0300 Subject: [PATCH] cutting new version --- src/crewai_tools/tools/pdf_search_tool/pdf_search_tool.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/crewai_tools/tools/pdf_search_tool/pdf_search_tool.py b/src/crewai_tools/tools/pdf_search_tool/pdf_search_tool.py index bb85673ba..e75cb8610 100644 --- a/src/crewai_tools/tools/pdf_search_tool/pdf_search_tool.py +++ b/src/crewai_tools/tools/pdf_search_tool/pdf_search_tool.py @@ -9,7 +9,7 @@ from ..rag.rag_tool import RagTool class FixedPDFSearchToolSchema(BaseModel): """Input for PDFSearchTool.""" - search_query: str = Field(..., description="Mandatory search query you want to use to search the PDF's content") + query: str = Field(..., description="Mandatory query you want to use to search the PDF's content") class PDFSearchToolSchema(FixedPDFSearchToolSchema): """Input for PDFSearchTool.""" @@ -32,10 +32,10 @@ class PDFSearchTool(RagTool): def _run( self, - search_query: str, + query: str, **kwargs: Any, ) -> Any: pdf = kwargs.get('pdf', self.pdf) self.app = App() self.app.add(pdf, data_type=DataType.PDF_FILE) - return super()._run(query=search_query) \ No newline at end of file + return super()._run(query=query) \ No newline at end of file