dtlpymcp 0.1.10__py3-none-any.whl → 0.1.11__py3-none-any.whl

dtlpymcp/__init__.py

@@ -1,3 +1,3 @@
 from .utils.dtlpy_context import DataloopContext, MCPSource
 
-__version__ = "0.1.10"
+__version__ = "0.1.11"

dtlpymcp/min_proxy.py

@@ -28,7 +28,7 @@ def create_dataloop_mcp_server() -> FastMCP:
         log_level="DEBUG",
     )
     tool_name = "test"
-    input_schema = {"properties": {"ping": {"type": "string", "default": "pong"}}, "required": ["ping"]}
+    input_schema = {"type": "object", "properties": {"ping": {"type": "string", "default": "pong"}}, "required": ["ping"]}
     # Create Dataloop context
     dynamic_pydantic_model_params = DataloopContext.build_pydantic_fields_from_schema(input_schema)
     arguments_model = create_model(f"{tool_name}Arguments", **dynamic_pydantic_model_params, __base__=ArgModelBase)

dtlpymcp/proxy.py

@@ -118,7 +118,7 @@ def create_dataloop_mcp_server(sources_file: Optional[str] = None, init_timeout:
         return result
 
     tool_name = "test"
-    input_schema = {"properties": {"ping": {"type": "string"}}, "required": ["ping"]}
+    input_schema = {"type": "object", "properties": {"ping": {"type": "string"}}, "required": ["ping"]}
     dynamic_pydantic_model_params = DataloopContext.build_pydantic_fields_from_schema(input_schema)
     arguments_model = create_model(f"{tool_name}Arguments", **dynamic_pydantic_model_params, __base__=ArgModelBase)
     resp = FuncMetadata(arg_model=arguments_model)

dtlpymcp/utils/dtlpy_context.py

@@ -95,6 +95,10 @@ class DataloopContext:
             ns_tool_name = f"{server_name}.{tool_name}"
             description = tool.description
             input_schema = tool.inputSchema
+            
+            # Normalize input schema to ensure it has "type": "object" at root level
+            # This is required by the MCP specification
+            input_schema = self.normalize_input_schema(input_schema)
 
             def build_handler(tool_name):
                 async def inner(**kwargs):
@@ -230,6 +234,36 @@ class DataloopContext:
                 logger.info(f"Discovered {len(tools.tools)} tools for source {source.dpk_name}")
                 return (source.dpk_name, tools, call_fn)
 
+    @staticmethod
+    def normalize_input_schema(input_schema: dict) -> dict:
+        """
+        Normalize input schema to ensure it conforms to MCP specification.
+        The schema must have "type": "object" at the root level.
+        """
+        if not isinstance(input_schema, dict):
+            return {"type": "object", "properties": {}, "required": []}
+        
+        # Create a copy to avoid mutating the original
+        normalized = input_schema.copy()
+        
+        # Ensure type is "object" at root level
+        if "type" not in normalized:
+            normalized["type"] = "object"
+        elif normalized.get("type") != "object":
+            # If type exists but isn't "object", log a warning and fix it
+            logger.warning(f"Input schema has type '{normalized.get('type')}' instead of 'object', fixing...")
+            normalized["type"] = "object"
+        
+        # Ensure properties exist
+        if "properties" not in normalized:
+            normalized["properties"] = {}
+        
+        # Ensure required exists (can be empty list)
+        if "required" not in normalized:
+            normalized["required"] = []
+        
+        return normalized
+
     @staticmethod
     def openapi_type_to_python(type_str):
         return {"string": str, "integer": int, "number": float, "boolean": bool, "array": list, "object": dict}.get(

{dtlpymcp-0.1.10.dist-info → dtlpymcp-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dtlpymcp
-Version: 0.1.10
+Version: 0.1.11
 Summary: STDIO MCP proxy server for Dataloop platform.
 Author-email: Your Name <your.email@example.com>
 Classifier: Programming Language :: Python :: 3.10

dtlpymcp-0.1.11.dist-info/RECORD

@@ -0,0 +1,10 @@
+dtlpymcp/__init__.py,sha256=-iPfWD7k43HTz0akAm7BHONlJipmtYiMjmF1CCEj0us,87
+dtlpymcp/__main__.py,sha256=ZsXN8guga8Qo-94bSvgC6u9s5gmmdyppUijb-_bCxCw,1347
+dtlpymcp/min_proxy.py,sha256=QYGoqclGNN7ZMjoLzWBLCxUm4yT2xCoLTbtdFRsqlEk,2572
+dtlpymcp/proxy.py,sha256=IzZrK6s37sXM-AM2L2m_K9AveT7uX_VSxwgrwggJYV4,6213
+dtlpymcp/utils/dtlpy_context.py,sha256=6ad136vmk0Cstp10cAcY6YaZkUTgFzyvMlocIP6DwiE,12112
+dtlpymcp-0.1.11.dist-info/METADATA,sha256=G895DOpzSG2bf9EzmzGIIO2MJgH0RtX6W48Z_eXqnRU,2190
+dtlpymcp-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dtlpymcp-0.1.11.dist-info/entry_points.txt,sha256=6hRVZNTjQevj7erwt9dAOURtPVrSrYu6uHXhAlhTaXQ,52
+dtlpymcp-0.1.11.dist-info/top_level.txt,sha256=z85v20pIEnY3cBaWgwhU3EZS4WAZRywejhIutwd-iHk,9
+dtlpymcp-0.1.11.dist-info/RECORD,,

dtlpymcp-0.1.10.dist-info/RECORD

@@ -1,10 +0,0 @@
-dtlpymcp/__init__.py,sha256=H33i9m9ILJX_o4GWGWBcCT_QoijFOU38yXubkbRRoYQ,87
-dtlpymcp/__main__.py,sha256=ZsXN8guga8Qo-94bSvgC6u9s5gmmdyppUijb-_bCxCw,1347
-dtlpymcp/min_proxy.py,sha256=n9KzWwaGU-8uUKRJigGHAnyZ6GXfNl5bwMpULfBrbEc,2554
-dtlpymcp/proxy.py,sha256=5r6T6kKo_liZI-5I0P7D8RaShvVTcsAOVZvnbPjJCDc,6195
-dtlpymcp/utils/dtlpy_context.py,sha256=qeKwptMx_fenYTRs5-l3711yPyo2iof_FoQX8jM3fmE,10670
-dtlpymcp-0.1.10.dist-info/METADATA,sha256=EA20TrysTutuhcOb-UG-7E1rDxW0WqoWd-0wMYP6880,2190
-dtlpymcp-0.1.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dtlpymcp-0.1.10.dist-info/entry_points.txt,sha256=6hRVZNTjQevj7erwt9dAOURtPVrSrYu6uHXhAlhTaXQ,52
-dtlpymcp-0.1.10.dist-info/top_level.txt,sha256=z85v20pIEnY3cBaWgwhU3EZS4WAZRywejhIutwd-iHk,9
-dtlpymcp-0.1.10.dist-info/RECORD,,