Merge pull request #58 from strnad/fix-ragtool-parameters

fix: Ensure tools handle parameters passed post-creation correctly
This commit is contained in:
João Moura
2024-07-14 12:08:11 -07:00
committed by GitHub
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)