deepset-mcp 0.0.9__tar.gz → 0.0.10__tar.gz

{deepset_mcp-0.0.9 → deepset_mcp-0.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deepset-mcp
-Version: 0.0.9
+Version: 0.0.10
 Summary: Collection of MCP tools and Agents to work with the deepset AI platform. Create, debug or learn about pipelines on the platform. Useable from the CLI, Cursor, Claude Code, or other MCP clients.
 Project-URL: Homepage, https://deepset.ai
 Author-email: Mathis Lucka <mathis.lucka@deepset.ai>, Tanay Soni <tanay.soni@deepset.ai>

{deepset_mcp-0.0.9 → deepset_mcp-0.0.10}/src/deepset_mcp/mcp/tool_factory.py

@@ -193,8 +193,15 @@ def apply_client(
         ctx_param = inspect.Parameter(name="ctx", kind=inspect.Parameter.KEYWORD_ONLY, annotation=Context)
         new_params.append(ctx_param)
         client_wrapper_with_context.__signature__ = original_sig.replace(parameters=new_params)  # type: ignore
+
+        # Remove client from docstring
         client_wrapper_with_context.__doc__ = remove_params_from_docstring(base_func.__doc__, {"client"})
 
+        # Remove client from annotations and add ctx
+        new_annotations = {k: v for k, v in base_func.__annotations__.items() if k != "client"}
+        new_annotations["ctx"] = Context
+        client_wrapper_with_context.__annotations__ = new_annotations
+
         return client_wrapper_with_context
     else:
 
@@ -214,6 +221,10 @@ def apply_client(
         # Remove client from docstring
         client_wrapper_without_context.__doc__ = remove_params_from_docstring(base_func.__doc__, {"client"})
 
+        # Remove client from annotations
+        new_annotations = {k: v for k, v in base_func.__annotations__.items() if k != "client"}
+        client_wrapper_without_context.__annotations__ = new_annotations
+
         return client_wrapper_without_context
 
 

{deepset_mcp-0.0.9 → deepset_mcp-0.0.10}/test/unit/test_tool_factory.py

@@ -9,6 +9,7 @@ from typing import Any
 from unittest.mock import MagicMock, patch
 
 import pytest
+from mcp.server.fastmcp import Context
 
 from deepset_mcp.api.protocols import AsyncClientProtocol
 from deepset_mcp.mcp.tool_factory import (
@@ -265,6 +266,11 @@ class TestApplyClient:
         assert ":param client:" not in result.__doc__
         assert ":param a:" in result.__doc__
 
+        # Check annotations were updated
+        assert "client" not in result.__annotations__
+        assert "ctx" in result.__annotations__
+        assert result.__annotations__["ctx"] == Context
+
     def test_client_signature_updated_without_context(self) -> None:
         """Test that client parameter is removed without ctx."""
 
@@ -289,6 +295,10 @@ class TestApplyClient:
         assert result.__doc__ is not None
         assert ":param client:" not in result.__doc__
 
+        # Check annotations were updated
+        assert "client" not in result.__annotations__
+        assert "ctx" not in result.__annotations__
+
     @pytest.mark.asyncio
     async def test_client_context_missing_raises_error(self) -> None:
         """Test that missing context raises ValueError."""