@mistralai/mistralai 2.1.2 → 2.2.1

package/src/models/components/activitytaskfailedresponse.ts

@@ -0,0 +1,97 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: 386a95e00e4c
+ */
+
+import * as z from "zod/v4";
+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 {
+  ActivityTaskFailedAttributes,
+  ActivityTaskFailedAttributes$inboundSchema,
+} from "./activitytaskfailedattributes.js";
+
+/**
+ * Emitted when an activity task fails after exhausting all retry attempts.
+ *
+ * @remarks
+ *
+ * This is a terminal event indicating the activity could not complete successfully.
+ */
+export type ActivityTaskFailedResponse = {
+  /**
+   * Unique identifier for this event instance.
+   */
+  eventId: string;
+  /**
+   * Unix timestamp in nanoseconds when the event was created.
+   */
+  eventTimestamp: number;
+  /**
+   * Execution ID of the root workflow that initiated this execution chain.
+   */
+  rootWorkflowExecId: string;
+  /**
+   * Execution ID of the parent workflow that initiated this execution. If this is a root workflow, this field is not set.
+   */
+  parentWorkflowExecId: string | null;
+  /**
+   * Execution ID of the workflow that emitted this event.
+   */
+  workflowExecId: string;
+  /**
+   * Run ID of the workflow execution. Changes on continue-as-new while workflow_exec_id stays the same.
+   */
+  workflowRunId: string;
+  /**
+   * The registered name of the workflow that emitted this event.
+   */
+  workflowName: string;
+  /**
+   * Event type discriminator.
+   */
+  eventType: "ACTIVITY_TASK_FAILED";
+  /**
+   * Attributes for activity task failed events (final failure after all retries).
+   */
+  attributes: ActivityTaskFailedAttributes;
+};
+
+/** @internal */
+export const ActivityTaskFailedResponse$inboundSchema: z.ZodType<
+  ActivityTaskFailedResponse,
+  unknown
+> = z.object({
+  event_id: z.string(),
+  event_timestamp: z.int(),
+  root_workflow_exec_id: z.string(),
+  parent_workflow_exec_id: z.nullable(z.string()),
+  workflow_exec_id: z.string(),
+  workflow_run_id: z.string(),
+  workflow_name: z.string(),
+  event_type: z.literal("ACTIVITY_TASK_FAILED").default("ACTIVITY_TASK_FAILED"),
+  attributes: ActivityTaskFailedAttributes$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "event_id": "eventId",
+    "event_timestamp": "eventTimestamp",
+    "root_workflow_exec_id": "rootWorkflowExecId",
+    "parent_workflow_exec_id": "parentWorkflowExecId",
+    "workflow_exec_id": "workflowExecId",
+    "workflow_run_id": "workflowRunId",
+    "workflow_name": "workflowName",
+    "event_type": "eventType",
+  });
+});
+
+export function activityTaskFailedResponseFromJSON(
+  jsonString: string,
+): SafeParseResult<ActivityTaskFailedResponse, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ActivityTaskFailedResponse$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ActivityTaskFailedResponse' from JSON`,
+  );
+}

package/src/models/components/activitytaskretryingattributes.ts

@@ -0,0 +1,59 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: a0da7dd8bb99
+ */
+
+import * as z from "zod/v4";
+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 { Failure, Failure$inboundSchema } from "./failure.js";
+
+/**
+ * Attributes for activity task retrying events.
+ */
+export type ActivityTaskRetryingAttributes = {
+  /**
+   * Unique identifier for the activity task within the workflow.
+   */
+  taskId: string;
+  /**
+   * The registered name of the activity being executed.
+   */
+  activityName: string;
+  /**
+   * The attempt number that failed (1-indexed).
+   */
+  attempt: number;
+  /**
+   * Represents an error or exception that occurred during execution.
+   */
+  failure: Failure;
+};
+
+/** @internal */
+export const ActivityTaskRetryingAttributes$inboundSchema: z.ZodType<
+  ActivityTaskRetryingAttributes,
+  unknown
+> = z.object({
+  task_id: z.string(),
+  activity_name: z.string(),
+  attempt: z.int(),
+  failure: Failure$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "task_id": "taskId",
+    "activity_name": "activityName",
+  });
+});
+
+export function activityTaskRetryingAttributesFromJSON(
+  jsonString: string,
+): SafeParseResult<ActivityTaskRetryingAttributes, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ActivityTaskRetryingAttributes$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ActivityTaskRetryingAttributes' from JSON`,
+  );
+}

