Fix: Use enum for server type

Author: cristipufu

src/uipath_mcp/_cli/_runtime/_context.py

@@ -1,3 +1,4 @@
+from enum import Enum
 from typing import Optional
 
 from uipath._cli._runtime._contracts import UiPathRuntimeContext
@@ -7,4 +8,12 @@
 
 class UiPathMcpRuntimeContext(UiPathRuntimeContext):
     """Context information passed throughout the runtime execution."""
+
     config: Optional[McpConfig] = None
+
+
+class UiPathServerType(Enum):
+    UiPath = 0  # Processes, Agents, Activities
+    External = 1  # npx, uvx
+    Local = 2  # PackageType.MCPServer
+    Hosted = 3  # tunnel to externally hosted server

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -19,7 +19,7 @@
 from uipath.tracing import wait_for_tracers
 
 from .._utils._config import McpServer
-from ._context import UiPathMcpRuntimeContext
+from ._context import UiPathMcpRuntimeContext, UiPathServerType
 from ._exception import UiPathMcpRuntimeError
 from ._session import SessionServer
 from ._stdio_client import stdio_client
@@ -313,14 +313,13 @@ async def _register(self) -> None:
                     "Name": self._server.name,
                     "Slug": self._server.name,
                     "Version": "1.0.0",
-                    "Type": 1 if self.sandboxed else 3,
+                    "Type": UiPathServerType.External.value if self.sandboxed else UiPathServerType.Hosted.value,
                 },
                 "tools": [],
             }
 
             for tool in tools_result.tools:
                 tool_info = {
-                    "Type": 1,
                     "Name": tool.name,
                     "ProcessType": "Tool",
                     "Description": tool.description,