fix: Apply pre-commit hooks formatting fixes

- Remove trailing whitespace from all files
- Fix ruff formatting issues (2 errors resolved)
- Apply consistent code formatting across 3 files
- Ensure all files pass pre-commit validation

This resolves all CI formatting failures.
This commit is contained in:
aakash
2025-10-06 13:47:16 -07:00
parent d2432b45f6
commit fe904ec992
3 changed files with 103 additions and 108 deletions

View File

@@ -19,7 +19,6 @@ from pathlib import Path
sys.path.append(str(Path(__file__).parent.parent)) sys.path.append(str(Path(__file__).parent.parent))
async def demo_slack_mcp(): async def demo_slack_mcp():
"""Demonstrate Slack MCP integration.""" """Demonstrate Slack MCP integration."""
print("=" * 60) print("=" * 60)

View File

@@ -13,8 +13,8 @@ from pathlib import Path
sys.path.append(str(Path(__file__).parent.parent)) sys.path.append(str(Path(__file__).parent.parent))
from apps.slack_data.slack_mcp_reader import SlackMCPReader from apps.slack_data.slack_mcp_reader import SlackMCPReader
from apps.twitter_data.twitter_mcp_reader import TwitterMCPReader
from apps.slack_rag import SlackMCPRAG from apps.slack_rag import SlackMCPRAG
from apps.twitter_data.twitter_mcp_reader import TwitterMCPReader
from apps.twitter_rag import TwitterMCPRAG from apps.twitter_rag import TwitterMCPRAG
@@ -33,7 +33,7 @@ def test_slack_reader_initialization():
"custom-slack-server", "custom-slack-server",
workspace_name="test-workspace", workspace_name="test-workspace",
concatenate_conversations=False, concatenate_conversations=False,
max_messages_per_conversation=50 max_messages_per_conversation=50,
) )
assert reader.workspace_name == "test-workspace" assert reader.workspace_name == "test-workspace"
assert not reader.concatenate_conversations assert not reader.concatenate_conversations
@@ -59,7 +59,7 @@ def test_twitter_reader_initialization():
username="testuser", username="testuser",
include_tweet_content=False, include_tweet_content=False,
include_metadata=False, include_metadata=False,
max_bookmarks=500 max_bookmarks=500,
) )
assert reader.username == "testuser" assert reader.username == "testuser"
assert not reader.include_tweet_content assert not reader.include_tweet_content
@@ -80,7 +80,7 @@ def test_slack_message_formatting():
"text": "Hello, world!", "text": "Hello, world!",
"user": "john_doe", "user": "john_doe",
"channel": "general", "channel": "general",
"ts": "1234567890.123456" "ts": "1234567890.123456",
} }
formatted = reader._format_message(message) formatted = reader._format_message(message)
@@ -110,7 +110,7 @@ def test_twitter_bookmark_formatting():
"created_at": "2024-01-01T12:00:00Z", "created_at": "2024-01-01T12:00:00Z",
"url": "https://twitter.com/ai_researcher/status/123456789", "url": "https://twitter.com/ai_researcher/status/123456789",
"likes": 42, "likes": 42,
"retweets": 15 "retweets": 15,
} }
formatted = reader._format_bookmark(bookmark) formatted = reader._format_bookmark(bookmark)
@@ -137,7 +137,7 @@ def test_slack_rag_initialization():
app = SlackMCPRAG() app = SlackMCPRAG()
assert app.default_index_name == "slack_messages" assert app.default_index_name == "slack_messages"
assert hasattr(app, 'parser') assert hasattr(app, "parser")
print("✅ SlackMCPRAG initialization tests passed") print("✅ SlackMCPRAG initialization tests passed")
@@ -148,7 +148,7 @@ def test_twitter_rag_initialization():
app = TwitterMCPRAG() app = TwitterMCPRAG()
assert app.default_index_name == "twitter_bookmarks" assert app.default_index_name == "twitter_bookmarks"
assert hasattr(app, 'parser') assert hasattr(app, "parser")
print("✅ TwitterMCPRAG initialization tests passed") print("✅ TwitterMCPRAG initialization tests passed")
@@ -162,7 +162,7 @@ def test_concatenated_content_creation():
messages = [ messages = [
{"text": "First message", "user": "alice", "ts": "1000"}, {"text": "First message", "user": "alice", "ts": "1000"},
{"text": "Second message", "user": "bob", "ts": "2000"}, {"text": "Second message", "user": "bob", "ts": "2000"},
{"text": "Third message", "user": "charlie", "ts": "3000"} {"text": "Third message", "user": "charlie", "ts": "3000"},
] ]
content = reader._create_concatenated_content(messages, "general") content = reader._create_concatenated_content(messages, "general")

