@aws-sdk/client-sagemaker-featurestore-runtime 3.1075.0 → 3.1077.0

package/dist-types/commands/ListRecordsCommand.d.ts

@@ -0,0 +1,120 @@
+import { Command as $Command } from "@smithy/core/client";
+import type { MetadataBearer as __MetadataBearer } from "@smithy/types";
+import type { ListRecordsRequest, ListRecordsResponse } from "../models/models_0";
+import type { SageMakerFeatureStoreRuntimeClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerFeatureStoreRuntimeClient";
+/**
+ * @public
+ */
+export type { __MetadataBearer };
+export { $Command };
+/**
+ * @public
+ *
+ * The input for {@link ListRecordsCommand}.
+ */
+export interface ListRecordsCommandInput extends ListRecordsRequest {
+}
+/**
+ * @public
+ *
+ * The output of {@link ListRecordsCommand}.
+ */
+export interface ListRecordsCommandOutput extends ListRecordsResponse, __MetadataBearer {
+}
+declare const ListRecordsCommand_base: {
+    new (input: ListRecordsCommandInput): import("@smithy/core/client").CommandImpl<ListRecordsCommandInput, ListRecordsCommandOutput, SageMakerFeatureStoreRuntimeClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes>;
+    new (input: ListRecordsCommandInput): import("@smithy/core/client").CommandImpl<ListRecordsCommandInput, ListRecordsCommandOutput, SageMakerFeatureStoreRuntimeClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes>;
+    getEndpointParameterInstructions(): {
+        [x: string]: unknown;
+    };
+};
+/**
+ * <p>Lists the <code>RecordIdentifier</code> values of all records stored in a
+ *             <code>FeatureGroup</code>'s <code>OnlineStore</code>. This enables you to discover which
+ *          records exist without retrieving the full record data.</p>
+ * @example
+ * Use a bare-bones client and the command you need to make an API call.
+ * ```javascript
+ * import { SageMakerFeatureStoreRuntimeClient, ListRecordsCommand } from "@aws-sdk/client-sagemaker-featurestore-runtime"; // ES Modules import
+ * // const { SageMakerFeatureStoreRuntimeClient, ListRecordsCommand } = require("@aws-sdk/client-sagemaker-featurestore-runtime"); // CommonJS import
+ * // import type { SageMakerFeatureStoreRuntimeClientConfig } from "@aws-sdk/client-sagemaker-featurestore-runtime";
+ * const config = {}; // type is SageMakerFeatureStoreRuntimeClientConfig
+ * const client = new SageMakerFeatureStoreRuntimeClient(config);
+ * const input = { // ListRecordsRequest
+ *   FeatureGroupName: "STRING_VALUE", // required
+ *   MaxResults: Number("int"),
+ *   NextToken: "STRING_VALUE",
+ *   IncludeSoftDeletedRecords: true || false,
+ * };
+ * const command = new ListRecordsCommand(input);
+ * const response = await client.send(command);
+ * // { // ListRecordsResponse
+ * //   RecordIdentifiers: [ // RecordIdentifierList // required
+ * //     "STRING_VALUE",
+ * //   ],
+ * //   NextToken: "STRING_VALUE",
+ * // };
+ *
+ * ```
+ *
+ * @param ListRecordsCommandInput - {@link ListRecordsCommandInput}
+ * @returns {@link ListRecordsCommandOutput}
+ * @see {@link ListRecordsCommandInput} for command's `input` shape.
+ * @see {@link ListRecordsCommandOutput} for command's `response` shape.
+ * @see {@link SageMakerFeatureStoreRuntimeClientResolvedConfig | config} for SageMakerFeatureStoreRuntimeClient's `config` shape.
+ *
+ * @throws {@link AccessForbidden} (client fault)
+ *  <p>You do not have permission to perform an action.</p>
+ *
+ * @throws {@link InternalFailure} (server fault)
+ *  <p>An internal failure occurred. Try your request again. If the problem persists, contact
+ *             Amazon Web Services customer support.</p>
+ *
+ * @throws {@link ResourceNotFound} (client fault)
+ *  <p>A resource that is required to perform an action was not found.</p>
+ *
+ * @throws {@link ServiceUnavailable} (server fault)
+ *  <p>The service is currently unavailable.</p>
+ *
+ * @throws {@link ValidationError} (client fault)
+ *  <p>There was an error validating your request.</p>
+ *
+ * @throws {@link SageMakerFeatureStoreRuntimeServiceException}
+ * <p>Base exception class for all service exceptions from SageMakerFeatureStoreRuntime service.</p>
+ *
+ *
+ * @example List record identifiers from a feature group
+ * ```javascript
+ * //
+ * const input = {
+ *   FeatureGroupName: "my-feature-group",
+ *   MaxResults: 10
+ * };
+ * const command = new ListRecordsCommand(input);
+ * const response = await client.send(command);
+ * /* response is
+ * {
+ *   NextToken: "eyJsYXN0RXZhbHVhdGVkS2V5IjoiYWJjMTIzIn0=",
+ *   RecordIdentifiers: [
+ *     "record-id-1",
+ *     "record-id-2"
+ *   ]
+ * }
+ * *\/
+ * ```
+ *
+ * @public
+ */
+export declare class ListRecordsCommand extends ListRecordsCommand_base {
+    /** @internal type navigation helper, not in runtime. */
+    protected static __types: {
+        api: {
+            input: ListRecordsRequest;
+            output: ListRecordsResponse;
+        };
+        sdk: {
+            input: ListRecordsCommandInput;
+            output: ListRecordsCommandOutput;
+        };
+    };
+}

package/dist-types/commands/index.d.ts

@@ -1,4 +1,6 @@
 export * from "./BatchGetRecordCommand";
+export * from "./BatchWriteRecordCommand";
 export * from "./DeleteRecordCommand";
 export * from "./GetRecordCommand";
+export * from "./ListRecordsCommand";
 export * from "./PutRecordCommand";

package/dist-types/index.d.ts

@@ -36,6 +36,7 @@ export type { RuntimeExtension } from "./runtimeExtensions";
 export type { SageMakerFeatureStoreRuntimeExtensionConfiguration } from "./extensionConfiguration";
 export * from "./commands";
 export * from "./schemas/schemas_0";
+export * from "./pagination";
 export * from "./models/enums";
 export * from "./models/errors";
 export * from "./models/models_0";

package/dist-types/models/enums.d.ts

@@ -10,18 +10,6 @@ export declare const ExpirationTimeResponse: {
  * @public
  */
 export type ExpirationTimeResponse = (typeof ExpirationTimeResponse)[keyof typeof ExpirationTimeResponse];
-/**
- * @public
- * @enum
- */
-export declare const DeletionMode: {
-    readonly HARD_DELETE: "HardDelete";
-    readonly SOFT_DELETE: "SoftDelete";
-};
-/**
- * @public
- */
-export type DeletionMode = (typeof DeletionMode)[keyof typeof DeletionMode];
 /**
  * @public
  * @enum
@@ -49,3 +37,15 @@ export declare const TtlDurationUnit: {
  * @public
  */
 export type TtlDurationUnit = (typeof TtlDurationUnit)[keyof typeof TtlDurationUnit];
+/**
+ * @public
+ * @enum
+ */
+export declare const DeletionMode: {
+    readonly HARD_DELETE: "HardDelete";
+    readonly SOFT_DELETE: "SoftDelete";
+};
+/**
+ * @public
+ */
+export type DeletionMode = (typeof DeletionMode)[keyof typeof DeletionMode];

package/dist-types/models/models_0.d.ts

@@ -148,6 +148,112 @@ export interface BatchGetRecordResponse {
      */
     UnprocessedIdentifiers: BatchGetRecordIdentifier[] | undefined;
 }
+/**
+ * <p>Time to live duration, where the record is hard deleted after the expiration time is
+ *          reached; <code>ExpiresAt</code> = <code>EventTime</code> + <code>TtlDuration</code>. For
+ *          information on HardDelete, see the <a href="https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_feature_store_DeleteRecord.html">DeleteRecord</a> API in the Amazon SageMaker API Reference guide.</p>
+ * @public
+ */
+export interface TtlDuration {
+    /**
+     * <p>
+     *             <code>TtlDuration</code> time unit.</p>
+     * @public
+     */
+    Unit: TtlDurationUnit | undefined;
+    /**
+     * <p>
+     *             <code>TtlDuration</code> time value.</p>
+     * @public
+     */
+    Value: number | undefined;
+}
+/**
+ * <p>An entry to write as part of a <code>BatchWriteRecord</code> request.</p>
+ * @public
+ */
+export interface BatchWriteRecordEntry {
+    /**
+     * <p>The name or Amazon Resource Name (ARN) of the <code>FeatureGroup</code> to write the
+     *          record to.</p>
+     * @public
+     */
+    FeatureGroupName: string | undefined;
+    /**
+     * <p>List of FeatureValues to be inserted. This will be a full over-write.</p>
+     * @public
+     */
+    Record: FeatureValue[] | undefined;
+    /**
+     * <p>A list of stores to which you're adding the record. By default, Feature Store adds the
+     *          record to all of the stores that you're using for the
+     *             <code>FeatureGroup</code>.</p>
+     * @public
+     */
+    TargetStores?: TargetStore[] | undefined;
+    /**
+     * <p>Time to live duration for this entry, where the record is hard deleted after the
+     *          expiration time is reached; <code>ExpiresAt</code> = <code>EventTime</code> +
+     *             <code>TtlDuration</code>. This overrides the request level
+     *          <code>TtlDuration</code>.</p>
+     * @public
+     */
+    TtlDuration?: TtlDuration | undefined;
+}
+/**
+ * @public
+ */
+export interface BatchWriteRecordRequest {
+    /**
+     * <p>A list of records to write. Each entry specifies the <code>FeatureGroup</code>, the
+     *          record data, and optionally target stores and a TTL duration.</p>
+     * @public
+     */
+    Entries: BatchWriteRecordEntry[] | undefined;
+    /**
+     * <p>Time to live duration applied to all entries in the batch that do not specify their own
+     *             <code>TtlDuration</code>; <code>ExpiresAt</code> = <code>EventTime</code> +
+     *             <code>TtlDuration</code>. For information on HardDelete, see the <a href="https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_feature_store_DeleteRecord.html">DeleteRecord</a> API in the Amazon SageMaker API Reference guide.</p>
+     * @public
+     */
+    TtlDuration?: TtlDuration | undefined;
+}
+/**
+ * <p>The error that has occurred when attempting to write a record in a batch.</p>
+ * @public
+ */
+export interface BatchWriteRecordError {
+    /**
+     * <p>The entry that failed to be written.</p>
+     * @public
+     */
+    Entry: BatchWriteRecordEntry | undefined;
+    /**
+     * <p>The error code for the failed record write.</p>
+     * @public
+     */
+    ErrorCode: string | undefined;
+    /**
+     * <p>The error message for the failed record write.</p>
+     * @public
+     */
+    ErrorMessage: string | undefined;
+}
+/**
+ * @public
+ */
+export interface BatchWriteRecordResponse {
+    /**
+     * <p>A list of errors that occurred when writing records in the batch.</p>
+     * @public
+     */
+    Errors: BatchWriteRecordError[] | undefined;
+    /**
+     * <p>A list of entries that were not processed. These entries can be retried.</p>
+     * @public
+     */
+    UnprocessedEntries: BatchWriteRecordEntry[] | undefined;
+}
 /**
  * @public
  */
@@ -230,24 +336,47 @@ export interface GetRecordResponse {
     ExpiresAt?: string | undefined;
 }
 /**
- * <p>Time to live duration, where the record is hard deleted after the expiration time is
- *          reached; <code>ExpiresAt</code> = <code>EventTime</code> + <code>TtlDuration</code>. For
- *          information on HardDelete, see the <a href="https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_feature_store_DeleteRecord.html">DeleteRecord</a> API in the Amazon SageMaker API Reference guide.</p>
  * @public
  */
-export interface TtlDuration {
+export interface ListRecordsRequest {
     /**
-     * <p>
-     *             <code>TtlDuration</code> time unit.</p>
+     * <p>The name or Amazon Resource Name (ARN) of the feature group to list records from.</p>
      * @public
      */
-    Unit: TtlDurationUnit | undefined;
+    FeatureGroupName: string | undefined;
     /**
-     * <p>
-     *             <code>TtlDuration</code> time value.</p>
+     * <p>The maximum number of record identifiers to return in a single page of results. For the <code>InMemory</code> tier, this value is a hint and not a strict requirement. The response may contain more or fewer results than the specified <code>MaxResults</code>.</p>
      * @public
      */
-    Value: number | undefined;
+    MaxResults?: number | undefined;
+    /**
+     * <p>A token to resume pagination of <code>ListRecords</code> results.</p>
+     * @public
+     */
+    NextToken?: string | undefined;
+    /**
+     * <p>If set to <code>true</code>, the result includes records that have been soft
+     *          deleted.</p>
+     * @public
+     */
+    IncludeSoftDeletedRecords?: boolean | undefined;
+}
+/**
+ * @public
+ */
+export interface ListRecordsResponse {
+    /**
+     * <p>A list of record identifier values for the records stored in the
+     *             <code>OnlineStore</code>.</p>
+     * @public
+     */
+    RecordIdentifiers: string[] | undefined;
+    /**
+     * <p>A token to resume pagination if the response includes more record identifiers than
+     *             <code>MaxResults</code>.</p>
+     * @public
+     */
+    NextToken?: string | undefined;
 }
 /**
  * @public

package/dist-types/pagination/Interfaces.d.ts

@@ -0,0 +1,8 @@
+import type { PaginationConfiguration } from "@smithy/types";
+import { SageMakerFeatureStoreRuntimeClient } from "../SageMakerFeatureStoreRuntimeClient";
+/**
+ * @public
+ */
+export interface SageMakerFeatureStoreRuntimePaginationConfiguration extends PaginationConfiguration {
+    client: SageMakerFeatureStoreRuntimeClient;
+}

package/dist-types/pagination/ListRecordsPaginator.d.ts

@@ -0,0 +1,7 @@
+import type { Paginator } from "@smithy/types";
+import { ListRecordsCommandInput, ListRecordsCommandOutput } from "../commands/ListRecordsCommand";
+import type { SageMakerFeatureStoreRuntimePaginationConfiguration } from "./Interfaces";
+/**
+ * @public
+ */
+export declare const paginateListRecords: (config: SageMakerFeatureStoreRuntimePaginationConfiguration, input: ListRecordsCommandInput, ...rest: any[]) => Paginator<ListRecordsCommandOutput>;

package/dist-types/pagination/index.d.ts

@@ -0,0 +1,2 @@
+export * from "./Interfaces";
+export * from "./ListRecordsPaginator";

package/dist-types/runtimeConfig.browser.d.ts

@@ -13,8 +13,7 @@ export declare const getRuntimeConfig: (config: SageMakerFeatureStoreRuntimeClie
     region: string | import("@smithy/types").Provider<any>;
     requestHandler: import("@smithy/core/protocols").HttpHandler<any> | RequestHandler;
     retryMode: string | import("@smithy/types").Provider<string>;
-    sha256: import("@smithy/types").HashConstructor;
-    streamCollector: import("@smithy/types").StreamCollector;
+    streamCollector: (stream: import("stream").Readable | import("stream/web").ReadableStream | ReadableStream | Blob) => Promise<Uint8Array>;
     useDualstackEndpoint: (boolean | import("@smithy/types").Provider<boolean>) & (boolean | import("@smithy/types").Provider<boolean | undefined>);
     useFipsEndpoint: (boolean | import("@smithy/types").Provider<boolean>) & (boolean | import("@smithy/types").Provider<boolean | undefined>);
     cacheMiddleware?: boolean | undefined;
@@ -24,6 +23,7 @@ export declare const getRuntimeConfig: (config: SageMakerFeatureStoreRuntimeClie
         [setting: string]: unknown;
     };
     apiVersion: string;
+    sha256: import("@smithy/types").HashConstructor;
     urlParser: import("@smithy/types").UrlParser;
     base64Decoder: import("@smithy/types").Decoder;
     base64Encoder: (_input: Uint8Array | string) => string;

package/dist-types/runtimeConfig.d.ts

@@ -14,8 +14,7 @@ export declare const getRuntimeConfig: (config: SageMakerFeatureStoreRuntimeClie
     region: string | import("@smithy/types").Provider<string>;
     requestHandler: RequestHandler | import("@smithy/core/protocols").HttpHandler<any>;
     retryMode: string | import("@smithy/types").Provider<string>;
-    sha256: import("@smithy/types").HashConstructor;
-    streamCollector: import("@smithy/types").StreamCollector;
+    streamCollector: (stream: import("stream").Readable | import("stream/web").ReadableStream | ReadableStream | Blob) => Promise<Uint8Array>;
     useDualstackEndpoint: boolean | import("@smithy/types").Provider<boolean>;
     useFipsEndpoint: boolean | import("@smithy/types").Provider<boolean>;
     userAgentAppId: string | import("@smithy/types").Provider<string | undefined>;
@@ -26,6 +25,7 @@ export declare const getRuntimeConfig: (config: SageMakerFeatureStoreRuntimeClie
         [setting: string]: unknown;
     };
     apiVersion: string;
+    sha256: import("@smithy/types").HashConstructor;
     urlParser: import("@smithy/types").UrlParser;
     base64Decoder: import("@smithy/types").Decoder;
     base64Encoder: (_input: Uint8Array | string) => string;

package/dist-types/runtimeConfig.native.d.ts

@@ -4,7 +4,6 @@ import type { SageMakerFeatureStoreRuntimeClientConfig } from "./SageMakerFeatur
  */
 export declare const getRuntimeConfig: (config: SageMakerFeatureStoreRuntimeClientConfig) => {
     runtime: string;
-    sha256: import("@smithy/types").HashConstructor;
     requestHandler: import("@smithy/types").NodeHttpHandlerOptions | import("@smithy/types").FetchHttpHandlerOptions | Record<string, unknown> | import("@smithy/core/protocols").HttpHandler<any> | import("@smithy/fetch-http-handler").FetchHttpHandler;
     cacheMiddleware?: boolean;
     protocol: import("@smithy/types").ClientProtocol<any, any> | import("@smithy/types").ClientProtocolCtor<any, any> | typeof import("@aws-sdk/core/protocols").AwsRestJsonProtocol;
@@ -13,9 +12,10 @@ export declare const getRuntimeConfig: (config: SageMakerFeatureStoreRuntimeClie
         [setting: string]: unknown;
     };
     apiVersion: string;
+    sha256: import("@smithy/types").HashConstructor;
     urlParser: import("@smithy/types").UrlParser;
     bodyLengthChecker: import("@smithy/types").BodyLengthCalculator;
-    streamCollector: import("@smithy/types").StreamCollector;
+    streamCollector: (stream: import("stream").Readable | import("stream/web").ReadableStream | ReadableStream | Blob) => Promise<Uint8Array>;
     base64Decoder: import("@smithy/types").Decoder;
     base64Encoder: (_input: Uint8Array | string) => string;
     utf8Decoder: import("@smithy/types").Decoder;

package/dist-types/runtimeConfig.shared.d.ts

@@ -21,6 +21,7 @@ export declare const getRuntimeConfig: (config: SageMakerFeatureStoreRuntimeClie
         defaultNamespace?: string;
     };
     serviceId: string;
+    sha256: import("@smithy/types").HashConstructor;
     urlParser: import("@smithy/types").UrlParser;
     utf8Decoder: import("@smithy/types").Decoder;
     utf8Encoder: (input: Uint8Array | string) => string;

package/dist-types/schemas/schemas_0.d.ts

@@ -17,13 +17,21 @@ export declare var BatchGetRecordIdentifier$: StaticStructureSchema;
 export declare var BatchGetRecordRequest$: StaticStructureSchema;
 export declare var BatchGetRecordResponse$: StaticStructureSchema;
 export declare var BatchGetRecordResultDetail$: StaticStructureSchema;
+export declare var BatchWriteRecordEntry$: StaticStructureSchema;
+export declare var BatchWriteRecordError$: StaticStructureSchema;
+export declare var BatchWriteRecordRequest$: StaticStructureSchema;
+export declare var BatchWriteRecordResponse$: StaticStructureSchema;
 export declare var DeleteRecordRequest$: StaticStructureSchema;
 export declare var FeatureValue$: StaticStructureSchema;
 export declare var GetRecordRequest$: StaticStructureSchema;
 export declare var GetRecordResponse$: StaticStructureSchema;
+export declare var ListRecordsRequest$: StaticStructureSchema;
+export declare var ListRecordsResponse$: StaticStructureSchema;
 export declare var PutRecordRequest$: StaticStructureSchema;
 export declare var TtlDuration$: StaticStructureSchema;
 export declare var BatchGetRecord$: StaticOperationSchema;
+export declare var BatchWriteRecord$: StaticOperationSchema;
 export declare var DeleteRecord$: StaticOperationSchema;
 export declare var GetRecord$: StaticOperationSchema;
+export declare var ListRecords$: StaticOperationSchema;
 export declare var PutRecord$: StaticOperationSchema;

package/dist-types/ts3.4/SageMakerFeatureStoreRuntime.d.ts

@@ -1,8 +1,16 @@
-import { HttpHandlerOptions as __HttpHandlerOptions } from "@smithy/types";
+import {
+  HttpHandlerOptions as __HttpHandlerOptions,
+  PaginationConfiguration,
+  Paginator,
+} from "@smithy/types";
 import {
   BatchGetRecordCommandInput,
   BatchGetRecordCommandOutput,
 } from "./commands/BatchGetRecordCommand";
+import {
+  BatchWriteRecordCommandInput,
+  BatchWriteRecordCommandOutput,
+} from "./commands/BatchWriteRecordCommand";
 import {
   DeleteRecordCommandInput,
   DeleteRecordCommandOutput,
@@ -11,6 +19,10 @@ import {
   GetRecordCommandInput,
   GetRecordCommandOutput,
 } from "./commands/GetRecordCommand";
+import {
+  ListRecordsCommandInput,
+  ListRecordsCommandOutput,
+} from "./commands/ListRecordsCommand";
 import {
   PutRecordCommandInput,
   PutRecordCommandOutput,
@@ -30,6 +42,19 @@ export interface SageMakerFeatureStoreRuntime {
     options: __HttpHandlerOptions,
     cb: (err: any, data?: BatchGetRecordCommandOutput) => void
   ): void;
+  batchWriteRecord(
+    args: BatchWriteRecordCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<BatchWriteRecordCommandOutput>;
+  batchWriteRecord(
+    args: BatchWriteRecordCommandInput,
+    cb: (err: any, data?: BatchWriteRecordCommandOutput) => void
+  ): void;
+  batchWriteRecord(
+    args: BatchWriteRecordCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: BatchWriteRecordCommandOutput) => void
+  ): void;
   deleteRecord(
     args: DeleteRecordCommandInput,
     options?: __HttpHandlerOptions
@@ -56,6 +81,19 @@ export interface SageMakerFeatureStoreRuntime {
     options: __HttpHandlerOptions,
     cb: (err: any, data?: GetRecordCommandOutput) => void
   ): void;
+  listRecords(
+    args: ListRecordsCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<ListRecordsCommandOutput>;
+  listRecords(
+    args: ListRecordsCommandInput,
+    cb: (err: any, data?: ListRecordsCommandOutput) => void
+  ): void;
+  listRecords(
+    args: ListRecordsCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: ListRecordsCommandOutput) => void
+  ): void;
   putRecord(
     args: PutRecordCommandInput,
     options?: __HttpHandlerOptions
@@ -69,6 +107,13 @@ export interface SageMakerFeatureStoreRuntime {
     options: __HttpHandlerOptions,
     cb: (err: any, data?: PutRecordCommandOutput) => void
   ): void;
+  paginateListRecords(
+    args: ListRecordsCommandInput,
+    paginationConfig?: Pick<
+      PaginationConfiguration,
+      Exclude<keyof PaginationConfiguration, "client">
+    >
+  ): Paginator<ListRecordsCommandOutput>;
 }
 export declare class SageMakerFeatureStoreRuntime
   extends SageMakerFeatureStoreRuntimeClient

package/dist-types/ts3.4/SageMakerFeatureStoreRuntimeClient.d.ts

@@ -40,6 +40,10 @@ import {
   BatchGetRecordCommandInput,
   BatchGetRecordCommandOutput,
 } from "./commands/BatchGetRecordCommand";
+import {
+  BatchWriteRecordCommandInput,
+  BatchWriteRecordCommandOutput,
+} from "./commands/BatchWriteRecordCommand";
 import {
   DeleteRecordCommandInput,
   DeleteRecordCommandOutput,
@@ -48,6 +52,10 @@ import {
   GetRecordCommandInput,
   GetRecordCommandOutput,
 } from "./commands/GetRecordCommand";
+import {
+  ListRecordsCommandInput,
+  ListRecordsCommandOutput,
+} from "./commands/ListRecordsCommand";
 import {
   PutRecordCommandInput,
   PutRecordCommandOutput,
@@ -61,13 +69,17 @@ import { RuntimeExtension, RuntimeExtensionsConfig } from "./runtimeExtensions";
 export { __Client };
 export type ServiceInputTypes =
   | BatchGetRecordCommandInput
+  | BatchWriteRecordCommandInput
   | DeleteRecordCommandInput
   | GetRecordCommandInput
+  | ListRecordsCommandInput
   | PutRecordCommandInput;
 export type ServiceOutputTypes =
   | BatchGetRecordCommandOutput
+  | BatchWriteRecordCommandOutput
   | DeleteRecordCommandOutput
   | GetRecordCommandOutput
+  | ListRecordsCommandOutput
   | PutRecordCommandOutput;
 export interface ClientDefaults
   extends Partial<__SmithyConfiguration<__HttpHandlerOptions>> {

package/dist-types/ts3.4/commands/BatchWriteRecordCommand.d.ts

@@ -0,0 +1,52 @@
+import { Command as $Command } from "@smithy/core/client";
+import { MetadataBearer as __MetadataBearer } from "@smithy/types";
+import {
+  BatchWriteRecordRequest,
+  BatchWriteRecordResponse,
+} from "../models/models_0";
+import {
+  SageMakerFeatureStoreRuntimeClientResolvedConfig,
+  ServiceInputTypes,
+  ServiceOutputTypes,
+} from "../SageMakerFeatureStoreRuntimeClient";
+export { __MetadataBearer };
+export { $Command };
+export interface BatchWriteRecordCommandInput extends BatchWriteRecordRequest {}
+export interface BatchWriteRecordCommandOutput
+  extends BatchWriteRecordResponse,
+    __MetadataBearer {}
+declare const BatchWriteRecordCommand_base: {
+  new (
+    input: BatchWriteRecordCommandInput
+  ): import("@smithy/core/client").CommandImpl<
+    BatchWriteRecordCommandInput,
+    BatchWriteRecordCommandOutput,
+    SageMakerFeatureStoreRuntimeClientResolvedConfig,
+    ServiceInputTypes,
+    ServiceOutputTypes
+  >;
+  new (
+    input: BatchWriteRecordCommandInput
+  ): import("@smithy/core/client").CommandImpl<
+    BatchWriteRecordCommandInput,
+    BatchWriteRecordCommandOutput,
+    SageMakerFeatureStoreRuntimeClientResolvedConfig,
+    ServiceInputTypes,
+    ServiceOutputTypes
+  >;
+  getEndpointParameterInstructions(): {
+    [x: string]: unknown;
+  };
+};
+export declare class BatchWriteRecordCommand extends BatchWriteRecordCommand_base {
+  protected static __types: {
+    api: {
+      input: BatchWriteRecordRequest;
+      output: BatchWriteRecordResponse;
+    };
+    sdk: {
+      input: BatchWriteRecordCommandInput;
+      output: BatchWriteRecordCommandOutput;
+    };
+  };
+}