andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 16:48:30 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

cutting new version

Browse Source
This commit is contained in:
João Moura
2024-03-07 20:32:18 -03:00
parent 37aa8d6b63
commit cf4f49c6e9
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/crewai_tools/tools/pdf_search_tool/pdf_search_tool.py
View File

@@ -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)
return super()._run(query=query)