feat(client-braket): Adds ExperimentalCapabilities field to CreateQuantumTask request and GetQuantumTask response objects. Enables use of experimental software capabilities when creating quantum tasks.

awstools · awstools · commit 6c179ae76ce6 · 2025-11-10T19:18:38.000Z
diff --git a/clients/client-braket/src/commands/CreateQuantumTaskCommand.ts b/clients/client-braket/src/commands/CreateQuantumTaskCommand.ts
@@ -55,6 +55,9 @@ export interface CreateQuantumTaskCommandOutput extends CreateQuantumTaskRespons
  *       type: "STRING_VALUE", // required
  *     },
  *   ],
+ *   experimentalCapabilities: { // ExperimentalCapabilities Union: only one key present
+ *     enabled: "STRING_VALUE",
+ *   },
  * };
  * const command = new CreateQuantumTaskCommand(input);
  * const response = await client.send(command);
diff --git a/clients/client-braket/src/commands/GetQuantumTaskCommand.ts b/clients/client-braket/src/commands/GetQuantumTaskCommand.ts
@@ -78,6 +78,9 @@ export interface GetQuantumTaskCommandOutput extends GetQuantumTaskResponse, __M
  * //     programCount: Number("long"),
  * //     executableCount: Number("long"),
  * //   },
