feat: Add support for wait for operation complete for sandbox sub-modules

PiperOrigin-RevId: 908889456

Author: vertex-sdk-bot

src/converters/_sandboxes_converters.ts

@@ -30,6 +30,33 @@ export function createAgentEngineSandboxConfigToVertex(
     common.setValueByPath(parentObject, ['ttl'], fromTtl);
   }
 
+  const fromSandboxEnvironmentTemplate = common.getValueByPath(fromObject, [
+    'sandboxEnvironmentTemplate',
+  ]);
+  if (parentObject !== undefined && fromSandboxEnvironmentTemplate != null) {
+    common.setValueByPath(
+      parentObject,
+      ['sandboxEnvironmentTemplate'],
+      fromSandboxEnvironmentTemplate,
+    );
+  }
+
+  const fromSandboxEnvironmentSnapshot = common.getValueByPath(fromObject, [
+    'sandboxEnvironmentSnapshot',
+  ]);
+  if (parentObject !== undefined && fromSandboxEnvironmentSnapshot != null) {
+    common.setValueByPath(
+      parentObject,
+      ['sandboxEnvironmentSnapshot'],
+      fromSandboxEnvironmentSnapshot,
+    );
+  }
+
+  const fromOwner = common.getValueByPath(fromObject, ['owner']);
+  if (parentObject !== undefined && fromOwner != null) {
+    common.setValueByPath(parentObject, ['owner'], fromOwner);
+  }
+
   return toObject;
 }
 

src/types/common.ts

@@ -1916,6 +1916,14 @@ export declare interface CreateAgentEngineSandboxConfig {
   waitForCompletion?: boolean;
   /** The TTL for this resource. The expiration time is computed: now + TTL. */
   ttl?: string;
+  /** The name of the sandbox environment template to create the sandbox from. The sandbox environment template should be in the format:
+      projects/{project}/locations/{location}/agentEngines/{agent_engine}/sandboxEnvironmentTemplates/{sandbox_environment_template} */
+  sandboxEnvironmentTemplate?: string;
+  /** The name of the sandbox environment snapshot to restore the sandbox from. The sandbox environment snapshot should be in the format:
+      projects/{project}/locations/{location}/agentEngines/{agent_engine}/sandboxEnvironmentSnapshots/{sandbox_environment_snapshot} */
+  sandboxEnvironmentSnapshot?: string;
+  /** Owner information for this sandbox environment. A sandbox can only be restored from a snapshot belonging to the same owner. */
+  owner?: string;
 }
 
 /** Parameters for creating Agent Engine Sandboxes. */