package/src/models/components/activitytaskretryingresponse.ts

@@ -0,0 +1,99 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: 283a6c69c1b0
+ */
+
+import * as z from "zod/v4";
+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 {
+  ActivityTaskRetryingAttributes,
+  ActivityTaskRetryingAttributes$inboundSchema,
+} from "./activitytaskretryingattributes.js";
+
+/**
+ * Emitted when an activity task fails and will be retried.
+ *
+ * @remarks
+ *
+ * Contains information about the failed attempt and the error that occurred.
+ */
+export type ActivityTaskRetryingResponse = {
+  /**
+   * Unique identifier for this event instance.
+   */
+  eventId: string;
+  /**
+   * Unix timestamp in nanoseconds when the event was created.
+   */
+  eventTimestamp: number;
+  /**
+   * Execution ID of the root workflow that initiated this execution chain.
+   */
+  rootWorkflowExecId: string;
+  /**
+   * Execution ID of the parent workflow that initiated this execution. If this is a root workflow, this field is not set.
+   */
+  parentWorkflowExecId: string | null;
+  /**
+   * Execution ID of the workflow that emitted this event.
+   */
+  workflowExecId: string;
+  /**
+   * Run ID of the workflow execution. Changes on continue-as-new while workflow_exec_id stays the same.
+   */
+  workflowRunId: string;
+  /**
+   * The registered name of the workflow that emitted this event.
+   */
+  workflowName: string;
+  /**
+   * Event type discriminator.
+   */
+  eventType: "ACTIVITY_TASK_RETRYING";
+  /**
+   * Attributes for activity task retrying events.
+   */
+  attributes: ActivityTaskRetryingAttributes;
+};
+
+/** @internal */
+export const ActivityTaskRetryingResponse$inboundSchema: z.ZodType<
+  ActivityTaskRetryingResponse,
+  unknown
+> = z.object({
+  event_id: z.string(),
+  event_timestamp: z.int(),
+  root_workflow_exec_id: z.string(),
+  parent_workflow_exec_id: z.nullable(z.string()),
+  workflow_exec_id: z.string(),
+  workflow_run_id: z.string(),
+  workflow_name: z.string(),
+  event_type: z.literal("ACTIVITY_TASK_RETRYING").default(
+    "ACTIVITY_TASK_RETRYING",
+  ),
+  attributes: ActivityTaskRetryingAttributes$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "event_id": "eventId",
+    "event_timestamp": "eventTimestamp",
+    "root_workflow_exec_id": "rootWorkflowExecId",
+    "parent_workflow_exec_id": "parentWorkflowExecId",
+    "workflow_exec_id": "workflowExecId",
+    "workflow_run_id": "workflowRunId",
+    "workflow_name": "workflowName",
+    "event_type": "eventType",
+  });
+});
+
+export function activityTaskRetryingResponseFromJSON(
+  jsonString: string,
+): SafeParseResult<ActivityTaskRetryingResponse, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ActivityTaskRetryingResponse$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ActivityTaskRetryingResponse' from JSON`,
+  );
+}

package/src/models/components/activitytaskstartedattributesresponse.ts

