mcp-proxy 0.2.1__py3-none-any.whl

mcp_proxy/__init__.py

@@ -0,0 +1,143 @@
+"""Create a local server that proxies requests to a remote server over SSE."""
+
+import logging
+import typing as t
+
+from mcp import server, types
+from mcp.client.session import ClientSession
+
+logger = logging.getLogger(__name__)
+
+
+async def create_proxy_server(remote_app: ClientSession) -> server.Server:  # noqa: C901
+    """Create a server instance from a remote app."""
+    response = await remote_app.initialize()
+    capabilities = response.capabilities
+
+    app = server.Server(response.serverInfo.name)
+
+    if capabilities.prompts:
+
+        async def _list_prompts(_: t.Any) -> types.ServerResult:  # noqa: ANN401
+            result = await remote_app.list_prompts()
+            return types.ServerResult(result)
+
+        app.request_handlers[types.ListPromptsRequest] = _list_prompts
+
+        async def _get_prompt(req: types.GetPromptRequest) -> types.ServerResult:
+            result = await remote_app.get_prompt(req.params.name, req.params.arguments)
+            return types.ServerResult(result)
+
+        app.request_handlers[types.GetPromptRequest] = _get_prompt
+
+    if capabilities.resources:
+
+        async def _list_resources(_: t.Any) -> types.ServerResult:  # noqa: ANN401
+            result = await remote_app.list_resources()
+            return types.ServerResult(result)
+
+        app.request_handlers[types.ListResourcesRequest] = _list_resources
+
+        # list_resource_templates() is not implemented in the client
+        # async def _list_resource_templates(_: t.Any) -> types.ServerResult:
+        #     result = await remote_app.list_resource_templates()
+        #     return types.ServerResult(result)
+
+        # app.request_handlers[types.ListResourceTemplatesRequest] = _list_resource_templates
+
+        async def _read_resource(req: types.ReadResourceRequest) -> types.ServerResult:
+            result = await remote_app.read_resource(req.params.uri)
+            return types.ServerResult(result)
+
+        app.request_handlers[types.ReadResourceRequest] = _read_resource
+
+    if capabilities.logging:
+
+        async def _set_logging_level(req: types.SetLevelRequest) -> types.ServerResult:
+            await remote_app.set_logging_level(req.params.level)
+            return types.ServerResult(types.EmptyResult())
+
+        app.request_handlers[types.SetLevelRequest] = _set_logging_level
+
+    if capabilities.resources:
+
+        async def _subscribe_resource(req: types.SubscribeRequest) -> types.ServerResult:
+            await remote_app.subscribe_resource(req.params.uri)
+            return types.ServerResult(types.EmptyResult())
+
+        app.request_handlers[types.SubscribeRequest] = _subscribe_resource
+
+        async def _unsubscribe_resource(req: types.UnsubscribeRequest) -> types.ServerResult:
+            await remote_app.unsubscribe_resource(req.params.uri)
+            return types.ServerResult(types.EmptyResult())
+
+        app.request_handlers[types.UnsubscribeRequest] = _unsubscribe_resource
+
+    if capabilities.tools:
+
+        async def _list_tools(_: t.Any) -> types.ServerResult:  # noqa: ANN401
+            tools = await remote_app.list_tools()
+            return types.ServerResult(tools)
+
+        app.request_handlers[types.ListToolsRequest] = _list_tools
+
+        async def _call_tool(req: types.CallToolRequest) -> types.ServerResult:
+            try:
+                result = await remote_app.call_tool(
+                    req.params.name,
+                    (req.params.arguments or {}),
+                )
+                return types.ServerResult(result)
+            except Exception as e:  # noqa: BLE001
+                return types.ServerResult(
+                    types.CallToolResult(
+                        content=[types.TextContent(type="text", text=str(e))],
+                        isError=True,
+                    ),
+                )
+
+        app.request_handlers[types.CallToolRequest] = _call_tool
+
+    async def _send_progress_notification(req: types.ProgressNotification) -> None:
+        await remote_app.send_progress_notification(
+            req.params.progressToken,
+            req.params.progress,
+            req.params.total,
+        )
+
+    app.notification_handlers[types.ProgressNotification] = _send_progress_notification
+
+    async def _complete(req: types.CompleteRequest) -> types.ServerResult:
+        result = await remote_app.complete(
+            req.params.ref,
+            req.params.argument.model_dump(),
+        )
+        return types.ServerResult(result)
+
+    app.request_handlers[types.CompleteRequest] = _complete
+
+    return app
+
+
+async def run_sse_client(url: str, api_access_token: str | None = None) -> None:
+    """Run the SSE client.
+
+    Args:
+        url: The URL to connect to.
+        api_access_token: The API access token to use for authentication.
+
+    """
+    from mcp.client.sse import sse_client
+
+    headers = {}
+    if api_access_token is not None:
+        headers["Authorization"] = f"Bearer {api_access_token}"
+
+    async with sse_client(url=url, headers=headers) as streams, ClientSession(*streams) as session:
+        app = await create_proxy_server(session)
+        async with server.stdio_server() as (read_stream, write_stream):
+            await app.run(
+                read_stream,
+                write_stream,
+                app.create_initialization_options(),
+            )

