awslabs.valkey-mcp-server 0.1.1__tar.gz → 1.0.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (48) hide show
  1. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/Dockerfile +2 -2
  2. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/ELASTICACHECONNECT.md +4 -0
  3. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/PKG-INFO +3 -2
  4. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/README.md +2 -1
  5. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/main.py +4 -25
  6. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/pyproject.toml +2 -2
  7. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_main.py +0 -18
  8. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/.gitignore +0 -0
  9. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/.pre-commit-config.yaml +0 -0
  10. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/.python-version +0 -0
  11. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/CHANGELOG.md +0 -0
  12. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/LICENSE +0 -0
  13. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/NOTICE +0 -0
  14. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/__init__.py +0 -0
  15. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/__init__.py +0 -0
  16. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/common/__init__.py +0 -0
  17. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/common/config.py +0 -0
  18. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/common/connection.py +0 -0
  19. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/common/server.py +0 -0
  20. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/tools/__init__.py +0 -0
  21. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/tools/bitmap.py +0 -0
  22. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/tools/hash.py +0 -0
  23. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/tools/hyperloglog.py +0 -0
  24. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/tools/json.py +0 -0
  25. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/tools/list.py +0 -0
  26. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/tools/misc.py +0 -0
  27. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/tools/server_management.py +0 -0
  28. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/tools/set.py +0 -0
  29. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/tools/sorted_set.py +0 -0
  30. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/tools/stream.py +0 -0
  31. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/tools/string.py +0 -0
  32. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/awslabs/valkey_mcp_server/version.py +0 -0
  33. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/docker-healthcheck.sh +0 -0
  34. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_bitmap.py +0 -0
  35. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_config.py +0 -0
  36. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_connection.py +0 -0
  37. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_hash.py +0 -0
  38. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_hyperloglog.py +0 -0
  39. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_init.py +0 -0
  40. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_json.py +0 -0
  41. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_list.py +0 -0
  42. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_server_management.py +0 -0
  43. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_set.py +0 -0
  44. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_sorted_set.py +0 -0
  45. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_sorted_set_additional.py +0 -0
  46. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_stream.py +0 -0
  47. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/tests/test_string.py +0 -0
  48. {awslabs_valkey_mcp_server-0.1.1 → awslabs_valkey_mcp_server-1.0.0}/uv.lock +0 -0
@@ -10,7 +10,7 @@
10
10
  # and limitations under the License.
11
11
 
12
12
  #FROM public.ecr.aws/sam/build-python3.10:1.137.1-20250411084548
13
- FROM public.ecr.aws/sam/build-python3.10@sha256:04cdbe84bec08d17d621192bc3f0a9e4a85a83f2ac99aa9241659dfac0d845ea AS uv
13
+ FROM public.ecr.aws/sam/build-python3.10@sha256:e78695db10ca8cb129e59e30f7dc9789b0dbd0181dba195d68419c72bac51ac1 AS uv
14
14
 
15
15
  # Install the project into `/app`
16
16
  WORKDIR /app
@@ -44,7 +44,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
44
44
  # Make the directory just in case it doesn't exist
45
45
  RUN mkdir -p /root/.local
46
46
 
47
- FROM public.ecr.aws/sam/build-python3.10@sha256:04cdbe84bec08d17d621192bc3f0a9e4a85a83f2ac99aa9241659dfac0d845ea
47
+ FROM public.ecr.aws/sam/build-python3.10@sha256:e78695db10ca8cb129e59e30f7dc9789b0dbd0181dba195d68419c72bac51ac1
48
48
 
49
49
  # Place executables in the environment at the front of the path and include other binaries
50
50
  ENV PATH="/app/.venv/bin:$PATH:/usr/sbin"
@@ -1,5 +1,9 @@
1
1
  # How to connect to an Amazon ElastiCache Valkey datastore
2
2
 
3
+ Your Amazon ElastiCache for Valkey datastores are designed to be accessed through an Amazon EC2 instance. You can access your ElastiCache for Valkey datastores from an Amazon EC2 instance in the same Amazon VPC, or by using VPC peering, you can access your ElastiCache for Valkey datastores from an Amazon EC2 in a different Amazon VPC.
4
+
5
+ The following instructions will help you create an EC2 instance in the same VPC as your ElastiCache for Valkey datastore, and will guide you to configure the security groups required to access the cache from your desktop through an SSH tunnel.
6
+
3
7
  ## Launch and configure the EC2 instance
4
8
 
5
9
  Complete the following steps:
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: awslabs.valkey-mcp-server
3
- Version: 0.1.1
3
+ Version: 1.0.0
4
4
  Summary: An AWS Labs Model Context Protocol (MCP) server for valkey
5
5
  Project-URL: homepage, https://awslabs.github.io/mcp/
6
6
  Project-URL: docs, https://awslabs.github.io/mcp/servers/valkey-mcp-server/
@@ -55,7 +55,8 @@ This MCP server provides tools to operate on Valkey data types. For example, it
55
55
 