View File

@@ -6,8 +6,8 @@ This script tests the basic functionality of the MCP readers
without requiring LEANN core dependencies. without requiring LEANN core dependencies.
""" """
import sys
import json import json
import sys
from pathlib import Path from pathlib import Path
# Add the parent directory to the path so we can import from apps # Add the parent directory to the path so we can import from apps
@@ -30,7 +30,7 @@ def test_slack_reader_basic():
"text": "Hello team! How's the project going?", "text": "Hello team! How's the project going?",
"user": "john_doe", "user": "john_doe",
"channel": "general", "channel": "general",
"ts": "1234567890.123456" "ts": "1234567890.123456",
} }
formatted = reader._format_message(message) formatted = reader._format_message(message)
@@ -41,7 +41,7 @@ def test_slack_reader_basic():
# Test concatenated content creation # Test concatenated content creation
messages = [ messages = [
{"text": "First message", "user": "alice", "ts": "1000"}, {"text": "First message", "user": "alice", "ts": "1000"},
{"text": "Second message", "user": "bob", "ts": "2000"} {"text": "Second message", "user": "bob", "ts": "2000"},
] ]
content = reader._create_concatenated_content(messages, "dev-team") content = reader._create_concatenated_content(messages, "dev-team")
@@ -74,7 +74,7 @@ def test_twitter_reader_basic():
"retweets": 42, "retweets": 42,
"replies": 23, "replies": 23,
"hashtags": ["AI", "tech", "future"], "hashtags": ["AI", "tech", "future"],
"mentions": ["@openai", "@anthropic"] "mentions": ["@openai", "@anthropic"],
} }
formatted = reader._format_bookmark(bookmark) formatted = reader._format_bookmark(bookmark)
@@ -108,8 +108,8 @@ def test_mcp_request_format():
"params": { "params": {
"protocolVersion": "2024-11-05", "protocolVersion": "2024-11-05",
"capabilities": {}, "capabilities": {},
"clientInfo": {"name": "leann-slack-reader", "version": "1.0.0"} "clientInfo": {"name": "leann-slack-reader", "version": "1.0.0"},
} },
} }
# Verify it's valid JSON # Verify it's valid JSON
@@ -120,12 +120,7 @@ def test_mcp_request_format():
assert parsed["params"]["protocolVersion"] == "2024-11-05" assert parsed["params"]["protocolVersion"] == "2024-11-05"
# Test tools/list request # Test tools/list request
list_request = { list_request = {"jsonrpc": "2.0", "id": 2, "method": "tools/list", "params": {}}
"jsonrpc": "2.0",
"id": 2,
"method": "tools/list",
"params": {}
}
json_str = json.dumps(list_request) json_str = json.dumps(list_request)
parsed = json.loads(json_str) parsed = json.loads(json_str)
@@ -147,7 +142,7 @@ def test_data_processing():
messages_with_timestamps = [ messages_with_timestamps = [
{"text": "Meeting in 5 minutes", "user": "alice", "ts": "1000.123"}, {"text": "Meeting in 5 minutes", "user": "alice", "ts": "1000.123"},
{"text": "On my way!", "user": "bob", "ts": "1001.456"}, {"text": "On my way!", "user": "bob", "ts": "1001.456"},
{"text": "Starting now", "user": "charlie", "ts": "1002.789"} {"text": "Starting now", "user": "charlie", "ts": "1002.789"},
] ]
content = slack_reader._create_concatenated_content(messages_with_timestamps, "meetings") content = slack_reader._create_concatenated_content(messages_with_timestamps, "meetings")
@@ -163,7 +158,7 @@ def test_data_processing():
"author": "startup_founder", "author": "startup_founder",
"likes": 1250, "likes": 1250,
"retweets": 340, "retweets": 340,
"replies": 89 "replies": 89,
} }
formatted = twitter_reader._format_bookmark(high_engagement_bookmark) formatted = twitter_reader._format_bookmark(high_engagement_bookmark)
@@ -217,6 +212,7 @@ def main():
except Exception as e: except Exception as e:
print(f"\n❌ Test failed: {e}") print(f"\n❌ Test failed: {e}")
import traceback import traceback
traceback.print_exc() traceback.print_exc()
sys.exit(1) sys.exit(1)