@functional-systems/lambdadb 0.4.3 → 0.5.0-rc.1

package/src/compat/qdrant/client.ts

@@ -1,6 +1,10 @@
 import { LambdaDBClient, type LambdaDBClientOptions } from "../../client.js";
 import { ResourceNotFoundError } from "../../models/errors/index.js";
-import type { CreateCollectionInput, ListDocsInput } from "../../types/public.js";
+import type {
+  CreateCollectionInput,
+  ListDocsInput,
+  UpdateCollectionInput,
+} from "../../types/public.js";
 import {
   ORIGINAL_ID_FIELD,
   mergeIndexConfigs,
@@ -28,7 +32,7 @@ type ListDocsBody = ListDocsInput;
 
 type CollectionLike = {
   get(): Promise<{ collection: JsonObject }>;
-  update(body: { indexConfigs: IndexConfigs }): Promise<unknown>;
+  update(body: UpdateCollectionInput): Promise<unknown>;
   delete(): Promise<unknown>;
   query(body: JsonObject): Promise<{ docs: unknown[] }>;
   docs: {
@@ -184,7 +188,7 @@ export class QdrantCompatClient {
       collectionName,
       indexConfigs: indexConfigs as CreateCollectionInput["indexConfigs"],
     });
-    await this.waitForCollectionActive(collectionName, body.timeout);
+    await this.waitForCollectionAvailable(collectionName, body.timeout);
     return true;
   }
 
@@ -255,9 +259,12 @@ export class QdrantCompatClient {
     }
 
     await this.client.collection(collectionName).update({
-      indexConfigs: mergeIndexConfigs(existingIndexConfigs, payloadIndexConfig),
+      indexConfigs: mergeIndexConfigs(
+        existingIndexConfigs,
+        payloadIndexConfig,
+      ) as UpdateCollectionInput["indexConfigs"],
     });
-    await this.waitForCollectionActive(collectionName, timeout);
+    await this.waitForCollectionAvailable(collectionName, timeout);
     return true;
   }
 
@@ -690,24 +697,25 @@ export class QdrantCompatClient {
     return response.collection;
   }
 
