fix: Ensure tools handle parameters passed post-creation correctly (#3)

- Fixed an issue where multiple tools failed to function if parameters were provided after tool creation.
- Updated tools to correctly process source file/URL passed by the agent post-creation as per documentation.

Closes #<47>
This commit is contained in:
Jakub Strnad
2024-06-21 15:06:17 +02:00
committed by GitHub
parent bd13b55afd
commit 2f80840c74
13 changed files with 13 additions and 13 deletions

View File

@@ -57,4 +57,4 @@ class CodeDocsSearchTool(RagTool):
search_query: str,
**kwargs: Any,
) -> Any:
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)

View File

@@ -57,4 +57,4 @@ class CSVSearchTool(RagTool):
search_query: str,
**kwargs: Any,
) -> Any:
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)

View File

@@ -57,4 +57,4 @@ class DirectorySearchTool(RagTool):
search_query: str,
**kwargs: Any,
) -> Any:
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)

View File

@@ -63,4 +63,4 @@ class DOCXSearchTool(RagTool):
docx = kwargs.get("docx")
if docx is not None:
self.add(docx)
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)

View File

@@ -68,4 +68,4 @@ class GithubSearchTool(RagTool):
search_query: str,
**kwargs: Any,
) -> Any:
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)

View File

@@ -57,4 +57,4 @@ class JSONSearchTool(RagTool):
search_query: str,
**kwargs: Any,
) -> Any:
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)

View File

@@ -57,4 +57,4 @@ class MDXSearchTool(RagTool):
search_query: str,
**kwargs: Any,
) -> Any:
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)

View File

@@ -41,4 +41,4 @@ class PGSearchTool(RagTool):
search_query: str,
**kwargs: Any,
) -> Any:
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)

View File

@@ -57,4 +57,4 @@ class TXTSearchTool(RagTool):
search_query: str,
**kwargs: Any,
) -> Any:
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)

View File

@@ -57,4 +57,4 @@ class WebsiteSearchTool(RagTool):
search_query: str,
**kwargs: Any,
) -> Any:
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)

View File

@@ -57,4 +57,4 @@ class XMLSearchTool(RagTool):
search_query: str,
**kwargs: Any,
) -> Any:
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)

View File

@@ -60,4 +60,4 @@ class YoutubeChannelSearchTool(RagTool):
search_query: str,
**kwargs: Any,
) -> Any:
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)

View File

@@ -57,4 +57,4 @@ class YoutubeVideoSearchTool(RagTool):
search_query: str,
**kwargs: Any,
) -> Any:
return super()._run(query=search_query)
return super()._run(query=search_query, **kwargs)