@meetkai/mka1 0.54.6 → 0.54.8

package/src/models/components/method.ts

@@ -5,6 +5,8 @@
 import * as z from "zod/v3";
 import { remap as remap$ } from "../../lib/primitives.js";
 import { safeParse } from "../../lib/schemas.js";
+import * as discriminatedUnionTypes from "../../types/discriminatedUnion.js";
+import { discriminatedUnion } from "../../types/discriminatedUnion.js";
 import { ClosedEnum } from "../../types/enums.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
@@ -135,13 +137,25 @@ export type Graders1 = {
   operation: GradersOperation;
 };
 
-export type Graders = Graders1 | Graders2 | Graders3 | Graders4 | Graders5;
+export type Graders =
+  | Graders1
+  | Graders2
+  | Graders3
+  | Graders4
+  | Graders5
+  | discriminatedUnionTypes.Unknown<"type">;
 
 export type Grader2 = {
   type: "multi";
   name: string;
   graders: {
-    [k: string]: Graders1 | Graders2 | Graders3 | Graders4 | Graders5;
+    [k: string]:
+      | Graders1
+      | Graders2
+      | Graders3
+      | Graders4
+      | Graders5
+      | discriminatedUnionTypes.Unknown<"type">;
   };
   calculateOutput: string;
 };
@@ -237,12 +251,32 @@ export type One1 = {
   operation: Operation;
 };
 
-export type Grader1 = One1 | One2 | One3 | One4 | One5;
-
-export type Grader = Grader2 | One1 | One2 | One3 | One4 | One5;
+export type Grader1 =
+  | One1
+  | One2
+  | One3
+  | One4
+  | One5
+  | discriminatedUnionTypes.Unknown<"type">;
+
+export type Grader =
+  | Grader2
+  | One1
+  | One2
+  | One3
+  | One4
+  | One5
+  | discriminatedUnionTypes.Unknown<"type">;
 
 export type Reinforcement = {
-  grader: Grader2 | One1 | One2 | One3 | One4 | One5;
+  grader:
+    | Grader2
+    | One1
+    | One2
+    | One3
+    | One4
+    | One5
+    | discriminatedUnionTypes.Unknown<"type">;
   hyperparameters?: ReinforcementHyperparameters | null | undefined;
 };
 
