Skip to content

Conversation

@im-michaelc
Copy link

@im-michaelc im-michaelc commented Dec 8, 2025

Fixes

Summary

Changes

Please provide a summary of what's being changed

This pull request includes some important improvements to the Keyspaces MCP Server. Internally, we do not have an ASBX-approved MCP server available in the MCP Server Registry available for internal teams to use (including Keyspaces service team).

  • Fix pylint errors/warnings
  • Provide additional test cases to certain API calls/query analyzations
  • Address asyncore module and migrate to asyncio for future suport of Python 3.12 and higher
  • Query scrubbing/cleaning to address AppSec approval process for listing to registry
  • Update README.md to use Pyton uv package manager
  • Created Keyspaces exception class to modularize exception cases
  • Created input classes to improve input validation as per Keyspaces spec

User experience

Please share what the user experience looks like before and after this change

Customer experience will be roughly the same visually, as there are still the same number of MCP tools available for the LLM to use. There are some additional resources provided in the docstring that are picked up and can provide prompts more tailored toward Keyspace/Cassandra infrastructure. Customer's agent running the MCP server should have additional context on inputs/outputs/exceptions and how to interpret edge case responses returned from the server.

Please reach out to [email protected] or @mlynnc on slack for further clarification. Thanks!

Checklist

If your change doesn't seem to apply, please leave them unchecked.

  • I have reviewed the contributing guidelines
  • I have performed a self-review of this change
  • Changes have been tested
  • Changes are documented

Is this a breaking change? (Y/N) N

RFC issue number:

Checklist:

  • Migration process documented
  • Implement warnings (if it can live side by side)

Acknowledgment

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of the project license.

@codecov
Copy link

codecov bot commented Dec 8, 2025

Codecov Report

❌ Patch coverage is 80.95238% with 40 lines in your changes missing coverage. Please review.
✅ Project coverage is 90.47%. Comparing base (ffef17e) to head (8f741fb).

Files with missing lines Patch % Lines
...rver/awslabs/amazon_keyspaces_mcp_server/server.py 66.30% 31 Missing ⚠️
...rver/awslabs/amazon_keyspaces_mcp_server/client.py 86.56% 9 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main    #1908      +/-   ##
==========================================
+ Coverage   90.43%   90.47%   +0.04%     
==========================================
  Files         852      853       +1     
  Lines       63411    63490      +79     
  Branches    10271    10274       +3     
==========================================
+ Hits        57344    57445     +101     
+ Misses       3774     3757      -17     
+ Partials     2293     2288       -5     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
  • 📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.

@im-michaelc im-michaelc changed the title Updates to Keyspaces MCP Server refactor: update Keyspaces MCP Server Dec 8, 2025
@im-michaelc im-michaelc marked this pull request as draft December 9, 2025 18:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

Status: To triage

Development

Successfully merging this pull request may close these issues.

1 participant