@meetkai/mka1 0.51.20 → 0.51.22

package/src/funcs/servingDatasetsUploadContent.ts

@@ -0,0 +1,202 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { SDKCore } from "../core.js";
+import { encodeSimple } from "../lib/encodings.js";
+import { matchStatusCode } from "../lib/http.js";
+import * as M from "../lib/matchers.js";
+import { compactMap } from "../lib/primitives.js";
+import { safeParse } from "../lib/schemas.js";
+import { RequestOptions } from "../lib/sdks.js";
+import { extractSecurity, resolveGlobalSecurity } from "../lib/security.js";
+import { pathToFunc } from "../lib/url.js";
+import * as components from "../models/components/index.js";
+import {
+  ConnectionError,
+  InvalidRequestError,
+  RequestAbortedError,
+  RequestTimeoutError,
+  UnexpectedClientError,
+} from "../models/errors/httpclienterrors.js";
+import * as errors from "../models/errors/index.js";
+import { ResponseValidationError } from "../models/errors/responsevalidationerror.js";
+import { SDKError } from "../models/errors/sdkerror.js";
+import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
+import * as operations from "../models/operations/index.js";
+import { APICall, APIPromise } from "../types/async.js";
+import { Result } from "../types/fp.js";
+
+/**
+ * Upload dataset content through the API
+ *
+ * @remarks
+ * Stream the raw dataset bytes through the serving API into the dataset's object storage. Valid only while the dataset is `pending_upload`; repeating the upload before commit overwrites the previous bytes. The response receipt carries the stored size and SHA-256 digest so the client can verify the bytes before calling commit. Uploads are capped by the cluster's configured maximum dataset size.
+ */
+export function servingDatasetsUploadContent(
+  client: SDKCore,
+  request: operations.UploadServingDatasetContentRequest,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    components.DatasetUploadReceipt,
+    | errors.ServingError
+    | errors.ServingHTTPValidationError
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: SDKCore,
+  request: operations.UploadServingDatasetContentRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      components.DatasetUploadReceipt,
+      | errors.ServingError
+      | errors.ServingHTTPValidationError
+      | SDKError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const parsed = safeParse(
+    request,
+    (value) =>
+      operations.UploadServingDatasetContentRequest$outboundSchema.parse(value),
+    "Input validation failed",
+  );
+  if (!parsed.ok) {
+    return [parsed, { status: "invalid" }];
+  }
+  const payload = parsed.value;
+  const body = payload.RequestBody instanceof Uint8Array
+    ? new Uint8Array(payload.RequestBody).buffer
+    : payload.RequestBody;
+
+  const pathParams = {
+    dataset_id: encodeSimple("dataset_id", payload.dataset_id, {
+      explode: false,
+      charEncoding: "percent",
+    }),
+  };
+  const path = pathToFunc("/api/v1/serving/datasets/{dataset_id}/content")(
+    pathParams,
+  );
+
+  const headers = new Headers(compactMap({
+    "Content-Type": "application/octet-stream",
+    Accept: "application/json",
+    "X-On-Behalf-Of": encodeSimple(
+      "X-On-Behalf-Of",
+      payload["X-On-Behalf-Of"],
+      { explode: false, charEncoding: "none" },
+    ),
+  }));
+
+  const secConfig = await extractSecurity(client._options.bearerAuth);
+  const securityInput = secConfig == null ? {} : { bearerAuth: secConfig };
+  const requestSecurity = resolveGlobalSecurity(securityInput);
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "uploadServingDatasetContent",
+    oAuth2Scopes: null,
+
+    resolvedSecurity: requestSecurity,
+
+    securitySource: client._options.bearerAuth,
+    retryConfig: options?.retries
+      || client._options.retryConfig
+      || {
+        strategy: "backoff",
+        backoff: {
+          initialInterval: 500,
+          maxInterval: 5000,
+          exponent: 1.5,
+          maxElapsedTime: 15000,
+        },
+        retryConnectionErrors: true,
+      }
+      || { strategy: "none" },
+    retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+  };
+
+  const requestRes = client._createRequest(context, {
+    security: requestSecurity,
+    method: "PUT",
+    baseURL: options?.serverURL,
+    path: path,
+    headers: headers,
+    body: body,
+    userAgent: client._options.userAgent,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
+  }, options);
+  if (!requestRes.ok) {
+    return [requestRes, { status: "invalid" }];
+  }
+  const req = requestRes.value;
+
+  const doResult = await client._do(req, {
+    context,
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
+    retryConfig: context.retryConfig,
+    retryCodes: context.retryCodes,
+  });
+  if (!doResult.ok) {
+    return [doResult, { status: "request-error", request: req }];
+  }
+  const response = doResult.value;
+
+  const responseFields = {
+    HttpMeta: { Response: response, Request: req },
+  };
+
+  const [result] = await M.match<
+    components.DatasetUploadReceipt,
+    | errors.ServingError
+    | errors.ServingHTTPValidationError
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, components.DatasetUploadReceipt$inboundSchema),
+    M.jsonErr([403, 404, 409, 413], errors.ServingError$inboundSchema),
+    M.jsonErr(422, errors.ServingHTTPValidationError$inboundSchema),
+    M.fail([401, "4XX"]),
+    M.fail("5XX"),
+  )(response, req, { extraFields: responseFields });
+  if (!result.ok) {
+    return [result, { status: "complete", request: req, response }];
+  }
+
+  return [result, { status: "complete", request: req, response }];
+}