@@ -775,13 +809,13 @@ export function graders1FromJSON(
 
 /** @internal */
 export const Graders$inboundSchema: z.ZodType<Graders, z.ZodTypeDef, unknown> =
-  z.union([
-    z.lazy(() => Graders1$inboundSchema),
-    z.lazy(() => Graders2$inboundSchema),
-    z.lazy(() => Graders3$inboundSchema),
-    z.lazy(() => Graders4$inboundSchema),
-    z.lazy(() => Graders5$inboundSchema),
-  ]);
+  discriminatedUnion("type", {
+    string_check: z.lazy(() => Graders1$inboundSchema),
+    text_similarity: z.lazy(() => Graders2$inboundSchema),
+    python: z.lazy(() => Graders3$inboundSchema),
+    score_model: z.lazy(() => Graders4$inboundSchema),
+    label_model: z.lazy(() => Graders5$inboundSchema),
+  });
 /** @internal */
 export type Graders$Outbound =
   | Graders1$Outbound
@@ -821,15 +855,13 @@ export const Grader2$inboundSchema: z.ZodType<Grader2, z.ZodTypeDef, unknown> =
   z.object({
     type: z.literal("multi"),
     name: z.string(),
-    graders: z.record(
-      z.union([
-        z.lazy(() => Graders1$inboundSchema),
-        z.lazy(() => Graders2$inboundSchema),
-        z.lazy(() => Graders3$inboundSchema),
-        z.lazy(() => Graders4$inboundSchema),
-        z.lazy(() => Graders5$inboundSchema),
-      ]),
-    ),
+    graders: z.record(discriminatedUnion("type", {
+      string_check: z.lazy(() => Graders1$inboundSchema),
+      text_similarity: z.lazy(() => Graders2$inboundSchema),
+      python: z.lazy(() => Graders3$inboundSchema),
+      score_model: z.lazy(() => Graders4$inboundSchema),
+      label_model: z.lazy(() => Graders5$inboundSchema),
+    })),
     calculate_output: z.string(),
   }).transform((v) => {
     return remap$(v, {
@@ -1294,13 +1326,13 @@ export function one1FromJSON(
 
 /** @internal */
 export const Grader1$inboundSchema: z.ZodType<Grader1, z.ZodTypeDef, unknown> =
-  z.union([
-    z.lazy(() => One1$inboundSchema),
-    z.lazy(() => One2$inboundSchema),
-    z.lazy(() => One3$inboundSchema),
-    z.lazy(() => One4$inboundSchema),
-    z.lazy(() => One5$inboundSchema),
-  ]);
+  discriminatedUnion("type", {
+    string_check: z.lazy(() => One1$inboundSchema),
+    text_similarity: z.lazy(() => One2$inboundSchema),
+    python: z.lazy(() => One3$inboundSchema),
+    score_model: z.lazy(() => One4$inboundSchema),
+    label_model: z.lazy(() => One5$inboundSchema),
+  });
 /** @internal */
 export type Grader1$Outbound =
   | One1$Outbound
@@ -1339,13 +1371,13 @@ export function grader1FromJSON(
 export const Grader$inboundSchema: z.ZodType<Grader, z.ZodTypeDef, unknown> = z
   .union([
     z.lazy(() => Grader2$inboundSchema),
-    z.union([
-      z.lazy(() => One1$inboundSchema),
-      z.lazy(() => One2$inboundSchema),
-      z.lazy(() => One3$inboundSchema),
-      z.lazy(() => One4$inboundSchema),
-      z.lazy(() => One5$inboundSchema),
-    ]),
+    discriminatedUnion("type", {
+      string_check: z.lazy(() => One1$inboundSchema),
+      text_similarity: z.lazy(() => One2$inboundSchema),
+      python: z.lazy(() => One3$inboundSchema),
+      score_model: z.lazy(() => One4$inboundSchema),
+      label_model: z.lazy(() => One5$inboundSchema),
+    }),
   ]);
 /** @internal */
 export type Grader$Outbound =
@@ -1393,13 +1425,13 @@ export const Reinforcement$inboundSchema: z.ZodType<
 > = z.object({
   grader: z.union([
     z.lazy(() => Grader2$inboundSchema),
-    z.union([
-      z.lazy(() => One1$inboundSchema),
-      z.lazy(() => One2$inboundSchema),
-      z.lazy(() => One3$inboundSchema),
-      z.lazy(() => One4$inboundSchema),
-      z.lazy(() => One5$inboundSchema),
-    ]),
+    discriminatedUnion("type", {
+      string_check: z.lazy(() => One1$inboundSchema),
+      text_similarity: z.lazy(() => One2$inboundSchema),
+      python: z.lazy(() => One3$inboundSchema),
+      score_model: z.lazy(() => One4$inboundSchema),
+      label_model: z.lazy(() => One5$inboundSchema),
+    }),
   ]),
   hyperparameters: z.nullable(ReinforcementHyperparameters$inboundSchema)
     .optional(),

package/src/models/components/openaichatcompletionchoice.ts

@@ -5,6 +5,8 @@
 import * as z from "zod/v3";
 import { remap as remap$ } from "../../lib/primitives.js";
 import { safeParse } from "../../lib/schemas.js";
+import * as discriminatedUnionTypes from "../../types/discriminatedUnion.js";
+import { discriminatedUnion } from "../../types/discriminatedUnion.js";
 import { ClosedEnum } from "../../types/enums.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
@@ -134,7 +136,8 @@ export type Content2 =
   | OpenAIChatCompletionChoice21
   | OpenAIChatCompletionChoice22
   | OpenAIChatCompletionChoice23
-  | OpenAIChatCompletionChoice24;
+  | OpenAIChatCompletionChoice24
+  | discriminatedUnionTypes.Unknown<"type">;
 
 /**
  * The contents of the assistant message. Can be null when the message contains tool calls only.
@@ -146,6 +149,7 @@ export type OpenAIChatCompletionChoiceContent =
     | OpenAIChatCompletionChoice22
     | OpenAIChatCompletionChoice23
     | OpenAIChatCompletionChoice24
+    | discriminatedUnionTypes.Unknown<"type">
   >;
 
 /**
@@ -203,6 +207,7 @@ export type Message = {
       | OpenAIChatCompletionChoice22
       | OpenAIChatCompletionChoice23
       | OpenAIChatCompletionChoice24
+      | discriminatedUnionTypes.Unknown<"type">
     >
     | null;
   /**
@@ -732,12 +737,12 @@ export const Content2$inboundSchema: z.ZodType<
   Content2,
   z.ZodTypeDef,
   unknown
-> = z.union([
-  z.lazy(() => OpenAIChatCompletionChoice21$inboundSchema),
-  z.lazy(() => OpenAIChatCompletionChoice22$inboundSchema),
-  z.lazy(() => OpenAIChatCompletionChoice23$inboundSchema),
-  z.lazy(() => OpenAIChatCompletionChoice24$inboundSchema),
-]);
+> = discriminatedUnion("type", {
+  text: z.lazy(() => OpenAIChatCompletionChoice21$inboundSchema),
+  image_url: z.lazy(() => OpenAIChatCompletionChoice22$inboundSchema),
+  input_audio: z.lazy(() => OpenAIChatCompletionChoice23$inboundSchema),
+  file: z.lazy(() => OpenAIChatCompletionChoice24$inboundSchema),
+});
 /** @internal */
 export type Content2$Outbound =
   | OpenAIChatCompletionChoice21$Outbound
@@ -777,12 +782,12 @@ export const OpenAIChatCompletionChoiceContent$inboundSchema: z.ZodType<
   unknown
 > = z.union([
   z.string(),
-  z.array(z.union([
-    z.lazy(() => OpenAIChatCompletionChoice21$inboundSchema),
-    z.lazy(() => OpenAIChatCompletionChoice22$inboundSchema),
-    z.lazy(() => OpenAIChatCompletionChoice23$inboundSchema),
-    z.lazy(() => OpenAIChatCompletionChoice24$inboundSchema),
-  ])),
+  z.array(discriminatedUnion("type", {
+    text: z.lazy(() => OpenAIChatCompletionChoice21$inboundSchema),
+    image_url: z.lazy(() => OpenAIChatCompletionChoice22$inboundSchema),
+    input_audio: z.lazy(() => OpenAIChatCompletionChoice23$inboundSchema),
+    file: z.lazy(() => OpenAIChatCompletionChoice24$inboundSchema),
+  })),
 ]);
 /** @internal */
 export type OpenAIChatCompletionChoiceContent$Outbound =
@@ -930,14 +935,12 @@ export const Message$inboundSchema: z.ZodType<Message, z.ZodTypeDef, unknown> =
     content: z.nullable(
       z.union([
         z.string(),
-        z.array(
-          z.union([
-            z.lazy(() => OpenAIChatCompletionChoice21$inboundSchema),
-            z.lazy(() => OpenAIChatCompletionChoice22$inboundSchema),
-            z.lazy(() => OpenAIChatCompletionChoice23$inboundSchema),
-            z.lazy(() => OpenAIChatCompletionChoice24$inboundSchema),
-          ]),
-        ),
+        z.array(discriminatedUnion("type", {
+          text: z.lazy(() => OpenAIChatCompletionChoice21$inboundSchema),
+          image_url: z.lazy(() => OpenAIChatCompletionChoice22$inboundSchema),
+          input_audio: z.lazy(() => OpenAIChatCompletionChoice23$inboundSchema),
+          file: z.lazy(() => OpenAIChatCompletionChoice24$inboundSchema),
+        })),
       ]),
     ),
     reasoning: z.nullable(z.string()).optional(),

package/src/models/components/openaichatcompletionchunk.ts

@@ -5,6 +5,8 @@
 import * as z from "zod/v3";
 import { remap as remap$ } from "../../lib/primitives.js";
 import { safeParse } from "../../lib/schemas.js";
+import * as discriminatedUnionTypes from "../../types/discriminatedUnion.js";
+import { discriminatedUnion } from "../../types/discriminatedUnion.js";
 import { ClosedEnum } from "../../types/enums.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
@@ -140,7 +142,8 @@ export type OpenAIChatCompletionChunkContent2 =
   | OpenAIChatCompletionChunk21
   | OpenAIChatCompletionChunk22
   | OpenAIChatCompletionChunk23
-  | OpenAIChatCompletionChunk24;
+  | OpenAIChatCompletionChunk24
+  | discriminatedUnionTypes.Unknown<"type">;
 
 /**
  * The content of the chunk. May be a string, content parts, or null.
@@ -152,6 +155,7 @@ export type OpenAIChatCompletionChunkContent =
     | OpenAIChatCompletionChunk22
     | OpenAIChatCompletionChunk23
     | OpenAIChatCompletionChunk24
+    | discriminatedUnionTypes.Unknown<"type">
   >;
 
 /**
@@ -208,6 +212,7 @@ export type Delta = {
       | OpenAIChatCompletionChunk22
       | OpenAIChatCompletionChunk23
       | OpenAIChatCompletionChunk24
+      | discriminatedUnionTypes.Unknown<"type">
     >
     | null
     | undefined;
@@ -778,12 +783,12 @@ export const OpenAIChatCompletionChunkContent2$inboundSchema: z.ZodType<
   OpenAIChatCompletionChunkContent2,
   z.ZodTypeDef,
   unknown
-> = z.union([
-  z.lazy(() => OpenAIChatCompletionChunk21$inboundSchema),
-  z.lazy(() => OpenAIChatCompletionChunk22$inboundSchema),
-  z.lazy(() => OpenAIChatCompletionChunk23$inboundSchema),
-  z.lazy(() => OpenAIChatCompletionChunk24$inboundSchema),
-]);
+> = discriminatedUnion("type", {
+  text: z.lazy(() => OpenAIChatCompletionChunk21$inboundSchema),
+  image_url: z.lazy(() => OpenAIChatCompletionChunk22$inboundSchema),
+  input_audio: z.lazy(() => OpenAIChatCompletionChunk23$inboundSchema),
+  file: z.lazy(() => OpenAIChatCompletionChunk24$inboundSchema),
+});
 /** @internal */
 export type OpenAIChatCompletionChunkContent2$Outbound =
   | OpenAIChatCompletionChunk21$Outbound
@@ -829,12 +834,12 @@ export const OpenAIChatCompletionChunkContent$inboundSchema: z.ZodType<
   unknown
 > = z.union([
   z.string(),
-  z.array(z.union([
-    z.lazy(() => OpenAIChatCompletionChunk21$inboundSchema),
-    z.lazy(() => OpenAIChatCompletionChunk22$inboundSchema),
-    z.lazy(() => OpenAIChatCompletionChunk23$inboundSchema),
-    z.lazy(() => OpenAIChatCompletionChunk24$inboundSchema),
-  ])),
+  z.array(discriminatedUnion("type", {
+    text: z.lazy(() => OpenAIChatCompletionChunk21$inboundSchema),
+    image_url: z.lazy(() => OpenAIChatCompletionChunk22$inboundSchema),
+    input_audio: z.lazy(() => OpenAIChatCompletionChunk23$inboundSchema),
+    file: z.lazy(() => OpenAIChatCompletionChunk24$inboundSchema),
+  })),
 ]);
 /** @internal */
 export type OpenAIChatCompletionChunkContent$Outbound =
@@ -984,14 +989,12 @@ export const Delta$inboundSchema: z.ZodType<Delta, z.ZodTypeDef, unknown> = z
     content: z.nullable(
       z.union([
         z.string(),
-        z.array(
-          z.union([
-            z.lazy(() => OpenAIChatCompletionChunk21$inboundSchema),
-            z.lazy(() => OpenAIChatCompletionChunk22$inboundSchema),
-            z.lazy(() => OpenAIChatCompletionChunk23$inboundSchema),
-            z.lazy(() => OpenAIChatCompletionChunk24$inboundSchema),
-          ]),
-        ),
+        z.array(discriminatedUnion("type", {
+          text: z.lazy(() => OpenAIChatCompletionChunk21$inboundSchema),
+          image_url: z.lazy(() => OpenAIChatCompletionChunk22$inboundSchema),
+          input_audio: z.lazy(() => OpenAIChatCompletionChunk23$inboundSchema),
+          file: z.lazy(() => OpenAIChatCompletionChunk24$inboundSchema),
+        })),
       ]),
     ).optional(),
     reasoning: z.nullable(z.string()).optional(),

package/src/models/components/outputmessage.ts

@@ -4,6 +4,8 @@
 
 import * as z from "zod/v3";
 import { safeParse } from "../../lib/schemas.js";
+import * as discriminatedUnionTypes from "../../types/discriminatedUnion.js";
+import { discriminatedUnion } from "../../types/discriminatedUnion.js";
 import { ClosedEnum } from "../../types/enums.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
@@ -22,7 +24,8 @@ import {
 
 export type OutputMessageContent =
   | (OutputText & { type: "output_text" })
-  | (Refusal & { type: "refusal" });
+  | (Refusal & { type: "refusal" })
+  | discriminatedUnionTypes.Unknown<"type">;
 
 /**
  * The processing status of this message (in_progress, completed, or incomplete).
@@ -69,7 +72,9 @@ export type OutputMessage = {
    * Array of content items in the assistant's response. Can contain text outputs or refusal messages.
    */
   content: Array<
-    (OutputText & { type: "output_text" }) | (Refusal & { type: "refusal" })
+    | (OutputText & { type: "output_text" })
+    | (Refusal & { type: "refusal" })
+    | discriminatedUnionTypes.Unknown<"type">
   >;
   /**
    * The processing status of this message (in_progress, completed, or incomplete).
@@ -86,10 +91,12 @@ export const OutputMessageContent$inboundSchema: z.ZodType<
   OutputMessageContent,
   z.ZodTypeDef,
   unknown
-> = z.union([
-  OutputText$inboundSchema.and(z.object({ type: z.literal("output_text") })),
-  Refusal$inboundSchema.and(z.object({ type: z.literal("refusal") })),
-]);
+> = discriminatedUnion("type", {
+  output_text: OutputText$inboundSchema.and(
+    z.object({ type: z.literal("output_text") }),
+  ),
+  refusal: Refusal$inboundSchema.and(z.object({ type: z.literal("refusal") })),
+});
 /** @internal */
 export type OutputMessageContent$Outbound =
   | (OutputText$Outbound & { type: "output_text" })
@@ -149,12 +156,14 @@ export const OutputMessage$inboundSchema: z.ZodType<
   id: z.string(),
   role: z.literal("assistant"),
   content: z.array(
-    z.union([
-      OutputText$inboundSchema.and(
+    discriminatedUnion("type", {
+      output_text: OutputText$inboundSchema.and(
         z.object({ type: z.literal("output_text") }),
       ),
-      Refusal$inboundSchema.and(z.object({ type: z.literal("refusal") })),
-    ]),
+      refusal: Refusal$inboundSchema.and(
+        z.object({ type: z.literal("refusal") }),
+      ),
+    }),
   ),
   status: OutputMessageStatus$inboundSchema,
   phase: Phase$inboundSchema.optional(),

package/src/models/components/outputmessagewithid.ts

@@ -5,6 +5,8 @@
 import * as z from "zod/v3";
 import { remap as remap$ } from "../../lib/primitives.js";
 import { safeParse } from "../../lib/schemas.js";
+import * as discriminatedUnionTypes from "../../types/discriminatedUnion.js";
+import { discriminatedUnion } from "../../types/discriminatedUnion.js";
 import { ClosedEnum } from "../../types/enums.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
@@ -23,7 +25,8 @@ import {
 
 export type OutputMessageWithIdContent =
   | (OutputText & { type: "output_text" })
-  | (Refusal & { type: "refusal" });
+  | (Refusal & { type: "refusal" })
+  | discriminatedUnionTypes.Unknown<"type">;
 
 /**
  * The processing status of this message (in_progress, completed, or incomplete).
@@ -74,7 +77,9 @@ export type OutputMessageWithId = {
    * Array of content items in the assistant's response. Can contain text outputs or refusal messages.
    */
   content: Array<
-    (OutputText & { type: "output_text" }) | (Refusal & { type: "refusal" })
+    | (OutputText & { type: "output_text" })
+    | (Refusal & { type: "refusal" })
+    | discriminatedUnionTypes.Unknown<"type">
   >;
   /**
    * The processing status of this message (in_progress, completed, or incomplete).
@@ -99,10 +104,12 @@ export const OutputMessageWithIdContent$inboundSchema: z.ZodType<
   OutputMessageWithIdContent,
   z.ZodTypeDef,
   unknown
-> = z.union([
-  OutputText$inboundSchema.and(z.object({ type: z.literal("output_text") })),
-  Refusal$inboundSchema.and(z.object({ type: z.literal("refusal") })),
-]);
+> = discriminatedUnion("type", {
+  output_text: OutputText$inboundSchema.and(
+    z.object({ type: z.literal("output_text") }),
+  ),
+  refusal: Refusal$inboundSchema.and(z.object({ type: z.literal("refusal") })),
+});
 /** @internal */
 export type OutputMessageWithIdContent$Outbound =
   | (OutputText$Outbound & { type: "output_text" })
@@ -163,12 +170,14 @@ export const OutputMessageWithId$inboundSchema: z.ZodType<
   id: z.string(),
   role: z.literal("assistant"),
   content: z.array(
-    z.union([
-      OutputText$inboundSchema.and(
+    discriminatedUnion("type", {
+      output_text: OutputText$inboundSchema.and(
         z.object({ type: z.literal("output_text") }),
       ),
-      Refusal$inboundSchema.and(z.object({ type: z.literal("refusal") })),
-    ]),
+      refusal: Refusal$inboundSchema.and(
+        z.object({ type: z.literal("refusal") }),
+      ),
+    }),
   ),
   status: OutputMessageWithIdStatus$inboundSchema,
   phase: OutputMessageWithIdPhase$inboundSchema.optional(),

package/src/models/components/outputtext.ts

@@ -4,6 +4,8 @@
 
 import * as z from "zod/v3";
 import { safeParse } from "../../lib/schemas.js";
+import * as discriminatedUnionTypes from "../../types/discriminatedUnion.js";
+import { discriminatedUnion } from "../../types/discriminatedUnion.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 import {
@@ -41,7 +43,8 @@ export type Annotations =
   | FileCitation
   | URLCitation
   | ContainerFileCitation
-  | FilePath;
+  | FilePath
+  | discriminatedUnionTypes.Unknown<"type">;
 
 /**
  * Text content output from the model with optional annotations for citations and file references.
@@ -59,7 +62,13 @@ export type OutputText = {
    * Array of annotations providing citations and references for the text. Used to attribute sources and reference files mentioned in the output.
    */
   annotations?:
-    | Array<FileCitation | URLCitation | ContainerFileCitation | FilePath>
+    | Array<
+      | FileCitation
+      | URLCitation
+      | ContainerFileCitation
+      | FilePath
+      | discriminatedUnionTypes.Unknown<"type">
+    >
     | undefined;
   logprobs?: Array<LogProb> | undefined;
 };
@@ -69,12 +78,12 @@ export const Annotations$inboundSchema: z.ZodType<
   Annotations,
   z.ZodTypeDef,
   unknown
-> = z.union([
-  FileCitation$inboundSchema,
-  URLCitation$inboundSchema,
-  ContainerFileCitation$inboundSchema,
-  FilePath$inboundSchema,
-]);
+> = discriminatedUnion("type", {
+  file_citation: FileCitation$inboundSchema,
+  url_citation: URLCitation$inboundSchema,
+  container_file_citation: ContainerFileCitation$inboundSchema,
+  file_path: FilePath$inboundSchema,
+});
 /** @internal */
 export type Annotations$Outbound =
   | FileCitation$Outbound
@@ -115,14 +124,12 @@ export const OutputText$inboundSchema: z.ZodType<
 > = z.object({
   type: z.literal("output_text"),
   text: z.string(),
-  annotations: z.array(
-    z.union([
-      FileCitation$inboundSchema,
-      URLCitation$inboundSchema,
-      ContainerFileCitation$inboundSchema,
-      FilePath$inboundSchema,
-    ]),
-  ).optional(),
+  annotations: z.array(discriminatedUnion("type", {
+    file_citation: FileCitation$inboundSchema,
+    url_citation: URLCitation$inboundSchema,
+    container_file_citation: ContainerFileCitation$inboundSchema,
+    file_path: FilePath$inboundSchema,
+  })).optional(),
   logprobs: z.array(LogProb$inboundSchema).optional(),
 });
 /** @internal */

package/src/models/components/responsecontentpartaddedevent.ts

@@ -5,6 +5,8 @@
 import * as z from "zod/v3";
 import { remap as remap$ } from "../../lib/primitives.js";
 import { safeParse } from "../../lib/schemas.js";
+import * as discriminatedUnionTypes from "../../types/discriminatedUnion.js";
+import { discriminatedUnion } from "../../types/discriminatedUnion.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 import {
@@ -29,7 +31,8 @@ import {
 export type Part =
   | (OutputText & { type: "output_text" })
   | (Refusal & { type: "refusal" })
-  | (ReasoningText & { type: "reasoning_text" });
+  | (ReasoningText & { type: "reasoning_text" })
+  | discriminatedUnionTypes.Unknown<"type">;
 
 export type ResponseContentPartAddedEvent = {
   type: "response.content_part.added";
@@ -40,18 +43,23 @@ export type ResponseContentPartAddedEvent = {
   part:
     | (OutputText & { type: "output_text" })
     | (Refusal & { type: "refusal" })
-    | (ReasoningText & { type: "reasoning_text" });
+    | (ReasoningText & { type: "reasoning_text" })
+    | discriminatedUnionTypes.Unknown<"type">;
 };
 
 /** @internal */
-export const Part$inboundSchema: z.ZodType<Part, z.ZodTypeDef, unknown> = z
-  .union([
-    OutputText$inboundSchema.and(z.object({ type: z.literal("output_text") })),
-    Refusal$inboundSchema.and(z.object({ type: z.literal("refusal") })),
-    ReasoningText$inboundSchema.and(
+export const Part$inboundSchema: z.ZodType<Part, z.ZodTypeDef, unknown> =
+  discriminatedUnion("type", {
+    output_text: OutputText$inboundSchema.and(
+      z.object({ type: z.literal("output_text") }),
+    ),
+    refusal: Refusal$inboundSchema.and(
+      z.object({ type: z.literal("refusal") }),
+    ),
+    reasoning_text: ReasoningText$inboundSchema.and(
       z.object({ type: z.literal("reasoning_text") }),
     ),
-  ]);
+  });
 /** @internal */
 export type Part$Outbound =
   | (OutputText$Outbound & { type: "output_text" })
@@ -92,13 +100,17 @@ export const ResponseContentPartAddedEvent$inboundSchema: z.ZodType<
   item_id: z.string(),
   output_index: z.number().int(),
   content_index: z.number().int(),
-  part: z.union([
-    OutputText$inboundSchema.and(z.object({ type: z.literal("output_text") })),
-    Refusal$inboundSchema.and(z.object({ type: z.literal("refusal") })),
-    ReasoningText$inboundSchema.and(
+  part: discriminatedUnion("type", {
+    output_text: OutputText$inboundSchema.and(
+      z.object({ type: z.literal("output_text") }),
+    ),
+    refusal: Refusal$inboundSchema.and(
+      z.object({ type: z.literal("refusal") }),
+    ),
+    reasoning_text: ReasoningText$inboundSchema.and(
       z.object({ type: z.literal("reasoning_text") }),
     ),
-  ]),
+  }),
 }).transform((v) => {
   return remap$(v, {
     "sequence_number": "sequenceNumber",