refactor: flip spine to NemoClient + swap all SDK imports

Flip sdk_factory to return NemoClient/AsyncNemoClient instead of
NeMoPlatform/AsyncNeMoPlatform. Rewrite the bridge adapter to use
from_client() instead of reading Stainless private attrs. Swap all
528 consumer files from 'from nemo_platform' imports to typed client
imports (nemo_platform_plugin.client.client, .errors, .types, etc.).

Type annotations changed: NeMoPlatform -> NemoClient,
AsyncNeMoPlatform -> AsyncNemoClient throughout.

Method calls use new typed client method names (get_workspace,
create_guardrail_config, etc.) which are available on the typed
clients from PR 1. Old method names (retrieve, create, list, etc.)
are also available via the compat layer from PR 2.

The Stainless SDK remains in place (deletion is PR 10). 24 files
that still import from nemo_platform (enhanced NeMoPlatform client,
error handlers, code generators) are left as-is since the SDK is
still present and importable.

Pyproject.toml files are NOT changed (SDK deps kept) — removal
happens in PR 10 when the SDK is actually deleted.

AIRCORE-827

Signed-off-by: Max Dubrinsky <mdubrinsky@nvidia.com>

Author: maxdubrinsky

agents/nemo-studio-assistant/src/nemo_studio_assistant/register.py

@@ -10,22 +10,22 @@
 from urllib.parse import urlencode
 
 import httpx
-from nemo_platform import NeMoPlatform
+from nemo_platform_plugin.client.client import NemoClient
 from pydantic import BaseModel
 
 DEFAULT_WORKSPACE = "default"
 STUDIO_CALLBACK_PATH = "/studio/api/assistant/mcp/{session_id}"
 STUDIO_CALLBACK_TIMEOUT_SECONDS = 3600.0
 _READ_ONLY_SDK_ACTIONS = frozenset({"get", "get_logs", "get_status", "list", "read", "retrieve", "search"})
 
-_clients: dict[str, NeMoPlatform] = {}
+_clients: dict[str, NemoClient] = {}
 
 
 def _active_workspace() -> str:
     return os.environ.get("NMP_WORKSPACE") or DEFAULT_WORKSPACE
 
 
-def _get_client(workspace: str) -> NeMoPlatform:
+def _get_client(workspace: str) -> NemoClient:
     request_workspace = workspace.strip()
     if not request_workspace:
         raise ValueError("workspace is required")
@@ -34,7 +34,7 @@ def _get_client(workspace: str) -> NeMoPlatform:
         kwargs: dict[str, Any] = {"workspace": request_workspace}
         if base_url:
             kwargs["base_url"] = base_url
-        _clients[request_workspace] = NeMoPlatform(**kwargs)
+        _clients[request_workspace] = NemoClient(**kwargs)
     return _clients[request_workspace]
 
 
@@ -50,7 +50,7 @@ def _serialize(obj: Any) -> Any:
     return str(obj)
 
 
-def _resolve_resource(client: NeMoPlatform, resource_path: str) -> Any:
+def _resolve_resource(client: NemoClient, resource_path: str) -> Any:
     current = client
     for part in resource_path.split("."):
         if not part or part.startswith("_"):

docs/evaluator/test_doc_examples.py

@@ -30,7 +30,7 @@
 from nemo_evaluator.sdk import Evaluator
 from nemo_evaluator.shared.metric_bundles.bundles import MetricBundlePackagerPolicyError
 from nemo_evaluator_sdk.metrics.protocol import MetricInput, MetricOutput, MetricOutputSpec, MetricResult
-from nemo_platform import NeMoPlatform
+from nemo_platform_plugin.client.client import NemoClient
 
 
 class _CustomMetric:
@@ -88,7 +88,7 @@ def _evaluator() -> Evaluator:
     Client construction and the ``submit`` argument guard are both offline; the
     guard runs before any executor/HTTP work.
     """
-    client = NeMoPlatform(base_url="http://localhost:8080", workspace="default")
+    client = NemoClient(base_url="http://localhost:8080", workspace="default")
     return client.evaluator
 
 

e2e/agents_deploy_helpers.py

@@ -27,7 +27,7 @@
 import httpx
 import pytest
 from nemo_agents_plugin.entities import NAT_WORKFLOW_CONFIG_FORMAT, NEMO_AGENTS_SPEC_CONFIG_FORMAT
-from nemo_platform import NeMoPlatform
+from nemo_platform_plugin.client.client import NemoClient
 from nmp.testing import MockProviderResponse, add_mock_provider
 
 # The mocked completion the deployed agent must round-trip back to the caller.
@@ -116,23 +116,23 @@ def _page_data(page: Any) -> list[dict[str, Any]]:
     return data
 
 
-def delete_agent_if_exists(sdk: NeMoPlatform, *, workspace: str, name: str) -> None:
+def delete_agent_if_exists(sdk: NemoClient, *, workspace: str, name: str) -> None:
     try:
-        sdk.agents.delete(name, workspace=workspace)
+        sdk.agents.delete_agent(name, workspace=workspace)
     except httpx.HTTPStatusError as exc:
         if exc.response.status_code != 404:
             raise
 
 
-def delete_deployment_if_exists(sdk: NeMoPlatform, *, workspace: str, name: str) -> None:
+def delete_deployment_if_exists(sdk: NemoClient, *, workspace: str, name: str) -> None:
     try:
-        sdk.agents.deployments.delete(name, workspace=workspace)
+        sdk.agents.delete_deployment(name, workspace=workspace)
     except httpx.HTTPStatusError as exc:
         if exc.response.status_code != 404:
             raise
 
 
-def get_deployment_log_text(sdk: NeMoPlatform, *, workspace: str, name: str) -> str:
+def get_deployment_log_text(sdk: NemoClient, *, workspace: str, name: str) -> str:
     try:
         response = sdk._client.get(
             f"/apis/agents/v2/workspaces/{workspace}/deployments/{name}/logs",
@@ -148,7 +148,7 @@ def get_deployment_log_text(sdk: NeMoPlatform, *, workspace: str, name: str) ->
 
 
 def wait_for_deployment_deleted(
-    sdk: NeMoPlatform,
+    sdk: NemoClient,
     *,
     workspace: str,
     name: str,
@@ -158,7 +158,7 @@ def wait_for_deployment_deleted(
     last_status: str | None = None
     while time.monotonic() < deadline:
         try:
-            deployment = sdk.agents.deployments.get(name, workspace=workspace)
+            deployment = sdk.agents.get_deployment(name, workspace=workspace)
             last_status = deployment.get("status")
         except httpx.HTTPStatusError as exc:
             if exc.response.status_code == 404:
@@ -169,7 +169,7 @@ def wait_for_deployment_deleted(
 
 
 def wait_for_deployment_running(
-    sdk: NeMoPlatform,
+    sdk: NemoClient,
     *,
     workspace: str,
     name: str,
@@ -178,7 +178,7 @@ def wait_for_deployment_running(
     deadline = time.monotonic() + timeout_seconds
     last_deployment: dict[str, Any] | None = None
     while time.monotonic() < deadline:
-        deployment = sdk.agents.deployments.get(name, workspace=workspace)
+        deployment = sdk.agents.get_deployment(name, workspace=workspace)
         last_deployment = deployment
         status = deployment["status"]
         if status == "running":
@@ -191,7 +191,7 @@ def wait_for_deployment_running(
 
 
 def run_agent_deploy_and_invoke(
-    sdk: NeMoPlatform,
+    sdk: NemoClient,
     *,
     workspace: str,
     deployment_mode: str,
@@ -233,7 +233,7 @@ def run_agent_deploy_and_invoke(
         served_models={model_name: model_name},
     )
 
-    sdk.agents.create(
+    sdk.agents.create_agent(
         workspace=workspace,
         name=agent_name,
         config=_mock_backed_agent_config(
@@ -245,7 +245,7 @@ def run_agent_deploy_and_invoke(
     )
 
     try:
-        created = sdk.agents.deployments.create(
+        created = sdk.agents.create_deployment(
             workspace=workspace,
             agent=agent_name,
             name=deployment_name,
@@ -276,7 +276,7 @@ def run_agent_deploy_and_invoke(
 
         sdk.models.wait_for_openai_model(model_name, workspace=workspace)
 
-        response = sdk.agents.invoke(
+        response = sdk.agents.invoke_agent(
             workspace=workspace,
             agent=agent_name,
             input="What is 12 multiplied by 8?",
@@ -294,7 +294,7 @@ def run_agent_deploy_and_invoke(
 
 
 def run_container_agent_deploy_and_invoke(
-    sdk: NeMoPlatform,
+    sdk: NemoClient,
     *,
     workspace: str,
     deployment_mode: str,

e2e/auditor/conftest.py

@@ -4,10 +4,10 @@
 """Fixtures for auditor plugin e2e tests."""
 
 import pytest
-from nemo_platform import NeMoPlatform
+from nemo_platform_plugin.client.client import NemoClient
 
 
 @pytest.fixture
-def auditor_url(sdk: NeMoPlatform) -> str:
+def auditor_url(sdk: NemoClient) -> str:
     """Root URL for raw httpx calls to the auditor plugin (filter/sort params not in SDK)."""
     return str(sdk.base_url).rstrip("/") + "/apis/auditor"

e2e/auditor/test_audit_job.py

@@ -19,7 +19,7 @@
 from contextlib import suppress
 
 import pytest
-from nemo_platform import NeMoPlatform
+from nemo_platform_plugin.client.client import NemoClient
 from nmp.testing import add_mock_provider, short_unique_name
 
 from e2e.auditor.utils import minimal_audit_config, unique_name
@@ -44,25 +44,25 @@ def _chat_completion(content: str = "I'm happy to help!") -> dict:
     }
 
 
-def _wait_for_audit_job(sdk: NeMoPlatform, job_name: str, workspace: str) -> str:
+def _wait_for_audit_job(sdk: NemoClient, job_name: str, workspace: str) -> str:
     deadline = time.monotonic() + AUDIT_JOB_TIMEOUT_SECONDS
     while time.monotonic() < deadline:
-        status_resp = sdk.jobs.get_status(name=job_name, workspace=workspace)
+        status_resp = sdk.jobs.get_job_status(name=job_name, workspace=workspace)
         status = str(status_resp.status)
         if status in TERMINAL_STATUSES:
             return status
         time.sleep(AUDIT_JOB_POLL_INTERVAL_SECONDS)
     raise TimeoutError(f"Audit job {job_name!r} did not complete within {AUDIT_JOB_TIMEOUT_SECONDS}s")
 
 
-def _cleanup_audit_job(sdk: NeMoPlatform, job_name: str, workspace: str) -> None:
+def _cleanup_audit_job(sdk: NemoClient, job_name: str, workspace: str) -> None:
     with suppress(Exception):
-        sdk.jobs.cancel(name=job_name, workspace=workspace)
+        sdk.jobs.cancel_job(name=job_name, workspace=workspace)
     with suppress(Exception):
-        sdk.jobs.delete(name=job_name, workspace=workspace)
+        sdk.jobs.delete_job(name=job_name, workspace=workspace)
 
 
-def _add_mock_provider_or_skip(sdk: NeMoPlatform, workspace: str, name: str) -> str:
+def _add_mock_provider_or_skip(sdk: NemoClient, workspace: str, name: str) -> str:
     """Create a mock inference provider, skipping the test if the deployment doesn't support one."""
     try:
         provider = add_mock_provider(
@@ -86,27 +86,27 @@ def _add_mock_provider_or_skip(sdk: NeMoPlatform, workspace: str, name: str) ->
 
 
 @pytest.fixture(scope="module")
-def audit_workspace(sdk: NeMoPlatform) -> Iterator[str]:
+def audit_workspace(sdk: NemoClient) -> Iterator[str]:
     name = short_unique_name("e2e-audit")
-    sdk.workspaces.create(name=name)
+    sdk.workspaces.create_workspace(name=name)
     try:
         yield name
     finally:
         with suppress(Exception):
-            sdk.workspaces.delete(name)
+            sdk.workspaces.delete_workspace(name)
 
 
 @pytest.fixture(scope="module")
-def mock_provider_name(sdk: NeMoPlatform, audit_workspace: str) -> str:
+def mock_provider_name(sdk: NemoClient, audit_workspace: str) -> str:
     """Create a canned-response mock provider for the module; workspace deletion cascades cleanup."""
     provider_name = short_unique_name("audit-mock")
     return _add_mock_provider_or_skip(sdk, audit_workspace, provider_name)
 
 
 @pytest.fixture(scope="module")
-def audit_config_name(sdk: NeMoPlatform, audit_workspace: str) -> Iterator[str]:
+def audit_config_name(sdk: NemoClient, audit_workspace: str) -> Iterator[str]:
     name = short_unique_name("e2e-audit-cfg")
-    sdk.auditor.configs.create(
+    sdk.auditor.create_audit_config(
         workspace=audit_workspace,
         name=name,
         **minimal_audit_config(plugins={"probe_spec": "test.Test", "detector_spec": "auto"}),
@@ -115,13 +115,13 @@ def audit_config_name(sdk: NeMoPlatform, audit_workspace: str) -> Iterator[str]:
         yield name
     finally:
         with suppress(Exception):
-            sdk.auditor.configs.delete(workspace=audit_workspace, name=name)
+            sdk.auditor.delete_audit_config(workspace=audit_workspace, name=name)
 
 
 @pytest.fixture(scope="module")
-def audit_target_name(sdk: NeMoPlatform, audit_workspace: str, mock_provider_name: str) -> Iterator[str]:
+def audit_target_name(sdk: NemoClient, audit_workspace: str, mock_provider_name: str) -> Iterator[str]:
     name = short_unique_name("e2e-audit-tgt")
-    sdk.auditor.targets.create(
+    sdk.auditor.create_audit_target(
         workspace=audit_workspace,
         name=name,
         type="openai",
@@ -143,15 +143,15 @@ def audit_target_name(sdk: NeMoPlatform, audit_workspace: str, mock_provider_nam
         yield name
     finally:
         with suppress(Exception):
-            sdk.auditor.targets.delete(workspace=audit_workspace, name=name)
+            sdk.auditor.delete_audit_target(workspace=audit_workspace, name=name)
 
 
 # ---- Tests ----
 
 
 @pytest.mark.skip("re-enable after auditor image rebuilt")
 def test_audit_job_submit_blank_probe(
-    sdk: NeMoPlatform,
+    sdk: NemoClient,
     audit_workspace: str,
     mock_provider_name: str,
 ) -> None:
@@ -180,7 +180,7 @@ def test_audit_job_submit_blank_probe(
         },
     }
 
-    job = sdk.auditor.submit(config=config, target=target, workspace=audit_workspace)
+    job = sdk.auditor.submit_audit(config=config, target=target, workspace=audit_workspace)
     job_name = job.name
     try:
         final_status = _wait_for_audit_job(sdk, job_name, audit_workspace)
@@ -194,13 +194,13 @@ def test_audit_job_submit_blank_probe(
 
 @pytest.mark.skip("re-enable after auditor image rebuilt")
 def test_audit_job_submit_with_entity_refs(
-    sdk: NeMoPlatform,
+    sdk: NemoClient,
     audit_workspace: str,
     audit_config_name: str,
     audit_target_name: str,
 ) -> None:
     """Submit an audit job using stored entity name references and verify completion."""
-    job = sdk.auditor.submit(
+    job = sdk.auditor.submit_audit(
         config=f"{audit_workspace}/{audit_config_name}",
         target=f"{audit_workspace}/{audit_target_name}",
         workspace=audit_workspace,
@@ -216,13 +216,13 @@ def test_audit_job_submit_with_entity_refs(
 
 
 def test_audit_job_appears_in_list(
-    sdk: NeMoPlatform,
+    sdk: NemoClient,
     audit_workspace: str,
     audit_config_name: str,
     audit_target_name: str,
 ) -> None:
     """Submitted audit job appears in list_jobs() with its name."""
-    job = sdk.auditor.submit(
+    job = sdk.auditor.submit_audit(
         config=f"{audit_workspace}/{audit_config_name}",
         target=f"{audit_workspace}/{audit_target_name}",
         workspace=audit_workspace,

e2e/auditor/test_cli.py

@@ -11,13 +11,13 @@
 
 import json
 
-from nemo_platform import NeMoPlatform
+from nemo_platform_plugin.client.client import NemoClient
 from nmp.testing import assert_exit_0, run_nemo_local
 
 from e2e.auditor.utils import minimal_audit_config, minimal_audit_target, unique_name
 
 
-def test_cli_config_create_list_delete(sdk: NeMoPlatform, workspace: str) -> None:
+def test_cli_config_create_list_delete(sdk: NemoClient, workspace: str) -> None:
     name = unique_name("cli-cfg")
     base_url = str(sdk.base_url)
 
@@ -49,7 +49,7 @@ def test_cli_config_create_list_delete(sdk: NeMoPlatform, workspace: str) -> Non
     assert all(item["name"] != name for item in listed["data"])
 
 
-def test_cli_target_create_list_delete(sdk: NeMoPlatform, workspace: str) -> None:
+def test_cli_target_create_list_delete(sdk: NemoClient, workspace: str) -> None:
     name = unique_name("cli-tgt")
     base_url = str(sdk.base_url)
 
@@ -75,7 +75,7 @@ def test_cli_target_create_list_delete(sdk: NeMoPlatform, workspace: str) -> Non
     assert_exit_0(result, "CLI delete target")
 
 
-def test_cli_config_update(sdk: NeMoPlatform, workspace: str) -> None:
+def test_cli_config_update(sdk: NemoClient, workspace: str) -> None:
     name = unique_name("cli-upd")
     base_url = str(sdk.base_url)