package/src/funcs/servingTenantsDeleteOrganization.ts

@@ -31,7 +31,9 @@ import { Result } from "../types/fp.js";
  * Trigger organization deletion teardown
  *
  * @remarks
- * Durably records an organization.deleted.v1 event and starts (or adopts) the mandatory teardown cascade: the organization is fenced against new work and every deployment is tombstoned and wound down, while all pricing, budget, spend, and usage records are preserved. Idempotent by event id. Cluster administrators only.
+ * Durably records an organization.deleted.v1 event and starts (or adopts) the mandatory teardown cascade: the organization is fenced against new work and every deployment is tombstoned and wound down, while all pricing, budget, spend, and usage records are preserved. Idempotent by event id. Cluster administrators only. Deprecated: this operation has moved to the operator admin plane at `POST /api/v1/admin/tenants/{org_id}/deletion`. This path keeps working for one transition release (responses carry `Deprecation: true` and a `Link` successor header) and is removed from this document in the next SDK major.
+ *
+ * @deprecated method: This will be removed in a future release, please migrate away from it as soon as possible.
  */
 export function servingTenantsDeleteOrganization(
   client: SDKCore,

package/src/funcs/servingTenantsGet.ts

@@ -31,7 +31,9 @@ import { Result } from "../types/fp.js";
  * Get a serving tenant
  *
  * @remarks
- * Get one organization's serving enablement record. Cluster administrators only.
+ * Get one organization's serving enablement record. Cluster administrators only. Deprecated: this operation has moved to the operator admin plane at `GET /api/v1/admin/tenants/{org_id}`. This path keeps working for one transition release (responses carry `Deprecation: true` and a `Link` successor header) and is removed from this document in the next SDK major.
+ *
+ * @deprecated method: This will be removed in a future release, please migrate away from it as soon as possible.
  */
 export function servingTenantsGet(
   client: SDKCore,

package/src/funcs/servingTenantsGetOrganizationDeletion.ts

@@ -31,7 +31,9 @@ import { Result } from "../types/fp.js";
  * Get organization deletion teardown state
  *
  * @remarks
- * Returns the durable teardown-cascade state for an organization (running or complete), so a cluster administrator can confirm offboarding progressed to completion. 404 if the organization has no teardown. Cluster administrators only.
+ * Returns the durable teardown-cascade state for an organization (running or complete), so a cluster administrator can confirm offboarding progressed to completion. 404 if the organization has no teardown. Cluster administrators only. Deprecated: this operation has moved to the operator admin plane at `GET /api/v1/admin/tenants/{org_id}/deletion`. This path keeps working for one transition release (responses carry `Deprecation: true` and a `Link` successor header) and is removed from this document in the next SDK major.
+ *
+ * @deprecated method: This will be removed in a future release, please migrate away from it as soon as possible.
  */
 export function servingTenantsGetOrganizationDeletion(
   client: SDKCore,

package/src/funcs/servingTenantsList.ts

@@ -31,7 +31,9 @@ import { Result } from "../types/fp.js";
  * List serving tenants
  *
  * @remarks
- * List organization-wide serving enablement records. Cluster administrators only; a missing organization is not provisioned.
+ * List organization-wide serving enablement records. Cluster administrators only; a missing organization is not provisioned. Deprecated: this operation has moved to the operator admin plane at `GET /api/v1/admin/tenants`. This path keeps working for one transition release (responses carry `Deprecation: true` and a `Link` successor header) and is removed from this document in the next SDK major.
+ *
+ * @deprecated method: This will be removed in a future release, please migrate away from it as soon as possible.
  */
 export function servingTenantsList(
   client: SDKCore,

package/src/funcs/servingTenantsSet.ts

@@ -31,7 +31,9 @@ import { Result } from "../types/fp.js";
  * Set serving tenant enablement
  *
  * @remarks
- * Enable or disable serving for an organization and all of its teams. Disabling blocks product API access but does not stop existing workloads or callbacks. Cluster administrators only.
+ * Enable or disable serving for an organization and all of its teams. Disabling blocks product API access but does not stop existing workloads or callbacks. Cluster administrators only. Deprecated: this operation has moved to the operator admin plane at `PUT /api/v1/admin/tenants/{org_id}`. This path keeps working for one transition release (responses carry `Deprecation: true` and a `Link` successor header) and is removed from this document in the next SDK major.
+ *
+ * @deprecated method: This will be removed in a future release, please migrate away from it as soon as possible.
  */
 export function servingTenantsSet(
   client: SDKCore,

package/src/lib/config.ts

@@ -65,7 +65,7 @@ export function serverURLFromOptions(options: SDKOptions): URL | null {
 export const SDK_METADATA = {
   language: "typescript",
   openapiDocVersion: "1.1.0",
-  sdkVersion: "0.51.20",
+  sdkVersion: "0.51.22",
   genVersion: "2.924.0",
-  userAgent: "speakeasy-sdk/typescript 0.51.20 2.924.0 1.1.0 @meetkai/mka1",
+  userAgent: "speakeasy-sdk/typescript 0.51.22 2.924.0 1.1.0 @meetkai/mka1",
 } as const;

package/src/mcp-server/mcp-server.ts

@@ -19,7 +19,7 @@ const routes = buildRouteMap({
 export const app = buildApplication(routes, {
   name: "mcp",
   versionInfo: {
-    currentVersion: "0.51.20",
+    currentVersion: "0.51.22",
   },
 });
 

package/src/mcp-server/server.ts

@@ -308,6 +308,7 @@ import { tool$servingDatasetsCreate } from "./tools/servingDatasetsCreate.js";
 import { tool$servingDatasetsDelete } from "./tools/servingDatasetsDelete.js";
 import { tool$servingDatasetsGet } from "./tools/servingDatasetsGet.js";
 import { tool$servingDatasetsList } from "./tools/servingDatasetsList.js";
+import { tool$servingDatasetsUploadContent } from "./tools/servingDatasetsUploadContent.js";
 import { tool$servingDeploymentsCreate } from "./tools/servingDeploymentsCreate.js";
 import { tool$servingDeploymentsDelete } from "./tools/servingDeploymentsDelete.js";
 import { tool$servingDeploymentsDeregister } from "./tools/servingDeploymentsDeregister.js";
@@ -370,7 +371,7 @@ export function createMCPServer(deps: {
 }) {
   const server = new McpServer({
     name: "SDK",
-    version: "0.51.20",
+    version: "0.51.22",
   });
 
   const client = new SDKCore({
@@ -697,6 +698,7 @@ export function createMCPServer(deps: {
   tool(tool$servingDatasetsDelete);
   tool(tool$servingDatasetsGet);
   tool(tool$servingDatasetsCommit);
+  tool(tool$servingDatasetsUploadContent);
   tool(tool$servingDeploymentsList);
   tool(tool$servingDeploymentsCreate);
   tool(tool$servingDeploymentsDelete);

package/src/mcp-server/tools/servingDatasetsCreate.ts

@@ -14,7 +14,7 @@ export const tool$servingDatasetsCreate: ToolDefinition<typeof args> = {
   name: "serving-datasets-create",
   description: `Register or begin uploading a dataset
 
-Register a dataset by reference (\`source\`: an external dataset reference passed to the trainer as-is, or a \`repo://{repo}[@{revision}][/{path}]\` reference reserved for the versioned dataset store) — the dataset is \`ready\` immediately. Omit \`source\` to upload instead: the response is \`pending_upload\` and carries an \`upload\` block of presigned PUT targets; PUT the data, then call commit to seal the immutable manifest.`,
+Register a dataset by reference (\`source\`: an external dataset reference passed to the trainer as-is, or a \`repo://{repo}[@{revision}][/{path}]\` reference reserved for the versioned dataset store) — the dataset is \`ready\` immediately. Omit \`source\` to upload instead: the response is \`pending_upload\` and carries an \`upload\` block describing where to PUT the data. When the artifact store is not internet-exposed (the default posture) the block is \`mode: api\` and the target is the API upload path; when a public artifact endpoint is configured it is \`mode: presigned\` with time-bounded presigned PUT URLs. PUT the data, then call commit to seal the immutable manifest.`,
   args,
   tool: async (client, args, ctx) => {
     const [result, apiCall] = await servingDatasetsCreate(

package/src/mcp-server/tools/servingDatasetsUploadContent.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { servingDatasetsUploadContent } from "../../funcs/servingDatasetsUploadContent.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.UploadServingDatasetContentRequest$inboundSchema,
+};
+
+export const tool$servingDatasetsUploadContent: ToolDefinition<typeof args> = {
+  name: "serving-datasets-upload-content",
+  description: `Upload dataset content through the API
+
+Stream the raw dataset bytes through the serving API into the dataset's object storage. Valid only while the dataset is \`pending_upload\`; repeating the upload before commit overwrites the previous bytes. The response receipt carries the stored size and SHA-256 digest so the client can verify the bytes before calling commit. Uploads are capped by the cluster's configured maximum dataset size.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await servingDatasetsUploadContent(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/src/mcp-server/tools/servingTenantsDeleteOrganization.ts

@@ -16,7 +16,7 @@ export const tool$servingTenantsDeleteOrganization: ToolDefinition<
   name: "serving-tenants-delete-organization",
   description: `Trigger organization deletion teardown
 
-Durably records an organization.deleted.v1 event and starts (or adopts) the mandatory teardown cascade: the organization is fenced against new work and every deployment is tombstoned and wound down, while all pricing, budget, spend, and usage records are preserved. Idempotent by event id. Cluster administrators only.`,
+Durably records an organization.deleted.v1 event and starts (or adopts) the mandatory teardown cascade: the organization is fenced against new work and every deployment is tombstoned and wound down, while all pricing, budget, spend, and usage records are preserved. Idempotent by event id. Cluster administrators only. Deprecated: this operation has moved to the operator admin plane at \`POST /api/v1/admin/tenants/{org_id}/deletion\`. This path keeps working for one transition release (responses carry \`Deprecation: true\` and a \`Link\` successor header) and is removed from this document in the next SDK major.`,
   args,
   tool: async (client, args, ctx) => {
     const [result, apiCall] = await servingTenantsDeleteOrganization(

package/src/mcp-server/tools/servingTenantsGet.ts

@@ -14,7 +14,7 @@ export const tool$servingTenantsGet: ToolDefinition<typeof args> = {
   name: "serving-tenants-get",
   description: `Get a serving tenant
 
-Get one organization's serving enablement record. Cluster administrators only.`,
+Get one organization's serving enablement record. Cluster administrators only. Deprecated: this operation has moved to the operator admin plane at \`GET /api/v1/admin/tenants/{org_id}\`. This path keeps working for one transition release (responses carry \`Deprecation: true\` and a \`Link\` successor header) and is removed from this document in the next SDK major.`,
   args,
   tool: async (client, args, ctx) => {
     const [result, apiCall] = await servingTenantsGet(

package/src/mcp-server/tools/servingTenantsGetOrganizationDeletion.ts

@@ -16,7 +16,7 @@ export const tool$servingTenantsGetOrganizationDeletion: ToolDefinition<
   name: "serving-tenants-get-organization-deletion",
   description: `Get organization deletion teardown state
 
-Returns the durable teardown-cascade state for an organization (running or complete), so a cluster administrator can confirm offboarding progressed to completion. 404 if the organization has no teardown. Cluster administrators only.`,
+Returns the durable teardown-cascade state for an organization (running or complete), so a cluster administrator can confirm offboarding progressed to completion. 404 if the organization has no teardown. Cluster administrators only. Deprecated: this operation has moved to the operator admin plane at \`GET /api/v1/admin/tenants/{org_id}/deletion\`. This path keeps working for one transition release (responses carry \`Deprecation: true\` and a \`Link\` successor header) and is removed from this document in the next SDK major.`,
   args,
   tool: async (client, args, ctx) => {
     const [result, apiCall] = await servingTenantsGetOrganizationDeletion(

package/src/mcp-server/tools/servingTenantsList.ts

@@ -14,7 +14,7 @@ export const tool$servingTenantsList: ToolDefinition<typeof args> = {
   name: "serving-tenants-list",
   description: `List serving tenants
 
-List organization-wide serving enablement records. Cluster administrators only; a missing organization is not provisioned.`,
+List organization-wide serving enablement records. Cluster administrators only; a missing organization is not provisioned. Deprecated: this operation has moved to the operator admin plane at \`GET /api/v1/admin/tenants\`. This path keeps working for one transition release (responses carry \`Deprecation: true\` and a \`Link\` successor header) and is removed from this document in the next SDK major.`,
   args,
   tool: async (client, args, ctx) => {
     const [result, apiCall] = await servingTenantsList(

package/src/mcp-server/tools/servingTenantsSet.ts

@@ -14,7 +14,7 @@ export const tool$servingTenantsSet: ToolDefinition<typeof args> = {
   name: "serving-tenants-set",
   description: `Set serving tenant enablement
 
-Enable or disable serving for an organization and all of its teams. Disabling blocks product API access but does not stop existing workloads or callbacks. Cluster administrators only.`,
+Enable or disable serving for an organization and all of its teams. Disabling blocks product API access but does not stop existing workloads or callbacks. Cluster administrators only. Deprecated: this operation has moved to the operator admin plane at \`PUT /api/v1/admin/tenants/{org_id}\`. This path keeps working for one transition release (responses carry \`Deprecation: true\` and a \`Link\` successor header) and is removed from this document in the next SDK major.`,
   args,
   tool: async (client, args, ctx) => {
     const [result, apiCall] = await servingTenantsSet(

package/src/models/components/acceleratortype.ts

@@ -27,14 +27,14 @@ import {
 } from "./acquirability.js";
 
 /**
- * Coarse accelerator class. npu/tpu are reserved for future catalog entries.
+ * Coarse accelerator class. npu/tpu are reserved for future catalog entries. No cpu-class type is listed: CPU is a class, not an accelerator type, and cpu-class serving is requested through the deployment accelerator spec's class field rather than by naming a type here.
  */
 export const Class = {
   Gpu: "gpu",
   Cpu: "cpu",
 } as const;
 /**
- * Coarse accelerator class. npu/tpu are reserved for future catalog entries.
+ * Coarse accelerator class. npu/tpu are reserved for future catalog entries. No cpu-class type is listed: CPU is a class, not an accelerator type, and cpu-class serving is requested through the deployment accelerator spec's class field rather than by naming a type here.
  */
 export type Class = ClosedEnum<typeof Class>;
 
@@ -75,7 +75,7 @@ export type AcceleratorType = {
    */
   acquirableCount?: number | undefined;
   /**
-   * Coarse accelerator class. npu/tpu are reserved for future catalog entries.
+   * Coarse accelerator class. npu/tpu are reserved for future catalog entries. No cpu-class type is listed: CPU is a class, not an accelerator type, and cpu-class serving is requested through the deployment accelerator spec's class field rather than by naming a type here.
    */
   class: Class;
   /**
@@ -85,11 +85,11 @@ export type AcceleratorType = {
    */
   kind?: Kind | undefined;
   /**
-   * Max accelerators of this type per replica. For class cpu the unit is vCPUs: the maximum rentable vCPU count per replica.
+   * Max accelerators of this type per replica.
    */
   maxCount: number;
   /**
-   * Per-accelerator memory in GB. Absent for class cpu: CPU capacity is sized per request (min_vcpus/min_mem_gb), not by a fixed module.
+   * Per-accelerator memory in GB. Absent for cpu-class entries, which have no fixed memory module.
    */
   memoryGb?: number | undefined;
   name: string;

package/src/models/components/datasetupload.ts

@@ -4,6 +4,7 @@
 
 import * as z from "zod/v3";
 import { safeParse } from "../../lib/schemas.js";
+import { ClosedEnum } from "../../types/enums.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 import {
@@ -14,22 +15,49 @@ import {
 } from "./datasetuploadtarget.js";
 
 /**
- * Presigned upload targets for an upload-mode dataset.
+ * How the targets are addressed: `api` (PUT through the serving API) or `presigned` (direct PUT to a presigned object-store URL).
+ */
+export const DatasetUploadMode = {
+  Api: "api",
+  Presigned: "presigned",
+} as const;
+/**
+ * How the targets are addressed: `api` (PUT through the serving API) or `presigned` (direct PUT to a presigned object-store URL).
+ */
+export type DatasetUploadMode = ClosedEnum<typeof DatasetUploadMode>;
+
+/**
+ * Upload instructions for an upload-mode dataset. `mode` is `api` when the artifact store is not internet-exposed (the default posture): PUT the raw bytes to the API-relative `targets[].url` path. `mode` is `presigned` only when the cluster exposes a public artifact endpoint: targets carry time-bounded presigned URLs.
  */
 export type DatasetUpload = {
+  /**
+   * How the targets are addressed: `api` (PUT through the serving API) or `presigned` (direct PUT to a presigned object-store URL).
+   */
+  mode: DatasetUploadMode;
   targets: Array<DatasetUploadTarget>;
 };
 
+/** @internal */
+export const DatasetUploadMode$inboundSchema: z.ZodNativeEnum<
+  typeof DatasetUploadMode
+> = z.nativeEnum(DatasetUploadMode);
+/** @internal */
+export const DatasetUploadMode$outboundSchema: z.ZodNativeEnum<
+  typeof DatasetUploadMode
+> = DatasetUploadMode$inboundSchema;
+
 /** @internal */
 export const DatasetUpload$inboundSchema: z.ZodType<
   DatasetUpload,
   z.ZodTypeDef,
   unknown
 > = z.object({
+  mode: DatasetUploadMode$inboundSchema,
   targets: z.array(DatasetUploadTarget$inboundSchema),
 });
 /** @internal */
 export type DatasetUpload$Outbound = {
+  mode: string;
   targets: Array<DatasetUploadTarget$Outbound>;
 };
 
@@ -39,6 +67,7 @@ export const DatasetUpload$outboundSchema: z.ZodType<
   z.ZodTypeDef,
   DatasetUpload
 > = z.object({
+  mode: DatasetUploadMode$outboundSchema,
   targets: z.array(DatasetUploadTarget$outboundSchema),
 });
 

package/src/models/components/datasetuploadreceipt.ts

@@ -0,0 +1,80 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+/**
+ * Receipt for one dataset object streamed through the API upload endpoint.
+ */
+export type DatasetUploadReceipt = {
+  /**
+   * The manifest object key the bytes were stored under.
+   */
+  key: string;
+  /**
+   * Hex SHA-256 digest computed while the upload streamed to the artifact store; verify it before committing.
+   */
+  sha256: string;
+  /**
+   * Number of bytes stored.
+   */
+  sizeBytes: number;
+};
+
+/** @internal */
+export const DatasetUploadReceipt$inboundSchema: z.ZodType<
+  DatasetUploadReceipt,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  key: z.string(),
+  sha256: z.string(),
+  size_bytes: z.number().int(),
+}).transform((v) => {
+  return remap$(v, {
+    "size_bytes": "sizeBytes",
+  });
+});
+/** @internal */
+export type DatasetUploadReceipt$Outbound = {
+  key: string;
+  sha256: string;
+  size_bytes: number;
+};
+
+/** @internal */
+export const DatasetUploadReceipt$outboundSchema: z.ZodType<
+  DatasetUploadReceipt$Outbound,
+  z.ZodTypeDef,
+  DatasetUploadReceipt
+> = z.object({
+  key: z.string(),
+  sha256: z.string(),
+  sizeBytes: z.number().int(),
+}).transform((v) => {
+  return remap$(v, {
+    sizeBytes: "size_bytes",
+  });
+});
+
+export function datasetUploadReceiptToJSON(
+  datasetUploadReceipt: DatasetUploadReceipt,
+): string {
+  return JSON.stringify(
+    DatasetUploadReceipt$outboundSchema.parse(datasetUploadReceipt),
+  );
+}
+export function datasetUploadReceiptFromJSON(
+  jsonString: string,
+): SafeParseResult<DatasetUploadReceipt, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => DatasetUploadReceipt$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'DatasetUploadReceipt' from JSON`,
+  );
+}

package/src/models/components/datasetuploadtarget.ts

@@ -17,14 +17,17 @@ export type DatasetUploadTargetMethod = ClosedEnum<
 >;
 
 export type DatasetUploadTarget = {
-  expiresAt: Date;
+  /**
+   * When a presigned target stops accepting uploads. Absent for `mode: api` targets, which remain valid while the dataset is `pending_upload`.
+   */
+  expiresAt?: Date | null | undefined;
   /**
    * The manifest object key this target uploads.
    */
   key: string;
   method: DatasetUploadTargetMethod;
   /**
-   * Presigned URL to PUT the object bytes to.
+   * Where to PUT the object bytes: a presigned object-store URL (`mode: presigned`) or an API-relative path such as `/api/v1/serving/datasets/{dataset_id}/content` (`mode: api`).
    */
   url: string;
 };
@@ -44,7 +47,9 @@ export const DatasetUploadTarget$inboundSchema: z.ZodType<
   z.ZodTypeDef,
   unknown
 > = z.object({
-  expires_at: z.string().datetime({ offset: true }).transform(v => new Date(v)),
+  expires_at: z.nullable(
+    z.string().datetime({ offset: true }).transform(v => new Date(v)),
+  ).optional(),
   key: z.string(),
   method: DatasetUploadTargetMethod$inboundSchema,
   url: z.string(),
@@ -55,7 +60,7 @@ export const DatasetUploadTarget$inboundSchema: z.ZodType<
 });
 /** @internal */
 export type DatasetUploadTarget$Outbound = {
-  expires_at: string;
+  expires_at?: string | null | undefined;
   key: string;
   method: string;
   url: string;
@@ -67,7 +72,7 @@ export const DatasetUploadTarget$outboundSchema: z.ZodType<
   z.ZodTypeDef,
   DatasetUploadTarget
 > = z.object({
-  expiresAt: z.date().transform(v => v.toISOString()),
+  expiresAt: z.nullable(z.date().transform(v => v.toISOString())).optional(),
   key: z.string(),
   method: DatasetUploadTargetMethod$outboundSchema,
   url: z.string(),

package/src/models/components/index.ts

@@ -137,6 +137,7 @@ export * from "./datasetcreate.js";
 export * from "./datasetmanifest.js";
 export * from "./datasetmanifestobject.js";
 export * from "./datasetupload.js";
+export * from "./datasetuploadreceipt.js";
 export * from "./datasetuploadtarget.js";
 export * from "./deactivateregistryentryresponse.js";
 export * from "./decideagentrunmcpapprovalrequest.js";