56
56
  1. Install `uv` from [Astral](https://docs.astral.sh/uv/getting-started/installation/) or the [GitHub README](https://github.com/astral-sh/uv#installation)
57
57
  2. Install Python using `uv python install 3.10`
58
- 3. Access to a Valkey datastore instance. For instructions to connect to an Amazon ElastiCache/MemoryDB Valkey cache [click here](ELASTICACHECONNECT.md).
58
+ 3. Access to a Valkey datastore.
59
+ 4. For instructions to connect to an Amazon ElastiCache/MemoryDB Valkey datastore [click here](https://github.com/awslabs/mcp/blob/main/src/valkey-mcp-server/ELASTICACHECONNECT.md).
59
60
 
60
61
 
61
62
  ## Installation
@@ -24,7 +24,8 @@ This MCP server provides tools to operate on Valkey data types. For example, it
24
24
 
25
25
  1. Install `uv` from [Astral](https://docs.astral.sh/uv/getting-started/installation/) or the [GitHub README](https://github.com/astral-sh/uv#installation)
26
26
  2. Install Python using `uv python install 3.10`
27
- 3. Access to a Valkey datastore instance. For instructions to connect to an Amazon ElastiCache/MemoryDB Valkey cache [click here](ELASTICACHECONNECT.md).
27
+ 3. Access to a Valkey datastore.
28
+ 4. For instructions to connect to an Amazon ElastiCache/MemoryDB Valkey datastore [click here](https://github.com/awslabs/mcp/blob/main/src/valkey-mcp-server/ELASTICACHECONNECT.md).
28
29
 
29
30
 
30
31
  ## Installation
@@ -11,7 +11,6 @@
11
11
 
12
12
  """awslabs valkey MCP Server implementation."""
13
13
 
14
- import argparse
15
14
  from awslabs.valkey_mcp_server.common.server import mcp
16
15
  from awslabs.valkey_mcp_server.tools import (
17
16
  bitmap, # noqa: F401
@@ -44,39 +43,19 @@ async def health_check(request):
44
43
  class ValkeyMCPServer:
45
44
  """Valkey MCP Server wrapper."""
46
45
 
47
- def __init__(self, sse=False, port=8888):
48
- """Initialize MCP Server wrapper.
49
-
50
- Args:
51
- sse: Whether to use SSE transport
52
- port: Port to run the server on (default: 8888)
53
- """
54
- self.sse = sse
55
- self.port = port
46
+ def __init__(self):
47
+ """Initialize MCP Server wrapper."""
56
48
 
57
49
  def run(self):
58
50
  """Run server with appropriate transport."""
59
- if self.sse:
60
- if self.port is not None:
61
- mcp.settings.port = self.port
62
- mcp.run(transport='sse')
63
- else:
64
- mcp.run()
51
+ mcp.run()
65
52
 
66
53
 
67
54
  def main():
68
55
  """Run the MCP server with CLI argument support."""
69
- parser = argparse.ArgumentParser(
70
- description='An AWS Labs Model Context Protocol (MCP) server for valkey'
71
- )
72
- parser.add_argument('--sse', action='store_true', help='Use SSE transport')
73
- parser.add_argument('--port', type=int, default=8888, help='Port to run the server on')
74
-
75
- args = parser.parse_args()
76
-
77
56
  logger.info('Amazon ElastiCache/MemoryDB Valkey MCP Server Started...')
78
57
 
79
- server = ValkeyMCPServer(args.sse, args.port)
58
+ server = ValkeyMCPServer()
80
59
  server.run()
81
60
 
82
61
 
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "awslabs.valkey-mcp-server"
3
- version = "0.1.1"
3
+ version = "1.0.0"
4
4
  description = "An AWS Labs Model Context Protocol (MCP) server for valkey"
5
5
  readme = "README.md"
6
6
  requires-python = ">=3.10"
@@ -38,7 +38,7 @@ repository = "https://github.com/awslabs/mcp.git"
38
38
  changelog = "https://github.com/awslabs/mcp/blob/main/src/valkey-mcp-server/CHANGELOG.md"
39
39
 
40
40
  [project.scripts]
41
- "awslabs.valkey-mcp-server" = "awslabs.valkey_mcp_server.server:main"
41
+ "awslabs.valkey-mcp-server" = "awslabs.valkey_mcp_server.main:main"
42
42
 
43
43
  [dependency-groups]
44
44
  dev = [
@@ -19,7 +19,6 @@ class TestMain:
19
19
  """Tests for the main function."""
20
20
 
21
21
  @patch('awslabs.valkey_mcp_server.common.server.mcp.run')
22
- @patch('sys.argv', ['awslabs.valkey-mcp-server'])
23
22
  def test_main_default(self, mock_run):
24
23
  """Test main function with default arguments."""
25
24
  # Call the main function
@@ -28,23 +27,6 @@ class TestMain:
28
27
 
29
28
  # Check that mcp.run was called with the correct arguments
30
29
  mock_run.assert_called_once()
31
- assert mock_run.call_args[1].get('transport') is None
32
-
33
- @patch('awslabs.valkey_mcp_server.common.server.mcp.run')
34
- @patch('sys.argv', ['awslabs.valkey-mcp-server', '--sse', '--port', '9999'])
35
- def test_main_sse(self, mock_run):
36
- """Test main function with SSE transport."""
37
- # Call the main function
38
- main()
39
-
40
- # Check that mcp.run was called with the correct arguments
41
- mock_run.assert_called_once()
42
- assert mock_run.call_args[1].get('transport') == 'sse'
43
-
44
- # Check that the port was set correctly
45
- from awslabs.valkey_mcp_server.common.server import mcp
46
-
47
- assert mcp.settings.port == 9999
48
30
 
49
31
  def test_module_execution(self):
50
32
  """Test the module execution when run as __main__."""