From 7c6b1249fb285f41fb64d3918d2ca6f7259ad98c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Mon, 6 May 2024 14:02:09 -0300 Subject: [PATCH] cutring new version with improved docs search --- .../tools/docx_search_tool/docx_search_tool.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/src/crewai_tools/tools/docx_search_tool/docx_search_tool.py b/src/crewai_tools/tools/docx_search_tool/docx_search_tool.py index e6f5b2d55..ed2d4daad 100644 --- a/src/crewai_tools/tools/docx_search_tool/docx_search_tool.py +++ b/src/crewai_tools/tools/docx_search_tool/docx_search_tool.py @@ -8,18 +8,18 @@ from ..rag.rag_tool import RagTool class FixedDOCXSearchToolSchema(BaseModel): """Input for DOCXSearchTool.""" - + docx: str = Optional[Field](..., description="Mandatory docx path you want to search") search_query: str = Field( ..., description="Mandatory search query you want to use to search the DOCX's content", ) - class DOCXSearchToolSchema(FixedDOCXSearchToolSchema): """Input for DOCXSearchTool.""" - - docx: str = Field(..., description="Mandatory docx path you want to search") - + search_query: str = Field( + ..., + description="Mandatory search query you want to use to search the DOCX's content", + ) class DOCXSearchTool(RagTool): name: str = "Search a DOCX's content" @@ -54,7 +54,13 @@ class DOCXSearchTool(RagTool): def _run( self, - search_query: str, **kwargs: Any, ) -> Any: + search_query = kwargs.get('search_query') + if search_query is None: + search_query = kwargs.get('query') + + docx = kwargs.get("docx") + if docx is not None: + self.add(docx) return super()._run(query=search_query)