dtlpymcp 0.1.4__py3-none-any.whl → 0.1.7__py3-none-any.whl

dtlpymcp/__init__.py

@@ -3,5 +3,5 @@ SOURCES_FILEPATH = os.path.join(os.path.dirname(__file__), "default_sources.json
 
 from .utils.dtlpy_context import DataloopContext, MCPSource
 
-__version__ = "0.1.4"
+__version__ = "0.1.7"
 

dtlpymcp/default_sources.json

@@ -16,5 +16,17 @@
         "app_url": null,
         "server_url": null,
         "app_jwt": null
+    },
+    {
+        "dpk_name": "dataloop-mcp-data",
+        "app_url": null,
+        "server_url": null,
+        "app_jwt": null
+    },
+    {
+        "dpk_name": "dataloop-mcp-dtlpy-sandbox",
+        "app_url": null,
+        "server_url": null,
+        "app_jwt": null
     }
 ]

dtlpymcp/utils/dtlpy_context.py

@@ -69,10 +69,13 @@ class DataloopContext:
             data = json.load(f)
         logger.info(f"Loading MCP sources from {sources_file}")
         for entry in data:
-            if not isinstance(entry, dict):
-                raise ValueError(f"Invalid source entry: {entry}")
-            logger.info(f"Adding MCP source: {entry.get('dpk_name')}, url: {entry.get('server_url')}")
-            await self.add_mcp_source(MCPSource(**entry))
+            try:
+                if not isinstance(entry, dict):
+                    raise ValueError(f"Invalid source entry: {entry}")
+                logger.info(f"Adding MCP source: {entry.get('dpk_name')}, url: {entry.get('server_url')}")
+                await self.add_mcp_source(MCPSource(**entry))
+            except Exception as e:
+                logger.error(f"Failed to add MCP source: {entry}\n{traceback.format_exc()}")
 
     async def add_mcp_source(self, mcp_source: MCPSource):
         self.mcp_sources.append(mcp_source)
@@ -129,9 +132,15 @@ class DataloopContext:
         try:
             dl.setenv(self.env)
             dl.client_api.token = self.token
-            dpk = dl.dpks.get(dpk_name=source.dpk_name)
-            apps_filters = dl.Filters(field='dpkName', values=dpk.name, resource='apps')
-            app = dl.apps.list(filters=apps_filters).items[0]
+            filters = dl.Filters(resource='apps')
+            filters.add(field="dpkName", values=source.dpk_name)
+            filters.add(field="scope", values="system")
+            apps = list(dl.apps.list(filters=filters).all())
+            if len(apps) == 0:
+                raise ValueError(f"No app found for DPK name: {source.dpk_name}")
+            if len(apps) > 1:
+                logger.warning(f"Multiple apps found for DPK name: {source.dpk_name}, using first one")
+            app = apps[0]
             logger.info(f"App: {app.name}")
             source.app_url = next(iter(app.routes.values()))
             session = requests.Session()
@@ -139,9 +148,9 @@ class DataloopContext:
             logger.info(f"App route URL: {response.url}")
             source.server_url = response.url
             source.app_jwt = session.cookies.get("JWT-APP")
-        except Exception:
+        except Exception as e:
             logger.error(f"Failed getting app info: {traceback.format_exc()}")
-            raise ValueError(f"Failed getting app info: {traceback.format_exc()}")
+            raise Exception(f"Failed getting app info: {traceback.format_exc()}") from e
 
     @staticmethod
     def is_expired(app_jwt: str) -> bool:

{dtlpymcp-0.1.4.dist-info → dtlpymcp-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dtlpymcp
-Version: 0.1.4
+Version: 0.1.7
 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.7.dist-info/RECORD

@@ -0,0 +1,10 @@
+dtlpymcp/__init__.py,sha256=tUwpqP2EpOpWItUNbzkuuC10aidfIn1SVheVwMf61bY,185
+dtlpymcp/__main__.py,sha256=1lo4qoZAoUHl9rkt1YGB2kCpKg5cIrTSBSrznxyk6F4,884
+dtlpymcp/default_sources.json,sha256=Kg6ArvMscwevXZp5zsnrnSSf8ds8M7AZDtKCYLJEly4,701
+dtlpymcp/proxy.py,sha256=pE-NAJI0A9wKcBJJt18_aSy1BskZKoxaK3f6VW7u44A,3608
+dtlpymcp/utils/dtlpy_context.py,sha256=WmLOoIpujudJ7pDvb07X8_jDAM_ndAi-L4Kqu4T3lFI,10231
+dtlpymcp-0.1.7.dist-info/METADATA,sha256=bkJWJnqSKUCJy2yvUaRHz-IL7DWESe2KyUq6yJnsgN8,1677
+dtlpymcp-0.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dtlpymcp-0.1.7.dist-info/entry_points.txt,sha256=6hRVZNTjQevj7erwt9dAOURtPVrSrYu6uHXhAlhTaXQ,52
+dtlpymcp-0.1.7.dist-info/top_level.txt,sha256=z85v20pIEnY3cBaWgwhU3EZS4WAZRywejhIutwd-iHk,9
+dtlpymcp-0.1.7.dist-info/RECORD,,

dtlpymcp-0.1.4.dist-info/RECORD

@@ -1,10 +0,0 @@
-dtlpymcp/__init__.py,sha256=jz6jKLs8djD6u5xy04_NQO7hgKyFea_oPa-yaiQQq0M,185
-dtlpymcp/__main__.py,sha256=1lo4qoZAoUHl9rkt1YGB2kCpKg5cIrTSBSrznxyk6F4,884
-dtlpymcp/default_sources.json,sha256=Es3XZcdMpe6o5FyOoqW9FG_oUjC5gkNbBC6N9eBqpQs,418
-dtlpymcp/proxy.py,sha256=pE-NAJI0A9wKcBJJt18_aSy1BskZKoxaK3f6VW7u44A,3608
-dtlpymcp/utils/dtlpy_context.py,sha256=0NW5FKFpzUl6gTuJGPPPMqklnNuecgBdwLdW-68YUXE,9752
-dtlpymcp-0.1.4.dist-info/METADATA,sha256=_CYzt19o00e_UIw9SeiQXW7lKp5q1Bq2xSSbyF7W3Yc,1677
-dtlpymcp-0.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dtlpymcp-0.1.4.dist-info/entry_points.txt,sha256=6hRVZNTjQevj7erwt9dAOURtPVrSrYu6uHXhAlhTaXQ,52
-dtlpymcp-0.1.4.dist-info/top_level.txt,sha256=z85v20pIEnY3cBaWgwhU3EZS4WAZRywejhIutwd-iHk,9
-dtlpymcp-0.1.4.dist-info/RECORD,,