feat(cli): UME lazy exec via shared profiles and ume_ne_id MCP support

Co-authored-by: Cursor <cursoragent@cursor.com>

netx_api/cli_schemas.py

@@ -0,0 +1,122 @@
+from __future__ import annotations
+
+from datetime import datetime
+
+from pydantic import BaseModel, Field, field_validator
+
+from .device_types import SUPPORTED_VENDORS
+
+
+class CliConnectProfileCreate(BaseModel):
+    name: str
+    username: str
+    password: str = ""
+    port: int = 22
+    protocol: str = "ssh"
+    device_type_default: str = "zte_zxros"
+    vendor_default: str = "ZTE"
+    ne_type_rules: str = ""
+    is_default: bool = False
+    hop_enabled: bool = False
+    hop_vendor: str = "zte"
+    hop_host: str = ""
+    hop_port: int = 22
+    hop_protocol: str = "ssh"
+    hop_username: str = ""
+    hop_password: str = ""
+    hop_command_template: str = ""
+    hop_vrf: str = ""
+    hop_target_auth_mode: str = "bastion_managed"
+
+    @field_validator("vendor_default")
+    @classmethod
+    def normalize_vendor(cls, v: str) -> str:
+        raw = str(v or "").strip()
+        if not raw:
+            return "ZTE"
+        for item in SUPPORTED_VENDORS:
+            if item.lower() == raw.lower():
+                return item
+        return "Other"
+
+
+class CliConnectProfileUpdate(BaseModel):
+    name: str | None = None
+    username: str | None = None
+    password: str | None = None
+    port: int | None = None
+    protocol: str | None = None
+    device_type_default: str | None = None
+    vendor_default: str | None = None
+    ne_type_rules: str | None = None
+    is_default: bool | None = None
+    hop_enabled: bool | None = None
+    hop_vendor: str | None = None
+    hop_host: str | None = None
+    hop_port: int | None = None
+    hop_protocol: str | None = None
+    hop_username: str | None = None
+    hop_password: str | None = None
+    hop_command_template: str | None = None
+    hop_vrf: str | None = None
+    hop_target_auth_mode: str | None = None
+
+
+class CliConnectProfileOut(BaseModel):
+    id: str
+    name: str
+    is_default: bool
+    username: str
+    port: int
+    protocol: str
+    device_type_default: str
+    vendor_default: str
+    ne_type_rules: str
+    hop_enabled: bool
+    hop_vendor: str
+    hop_host: str
+    hop_port: int
+    hop_protocol: str
+    hop_username: str
+    hop_command_template: str
+    hop_vrf: str
+    hop_target_auth_mode: str
+    created_at: datetime
+    updated_at: datetime
+
+
+class UmeCliOverrideUpdate(BaseModel):
+    profile_id: str | None = None
+    username_override: str | None = None
+    device_type_override: str | None = None
+    vendor_override: str | None = None
+
+
+class UmeCliOverrideOut(BaseModel):
+    ume_ne_id: str
+    profile_id: str | None
+    username_override: str
+    device_type_override: str
+    vendor_override: str
+    connect_status: str
+    connect_message: str
+    connect_detail: str
+    connect_tested_at: datetime | None
+    updated_at: datetime
+
+
+class UmeConnectTestRequest(BaseModel):
+    ume_ne_ids: list[str] = Field(min_length=1)
+
+
+class CliTargetOut(BaseModel):
+    source: str
+    id: str
+    ume_ne_id: str | None = None
+    name: str
+    ip_address: str
+    ne_type: str = ""
+    vendor: str = ""
+    device_type: str = ""
+    connect_status: str = "unknown"
+    cli_profile_ready: bool = False