awslabs-core-mcp-server

import asyncio import websockets from awslabs_core_mcp_server.model_context_protocol import InMemoryModelContextProtocolServer async def run_server(): # Example initial data (can be empty or loaded from a file) initial_data = { "context_id_1": {"model_id": "model_A", "context": {"key": "value"}} } # Initialize the in-memory MCP server with optional initial data server_instance = InMemoryModelContextProtocolServer(initial_data=initial_data) # Start the WebSocket server port = 8080 host = "localhost" print(f"Starting MCP server on ws://{host}:{port}") async with websockets.serve(server_instance.handle_request, host, port): await asyncio.Future() # Run forever if __name__ == "__main__": try: asyncio.run(run_server()) except KeyboardInterrupt: print("\nServer stopped gracefully.")