@@ -0,0 +1,62 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: 4b5bd5f1fdb6
+ */
+
+import * as z from "zod/v4";
+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 {
+  JSONPayloadResponse,
+  JSONPayloadResponse$inboundSchema,
+} from "./jsonpayloadresponse.js";
+
+/**
+ * Attributes for activity task started events.
+ */
+export type ActivityTaskStartedAttributesResponse = {
+  /**
+   * Unique identifier for the activity task within the workflow.
+   */
+  taskId: string;
+  /**
+   * The registered name of the activity being executed.
+   */
+  activityName: string;
+  /**
+   * A payload containing arbitrary JSON data.
+   *
+   * @remarks
+   *
+   * Used for complete state snapshots or final results.
+   */
+  input: JSONPayloadResponse;
+};
+
+/** @internal */
+export const ActivityTaskStartedAttributesResponse$inboundSchema: z.ZodType<
+  ActivityTaskStartedAttributesResponse,
+  unknown
+> = z.object({
+  task_id: z.string(),
+  activity_name: z.string(),
+  input: JSONPayloadResponse$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "task_id": "taskId",
+    "activity_name": "activityName",
+  });
+});
+
+export function activityTaskStartedAttributesResponseFromJSON(
+  jsonString: string,
+): SafeParseResult<ActivityTaskStartedAttributesResponse, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) =>
+      ActivityTaskStartedAttributesResponse$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ActivityTaskStartedAttributesResponse' from JSON`,
+  );
+}

package/src/models/components/activitytaskstartedresponse.ts

@@ -0,0 +1,100 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: 40e1a43d6b75
+ */
+
+import * as z from "zod/v4";
+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 {
+  ActivityTaskStartedAttributesResponse,
+  ActivityTaskStartedAttributesResponse$inboundSchema,
+} from "./activitytaskstartedattributesresponse.js";
+
+/**
+ * Emitted when an activity task begins execution.
+ *
+ * @remarks
+ *
+ * This is the first event for an activity, emitted on the first attempt only.
+ * Subsequent retry attempts emit ACTIVITY_TASK_RETRYING instead.
+ */
+export type ActivityTaskStartedResponse = {
+  /**
+   * Unique identifier for this event instance.
+   */
+  eventId: string;
+  /**
+   * Unix timestamp in nanoseconds when the event was created.
+   */
+  eventTimestamp: number;
+  /**
+   * Execution ID of the root workflow that initiated this execution chain.
+   */
+  rootWorkflowExecId: string;
+  /**
+   * Execution ID of the parent workflow that initiated this execution. If this is a root workflow, this field is not set.
+   */
+  parentWorkflowExecId: string | null;
+  /**
+   * Execution ID of the workflow that emitted this event.
+   */
+  workflowExecId: string;
+  /**
+   * Run ID of the workflow execution. Changes on continue-as-new while workflow_exec_id stays the same.
+   */
+  workflowRunId: string;
+  /**
+   * The registered name of the workflow that emitted this event.
+   */
+  workflowName: string;
+  /**
+   * Event type discriminator.
+   */
+  eventType: "ACTIVITY_TASK_STARTED";
+  /**
+   * Attributes for activity task started events.
+   */
+  attributes: ActivityTaskStartedAttributesResponse;
+};
+
+/** @internal */
+export const ActivityTaskStartedResponse$inboundSchema: z.ZodType<
+  ActivityTaskStartedResponse,
+  unknown
+> = z.object({
+  event_id: z.string(),
+  event_timestamp: z.int(),
+  root_workflow_exec_id: z.string(),
+  parent_workflow_exec_id: z.nullable(z.string()),
+  workflow_exec_id: z.string(),
+  workflow_run_id: z.string(),
+  workflow_name: z.string(),
+  event_type: z.literal("ACTIVITY_TASK_STARTED").default(
+    "ACTIVITY_TASK_STARTED",
+  ),
+  attributes: ActivityTaskStartedAttributesResponse$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "event_id": "eventId",
+    "event_timestamp": "eventTimestamp",
+    "root_workflow_exec_id": "rootWorkflowExecId",
+    "parent_workflow_exec_id": "parentWorkflowExecId",
+    "workflow_exec_id": "workflowExecId",
+    "workflow_run_id": "workflowRunId",
+    "workflow_name": "workflowName",
+    "event_type": "eventType",
+  });
+});
+
+export function activityTaskStartedResponseFromJSON(
+  jsonString: string,
+): SafeParseResult<ActivityTaskStartedResponse, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ActivityTaskStartedResponse$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ActivityTaskStartedResponse' from JSON`,
+  );
+}

