regen rest api bindings

kurtosis-tech · Feb 14, 2024 · 0738875 · 0738875
1 parent c43037c
commit 0738875
Show file tree

Hide file tree

Showing 8 changed files with 99 additions and 78 deletions.
diff --git a/api/golang/core/lib/binding_constructors/binding_constructors.go b/api/golang/core/lib/binding_constructors/binding_constructors.go
@@ -116,6 +116,7 @@ func NewRunStarlarkRemotePackageArgs(
 	cloudInstanceId string,
 	cloudUserId string,
 	imageDownloadMode kurtosis_core_rpc_api_bindings.ImageDownloadMode,
+	nonBlockingMode bool,
 ) *kurtosis_core_rpc_api_bindings.RunStarlarkPackageArgs {
 	parallelismCopy := new(int32)
 	*parallelismCopy = parallelism
@@ -139,6 +140,7 @@ func NewRunStarlarkRemotePackageArgs(
 		CloudInstanceId:        cloudInstanceIdCopy,
 		CloudUserId:            cloudUserIdCopy,
 		ImageDownloadMode:      imageDownloadModeCopy,
+		NonBlockingMode:        &nonBlockingMode,
 	}
 }
 

diff --git a/api/golang/core/lib/enclaves/enclave_context.go b/api/golang/core/lib/enclaves/enclave_context.go
@@ -308,7 +308,7 @@ func (enclaveCtx *EnclaveContext) RunStarlarkRemotePackage(
 	}()
 
 	starlarkResponseLineChan := make(chan *kurtosis_core_rpc_api_bindings.StarlarkRunResponseLine)
-	executeStartosisScriptArgs := binding_constructors.NewRunStarlarkRemotePackageArgs(packageId, runConfig.RelativePathToMainFile, runConfig.MainFunctionName, serializedParams, runConfig.DryRun, runConfig.Parallelism, runConfig.ExperimentalFeatureFlags, runConfig.CloudInstanceId, runConfig.CloudUserId, runConfig.ImageDownload)
+	executeStartosisScriptArgs := binding_constructors.NewRunStarlarkRemotePackageArgs(packageId, runConfig.RelativePathToMainFile, runConfig.MainFunctionName, serializedParams, runConfig.DryRun, runConfig.Parallelism, runConfig.ExperimentalFeatureFlags, runConfig.CloudInstanceId, runConfig.CloudUserId, runConfig.ImageDownload, runConfig.NonBlockingMode)
 
 	stream, err := enclaveCtx.client.RunStarlarkPackage(ctxWithCancel, executeStartosisScriptArgs)
 	if err != nil {

diff --git a/api/golang/core/lib/starlark_run_config/starlark_run_config.go b/api/golang/core/lib/starlark_run_config/starlark_run_config.go
@@ -42,6 +42,7 @@ func NewRunStarlarkConfig(opts ...starlarkRunConfigOption) *StarlarkRunConfig {
 		CloudInstanceId:          defaultCloudInstanceId,
 		CloudUserId:              defaultCloudUserId,
 		ImageDownload:            defaultImageDownload,
+		NonBlockingMode:          defaultNonBlockingMode,
 	}
 
 	for _, opt := range opts {

diff --git a/api/golang/http_rest/api_types/api_types.gen.go b/api/golang/http_rest/api_types/api_types.gen.go
diff --git a/api/golang/http_rest/server/core_rest_api/api_container_server.gen.go b/api/golang/http_rest/server/core_rest_api/api_container_server.gen.go
diff --git a/api/openapi/specs/kurtosis_api.yaml b/api/openapi/specs/kurtosis_api.yaml
@@ -1125,6 +1125,9 @@ components:
           description: Defaults to empty
         image_download_mode:
           $ref: "#/components/schemas/ImageDownloadMode"
+        non_blocking_mode:
+          type: boolean
+          description: Defaults to false
       required:
         - serialized_script
 
@@ -1168,6 +1171,9 @@ components:
           description: Defaults to empty
         image_download_mode:
           $ref: "#/components/schemas/ImageDownloadMode"
+        non_blocking_mode:
+          type: boolean
+          description: Defaults to false
 
     KurtosisFeatureFlag:
       type: string

diff --git a/api/typescript/src/engine/rest_api_bindings/types.d.ts b/api/typescript/src/engine/rest_api_bindings/types.d.ts
@@ -1326,6 +1326,8 @@ export interface components {
             /** @description Defaults to empty */
             cloud_user_id?: string;
             image_download_mode?: components["schemas"]["ImageDownloadMode"];
+            /** @description Defaults to false */
+            non_blocking_mode?: boolean;
         };
         RunStarlarkPackage: {
             /** @description Parameters data for the Starlark package main function */
@@ -1354,6 +1356,8 @@ export interface components {
             /** @description Defaults to empty */
             cloud_user_id?: string;
             image_download_mode?: components["schemas"]["ImageDownloadMode"];
+            /** @description Defaults to false */
+            non_blocking_mode?: boolean;
         };
         /**
          * @description 0 - NO_INSTRUCTIONS_CACHING

diff --git a/engine/server/engine/server/enclave_rest_api_handler.go b/engine/server/engine/server/enclave_rest_api_handler.go
@@ -601,6 +601,7 @@ func (manager *enclaveRuntime) PostEnclavesEnclaveIdentifierStarlarkPackagesPack
 		// RunStarlarkPackage. If the package is remote and must be cloned within the APIC, use the standalone boolean flag
 		// clone_package below
 		StarlarkPackageContent: nil,
+		NonBlockingMode:        request.Body.NonBlockingMode,
 	}
 
 	ctxWithCancel, cancelCtxFunc := context.WithCancel(context.Background())
@@ -673,6 +674,7 @@ func (manager *enclaveRuntime) PostEnclavesEnclaveIdentifierStarlarkScripts(ctx
 		CloudInstanceId:      request.Body.CloudInstanceId,
 		CloudUserId:          request.Body.CloudUserId,
 		ImageDownloadMode:    utils.MapPointer(request.Body.ImageDownloadMode, to_grpc.ToGrpcImageDownloadMode),
+		NonBlockingMode:      request.Body.NonBlockingMode,
 	}
 
 	ctxWithCancel, cancelCtxFunc := context.WithCancel(context.Background())