@orq-ai/node 4.14.0-rc.1 → 4.14.0-rc.3

package/src/models/components/querymemorystoretoolinput.ts

@@ -0,0 +1,51 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
+
+/**
+ * Queries agent memory stores for context
+ */
+export type QueryMemoryStoreToolInput = {
+  type: "query_memory_store";
+  /**
+   * Whether this tool requires approval before execution
+   */
+  requiresApproval?: boolean | undefined;
+  /**
+   * Static tool configuration set at design time. Merged over LLM-provided arguments at execution time.
+   */
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export type QueryMemoryStoreToolInput$Outbound = {
+  type: "query_memory_store";
+  requires_approval?: boolean | undefined;
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export const QueryMemoryStoreToolInput$outboundSchema: z.ZodType<
+  QueryMemoryStoreToolInput$Outbound,
+  z.ZodTypeDef,
+  QueryMemoryStoreToolInput
+> = z.object({
+  type: z.literal("query_memory_store"),
+  requiresApproval: z.boolean().optional(),
+  configuration: z.record(z.any()).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    requiresApproval: "requires_approval",
+  });
+});
+
+export function queryMemoryStoreToolInputToJSON(
+  queryMemoryStoreToolInput: QueryMemoryStoreToolInput,
+): string {
+  return JSON.stringify(
+    QueryMemoryStoreToolInput$outboundSchema.parse(queryMemoryStoreToolInput),
+  );
+}

package/src/models/components/reasoningpart.ts

