Fix RAG tools

2026-05-01 15:22:37 +00:00 · 2024-04-04 13:42:30 -03:00
parent 80f9613959
commit 4e9709b8fb
13 changed files with 92 additions and 1 deletions
--- a/src/crewai_tools/tools/pg_seach_tool/pg_search_tool.py
+++ b/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)