Publish curated ITWorx_MCP_Hub source from 6c7434368ce2a594005983c5a6601a31a639e7f1

services/control_api/persistence/machine_client.py

@@ -8,6 +8,8 @@ from uuid import NAMESPACE_URL, UUID, uuid5
 from sqlalchemy import select
 from sqlalchemy.orm import Session
 
+from packages.hub_clients import ClientKind
+
 from .models import (
     ClientBindingRow,
     ClientRow,
@@ -114,6 +116,12 @@ def provision_machine_client(
             or by_id.status != "pending"
         ):
             raise ValueError("existing Hub client is not an activatable pending identity")
+        try:
+            kind = ClientKind(existing_metadata.get("kind", ClientKind.STDIO_BRIDGE.value))
+        except (TypeError, ValueError) as exc:
+            raise ValueError("pending machine client has an invalid client kind") from exc
+    else:
+        kind = ClientKind.STDIO_BRIDGE
     client = principal or by_id
     metadata = {
         "managedBy": "authentik-machine-provisioning",
@@ -121,7 +129,7 @@ def provision_machine_client(
         "name": name,
         "project": project_slug,
         "environment": environment_name,
-        "kind": "stdio-bridge",
+        "kind": kind.value,
         "oauthUsername": oauth_username,
         "credentialReference": f"authentik://service-account/{oauth_username}",
         "operatorDisabled": False,