codeocean-mcp-server 0.12.1__tar.gz → 0.12.2__tar.gz

{codeocean_mcp_server-0.12.1 → codeocean_mcp_server-0.12.2}/CHANGELOG.md

@@ -1,6 +1,10 @@
 CHANGELOG
 =========
 
+## 0.12.2 (2026-09-03)
+
+- [#42](https://github.com/codeocean/codeocean-mcp-server/pull/42) fix: run synchronous tool bodies in a worker thread
+
 ## 0.12.1 (2026-08-28)
 
 - [#40](https://github.com/codeocean/codeocean-mcp-server/pull/40) fix: apply `LOG_FORMAT` to uvicorn's own log records

{codeocean_mcp_server-0.12.1 → codeocean_mcp_server-0.12.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: codeocean-mcp-server
-Version: 0.12.1
+Version: 0.12.2
 Summary: Code Ocean MCP Server
 Project-URL: Homepage, https://github.com/codeocean/codeocean-mcp-server
 Project-URL: Issues, https://github.com/codeocean/codeocean-mcp-server/issues
@@ -12,6 +12,7 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.10
+Requires-Dist: anyio>=4.5
 Requires-Dist: codeocean<0.15.0,>=0.14.0
 Requires-Dist: mcp<1.24.0,>=1.23.0
 Description-Content-Type: text/markdown

{codeocean_mcp_server-0.12.1 → codeocean_mcp_server-0.12.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codeocean-mcp-server"
-version = "0.12.1"
+version = "0.12.2"
 authors = [{ name = "Code Ocean", email = "dev@codeocean.com" }]
 description = "Code Ocean MCP Server"
 readme = "README.md"
@@ -11,6 +11,7 @@ classifiers = [
   "Operating System :: OS Independent",
 ]
 dependencies = [
+  "anyio>=4.5",
   "codeocean>=0.14.0,<0.15.0",
   "mcp>=1.23.0,<1.24.0",
 ]

{codeocean_mcp_server-0.12.1 → codeocean_mcp_server-0.12.2}/src/codeocean_mcp_server/server.py

@@ -1,6 +1,10 @@
 import argparse
+import functools
+import inspect
 import os
+from collections.abc import Callable
 
+import anyio
 from codeocean import CodeOcean
 from mcp.server.fastmcp import FastMCP
 
@@ -14,6 +18,34 @@ from codeocean_mcp_server.tools import (
 )
 
 
+class ThreadedFastMCP(FastMCP):
+    """A server that runs synchronous tool bodies in a worker thread.
+
+    FastMCP awaits an asynchronous tool but calls a synchronous one on the event loop, where it
+    blocks the whole process for as long as it runs: no other tool call, and no other session's
+    requests, are served meanwhile. The tools here are synchronous and some wait on a computation,
+    so they are handed to a thread instead. anyio copies the context into it, leaving each call's
+    per-request credentials in place.
+    """
+
+    def tool(self, *args, **kwargs) -> Callable[[Callable], Callable]:
+        """Register a tool, moving a synchronous one off the event loop."""
+        register = super().tool(*args, **kwargs)
+
+        def decorator(fn: Callable) -> Callable:
+            if inspect.iscoroutinefunction(fn):
+                return register(fn)
+
+            @functools.wraps(fn)
+            async def in_worker_thread(**arguments):
+                return await anyio.to_thread.run_sync(functools.partial(fn, **arguments))
+
+            register(in_worker_thread)
+            return fn
+
+        return decorator
+
+
 def parse_args(argv: list[str] | None = None) -> argparse.Namespace:
     """Parse command line arguments."""
     parser = argparse.ArgumentParser(prog="codeocean-mcp-server", description="Code Ocean MCP Server")
@@ -40,7 +72,7 @@ def main():
         raise ValueError("Environment variable CODEOCEAN_DOMAIN must be set.")
     agent_id = os.getenv("AGENT_ID", "AI Agent")
 
-    mcp = FastMCP(
+    mcp = ThreadedFastMCP(
         name="Code Ocean",
         instructions=(
             f"MCP server for Code Ocean: search & run capsules, pipelines, and assets using Code Ocean domain {domain}."

{codeocean_mcp_server-0.12.1 → codeocean_mcp_server-0.12.2}/tests/test_http_transport.py

@@ -23,12 +23,17 @@ from mcp_client import get_tools
 
 SERVER_SCRIPT_PATH = str(Path(__file__).parent.parent / "src" / "codeocean_mcp_server" / "server.py")
 
+# How long the stub API takes to answer a token named "slow-...", standing in for a tool that waits.
+SLOW_API_SECONDS = 3.0
+
 
 class _EchoTokenHandler(BaseHTTPRequestHandler):
     """Answer the custom metadata endpoint with the basic-auth user, which is the API token."""
 
     def do_GET(self):  # noqa: D102, N802
         user = base64.b64decode(self.headers["Authorization"].split(" ")[1]).decode().split(":")[0]
+        if user.startswith("slow-"):
+            time.sleep(SLOW_API_SECONDS)
         body = json.dumps({"categories": [user]}).encode()
         self.send_response(200)
         self.send_header("Content-Type", "application/json")
@@ -110,6 +115,23 @@ def test_concurrent_requests_use_their_own_token(http_server):
     assert bob.structuredContent["categories"] == ["bob-token"]
 
 
+def test_a_slow_tool_does_not_hold_up_other_requests(http_server):
+    """A tool call still waiting on Code Ocean leaves the server free to serve another session."""
+    url, _ = http_server
+
+    async def call_and_time(token: str):
+        result = await _call_get_custom_metadata(url, token)
+        return result, time.monotonic()
+
+    async def both():
+        return await asyncio.gather(call_and_time("slow-token"), call_and_time("fast-token"))
+
+    (slow, slow_finished), (fast, fast_finished) = asyncio.run(both())
+    assert slow.structuredContent["categories"] == ["slow-token"]
+    assert fast.structuredContent["categories"] == ["fast-token"]
+    assert fast_finished < slow_finished - 1
+
+
 def test_request_without_credential_is_refused(http_server):
     """A request carrying no token is refused rather than served with the environment's client."""
     url, _ = http_server

{codeocean_mcp_server-0.12.1 → codeocean_mcp_server-0.12.2}/uv.lock

@@ -558,9 +558,10 @@ wheels = [
 
 [[package]]
 name = "codeocean-mcp-server"
-version = "0.12.1"
+version = "0.12.2"
 source = { editable = "." }
 dependencies = [
+    { name = "anyio" },
     { name = "codeocean" },
     { name = "mcp" },
 ]
@@ -577,6 +578,7 @@ dev = [
 
 [package.metadata]
 requires-dist = [
+    { name = "anyio", specifier = ">=4.5" },
     { name = "codeocean", specifier = ">=0.14.0,<0.15.0" },
     { name = "mcp", specifier = ">=1.23.0,<1.24.0" },
 ]