TBNilles/LEANN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Update MCP RAG classes to match BaseRAGExample signature

Browse Source 
- Fix SlackMCPRAG and TwitterMCPRAG __init__ methods to provide required parameters
- Add name, description, and default_index_name to super().__init__ calls
- Resolves test failures: test_slack_rag_initialization and test_twitter_rag_initialization

This fixes the TypeError caused by BaseRAGExample requiring additional parameters.
This commit is contained in:
aakash
2025-10-06 13:57:07 -07:00
parent fe904ec992
commit 99bb98748d
2 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
apps/slack_rag.py
View File

@@ -25,8 +25,11 @@ class SlackMCPRAG(BaseRAGExample):
""" """
def __init__(self): def __init__(self):
super().__init__() super().__init__(
self.default_index_name = "slack_messages" name="Slack MCP RAG",
description="RAG application for Slack messages via MCP servers",
default_index_name="slack_messages"
)
def _add_specific_arguments(self, parser: argparse.ArgumentParser): def _add_specific_arguments(self, parser: argparse.ArgumentParser):
"""Add Slack MCP-specific arguments.""" """Add Slack MCP-specific arguments."""

7
apps/twitter_rag.py
View File

@@ -25,8 +25,11 @@ class TwitterMCPRAG(BaseRAGExample):
""" """
def __init__(self): def __init__(self):
super().__init__() super().__init__(
self.default_index_name = "twitter_bookmarks" name="Twitter MCP RAG",
description="RAG application for Twitter bookmarks via MCP servers",
default_index_name="twitter_bookmarks"
)
def _add_specific_arguments(self, parser: argparse.ArgumentParser): def _add_specific_arguments(self, parser: argparse.ArgumentParser):
"""Add Twitter MCP-specific arguments.""" """Add Twitter MCP-specific arguments."""