codeocean-mcp-server 0.8.2__tar.gz → 0.9.0__tar.gz

{codeocean_mcp_server-0.8.2 → codeocean_mcp_server-0.9.0}/CHANGELOG.md

@@ -1,6 +1,11 @@
 CHANGELOG
 =========
 
+## 0.9.0 (2026-01-30)
+
+- [#29](https://github.com/codeocean/codeocean-mcp-server/pull/29) feat: upgrade codeocean SDK to v0.14.0
+- **Minimum Code Ocean platform version updated to `4.2.0`.**
+
 ## 0.8.2 (2026-01-29)
 
 - [#27](https://github.com/codeocean/codeocean-mcp-server/pull/27) fix: update documentation for search results to include pagination details

{codeocean_mcp_server-0.8.2 → codeocean_mcp_server-0.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codeocean-mcp-server
-Version: 0.8.2
+Version: 0.9.0
 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,7 +12,7 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.10
-Requires-Dist: codeocean<0.13.0,>=0.12.0
+Requires-Dist: codeocean<0.15.0,>=0.14.0
 Requires-Dist: mcp<1.13.0,>=1.12.0
 Description-Content-Type: text/markdown
 

{codeocean_mcp_server-0.8.2 → codeocean_mcp_server-0.9.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codeocean-mcp-server"
-version = "0.8.2"
+version = "0.9.0"
 authors = [{ name = "Code Ocean", email = "dev@codeocean.com" }]
 description = "Code Ocean MCP Server"
 readme = "README.md"
@@ -11,7 +11,7 @@ classifiers = [
   "Operating System :: OS Independent",
 ]
 dependencies = [
-  "codeocean>=0.12.0,<0.13.0",
+  "codeocean>=0.14.0,<0.15.0",
   "mcp>=1.12.0,<1.13.0",
 ]
 license = "MIT"

{codeocean_mcp_server-0.8.2 → codeocean_mcp_server-0.9.0}/src/codeocean_mcp_server/tools/capsules.py

@@ -30,14 +30,14 @@ def add_tools(mcp: FastMCP, client: CodeOcean):
         results = client.capsules.search_capsules(params)
         return CapsuleSearchResults.from_sdk_results(results, include_field_names)
 
-    @mcp.tool(description=(str(client.capsules.search_pipelines.__doc__) + " " + str(CapsuleSearchResults.__doc__)))
+    @mcp.tool(description=(str(client.pipelines.search_pipelines.__doc__) + " " + str(CapsuleSearchResults.__doc__)))
     def search_pipelines(
         search_params: CapsuleSearchParamsModel,
         include_field_names: bool = False,
     ) -> CapsuleSearchResults:
         """Search for pipelines matching specified criteria."""
         params = CapsuleSearchParams(**search_params.model_dump(exclude_none=True))
-        results = client.capsules.search_pipelines(params)
+        results = client.pipelines.search_pipelines(params)
         return CapsuleSearchResults.from_sdk_results(results, include_field_names)
 
     @mcp.tool(
@@ -74,3 +74,8 @@ def add_tools(mcp: FastMCP, client: CodeOcean):
     def get_capsule_app_panel(capsule_id: str, version: int | None = None) -> AppPanelModel:
         """Retrieve the app panel for a capsule, optionally for a specific version."""
         return client.capsules.get_capsule_app_panel(capsule_id, version)
+
+    @mcp.tool(description=client.capsules.detach_data_assets.__doc__)
+    def detach_data_assets(capsule_id: str, data_assets: list[str]) -> None:
+        """Remove attached data assets from a capsule."""
+        client.capsules.detach_data_assets(capsule_id, data_assets)

{codeocean_mcp_server-0.8.2 → codeocean_mcp_server-0.9.0}/src/codeocean_mcp_server/tools/computations.py

@@ -5,12 +5,14 @@ from codeocean.computation import (
     Folder,
     RunParams,
 )
+from codeocean.data_asset import DataAssetAttachParams, DataAssetAttachResults
 from mcp.server.fastmcp import FastMCP
 
 from codeocean_mcp_server.file_utils import download_and_read_file
 from codeocean_mcp_server.models import dataclass_to_pydantic
 
 RunParamsModel = dataclass_to_pydantic(RunParams)
+DataAssetAttachParamsModel = dataclass_to_pydantic(DataAssetAttachParams)
 
 
 def add_tools(mcp: FastMCP, client: CodeOcean):
@@ -59,3 +61,28 @@ def add_tools(mcp: FastMCP, client: CodeOcean):
         """Download a file using the provided URL and return its content."""
         file_urls = client.computations.get_result_file_urls(computation_id, file_path)
         return download_and_read_file(file_urls.download_url)
+
+    @mcp.tool(description=client.computations.rename_computation.__doc__)
+    def rename_computation(computation_id: str, name: str) -> None:
+        """Rename an existing computation."""
+        client.computations.rename_computation(computation_id, name)
+
+    @mcp.tool(
+        description=(
+            str(client.computations.attach_data_assets.__doc__)
+            + " Accepts a list of parameter objects (e.g. [{'id': '...'}]). "
+            "Use for cloud workstation sessions."
+        )
+    )
+    def attach_computation_data_assets(
+        computation_id: str,
+        attach_params: list[DataAssetAttachParamsModel],
+    ) -> list[DataAssetAttachResults]:
+        """Attach data assets to a cloud workstation session."""
+        params = [DataAssetAttachParams(**p.model_dump(exclude_none=True)) for p in attach_params]
+        return client.computations.attach_data_assets(computation_id, params)
+
+    @mcp.tool(description=client.computations.detach_data_assets.__doc__)
+    def detach_computation_data_assets(computation_id: str, data_assets: list[str]) -> None:
+        """Detach data assets from a cloud workstation session."""
+        client.computations.detach_data_assets(computation_id, data_assets)

{codeocean_mcp_server-0.8.2 → codeocean_mcp_server-0.9.0}/tests/test_tool_calling.py

@@ -427,6 +427,79 @@ test_response = [
             },
         },
     },
+    # =================================================================
+    # Pipeline tests
+    # --------------------------------------------------------------
+    # search_pipelines
+    {
+        "id": "search_pipelines",
+        "prompt": "search for pipelines with query 'analysis' and return the first 5 results",
+        "expected": {
+            "name": "search_pipelines",
+            "input": {
+                "search_params": {
+                    "query": "analysis",
+                    "limit": 5,
+                }
+            },
+        },
+    },
+    # =================================================================
+    # Capsule detach tests
+    # --------------------------------------------------------------
+    # detach_data_assets from capsule
+    {
+        "id": "detach_data_assets",
+        "prompt": "detach data assets with ids: asset123, asset456 from capsule with id: capsule789",
+        "expected": {
+            "name": "detach_data_assets",
+            "input": {
+                "capsule_id": "capsule789",
+                "data_assets": ["asset123", "asset456"],
+            },
+        },
+    },
+    # =================================================================
+    # Computation management tests
+    # --------------------------------------------------------------
+    # rename_computation
+    {
+        "id": "rename_computation",
+        "prompt": "rename computation with id: comp123 to 'Analysis Run 2024'",
+        "expected": {
+            "name": "rename_computation",
+            "input": {
+                "computation_id": "comp123",
+                "name": "Analysis Run 2024",
+            },
+        },
+    },
+    # --------------------------------------------------------------
+    # attach_computation_data_assets
+    {
+        "id": "attach_computation_data_assets",
+        "prompt": "attach data assets with ids: data789, data101 to cloud workstation computation with id: ws_comp456",
+        "expected": {
+            "name": "attach_computation_data_assets",
+            "input": {
+                "computation_id": "ws_comp456",
+                "attach_params": [{"id": "data789"}, {"id": "data101"}],
+            },
+        },
+    },
+    # --------------------------------------------------------------
+    # detach_computation_data_assets
+    {
+        "id": "detach_computation_data_assets",
+        "prompt": "detach data assets with ids: data222, data333 from cloud workstation with id: ws_comp789",
+        "expected": {
+            "name": "detach_computation_data_assets",
+            "input": {
+                "computation_id": "ws_comp789",
+                "data_assets": ["data222", "data333"],
+            },
+        },
+    },
 ]
 
 

{codeocean_mcp_server-0.8.2 → codeocean_mcp_server-0.9.0}/uv.lock

@@ -514,7 +514,7 @@ wheels = [
 
 [[package]]
 name = "codeocean"
-version = "0.12.0"
+version = "0.14.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "backports-strenum", marker = "python_full_version < '3.11'" },
@@ -522,14 +522,14 @@ dependencies = [
     { name = "requests" },
     { name = "requests-toolbelt" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/1e/87/63203a9bd4c9f160b02d60016c6222e0836a192d74b624445540667a9a34/codeocean-0.12.0.tar.gz", hash = "sha256:f124a0c090dfdecbbce5f2b033673142820cc200a4a4ed12239027e0eff0ada8", size = 19465, upload-time = "2025-10-21T07:28:06.618Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/c2/de/5d0ed25123b5e903eec6325249c23d3ae76d1ad5aceff000c9544e242700/codeocean-0.14.0.tar.gz", hash = "sha256:90d2a58a9c8d05edc53becc0b63773c5b8cd44423de19afa3b0ce1d82ebfadc6", size = 21461, upload-time = "2026-01-29T23:11:34.016Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/0c/67/86aaaa16e089883035cc45e1d93e5861cdf912e98c3f69dfaf7c7fa9b9a9/codeocean-0.12.0-py3-none-any.whl", hash = "sha256:c9135d34d0920dae5298095bb7ad03fffff5b5fd4c5e941a0200c3c38090df1c", size = 20417, upload-time = "2025-10-21T07:28:07.596Z" },
+    { url = "https://files.pythonhosted.org/packages/f8/d1/ef0027ce1a17eedd8c671495d780fffa0590d017061337e4ff86a0a0054e/codeocean-0.14.0-py3-none-any.whl", hash = "sha256:33ea9cde80df0c7d986e493a88edbe00f13b5d0cf92cfaed8a518f7314dbf4e1", size = 25201, upload-time = "2026-01-29T23:11:36.219Z" },
 ]
 
 [[package]]
 name = "codeocean-mcp-server"
-version = "0.8.2"
+version = "0.9.0"
 source = { editable = "." }
 dependencies = [
     { name = "codeocean" },
@@ -548,7 +548,7 @@ dev = [
 
 [package.metadata]
 requires-dist = [
-    { name = "codeocean", specifier = ">=0.12.0,<0.13.0" },
+    { name = "codeocean", specifier = ">=0.14.0,<0.15.0" },
     { name = "mcp", specifier = ">=1.12.0,<1.13.0" },
 ]