nemo-platform-plugin 0.1.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- nemo_platform/__init__.py +117 -0
- nemo_platform/_base_client.py +2146 -0
- nemo_platform/_client.py +1200 -0
- nemo_platform/_compat.py +241 -0
- nemo_platform/_constants.py +29 -0
- nemo_platform/_decoders/jsonl.py +138 -0
- nemo_platform/_exceptions.py +123 -0
- nemo_platform/_files.py +188 -0
- nemo_platform/_models.py +967 -0
- nemo_platform/_qs.py +164 -0
- nemo_platform/_resource.py +58 -0
- nemo_platform/_response.py +872 -0
- nemo_platform/_streaming.py +391 -0
- nemo_platform/_types.py +288 -0
- nemo_platform/_utils/__init__.py +79 -0
- nemo_platform/_utils/_compat.py +60 -0
- nemo_platform/_utils/_datetime_parse.py +151 -0
- nemo_platform/_utils/_json.py +50 -0
- nemo_platform/_utils/_logs.py +40 -0
- nemo_platform/_utils/_path.py +142 -0
- nemo_platform/_utils/_proxy.py +80 -0
- nemo_platform/_utils/_reflection.py +57 -0
- nemo_platform/_utils/_resources_proxy.py +39 -0
- nemo_platform/_utils/_streams.py +27 -0
- nemo_platform/_utils/_sync.py +73 -0
- nemo_platform/_utils/_transform.py +472 -0
- nemo_platform/_utils/_typing.py +171 -0
- nemo_platform/_utils/_utils.py +448 -0
- nemo_platform/_version.py +21 -0
- nemo_platform/auth/__init__.py +15 -0
- nemo_platform/auth/device_flow.py +303 -0
- nemo_platform/auth/helpers.py +211 -0
- nemo_platform/auth/token_provider.py +239 -0
- nemo_platform/beta/evaluator/__init__.py +120 -0
- nemo_platform/beta/evaluator/agent_inference.py +360 -0
- nemo_platform/beta/evaluator/constants.py +4 -0
- nemo_platform/beta/evaluator/dataset_schemas/common.py +177 -0
- nemo_platform/beta/evaluator/dataset_schemas/compatibility.py +353 -0
- nemo_platform/beta/evaluator/dataset_schemas/templates.py +389 -0
- nemo_platform/beta/evaluator/datasets/__init__.py +28 -0
- nemo_platform/beta/evaluator/datasets/loader.py +415 -0
- nemo_platform/beta/evaluator/enums.py +64 -0
- nemo_platform/beta/evaluator/execution/_protocols.py +20 -0
- nemo_platform/beta/evaluator/execution/backends/base.py +86 -0
- nemo_platform/beta/evaluator/execution/backends/local/backend.py +132 -0
- nemo_platform/beta/evaluator/execution/benchmark_execution.py +666 -0
- nemo_platform/beta/evaluator/execution/config.py +64 -0
- nemo_platform/beta/evaluator/execution/evaluator.py +289 -0
- nemo_platform/beta/evaluator/execution/job_poll.py +98 -0
- nemo_platform/beta/evaluator/execution/metric_execution.py +878 -0
- nemo_platform/beta/evaluator/execution/pipeline.py +64 -0
- nemo_platform/beta/evaluator/execution/runs.py +143 -0
- nemo_platform/beta/evaluator/execution/scoring.py +224 -0
- nemo_platform/beta/evaluator/execution/utils.py +116 -0
- nemo_platform/beta/evaluator/execution/values.py +53 -0
- nemo_platform/beta/evaluator/inference.py +456 -0
- nemo_platform/beta/evaluator/metrics/aggregation.py +451 -0
- nemo_platform/beta/evaluator/metrics/bleu.py +94 -0
- nemo_platform/beta/evaluator/metrics/exact_match.py +52 -0
- nemo_platform/beta/evaluator/metrics/f1.py +50 -0
- nemo_platform/beta/evaluator/metrics/hooks.py +41 -0
- nemo_platform/beta/evaluator/metrics/llm_judge.py +392 -0
- nemo_platform/beta/evaluator/metrics/number_check.py +95 -0
- nemo_platform/beta/evaluator/metrics/protocol.py +241 -0
- nemo_platform/beta/evaluator/metrics/ragas/__init__.py +41 -0
- nemo_platform/beta/evaluator/metrics/ragas/base.py +521 -0
- nemo_platform/beta/evaluator/metrics/ragas/git_patch.py +36 -0
- nemo_platform/beta/evaluator/metrics/ragas/imports.py +192 -0
- nemo_platform/beta/evaluator/metrics/ragas/metrics.py +183 -0
- nemo_platform/beta/evaluator/metrics/remote.py +250 -0
- nemo_platform/beta/evaluator/metrics/rouge.py +71 -0
- nemo_platform/beta/evaluator/metrics/string_check.py +69 -0
- nemo_platform/beta/evaluator/metrics/template_rendering.py +141 -0
- nemo_platform/beta/evaluator/metrics/tool_calling.py +173 -0
- nemo_platform/beta/evaluator/metrics/types.py +62 -0
- nemo_platform/beta/evaluator/metrics/utils.py +54 -0
- nemo_platform/beta/evaluator/resilience/api.py +142 -0
- nemo_platform/beta/evaluator/resilience/classifier.py +142 -0
- nemo_platform/beta/evaluator/resilience/config.py +69 -0
- nemo_platform/beta/evaluator/resilience/errors.py +117 -0
- nemo_platform/beta/evaluator/resilience/policy.py +50 -0
- nemo_platform/beta/evaluator/resilience/scheduler.py +459 -0
- nemo_platform/beta/evaluator/resilience/types.py +93 -0
- nemo_platform/beta/evaluator/structured_output.py +184 -0
- nemo_platform/beta/evaluator/templates.py +115 -0
- nemo_platform/beta/evaluator/values/__init__.py +162 -0
- nemo_platform/beta/evaluator/values/agents.py +84 -0
- nemo_platform/beta/evaluator/values/common.py +28 -0
- nemo_platform/beta/evaluator/values/dataset_schemas.py +112 -0
- nemo_platform/beta/evaluator/values/datasets.py +28 -0
- nemo_platform/beta/evaluator/values/metrics.py +587 -0
- nemo_platform/beta/evaluator/values/models.py +157 -0
- nemo_platform/beta/evaluator/values/multi_metric_results.py +359 -0
- nemo_platform/beta/evaluator/values/params.py +109 -0
- nemo_platform/beta/evaluator/values/results.py +699 -0
- nemo_platform/beta/evaluator/values/scores.py +300 -0
- nemo_platform/beta/safe_synthesizer/__init__.py +4 -0
- nemo_platform/beta/safe_synthesizer/config.py +29 -0
- nemo_platform/beta/safe_synthesizer/job.py +309 -0
- nemo_platform/beta/safe_synthesizer/job_builder.py +355 -0
- nemo_platform/cli/__init__.py +4 -0
- nemo_platform/cli/app.py +321 -0
- nemo_platform/cli/commands/__init__.py +15 -0
- nemo_platform/cli/commands/api/__init__.py +114 -0
- nemo_platform/cli/commands/api/adapters.py +384 -0
- nemo_platform/cli/commands/api/files/__init__.py +237 -0
- nemo_platform/cli/commands/api/files/filesets.py +386 -0
- nemo_platform/cli/commands/api/files/otlp/__init__.py +12 -0
- nemo_platform/cli/commands/api/files/otlp/logs.py +116 -0
- nemo_platform/cli/commands/api/guardrail/__init__.py +270 -0
- nemo_platform/cli/commands/api/guardrail/configs.py +311 -0
- nemo_platform/cli/commands/api/iam/__init__.py +12 -0
- nemo_platform/cli/commands/api/iam/role_bindings.py +264 -0
- nemo_platform/cli/commands/api/inference/__init__.py +76 -0
- nemo_platform/cli/commands/api/inference/deployment_configs/__init__.py +370 -0
- nemo_platform/cli/commands/api/inference/deployment_configs/versions.py +132 -0
- nemo_platform/cli/commands/api/inference/deployments/__init__.py +565 -0
- nemo_platform/cli/commands/api/inference/deployments/versions.py +141 -0
- nemo_platform/cli/commands/api/inference/gateway/__init__.py +14 -0
- nemo_platform/cli/commands/api/inference/gateway/model.py +324 -0
- nemo_platform/cli/commands/api/inference/gateway/openai/__init__.py +12 -0
- nemo_platform/cli/commands/api/inference/gateway/openai/v1/__init__.py +12 -0
- nemo_platform/cli/commands/api/inference/gateway/openai/v1/models.py +106 -0
- nemo_platform/cli/commands/api/inference/gateway/provider.py +327 -0
- nemo_platform/cli/commands/api/inference/models.py +106 -0
- nemo_platform/cli/commands/api/inference/providers.py +612 -0
- nemo_platform/cli/commands/api/inference/virtual_models.py +412 -0
- nemo_platform/cli/commands/api/intake/__init__.py +18 -0
- nemo_platform/cli/commands/api/intake/apps/__init__.py +326 -0
- nemo_platform/cli/commands/api/intake/apps/tasks.py +340 -0
- nemo_platform/cli/commands/api/intake/entries/__init__.py +445 -0
- nemo_platform/cli/commands/api/intake/entries/events.py +118 -0
- nemo_platform/cli/commands/api/intake/evaluator_results.py +267 -0
- nemo_platform/cli/commands/api/intake/exports/__init__.py +63 -0
- nemo_platform/cli/commands/api/intake/exports/jobs.py +245 -0
- nemo_platform/cli/commands/api/intake/ingest/__init__.py +14 -0
- nemo_platform/cli/commands/api/intake/ingest/atif.py +115 -0
- nemo_platform/cli/commands/api/intake/ingest/chat_completions.py +125 -0
- nemo_platform/cli/commands/api/intake/ingest/otlp/__init__.py +12 -0
- nemo_platform/cli/commands/api/intake/ingest/otlp/v1/__init__.py +12 -0
- nemo_platform/cli/commands/api/intake/ingest/otlp/v1/traces.py +72 -0
- nemo_platform/cli/commands/api/intake/spans/__init__.py +202 -0
- nemo_platform/cli/commands/api/intake/spans/evaluator_results.py +65 -0
- nemo_platform/cli/commands/api/intake/traces.py +183 -0
- nemo_platform/cli/commands/api/jobs/__init__.py +576 -0
- nemo_platform/cli/commands/api/jobs/results.py +211 -0
- nemo_platform/cli/commands/api/jobs/steps.py +243 -0
- nemo_platform/cli/commands/api/jobs/tasks.py +210 -0
- nemo_platform/cli/commands/api/members.py +304 -0
- nemo_platform/cli/commands/api/models/__init__.py +555 -0
- nemo_platform/cli/commands/api/models/adapters.py +261 -0
- nemo_platform/cli/commands/api/projects.py +332 -0
- nemo_platform/cli/commands/api/safe_synthesizer/__init__.py +12 -0
- nemo_platform/cli/commands/api/safe_synthesizer/jobs/__init__.py +417 -0
- nemo_platform/cli/commands/api/safe_synthesizer/jobs/results.py +211 -0
- nemo_platform/cli/commands/api/secrets/__init__.py +315 -0
- nemo_platform/cli/commands/api/secrets/admin.py +44 -0
- nemo_platform/cli/commands/api/workspaces.py +339 -0
- nemo_platform/cli/commands/auth.py +803 -0
- nemo_platform/cli/commands/config.py +308 -0
- nemo_platform/cli/commands/docs.py +277 -0
- nemo_platform/cli/commands/manifest_registry.py +180 -0
- nemo_platform/cli/commands/plugins.py +87 -0
- nemo_platform/cli/commands/quickstart/__init__.py +15 -0
- nemo_platform/cli/commands/quickstart/cli.py +1178 -0
- nemo_platform/cli/commands/services/_process.py +472 -0
- nemo_platform/cli/commands/services/cli.py +694 -0
- nemo_platform/cli/commands/setup.py +1947 -0
- nemo_platform/cli/commands/skills/agents/claude.py +29 -0
- nemo_platform/cli/commands/skills/agents/codex.py +32 -0
- nemo_platform/cli/commands/skills/agents/cursor.py +18 -0
- nemo_platform/cli/commands/skills/agents/opencode.py +20 -0
- nemo_platform/cli/commands/skills/base.py +53 -0
- nemo_platform/cli/commands/skills/cli.py +313 -0
- nemo_platform/cli/commands/skills/installer.py +51 -0
- nemo_platform/cli/commands/skills/registry.py +396 -0
- nemo_platform/cli/commands/use_cases/__init__.py +15 -0
- nemo_platform/cli/commands/use_cases/agent.py +245 -0
- nemo_platform/cli/commands/use_cases/chat.py +965 -0
- nemo_platform/cli/commands/use_cases/wait.py +151 -0
- nemo_platform/cli/core/__init__.py +8 -0
- nemo_platform/cli/core/agent_helpers.py +84 -0
- nemo_platform/cli/core/api.py +160 -0
- nemo_platform/cli/core/autocomplete.py +63 -0
- nemo_platform/cli/core/code_generator.py +306 -0
- nemo_platform/cli/core/context.py +179 -0
- nemo_platform/cli/core/errors.py +349 -0
- nemo_platform/cli/core/formatters.py +718 -0
- nemo_platform/cli/core/help_formatter.py +868 -0
- nemo_platform/cli/core/lazy_load.py +187 -0
- nemo_platform/cli/core/logging.py +24 -0
- nemo_platform/cli/core/pagination.py +255 -0
- nemo_platform/cli/core/stdin_utils.py +233 -0
- nemo_platform/cli/core/streaming.py +30 -0
- nemo_platform/cli/core/table_config.py +302 -0
- nemo_platform/cli/core/timestamp_formatter.py +127 -0
- nemo_platform/cli/core/types.py +112 -0
- nemo_platform/cli/core/waiters.py +339 -0
- nemo_platform/cli/manifest.py +108 -0
- nemo_platform/client/__init__.py +15 -0
- nemo_platform/client/factory.py +597 -0
- nemo_platform/config/README.md +168 -0
- nemo_platform/config/__init__.py +23 -0
- nemo_platform/config/config.py +538 -0
- nemo_platform/config/models.py +377 -0
- nemo_platform/config/types.py +7 -0
- nemo_platform/filesets/__init__.py +19 -0
- nemo_platform/filesets/filesystem/callbacks.py +229 -0
- nemo_platform/filesets/filesystem/filesystem.py +940 -0
- nemo_platform/filesets/resources.py +1120 -0
- nemo_platform/lib/.keep +4 -0
- nemo_platform/models/__init__.py +14 -0
- nemo_platform/models/resources.py +836 -0
- nemo_platform/pagination.py +149 -0
- nemo_platform/py.typed +0 -0
- nemo_platform/quickstart/__init__.py +99 -0
- nemo_platform/quickstart/cluster.py +198 -0
- nemo_platform/quickstart/config.py +402 -0
- nemo_platform/quickstart/container.py +758 -0
- nemo_platform/quickstart/gpu_config.py +111 -0
- nemo_platform/quickstart/platform_config.py +82 -0
- nemo_platform/quickstart/preflight.py +451 -0
- nemo_platform/quickstart/prompts.py +426 -0
- nemo_platform/quickstart/storage.py +90 -0
- nemo_platform/quickstart/validators.py +268 -0
- nemo_platform/resources/__init__.py +16 -0
- nemo_platform/resources/adapters/__init__.py +34 -0
- nemo_platform/resources/adapters/adapters.py +704 -0
- nemo_platform/resources/adapters/api.md +15 -0
- nemo_platform/resources/audit/__init__.py +16 -0
- nemo_platform/resources/entities/__init__.py +34 -0
- nemo_platform/resources/entities/api.md +16 -0
- nemo_platform/resources/entities/entities.py +1014 -0
- nemo_platform/resources/evaluation/__init__.py +118 -0
- nemo_platform/resources/evaluation/api.md +303 -0
- nemo_platform/resources/evaluation/benchmark_job_results.py +541 -0
- nemo_platform/resources/evaluation/benchmark_jobs/__init__.py +48 -0
- nemo_platform/resources/evaluation/benchmark_jobs/benchmark_jobs.py +900 -0
- nemo_platform/resources/evaluation/benchmark_jobs/results/__init__.py +76 -0
- nemo_platform/resources/evaluation/benchmark_jobs/results/aggregate_scores.py +197 -0
- nemo_platform/resources/evaluation/benchmark_jobs/results/artifacts.py +206 -0
- nemo_platform/resources/evaluation/benchmark_jobs/results/results.py +516 -0
- nemo_platform/resources/evaluation/benchmark_jobs/results/row_scores.py +213 -0
- nemo_platform/resources/evaluation/benchmarks.py +690 -0
- nemo_platform/resources/evaluation/evaluation.py +277 -0
- nemo_platform/resources/evaluation/metric_job_results.py +533 -0
- nemo_platform/resources/evaluation/metric_jobs/__init__.py +48 -0
- nemo_platform/resources/evaluation/metric_jobs/metric_jobs.py +888 -0
- nemo_platform/resources/evaluation/metric_jobs/results/__init__.py +76 -0
- nemo_platform/resources/evaluation/metric_jobs/results/aggregate_scores.py +197 -0
- nemo_platform/resources/evaluation/metric_jobs/results/artifacts.py +206 -0
- nemo_platform/resources/evaluation/metric_jobs/results/results.py +512 -0
- nemo_platform/resources/evaluation/metric_jobs/results/row_scores.py +213 -0
- nemo_platform/resources/evaluation/metrics.py +3495 -0
- nemo_platform/resources/files/__init__.py +62 -0
- nemo_platform/resources/files/api.md +65 -0
- nemo_platform/resources/files/files.py +683 -0
- nemo_platform/resources/files/filesets.py +783 -0
- nemo_platform/resources/files/otlp/__init__.py +48 -0
- nemo_platform/resources/files/otlp/logs.py +337 -0
- nemo_platform/resources/files/otlp/otlp.py +117 -0
- nemo_platform/resources/guardrail/__init__.py +48 -0
- nemo_platform/resources/guardrail/api.md +116 -0
- nemo_platform/resources/guardrail/configs.py +679 -0
- nemo_platform/resources/guardrail/guardrail.py +463 -0
- nemo_platform/resources/iam/__init__.py +48 -0
- nemo_platform/resources/iam/api.md +22 -0
- nemo_platform/resources/iam/iam.py +117 -0
- nemo_platform/resources/iam/role_bindings.py +537 -0
- nemo_platform/resources/inference/__init__.py +118 -0
- nemo_platform/resources/inference/api.md +224 -0
- nemo_platform/resources/inference/deployment_configs/__init__.py +48 -0
- nemo_platform/resources/inference/deployment_configs/deployment_configs.py +756 -0
- nemo_platform/resources/inference/deployment_configs/versions.py +418 -0
- nemo_platform/resources/inference/deployments/__init__.py +48 -0
- nemo_platform/resources/inference/deployments/deployments.py +1017 -0
- nemo_platform/resources/inference/deployments/versions.py +434 -0
- nemo_platform/resources/inference/gateway/__init__.py +76 -0
- nemo_platform/resources/inference/gateway/gateway.py +181 -0
- nemo_platform/resources/inference/gateway/model.py +708 -0
- nemo_platform/resources/inference/gateway/openai/__init__.py +48 -0
- nemo_platform/resources/inference/gateway/openai/openai.py +700 -0
- nemo_platform/resources/inference/gateway/openai/v1/__init__.py +48 -0
- nemo_platform/resources/inference/gateway/openai/v1/models.py +290 -0
- nemo_platform/resources/inference/gateway/openai/v1/v1.py +117 -0
- nemo_platform/resources/inference/gateway/provider.py +749 -0
- nemo_platform/resources/inference/inference.py +277 -0
- nemo_platform/resources/inference/models.py +290 -0
- nemo_platform/resources/inference/providers.py +1049 -0
- nemo_platform/resources/inference/virtual_models.py +846 -0
- nemo_platform/resources/intake/__init__.py +132 -0
- nemo_platform/resources/intake/api.md +268 -0
- nemo_platform/resources/intake/apps/__init__.py +48 -0
- nemo_platform/resources/intake/apps/apps.py +718 -0
- nemo_platform/resources/intake/apps/tasks.py +731 -0
- nemo_platform/resources/intake/entries/__init__.py +48 -0
- nemo_platform/resources/intake/entries/entries.py +855 -0
- nemo_platform/resources/intake/entries/events.py +320 -0
- nemo_platform/resources/intake/evaluator_results.py +512 -0
- nemo_platform/resources/intake/exports/__init__.py +48 -0
- nemo_platform/resources/intake/exports/exports.py +229 -0
- nemo_platform/resources/intake/exports/jobs.py +464 -0
- nemo_platform/resources/intake/ingest/__init__.py +76 -0
- nemo_platform/resources/intake/ingest/atif.py +242 -0
- nemo_platform/resources/intake/ingest/chat_completions.py +270 -0
- nemo_platform/resources/intake/ingest/ingest.py +181 -0
- nemo_platform/resources/intake/ingest/otlp/__init__.py +48 -0
- nemo_platform/resources/intake/ingest/otlp/otlp.py +117 -0
- nemo_platform/resources/intake/ingest/otlp/v1/__init__.py +48 -0
- nemo_platform/resources/intake/ingest/otlp/v1/traces.py +183 -0
- nemo_platform/resources/intake/ingest/otlp/v1/v1.py +117 -0
- nemo_platform/resources/intake/intake.py +309 -0
- nemo_platform/resources/intake/spans/__init__.py +48 -0
- nemo_platform/resources/intake/spans/evaluator_results.py +197 -0
- nemo_platform/resources/intake/spans/spans.py +367 -0
- nemo_platform/resources/intake/traces.py +351 -0
- nemo_platform/resources/jobs/__init__.py +76 -0
- nemo_platform/resources/jobs/api.md +118 -0
- nemo_platform/resources/jobs/jobs.py +1300 -0
- nemo_platform/resources/jobs/results.py +555 -0
- nemo_platform/resources/jobs/steps.py +491 -0
- nemo_platform/resources/jobs/tasks.py +470 -0
- nemo_platform/resources/members/__init__.py +34 -0
- nemo_platform/resources/members/api.md +19 -0
- nemo_platform/resources/members/members.py +634 -0
- nemo_platform/resources/models/__init__.py +48 -0
- nemo_platform/resources/models/adapters.py +514 -0
- nemo_platform/resources/models/api.md +44 -0
- nemo_platform/resources/models/models.py +981 -0
- nemo_platform/resources/projects/__init__.py +34 -0
- nemo_platform/resources/projects/api.md +21 -0
- nemo_platform/resources/projects/projects.py +736 -0
- nemo_platform/resources/safe_synthesizer/__init__.py +48 -0
- nemo_platform/resources/safe_synthesizer/api.md +62 -0
- nemo_platform/resources/safe_synthesizer/jobs/__init__.py +48 -0
- nemo_platform/resources/safe_synthesizer/jobs/jobs.py +895 -0
- nemo_platform/resources/safe_synthesizer/jobs/results.py +819 -0
- nemo_platform/resources/safe_synthesizer/safe_synthesizer.py +117 -0
- nemo_platform/resources/secrets/__init__.py +48 -0
- nemo_platform/resources/secrets/admin.py +150 -0
- nemo_platform/resources/secrets/api.md +34 -0
- nemo_platform/resources/secrets/secrets.py +756 -0
- nemo_platform/resources/workspaces/__init__.py +34 -0
- nemo_platform/resources/workspaces/api.md +20 -0
- nemo_platform/resources/workspaces/workspaces.py +734 -0
- nemo_platform/skills/__init__.py +28 -0
- nemo_platform/skills/inference/SKILL.md +564 -0
- nemo_platform/skills/nemo-build-agent/SKILL.md +221 -0
- nemo_platform/skills/nemo-build-agent/references/templates/data-designer-config.py +80 -0
- nemo_platform/skills/nemo-evaluator/SKILL.md +140 -0
- nemo_platform/skills/nemo-explore/SKILL.md +70 -0
- nemo_platform/skills/nemo-files/SKILL.md +81 -0
- nemo_platform/skills/nemo-fine-tune/SKILL.md +43 -0
- nemo_platform/skills/nemo-guardrails/SKILL.md +519 -0
- nemo_platform/skills/nemo-secrets/SKILL.md +60 -0
- nemo_platform/skills/nemo-skill-selection/SKILL.md +146 -0
- nemo_platform/skills/nemo-spec/SKILL.md +76 -0
- nemo_platform/skills/nemo-spec/references/templates/agent-spec.md +68 -0
- nemo_platform/skills/nemo-status/SKILL.md +115 -0
- nemo_platform/skills/nemo-teardown/SKILL.md +185 -0
- nemo_platform/skills/nemo-try-agent/SKILL.md +102 -0
- nemo_platform/types/__init__.py +54 -0
- nemo_platform/types/adapters/__init__.py +24 -0
- nemo_platform/types/adapters/adapter_create_params.py +63 -0
- nemo_platform/types/adapters/adapter_entity_filter_param.py +56 -0
- nemo_platform/types/adapters/adapter_list_params.py +46 -0
- nemo_platform/types/adapters/adapter_patch_params.py +35 -0
- nemo_platform/types/adapters/adapters_page.py +37 -0
- nemo_platform/types/audit/__init__.py +18 -0
- nemo_platform/types/audit/jobs/__init__.py +18 -0
- nemo_platform/types/entities/__init__.py +26 -0
- nemo_platform/types/entities/entities_page.py +37 -0
- nemo_platform/types/entities/entity.py +63 -0
- nemo_platform/types/entities/entity_create_params.py +46 -0
- nemo_platform/types/entities/entity_delete_entity_by_name_params.py +31 -0
- nemo_platform/types/entities/entity_get_entity_by_name_params.py +31 -0
- nemo_platform/types/entities/entity_list_params.py +47 -0
- nemo_platform/types/entities/entity_update_entity_by_name_params.py +52 -0
- nemo_platform/types/evaluation/__init__.py +242 -0
- nemo_platform/types/evaluation/agent.py +68 -0
- nemo_platform/types/evaluation/agent_goal_accuracy_metric.py +92 -0
- nemo_platform/types/evaluation/agent_goal_accuracy_metric_param.py +69 -0
- nemo_platform/types/evaluation/agent_goal_accuracy_metric_param_param.py +70 -0
- nemo_platform/types/evaluation/agent_goal_accuracy_metric_response.py +88 -0
- nemo_platform/types/evaluation/agent_param.py +69 -0
- nemo_platform/types/evaluation/aggregate_range_score.py +65 -0
- nemo_platform/types/evaluation/aggregate_rubric_score.py +64 -0
- nemo_platform/types/evaluation/aggregated_metric_result.py +34 -0
- nemo_platform/types/evaluation/answer_accuracy_metric.py +89 -0
- nemo_platform/types/evaluation/answer_accuracy_metric_param.py +66 -0
- nemo_platform/types/evaluation/answer_accuracy_metric_param_param.py +67 -0
- nemo_platform/types/evaluation/answer_accuracy_metric_response.py +85 -0
- nemo_platform/types/evaluation/benchmark.py +81 -0
- nemo_platform/types/evaluation/benchmark_create_params.py +64 -0
- nemo_platform/types/evaluation/benchmark_create_response.py +26 -0
- nemo_platform/types/evaluation/benchmark_evaluation_job.py +75 -0
- nemo_platform/types/evaluation/benchmark_evaluation_jobs_list_filter_param.py +49 -0
- nemo_platform/types/evaluation/benchmark_evaluation_jobs_page.py +37 -0
- nemo_platform/types/evaluation/benchmark_evaluation_jobs_sort_field.py +22 -0
- nemo_platform/types/evaluation/benchmark_job_create_params.py +58 -0
- nemo_platform/types/evaluation/benchmark_job_get_logs_params.py +30 -0
- nemo_platform/types/evaluation/benchmark_job_list_params.py +44 -0
- nemo_platform/types/evaluation/benchmark_job_result.py +93 -0
- nemo_platform/types/evaluation/benchmark_job_result_list_params.py +116 -0
- nemo_platform/types/evaluation/benchmark_job_result_retrieve_params.py +50 -0
- nemo_platform/types/evaluation/benchmark_job_results_list_response.py +37 -0
- nemo_platform/types/evaluation/benchmark_jobs/__init__.py +18 -0
- nemo_platform/types/evaluation/benchmark_jobs/results/__init__.py +22 -0
- nemo_platform/types/evaluation/benchmark_jobs/results/benchmark_evaluation_result.py +30 -0
- nemo_platform/types/evaluation/benchmark_jobs/results/benchmark_metric_result.py +44 -0
- nemo_platform/types/evaluation/benchmark_jobs/results/row_score_download_params.py +28 -0
- nemo_platform/types/evaluation/benchmark_list_params.py +84 -0
- nemo_platform/types/evaluation/benchmark_offline_job.py +43 -0
- nemo_platform/types/evaluation/benchmark_offline_job_param.py +44 -0
- nemo_platform/types/evaluation/benchmark_online_agent_job.py +71 -0
- nemo_platform/types/evaluation/benchmark_online_agent_job_param.py +74 -0
- nemo_platform/types/evaluation/benchmark_online_job.py +66 -0
- nemo_platform/types/evaluation/benchmark_online_job_param.py +68 -0
- nemo_platform/types/evaluation/benchmark_ref.py +22 -0
- nemo_platform/types/evaluation/benchmark_retrieve_params.py +29 -0
- nemo_platform/types/evaluation/benchmark_retrieve_response.py +27 -0
- nemo_platform/types/evaluation/benchmarks_list_response.py +42 -0
- nemo_platform/types/evaluation/bleu_metric.py +76 -0
- nemo_platform/types/evaluation/bleu_metric_param.py +50 -0
- nemo_platform/types/evaluation/bleu_metric_param_param.py +52 -0
- nemo_platform/types/evaluation/bleu_metric_response.py +69 -0
- nemo_platform/types/evaluation/built_in_dataset.py +46 -0
- nemo_platform/types/evaluation/context_entity_recall_metric.py +89 -0
- nemo_platform/types/evaluation/context_entity_recall_metric_param.py +66 -0
- nemo_platform/types/evaluation/context_entity_recall_metric_param_param.py +67 -0
- nemo_platform/types/evaluation/context_entity_recall_metric_response.py +85 -0
- nemo_platform/types/evaluation/context_precision_metric.py +89 -0
- nemo_platform/types/evaluation/context_precision_metric_param.py +66 -0
- nemo_platform/types/evaluation/context_precision_metric_param_param.py +67 -0
- nemo_platform/types/evaluation/context_precision_metric_response.py +85 -0
- nemo_platform/types/evaluation/context_recall_metric.py +89 -0
- nemo_platform/types/evaluation/context_recall_metric_param.py +66 -0
- nemo_platform/types/evaluation/context_recall_metric_param_param.py +67 -0
- nemo_platform/types/evaluation/context_recall_metric_response.py +85 -0
- nemo_platform/types/evaluation/context_relevance_metric.py +89 -0
- nemo_platform/types/evaluation/context_relevance_metric_param.py +66 -0
- nemo_platform/types/evaluation/context_relevance_metric_param_param.py +67 -0
- nemo_platform/types/evaluation/context_relevance_metric_response.py +85 -0
- nemo_platform/types/evaluation/dataset_rows.py +35 -0
- nemo_platform/types/evaluation/dataset_rows_param.py +36 -0
- nemo_platform/types/evaluation/evaluate_dataset_rows_param.py +33 -0
- nemo_platform/types/evaluation/exact_match_metric.py +79 -0
- nemo_platform/types/evaluation/exact_match_metric_param.py +53 -0
- nemo_platform/types/evaluation/exact_match_metric_param_param.py +53 -0
- nemo_platform/types/evaluation/exact_match_metric_response.py +72 -0
- nemo_platform/types/evaluation/extended_benchmark.py +130 -0
- nemo_platform/types/evaluation/f1_metric.py +76 -0
- nemo_platform/types/evaluation/f1_metric_param.py +50 -0
- nemo_platform/types/evaluation/f1_metric_param_param.py +50 -0
- nemo_platform/types/evaluation/f1_metric_response.py +69 -0
- nemo_platform/types/evaluation/faithfulness_metric.py +89 -0
- nemo_platform/types/evaluation/faithfulness_metric_param.py +66 -0
- nemo_platform/types/evaluation/faithfulness_metric_param_param.py +67 -0
- nemo_platform/types/evaluation/faithfulness_metric_response.py +85 -0
- nemo_platform/types/evaluation/field_mapping.py +56 -0
- nemo_platform/types/evaluation/field_mapping_param.py +57 -0
- nemo_platform/types/evaluation/fileset.py +48 -0
- nemo_platform/types/evaluation/fileset_param.py +49 -0
- nemo_platform/types/evaluation/fileset_ref.py +22 -0
- nemo_platform/types/evaluation/histogram.py +30 -0
- nemo_platform/types/evaluation/histogram_bin.py +33 -0
- nemo_platform/types/evaluation/inference_params.py +65 -0
- nemo_platform/types/evaluation/inference_params_param.py +53 -0
- nemo_platform/types/evaluation/json_score_parser.py +35 -0
- nemo_platform/types/evaluation/json_score_parser_param.py +34 -0
- nemo_platform/types/evaluation/llm_judge_metric.py +126 -0
- nemo_platform/types/evaluation/llm_judge_metric_param.py +103 -0
- nemo_platform/types/evaluation/llm_judge_metric_param_param.py +105 -0
- nemo_platform/types/evaluation/llm_judge_metric_response.py +120 -0
- nemo_platform/types/evaluation/metric_create_params.py +959 -0
- nemo_platform/types/evaluation/metric_create_response.py +75 -0
- nemo_platform/types/evaluation/metric_evaluate_params.py +111 -0
- nemo_platform/types/evaluation/metric_evaluation_job.py +65 -0
- nemo_platform/types/evaluation/metric_evaluation_jobs_list_filter_param.py +49 -0
- nemo_platform/types/evaluation/metric_evaluation_jobs_page.py +37 -0
- nemo_platform/types/evaluation/metric_evaluation_jobs_sort_field.py +22 -0
- nemo_platform/types/evaluation/metric_evaluation_response.py +98 -0
- nemo_platform/types/evaluation/metric_evaluation_row_score.py +45 -0
- nemo_platform/types/evaluation/metric_job_create_params.py +48 -0
- nemo_platform/types/evaluation/metric_job_get_logs_params.py +30 -0
- nemo_platform/types/evaluation/metric_job_list_params.py +44 -0
- nemo_platform/types/evaluation/metric_job_result.py +93 -0
- nemo_platform/types/evaluation/metric_job_result_list_params.py +114 -0
- nemo_platform/types/evaluation/metric_job_result_retrieve_params.py +50 -0
- nemo_platform/types/evaluation/metric_job_results_list_response.py +37 -0
- nemo_platform/types/evaluation/metric_jobs/__init__.py +18 -0
- nemo_platform/types/evaluation/metric_jobs/results/__init__.py +20 -0
- nemo_platform/types/evaluation/metric_jobs/results/row_score_download_params.py +28 -0
- nemo_platform/types/evaluation/metric_list_params.py +73 -0
- nemo_platform/types/evaluation/metric_offline_job.py +106 -0
- nemo_platform/types/evaluation/metric_offline_job_param.py +107 -0
- nemo_platform/types/evaluation/metric_online_agent_job.py +133 -0
- nemo_platform/types/evaluation/metric_online_agent_job_param.py +135 -0
- nemo_platform/types/evaluation/metric_online_job.py +127 -0
- nemo_platform/types/evaluation/metric_online_job_param.py +129 -0
- nemo_platform/types/evaluation/metric_output.py +30 -0
- nemo_platform/types/evaluation/metric_ref.py +22 -0
- nemo_platform/types/evaluation/metric_retrieve_response.py +98 -0
- nemo_platform/types/evaluation/metric_retriever_job.py +64 -0
- nemo_platform/types/evaluation/metric_retriever_job_param.py +65 -0
- nemo_platform/types/evaluation/metric_type.py +47 -0
- nemo_platform/types/evaluation/metrics_list_response.py +90 -0
- nemo_platform/types/evaluation/model.py +44 -0
- nemo_platform/types/evaluation/model_param.py +44 -0
- nemo_platform/types/evaluation/model_ref.py +22 -0
- nemo_platform/types/evaluation/nemo_agent_toolkit_remote_metric.py +87 -0
- nemo_platform/types/evaluation/nemo_agent_toolkit_remote_metric_param.py +64 -0
- nemo_platform/types/evaluation/nemo_agent_toolkit_remote_metric_param_param.py +65 -0
- nemo_platform/types/evaluation/nemo_agent_toolkit_remote_metric_response.py +80 -0
- nemo_platform/types/evaluation/noise_sensitivity_metric.py +89 -0
- nemo_platform/types/evaluation/noise_sensitivity_metric_param.py +66 -0
- nemo_platform/types/evaluation/noise_sensitivity_metric_param_param.py +67 -0
- nemo_platform/types/evaluation/noise_sensitivity_metric_response.py +85 -0
- nemo_platform/types/evaluation/number_check_metric.py +104 -0
- nemo_platform/types/evaluation/number_check_metric_param.py +81 -0
- nemo_platform/types/evaluation/number_check_metric_param_param.py +83 -0
- nemo_platform/types/evaluation/number_check_metric_response.py +97 -0
- nemo_platform/types/evaluation/parameter.py +42 -0
- nemo_platform/types/evaluation/parameter_param.py +40 -0
- nemo_platform/types/evaluation/percentiles.py +54 -0
- nemo_platform/types/evaluation/range_score.py +56 -0
- nemo_platform/types/evaluation/range_score_param.py +57 -0
- nemo_platform/types/evaluation/reasoning_params.py +41 -0
- nemo_platform/types/evaluation/reasoning_params_param.py +41 -0
- nemo_platform/types/evaluation/regex_score_parser.py +38 -0
- nemo_platform/types/evaluation/regex_score_parser_param.py +37 -0
- nemo_platform/types/evaluation/remote_metric.py +91 -0
- nemo_platform/types/evaluation/remote_metric_param.py +68 -0
- nemo_platform/types/evaluation/remote_metric_param_param.py +69 -0
- nemo_platform/types/evaluation/remote_metric_response.py +84 -0
- nemo_platform/types/evaluation/remote_score.py +49 -0
- nemo_platform/types/evaluation/remote_score_param.py +50 -0
- nemo_platform/types/evaluation/response_groundedness_metric.py +89 -0
- nemo_platform/types/evaluation/response_groundedness_metric_param.py +66 -0
- nemo_platform/types/evaluation/response_groundedness_metric_param_param.py +67 -0
- nemo_platform/types/evaluation/response_groundedness_metric_response.py +85 -0
- nemo_platform/types/evaluation/response_relevancy_metric.py +95 -0
- nemo_platform/types/evaluation/response_relevancy_metric_param.py +74 -0
- nemo_platform/types/evaluation/response_relevancy_metric_param_param.py +75 -0
- nemo_platform/types/evaluation/response_relevancy_metric_response.py +93 -0
- nemo_platform/types/evaluation/retriever_pipeline.py +34 -0
- nemo_platform/types/evaluation/retriever_pipeline_param.py +35 -0
- nemo_platform/types/evaluation/rouge_metric.py +76 -0
- nemo_platform/types/evaluation/rouge_metric_param.py +53 -0
- nemo_platform/types/evaluation/rouge_metric_param_param.py +53 -0
- nemo_platform/types/evaluation/rouge_metric_response.py +69 -0
- nemo_platform/types/evaluation/row_score.py +59 -0
- nemo_platform/types/evaluation/rubric.py +40 -0
- nemo_platform/types/evaluation/rubric_param.py +40 -0
- nemo_platform/types/evaluation/rubric_score.py +54 -0
- nemo_platform/types/evaluation/rubric_score_param.py +55 -0
- nemo_platform/types/evaluation/rubric_score_stat.py +38 -0
- nemo_platform/types/evaluation/run_config.py +39 -0
- nemo_platform/types/evaluation/run_config_online.py +51 -0
- nemo_platform/types/evaluation/run_config_online_model.py +73 -0
- nemo_platform/types/evaluation/run_config_online_model_param.py +75 -0
- nemo_platform/types/evaluation/run_config_online_param.py +51 -0
- nemo_platform/types/evaluation/run_config_param.py +39 -0
- nemo_platform/types/evaluation/string_check_metric.py +82 -0
- nemo_platform/types/evaluation/string_check_metric_param.py +59 -0
- nemo_platform/types/evaluation/string_check_metric_param_param.py +61 -0
- nemo_platform/types/evaluation/string_check_metric_response.py +75 -0
- nemo_platform/types/evaluation/system_benchmark.py +71 -0
- nemo_platform/types/evaluation/system_benchmark_offline_job.py +58 -0
- nemo_platform/types/evaluation/system_benchmark_offline_job_param.py +60 -0
- nemo_platform/types/evaluation/system_benchmark_online_job.py +54 -0
- nemo_platform/types/evaluation/system_benchmark_online_job_param.py +55 -0
- nemo_platform/types/evaluation/system_metric.py +70 -0
- nemo_platform/types/evaluation/system_metric_param.py +50 -0
- nemo_platform/types/evaluation/system_metric_param_param.py +51 -0
- nemo_platform/types/evaluation/system_metric_response.py +66 -0
- nemo_platform/types/evaluation/tool_call_accuracy_metric.py +70 -0
- nemo_platform/types/evaluation/tool_call_accuracy_metric_param.py +44 -0
- nemo_platform/types/evaluation/tool_call_accuracy_metric_param_param.py +44 -0
- nemo_platform/types/evaluation/tool_call_accuracy_metric_response.py +63 -0
- nemo_platform/types/evaluation/tool_calling_metric.py +70 -0
- nemo_platform/types/evaluation/tool_calling_metric_param.py +47 -0
- nemo_platform/types/evaluation/tool_calling_metric_param_param.py +47 -0
- nemo_platform/types/evaluation/tool_calling_metric_response.py +63 -0
- nemo_platform/types/evaluation/topic_adherence_metric.py +92 -0
- nemo_platform/types/evaluation/topic_adherence_metric_param.py +69 -0
- nemo_platform/types/evaluation/topic_adherence_metric_param_param.py +70 -0
- nemo_platform/types/evaluation/topic_adherence_metric_response.py +88 -0
- nemo_platform/types/files/__init__.py +45 -0
- nemo_platform/types/files/cache_status.py +22 -0
- nemo_platform/types/files/dataset_metadata_content.py +46 -0
- nemo_platform/types/files/dataset_metadata_content_param.py +45 -0
- nemo_platform/types/files/file_list_files_params.py +35 -0
- nemo_platform/types/files/file_upload_file_params.py +28 -0
- nemo_platform/types/files/fileset.py +62 -0
- nemo_platform/types/files/fileset_create_params.py +74 -0
- nemo_platform/types/files/fileset_file.py +36 -0
- nemo_platform/types/files/fileset_filter_param.py +54 -0
- nemo_platform/types/files/fileset_list_params.py +47 -0
- nemo_platform/types/files/fileset_metadata_param.py +46 -0
- nemo_platform/types/files/fileset_metadata_param_param.py +47 -0
- nemo_platform/types/files/fileset_outputs_page.py +37 -0
- nemo_platform/types/files/fileset_purpose.py +22 -0
- nemo_platform/types/files/fileset_update_params.py +49 -0
- nemo_platform/types/files/huggingface_storage_config.py +60 -0
- nemo_platform/types/files/huggingface_storage_config_param.py +60 -0
- nemo_platform/types/files/list_fileset_files_response.py +27 -0
- nemo_platform/types/files/local_storage_config.py +39 -0
- nemo_platform/types/files/local_storage_config_param.py +38 -0
- nemo_platform/types/files/ngc_storage_config.py +66 -0
- nemo_platform/types/files/ngc_storage_config_param.py +66 -0
- nemo_platform/types/files/otlp/__init__.py +22 -0
- nemo_platform/types/files/otlp/log_query_params.py +36 -0
- nemo_platform/types/files/otlp/otel_export_logs_partial_success.py +34 -0
- nemo_platform/types/files/otlp/otel_export_logs_service_response.py +35 -0
- nemo_platform/types/files/s3_storage_config.py +85 -0
- nemo_platform/types/files/s3_storage_config_param.py +85 -0
- nemo_platform/types/files/secret_ref.py +22 -0
- nemo_platform/types/files/storage_config_type.py +22 -0
- nemo_platform/types/guardrail/__init__.py +165 -0
- nemo_platform/types/guardrail/action_rails.py +36 -0
- nemo_platform/types/guardrail/action_rails_param.py +38 -0
- nemo_platform/types/guardrail/activated_rail.py +61 -0
- nemo_platform/types/guardrail/ai_defense_rail_config.py +36 -0
- nemo_platform/types/guardrail/ai_defense_rail_config_param.py +36 -0
- nemo_platform/types/guardrail/auto_align_options.py +29 -0
- nemo_platform/types/guardrail/auto_align_options_param.py +30 -0
- nemo_platform/types/guardrail/auto_align_rail_config.py +35 -0
- nemo_platform/types/guardrail/auto_align_rail_config_param.py +37 -0
- nemo_platform/types/guardrail/cache_stats_config.py +32 -0
- nemo_platform/types/guardrail/cache_stats_config_param.py +32 -0
- nemo_platform/types/guardrail/chat_completion_assistant_message_param.py +45 -0
- nemo_platform/types/guardrail/chat_completion_content_part_image_param.py +34 -0
- nemo_platform/types/guardrail/chat_completion_content_part_text_param.py +32 -0
- nemo_platform/types/guardrail/chat_completion_function_message_param.py +35 -0
- nemo_platform/types/guardrail/chat_completion_message_tool_call_param.py +37 -0
- nemo_platform/types/guardrail/chat_completion_system_message_param.py +35 -0
- nemo_platform/types/guardrail/chat_completion_tool_message_param.py +35 -0
- nemo_platform/types/guardrail/chat_completion_user_message_param.py +41 -0
- nemo_platform/types/guardrail/clavata_rail_config.py +47 -0
- nemo_platform/types/guardrail/clavata_rail_config_param.py +48 -0
- nemo_platform/types/guardrail/clavata_rail_options.py +36 -0
- nemo_platform/types/guardrail/clavata_rail_options_param.py +38 -0
- nemo_platform/types/guardrail/config_create_params.py +36 -0
- nemo_platform/types/guardrail/config_list_params.py +47 -0
- nemo_platform/types/guardrail/config_update_params.py +33 -0
- nemo_platform/types/guardrail/content_safety_config.py +34 -0
- nemo_platform/types/guardrail/content_safety_config_param.py +35 -0
- nemo_platform/types/guardrail/crowd_strike_aidr_rail_config.py +29 -0
- nemo_platform/types/guardrail/crowd_strike_aidr_rail_config_param.py +29 -0
- nemo_platform/types/guardrail/dialog_rails.py +34 -0
- nemo_platform/types/guardrail/dialog_rails_param.py +35 -0
- nemo_platform/types/guardrail/executed_action.py +48 -0
- nemo_platform/types/guardrail/fact_checking_rail_config.py +31 -0
- nemo_platform/types/guardrail/fact_checking_rail_config_param.py +32 -0
- nemo_platform/types/guardrail/fiddler_guardrails.py +35 -0
- nemo_platform/types/guardrail/fiddler_guardrails_param.py +35 -0
- nemo_platform/types/guardrail/function_call_param.py +35 -0
- nemo_platform/types/guardrail/function_param.py +35 -0
- nemo_platform/types/guardrail/g_li_ner_detection.py +51 -0
- nemo_platform/types/guardrail/g_li_ner_detection_options.py +29 -0
- nemo_platform/types/guardrail/g_li_ner_detection_options_param.py +31 -0
- nemo_platform/types/guardrail/g_li_ner_detection_param.py +52 -0
- nemo_platform/types/guardrail/generation_log.py +44 -0
- nemo_platform/types/guardrail/generation_log_options_param.py +50 -0
- nemo_platform/types/guardrail/generation_options_param.py +55 -0
- nemo_platform/types/guardrail/generation_rails_options_param.py +53 -0
- nemo_platform/types/guardrail/generation_stats.py +56 -0
- nemo_platform/types/guardrail/guardrail_check_params.py +164 -0
- nemo_platform/types/guardrail/guardrail_check_response.py +36 -0
- nemo_platform/types/guardrail/guardrail_config.py +59 -0
- nemo_platform/types/guardrail/guardrail_config_filter_param.py +49 -0
- nemo_platform/types/guardrail/guardrail_configs_page.py +37 -0
- nemo_platform/types/guardrail/guardrails_ai_rail_config.py +33 -0
- nemo_platform/types/guardrail/guardrails_ai_rail_config_param.py +35 -0
- nemo_platform/types/guardrail/guardrails_ai_validator_config.py +44 -0
- nemo_platform/types/guardrail/guardrails_ai_validator_config_param.py +45 -0
- nemo_platform/types/guardrail/guardrails_data.py +40 -0
- nemo_platform/types/guardrail/guardrails_data_param.py +67 -0
- nemo_platform/types/guardrail/image_u_rl_param.py +32 -0
- nemo_platform/types/guardrail/injection_detection.py +49 -0
- nemo_platform/types/guardrail/injection_detection_param.py +52 -0
- nemo_platform/types/guardrail/input_rails.py +32 -0
- nemo_platform/types/guardrail/input_rails_param.py +34 -0
- nemo_platform/types/guardrail/instruction.py +30 -0
- nemo_platform/types/guardrail/instruction_param.py +32 -0
- nemo_platform/types/guardrail/jailbreak_detection_config.py +52 -0
- nemo_platform/types/guardrail/jailbreak_detection_config_param.py +58 -0
- nemo_platform/types/guardrail/llm_call_info.py +63 -0
- nemo_platform/types/guardrail/log_adapter_config.py +41 -0
- nemo_platform/types/guardrail/log_adapter_config_param.py +27 -0
- nemo_platform/types/guardrail/message_template.py +30 -0
- nemo_platform/types/guardrail/message_template_param.py +32 -0
- nemo_platform/types/guardrail/model.py +55 -0
- nemo_platform/types/guardrail/model_cache_config.py +36 -0
- nemo_platform/types/guardrail/model_cache_config_param.py +37 -0
- nemo_platform/types/guardrail/model_param.py +55 -0
- nemo_platform/types/guardrail/model_parameters.py +51 -0
- nemo_platform/types/guardrail/model_parameters_param.py +38 -0
- nemo_platform/types/guardrail/multilingual_config.py +40 -0
- nemo_platform/types/guardrail/multilingual_config_param.py +41 -0
- nemo_platform/types/guardrail/output_rails.py +44 -0
- nemo_platform/types/guardrail/output_rails_param.py +46 -0
- nemo_platform/types/guardrail/output_rails_streaming_config.py +44 -0
- nemo_platform/types/guardrail/output_rails_streaming_config_param.py +44 -0
- nemo_platform/types/guardrail/pangea_rail_config.py +33 -0
- nemo_platform/types/guardrail/pangea_rail_config_param.py +34 -0
- nemo_platform/types/guardrail/pangea_rail_options.py +31 -0
- nemo_platform/types/guardrail/pangea_rail_options_param.py +33 -0
- nemo_platform/types/guardrail/patronus_evaluate_api_params.py +38 -0
- nemo_platform/types/guardrail/patronus_evaluate_api_params_param.py +40 -0
- nemo_platform/types/guardrail/patronus_evaluate_config.py +30 -0
- nemo_platform/types/guardrail/patronus_evaluate_config_param.py +31 -0
- nemo_platform/types/guardrail/patronus_evaluation_success_strategy.py +22 -0
- nemo_platform/types/guardrail/patronus_rail_config.py +33 -0
- nemo_platform/types/guardrail/patronus_rail_config_param.py +34 -0
- nemo_platform/types/guardrail/private_ai_detection.py +39 -0
- nemo_platform/types/guardrail/private_ai_detection_options.py +29 -0
- nemo_platform/types/guardrail/private_ai_detection_options_param.py +31 -0
- nemo_platform/types/guardrail/private_ai_detection_param.py +40 -0
- nemo_platform/types/guardrail/rail_status.py +26 -0
- nemo_platform/types/guardrail/rails.py +73 -0
- nemo_platform/types/guardrail/rails_config.py +79 -0
- nemo_platform/types/guardrail/rails_config_data.py +94 -0
- nemo_platform/types/guardrail/rails_config_data_param.py +95 -0
- nemo_platform/types/guardrail/rails_config_param.py +81 -0
- nemo_platform/types/guardrail/rails_param.py +74 -0
- nemo_platform/types/guardrail/reasoning_config.py +32 -0
- nemo_platform/types/guardrail/reasoning_config_param.py +32 -0
- nemo_platform/types/guardrail/regex_detection.py +36 -0
- nemo_platform/types/guardrail/regex_detection_options.py +32 -0
- nemo_platform/types/guardrail/regex_detection_options_param.py +34 -0
- nemo_platform/types/guardrail/regex_detection_param.py +37 -0
- nemo_platform/types/guardrail/retrieval_rails.py +29 -0
- nemo_platform/types/guardrail/retrieval_rails_param.py +31 -0
- nemo_platform/types/guardrail/sensitive_data_detection.py +43 -0
- nemo_platform/types/guardrail/sensitive_data_detection_options.py +37 -0
- nemo_platform/types/guardrail/sensitive_data_detection_options_param.py +39 -0
- nemo_platform/types/guardrail/sensitive_data_detection_param.py +45 -0
- nemo_platform/types/guardrail/single_call_config.py +31 -0
- nemo_platform/types/guardrail/single_call_config_param.py +31 -0
- nemo_platform/types/guardrail/status_enum.py +22 -0
- nemo_platform/types/guardrail/task_prompt.py +63 -0
- nemo_platform/types/guardrail/task_prompt_param.py +65 -0
- nemo_platform/types/guardrail/tool_input_rails.py +36 -0
- nemo_platform/types/guardrail/tool_input_rails_param.py +38 -0
- nemo_platform/types/guardrail/tool_output_rails.py +36 -0
- nemo_platform/types/guardrail/tool_output_rails_param.py +38 -0
- nemo_platform/types/guardrail/tracing_config.py +48 -0
- nemo_platform/types/guardrail/tracing_config_param.py +50 -0
- nemo_platform/types/guardrail/trend_micro_rail_config.py +51 -0
- nemo_platform/types/guardrail/trend_micro_rail_config_param.py +51 -0
- nemo_platform/types/guardrail/user_messages_config.py +42 -0
- nemo_platform/types/guardrail/user_messages_config_param.py +42 -0
- nemo_platform/types/iam/__init__.py +26 -0
- nemo_platform/types/iam/date_range_filter_param.py +36 -0
- nemo_platform/types/iam/role_binding.py +42 -0
- nemo_platform/types/iam/role_binding_create_params.py +39 -0
- nemo_platform/types/iam/role_binding_delete_params.py +30 -0
- nemo_platform/types/iam/role_binding_filter_param.py +49 -0
- nemo_platform/types/iam/role_binding_list_params.py +44 -0
- nemo_platform/types/iam/role_bindings_page.py +37 -0
- nemo_platform/types/inference/__init__.py +60 -0
- nemo_platform/types/inference/deployment_config_create_params.py +47 -0
- nemo_platform/types/inference/deployment_config_list_params.py +46 -0
- nemo_platform/types/inference/deployment_config_update_params.py +37 -0
- nemo_platform/types/inference/deployment_configs/__init__.py +20 -0
- nemo_platform/types/inference/deployment_configs/version_list_response.py +25 -0
- nemo_platform/types/inference/deployment_create_params.py +45 -0
- nemo_platform/types/inference/deployment_list_models_response.py +23 -0
- nemo_platform/types/inference/deployment_list_params.py +52 -0
- nemo_platform/types/inference/deployment_update_params.py +35 -0
- nemo_platform/types/inference/deployment_update_status_params.py +42 -0
- nemo_platform/types/inference/deployments/__init__.py +20 -0
- nemo_platform/types/inference/deployments/version_list_response.py +25 -0
- nemo_platform/types/inference/gateway/__init__.py +38 -0
- nemo_platform/types/inference/gateway/model_patch_params.py +31 -0
- nemo_platform/types/inference/gateway/model_patch_response.py +23 -0
- nemo_platform/types/inference/gateway/model_post_params.py +31 -0
- nemo_platform/types/inference/gateway/model_post_response.py +23 -0
- nemo_platform/types/inference/gateway/model_put_params.py +31 -0
- nemo_platform/types/inference/gateway/model_put_response.py +23 -0
- nemo_platform/types/inference/gateway/openai/__init__.py +18 -0
- nemo_platform/types/inference/gateway/openai/v1/__init__.py +21 -0
- nemo_platform/types/inference/gateway/openai/v1/openai_list_models_resp.py +31 -0
- nemo_platform/types/inference/gateway/openai/v1/openai_model_resp.py +34 -0
- nemo_platform/types/inference/gateway/openai_patch_params.py +29 -0
- nemo_platform/types/inference/gateway/openai_patch_response.py +23 -0
- nemo_platform/types/inference/gateway/openai_post_params.py +29 -0
- nemo_platform/types/inference/gateway/openai_post_response.py +23 -0
- nemo_platform/types/inference/gateway/openai_put_params.py +29 -0
- nemo_platform/types/inference/gateway/openai_put_response.py +23 -0
- nemo_platform/types/inference/gateway/provider_patch_params.py +31 -0
- nemo_platform/types/inference/gateway/provider_patch_response.py +23 -0
- nemo_platform/types/inference/gateway/provider_post_params.py +31 -0
- nemo_platform/types/inference/gateway/provider_post_response.py +23 -0
- nemo_platform/types/inference/gateway/provider_put_params.py +31 -0
- nemo_platform/types/inference/gateway/provider_put_response.py +23 -0
- nemo_platform/types/inference/gateway/provider_ready_response.py +23 -0
- nemo_platform/types/inference/k8s_nim_operator_config.py +57 -0
- nemo_platform/types/inference/k8s_nim_operator_config_param.py +58 -0
- nemo_platform/types/inference/middleware_call.py +55 -0
- nemo_platform/types/inference/middleware_call_param.py +56 -0
- nemo_platform/types/inference/model_deployment.py +97 -0
- nemo_platform/types/inference/model_deployment_config.py +76 -0
- nemo_platform/types/inference/model_deployment_config_filter_param.py +49 -0
- nemo_platform/types/inference/model_deployment_configs_page.py +37 -0
- nemo_platform/types/inference/model_deployment_filter_param.py +56 -0
- nemo_platform/types/inference/model_deployment_status.py +24 -0
- nemo_platform/types/inference/model_deployment_status_history_item.py +37 -0
- nemo_platform/types/inference/model_deployments_page.py +37 -0
- nemo_platform/types/inference/model_provider.py +134 -0
- nemo_platform/types/inference/model_provider_filter_param.py +56 -0
- nemo_platform/types/inference/model_provider_sort.py +24 -0
- nemo_platform/types/inference/model_provider_status.py +24 -0
- nemo_platform/types/inference/model_providers_page.py +37 -0
- nemo_platform/types/inference/model_type.py +22 -0
- nemo_platform/types/inference/nim_deployment.py +96 -0
- nemo_platform/types/inference/nim_deployment_param.py +93 -0
- nemo_platform/types/inference/provider_create_params.py +94 -0
- nemo_platform/types/inference/provider_list_params.py +47 -0
- nemo_platform/types/inference/provider_update_params.py +87 -0
- nemo_platform/types/inference/provider_update_status_params.py +48 -0
- nemo_platform/types/inference/served_model_mapping.py +35 -0
- nemo_platform/types/inference/served_model_mapping_param.py +32 -0
- nemo_platform/types/inference/virtual_model.py +91 -0
- nemo_platform/types/inference/virtual_model_create_params.py +83 -0
- nemo_platform/types/inference/virtual_model_inference_config.py +32 -0
- nemo_platform/types/inference/virtual_model_inference_config_param.py +34 -0
- nemo_platform/types/inference/virtual_model_list_params.py +35 -0
- nemo_platform/types/inference/virtual_model_patch_params.py +80 -0
- nemo_platform/types/inference/virtual_models_page.py +37 -0
- nemo_platform/types/intake/__init__.py +86 -0
- nemo_platform/types/intake/app.py +51 -0
- nemo_platform/types/intake/app_create_params.py +38 -0
- nemo_platform/types/intake/app_filter_param.py +46 -0
- nemo_platform/types/intake/app_list_params.py +44 -0
- nemo_platform/types/intake/app_patch_params.py +35 -0
- nemo_platform/types/intake/app_sort_field.py +22 -0
- nemo_platform/types/intake/apps/__init__.py +26 -0
- nemo_platform/types/intake/apps/task.py +54 -0
- nemo_platform/types/intake/apps/task_create_params.py +40 -0
- nemo_platform/types/intake/apps/task_filter_param.py +49 -0
- nemo_platform/types/intake/apps/task_list_params.py +44 -0
- nemo_platform/types/intake/apps/task_patch_params.py +37 -0
- nemo_platform/types/intake/apps/task_sort_field.py +22 -0
- nemo_platform/types/intake/apps/tasks_page.py +37 -0
- nemo_platform/types/intake/apps_page.py +37 -0
- nemo_platform/types/intake/entries/__init__.py +20 -0
- nemo_platform/types/intake/entries/event_create_params.py +40 -0
- nemo_platform/types/intake/entry.py +98 -0
- nemo_platform/types/intake/entry_context.py +79 -0
- nemo_platform/types/intake/entry_context_filter_param.py +41 -0
- nemo_platform/types/intake/entry_context_param.py +82 -0
- nemo_platform/types/intake/entry_create_params.py +85 -0
- nemo_platform/types/intake/entry_data.py +48 -0
- nemo_platform/types/intake/entry_data_param.py +51 -0
- nemo_platform/types/intake/entry_filter_param.py +85 -0
- nemo_platform/types/intake/entry_list_params.py +47 -0
- nemo_platform/types/intake/entry_patch_params.py +79 -0
- nemo_platform/types/intake/entry_sort_field.py +22 -0
- nemo_platform/types/intake/entry_user_rating_filter_param.py +31 -0
- nemo_platform/types/intake/entrys_page.py +37 -0
- nemo_platform/types/intake/evaluation_context_param.py +41 -0
- nemo_platform/types/intake/evaluator_result.py +52 -0
- nemo_platform/types/intake/evaluator_result_create_params.py +52 -0
- nemo_platform/types/intake/evaluator_result_data_type.py +22 -0
- nemo_platform/types/intake/evaluator_result_event.py +67 -0
- nemo_platform/types/intake/evaluator_result_event_param.py +69 -0
- nemo_platform/types/intake/evaluator_result_filter_param.py +49 -0
- nemo_platform/types/intake/evaluator_result_list_params.py +43 -0
- nemo_platform/types/intake/evaluator_result_sort_field.py +22 -0
- nemo_platform/types/intake/evaluator_results_page.py +37 -0
- nemo_platform/types/intake/export_config_param.py +44 -0
- nemo_platform/types/intake/export_config_param_param.py +45 -0
- nemo_platform/types/intake/export_preview_params.py +34 -0
- nemo_platform/types/intake/export_preview_response.py +39 -0
- nemo_platform/types/intake/exports/__init__.py +28 -0
- nemo_platform/types/intake/exports/export_config.py +44 -0
- nemo_platform/types/intake/exports/export_job.py +76 -0
- nemo_platform/types/intake/exports/export_job_filter_param.py +50 -0
- nemo_platform/types/intake/exports/export_job_sort_field.py +22 -0
- nemo_platform/types/intake/exports/export_jobs_page.py +37 -0
- nemo_platform/types/intake/exports/export_status_details.py +35 -0
- nemo_platform/types/intake/exports/job_create_params.py +40 -0
- nemo_platform/types/intake/exports/job_list_params.py +46 -0
- nemo_platform/types/intake/exports/job_status.py +22 -0
- nemo_platform/types/intake/flexible_entry_request.py +63 -0
- nemo_platform/types/intake/flexible_entry_request_param.py +51 -0
- nemo_platform/types/intake/flexible_entry_response.py +62 -0
- nemo_platform/types/intake/flexible_entry_response_param.py +49 -0
- nemo_platform/types/intake/flexible_message.py +56 -0
- nemo_platform/types/intake/flexible_message_param.py +43 -0
- nemo_platform/types/intake/float_filter_param.py +32 -0
- nemo_platform/types/intake/ingest/__init__.py +37 -0
- nemo_platform/types/intake/ingest/atif_agent_param.py +35 -0
- nemo_platform/types/intake/ingest/atif_content_part_image_param.py +30 -0
- nemo_platform/types/intake/ingest/atif_content_part_param.py +28 -0
- nemo_platform/types/intake/ingest/atif_content_part_text_param.py +28 -0
- nemo_platform/types/intake/ingest/atif_create_params.py +52 -0
- nemo_platform/types/intake/ingest/atif_final_metrics_param.py +37 -0
- nemo_platform/types/intake/ingest/atif_image_source_param.py +28 -0
- nemo_platform/types/intake/ingest/atif_metrics_param.py +41 -0
- nemo_platform/types/intake/ingest/atif_observation_param.py +29 -0
- nemo_platform/types/intake/ingest/atif_observation_result_param.py +36 -0
- nemo_platform/types/intake/ingest/atif_step_agent_param.py +58 -0
- nemo_platform/types/intake/ingest/atif_step_param.py +29 -0
- nemo_platform/types/intake/ingest/atif_step_system_param.py +43 -0
- nemo_platform/types/intake/ingest/atif_step_user_param.py +43 -0
- nemo_platform/types/intake/ingest/atif_subagent_trajectory_ref_param.py +33 -0
- nemo_platform/types/intake/ingest/atif_tool_call_param.py +31 -0
- nemo_platform/types/intake/ingest/chat_completion_create_params.py +66 -0
- nemo_platform/types/intake/ingest/chat_completions_ingest_response.py +26 -0
- nemo_platform/types/intake/ingest/otlp/__init__.py +18 -0
- nemo_platform/types/intake/ingest/otlp/v1/__init__.py +20 -0
- nemo_platform/types/intake/ingest/otlp/v1/ingest_response.py +26 -0
- nemo_platform/types/intake/message_role.py +22 -0
- nemo_platform/types/intake/reviewer_annotation_event.py +95 -0
- nemo_platform/types/intake/reviewer_annotation_event_param.py +97 -0
- nemo_platform/types/intake/span.py +100 -0
- nemo_platform/types/intake/span_evaluation_context.py +40 -0
- nemo_platform/types/intake/span_filter_param.py +95 -0
- nemo_platform/types/intake/span_kind.py +24 -0
- nemo_platform/types/intake/span_list_params.py +46 -0
- nemo_platform/types/intake/span_sort_field.py +22 -0
- nemo_platform/types/intake/span_status.py +22 -0
- nemo_platform/types/intake/spans/__init__.py +20 -0
- nemo_platform/types/intake/spans/evaluator_result_list_response.py +25 -0
- nemo_platform/types/intake/spans_page.py +37 -0
- nemo_platform/types/intake/thumb_direction.py +22 -0
- nemo_platform/types/intake/trace.py +65 -0
- nemo_platform/types/intake/trace_filter_param.py +60 -0
- nemo_platform/types/intake/trace_list_params.py +49 -0
- nemo_platform/types/intake/trace_retrieve_params.py +32 -0
- nemo_platform/types/intake/trace_sort_field.py +22 -0
- nemo_platform/types/intake/traces_page.py +37 -0
- nemo_platform/types/intake/usage.py +65 -0
- nemo_platform/types/intake/usage_param.py +68 -0
- nemo_platform/types/intake/user_action_event.py +65 -0
- nemo_platform/types/intake/user_action_event_param.py +68 -0
- nemo_platform/types/intake/user_feedback_event.py +84 -0
- nemo_platform/types/intake/user_feedback_event_param.py +86 -0
- nemo_platform/types/intake/user_rating.py +68 -0
- nemo_platform/types/intake/user_rating_param.py +70 -0
- nemo_platform/types/jobs/__init__.py +95 -0
- nemo_platform/types/jobs/compute_resource_spec.py +32 -0
- nemo_platform/types/jobs/compute_resource_spec_param.py +32 -0
- nemo_platform/types/jobs/compute_resources.py +46 -0
- nemo_platform/types/jobs/compute_resources_param.py +47 -0
- nemo_platform/types/jobs/container_spec.py +35 -0
- nemo_platform/types/jobs/container_spec_param.py +37 -0
- nemo_platform/types/jobs/cpu_execution_provider.py +46 -0
- nemo_platform/types/jobs/cpu_execution_provider_param.py +46 -0
- nemo_platform/types/jobs/distributed_gpu_execution_provider.py +46 -0
- nemo_platform/types/jobs/distributed_gpu_execution_provider_param.py +46 -0
- nemo_platform/types/jobs/docker_job_execution_profile.py +43 -0
- nemo_platform/types/jobs/docker_job_execution_profile_config.py +52 -0
- nemo_platform/types/jobs/docker_job_network_config.py +27 -0
- nemo_platform/types/jobs/docker_job_storage_config.py +36 -0
- nemo_platform/types/jobs/docker_volume_mount.py +47 -0
- nemo_platform/types/jobs/e2e_job_execution_profile.py +43 -0
- nemo_platform/types/jobs/gpu_execution_provider.py +46 -0
- nemo_platform/types/jobs/gpu_execution_provider_param.py +46 -0
- nemo_platform/types/jobs/image_pull_secret.py +27 -0
- nemo_platform/types/jobs/job_create_params.py +46 -0
- nemo_platform/types/jobs/job_execution_profile_config.py +42 -0
- nemo_platform/types/jobs/job_get_logs_params.py +41 -0
- nemo_platform/types/jobs/job_list_execution_profiles_response.py +37 -0
- nemo_platform/types/jobs/job_list_params.py +47 -0
- nemo_platform/types/jobs/job_update_status_details_params.py +29 -0
- nemo_platform/types/jobs/kubernetes_empty_dir_volume.py +32 -0
- nemo_platform/types/jobs/kubernetes_job_execution_profile.py +43 -0
- nemo_platform/types/jobs/kubernetes_job_execution_profile_config.py +101 -0
- nemo_platform/types/jobs/kubernetes_job_storage_config.py +40 -0
- nemo_platform/types/jobs/kubernetes_object_metadata.py +28 -0
- nemo_platform/types/jobs/kubernetes_persistent_volume_claim.py +32 -0
- nemo_platform/types/jobs/kubernetes_volume.py +37 -0
- nemo_platform/types/jobs/kubernetes_volume_mount.py +38 -0
- nemo_platform/types/jobs/platform_job_environment_variable.py +36 -0
- nemo_platform/types/jobs/platform_job_environment_variable_param.py +37 -0
- nemo_platform/types/jobs/platform_job_list_task_response.py +29 -0
- nemo_platform/types/jobs/platform_job_response.py +75 -0
- nemo_platform/types/jobs/platform_job_responses_page.py +37 -0
- nemo_platform/types/jobs/platform_job_secret_environment_variable_ref.py +27 -0
- nemo_platform/types/jobs/platform_job_secret_environment_variable_ref_param.py +29 -0
- nemo_platform/types/jobs/platform_job_sort_field.py +22 -0
- nemo_platform/types/jobs/platform_job_spec.py +30 -0
- nemo_platform/types/jobs/platform_job_spec_param.py +32 -0
- nemo_platform/types/jobs/platform_job_step.py +75 -0
- nemo_platform/types/jobs/platform_job_step_spec.py +63 -0
- nemo_platform/types/jobs/platform_job_step_spec_param.py +65 -0
- nemo_platform/types/jobs/platform_job_step_with_context.py +67 -0
- nemo_platform/types/jobs/platform_job_step_with_contexts_page.py +37 -0
- nemo_platform/types/jobs/platform_job_steps_list_filter_param.py +38 -0
- nemo_platform/types/jobs/platform_job_task.py +75 -0
- nemo_platform/types/jobs/platform_jobs_list_filter_param.py +51 -0
- nemo_platform/types/jobs/result_create_params.py +34 -0
- nemo_platform/types/jobs/result_list_params.py +34 -0
- nemo_platform/types/jobs/step_lifecycle.py +37 -0
- nemo_platform/types/jobs/step_lifecycle_param.py +37 -0
- nemo_platform/types/jobs/step_list_params.py +44 -0
- nemo_platform/types/jobs/step_update_status_params.py +44 -0
- nemo_platform/types/jobs/subprocess_execution_provider.py +33 -0
- nemo_platform/types/jobs/subprocess_execution_provider_param.py +34 -0
- nemo_platform/types/jobs/subprocess_job_execution_profile.py +36 -0
- nemo_platform/types/jobs/subprocess_job_execution_profile_config.py +49 -0
- nemo_platform/types/jobs/task_create_or_update_params.py +46 -0
- nemo_platform/types/jobs/volcano_job_execution_profile.py +39 -0
- nemo_platform/types/jobs/volcano_job_execution_profile_config.py +114 -0
- nemo_platform/types/members/__init__.py +24 -0
- nemo_platform/types/members/member_create_params.py +40 -0
- nemo_platform/types/members/member_delete_params.py +32 -0
- nemo_platform/types/members/member_update_params.py +37 -0
- nemo_platform/types/members/workspace_member.py +39 -0
- nemo_platform/types/members/workspace_member_list_response.py +29 -0
- nemo_platform/types/models/__init__.py +34 -0
- nemo_platform/types/models/adapter.py +73 -0
- nemo_platform/types/models/adapter_create_params.py +54 -0
- nemo_platform/types/models/adapter_update_params.py +37 -0
- nemo_platform/types/models/base_model_filter_param.py +29 -0
- nemo_platform/types/models/finetuning_type_filter_param.py +31 -0
- nemo_platform/types/models/lora.py +30 -0
- nemo_platform/types/models/lora_param.py +30 -0
- nemo_platform/types/models/model_create_params.py +93 -0
- nemo_platform/types/models/model_entity.py +112 -0
- nemo_platform/types/models/model_entity_filter_param.py +72 -0
- nemo_platform/types/models/model_entity_sort_field.py +22 -0
- nemo_platform/types/models/model_entitys_page.py +37 -0
- nemo_platform/types/models/model_list_params.py +50 -0
- nemo_platform/types/models/model_retrieve_params.py +29 -0
- nemo_platform/types/models/model_update_params.py +86 -0
- nemo_platform/types/projects/__init__.py +25 -0
- nemo_platform/types/projects/project.py +45 -0
- nemo_platform/types/projects/project_create_params.py +37 -0
- nemo_platform/types/projects/project_list_params.py +49 -0
- nemo_platform/types/projects/project_sort_field.py +22 -0
- nemo_platform/types/projects/project_update_params.py +29 -0
- nemo_platform/types/projects/projects_page.py +37 -0
- nemo_platform/types/safe_synthesizer/__init__.py +68 -0
- nemo_platform/types/safe_synthesizer/classify_config.py +41 -0
- nemo_platform/types/safe_synthesizer/classify_config_param.py +43 -0
- nemo_platform/types/safe_synthesizer/column.py +46 -0
- nemo_platform/types/safe_synthesizer/column_actions.py +36 -0
- nemo_platform/types/safe_synthesizer/column_actions_param.py +38 -0
- nemo_platform/types/safe_synthesizer/column_param.py +49 -0
- nemo_platform/types/safe_synthesizer/data_parameters.py +69 -0
- nemo_platform/types/safe_synthesizer/data_parameters_param.py +69 -0
- nemo_platform/types/safe_synthesizer/differential_privacy_hyperparams.py +48 -0
- nemo_platform/types/safe_synthesizer/differential_privacy_hyperparams_param.py +48 -0
- nemo_platform/types/safe_synthesizer/evaluation_parameters.py +64 -0
- nemo_platform/types/safe_synthesizer/evaluation_parameters_param.py +66 -0
- nemo_platform/types/safe_synthesizer/generate_parameters.py +104 -0
- nemo_platform/types/safe_synthesizer/generate_parameters_param.py +102 -0
- nemo_platform/types/safe_synthesizer/gliner_config.py +41 -0
- nemo_platform/types/safe_synthesizer/gliner_config_param.py +41 -0
- nemo_platform/types/safe_synthesizer/globals.py +45 -0
- nemo_platform/types/safe_synthesizer/globals_param.py +47 -0
- nemo_platform/types/safe_synthesizer/job_create_params.py +46 -0
- nemo_platform/types/safe_synthesizer/job_get_logs_params.py +30 -0
- nemo_platform/types/safe_synthesizer/job_list_params.py +44 -0
- nemo_platform/types/safe_synthesizer/jobs/__init__.py +21 -0
- nemo_platform/types/safe_synthesizer/jobs/safe_synthesizer_summary.py +95 -0
- nemo_platform/types/safe_synthesizer/jobs/safe_synthesizer_timing.py +41 -0
- nemo_platform/types/safe_synthesizer/ner_config.py +42 -0
- nemo_platform/types/safe_synthesizer/ner_config_param.py +44 -0
- nemo_platform/types/safe_synthesizer/pii_replacer_config.py +40 -0
- nemo_platform/types/safe_synthesizer/pii_replacer_config_param.py +42 -0
- nemo_platform/types/safe_synthesizer/row.py +50 -0
- nemo_platform/types/safe_synthesizer/row_actions.py +33 -0
- nemo_platform/types/safe_synthesizer/row_actions_param.py +35 -0
- nemo_platform/types/safe_synthesizer/row_param.py +53 -0
- nemo_platform/types/safe_synthesizer/safe_synthesizer_job.py +63 -0
- nemo_platform/types/safe_synthesizer/safe_synthesizer_job_config.py +55 -0
- nemo_platform/types/safe_synthesizer/safe_synthesizer_job_config_param.py +56 -0
- nemo_platform/types/safe_synthesizer/safe_synthesizer_jobs_list_filter_param.py +49 -0
- nemo_platform/types/safe_synthesizer/safe_synthesizer_jobs_page.py +37 -0
- nemo_platform/types/safe_synthesizer/safe_synthesizer_jobs_sort_field.py +22 -0
- nemo_platform/types/safe_synthesizer/safe_synthesizer_parameters.py +91 -0
- nemo_platform/types/safe_synthesizer/safe_synthesizer_parameters_param.py +92 -0
- nemo_platform/types/safe_synthesizer/step_definition.py +39 -0
- nemo_platform/types/safe_synthesizer/step_definition_param.py +41 -0
- nemo_platform/types/safe_synthesizer/time_series_parameters.py +73 -0
- nemo_platform/types/safe_synthesizer/time_series_parameters_param.py +74 -0
- nemo_platform/types/safe_synthesizer/training_hyperparams.py +165 -0
- nemo_platform/types/safe_synthesizer/training_hyperparams_param.py +167 -0
- nemo_platform/types/safe_synthesizer/validation_parameters.py +54 -0
- nemo_platform/types/safe_synthesizer/validation_parameters_param.py +54 -0
- nemo_platform/types/secrets/__init__.py +28 -0
- nemo_platform/types/secrets/platform_secret_access_response.py +33 -0
- nemo_platform/types/secrets/platform_secret_admin_rotation_response.py +28 -0
- nemo_platform/types/secrets/platform_secret_response.py +40 -0
- nemo_platform/types/secrets/platform_secret_responses_page.py +37 -0
- nemo_platform/types/secrets/secret_create_params.py +39 -0
- nemo_platform/types/secrets/secret_list_params.py +32 -0
- nemo_platform/types/secrets/secret_update_params.py +32 -0
- nemo_platform/types/shared/__init__.py +50 -0
- nemo_platform/types/shared/api_endpoint_data.py +43 -0
- nemo_platform/types/shared/auth_context.py +48 -0
- nemo_platform/types/shared/auth_discovery_response.py +32 -0
- nemo_platform/types/shared/backend_format.py +22 -0
- nemo_platform/types/shared/datetime_filter.py +33 -0
- nemo_platform/types/shared/delete_response.py +33 -0
- nemo_platform/types/shared/error_response.py +46 -0
- nemo_platform/types/shared/field_error.py +28 -0
- nemo_platform/types/shared/file_storage_type.py +22 -0
- nemo_platform/types/shared/fileset_metadata.py +46 -0
- nemo_platform/types/shared/finetuning_type.py +48 -0
- nemo_platform/types/shared/generic_sort_field.py +22 -0
- nemo_platform/types/shared/http_validation_error.py +27 -0
- nemo_platform/types/shared/linear_layer_spec.py +33 -0
- nemo_platform/types/shared/mamba_config.py +44 -0
- nemo_platform/types/shared/mo_e_config.py +41 -0
- nemo_platform/types/shared/model_metadata_content.py +38 -0
- nemo_platform/types/shared/model_spec.py +111 -0
- nemo_platform/types/shared/oidc_discovery_response.py +42 -0
- nemo_platform/types/shared/pagination_data.py +37 -0
- nemo_platform/types/shared/platform_job_list_result_response.py +27 -0
- nemo_platform/types/shared/platform_job_log.py +34 -0
- nemo_platform/types/shared/platform_job_log_page.py +33 -0
- nemo_platform/types/shared/platform_job_result_response.py +44 -0
- nemo_platform/types/shared/platform_job_status.py +24 -0
- nemo_platform/types/shared/platform_job_status_response.py +48 -0
- nemo_platform/types/shared/platform_job_step_status_response.py +48 -0
- nemo_platform/types/shared/platform_job_task_status_response.py +47 -0
- nemo_platform/types/shared/prompt_data.py +47 -0
- nemo_platform/types/shared/sliding_window_config.py +27 -0
- nemo_platform/types/shared/tool_call_config.py +46 -0
- nemo_platform/types/shared/tool_calling_metadata_content.py +48 -0
- nemo_platform/types/shared/validation_error.py +34 -0
- nemo_platform/types/shared_params/__init__.py +33 -0
- nemo_platform/types/shared_params/api_endpoint_data.py +38 -0
- nemo_platform/types/shared_params/backend_format.py +24 -0
- nemo_platform/types/shared_params/datetime_filter.py +34 -0
- nemo_platform/types/shared_params/file_storage_type.py +24 -0
- nemo_platform/types/shared_params/finetuning_type.py +50 -0
- nemo_platform/types/shared_params/generic_sort_field.py +24 -0
- nemo_platform/types/shared_params/linear_layer_spec.py +35 -0
- nemo_platform/types/shared_params/mamba_config.py +44 -0
- nemo_platform/types/shared_params/mo_e_config.py +41 -0
- nemo_platform/types/shared_params/model_metadata_content.py +39 -0
- nemo_platform/types/shared_params/model_spec.py +113 -0
- nemo_platform/types/shared_params/platform_job_status.py +26 -0
- nemo_platform/types/shared_params/prompt_data.py +48 -0
- nemo_platform/types/shared_params/sliding_window_config.py +29 -0
- nemo_platform/types/shared_params/tool_call_config.py +46 -0
- nemo_platform/types/shared_params/tool_calling_metadata_content.py +48 -0
- nemo_platform/types/workspaces/__init__.py +24 -0
- nemo_platform/types/workspaces/workspace.py +48 -0
- nemo_platform/types/workspaces/workspace_create_params.py +41 -0
- nemo_platform/types/workspaces/workspace_list_params.py +47 -0
- nemo_platform/types/workspaces/workspace_update_params.py +27 -0
- nemo_platform/types/workspaces/workspaces_page.py +37 -0
- nemo_platform/ui/__init__.py +15 -0
- nemo_platform/ui/output.py +109 -0
- nemo_platform/ui/prompts.py +678 -0
- nemo_platform_plugin/.agents/skills/creating-a-plugin/SKILL.md +275 -0
- nemo_platform_plugin/.agents/skills/plugin-config/SKILL.md +168 -0
- nemo_platform_plugin/.agents/skills/plugin-controller/SKILL.md +320 -0
- nemo_platform_plugin/.agents/skills/plugin-entities/SKILL.md +290 -0
- nemo_platform_plugin/.agents/skills/plugin-function/SKILL.md +292 -0
- nemo_platform_plugin/.agents/skills/plugin-inference-middleware/SKILL.md +346 -0
- nemo_platform_plugin/.agents/skills/plugin-job/SKILL.md +247 -0
- nemo_platform_plugin/.agents/skills/plugin-platform-services/SKILL.md +239 -0
- nemo_platform_plugin/.agents/skills/plugin-platform-services/services-reference.md +192 -0
- nemo_platform_plugin/.agents/skills/plugin-service/SKILL.md +181 -0
- nemo_platform_plugin/.agents/skills/plugin-service/crud-example.md +353 -0
- nemo_platform_plugin/.agents/skills/plugin-testing/SKILL.md +206 -0
- nemo_platform_plugin/README.md +43 -0
- nemo_platform_plugin/_base.py +53 -0
- nemo_platform_plugin/_spec_flags.py +573 -0
- nemo_platform_plugin/api/filter.py +189 -0
- nemo_platform_plugin/api/parsed_filter.py +288 -0
- nemo_platform_plugin/api/text_filter.py +291 -0
- nemo_platform_plugin/cli.py +152 -0
- nemo_platform_plugin/cli_errors.py +232 -0
- nemo_platform_plugin/cli_renderer.py +92 -0
- nemo_platform_plugin/commands.py +1433 -0
- nemo_platform_plugin/config.py +844 -0
- nemo_platform_plugin/controller.py +148 -0
- nemo_platform_plugin/dependencies.py +64 -0
- nemo_platform_plugin/discovery.py +484 -0
- nemo_platform_plugin/docs/ARCHITECTURE.md +160 -0
- nemo_platform_plugin/docs/CONFIG.md +151 -0
- nemo_platform_plugin/docs/CONTROLLER.md +209 -0
- nemo_platform_plugin/docs/ENTITY.md +186 -0
- nemo_platform_plugin/docs/INFERENCE_MIDDLEWARE.md +408 -0
- nemo_platform_plugin/docs/JOB.md +237 -0
- nemo_platform_plugin/docs/QUICKSTART.md +306 -0
- nemo_platform_plugin/docs/SERVICE.md +302 -0
- nemo_platform_plugin/entities.py +793 -0
- nemo_platform_plugin/entity.py +79 -0
- nemo_platform_plugin/entity_client.py +54 -0
- nemo_platform_plugin/filter_ops.py +169 -0
- nemo_platform_plugin/function.py +299 -0
- nemo_platform_plugin/function_context.py +61 -0
- nemo_platform_plugin/functions/frames.py +69 -0
- nemo_platform_plugin/functions/routes.py +454 -0
- nemo_platform_plugin/inference_middleware.py +1191 -0
- nemo_platform_plugin/interface.py +28 -0
- nemo_platform_plugin/job.py +370 -0
- nemo_platform_plugin/job_context.py +86 -0
- nemo_platform_plugin/job_results.py +209 -0
- nemo_platform_plugin/jobs/_cli_options.py +204 -0
- nemo_platform_plugin/jobs/api_factory.py +1178 -0
- nemo_platform_plugin/jobs/constants.py +27 -0
- nemo_platform_plugin/jobs/docker.py +76 -0
- nemo_platform_plugin/jobs/exceptions.py +8 -0
- nemo_platform_plugin/jobs/file_manager.py +238 -0
- nemo_platform_plugin/jobs/openapi_utils.py +174 -0
- nemo_platform_plugin/jobs/profile.py +110 -0
- nemo_platform_plugin/jobs/result_manager.py +291 -0
- nemo_platform_plugin/jobs/routes.py +297 -0
- nemo_platform_plugin/jobs/schemas.py +174 -0
- nemo_platform_plugin/refs.py +154 -0
- nemo_platform_plugin/run_dependencies.py +137 -0
- nemo_platform_plugin/scheduler.py +537 -0
- nemo_platform_plugin/schema.py +216 -0
- nemo_platform_plugin/sdk.py +34 -0
- nemo_platform_plugin/seed.py +71 -0
- nemo_platform_plugin/service.py +130 -0
- nemo_platform_plugin/tasks/dispatcher.py +226 -0
- nemo_platform_plugin-0.1.0.dist-info/METADATA +88 -0
- nemo_platform_plugin-0.1.0.dist-info/RECORD +1224 -0
- nemo_platform_plugin-0.1.0.dist-info/WHEEL +4 -0
|
@@ -0,0 +1,803 @@
|
|
|
1
|
+
# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
2
|
+
# SPDX-License-Identifier: Apache-2.0
|
|
3
|
+
|
|
4
|
+
"""Authentication commands for NeMo Platform CLI.
|
|
5
|
+
|
|
6
|
+
Scope normalization, JWT decode, and scope validation logic live in
|
|
7
|
+
nemo_platform.auth.helpers.
|
|
8
|
+
"""
|
|
9
|
+
|
|
10
|
+
from __future__ import annotations
|
|
11
|
+
|
|
12
|
+
import asyncio
|
|
13
|
+
import time
|
|
14
|
+
from typing import Annotated, cast
|
|
15
|
+
|
|
16
|
+
import httpx
|
|
17
|
+
import typer
|
|
18
|
+
|
|
19
|
+
from nemo_platform.auth.helpers import (
|
|
20
|
+
AuthError,
|
|
21
|
+
build_effective_scope,
|
|
22
|
+
decode_jwt_claims,
|
|
23
|
+
discover_nmp_config,
|
|
24
|
+
generate_unsigned_jwt,
|
|
25
|
+
is_unsigned_jwt,
|
|
26
|
+
normalize_scope_prefix,
|
|
27
|
+
validate_requested_scopes_granted,
|
|
28
|
+
)
|
|
29
|
+
from nemo_platform.auth.token_provider import OIDCTokenProvider, TokenSet
|
|
30
|
+
from nemo_platform.cli.core.context import CLIContext
|
|
31
|
+
from nemo_platform.cli.core.errors import handle_errors
|
|
32
|
+
from nemo_platform.cli.core.help_formatter import create_typer_app
|
|
33
|
+
from nemo_platform.config.models import ConfigParams, Context
|
|
34
|
+
|
|
35
|
+
app = create_typer_app(
|
|
36
|
+
name="auth",
|
|
37
|
+
help="Manage authentication for NeMo Platform.",
|
|
38
|
+
)
|
|
39
|
+
|
|
40
|
+
|
|
41
|
+
def is_auth_disabled(base_url: str, timeout: float = 3.0) -> bool | None:
|
|
42
|
+
"""Check whether authentication is disabled on the cluster.
|
|
43
|
+
|
|
44
|
+
Returns:
|
|
45
|
+
True if auth is definitely disabled, False if enabled, None if unreachable.
|
|
46
|
+
"""
|
|
47
|
+
try:
|
|
48
|
+
return not discover_nmp_config(base_url, timeout=timeout).auth_enabled
|
|
49
|
+
except httpx.HTTPError:
|
|
50
|
+
return None
|
|
51
|
+
|
|
52
|
+
|
|
53
|
+
def ensure_valid_token(context: Context, refresh_buffer_seconds: int = 300) -> bool:
|
|
54
|
+
"""
|
|
55
|
+
Check if the current token is valid and refresh if needed.
|
|
56
|
+
|
|
57
|
+
This function should be called before making API requests to ensure
|
|
58
|
+
the access token is valid. If the token is expired or about to expire
|
|
59
|
+
(within refresh_buffer_seconds), it will automatically refresh using
|
|
60
|
+
the stored refresh token.
|
|
61
|
+
|
|
62
|
+
Args:
|
|
63
|
+
refresh_buffer_seconds: Refresh the token if it expires within this many seconds.
|
|
64
|
+
Default is 300 (5 minutes).
|
|
65
|
+
|
|
66
|
+
Returns:
|
|
67
|
+
True if a valid token is available (possibly after refresh), False otherwise.
|
|
68
|
+
"""
|
|
69
|
+
from datetime import datetime, timezone
|
|
70
|
+
|
|
71
|
+
from nemo_platform.config.config import Config
|
|
72
|
+
from nemo_platform.config.models import OAuthUser
|
|
73
|
+
|
|
74
|
+
# Only OAuthUser supports token refresh
|
|
75
|
+
if not context.user or not isinstance(context.user, OAuthUser):
|
|
76
|
+
return True # Non-OAuth users don't need refresh
|
|
77
|
+
|
|
78
|
+
# Check if token is expired or about to expire
|
|
79
|
+
token = context.user.token.get_secret_value()
|
|
80
|
+
claims = decode_jwt_claims(token)
|
|
81
|
+
if not claims:
|
|
82
|
+
# Not a JWT, can't check expiry
|
|
83
|
+
return True
|
|
84
|
+
|
|
85
|
+
exp = claims.get("exp")
|
|
86
|
+
if not exp:
|
|
87
|
+
return True
|
|
88
|
+
|
|
89
|
+
exp_dt = datetime.fromtimestamp(exp, tz=timezone.utc)
|
|
90
|
+
now = datetime.now(tz=timezone.utc)
|
|
91
|
+
|
|
92
|
+
# Token is still valid with buffer
|
|
93
|
+
if exp_dt > now + __import__("datetime").timedelta(seconds=refresh_buffer_seconds):
|
|
94
|
+
return True
|
|
95
|
+
|
|
96
|
+
# Token expired or about to expire - try to refresh
|
|
97
|
+
if not context.user.refresh_token:
|
|
98
|
+
return exp_dt > now # Return True only if not yet expired
|
|
99
|
+
|
|
100
|
+
base_url = str(context.cluster.base_url).rstrip("/")
|
|
101
|
+
try:
|
|
102
|
+
nmp_config = discover_nmp_config(base_url)
|
|
103
|
+
except httpx.HTTPError:
|
|
104
|
+
return exp_dt > now
|
|
105
|
+
|
|
106
|
+
if not nmp_config.client_id or not nmp_config.token_endpoint:
|
|
107
|
+
return exp_dt > now
|
|
108
|
+
|
|
109
|
+
effective_scope = build_effective_scope(nmp_config.default_scopes, nmp_config.scope_prefix)
|
|
110
|
+
|
|
111
|
+
try:
|
|
112
|
+
provider = OIDCTokenProvider(
|
|
113
|
+
token_endpoint=nmp_config.token_endpoint,
|
|
114
|
+
client_id=nmp_config.client_id,
|
|
115
|
+
tokens=TokenSet.from_access_token(
|
|
116
|
+
context.user.token.get_secret_value(),
|
|
117
|
+
context.user.refresh_token.get_secret_value(),
|
|
118
|
+
),
|
|
119
|
+
refresh_scope=effective_scope,
|
|
120
|
+
refresh_margin_seconds=float(refresh_buffer_seconds),
|
|
121
|
+
)
|
|
122
|
+
provider.force_refresh()
|
|
123
|
+
|
|
124
|
+
config_params = {
|
|
125
|
+
"access_token": provider.tokens.access_token,
|
|
126
|
+
}
|
|
127
|
+
if provider.tokens.refresh_token:
|
|
128
|
+
config_params["refresh_token"] = provider.tokens.refresh_token
|
|
129
|
+
|
|
130
|
+
Config.write(config_params, context_name=context.context_name) # type: ignore[arg-type]
|
|
131
|
+
typer.echo("[Auto-refreshed expired token]", err=True)
|
|
132
|
+
|
|
133
|
+
return True
|
|
134
|
+
except Exception:
|
|
135
|
+
return exp_dt > now
|
|
136
|
+
|
|
137
|
+
|
|
138
|
+
@app.callback(invoke_without_command=True)
|
|
139
|
+
def auth_callback(ctx: typer.Context) -> None:
|
|
140
|
+
"""Manage authentication for NeMo Platform."""
|
|
141
|
+
if ctx.invoked_subcommand is None:
|
|
142
|
+
typer.echo(ctx.get_help())
|
|
143
|
+
|
|
144
|
+
|
|
145
|
+
@app.command("login")
|
|
146
|
+
@handle_errors
|
|
147
|
+
def login(
|
|
148
|
+
ctx: typer.Context,
|
|
149
|
+
context_name: Annotated[
|
|
150
|
+
str | None,
|
|
151
|
+
typer.Option(
|
|
152
|
+
"--context",
|
|
153
|
+
help="Context to use for this login command.",
|
|
154
|
+
),
|
|
155
|
+
] = None,
|
|
156
|
+
base_url: Annotated[
|
|
157
|
+
str | None,
|
|
158
|
+
typer.Option(
|
|
159
|
+
"--base-url",
|
|
160
|
+
help="Set cluster base URL for the selected context before login",
|
|
161
|
+
),
|
|
162
|
+
] = None,
|
|
163
|
+
no_browser: Annotated[
|
|
164
|
+
bool,
|
|
165
|
+
typer.Option("--no-browser", help="Don't open browser (device flow only)"),
|
|
166
|
+
] = False,
|
|
167
|
+
scope: Annotated[
|
|
168
|
+
str | None,
|
|
169
|
+
typer.Option(
|
|
170
|
+
"--scope",
|
|
171
|
+
help='OAuth scopes to request (space-separated; quote for multiple, e.g. --scope "platform:read secrets:write")',
|
|
172
|
+
),
|
|
173
|
+
] = None,
|
|
174
|
+
username: Annotated[
|
|
175
|
+
str | None,
|
|
176
|
+
typer.Option("--username", help="Username for password grant (CI / non-interactive)"),
|
|
177
|
+
] = None,
|
|
178
|
+
password: Annotated[
|
|
179
|
+
str | None,
|
|
180
|
+
typer.Option("--password", help="Password for password grant (prefer env NMP_OIDC_PASSWORD)"),
|
|
181
|
+
] = None,
|
|
182
|
+
unsigned_token: Annotated[
|
|
183
|
+
bool,
|
|
184
|
+
typer.Option(
|
|
185
|
+
"--unsigned-token",
|
|
186
|
+
help="Generate and save an unsigned JWT for local/testing authentication.",
|
|
187
|
+
rich_help_panel="Unsigned Token Options",
|
|
188
|
+
),
|
|
189
|
+
] = False,
|
|
190
|
+
principal_id: Annotated[
|
|
191
|
+
str | None,
|
|
192
|
+
typer.Option(
|
|
193
|
+
"--principal-id",
|
|
194
|
+
help="Principal ID for the unsigned token (`sub` claim). Defaults to --email.",
|
|
195
|
+
rich_help_panel="Unsigned Token Options",
|
|
196
|
+
),
|
|
197
|
+
] = None,
|
|
198
|
+
email: Annotated[
|
|
199
|
+
str | None,
|
|
200
|
+
typer.Option(
|
|
201
|
+
"--email",
|
|
202
|
+
help="Email claim for the unsigned token (required with --unsigned-token).",
|
|
203
|
+
rich_help_panel="Unsigned Token Options",
|
|
204
|
+
),
|
|
205
|
+
] = None,
|
|
206
|
+
groups: Annotated[
|
|
207
|
+
list[str] | None,
|
|
208
|
+
typer.Option(
|
|
209
|
+
"--group",
|
|
210
|
+
help="Group claim value for unsigned token (repeat for multiple).",
|
|
211
|
+
rich_help_panel="Unsigned Token Options",
|
|
212
|
+
),
|
|
213
|
+
] = None,
|
|
214
|
+
expires_in: Annotated[
|
|
215
|
+
int,
|
|
216
|
+
typer.Option(
|
|
217
|
+
"--expires-in",
|
|
218
|
+
help="Unsigned token expiry in seconds from now.",
|
|
219
|
+
rich_help_panel="Unsigned Token Options",
|
|
220
|
+
),
|
|
221
|
+
] = 3600,
|
|
222
|
+
no_exp: Annotated[
|
|
223
|
+
bool,
|
|
224
|
+
typer.Option(
|
|
225
|
+
"--no-exp",
|
|
226
|
+
help="Omit the exp claim from the unsigned token.",
|
|
227
|
+
rich_help_panel="Unsigned Token Options",
|
|
228
|
+
),
|
|
229
|
+
] = False,
|
|
230
|
+
audience: Annotated[
|
|
231
|
+
str | None,
|
|
232
|
+
typer.Option(
|
|
233
|
+
"--audience",
|
|
234
|
+
help="Audience (`aud`) claim for unsigned token.",
|
|
235
|
+
rich_help_panel="Unsigned Token Options",
|
|
236
|
+
),
|
|
237
|
+
] = None,
|
|
238
|
+
issuer: Annotated[
|
|
239
|
+
str | None,
|
|
240
|
+
typer.Option(
|
|
241
|
+
"--issuer",
|
|
242
|
+
help="Issuer (`iss`) claim for unsigned token.",
|
|
243
|
+
rich_help_panel="Unsigned Token Options",
|
|
244
|
+
),
|
|
245
|
+
] = None,
|
|
246
|
+
) -> None:
|
|
247
|
+
"""Authenticate with the NeMo Platform cluster.
|
|
248
|
+
|
|
249
|
+
Uses device flow (browser) by default, or password grant when username and password are provided (e.g. for CI).
|
|
250
|
+
|
|
251
|
+
For quickstart, use [cyan]`--unsigned-token`[/] to generate an unsigned JWT.
|
|
252
|
+
|
|
253
|
+
Examples:
|
|
254
|
+
# Set base URL and log in
|
|
255
|
+
nemo auth login --base-url https://nemo.example.com
|
|
256
|
+
# Context-specific login
|
|
257
|
+
nemo auth login --context dev --base-url https://nemo.dev.example.com
|
|
258
|
+
# Device flow, open browser
|
|
259
|
+
nemo auth login
|
|
260
|
+
# Device flow, show code only
|
|
261
|
+
nemo auth login --no-browser
|
|
262
|
+
"""
|
|
263
|
+
import os
|
|
264
|
+
|
|
265
|
+
from rich.console import Console
|
|
266
|
+
|
|
267
|
+
from nemo_platform.auth.device_flow import (
|
|
268
|
+
DeviceFlowError,
|
|
269
|
+
authenticate_with_device_flow,
|
|
270
|
+
authenticate_with_password_grant,
|
|
271
|
+
)
|
|
272
|
+
from nemo_platform.config.config import Config
|
|
273
|
+
|
|
274
|
+
cli_context: CLIContext = ctx.obj
|
|
275
|
+
selected_context = cast(str | None, cli_context.overrides.get("current_context"))
|
|
276
|
+
|
|
277
|
+
if context_name is not None:
|
|
278
|
+
selected_context = context_name
|
|
279
|
+
cli_context.overrides["current_context"] = context_name
|
|
280
|
+
cli_context.reset_sdk_context()
|
|
281
|
+
|
|
282
|
+
if base_url is not None:
|
|
283
|
+
base_url_params: ConfigParams = {"base_url": base_url}
|
|
284
|
+
if selected_context is not None:
|
|
285
|
+
base_url_params["current_context"] = selected_context
|
|
286
|
+
try:
|
|
287
|
+
Config.write(base_url_params, context_name=selected_context)
|
|
288
|
+
except Exception as exc:
|
|
289
|
+
context_label = selected_context or "default"
|
|
290
|
+
raise AuthError(f"Failed to set base URL for context '{context_label}': {exc}") from exc
|
|
291
|
+
cli_context.reset_sdk_context()
|
|
292
|
+
|
|
293
|
+
console = Console()
|
|
294
|
+
|
|
295
|
+
unsigned_option_flags = {
|
|
296
|
+
"principal_id": "--principal-id",
|
|
297
|
+
"email": "--email",
|
|
298
|
+
"groups": "--group",
|
|
299
|
+
"expires_in": "--expires-in",
|
|
300
|
+
"no_exp": "--no-exp",
|
|
301
|
+
"audience": "--audience",
|
|
302
|
+
"issuer": "--issuer",
|
|
303
|
+
}
|
|
304
|
+
|
|
305
|
+
if not unsigned_token:
|
|
306
|
+
from click.core import ParameterSource
|
|
307
|
+
|
|
308
|
+
unsigned_options_used = [
|
|
309
|
+
flag
|
|
310
|
+
for param_name, flag in unsigned_option_flags.items()
|
|
311
|
+
if ctx.get_parameter_source(param_name) != ParameterSource.DEFAULT
|
|
312
|
+
]
|
|
313
|
+
if unsigned_options_used:
|
|
314
|
+
raise AuthError(f"Unsigned token option(s) {', '.join(unsigned_options_used)} require --unsigned-token.")
|
|
315
|
+
|
|
316
|
+
if unsigned_token:
|
|
317
|
+
if username or password:
|
|
318
|
+
raise AuthError("Cannot combine --unsigned-token with --username/--password.")
|
|
319
|
+
if email is None:
|
|
320
|
+
raise AuthError("--email is required when using --unsigned-token.")
|
|
321
|
+
|
|
322
|
+
effective_principal_id = principal_id or email
|
|
323
|
+
|
|
324
|
+
context = cli_context.get_sdk_context()
|
|
325
|
+
base_url = str(context.cluster.base_url).rstrip("/")
|
|
326
|
+
|
|
327
|
+
try:
|
|
328
|
+
oidc_config = discover_nmp_config(base_url)
|
|
329
|
+
oidc_login_configured = bool(oidc_config.token_endpoint and oidc_config.client_id)
|
|
330
|
+
if oidc_login_configured:
|
|
331
|
+
raise AuthError(
|
|
332
|
+
"Cluster has OIDC authentication configured. Use 'nemo auth login' instead of '--unsigned-token'."
|
|
333
|
+
)
|
|
334
|
+
if oidc_config.auth_enabled:
|
|
335
|
+
console.print(
|
|
336
|
+
"Cluster authentication is enabled. Ensure [cyan]auth.allow_unsigned_jwt=true[/] on the cluster."
|
|
337
|
+
)
|
|
338
|
+
except httpx.HTTPError:
|
|
339
|
+
console.print("[yellow]Could not verify cluster auth settings via discovery endpoint.[/]")
|
|
340
|
+
|
|
341
|
+
requested_scopes = scope.split() if scope else None
|
|
342
|
+
token = generate_unsigned_jwt(
|
|
343
|
+
principal_id=effective_principal_id,
|
|
344
|
+
email=email,
|
|
345
|
+
groups=groups,
|
|
346
|
+
scopes=requested_scopes,
|
|
347
|
+
expires_in_seconds=None if no_exp else expires_in,
|
|
348
|
+
audience=audience,
|
|
349
|
+
issuer=issuer,
|
|
350
|
+
)
|
|
351
|
+
|
|
352
|
+
config_params: ConfigParams = {"access_token": token}
|
|
353
|
+
if selected_context is not None:
|
|
354
|
+
config_params["current_context"] = context.context_name
|
|
355
|
+
Config.write(config_params, context_name=context.context_name)
|
|
356
|
+
|
|
357
|
+
console.print("\n[bold yellow]Warning:[/] Generated an unsigned JWT (`alg=none`).")
|
|
358
|
+
console.print("Use this only for local/testing environments.")
|
|
359
|
+
|
|
360
|
+
console.print("\n[bold green]Unsigned token saved to config file.[/]")
|
|
361
|
+
console.print(f"Principal: [cyan]{effective_principal_id}[/]")
|
|
362
|
+
console.print("\n[dim]Run 'nemo auth status' to inspect the token.[/]")
|
|
363
|
+
return
|
|
364
|
+
|
|
365
|
+
context = cli_context.get_sdk_context()
|
|
366
|
+
base_url = str(context.cluster.base_url).rstrip("/")
|
|
367
|
+
|
|
368
|
+
console.print(f"\nDiscovering auth configuration from {base_url}...")
|
|
369
|
+
|
|
370
|
+
try:
|
|
371
|
+
oidc_config = discover_nmp_config(base_url)
|
|
372
|
+
except httpx.HTTPError as exc:
|
|
373
|
+
raise AuthError(f"Failed to discover auth configuration: {exc}") from exc
|
|
374
|
+
|
|
375
|
+
if not oidc_config.auth_enabled:
|
|
376
|
+
console.print("[yellow]Authentication is not enabled on this cluster.[/]")
|
|
377
|
+
console.print("You can use the API without authentication.")
|
|
378
|
+
raise typer.Exit(0)
|
|
379
|
+
|
|
380
|
+
if not oidc_config.token_endpoint:
|
|
381
|
+
raise AuthError(
|
|
382
|
+
"This cluster does not have OIDC token endpoint configured.\n"
|
|
383
|
+
"Use OIDC configuration for device/password login, or for local testing use:\n"
|
|
384
|
+
"nemo auth login --unsigned-token --email <email>"
|
|
385
|
+
)
|
|
386
|
+
|
|
387
|
+
login_username = username or os.environ.get("NMP_OIDC_USERNAME")
|
|
388
|
+
login_password = password or os.environ.get("NMP_OIDC_PASSWORD")
|
|
389
|
+
use_password_grant = bool(login_username and login_password)
|
|
390
|
+
|
|
391
|
+
if use_password_grant:
|
|
392
|
+
if not oidc_config.client_id:
|
|
393
|
+
raise AuthError("OIDC client_id is required for password grant.")
|
|
394
|
+
else:
|
|
395
|
+
if not oidc_config.device_authorization_endpoint:
|
|
396
|
+
raise AuthError(
|
|
397
|
+
"This cluster does not support device flow authentication.\n"
|
|
398
|
+
"For non-interactive login use: nemo auth login --username <user> --password <pass>\n"
|
|
399
|
+
"Or set NMP_OIDC_USERNAME and NMP_OIDC_PASSWORD (e.g. in CI)."
|
|
400
|
+
)
|
|
401
|
+
|
|
402
|
+
console.print(f"[green]Found OIDC configuration[/] (issuer: {oidc_config.issuer})")
|
|
403
|
+
|
|
404
|
+
# Use only generic scopes from cluster defaults (exclude platform/custom scopes like platform:read)
|
|
405
|
+
# so platform scopes come only from --scope. Merge with --scope if provided.
|
|
406
|
+
raw_defaults = oidc_config.default_scopes
|
|
407
|
+
default_baseline = " ".join(s for s in raw_defaults.split() if ":" not in s)
|
|
408
|
+
if scope:
|
|
409
|
+
seen: set[str] = set()
|
|
410
|
+
parts: list[str] = []
|
|
411
|
+
for s in default_baseline.split():
|
|
412
|
+
if s not in seen:
|
|
413
|
+
seen.add(s)
|
|
414
|
+
parts.append(s)
|
|
415
|
+
for s in scope.split():
|
|
416
|
+
if s not in seen:
|
|
417
|
+
seen.add(s)
|
|
418
|
+
parts.append(s)
|
|
419
|
+
requested_scopes = " ".join(parts)
|
|
420
|
+
else:
|
|
421
|
+
requested_scopes = raw_defaults
|
|
422
|
+
|
|
423
|
+
scope_prefix = normalize_scope_prefix(oidc_config.scope_prefix)
|
|
424
|
+
effective_scope = build_effective_scope(requested_scopes, oidc_config.scope_prefix)
|
|
425
|
+
|
|
426
|
+
# Display the scopes being requested
|
|
427
|
+
console.print("\n[bold]Requesting scopes:[/]")
|
|
428
|
+
for s in requested_scopes.split():
|
|
429
|
+
if scope_prefix and (":" in s or s.endswith(".default")):
|
|
430
|
+
console.print(f" [cyan]{s}[/] [dim]({scope_prefix}{s})[/]")
|
|
431
|
+
else:
|
|
432
|
+
console.print(f" [cyan]{s}[/]")
|
|
433
|
+
console.print()
|
|
434
|
+
|
|
435
|
+
if use_password_grant:
|
|
436
|
+
if login_username is None or login_password is None:
|
|
437
|
+
raise AuthError("Username and password are required for password grant.")
|
|
438
|
+
client_id = cast(str, oidc_config.client_id)
|
|
439
|
+
try:
|
|
440
|
+
token_response = authenticate_with_password_grant(
|
|
441
|
+
token_endpoint=oidc_config.token_endpoint,
|
|
442
|
+
client_id=client_id,
|
|
443
|
+
username=login_username,
|
|
444
|
+
password=login_password,
|
|
445
|
+
scope=effective_scope,
|
|
446
|
+
)
|
|
447
|
+
except DeviceFlowError as exc:
|
|
448
|
+
raise AuthError(f"Authentication failed: {exc}") from exc
|
|
449
|
+
else:
|
|
450
|
+
if oidc_config.device_authorization_endpoint is None:
|
|
451
|
+
raise AuthError("This cluster does not support device flow authentication.")
|
|
452
|
+
client_id = cast(str, oidc_config.client_id)
|
|
453
|
+
device_authorization_endpoint = cast(str, oidc_config.device_authorization_endpoint)
|
|
454
|
+
try:
|
|
455
|
+
token_response = asyncio.run(
|
|
456
|
+
authenticate_with_device_flow(
|
|
457
|
+
device_authorization_endpoint=device_authorization_endpoint,
|
|
458
|
+
token_endpoint=oidc_config.token_endpoint,
|
|
459
|
+
client_id=client_id,
|
|
460
|
+
scope=effective_scope,
|
|
461
|
+
open_browser=not no_browser,
|
|
462
|
+
)
|
|
463
|
+
)
|
|
464
|
+
except DeviceFlowError as exc:
|
|
465
|
+
raise AuthError(f"Authentication failed: {exc}") from exc
|
|
466
|
+
|
|
467
|
+
token = token_response.token_for_nmp
|
|
468
|
+
|
|
469
|
+
claims = decode_jwt_claims(token)
|
|
470
|
+
user_email = claims.get("upn") or claims.get("email") or claims.get("preferred_username")
|
|
471
|
+
granted_scopes = claims.get("scp") or claims.get("scope") or ""
|
|
472
|
+
if isinstance(granted_scopes, str):
|
|
473
|
+
granted_scopes = granted_scopes.split()
|
|
474
|
+
|
|
475
|
+
validate_requested_scopes_granted(effective_scope, granted_scopes, scope_prefix)
|
|
476
|
+
|
|
477
|
+
config_params: ConfigParams = {"access_token": token}
|
|
478
|
+
if token_response.refresh_token:
|
|
479
|
+
config_params["refresh_token"] = token_response.refresh_token
|
|
480
|
+
if selected_context is not None:
|
|
481
|
+
config_params["current_context"] = context.context_name
|
|
482
|
+
Config.write(config_params, context_name=context.context_name)
|
|
483
|
+
|
|
484
|
+
console.print("\n[bold green]Authentication successful![/]")
|
|
485
|
+
if user_email:
|
|
486
|
+
console.print(f" Logged in as: [cyan]{user_email}[/]")
|
|
487
|
+
|
|
488
|
+
if granted_scopes:
|
|
489
|
+
# Normalize scopes by stripping prefix for display
|
|
490
|
+
display_scopes = []
|
|
491
|
+
for s in granted_scopes:
|
|
492
|
+
if scope_prefix and s.startswith(scope_prefix):
|
|
493
|
+
display_scopes.append(s[len(scope_prefix) :])
|
|
494
|
+
else:
|
|
495
|
+
display_scopes.append(s)
|
|
496
|
+
console.print(f" Granted scopes: [cyan]{' '.join(display_scopes)}[/]")
|
|
497
|
+
|
|
498
|
+
if token_response.refresh_token:
|
|
499
|
+
console.print(" Refresh token: [green]saved[/] (enables automatic token renewal)")
|
|
500
|
+
else:
|
|
501
|
+
console.print(" Refresh token: [yellow]not available[/] (add 'offline_access' scope to enable)")
|
|
502
|
+
|
|
503
|
+
console.print("\n[bold green]Credentials saved to config file.[/]")
|
|
504
|
+
console.print("\n[dim]Run 'nemo workspaces list' to verify your access.[/]")
|
|
505
|
+
|
|
506
|
+
|
|
507
|
+
@app.command("logout")
|
|
508
|
+
@handle_errors
|
|
509
|
+
def logout(ctx: typer.Context) -> None:
|
|
510
|
+
"""Remove stored credentials for the current context."""
|
|
511
|
+
from rich.console import Console
|
|
512
|
+
|
|
513
|
+
from nemo_platform.config.config import Config
|
|
514
|
+
|
|
515
|
+
cli_context: CLIContext = ctx.obj
|
|
516
|
+
context = cli_context.get_sdk_context()
|
|
517
|
+
|
|
518
|
+
console = Console()
|
|
519
|
+
|
|
520
|
+
base_url = str(context.cluster.base_url).rstrip("/")
|
|
521
|
+
if is_auth_disabled(base_url) is True:
|
|
522
|
+
console.print("[yellow]Authentication is disabled on this cluster — nothing to log out from.[/]")
|
|
523
|
+
return
|
|
524
|
+
|
|
525
|
+
logout_params = cast(
|
|
526
|
+
ConfigParams,
|
|
527
|
+
{"access_token": None, "refresh_token": None},
|
|
528
|
+
)
|
|
529
|
+
Config.write(logout_params, context_name=context.context_name)
|
|
530
|
+
console.print("[green]Logged out successfully.[/]")
|
|
531
|
+
|
|
532
|
+
|
|
533
|
+
@app.command("refresh")
|
|
534
|
+
@handle_errors
|
|
535
|
+
def refresh(ctx: typer.Context) -> None:
|
|
536
|
+
"""Refresh the current access token.
|
|
537
|
+
|
|
538
|
+
This command uses the saved refresh token to obtain a new access token without requiring you to re-authenticate through the browser.
|
|
539
|
+
"""
|
|
540
|
+
from rich.console import Console
|
|
541
|
+
|
|
542
|
+
from nemo_platform.config.config import Config
|
|
543
|
+
from nemo_platform.config.models import OAuthUser
|
|
544
|
+
|
|
545
|
+
cli_context: CLIContext = ctx.obj
|
|
546
|
+
context = cli_context.get_sdk_context()
|
|
547
|
+
|
|
548
|
+
console = Console()
|
|
549
|
+
|
|
550
|
+
# Only OAuthUser stores tokens in config
|
|
551
|
+
if not context.user or not isinstance(context.user, OAuthUser):
|
|
552
|
+
raise AuthError("No OAuth authentication configured. Run 'nemo auth login' first.")
|
|
553
|
+
|
|
554
|
+
token_value = context.user.token.get_secret_value()
|
|
555
|
+
|
|
556
|
+
if is_unsigned_jwt(token_value):
|
|
557
|
+
claims = decode_jwt_claims(token_value)
|
|
558
|
+
principal_id = claims.get("sub")
|
|
559
|
+
if not isinstance(principal_id, str) or not principal_id:
|
|
560
|
+
raise AuthError("Unsigned token is missing required 'sub' claim.")
|
|
561
|
+
|
|
562
|
+
email_claim = claims.get("email")
|
|
563
|
+
email = email_claim if isinstance(email_claim, str) else None
|
|
564
|
+
|
|
565
|
+
groups_claim = claims.get("groups")
|
|
566
|
+
if isinstance(groups_claim, str):
|
|
567
|
+
groups = [groups_claim]
|
|
568
|
+
elif isinstance(groups_claim, list) and all(isinstance(group, str) for group in groups_claim):
|
|
569
|
+
groups = groups_claim or None
|
|
570
|
+
else:
|
|
571
|
+
groups = None
|
|
572
|
+
|
|
573
|
+
scope_claim = claims.get("scope")
|
|
574
|
+
scopes = scope_claim.split() if isinstance(scope_claim, str) and scope_claim else None
|
|
575
|
+
|
|
576
|
+
expires_in_seconds = None
|
|
577
|
+
exp_claim = claims.get("exp")
|
|
578
|
+
iat_claim = claims.get("iat")
|
|
579
|
+
if isinstance(exp_claim, int):
|
|
580
|
+
if isinstance(iat_claim, int):
|
|
581
|
+
expires_in_seconds = max(exp_claim - iat_claim, 0)
|
|
582
|
+
else:
|
|
583
|
+
expires_in_seconds = max(exp_claim - int(time.time()), 0)
|
|
584
|
+
|
|
585
|
+
audience = claims.get("aud") if isinstance(claims.get("aud"), str) else None
|
|
586
|
+
issuer = claims.get("iss") if isinstance(claims.get("iss"), str) else None
|
|
587
|
+
|
|
588
|
+
excluded_claims = {
|
|
589
|
+
"sub",
|
|
590
|
+
"email",
|
|
591
|
+
"groups",
|
|
592
|
+
"scope",
|
|
593
|
+
"iat",
|
|
594
|
+
"exp",
|
|
595
|
+
"aud",
|
|
596
|
+
"iss",
|
|
597
|
+
}
|
|
598
|
+
extra_claims = {k: v for k, v in claims.items() if k not in excluded_claims}
|
|
599
|
+
if "aud" in claims and not isinstance(claims.get("aud"), str):
|
|
600
|
+
extra_claims["aud"] = claims.get("aud")
|
|
601
|
+
|
|
602
|
+
refreshed_token = generate_unsigned_jwt(
|
|
603
|
+
principal_id=principal_id,
|
|
604
|
+
email=email,
|
|
605
|
+
groups=groups,
|
|
606
|
+
scopes=scopes,
|
|
607
|
+
expires_in_seconds=expires_in_seconds,
|
|
608
|
+
audience=audience,
|
|
609
|
+
issuer=issuer,
|
|
610
|
+
extra_claims=extra_claims or None,
|
|
611
|
+
)
|
|
612
|
+
|
|
613
|
+
Config.write(
|
|
614
|
+
cast(ConfigParams, {"access_token": refreshed_token, "refresh_token": None}),
|
|
615
|
+
context_name=context.context_name,
|
|
616
|
+
)
|
|
617
|
+
|
|
618
|
+
refreshed_claims = decode_jwt_claims(refreshed_token)
|
|
619
|
+
exp = refreshed_claims.get("exp")
|
|
620
|
+
if exp:
|
|
621
|
+
from datetime import datetime, timezone
|
|
622
|
+
|
|
623
|
+
exp_dt = datetime.fromtimestamp(exp, tz=timezone.utc)
|
|
624
|
+
console.print(f"[green]Unsigned token refreshed successfully![/] Expires: {exp_dt.isoformat()}")
|
|
625
|
+
else:
|
|
626
|
+
console.print("[green]Unsigned token refreshed successfully![/]")
|
|
627
|
+
return
|
|
628
|
+
|
|
629
|
+
if not context.user.refresh_token:
|
|
630
|
+
raise AuthError("No refresh token available. Re-run 'nemo auth login' with 'offline_access' scope.")
|
|
631
|
+
|
|
632
|
+
# Fetch client_id from cluster discovery
|
|
633
|
+
base_url = str(context.cluster.base_url).rstrip("/")
|
|
634
|
+
try:
|
|
635
|
+
oidc_config = discover_nmp_config(base_url)
|
|
636
|
+
except httpx.HTTPError as e:
|
|
637
|
+
raise AuthError(f"Failed to discover auth configuration: {e}")
|
|
638
|
+
|
|
639
|
+
if not oidc_config.client_id or not oidc_config.token_endpoint:
|
|
640
|
+
raise AuthError("OIDC not configured on cluster.")
|
|
641
|
+
|
|
642
|
+
effective_scope = build_effective_scope(oidc_config.default_scopes, oidc_config.scope_prefix)
|
|
643
|
+
|
|
644
|
+
console.print("Refreshing access token...")
|
|
645
|
+
|
|
646
|
+
provider = OIDCTokenProvider(
|
|
647
|
+
token_endpoint=oidc_config.token_endpoint,
|
|
648
|
+
client_id=oidc_config.client_id,
|
|
649
|
+
tokens=TokenSet.from_access_token(
|
|
650
|
+
context.user.token.get_secret_value(),
|
|
651
|
+
context.user.refresh_token.get_secret_value(),
|
|
652
|
+
),
|
|
653
|
+
refresh_scope=effective_scope,
|
|
654
|
+
)
|
|
655
|
+
|
|
656
|
+
try:
|
|
657
|
+
provider.force_refresh()
|
|
658
|
+
except RuntimeError as e:
|
|
659
|
+
raise AuthError(f"Token refresh failed: {e}") from e
|
|
660
|
+
|
|
661
|
+
# Save new tokens (refresh token may be rotated)
|
|
662
|
+
config_params = {"access_token": provider.tokens.access_token}
|
|
663
|
+
if provider.tokens.refresh_token:
|
|
664
|
+
config_params["refresh_token"] = provider.tokens.refresh_token
|
|
665
|
+
Config.write(config_params, context_name=context.context_name) # type: ignore[arg-type]
|
|
666
|
+
|
|
667
|
+
# Show new token info
|
|
668
|
+
claims = decode_jwt_claims(provider.tokens.access_token)
|
|
669
|
+
exp = claims.get("exp")
|
|
670
|
+
if exp:
|
|
671
|
+
from datetime import datetime, timezone
|
|
672
|
+
|
|
673
|
+
exp_dt = datetime.fromtimestamp(exp, tz=timezone.utc)
|
|
674
|
+
console.print(f"[green]Token refreshed successfully![/] Expires: {exp_dt.isoformat()}")
|
|
675
|
+
else:
|
|
676
|
+
console.print("[green]Token refreshed successfully![/]")
|
|
677
|
+
|
|
678
|
+
|
|
679
|
+
@app.command("token")
|
|
680
|
+
@handle_errors
|
|
681
|
+
def token(ctx: typer.Context) -> None:
|
|
682
|
+
"""Print the current access token (for use with SDK or curl).
|
|
683
|
+
|
|
684
|
+
This outputs the raw token to stdout, suitable for piping or capture.
|
|
685
|
+
|
|
686
|
+
Examples:
|
|
687
|
+
# Print token
|
|
688
|
+
nemo auth token
|
|
689
|
+
# Capture in env var
|
|
690
|
+
export TOKEN=$(nemo auth token)
|
|
691
|
+
curl -H "Authorization: Bearer $(nemo auth token)" ...
|
|
692
|
+
"""
|
|
693
|
+
from nemo_platform.config.models import OAuthUser
|
|
694
|
+
|
|
695
|
+
cli_context: CLIContext = ctx.obj
|
|
696
|
+
context = cli_context.get_sdk_context()
|
|
697
|
+
|
|
698
|
+
if not context.user:
|
|
699
|
+
raise AuthError("No authentication configured. Run 'nemo auth login' first.")
|
|
700
|
+
|
|
701
|
+
if isinstance(context.user, OAuthUser):
|
|
702
|
+
typer.echo(context.user.token.get_secret_value())
|
|
703
|
+
else:
|
|
704
|
+
raise AuthError("No token available for current user type.")
|
|
705
|
+
|
|
706
|
+
|
|
707
|
+
@app.command("status")
|
|
708
|
+
@handle_errors
|
|
709
|
+
def status(ctx: typer.Context) -> None:
|
|
710
|
+
"""Show current authentication status."""
|
|
711
|
+
from datetime import datetime, timezone
|
|
712
|
+
|
|
713
|
+
from rich.console import Console
|
|
714
|
+
from rich.table import Table
|
|
715
|
+
|
|
716
|
+
from nemo_platform.config.models import OAuthUser
|
|
717
|
+
|
|
718
|
+
cli_context: CLIContext = ctx.obj
|
|
719
|
+
context = cli_context.get_sdk_context()
|
|
720
|
+
|
|
721
|
+
console = Console()
|
|
722
|
+
|
|
723
|
+
# Check whether the cluster has auth enabled before showing token details.
|
|
724
|
+
base_url = str(context.cluster.base_url).rstrip("/")
|
|
725
|
+
if is_auth_disabled(base_url) is True:
|
|
726
|
+
console.print()
|
|
727
|
+
console.print(f"[cyan]Cluster:[/] {base_url}")
|
|
728
|
+
console.print(f"[cyan]Context:[/] {context.context_name}")
|
|
729
|
+
console.print()
|
|
730
|
+
console.print("[green]Authentication is disabled on this cluster.[/]")
|
|
731
|
+
console.print("All API requests are accepted without credentials.")
|
|
732
|
+
return
|
|
733
|
+
|
|
734
|
+
table = Table(title="Authentication Status", show_header=False)
|
|
735
|
+
table.add_column("Property", style="cyan")
|
|
736
|
+
table.add_column("Value")
|
|
737
|
+
|
|
738
|
+
table.add_row("Cluster", str(context.cluster.base_url))
|
|
739
|
+
table.add_row("Context", context.context_name)
|
|
740
|
+
|
|
741
|
+
if context.user:
|
|
742
|
+
table.add_row("Auth Type", context.user.type)
|
|
743
|
+
|
|
744
|
+
if isinstance(context.user, OAuthUser):
|
|
745
|
+
# OAuth token authentication
|
|
746
|
+
token_value = context.user.token.get_secret_value()
|
|
747
|
+
if is_unsigned_jwt(token_value):
|
|
748
|
+
table.add_row("Warning", "[yellow]Unsigned JWT (alg=none). Use only for local/testing.[/]")
|
|
749
|
+
|
|
750
|
+
claims = decode_jwt_claims(token_value)
|
|
751
|
+
|
|
752
|
+
if claims:
|
|
753
|
+
# Show decoded JWT info
|
|
754
|
+
email = claims.get("upn") or claims.get("email") or claims.get("preferred_username")
|
|
755
|
+
if email:
|
|
756
|
+
table.add_row("Email", email)
|
|
757
|
+
|
|
758
|
+
subject = claims.get("oid") or claims.get("sub")
|
|
759
|
+
if subject:
|
|
760
|
+
table.add_row("User ID", subject)
|
|
761
|
+
|
|
762
|
+
scopes = claims.get("scp") or claims.get("scope") or ""
|
|
763
|
+
if isinstance(scopes, str):
|
|
764
|
+
scopes = scopes.split()
|
|
765
|
+
if scopes:
|
|
766
|
+
table.add_row("Scopes", " ".join(scopes))
|
|
767
|
+
else:
|
|
768
|
+
table.add_row("Scopes", "[dim]none[/]")
|
|
769
|
+
|
|
770
|
+
groups = claims.get("groups") or claims.get("cognito:groups") or []
|
|
771
|
+
if isinstance(groups, str):
|
|
772
|
+
groups = [groups]
|
|
773
|
+
if groups:
|
|
774
|
+
table.add_row("Groups", ", ".join(groups))
|
|
775
|
+
|
|
776
|
+
exp = claims.get("exp")
|
|
777
|
+
if exp:
|
|
778
|
+
exp_dt = datetime.fromtimestamp(exp, tz=timezone.utc)
|
|
779
|
+
now = datetime.now(tz=timezone.utc)
|
|
780
|
+
if exp_dt > now:
|
|
781
|
+
remaining = exp_dt - now
|
|
782
|
+
hours, remainder = divmod(int(remaining.total_seconds()), 3600)
|
|
783
|
+
minutes, _ = divmod(remainder, 60)
|
|
784
|
+
table.add_row("Expires", f"{exp_dt.isoformat()} ({hours}h {minutes}m remaining)")
|
|
785
|
+
else:
|
|
786
|
+
table.add_row("Expires", f"[red]EXPIRED[/] ({exp_dt.isoformat()})")
|
|
787
|
+
|
|
788
|
+
# Show refresh token status
|
|
789
|
+
if context.user.refresh_token:
|
|
790
|
+
table.add_row("Refresh Token", "[green]available[/] (run 'nemo auth refresh' to renew)")
|
|
791
|
+
else:
|
|
792
|
+
table.add_row("Refresh Token", "[yellow]not available[/]")
|
|
793
|
+
|
|
794
|
+
# Show redacted token
|
|
795
|
+
redacted = f"{token_value[:20]}...{token_value[-10:]}" if len(token_value) > 30 else "***"
|
|
796
|
+
table.add_row("Token", redacted)
|
|
797
|
+
|
|
798
|
+
else:
|
|
799
|
+
table.add_row("User", "None")
|
|
800
|
+
table.add_row("Auth Type", "no-auth")
|
|
801
|
+
|
|
802
|
+
console.print()
|
|
803
|
+
console.print(table)
|