+ * //   experimentalCapabilities: { // ExperimentalCapabilities Union: only one key present
+ * //     enabled: "STRING_VALUE",
+ * //   },
  * // };
  *
  * ```
diff --git a/clients/client-braket/src/models/models_0.ts b/clients/client-braket/src/models/models_0.ts
@@ -38,13 +38,13 @@ export interface ActionMetadata {
   actionType: string | undefined;
 
   /**
-   * <p>The number of programs in a program set. This is only available for a Program Set.</p>
+   * <p>The number of programs in a program set. This is only available for a program set.</p>
    * @public
    */
   programCount?: number | undefined;
 
   /**
-   * <p>The number of executables in a program set. This is only available for a Program Set.</p>
+   * <p>The number of executables in a program set. This is only available for a program set.</p>
    * @public
    */
   executableCount?: number | undefined;
@@ -1356,6 +1356,59 @@ export interface CancelQuantumTaskResponse {
   cancellationStatus: CancellationStatus | undefined;
 }
 
+/**
+ * @public
+ * @enum
+ */
+export const ExperimentalCapabilitiesEnablementType = {
+  ALL: "ALL",
+  NONE: "NONE",
+} as const;
+
+/**
+ * @public
+ */
+export type ExperimentalCapabilitiesEnablementType =
+  (typeof ExperimentalCapabilitiesEnablementType)[keyof typeof ExperimentalCapabilitiesEnablementType];
+
+/**
+ * <p>Enabled experimental capabilities for quantum hardware. Note that the use of these features may impact device capabilities and performance beyond its standard specifications.</p>
+ * @public
+ */
+export type ExperimentalCapabilities = ExperimentalCapabilities.EnabledMember | ExperimentalCapabilities.$UnknownMember;
+
+/**
+ * @public
+ */
+export namespace ExperimentalCapabilities {
+  /**
+   * <p>Enabled experimental capabilities.</p>
+   * @public
+   */
+  export interface EnabledMember {
+    enabled: ExperimentalCapabilitiesEnablementType;
+    $unknown?: never;
+  }
+
+  /**
+   * @public
+   */
+  export interface $UnknownMember {
+    enabled?: never;
+    $unknown: [string, any];
+  }
+
+  export interface Visitor<T> {
+    enabled: (value: ExperimentalCapabilitiesEnablementType) => T;
+    _: (name: string, value: any) => T;
+  }
+
+  export const visit = <T>(value: ExperimentalCapabilities, visitor: Visitor<T>): T => {
+    if (value.enabled !== undefined) return visitor.enabled(value.enabled);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  };
+}
+
 /**
  * @public
  */
@@ -1419,6 +1472,12 @@ export interface CreateQuantumTaskRequest {
    * @public
    */
   associations?: Association[] | undefined;
+
+  /**
+   * <p>Enable experimental capabilities for the quantum task.</p>
+   * @public
+   */
+  experimentalCapabilities?: ExperimentalCapabilities | undefined;
 }
 
 /**
@@ -1611,6 +1670,12 @@ export interface GetQuantumTaskResponse {
    * @public
    */
   actionMetadata?: ActionMetadata | undefined;
+
+  /**
+   * <p>Enabled experimental capabilities for the quantum task, if any.</p>
+   * @public
+   */
+  experimentalCapabilities?: ExperimentalCapabilities | undefined;
 }
 
 /**
diff --git a/clients/client-braket/src/protocols/Aws_restJson1.ts b/clients/client-braket/src/protocols/Aws_restJson1.ts
@@ -1,5 +1,10 @@
 // smithy-typescript generated code
-import { loadRestJsonErrorCode, parseJsonBody as parseBody, parseJsonErrorBody as parseErrorBody } from "@aws-sdk/core";
+import {
+  awsExpectUnion as __expectUnion,
+  loadRestJsonErrorCode,
+  parseJsonBody as parseBody,
+  parseJsonErrorBody as parseErrorBody,
+} from "@aws-sdk/core";
 import { requestBuilder as rb } from "@smithy/core";
 import { HttpRequest as __HttpRequest, HttpResponse as __HttpResponse } from "@smithy/protocol-http";
 import {
@@ -53,6 +58,7 @@ import {
   DeviceConfig,
   DeviceOfflineException,
   DeviceRetiredException,
+  ExperimentalCapabilities,
   InputFileConfig,
   InstanceConfig,
   InternalServiceException,
@@ -166,6 +172,7 @@ export const se_CreateQuantumTaskCommand = async (
       clientToken: [true, (_) => _ ?? generateIdempotencyToken()],
       deviceArn: [],
       deviceParameters: (_) => __LazyJsonString.from(_),
+      experimentalCapabilities: (_) => _json(_),
       jobToken: [],
       outputS3Bucket: [],
       outputS3KeyPrefix: [],
@@ -533,6 +540,7 @@ export const de_GetQuantumTaskCommand = async (
     deviceArn: __expectString,
     deviceParameters: __LazyJsonString.from,
     endedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+    experimentalCapabilities: (_) => _json(__expectUnion(_)),
     failureReason: __expectString,
     jobArn: __expectString,
     numSuccessfulShots: __expectLong,
@@ -902,6 +910,8 @@ const de_ValidationExceptionRes = async (parsedOutput: any, context: __SerdeCont
 
 // se_DeviceConfig omitted.
 
+// se_ExperimentalCapabilities omitted.
+
 // se_HyperParameters omitted.
 
 // se_InputConfigList omitted.
@@ -958,6 +968,8 @@ const de_ValidationExceptionRes = async (parsedOutput: any, context: __SerdeCont
 
 // de_DeviceSummaryList omitted.
 
+// de_ExperimentalCapabilities omitted.
+
 // de_HybridJobQueueInfo omitted.
 
 // de_HyperParameters omitted.
diff --git a/codegen/sdk-codegen/aws-models/braket.json b/codegen/sdk-codegen/aws-models/braket.json
@@ -27,13 +27,13 @@
         "programCount": {
           "target": "smithy.api#Long",
           "traits": {
-            "smithy.api#documentation": "<p>The number of programs in a program set. This is only available for a Program Set.</p>"
+            "smithy.api#documentation": "<p>The number of programs in a program set. This is only available for a program set.</p>"
           }
         },
         "executableCount": {
           "target": "smithy.api#Long",
           "traits": {
-            "smithy.api#documentation": "<p>The number of executables in a program set. This is only available for a Program Set.</p>"
+            "smithy.api#documentation": "<p>The number of executables in a program set. This is only available for a program set.</p>"
           }
         }
       },
@@ -1277,6 +1277,12 @@
               "max": 1
             }
           }
+        },
+        "experimentalCapabilities": {
+          "target": "com.amazonaws.braket#ExperimentalCapabilities",
+          "traits": {
+            "smithy.api#documentation": "<p>Enable experimental capabilities for the quantum task.</p>"
+          }
         }
       }
     },
@@ -1498,6 +1504,35 @@
         ]
       }
     },
+    "com.amazonaws.braket#ExperimentalCapabilities": {
+      "type": "union",
+      "members": {
+        "enabled": {
+          "target": "com.amazonaws.braket#ExperimentalCapabilitiesEnablementType",
+          "traits": {
+            "smithy.api#documentation": "<p>Enabled experimental capabilities.</p>"
+          }
+        }
+      },
+      "traits": {
+        "smithy.api#documentation": "<p>Enabled experimental capabilities for quantum hardware. Note that the use of these features may impact device capabilities and performance beyond its standard specifications.</p>"
+      }
+    },
+    "com.amazonaws.braket#ExperimentalCapabilitiesEnablementType": {
+      "type": "string",
+      "traits": {
+        "smithy.api#enum": [
+          {
+            "value": "ALL",
+            "name": "ALL"
+          },
+          {
+            "value": "NONE",
+            "name": "NONE"
+          }
+        ]
+      }
+    },
     "com.amazonaws.braket#GetDevice": {
       "type": "operation",
       "input": {
@@ -1961,6 +1996,12 @@
           "traits": {
             "smithy.api#documentation": "<p>Metadata about the action performed by the quantum task, including information about the type of action and program counts.</p>"
           }
+        },
+        "experimentalCapabilities": {
+          "target": "com.amazonaws.braket#ExperimentalCapabilities",
+          "traits": {
+            "smithy.api#documentation": "<p>Enabled experimental capabilities for the quantum task, if any.</p>"
+          }
         }
       }
     },