-  private async waitForCollectionActive(
+  private async waitForCollectionAvailable(
     collectionName: string,
     timeoutSeconds: unknown,
   ): Promise<void> {
     const timeout = typeof timeoutSeconds === "number" ? timeoutSeconds : 60;
     const deadline = Date.now() + timeout * 1000;
-    let lastStatus = "unknown";
     while (true) {
-      const collection = await this.currentCollection(collectionName);
-      const status = collection["collectionStatus"];
-      if (status === undefined || status === "ACTIVE") return;
-      lastStatus = String(status);
-      if (Date.now() >= deadline) {
-        throw new QdrantCompatError(
-          `Collection ${collectionName} did not become ACTIVE within ${timeout}s; last status=${lastStatus}`,
-        );
+      try {
+        await this.currentCollection(collectionName);
+        return;
+      } catch (error) {
+        if (!isNotFoundError(error)) throw error;
+        if (Date.now() >= deadline) {
+          throw new QdrantCompatError(
+            `Collection ${collectionName} was not available within ${timeout}s`,
+          );
+        }
+        await sleep(500);
       }
-      await sleep(500);
     }
   }
 }
@@ -737,6 +745,7 @@ function normalizeConstructorOptions(
     "timeoutMs",
     "retryConfig",
     "httpClient",
+    "transferClient",
     "serverURL",
     "projectHost",
     "path",
@@ -779,6 +788,9 @@ function normalizeConstructorOptions(
   if (timeoutMs !== undefined) normalized.timeoutMs = timeoutMs;
   if (options.retryConfig !== undefined) normalized.retryConfig = options.retryConfig;
   if (options.httpClient !== undefined) normalized.httpClient = options.httpClient;
+  if (options.transferClient !== undefined) {
+    normalized.transferClient = options.transferClient;
+  }
   if (options.serverURL !== undefined) normalized.serverURL = options.serverURL;
   if (options.projectHost !== undefined) normalized.projectHost = options.projectHost;
   return normalized;

package/src/funcs/collectionsCreate.ts

@@ -175,7 +175,7 @@ async function $do(
     | UnexpectedClientError
     | SDKValidationError
   >(
-    M.json(202, operations.CreateCollectionResponse$inboundSchema),
+    M.json(201, operations.CreateCollectionResponse$inboundSchema),
     M.jsonErr(400, errors.BadRequestError$inboundSchema),
     M.jsonErr(401, errors.UnauthenticatedError$inboundSchema),
     M.jsonErr(409, errors.ResourceAlreadyExistsError$inboundSchema),

package/src/funcs/collectionsDelete.ts

@@ -179,7 +179,7 @@ async function $do(
     | UnexpectedClientError
     | SDKValidationError
   >(
-    M.json(202, models.MessageResponse$inboundSchema),
+    M.json(200, models.MessageResponse$inboundSchema),
     M.jsonErr(401, errors.UnauthenticatedError$inboundSchema),
     M.jsonErr(404, errors.ResourceNotFoundError$inboundSchema),
     M.jsonErr(429, errors.TooManyRequestsError$inboundSchema),

package/src/funcs/collectionsDocsGetBulkUpsert.ts

@@ -3,7 +3,7 @@
  */
 
 import { LambdaDBCore } from "../core.js";
-import { encodeSimple } from "../lib/encodings.js";
+import { encodeFormQuery, encodeSimple } from "../lib/encodings.js";
 import * as M from "../lib/matchers.js";
 import { compactMap } from "../lib/primitives.js";
 import { safeParse } from "../lib/schemas.js";
@@ -104,6 +104,10 @@ async function $do(
     pathParams,
   );
 
+  const query = encodeFormQuery({
+    "branch": payload.branch,
+  });
+
   const headers = new Headers(compactMap({
     Accept: "application/json",
   }));
@@ -142,6 +146,7 @@ async function $do(
     method: "GET",
     baseURL: options?.serverURL,
     path: path,
+    query: query,
     headers: headers,
     body: body,
     userAgent: client._options.userAgent,

package/src/funcs/collectionsDocsListDocs.ts

@@ -105,6 +105,8 @@ async function $do(
   const query = encodeFormQuery({
     "includeVectors": payload.includeVectors,
     "pageToken": payload.pageToken,
+    "refKind": payload.refKind,
+    "refName": payload.refName,
     "size": payload.size,
   });
 

package/src/index.ts

@@ -12,12 +12,29 @@ export * from "./sdk/sdk.js";
 export {
   LambdaDBClient,
   CollectionHandle,
+  CollectionDocs,
   DEFAULT_BASE_URL,
   DEFAULT_PROJECT_NAME,
   type RequestOptions as ClientRequestOptions,
   type LambdaDBClientOptions,
 } from "./client.js";
 
+/** Data Versioning lifecycle clients and validated ref/source/target helpers. */
+export {
+  CollectionAliases,
+  CollectionBranches,
+  CollectionTags,
+  DATA_VERSIONING_CONTRACT_REVISION,
+  aliasRef,
+  branchRef,
+  branchSource,
+  branchTarget,
+  tagRef,
+  tagSource,
+  tagTarget,
+} from "./versioning.js";
+export type { VersioningError } from "./versioning.js";
+
 /** Public API types (inputs and responses for the collection-scoped client). */
 export type * from "./types/public.js";
 

package/src/lib/config.ts

@@ -73,8 +73,8 @@ export function serverURLFromOptions(options: SDKOptions): URL | null {
 export const SDK_METADATA = {
   language: "typescript",
   openapiDocVersion: "1.1.1",
-  sdkVersion: "0.4.3",
+  sdkVersion: "0.5.0-rc.1",
   genVersion: "2.798.0",
   userAgent:
-    "speakeasy-sdk/typescript 0.4.3 2.798.0 1.1.1 @functional-systems/lambdadb",
+    "speakeasy-sdk/typescript 0.5.0-rc.1 2.798.0 1.1.1 @functional-systems/lambdadb",
 } as const;

package/src/lib/queryInput.ts

@@ -2,20 +2,13 @@
  * Helper to build QueryCollectionInput for collection.query() / collection.querySafe().
  */
 
-import type { FieldsSelectorUnion, PartitionFilter } from "../models/index.js";
 import type { QueryCollectionInput } from "../types/public.js";
 
 /**
  * Options for createQueryInput (all optional except query).
  */
-export type CreateQueryInputOptions = {
-  size?: number;
-  consistentRead?: boolean;
-  includeVectors?: boolean;
-  sort?: Array<{ [k: string]: unknown }>;
-  fields?: FieldsSelectorUnion;
-  partitionFilter?: PartitionFilter;
-};
+type WithoutQuery<T> = T extends unknown ? Omit<T, "query"> : never;
+export type CreateQueryInputOptions = WithoutQuery<QueryCollectionInput>;
 
 /**
  * Build a query input object for collection.query() or collection.querySafe().
@@ -29,5 +22,5 @@ export function createQueryInput(
   query: { [k: string]: unknown },
   options?: CreateQueryInputOptions,
 ): QueryCollectionInput {
-  return { query, ...options };
+  return { query, ...options } as QueryCollectionInput;
 }

package/src/models/collectionresponse.ts

@@ -14,7 +14,6 @@ import {
   PartitionConfig,
   PartitionConfig$inboundSchema,
 } from "./partitionconfig.js";
-import { Status, Status$inboundSchema } from "./status.js";
 
 export type CollectionResponse = {
   /**
@@ -26,6 +25,10 @@ export type CollectionResponse = {
    */
   collectionName: string;
   indexConfigs: { [k: string]: IndexConfigsUnion };
+  /** Collection description. */
+  description: string;
+  /** Collection metadata tags. */
+  tags: Record<string, string>;
   partitionConfig?: PartitionConfig | undefined;
   /**
    * Total number of partitions including the default partition.
@@ -35,34 +38,22 @@ export type CollectionResponse = {
    * Total number of documents.
    */
   numDocs: number;
+  /** Default writable Branch. */
+  defaultBranchName: "main";
+  /** Number of days that committed snapshots are retained. */
+  snapshotRetentionInDays: number;
   /**
-   * Source project name.
-   */
-  sourceProjectName?: string | undefined;
-  /**
-   * Source collection name.
-   */
-  sourceCollectionName?: string | undefined;
-  /**
-   * Source collection version.
-   */
-  sourceCollectionVersionId?: string | undefined;
-  /**
-   * Status
-   */
-  collectionStatus: Status;
-  /**
-   * Collection creation time in seconds since the Unix epoch.
+   * Collection creation time in milliseconds since the Unix epoch.
    */
   createdAt: number;
   /**
-   * Collection last update time in seconds since the Unix epoch.
+   * Collection last update time in milliseconds since the Unix epoch.
    */
   updatedAt: number;
   /**
-   * Collection data last update time in seconds since the Unix epoch.
+   * Collection data last update time in milliseconds since the Unix epoch.
    */
-  dataUpdatedAt: number;
+  dataUpdatedAt?: number | undefined;
 };
 
 /** @internal */
@@ -74,16 +65,16 @@ export const CollectionResponse$inboundSchema: z.ZodType<
   projectName: z.string(),
   collectionName: z.string(),
   indexConfigs: z.record(IndexConfigsUnion$inboundSchema),
+  description: z.string(),
+  tags: z.record(z.string()),
   partitionConfig: nullToUndefined(PartitionConfig$inboundSchema.optional()),
   numPartitions: z.number().int(),
   numDocs: z.number().int(),
-  sourceProjectName: nullToUndefined(z.string().optional()),
-  sourceCollectionName: nullToUndefined(z.string().optional()),
-  sourceCollectionVersionId: nullToUndefined(z.string().optional()),
-  collectionStatus: Status$inboundSchema,
+  defaultBranchName: z.literal("main"),
+  snapshotRetentionInDays: z.number().int().min(1).max(31),
   createdAt: z.number().int(),
   updatedAt: z.number().int(),
-  dataUpdatedAt: z.number().int(),
+  dataUpdatedAt: nullToUndefined(z.number().int().optional()),
 });
 
 export function collectionResponseFromJSON(

package/src/models/index.ts

@@ -10,3 +10,4 @@ export * from "./partitionconfig.js";
 export * from "./partitionfilter.js";
 export * from "./security.js";
 export * from "./status.js";
+export * from "./versioning.js";

package/src/models/operations/bulkupsertdocs.ts

@@ -10,6 +10,10 @@ export type BulkUpsertDocsRequestBody = {
    * Object key uploaded based on bulk upsert info.
    */
   objectKey: string;
+  /** Content type supplied by the bulk-upload info response. */
+  type?: "application/json" | undefined;
+  /** Write target Branch. Must match the bulk-upload info request. */
+  branch?: string | undefined;
 };
 
 export type BulkUpsertDocsRequest = {
@@ -23,6 +27,8 @@ export type BulkUpsertDocsRequest = {
 /** @internal */
 export type BulkUpsertDocsRequestBody$Outbound = {
   objectKey: string;
+  type: "application/json";
+  branch?: string | undefined;
 };
 
 /** @internal */
@@ -32,6 +38,8 @@ export const BulkUpsertDocsRequestBody$outboundSchema: z.ZodType<
   BulkUpsertDocsRequestBody
 > = z.object({
   objectKey: z.string(),
+  type: z.literal("application/json").default("application/json"),
+  branch: z.string().regex(/^[a-zA-Z0-9_-]{3,52}$/).optional(),
 });
 
 export function bulkUpsertDocsRequestBodyToJSON(

package/src/models/operations/createcollection.ts

@@ -13,30 +13,38 @@ export type CreateCollectionRequest = {
    * Collection name must be unique within a project and the supported maximum length is 52.
    */
   collectionName: string;
-  indexConfigs?: { [k: string]: models.IndexConfigsUnion } | undefined;
+  indexConfigs: { [k: string]: models.IndexConfigsUnion };
+  description?: string | undefined;
+  tags?: Record<string, string> | undefined;
   partitionConfig?: models.PartitionConfig | undefined;
-  sourceProjectName?: string | undefined;
-  sourceCollectionName?: string | undefined;
-  sourceDatetime?: string | undefined;
-  sourceProjectApiKey?: string | undefined;
+  snapshotRetentionInDays?: number | undefined;
+};
+
+export type CreatedCollection = {
+  collectionName: string;
+  description: string;
+  tags: Record<string, string>;
+  defaultBranchName: "main";
+  snapshotRetentionInDays: number;
+  /** Collection creation time as Unix epoch milliseconds. */
+  createdAt: number;
 };
 
 /**
  * Created collection
  */
 export type CreateCollectionResponse = {
-  collection: models.CollectionResponse;
+  collection: CreatedCollection;
 };
 
 /** @internal */
 export type CreateCollectionRequest$Outbound = {
   collectionName: string;
-  indexConfigs?: { [k: string]: models.IndexConfigsUnion$Outbound } | undefined;
+  indexConfigs: { [k: string]: models.IndexConfigsUnion$Outbound };
+  description?: string | undefined;
+  tags?: Record<string, string> | undefined;
   partitionConfig?: models.PartitionConfig$Outbound | undefined;
-  sourceProjectName?: string | undefined;
-  sourceCollectionName?: string | undefined;
-  sourceDatetime?: string | undefined;
-  sourceProjectApiKey?: string | undefined;
+  snapshotRetentionInDays?: number | undefined;
 };
 
 /** @internal */
@@ -45,13 +53,19 @@ export const CreateCollectionRequest$outboundSchema: z.ZodType<
   z.ZodTypeDef,
   CreateCollectionRequest
 > = z.object({
-  collectionName: z.string(),
-  indexConfigs: z.record(models.IndexConfigsUnion$outboundSchema).optional(),
+  collectionName: z.string().regex(/^[a-zA-Z0-9_-]{3,52}$/),
+  indexConfigs: z.record(models.IndexConfigsUnion$outboundSchema),
+  description: z.string().max(255).optional(),
+  tags: z.record(
+    z.string().min(1).max(127).regex(/^[^:#,]+$/),
+  ).refine((value) => Object.keys(value).length <= 5, {
+    message: "Collection tags support at most five entries",
+  }).refine(
+    (value) => Object.keys(value).every((key) => /^[A-Za-z0-9_.-]{1,63}$/.test(key)),
+    { message: "Invalid collection tag key" },
+  ).optional(),
   partitionConfig: models.PartitionConfig$outboundSchema.optional(),
-  sourceProjectName: z.string().optional(),
-  sourceCollectionName: z.string().optional(),
-  sourceDatetime: z.string().optional(),
-  sourceProjectApiKey: z.string().optional(),
+  snapshotRetentionInDays: z.number().int().min(1).max(31).optional(),
 });
 
 export function createCollectionRequestToJSON(
@@ -68,7 +82,14 @@ export const CreateCollectionResponse$inboundSchema: z.ZodType<
   z.ZodTypeDef,
   unknown
 > = z.object({
-  collection: models.CollectionResponse$inboundSchema,
+  collection: z.object({
+    collectionName: z.string(),
+    description: z.string(),
+    tags: z.record(z.string()),
+    defaultBranchName: z.literal("main"),
+    snapshotRetentionInDays: z.number().int().min(1).max(31),
+    createdAt: z.number().int(),
+  }),
 });
 
 export function createCollectionResponseFromJSON(

package/src/models/operations/deletedocs.ts

@@ -16,6 +16,8 @@ export type DeleteDocsRequestBody = {
    */
   filter?: { [k: string]: any } | undefined;
   partitionFilter?: models.PartitionFilter | undefined;
+  /** Write target Branch. Omitting it writes to main. */
+  branch?: string | undefined;
 };
 
 export type DeleteDocsRequest = {
@@ -31,6 +33,7 @@ export type DeleteDocsRequestBody$Outbound = {
   ids?: Array<string> | undefined;
   filter?: { [k: string]: any } | undefined;
   partitionFilter?: models.PartitionFilter$Outbound | undefined;
+  branch?: string | undefined;
 };
 
 /** @internal */
@@ -42,6 +45,7 @@ export const DeleteDocsRequestBody$outboundSchema: z.ZodType<
   ids: z.array(z.string()).optional(),
   filter: z.record(z.any()).optional(),
   partitionFilter: models.PartitionFilter$outboundSchema.optional(),
+  branch: z.string().regex(/^[a-zA-Z0-9_-]{3,52}$/).optional(),
 });
 
 export function deleteDocsRequestBodyToJSON(

package/src/models/operations/fetchdocs.ts

@@ -27,6 +27,8 @@ export type FetchDocsRequestBody = {
    */
   fields?: models.FieldsSelectorUnion | undefined;
   partitionFilter?: models.PartitionFilter | undefined;
+  /** Read target. Omitting it reads from main. */
+  ref?: models.ReadRef | undefined;
 };
 
 export type FetchDocsRequest = {
@@ -72,6 +74,7 @@ export type FetchDocsRequestBody$Outbound = {
   includeVectors: boolean;
   fields?: models.FieldsSelectorUnion$Outbound | undefined;
   partitionFilter?: models.PartitionFilter$Outbound | undefined;
+  ref?: models.ReadRef | undefined;
 };
 
 /** @internal */
@@ -85,6 +88,15 @@ export const FetchDocsRequestBody$outboundSchema: z.ZodType<
   includeVectors: z.boolean().default(false),
   fields: models.FieldsSelectorUnion$outboundSchema.optional(),
   partitionFilter: models.PartitionFilter$outboundSchema.optional(),
+  ref: models.ReadRef$schema.optional(),
+}).superRefine((value, context) => {
+  if (value.consistentRead && value.ref != null && value.ref.kind !== "branch") {
+    context.addIssue({
+      code: z.ZodIssueCode.custom,
+      message: "consistentRead is supported only for direct Branch reads",
+      path: ["consistentRead"],
+    });
+  }
 });
 
 export function fetchDocsRequestBodyToJSON(

package/src/models/operations/getbulkupsertdocs.ts

@@ -3,7 +3,7 @@
  */
 
 import * as z from "zod/v3";
-import { nullToUndefined, safeParse } from "../../lib/schemas.js";
+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";
@@ -13,6 +13,8 @@ export type GetBulkUpsertDocsRequest = {
    * Collection name.
    */
   collectionName: string;
+  /** Write target Branch. Omitting it writes to main. */
+  branch?: string | undefined;
 };
 
 /**
@@ -61,11 +63,14 @@ export type GetBulkUpsertDocsResponse = {
    * Object size limit in bytes.
    */
   sizeLimitBytes: number;
+  /** Signed headers to forward unchanged to the presigned upload URL. */
+  headers: Record<string, string>;
 };
 
 /** @internal */
 export type GetBulkUpsertDocsRequest$Outbound = {
   collectionName: string;
+  branch?: string | undefined;
 };
 
 /** @internal */
@@ -75,6 +80,7 @@ export const GetBulkUpsertDocsRequest$outboundSchema: z.ZodType<
   GetBulkUpsertDocsRequest
 > = z.object({
   collectionName: z.string(),
+  branch: z.string().regex(/^[a-zA-Z0-9_-]{3,52}$/).optional(),
 });
 
 export function getBulkUpsertDocsRequestToJSON(
@@ -101,10 +107,11 @@ export const GetBulkUpsertDocsResponse$inboundSchema: z.ZodType<
   unknown
 > = z.object({
   url: z.string(),
-  type: nullToUndefined(Type$inboundSchema.default("application/json")),
-  httpMethod: nullToUndefined(HttpMethod$inboundSchema.default("PUT")),
+  type: Type$inboundSchema,
+  httpMethod: HttpMethod$inboundSchema,
   objectKey: z.string(),
-  sizeLimitBytes: nullToUndefined(z.number().int().default(209715200)),
+  sizeLimitBytes: z.number().int(),
+  headers: z.record(z.string()),
 });
 
 export function getBulkUpsertDocsResponseFromJSON(

package/src/models/operations/listdocs.ts

@@ -26,6 +26,10 @@ export type ListDocsRequest = {
    * Set to true to include vector values in the response. Defaults to false.
    */
   includeVectors?: boolean | undefined;
+  /** Read ref kind. Must be supplied together with refName. */
+  refKind?: models.ReadRef["kind"] | undefined;
+  /** Read ref name. Must be supplied together with refKind. */
+  refName?: string | undefined;
 };
 
 export type ListDocsDoc = {
@@ -59,6 +63,8 @@ export type ListDocsRequest$Outbound = {
   size?: number | undefined;
   pageToken?: string | undefined;
   includeVectors: boolean;
+  refKind?: models.ReadRef["kind"] | undefined;
+  refName?: string | undefined;
 };
 
 /** @internal */
@@ -71,6 +77,16 @@ export const ListDocsRequest$outboundSchema: z.ZodType<
   size: z.number().int().optional(),
   pageToken: z.string().optional(),
   includeVectors: z.boolean().default(false),
+  refKind: z.enum(["branch", "tag", "alias"]).optional(),
+  refName: z.string().regex(/^[a-zA-Z0-9_-]{3,52}$/).optional(),
+}).superRefine((value, context) => {
+  if ((value.refKind == null) !== (value.refName == null)) {
+    context.addIssue({
+      code: z.ZodIssueCode.custom,
+      message: "refKind and refName must be supplied together",
+      path: [value.refKind == null ? "refKind" : "refName"],
+    });
+  }
 });
 
 export function listDocsRequestToJSON(
@@ -101,6 +117,8 @@ export type ListDocsExtendedRequestBody = {
    * Set to true to include vector values in the response. Defaults to false.
    */
   includeVectors?: boolean | undefined;
+  /** Read target. Omitting it reads from main. */
+  ref?: models.ReadRef | undefined;
 };
 
 export type ListDocsExtendedRequest = {
@@ -119,6 +137,7 @@ export type ListDocsExtendedRequestBody$Outbound = {
   partitionFilter?: models.PartitionFilter$Outbound | undefined;
   fields?: models.FieldsSelectorUnion$Outbound | undefined;
   includeVectors: boolean;
+  ref?: models.ReadRef | undefined;
 };
 
 /** @internal */
@@ -133,6 +152,7 @@ export const ListDocsExtendedRequestBody$outboundSchema: z.ZodType<
   partitionFilter: models.PartitionFilter$outboundSchema.optional(),
   fields: models.FieldsSelectorUnion$outboundSchema.optional(),
   includeVectors: z.boolean().default(false),
+  ref: models.ReadRef$schema.optional(),
 });
 
 export function listDocsExtendedRequestBodyToJSON(

package/src/models/operations/querycollection.ts

@@ -35,6 +35,8 @@ export type QueryCollectionRequestBody = {
    */
   fields?: models.FieldsSelectorUnion | undefined;
   partitionFilter?: models.PartitionFilter | undefined;
+  /** Read target. Omitting it reads from main. */
+  ref?: models.ReadRef | undefined;
 };
 
 export type QueryCollectionRequest = {
@@ -96,6 +98,7 @@ export type QueryCollectionRequestBody$Outbound = {
   sort?: Array<{ [k: string]: any }> | undefined;
   fields?: models.FieldsSelectorUnion$Outbound | undefined;
   partitionFilter?: models.PartitionFilter$Outbound | undefined;
+  ref?: models.ReadRef | undefined;
 };
 
 /** @internal */
@@ -111,6 +114,15 @@ export const QueryCollectionRequestBody$outboundSchema: z.ZodType<
   sort: z.array(z.record(z.any())).optional(),
   fields: models.FieldsSelectorUnion$outboundSchema.optional(),
   partitionFilter: models.PartitionFilter$outboundSchema.optional(),
+  ref: models.ReadRef$schema.optional(),
+}).superRefine((value, context) => {
+  if (value.consistentRead && value.ref != null && value.ref.kind !== "branch") {
+    context.addIssue({
+      code: z.ZodIssueCode.custom,
+      message: "consistentRead is supported only for direct Branch reads",
+      path: ["consistentRead"],
+    });
+  }
 });
 
 export function queryCollectionRequestBodyToJSON(