diff --git a/client/src/features/sessionsV2/api/computeResources.generated-api.ts b/client/src/features/sessionsV2/api/computeResources.generated-api.ts index 9e5376adbe..98c48a1239 100644 --- a/client/src/features/sessionsV2/api/computeResources.generated-api.ts +++ b/client/src/features/sessionsV2/api/computeResources.generated-api.ts @@ -655,7 +655,16 @@ export type RemoteConfigurationFirecrest = { system_name: RemoteConfigurationFirecrestSystemName; partition?: RemoteConfigurationFirecrestPartition; }; -export type RemoteConfiguration = RemoteConfigurationFirecrest; +export type RemoteConfigurationRunaiApiUrl = string; +export type RemoteConfigurationRunai = { + /** Kind of remote resource pool */ + kind: "runai"; + base_url: RemoteConfigurationRunaiApiUrl; + provider_id?: RemoteConfigurationFirecrestProviderId; +}; +export type RemoteConfiguration = + | RemoteConfigurationFirecrest + | RemoteConfigurationRunai; export type IdleThreshold = number; export type HibernationThreshold = number; export type HibernationWarningPeriod = number; @@ -768,9 +777,16 @@ export type RemoteConfigurationFirecrestPatch = { system_name?: RemoteConfigurationFirecrestSystemName; partition?: RemoteConfigurationFirecrestPartition; }; +export type RemoteConfigurationRunaiPatch = { + /** Kind of remote resource pool */ + kind?: "runai"; + base_url?: RemoteConfigurationRunaiApiUrl; + provider_id?: RemoteConfigurationFirecrestProviderId; +}; export type RemoteConfigurationPatch = | RemoteConfigurationPatchReset - | RemoteConfigurationFirecrestPatch; + | RemoteConfigurationFirecrestPatch + | RemoteConfigurationRunaiPatch; export type ResourcePoolPatch = { quota?: QuotaPatch; classes?: ResourceClassesPatchWithId; diff --git a/client/src/features/sessionsV2/api/computeResources.openapi.json b/client/src/features/sessionsV2/api/computeResources.openapi.json index 745aa7c543..4363155b77 100644 --- a/client/src/features/sessionsV2/api/computeResources.openapi.json +++ b/client/src/features/sessionsV2/api/computeResources.openapi.json @@ -2622,6 +2622,9 @@ "oneOf": [ { "$ref": "#/components/schemas/RemoteConfigurationFirecrest" + }, + { + "$ref": "#/components/schemas/RemoteConfigurationRunai" } ] }, @@ -2671,6 +2674,31 @@ "description": "The partition to use when submitting jobs", "example": "normal" }, + "RemoteConfigurationRunai": { + "type": "object", + "description": "The configuration for starting sessions remotely using Runai\n", + "additionalProperties": false, + "properties": { + "kind": { + "type": "string", + "enum": ["runai"], + "description": "Kind of remote resource pool", + "example": "runai" + }, + "base_url": { + "$ref": "#/components/schemas/RemoteConfigurationRunaiApiUrl" + }, + "provider_id": { + "$ref": "#/components/schemas/RemoteConfigurationFirecrestProviderId" + } + }, + "required": ["kind", "base_url"] + }, + "RemoteConfigurationRunaiApiUrl": { + "type": "string", + "description": "The base URL of the Runai server", + "example": "https://sdsc.run.ai" + }, "RemoteConfigurationPatch": { "type": "object", "description": "Patch for the configuration used by to start sessions remotely\n", @@ -2680,6 +2708,9 @@ }, { "$ref": "#/components/schemas/RemoteConfigurationFirecrestPatch" + }, + { + "$ref": "#/components/schemas/RemoteConfigurationRunaiPatch" } ] }, @@ -2713,6 +2744,25 @@ } } }, + "RemoteConfigurationRunaiPatch": { + "type": "object", + "description": "The configuration for starting sessions remotely using Runai\n", + "additionalProperties": false, + "properties": { + "kind": { + "type": "string", + "enum": ["runai"], + "description": "Kind of remote resource pool", + "example": "runai" + }, + "base_url": { + "$ref": "#/components/schemas/RemoteConfigurationRunaiApiUrl" + }, + "provider_id": { + "$ref": "#/components/schemas/RemoteConfigurationFirecrestProviderId" + } + } + }, "IdleThreshold": { "type": "integer", "description": "A threshold in seconds after which a session gets hibernated (0 means no threshold)",