Merge pull request #21 from joaomdmoura/gui/fix-rag-tools

Fix RAG tools

src/crewai_tools/tools/code_docs_search_tool/code_docs_search_tool.py

@@ -50,3 +50,10 @@ class CodeDocsSearchTool(RagTool):
     ) -> Any:
         if "docs_url" in kwargs:
             self.add(kwargs["docs_url"])
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)

src/crewai_tools/tools/csv_search_tool/csv_search_tool.py

@@ -50,3 +50,10 @@ class CSVSearchTool(RagTool):
     ) -> Any:
         if "csv" in kwargs:
             self.add(kwargs["csv"])
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)

src/crewai_tools/tools/directory_search_tool/directory_search_tool.py

@@ -50,3 +50,10 @@ class DirectorySearchTool(RagTool):
     ) -> Any:
         if "directory" in kwargs:
             self.add(kwargs["directory"])
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)

src/crewai_tools/tools/docx_search_tool/docx_search_tool.py

@@ -50,3 +50,10 @@ class DOCXSearchTool(RagTool):
     ) -> Any:
         if "docx" in kwargs:
             self.add(kwargs["docx"])
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)

src/crewai_tools/tools/github_search_tool/github_search_tool.py

@@ -21,7 +21,7 @@ class GithubSearchToolSchema(FixedGithubSearchToolSchema):
     github_repo: str = Field(..., description="Mandatory github you want to search")
     content_types: List[str] = Field(
         ...,
-        description="Mandatory content types you want to be inlcuded search, options: [code, repo, pr, issue]",
+        description="Mandatory content types you want to be included search, options: [code, repo, pr, issue]",
     )
 
 
@@ -56,3 +56,10 @@ class GithubSearchTool(RagTool):
     ) -> Any:
         if "github_repo" in kwargs:
             self.add(kwargs["github_repo"])
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)

src/crewai_tools/tools/json_search_tool/json_search_tool.py

@@ -50,3 +50,10 @@ class JSONSearchTool(RagTool):
     ) -> Any:
         if "json_path" in kwargs:
             self.add(kwargs["json_path"])
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)

src/crewai_tools/tools/mdx_seach_tool/mdx_search_tool.py

@@ -50,3 +50,10 @@ class MDXSearchTool(RagTool):
     ) -> Any:
         if "mdx" in kwargs:
             self.add(kwargs["mdx"])
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)

src/crewai_tools/tools/pg_seach_tool/pg_search_tool.py

@@ -35,3 +35,10 @@ class PGSearchTool(RagTool):
         kwargs["data_type"] = "postgres"
         kwargs["loader"] = PostgresLoader(config=dict(url=self.db_uri))
         super().add(f"SELECT * FROM {table_name};", **kwargs)
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)

src/crewai_tools/tools/txt_search_tool/txt_search_tool.py

@@ -50,3 +50,10 @@ class TXTSearchTool(RagTool):
     ) -> Any:
         if "txt" in kwargs:
             self.add(kwargs["txt"])
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)

src/crewai_tools/tools/website_search/website_search_tool.py

@@ -50,3 +50,10 @@ class WebsiteSearchTool(RagTool):
     ) -> Any:
         if "website" in kwargs:
             self.add(kwargs["website"])
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)

src/crewai_tools/tools/xml_search_tool/xml_search_tool.py

@@ -50,3 +50,10 @@ class XMLSearchTool(RagTool):
     ) -> Any:
         if "xml" in kwargs:
             self.add(kwargs["xml"])
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)

src/crewai_tools/tools/youtube_channel_search_tool/youtube_channel_search_tool.py

@@ -53,3 +53,10 @@ class YoutubeChannelSearchTool(RagTool):
     ) -> Any:
         if "youtube_channel_handle" in kwargs:
             self.add(kwargs["youtube_channel_handle"])
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)

src/crewai_tools/tools/youtube_video_search_tool/youtube_video_search_tool.py

@@ -50,3 +50,10 @@ class YoutubeVideoSearchTool(RagTool):
     ) -> Any:
         if "youtube_video_url" in kwargs:
             self.add(kwargs["youtube_video_url"])
+
+    def _run(
+        self,
+        search_query: str,
+        **kwargs: Any,
+    ) -> Any:
+        return super()._run(query=search_query)