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

cutring new version with improved docs search

Browse Source
This commit is contained in:
João Moura
2024-05-06 14:02:09 -03:00
parent 037e80c6a3
commit 7c6b1249fb
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/crewai_tools/tools/docx_search_tool/docx_search_tool.py
View File

@@ -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)