mcp_proxy/__main__.py

@@ -0,0 +1,30 @@
+"""The entry point for the mcp-proxy application. It sets up the logging and runs the main function.
+
+Two ways to run the application:
+1. Run the application as a module `uv run -m mcp_proxy`
+2. Run the application as a package `uv run mcp-proxy`
+
+"""
+
+import asyncio
+import logging
+import os
+import typing as t
+
+from . import run_sse_client
+
+logging.basicConfig(level=logging.DEBUG)
+SSE_URL: t.Final[str] = os.getenv("SSE_URL", "")
+API_ACCESS_TOKEN: t.Final[str | None] = os.getenv("API_ACCESS_TOKEN", None)
+
+if not SSE_URL:
+    raise ValueError("SSE_URL environment variable is not set")
+
+
+def main() -> None:
+    """Start the client using asyncio."""
+    asyncio.run(run_sse_client(SSE_URL, api_access_token=API_ACCESS_TOKEN))
+
+
+if __name__ == "__main__":
+    main()

mcp_proxy-0.2.1.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Sergey Parfenyuk
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

mcp_proxy-0.2.1.dist-info/METADATA

@@ -0,0 +1,7 @@
+Metadata-Version: 2.1
+Name: mcp-proxy
+Version: 0.2.1
+Author-email: Sergey Parfenyuk <sergey.parfenyuk@gmail.com>
+Requires-Python: >=3.11
+License-File: LICENSE
+Requires-Dist: mcp

mcp_proxy-0.2.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+mcp_proxy/__init__.py,sha256=-vnAAZ56FcdkI1TKI5Ync8lojKHHy__s2CLWTYS7_no,5319
+mcp_proxy/__main__.py,sha256=7IKy0orhu67F9-lsPCps9CqbZ3kVN4_i-BBAFA1t34E,770
+mcp_proxy/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mcp_proxy-0.2.1.dist-info/LICENSE,sha256=IB5Kq1DC-6sU8yGp_Hzv6SXBKRrlSbYcKCLNkciNBUg,1073
+mcp_proxy-0.2.1.dist-info/METADATA,sha256=60-QQSE3yg5O2IxWgt67l3CSHSgA-O5zNwWLeBUrhEc,178
+mcp_proxy-0.2.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+mcp_proxy-0.2.1.dist-info/entry_points.txt,sha256=6wHxBfNcv1bdXrB27vVQ3RdvpGlP1a-qx3hviRXwCAo,54
+mcp_proxy-0.2.1.dist-info/top_level.txt,sha256=Sc_nmlR8kpkoOyksgzzlMw7rdAn5ZsPfRsd_tVWgiBI,10
+mcp_proxy-0.2.1.dist-info/RECORD,,

mcp_proxy-0.2.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.6.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

mcp_proxy-0.2.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+mcp-proxy = mcp_proxy.__main__:main

mcp_proxy-0.2.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+mcp_proxy