@@ -40,7 +40,7 @@ export const ReasoningPart$inboundSchema: z.ZodType<
   z.ZodTypeDef,
   unknown
 > = z.object({
-  _id: z.string().default("reasoning_01kzm7pmm4cdjy2mnm3vby0mye"),
+  _id: z.string().default("reasoning_01kznzewms7eyy9hw3yrxgszrx"),
   metadata: z.record(z.any()).optional(),
   kind: z.literal("reasoning"),
   reasoning: z.string(),

package/src/models/components/retrieveagentstoolinput.ts

@@ -0,0 +1,51 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
+
+/**
+ * Retrieves available agents in the system
+ */
+export type RetrieveAgentsToolInput = {
+  type: "retrieve_agents";
+  /**
+   * Whether this tool requires approval before execution
+   */
+  requiresApproval?: boolean | undefined;
+  /**
+   * Static tool configuration set at design time. Merged over LLM-provided arguments at execution time.
+   */
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export type RetrieveAgentsToolInput$Outbound = {
+  type: "retrieve_agents";
+  requires_approval?: boolean | undefined;
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export const RetrieveAgentsToolInput$outboundSchema: z.ZodType<
+  RetrieveAgentsToolInput$Outbound,
+  z.ZodTypeDef,
+  RetrieveAgentsToolInput
+> = z.object({
+  type: z.literal("retrieve_agents"),
+  requiresApproval: z.boolean().optional(),
+  configuration: z.record(z.any()).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    requiresApproval: "requires_approval",
+  });
+});
+
+export function retrieveAgentsToolInputToJSON(
+  retrieveAgentsToolInput: RetrieveAgentsToolInput,
+): string {
+  return JSON.stringify(
+    RetrieveAgentsToolInput$outboundSchema.parse(retrieveAgentsToolInput),
+  );
+}

package/src/models/components/retrieveknowledgebasestoolinput.ts

@@ -0,0 +1,53 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
+
+/**
+ * Lists available knowledge bases
+ */
+export type RetrieveKnowledgeBasesToolInput = {
+  type: "retrieve_knowledge_bases";
+  /**
+   * Whether this tool requires approval before execution
+   */
+  requiresApproval?: boolean | undefined;
+  /**
+   * Static tool configuration set at design time. Merged over LLM-provided arguments at execution time.
+   */
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export type RetrieveKnowledgeBasesToolInput$Outbound = {
+  type: "retrieve_knowledge_bases";
+  requires_approval?: boolean | undefined;
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export const RetrieveKnowledgeBasesToolInput$outboundSchema: z.ZodType<
+  RetrieveKnowledgeBasesToolInput$Outbound,
+  z.ZodTypeDef,
+  RetrieveKnowledgeBasesToolInput
+> = z.object({
+  type: z.literal("retrieve_knowledge_bases"),
+  requiresApproval: z.boolean().optional(),
+  configuration: z.record(z.any()).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    requiresApproval: "requires_approval",
+  });
+});
+
+export function retrieveKnowledgeBasesToolInputToJSON(
+  retrieveKnowledgeBasesToolInput: RetrieveKnowledgeBasesToolInput,
+): string {
+  return JSON.stringify(
+    RetrieveKnowledgeBasesToolInput$outboundSchema.parse(
+      retrieveKnowledgeBasesToolInput,
+    ),
+  );
+}

package/src/models/components/retrievememorystorestoolinput.ts

@@ -0,0 +1,53 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
+
+/**
+ * Lists available memory stores
+ */
+export type RetrieveMemoryStoresToolInput = {
+  type: "retrieve_memory_stores";
+  /**
+   * Whether this tool requires approval before execution
+   */
+  requiresApproval?: boolean | undefined;
+  /**
+   * Static tool configuration set at design time. Merged over LLM-provided arguments at execution time.
+   */
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export type RetrieveMemoryStoresToolInput$Outbound = {
+  type: "retrieve_memory_stores";
+  requires_approval?: boolean | undefined;
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export const RetrieveMemoryStoresToolInput$outboundSchema: z.ZodType<
+  RetrieveMemoryStoresToolInput$Outbound,
+  z.ZodTypeDef,
+  RetrieveMemoryStoresToolInput
+> = z.object({
+  type: z.literal("retrieve_memory_stores"),
+  requiresApproval: z.boolean().optional(),
+  configuration: z.record(z.any()).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    requiresApproval: "requires_approval",
+  });
+});
+
+export function retrieveMemoryStoresToolInputToJSON(
+  retrieveMemoryStoresToolInput: RetrieveMemoryStoresToolInput,
+): string {
+  return JSON.stringify(
+    RetrieveMemoryStoresToolInput$outboundSchema.parse(
+      retrieveMemoryStoresToolInput,
+    ),
+  );
+}

package/src/models/components/routingruleexpression.ts

@@ -7,10 +7,10 @@ import { safeParse } from "../../lib/schemas.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 
-export type Config = {};
+export type RoutingRuleExpressionConfig = {};
 
 export type RoutingRuleExpression = {
-  config?: Config | undefined;
+  config?: RoutingRuleExpressionConfig | undefined;
   /**
    * CEL expression used to match requests. Empty means the rule always matches.
    */
@@ -18,28 +18,37 @@ export type RoutingRuleExpression = {
 };
 
 /** @internal */
-export const Config$inboundSchema: z.ZodType<Config, z.ZodTypeDef, unknown> = z
-  .object({});
+export const RoutingRuleExpressionConfig$inboundSchema: z.ZodType<
+  RoutingRuleExpressionConfig,
+  z.ZodTypeDef,
+  unknown
+> = z.object({});
 /** @internal */
-export type Config$Outbound = {};
+export type RoutingRuleExpressionConfig$Outbound = {};
 
 /** @internal */
-export const Config$outboundSchema: z.ZodType<
-  Config$Outbound,
+export const RoutingRuleExpressionConfig$outboundSchema: z.ZodType<
+  RoutingRuleExpressionConfig$Outbound,
   z.ZodTypeDef,
-  Config
+  RoutingRuleExpressionConfig
 > = z.object({});
 
-export function configToJSON(config: Config): string {
-  return JSON.stringify(Config$outboundSchema.parse(config));
+export function routingRuleExpressionConfigToJSON(
+  routingRuleExpressionConfig: RoutingRuleExpressionConfig,
+): string {
+  return JSON.stringify(
+    RoutingRuleExpressionConfig$outboundSchema.parse(
+      routingRuleExpressionConfig,
+    ),
+  );
 }
-export function configFromJSON(
+export function routingRuleExpressionConfigFromJSON(
   jsonString: string,
-): SafeParseResult<Config, SDKValidationError> {
+): SafeParseResult<RoutingRuleExpressionConfig, SDKValidationError> {
   return safeParse(
     jsonString,
-    (x) => Config$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'Config' from JSON`,
+    (x) => RoutingRuleExpressionConfig$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'RoutingRuleExpressionConfig' from JSON`,
   );
 }
 
@@ -49,12 +58,12 @@ export const RoutingRuleExpression$inboundSchema: z.ZodType<
   z.ZodTypeDef,
   unknown
 > = z.object({
-  config: z.lazy(() => Config$inboundSchema).optional(),
+  config: z.lazy(() => RoutingRuleExpressionConfig$inboundSchema).optional(),
   cel: z.string().optional(),
 });
 /** @internal */
 export type RoutingRuleExpression$Outbound = {
-  config?: Config$Outbound | undefined;
+  config?: RoutingRuleExpressionConfig$Outbound | undefined;
   cel?: string | undefined;
 };
 
@@ -64,7 +73,7 @@ export const RoutingRuleExpression$outboundSchema: z.ZodType<
   z.ZodTypeDef,
   RoutingRuleExpression
 > = z.object({
-  config: z.lazy(() => Config$outboundSchema).optional(),
+  config: z.lazy(() => RoutingRuleExpressionConfig$outboundSchema).optional(),
   cel: z.string().optional(),
 });
 

package/src/models/components/sidekicktoolinput.ts

@@ -0,0 +1,51 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
+
+/**
+ * Delegate a subtask to a secondary model for execution
+ */
+export type SidekickToolInput = {
+  type: "sidekick";
+  /**
+   * Whether this tool requires approval before execution
+   */
+  requiresApproval?: boolean | undefined;
+  /**
+   * Static tool configuration set at design time. Merged over LLM-provided arguments at execution time.
+   */
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export type SidekickToolInput$Outbound = {
+  type: "sidekick";
+  requires_approval?: boolean | undefined;
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export const SidekickToolInput$outboundSchema: z.ZodType<
+  SidekickToolInput$Outbound,
+  z.ZodTypeDef,
+  SidekickToolInput
+> = z.object({
+  type: z.literal("sidekick"),
+  requiresApproval: z.boolean().optional(),
+  configuration: z.record(z.any()).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    requiresApproval: "requires_approval",
+  });
+});
+
+export function sidekickToolInputToJSON(
+  sidekickToolInput: SidekickToolInput,
+): string {
+  return JSON.stringify(
+    SidekickToolInput$outboundSchema.parse(sidekickToolInput),
+  );
+}

package/src/models/components/updateguardrailrulerequest.ts

@@ -0,0 +1,69 @@
+/*
+ * 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 {
+  GuardrailRuleExpression,
+  GuardrailRuleExpression$Outbound,
+  GuardrailRuleExpression$outboundSchema,
+} from "./guardrailruleexpression.js";
+import {
+  GuardrailRuleGuardrail,
+  GuardrailRuleGuardrail$Outbound,
+  GuardrailRuleGuardrail$outboundSchema,
+} from "./guardrailruleguardrail.js";
+import {
+  GuardrailRulePlugin,
+  GuardrailRulePlugin$Outbound,
+  GuardrailRulePlugin$outboundSchema,
+} from "./guardrailruleplugin.js";
+
+export type UpdateGuardrailRuleRequest = {
+  displayName?: string | undefined;
+  description?: string | undefined;
+  enabled?: boolean | undefined;
+  expression?: GuardrailRuleExpression | undefined;
+  guardrails?: Array<GuardrailRuleGuardrail> | undefined;
+  plugins?: Array<GuardrailRulePlugin> | undefined;
+  timeout?: number | undefined;
+};
+
+/** @internal */
+export type UpdateGuardrailRuleRequest$Outbound = {
+  display_name?: string | undefined;
+  description?: string | undefined;
+  enabled?: boolean | undefined;
+  expression?: GuardrailRuleExpression$Outbound | undefined;
+  guardrails?: Array<GuardrailRuleGuardrail$Outbound> | undefined;
+  plugins?: Array<GuardrailRulePlugin$Outbound> | undefined;
+  timeout?: number | undefined;
+};
+
+/** @internal */
+export const UpdateGuardrailRuleRequest$outboundSchema: z.ZodType<
+  UpdateGuardrailRuleRequest$Outbound,
+  z.ZodTypeDef,
+  UpdateGuardrailRuleRequest
+> = z.object({
+  displayName: z.string().optional(),
+  description: z.string().optional(),
+  enabled: z.boolean().optional(),
+  expression: GuardrailRuleExpression$outboundSchema.optional(),
+  guardrails: z.array(GuardrailRuleGuardrail$outboundSchema).optional(),
+  plugins: z.array(GuardrailRulePlugin$outboundSchema).optional(),
+  timeout: z.number().int().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    displayName: "display_name",
+  });
+});
+
+export function updateGuardrailRuleRequestToJSON(
+  updateGuardrailRuleRequest: UpdateGuardrailRuleRequest,
+): string {
+  return JSON.stringify(
+    UpdateGuardrailRuleRequest$outboundSchema.parse(updateGuardrailRuleRequest),
+  );
+}

package/src/models/components/updateguardrailruleresponse.ts

@@ -0,0 +1,37 @@
+/*
+ * 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";
+import { GuardrailRule, GuardrailRule$inboundSchema } from "./guardrailrule.js";
+
+export type UpdateGuardrailRuleResponse = {
+  guardrailRule: GuardrailRule;
+};
+
+/** @internal */
+export const UpdateGuardrailRuleResponse$inboundSchema: z.ZodType<
+  UpdateGuardrailRuleResponse,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  guardrail_rule: GuardrailRule$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "guardrail_rule": "guardrailRule",
+  });
+});
+
+export function updateGuardrailRuleResponseFromJSON(
+  jsonString: string,
+): SafeParseResult<UpdateGuardrailRuleResponse, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => UpdateGuardrailRuleResponse$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'UpdateGuardrailRuleResponse' from JSON`,
+  );
+}

package/src/models/components/webscrapertoolinput.ts

@@ -0,0 +1,51 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
+
+/**
+ * Scrapes and extracts content from web pages
+ */
+export type WebScraperToolInput = {
+  type: "web_scraper";
+  /**
+   * Whether this tool requires approval before execution
+   */
+  requiresApproval?: boolean | undefined;
+  /**
+   * Static tool configuration set at design time. Merged over LLM-provided arguments at execution time.
+   */
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export type WebScraperToolInput$Outbound = {
+  type: "web_scraper";
+  requires_approval?: boolean | undefined;
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export const WebScraperToolInput$outboundSchema: z.ZodType<
+  WebScraperToolInput$Outbound,
+  z.ZodTypeDef,
+  WebScraperToolInput
+> = z.object({
+  type: z.literal("web_scraper"),
+  requiresApproval: z.boolean().optional(),
+  configuration: z.record(z.any()).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    requiresApproval: "requires_approval",
+  });
+});
+
+export function webScraperToolInputToJSON(
+  webScraperToolInput: WebScraperToolInput,
+): string {
+  return JSON.stringify(
+    WebScraperToolInput$outboundSchema.parse(webScraperToolInput),
+  );
+}

package/src/models/components/writememorystoretoolinput.ts

@@ -0,0 +1,51 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
+
+/**
+ * Writes information to agent memory stores
+ */
+export type WriteMemoryStoreToolInput = {
+  type: "write_memory_store";
+  /**
+   * Whether this tool requires approval before execution
+   */
+  requiresApproval?: boolean | undefined;
+  /**
+   * Static tool configuration set at design time. Merged over LLM-provided arguments at execution time.
+   */
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export type WriteMemoryStoreToolInput$Outbound = {
+  type: "write_memory_store";
+  requires_approval?: boolean | undefined;
+  configuration?: { [k: string]: any } | undefined;
+};
+
+/** @internal */
+export const WriteMemoryStoreToolInput$outboundSchema: z.ZodType<
+  WriteMemoryStoreToolInput$Outbound,
+  z.ZodTypeDef,
+  WriteMemoryStoreToolInput
+> = z.object({
+  type: z.literal("write_memory_store"),
+  requiresApproval: z.boolean().optional(),
+  configuration: z.record(z.any()).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    requiresApproval: "requires_approval",
+  });
+});
+
+export function writeMemoryStoreToolInputToJSON(
+  writeMemoryStoreToolInput: WriteMemoryStoreToolInput,
+): string {
+  return JSON.stringify(
+    WriteMemoryStoreToolInput$outboundSchema.parse(writeMemoryStoreToolInput),
+  );
+}

package/src/models/operations/createannotationqueue.ts

@@ -151,7 +151,7 @@ export const CreateAnnotationQueueResponseBody$inboundSchema: z.ZodType<
   created: z.string().datetime({ offset: true }).transform(v => new Date(v))
     .optional(),
   updated: z.string().datetime({ offset: true }).default(
-    "2026-08-09T21:43:29.724Z",
+    "2026-08-10T13:57:55.635Z",
   ).transform(v => new Date(v)),
 }).transform((v) => {
   return remap$(v, {

package/src/models/operations/createdataset.ts

@@ -138,7 +138,7 @@ export const CreateDatasetResponseBody$inboundSchema: z.ZodType<
   created: z.string().datetime({ offset: true }).transform(v => new Date(v))
     .optional(),
   updated: z.string().datetime({ offset: true }).default(
-    "2026-08-09T21:43:29.724Z",
+    "2026-08-10T13:57:55.635Z",
   ).transform(v => new Date(v)),
 }).transform((v) => {
   return remap$(v, {

package/src/models/operations/createdatasetitem.ts

@@ -2732,7 +2732,7 @@ export const Evaluations4$inboundSchema: z.ZodType<
   explanation: z.string().optional(),
   reviewed_by_id: z.string(),
   reviewed_at: z.string().datetime({ offset: true }).default(
-    "2026-08-09T21:43:44.538Z",
+    "2026-08-10T13:58:10.857Z",
   ).transform(v => new Date(v)),
   type: z.literal("string_array"),
   values: z.array(z.string()),
@@ -2822,7 +2822,7 @@ export const Evaluations3$inboundSchema: z.ZodType<
   explanation: z.string().optional(),
   reviewed_by_id: z.string(),
   reviewed_at: z.string().datetime({ offset: true }).default(
-    "2026-08-09T21:43:44.538Z",
+    "2026-08-10T13:58:10.856Z",
   ).transform(v => new Date(v)),
   type: z.literal("boolean"),
   value: z.boolean(),
@@ -2907,7 +2907,7 @@ export const Evaluations2$inboundSchema: z.ZodType<
   explanation: z.string().optional(),
   reviewed_by_id: z.string(),
   reviewed_at: z.string().datetime({ offset: true }).default(
-    "2026-08-09T21:43:44.537Z",
+    "2026-08-10T13:58:10.856Z",
   ).transform(v => new Date(v)),
   type: z.literal("number"),
   value: z.number(),
@@ -2993,7 +2993,7 @@ export const Evaluations1$inboundSchema: z.ZodType<
   explanation: z.string().optional(),
   reviewed_by_id: z.string(),
   reviewed_at: z.string().datetime({ offset: true }).default(
-    "2026-08-09T21:43:44.537Z",
+    "2026-08-10T13:58:10.855Z",
   ).transform(v => new Date(v)),
   type: z.literal("string"),
   value: z.string(),
@@ -3080,7 +3080,7 @@ export const CreateDatasetItemResponseBody$inboundSchema: z.ZodType<
   created: z.string().datetime({ offset: true }).transform(v => new Date(v))
     .optional(),
   updated: z.string().datetime({ offset: true }).default(
-    "2026-08-09T21:43:29.724Z",
+    "2026-08-10T13:57:55.635Z",
   ).transform(v => new Date(v)),
 }).transform((v) => {
   return remap$(v, {

package/src/models/operations/createdatasource.ts

@@ -487,7 +487,7 @@ export const CreateDatasourceResponseBody$inboundSchema: z.ZodType<
   z.ZodTypeDef,
   unknown
 > = z.object({
-  _id: z.string().default("01KZM7PN2MQTSBX9KTTFFE04KX"),
+  _id: z.string().default("01KZNZEX38VHTW4F91BYBPR78X"),
   display_name: z.string(),
   description: z.nullable(z.string()).optional(),
   status: CreateDatasourceStatus$inboundSchema,

package/src/models/operations/creatererank.ts

@@ -225,6 +225,10 @@ export type CreateRerankRequestBody = {
    * The number of most relevant documents or indices to return, defaults to the length of the documents
    */
   topN?: number | undefined;
+  /**
+   * Whether to return the documents in the response
+   */
+  returnDocuments?: boolean | undefined;
   /**
    * The filename of the document to rerank
    */
@@ -784,6 +788,7 @@ export type CreateRerankRequestBody$Outbound = {
   documents: Array<string>;
   model: string;
   top_n?: number | undefined;
+  return_documents?: boolean | undefined;
   filename?: string | null | undefined;
   name?: string | undefined;
   fallbacks?: Array<CreateRerankFallbacks$Outbound> | undefined;
@@ -810,7 +815,8 @@ export const CreateRerankRequestBody$outboundSchema: z.ZodType<
   query: z.string(),
   documents: z.array(z.string()),
   model: z.string(),
-  topN: z.number().optional(),
+  topN: z.number().int().optional(),
+  returnDocuments: z.boolean().optional(),
   filename: z.nullable(z.string()).optional(),
   name: z.string().optional(),
   fallbacks: z.array(z.lazy(() => CreateRerankFallbacks$outboundSchema))
@@ -831,6 +837,7 @@ export const CreateRerankRequestBody$outboundSchema: z.ZodType<
 }).transform((v) => {
   return remap$(v, {
     topN: "top_n",
+    returnDocuments: "return_documents",
     loadBalancer: "load_balancer",
   });
 });