dtlpymcp 0.1.12__tar.gz → 0.1.14__tar.gz

{dtlpymcp-0.1.12 → dtlpymcp-0.1.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dtlpymcp
-Version: 0.1.12
+Version: 0.1.14
 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.12 → dtlpymcp-0.1.14}/dtlpymcp/__init__.py

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

{dtlpymcp-0.1.12 → dtlpymcp-0.1.14}/dtlpymcp/utils/dtlpy_context.py

@@ -1,4 +1,3 @@
-from imp import source_from_cache
 from mcp.server.fastmcp.utilities.func_metadata import ArgModelBase, FuncMetadata
 from mcp.client.streamable_http import streamablehttp_client
 from typing import List, Tuple, Callable, Optional
@@ -96,7 +95,7 @@ class DataloopContext:
         server_name, tools, call_fn = result
         for tool in tools.tools:
             tool_name = tool.name
-            ns_tool_name = f"{server_name}.{tool_name}"
+            ns_tool_name = f"{server_name}__{tool_name}"
             description = tool.description
             input_schema = tool.inputSchema
             

{dtlpymcp-0.1.12 → dtlpymcp-0.1.14}/dtlpymcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dtlpymcp
-Version: 0.1.12
+Version: 0.1.14
 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.12 → dtlpymcp-0.1.14}/dtlpymcp.egg-info/SOURCES.txt

@@ -14,6 +14,7 @@ dtlpymcp.egg-info/top_level.txt
 dtlpymcp/utils/dtlpy_context.py
 tests/test_context.py
 tests/test_custom_sources_file.py
+tests/test_dlchat.py
 tests/test_list_platform_tools.py
 tests/test_proxy.py
 tests/test_run.py

{dtlpymcp-0.1.12 → dtlpymcp-0.1.14}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dtlpymcp"
-version = "0.1.12"
+version = "0.1.14"
 description = "STDIO MCP proxy server for Dataloop platform."
 authors = [
     { name = "Your Name", email = "your.email@example.com" }

dtlpymcp-0.1.14/tests/test_dlchat.py

@@ -0,0 +1,36 @@
+import asyncio
+import random
+import json
+import os
+from datetime import timedelta
+from mcp.client.stdio import stdio_client
+from mcp import ClientSession, StdioServerParameters
+import dtlpy as dl
+
+dl.setenv('prod')
+if dl.token_expired():
+    dl.login()
+
+# Create server parameters for stdio connection
+server_params = StdioServerParameters(
+    command="e:\\Applications\\dtlpy-mcp\\.venv\\Scripts\\python.exe",  # Executable
+    args=["-m", "dtlpymcp", "start", "-s", "tests/assets/sources.json"],  # Command line arguments
+    env={"DATALOOP_API_KEY": dl.token()},  # Optional environment variables
+    cwd=os.getcwd()
+)
+
+async def test_health_check():
+    print("[TEST CLIENT] Connecting to MCP server and calling test tool...")
+    async with stdio_client(server=server_params) as (read, write):
+        async with ClientSession(read, write, read_timeout_seconds=timedelta(seconds=60)) as session:
+            await session.initialize()
+            tools = await session.list_tools()
+            print([tool.name for tool in tools.tools])
+            tool_result = await session.call_tool("dl-internal-chat-prod__list_repos")
+            print("[RESULT]", tool_result)            
+            tool_result = await session.call_tool("dl-internal-chat-prod__ping")
+            print("[RESULT]", tool_result)
+
+if __name__ == "__main__":
+    asyncio.run(test_health_check()) 
+    # asyncio.run(test_ask_dataloop())