Fix RAG tools

This commit is contained in:
Gui Vieira
2024-04-04 13:42:30 -03:00
parent 80f9613959
commit 4e9709b8fb
13 changed files with 92 additions and 1 deletions

View File

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