andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 23:32:39 +00:00
Code Issues Actions 6 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): class FixedPDFSearchToolSchema(BaseModel):
"""Input for PDFSearchTool.""" """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): class PDFSearchToolSchema(FixedPDFSearchToolSchema):
"""Input for PDFSearchTool.""" """Input for PDFSearchTool."""
@@ -32,10 +32,10 @@ class PDFSearchTool(RagTool):
def _run( def _run(
self, self,
search_query: str, query: str,
**kwargs: Any, **kwargs: Any,
) -> Any: ) -> Any:
pdf = kwargs.get('pdf', self.pdf) pdf = kwargs.get('pdf', self.pdf)
self.app = App() self.app = App()
self.app.add(pdf, data_type=DataType.PDF_FILE) self.app.add(pdf, data_type=DataType.PDF_FILE)
return super()._run(query=search_query) return super()._run(query=query)