package/src/models/components/agent.ts

@@ -35,7 +35,6 @@ import {
   ImageGenerationTool,
   ImageGenerationTool$inboundSchema,
 } from "./imagegenerationtool.js";
-import { MetadataDict, MetadataDict$inboundSchema } from "./metadatadict.js";
 import {
   WebSearchPremiumTool,
   WebSearchPremiumTool$inboundSchema,
@@ -81,7 +80,7 @@ export type Agent = {
   name: string;
   description?: string | null | undefined;
   handoffs?: Array<string> | null | undefined;
-  metadata?: MetadataDict | null | undefined;
+  metadata?: { [k: string]: any } | null | undefined;
   object: "agent";
   id: string;
   version: number;
@@ -133,7 +132,7 @@ export const Agent$inboundSchema: z.ZodType<Agent, unknown> = z.object({
   name: z.string(),
   description: z.nullable(z.string()).optional(),
   handoffs: z.nullable(z.array(z.string())).optional(),
-  metadata: z.nullable(MetadataDict$inboundSchema).optional(),
+  metadata: z.nullable(z.record(z.string(), z.any())).optional(),
   object: z.literal("agent").default("agent"),
   id: z.string(),
   version: z.int(),

package/src/models/components/agentconversation.ts

@@ -9,7 +9,6 @@ import { safeParse } from "../../lib/schemas.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { smartUnion } from "../../types/smartUnion.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
-import { MetadataDict, MetadataDict$inboundSchema } from "./metadatadict.js";
 
 export type AgentConversationAgentVersion = string | number;
 
@@ -25,7 +24,7 @@ export type AgentConversation = {
   /**
    * Custom metadata for the conversation.
    */
-  metadata?: MetadataDict | null | undefined;
+  metadata?: { [k: string]: any } | null | undefined;
   object: "conversation";
   id: string;
   createdAt: Date;
@@ -57,7 +56,7 @@ export const AgentConversation$inboundSchema: z.ZodType<
 > = z.object({
   name: z.nullable(z.string()).optional(),
   description: z.nullable(z.string()).optional(),
-  metadata: z.nullable(MetadataDict$inboundSchema).optional(),
+  metadata: z.nullable(z.record(z.string(), z.any())).optional(),
   object: z.literal("conversation").default("conversation"),
   id: z.string(),
   created_at: z.iso.datetime({ offset: true }).transform(v => new Date(v)),

package/src/models/components/agentscompletionrequest.ts

@@ -11,11 +11,31 @@ import {
   AssistantMessage$Outbound,
   AssistantMessage$outboundSchema,
 } from "./assistantmessage.js";
+import {
+  CodeInterpreterTool,
+  CodeInterpreterTool$Outbound,
+  CodeInterpreterTool$outboundSchema,
+} from "./codeinterpretertool.js";
+import {
+  CustomConnector,
+  CustomConnector$Outbound,
+  CustomConnector$outboundSchema,
+} from "./customconnector.js";
+import {
+  DocumentLibraryTool,
+  DocumentLibraryTool$Outbound,
+  DocumentLibraryTool$outboundSchema,
+} from "./documentlibrarytool.js";
 import {
   GuardrailConfig,
   GuardrailConfig$Outbound,
   GuardrailConfig$outboundSchema,
 } from "./guardrailconfig.js";
+import {
+  ImageGenerationTool,
+  ImageGenerationTool$Outbound,
+  ImageGenerationTool$outboundSchema,
+} from "./imagegenerationtool.js";
 import {
   MistralPromptMode,
   MistralPromptMode$outboundSchema,
@@ -59,6 +79,16 @@ import {
   UserMessage$Outbound,
   UserMessage$outboundSchema,
 } from "./usermessage.js";
+import {
+  WebSearchPremiumTool,
+  WebSearchPremiumTool$Outbound,
+  WebSearchPremiumTool$outboundSchema,
+} from "./websearchpremiumtool.js";
+import {
+  WebSearchTool,
+  WebSearchTool$Outbound,
+  WebSearchTool$outboundSchema,
+} from "./websearchtool.js";
 
 /**
  * Stop generation if this token is detected. Or if one of these tokens is detected when providing an array
@@ -71,6 +101,15 @@ export type AgentsCompletionRequestMessage =
   | ToolMessage
   | UserMessage;
 
+export type AgentsCompletionRequestTool =
+  | Tool
+  | WebSearchTool
+  | WebSearchPremiumTool
+  | CodeInterpreterTool
+  | ImageGenerationTool
+  | DocumentLibraryTool
+  | CustomConnector;
+
 export type AgentsCompletionRequestToolChoice = ToolChoice | ToolChoiceEnum;
 
 export type AgentsCompletionRequest = {
@@ -85,7 +124,7 @@ export type AgentsCompletionRequest = {
   /**
    * Stop generation if this token is detected. Or if one of these tokens is detected when providing an array
    */
-  stop?: string | Array<string> | undefined;
+  stop?: string | Array<string> | null | undefined;
   /**
    * The seed to use for random sampling. If set, different calls will generate deterministic results.
    */
@@ -104,16 +143,27 @@ export type AgentsCompletionRequest = {
    * Specify the format that the model must output. By default it will use `{ "type": "text" }`. Setting to `{ "type": "json_object" }` enables JSON mode, which guarantees the message the model generates is in JSON. When using JSON mode you MUST also instruct the model to produce JSON yourself with a system or a user message. Setting to `{ "type": "json_schema" }` enables JSON schema mode, which guarantees the message the model generates is in JSON and follows the schema you provide.
    */
   responseFormat?: ResponseFormat | undefined;
-  tools?: Array<Tool> | null | undefined;
+  tools?:
+    | Array<
+      | Tool
+      | WebSearchTool
+      | WebSearchPremiumTool
+      | CodeInterpreterTool
+      | ImageGenerationTool
+      | DocumentLibraryTool
+      | CustomConnector
+    >
+    | null
+    | undefined;
   toolChoice?: ToolChoice | ToolChoiceEnum | undefined;
   /**
    * The `presence_penalty` determines how much the model penalizes the repetition of words or phrases. A higher presence penalty encourages the model to use a wider variety of words and phrases, making the output more diverse and creative.
    */
-  presencePenalty?: number | undefined;
+  presencePenalty?: number | null | undefined;
   /**
    * The `frequency_penalty` penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
    */
-  frequencyPenalty?: number | undefined;
+  frequencyPenalty?: number | null | undefined;
   /**
    * Number of completions to return for each request, input tokens are only billed once.
    */
@@ -184,6 +234,40 @@ export function agentsCompletionRequestMessageToJSON(
   );
 }
 
+/** @internal */
+export type AgentsCompletionRequestTool$Outbound =
+  | Tool$Outbound
+  | WebSearchTool$Outbound
+  | WebSearchPremiumTool$Outbound
+  | CodeInterpreterTool$Outbound
+  | ImageGenerationTool$Outbound
+  | DocumentLibraryTool$Outbound
+  | CustomConnector$Outbound;
+
+/** @internal */
+export const AgentsCompletionRequestTool$outboundSchema: z.ZodType<
+  AgentsCompletionRequestTool$Outbound,
+  AgentsCompletionRequestTool
+> = z.union([
+  Tool$outboundSchema,
+  WebSearchTool$outboundSchema,
+  WebSearchPremiumTool$outboundSchema,
+  CodeInterpreterTool$outboundSchema,
+  ImageGenerationTool$outboundSchema,
+  DocumentLibraryTool$outboundSchema,
+  CustomConnector$outboundSchema,
+]);
+
+export function agentsCompletionRequestToolToJSON(
+  agentsCompletionRequestTool: AgentsCompletionRequestTool,
+): string {
+  return JSON.stringify(
+    AgentsCompletionRequestTool$outboundSchema.parse(
+      agentsCompletionRequestTool,
+    ),
+  );
+}
+
 /** @internal */
 export type AgentsCompletionRequestToolChoice$Outbound =
   | ToolChoice$Outbound
@@ -209,7 +293,7 @@ export function agentsCompletionRequestToolChoiceToJSON(
 export type AgentsCompletionRequest$Outbound = {
   max_tokens?: number | null | undefined;
   stream: boolean;
-  stop?: string | Array<string> | undefined;
+  stop?: string | Array<string> | null | undefined;
   random_seed?: number | null | undefined;
   metadata?: { [k: string]: any } | null | undefined;
   messages: Array<
@@ -219,10 +303,21 @@ export type AgentsCompletionRequest$Outbound = {
     | UserMessage$Outbound
   >;
   response_format?: ResponseFormat$Outbound | undefined;
-  tools?: Array<Tool$Outbound> | null | undefined;
+  tools?:
+    | Array<
+      | Tool$Outbound
+      | WebSearchTool$Outbound
+      | WebSearchPremiumTool$Outbound
+      | CodeInterpreterTool$Outbound
+      | ImageGenerationTool$Outbound
+      | DocumentLibraryTool$Outbound
+      | CustomConnector$Outbound
+    >
+    | null
+    | undefined;
   tool_choice?: ToolChoice$Outbound | string | undefined;
-  presence_penalty?: number | undefined;
-  frequency_penalty?: number | undefined;
+  presence_penalty?: number | null | undefined;
+  frequency_penalty?: number | null | undefined;
   n?: number | null | undefined;
   prediction?: Prediction$Outbound | undefined;
   parallel_tool_calls?: boolean | undefined;
@@ -239,7 +334,7 @@ export const AgentsCompletionRequest$outboundSchema: z.ZodType<
 > = z.object({
   maxTokens: z.nullable(z.int()).optional(),
   stream: z.boolean().default(false),
-  stop: smartUnion([z.string(), z.array(z.string())]).optional(),
+  stop: z.nullable(smartUnion([z.string(), z.array(z.string())])).optional(),
   randomSeed: z.nullable(z.int()).optional(),
   metadata: z.nullable(z.record(z.string(), z.any())).optional(),
   messages: z.array(
@@ -253,13 +348,25 @@ export const AgentsCompletionRequest$outboundSchema: z.ZodType<
     ]),
   ),
   responseFormat: ResponseFormat$outboundSchema.optional(),
-  tools: z.nullable(z.array(Tool$outboundSchema)).optional(),
+  tools: z.nullable(
+    z.array(
+      z.union([
+        Tool$outboundSchema,
+        WebSearchTool$outboundSchema,
+        WebSearchPremiumTool$outboundSchema,
+        CodeInterpreterTool$outboundSchema,
+        ImageGenerationTool$outboundSchema,
+        DocumentLibraryTool$outboundSchema,
+        CustomConnector$outboundSchema,
+      ]),
+    ),
+  ).optional(),
   toolChoice: smartUnion([
     ToolChoice$outboundSchema,
     ToolChoiceEnum$outboundSchema,
   ]).optional(),
-  presencePenalty: z.number().optional(),
-  frequencyPenalty: z.number().optional(),
+  presencePenalty: z.nullable(z.number()).optional(),
+  frequencyPenalty: z.nullable(z.number()).optional(),
   n: z.nullable(z.int()).optional(),
   prediction: Prediction$outboundSchema.optional(),
   parallelToolCalls: z.boolean().optional(),