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

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