@orq-ai/node 4.1.0-rc.1 → 4.1.0-rc.13

package/src/models/operations/invokeagent.ts

@@ -7,6 +7,7 @@ import { remap as remap$ } from "../../lib/primitives.js";
 import { safeParse } from "../../lib/schemas.js";
 import { ClosedEnum } from "../../types/enums.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
+import * as components from "../components/index.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 
 /**
@@ -42,101 +43,13 @@ export type InvokeAgentRole =
   | InvokeAgentRoleUserMessage
   | InvokeAgentRoleToolMessage;
 
-export const InvokeAgentPublicMessagePartAgentsRequestKind = {
-  ToolResult: "tool_result",
-} as const;
-export type InvokeAgentPublicMessagePartAgentsRequestKind = ClosedEnum<
-  typeof InvokeAgentPublicMessagePartAgentsRequestKind
->;
-
-/**
- * Tool execution result part. Use this ONLY when providing results for a pending tool call from the agent. The tool_call_id must match the ID from the agent's tool call request.
- */
-export type PublicMessagePartToolResultPart = {
-  kind: InvokeAgentPublicMessagePartAgentsRequestKind;
-  toolCallId: string;
-  result?: any | undefined;
-  metadata?: { [k: string]: any } | undefined;
-};
-
-export const InvokeAgentPublicMessagePartAgentsKind = {
-  File: "file",
-} as const;
-export type InvokeAgentPublicMessagePartAgentsKind = ClosedEnum<
-  typeof InvokeAgentPublicMessagePartAgentsKind
->;
-
-/**
- * File in URI format. Check in the model's documentation for the supported mime types for the URI format
- */
-export type FileFileInURIFormat = {
-  /**
-   * URL for the File content
-   */
-  uri: string;
-  /**
-   * Optional mimeType for the file
-   */
-  mimeType?: string | undefined;
-  /**
-   * Optional name for the file
-   */
-  name?: string | undefined;
-};
-
-/**
- * Binary in base64 format. Check in the model's documentation for the supported mime types for the binary format.
- */
-export type FileBinaryFormat = {
-  /**
-   * base64 encoded content of the file
-   */
-  bytes: string;
-  /**
-   * Optional mimeType for the file
-   */
-  mimeType?: string | undefined;
-  /**
-   * Optional name for the file
-   */
-  name?: string | undefined;
-};
-
-export type InvokeAgentPublicMessagePartFile =
-  | FileBinaryFormat
-  | FileFileInURIFormat;
-
-/**
- * File attachment part. Use this to send files (images, documents, etc.) to the agent for processing.
- */
-export type PublicMessagePartFilePart = {
-  kind: InvokeAgentPublicMessagePartAgentsKind;
-  file: FileBinaryFormat | FileFileInURIFormat;
-  metadata?: { [k: string]: any } | undefined;
-};
-
-export const InvokeAgentPublicMessagePartKind = {
-  Text: "text",
-} as const;
-export type InvokeAgentPublicMessagePartKind = ClosedEnum<
-  typeof InvokeAgentPublicMessagePartKind
->;
-
-/**
- * Text content part. Use this to send text messages to the agent.
- */
-export type PublicMessagePartTextPart = {
-  kind: InvokeAgentPublicMessagePartKind;
-  text: string;
-};
-
 /**
  * Message part that can be provided by users. Use "text" for regular messages, "file" for attachments, or "tool_result" when responding to tool call requests.
  */
 export type InvokeAgentPublicMessagePart =
-  | PublicMessagePartTextPart
-  | PublicMessagePartFilePart
-  | PublicMessagePartToolResultPart;
+  | components.TextPart
+  | components.FilePart
+  | components.ToolResultPart;
 
 /**
  * The A2A message to send to the agent (user input or tool results)
@@ -154,9 +67,7 @@ export type InvokeAgentA2AMessage = {
    * A2A message parts (text, file, or tool_result only)
    */
   parts: Array<
-    | PublicMessagePartTextPart
-    | PublicMessagePartFilePart
-    | PublicMessagePartToolResultPart
+    components.TextPart | components.FilePart | components.ToolResultPart
   >;
 };
 
@@ -256,13 +167,13 @@ export type InvokeAgentRequest = {
 /**
  * A2A entity type identifier
  */
-export const InvokeAgentKind = {
+export const Kind = {
   Task: "task",
 } as const;
 /**
  * A2A entity type identifier
  */
-export type InvokeAgentKind = ClosedEnum<typeof InvokeAgentKind>;
+export type Kind = ClosedEnum<typeof Kind>;
 
 /**
  * Current state of the agent task execution. Values: submitted (queued), working (executing), input-required (awaiting user input), completed (finished successfully), failed (error occurred). Note: auth-required, canceled, and rejected statuses are defined for A2A protocol compatibility but are not currently supported in task execution.
@@ -282,10 +193,10 @@ export const TaskState = {
  */
 export type TaskState = ClosedEnum<typeof TaskState>;
 
-export const InvokeAgentAgentsKind = {
+export const InvokeAgentKind = {
   Message: "message",
 } as const;
-export type InvokeAgentAgentsKind = ClosedEnum<typeof InvokeAgentAgentsKind>;
+export type InvokeAgentKind = ClosedEnum<typeof InvokeAgentKind>;
 
 /**
  * Role of the message sender in the A2A protocol. Values: user (end user), agent (AI agent), tool (tool execution result), system (system instructions/prompts).
@@ -301,147 +212,29 @@ export const ExtendedMessageRole = {
  */
 export type ExtendedMessageRole = ClosedEnum<typeof ExtendedMessageRole>;
 
-export const InvokeAgentPartsAgentsResponse200Kind = {
-  ToolResult: "tool_result",
-} as const;
-export type InvokeAgentPartsAgentsResponse200Kind = ClosedEnum<
-  typeof InvokeAgentPartsAgentsResponse200Kind
->;
-
-/**
- * The result of a tool execution. Contains the tool call ID for correlation and the result data from the tool invocation.
- */
-export type PartsToolResultPart = {
-  kind: InvokeAgentPartsAgentsResponse200Kind;
-  toolCallId: string;
-  result?: any | undefined;
-  metadata?: { [k: string]: any } | undefined;
-};
-
-export const InvokeAgentPartsAgentsResponseKind = {
-  ToolCall: "tool_call",
-} as const;
-export type InvokeAgentPartsAgentsResponseKind = ClosedEnum<
-  typeof InvokeAgentPartsAgentsResponseKind
->;
-
-/**
- * A tool invocation request from an agent. Contains the tool name, unique call ID, and arguments for the tool execution.
- */
-export type ToolCallPart = {
-  kind: InvokeAgentPartsAgentsResponseKind;
-  toolName: string;
-  toolCallId: string;
-  arguments: { [k: string]: any };
-  metadata?: { [k: string]: any } | undefined;
-};
-
-export const InvokeAgentPartsAgentsKind = {
-  File: "file",
-} as const;
-export type InvokeAgentPartsAgentsKind = ClosedEnum<
-  typeof InvokeAgentPartsAgentsKind
->;
-
-/**
- * File in URI format. Check in the model's documentation for the supported mime types for the URI format
- */
-export type InvokeAgentFileFileInURIFormat = {
-  /**
-   * URL for the File content
-   */
-  uri: string;
-  /**
-   * Optional mimeType for the file
-   */
-  mimeType?: string | undefined;
-  /**
-   * Optional name for the file
-   */
-  name?: string | undefined;
-};
-
-/**
- * Binary in base64 format. Check in the model's documentation for the supported mime types for the binary format.
- */
-export type InvokeAgentFileBinaryFormat = {
-  /**
-   * base64 encoded content of the file
-   */
-  bytes: string;
-  /**
-   * Optional mimeType for the file
-   */
-  mimeType?: string | undefined;
-  /**
-   * Optional name for the file
-   */
-  name?: string | undefined;
-};
-
-export type PartsFile =
-  | InvokeAgentFileBinaryFormat
-  | InvokeAgentFileFileInURIFormat;
-
-/**
- * A file content part that can contain either base64-encoded bytes or a URI reference. Used for images, documents, and other binary content in agent communications.
- */
-export type PartsFilePart = {
-  kind: InvokeAgentPartsAgentsKind;
-  file: InvokeAgentFileBinaryFormat | InvokeAgentFileFileInURIFormat;
-  metadata?: { [k: string]: any } | undefined;
-};
-
-export const InvokeAgentPartsKind = {
-  Data: "data",
-} as const;
-export type InvokeAgentPartsKind = ClosedEnum<typeof InvokeAgentPartsKind>;
-
-/**
- * A structured data part containing JSON-serializable key-value pairs. Used for passing structured information between agents and tools.
- */
-export type DataPart = {
-  kind: InvokeAgentPartsKind;
-  data: { [k: string]: any };
-  metadata?: { [k: string]: any } | undefined;
-};
-
-export const PartsKind = {
-  Text: "text",
-} as const;
-export type PartsKind = ClosedEnum<typeof PartsKind>;
-
-/**
- * A text content part containing plain text or markdown. Used for agent messages, user input, and text-based responses.
- */
-export type PartsTextPart = {
-  kind: PartsKind;
-  text: string;
-};
-
-export type Parts =
-  | ToolCallPart
-  | PartsTextPart
-  | DataPart
-  | PartsFilePart
-  | PartsToolResultPart;
+export type InvokeAgentParts =
+  | (components.ToolCallPart & { kind: "tool_call" })
+  | (components.TextPart & { kind: "text" })
+  | (components.DataPart & { kind: "data" })
+  | (components.FilePart & { kind: "file" })
+  | (components.ToolResultPart & { kind: "tool_result" });
 
 /**
  * Optional A2A message providing additional context about the current status
  */
 export type TaskStatusMessage = {
-  kind: InvokeAgentAgentsKind;
+  kind: InvokeAgentKind;
   messageId: string;
   /**
    * Role of the message sender in the A2A protocol. Values: user (end user), agent (AI agent), tool (tool execution result), system (system instructions/prompts).
    */
   role: ExtendedMessageRole;
   parts: Array<
-    | ToolCallPart
-    | PartsTextPart
-    | DataPart
-    | PartsFilePart
-    | PartsToolResultPart
+    | (components.ToolCallPart & { kind: "tool_call" })
+    | (components.TextPart & { kind: "text" })
+    | (components.DataPart & { kind: "data" })
+    | (components.FilePart & { kind: "file" })
+    | (components.ToolResultPart & { kind: "tool_result" })
   >;
 };
 
@@ -478,7 +271,7 @@ export type InvokeAgentA2ATaskResponse = {
   /**
    * A2A entity type identifier
    */
-  kind: InvokeAgentKind;
+  kind: Kind;
   /**
    * Current task status information
    */
@@ -523,336 +316,24 @@ export type InvokeAgentRole$Outbound = string | string;
 export const InvokeAgentRole$outboundSchema: z.ZodType<
   InvokeAgentRole$Outbound,
   z.ZodTypeDef,
-  InvokeAgentRole
-> = z.union([
-  InvokeAgentRoleUserMessage$outboundSchema,
-  InvokeAgentRoleToolMessage$outboundSchema,
-]);
-
-export function invokeAgentRoleToJSON(
-  invokeAgentRole: InvokeAgentRole,
-): string {
-  return JSON.stringify(InvokeAgentRole$outboundSchema.parse(invokeAgentRole));
-}
-export function invokeAgentRoleFromJSON(
-  jsonString: string,
-): SafeParseResult<InvokeAgentRole, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => InvokeAgentRole$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'InvokeAgentRole' from JSON`,
-  );
-}
-
-/** @internal */
-export const InvokeAgentPublicMessagePartAgentsRequestKind$inboundSchema:
-  z.ZodNativeEnum<typeof InvokeAgentPublicMessagePartAgentsRequestKind> = z
-    .nativeEnum(InvokeAgentPublicMessagePartAgentsRequestKind);
-/** @internal */
-export const InvokeAgentPublicMessagePartAgentsRequestKind$outboundSchema:
-  z.ZodNativeEnum<typeof InvokeAgentPublicMessagePartAgentsRequestKind> =
-    InvokeAgentPublicMessagePartAgentsRequestKind$inboundSchema;
-
-/** @internal */
-export const PublicMessagePartToolResultPart$inboundSchema: z.ZodType<
-  PublicMessagePartToolResultPart,
-  z.ZodTypeDef,
-  unknown
-> = z.object({
-  kind: InvokeAgentPublicMessagePartAgentsRequestKind$inboundSchema,
-  tool_call_id: z.string(),
-  result: z.any().optional(),
-  metadata: z.record(z.any()).optional(),
-}).transform((v) => {
-  return remap$(v, {
-    "tool_call_id": "toolCallId",
-  });
-});
-/** @internal */
-export type PublicMessagePartToolResultPart$Outbound = {
-  kind: string;
-  tool_call_id: string;
-  result?: any | undefined;
-  metadata?: { [k: string]: any } | undefined;
-};
-
-/** @internal */
-export const PublicMessagePartToolResultPart$outboundSchema: z.ZodType<
-  PublicMessagePartToolResultPart$Outbound,
-  z.ZodTypeDef,
-  PublicMessagePartToolResultPart
-> = z.object({
-  kind: InvokeAgentPublicMessagePartAgentsRequestKind$outboundSchema,
-  toolCallId: z.string(),
-  result: z.any().optional(),
-  metadata: z.record(z.any()).optional(),
-}).transform((v) => {
-  return remap$(v, {
-    toolCallId: "tool_call_id",
-  });
-});
-
-export function publicMessagePartToolResultPartToJSON(
-  publicMessagePartToolResultPart: PublicMessagePartToolResultPart,
-): string {
-  return JSON.stringify(
-    PublicMessagePartToolResultPart$outboundSchema.parse(
-      publicMessagePartToolResultPart,
-    ),
-  );
-}
-export function publicMessagePartToolResultPartFromJSON(
-  jsonString: string,
-): SafeParseResult<PublicMessagePartToolResultPart, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => PublicMessagePartToolResultPart$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'PublicMessagePartToolResultPart' from JSON`,
-  );
-}
-
-/** @internal */
-export const InvokeAgentPublicMessagePartAgentsKind$inboundSchema:
-  z.ZodNativeEnum<typeof InvokeAgentPublicMessagePartAgentsKind> = z.nativeEnum(
-    InvokeAgentPublicMessagePartAgentsKind,
-  );
-/** @internal */
-export const InvokeAgentPublicMessagePartAgentsKind$outboundSchema:
-  z.ZodNativeEnum<typeof InvokeAgentPublicMessagePartAgentsKind> =
-    InvokeAgentPublicMessagePartAgentsKind$inboundSchema;
-
-/** @internal */
-export const FileFileInURIFormat$inboundSchema: z.ZodType<
-  FileFileInURIFormat,
-  z.ZodTypeDef,
-  unknown
-> = z.object({
-  uri: z.string(),
-  mimeType: z.string().optional(),
-  name: z.string().optional(),
-});
-/** @internal */
-export type FileFileInURIFormat$Outbound = {
-  uri: string;
-  mimeType?: string | undefined;
-  name?: string | undefined;
-};
-
-/** @internal */
-export const FileFileInURIFormat$outboundSchema: z.ZodType<
-  FileFileInURIFormat$Outbound,
-  z.ZodTypeDef,
-  FileFileInURIFormat
-> = z.object({
-  uri: z.string(),
-  mimeType: z.string().optional(),
-  name: z.string().optional(),
-});
-
-export function fileFileInURIFormatToJSON(
-  fileFileInURIFormat: FileFileInURIFormat,
-): string {
-  return JSON.stringify(
-    FileFileInURIFormat$outboundSchema.parse(fileFileInURIFormat),
-  );
-}
-export function fileFileInURIFormatFromJSON(
-  jsonString: string,
-): SafeParseResult<FileFileInURIFormat, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => FileFileInURIFormat$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'FileFileInURIFormat' from JSON`,
-  );
-}
-
-/** @internal */
-export const FileBinaryFormat$inboundSchema: z.ZodType<
-  FileBinaryFormat,
-  z.ZodTypeDef,
-  unknown
-> = z.object({
-  bytes: z.string(),
-  mimeType: z.string().optional(),
-  name: z.string().optional(),
-});
-/** @internal */
-export type FileBinaryFormat$Outbound = {
-  bytes: string;
-  mimeType?: string | undefined;
-  name?: string | undefined;
-};
-
-/** @internal */
-export const FileBinaryFormat$outboundSchema: z.ZodType<
-  FileBinaryFormat$Outbound,
-  z.ZodTypeDef,
-  FileBinaryFormat
-> = z.object({
-  bytes: z.string(),
-  mimeType: z.string().optional(),
-  name: z.string().optional(),
-});
-
-export function fileBinaryFormatToJSON(
-  fileBinaryFormat: FileBinaryFormat,
-): string {
-  return JSON.stringify(
-    FileBinaryFormat$outboundSchema.parse(fileBinaryFormat),
-  );
-}
-export function fileBinaryFormatFromJSON(
-  jsonString: string,
-): SafeParseResult<FileBinaryFormat, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => FileBinaryFormat$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'FileBinaryFormat' from JSON`,
-  );
-}
-
-/** @internal */
-export const InvokeAgentPublicMessagePartFile$inboundSchema: z.ZodType<
-  InvokeAgentPublicMessagePartFile,
-  z.ZodTypeDef,
-  unknown
-> = z.union([
-  z.lazy(() => FileBinaryFormat$inboundSchema),
-  z.lazy(() => FileFileInURIFormat$inboundSchema),
-]);
-/** @internal */
-export type InvokeAgentPublicMessagePartFile$Outbound =
-  | FileBinaryFormat$Outbound
-  | FileFileInURIFormat$Outbound;
-
-/** @internal */
-export const InvokeAgentPublicMessagePartFile$outboundSchema: z.ZodType<
-  InvokeAgentPublicMessagePartFile$Outbound,
-  z.ZodTypeDef,
-  InvokeAgentPublicMessagePartFile
-> = z.union([
-  z.lazy(() => FileBinaryFormat$outboundSchema),
-  z.lazy(() => FileFileInURIFormat$outboundSchema),
-]);
-
-export function invokeAgentPublicMessagePartFileToJSON(
-  invokeAgentPublicMessagePartFile: InvokeAgentPublicMessagePartFile,
-): string {
-  return JSON.stringify(
-    InvokeAgentPublicMessagePartFile$outboundSchema.parse(
-      invokeAgentPublicMessagePartFile,
-    ),
-  );
-}
-export function invokeAgentPublicMessagePartFileFromJSON(
-  jsonString: string,
-): SafeParseResult<InvokeAgentPublicMessagePartFile, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => InvokeAgentPublicMessagePartFile$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'InvokeAgentPublicMessagePartFile' from JSON`,
-  );
-}
-
-/** @internal */
-export const PublicMessagePartFilePart$inboundSchema: z.ZodType<
-  PublicMessagePartFilePart,
-  z.ZodTypeDef,
-  unknown
-> = z.object({
-  kind: InvokeAgentPublicMessagePartAgentsKind$inboundSchema,
-  file: z.union([
-    z.lazy(() => FileBinaryFormat$inboundSchema),
-    z.lazy(() => FileFileInURIFormat$inboundSchema),
-  ]),
-  metadata: z.record(z.any()).optional(),
-});
-/** @internal */
-export type PublicMessagePartFilePart$Outbound = {
-  kind: string;
-  file: FileBinaryFormat$Outbound | FileFileInURIFormat$Outbound;
-  metadata?: { [k: string]: any } | undefined;
-};
-
-/** @internal */
-export const PublicMessagePartFilePart$outboundSchema: z.ZodType<
-  PublicMessagePartFilePart$Outbound,
-  z.ZodTypeDef,
-  PublicMessagePartFilePart
-> = z.object({
-  kind: InvokeAgentPublicMessagePartAgentsKind$outboundSchema,
-  file: z.union([
-    z.lazy(() => FileBinaryFormat$outboundSchema),
-    z.lazy(() => FileFileInURIFormat$outboundSchema),
-  ]),
-  metadata: z.record(z.any()).optional(),
-});
-
-export function publicMessagePartFilePartToJSON(
-  publicMessagePartFilePart: PublicMessagePartFilePart,
-): string {
-  return JSON.stringify(
-    PublicMessagePartFilePart$outboundSchema.parse(publicMessagePartFilePart),
-  );
-}
-export function publicMessagePartFilePartFromJSON(
-  jsonString: string,
-): SafeParseResult<PublicMessagePartFilePart, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => PublicMessagePartFilePart$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'PublicMessagePartFilePart' from JSON`,
-  );
-}
-
-/** @internal */
-export const InvokeAgentPublicMessagePartKind$inboundSchema: z.ZodNativeEnum<
-  typeof InvokeAgentPublicMessagePartKind
-> = z.nativeEnum(InvokeAgentPublicMessagePartKind);
-/** @internal */
-export const InvokeAgentPublicMessagePartKind$outboundSchema: z.ZodNativeEnum<
-  typeof InvokeAgentPublicMessagePartKind
-> = InvokeAgentPublicMessagePartKind$inboundSchema;
-
-/** @internal */
-export const PublicMessagePartTextPart$inboundSchema: z.ZodType<
-  PublicMessagePartTextPart,
-  z.ZodTypeDef,
-  unknown
-> = z.object({
-  kind: InvokeAgentPublicMessagePartKind$inboundSchema,
-  text: z.string(),
-});
-/** @internal */
-export type PublicMessagePartTextPart$Outbound = {
-  kind: string;
-  text: string;
-};
-
-/** @internal */
-export const PublicMessagePartTextPart$outboundSchema: z.ZodType<
-  PublicMessagePartTextPart$Outbound,
-  z.ZodTypeDef,
-  PublicMessagePartTextPart
-> = z.object({
-  kind: InvokeAgentPublicMessagePartKind$outboundSchema,
-  text: z.string(),
-});
+  InvokeAgentRole
+> = z.union([
+  InvokeAgentRoleUserMessage$outboundSchema,
+  InvokeAgentRoleToolMessage$outboundSchema,
+]);
 
-export function publicMessagePartTextPartToJSON(
-  publicMessagePartTextPart: PublicMessagePartTextPart,
+export function invokeAgentRoleToJSON(
+  invokeAgentRole: InvokeAgentRole,
 ): string {
-  return JSON.stringify(
-    PublicMessagePartTextPart$outboundSchema.parse(publicMessagePartTextPart),
-  );
+  return JSON.stringify(InvokeAgentRole$outboundSchema.parse(invokeAgentRole));
 }
-export function publicMessagePartTextPartFromJSON(
+export function invokeAgentRoleFromJSON(
   jsonString: string,
-): SafeParseResult<PublicMessagePartTextPart, SDKValidationError> {
+): SafeParseResult<InvokeAgentRole, SDKValidationError> {
   return safeParse(
     jsonString,
-    (x) => PublicMessagePartTextPart$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'PublicMessagePartTextPart' from JSON`,
+    (x) => InvokeAgentRole$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'InvokeAgentRole' from JSON`,
   );
 }
 
@@ -862,15 +343,15 @@ export const InvokeAgentPublicMessagePart$inboundSchema: z.ZodType<
   z.ZodTypeDef,
   unknown
 > = z.union([
-  z.lazy(() => PublicMessagePartTextPart$inboundSchema),
-  z.lazy(() => PublicMessagePartFilePart$inboundSchema),
-  z.lazy(() => PublicMessagePartToolResultPart$inboundSchema),
+  components.TextPart$inboundSchema,
+  components.FilePart$inboundSchema,
+  components.ToolResultPart$inboundSchema,
 ]);
 /** @internal */
 export type InvokeAgentPublicMessagePart$Outbound =
-  | PublicMessagePartTextPart$Outbound
-  | PublicMessagePartFilePart$Outbound
-  | PublicMessagePartToolResultPart$Outbound;
+  | components.TextPart$Outbound
+  | components.FilePart$Outbound
+  | components.ToolResultPart$Outbound;
 
 /** @internal */
 export const InvokeAgentPublicMessagePart$outboundSchema: z.ZodType<
@@ -878,9 +359,9 @@ export const InvokeAgentPublicMessagePart$outboundSchema: z.ZodType<
   z.ZodTypeDef,
   InvokeAgentPublicMessagePart
 > = z.union([
-  z.lazy(() => PublicMessagePartTextPart$outboundSchema),
-  z.lazy(() => PublicMessagePartFilePart$outboundSchema),
-  z.lazy(() => PublicMessagePartToolResultPart$outboundSchema),
+  components.TextPart$outboundSchema,
+  components.FilePart$outboundSchema,
+  components.ToolResultPart$outboundSchema,
 ]);
 
 export function invokeAgentPublicMessagePartToJSON(
@@ -915,9 +396,9 @@ export const InvokeAgentA2AMessage$inboundSchema: z.ZodType<
   ]),
   parts: z.array(
     z.union([
-      z.lazy(() => PublicMessagePartTextPart$inboundSchema),
-      z.lazy(() => PublicMessagePartFilePart$inboundSchema),
-      z.lazy(() => PublicMessagePartToolResultPart$inboundSchema),
+      components.TextPart$inboundSchema,
+      components.FilePart$inboundSchema,
+      components.ToolResultPart$inboundSchema,
     ]),
   ),
 });
@@ -926,9 +407,9 @@ export type InvokeAgentA2AMessage$Outbound = {
   messageId?: string | undefined;
   role: string | string;
   parts: Array<
-    | PublicMessagePartTextPart$Outbound
-    | PublicMessagePartFilePart$Outbound
-    | PublicMessagePartToolResultPart$Outbound
+    | components.TextPart$Outbound
+    | components.FilePart$Outbound
+    | components.ToolResultPart$Outbound
   >;
 };
 
@@ -945,9 +426,9 @@ export const InvokeAgentA2AMessage$outboundSchema: z.ZodType<
   ]),
   parts: z.array(
     z.union([
-      z.lazy(() => PublicMessagePartTextPart$outboundSchema),
-      z.lazy(() => PublicMessagePartFilePart$outboundSchema),
-      z.lazy(() => PublicMessagePartToolResultPart$outboundSchema),
+      components.TextPart$outboundSchema,
+      components.FilePart$outboundSchema,
+      components.ToolResultPart$outboundSchema,
     ]),
   ),
 });
@@ -1238,13 +719,12 @@ export function invokeAgentRequestFromJSON(
 }
 
 /** @internal */
-export const InvokeAgentKind$inboundSchema: z.ZodNativeEnum<
-  typeof InvokeAgentKind
-> = z.nativeEnum(InvokeAgentKind);
+export const Kind$inboundSchema: z.ZodNativeEnum<typeof Kind> = z.nativeEnum(
+  Kind,
+);
 /** @internal */
-export const InvokeAgentKind$outboundSchema: z.ZodNativeEnum<
-  typeof InvokeAgentKind
-> = InvokeAgentKind$inboundSchema;
+export const Kind$outboundSchema: z.ZodNativeEnum<typeof Kind> =
+  Kind$inboundSchema;
 
 /** @internal */
 export const TaskState$inboundSchema: z.ZodNativeEnum<typeof TaskState> = z
@@ -1254,13 +734,13 @@ export const TaskState$outboundSchema: z.ZodNativeEnum<typeof TaskState> =
   TaskState$inboundSchema;
 
 /** @internal */
-export const InvokeAgentAgentsKind$inboundSchema: z.ZodNativeEnum<
-  typeof InvokeAgentAgentsKind
-> = z.nativeEnum(InvokeAgentAgentsKind);
+export const InvokeAgentKind$inboundSchema: z.ZodNativeEnum<
+  typeof InvokeAgentKind
+> = z.nativeEnum(InvokeAgentKind);
 /** @internal */
-export const InvokeAgentAgentsKind$outboundSchema: z.ZodNativeEnum<
-  typeof InvokeAgentAgentsKind
-> = InvokeAgentAgentsKind$inboundSchema;
+export const InvokeAgentKind$outboundSchema: z.ZodNativeEnum<
+  typeof InvokeAgentKind
+> = InvokeAgentKind$inboundSchema;
 
 /** @internal */
 export const ExtendedMessageRole$inboundSchema: z.ZodNativeEnum<
@@ -1272,461 +752,80 @@ export const ExtendedMessageRole$outboundSchema: z.ZodNativeEnum<
 > = ExtendedMessageRole$inboundSchema;
 
 /** @internal */
-export const InvokeAgentPartsAgentsResponse200Kind$inboundSchema:
-  z.ZodNativeEnum<typeof InvokeAgentPartsAgentsResponse200Kind> = z.nativeEnum(
-    InvokeAgentPartsAgentsResponse200Kind,
-  );
-/** @internal */
-export const InvokeAgentPartsAgentsResponse200Kind$outboundSchema:
-  z.ZodNativeEnum<typeof InvokeAgentPartsAgentsResponse200Kind> =
-    InvokeAgentPartsAgentsResponse200Kind$inboundSchema;
-
-/** @internal */
-export const PartsToolResultPart$inboundSchema: z.ZodType<
-  PartsToolResultPart,
-  z.ZodTypeDef,
-  unknown
-> = z.object({
-  kind: InvokeAgentPartsAgentsResponse200Kind$inboundSchema,
-  tool_call_id: z.string(),
-  result: z.any().optional(),
-  metadata: z.record(z.any()).optional(),
-}).transform((v) => {
-  return remap$(v, {
-    "tool_call_id": "toolCallId",
-  });
-});
-/** @internal */
-export type PartsToolResultPart$Outbound = {
-  kind: string;
-  tool_call_id: string;
-  result?: any | undefined;
-  metadata?: { [k: string]: any } | undefined;
-};
-
-/** @internal */
-export const PartsToolResultPart$outboundSchema: z.ZodType<
-  PartsToolResultPart$Outbound,
-  z.ZodTypeDef,
-  PartsToolResultPart
-> = z.object({
-  kind: InvokeAgentPartsAgentsResponse200Kind$outboundSchema,
-  toolCallId: z.string(),
-  result: z.any().optional(),
-  metadata: z.record(z.any()).optional(),
-}).transform((v) => {
-  return remap$(v, {
-    toolCallId: "tool_call_id",
-  });
-});
-
-export function partsToolResultPartToJSON(
-  partsToolResultPart: PartsToolResultPart,
-): string {
-  return JSON.stringify(
-    PartsToolResultPart$outboundSchema.parse(partsToolResultPart),
-  );
-}
-export function partsToolResultPartFromJSON(
-  jsonString: string,
-): SafeParseResult<PartsToolResultPart, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => PartsToolResultPart$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'PartsToolResultPart' from JSON`,
-  );
-}
-
-/** @internal */
-export const InvokeAgentPartsAgentsResponseKind$inboundSchema: z.ZodNativeEnum<
-  typeof InvokeAgentPartsAgentsResponseKind
-> = z.nativeEnum(InvokeAgentPartsAgentsResponseKind);
-/** @internal */
-export const InvokeAgentPartsAgentsResponseKind$outboundSchema: z.ZodNativeEnum<
-  typeof InvokeAgentPartsAgentsResponseKind
-> = InvokeAgentPartsAgentsResponseKind$inboundSchema;
-
-/** @internal */
-export const ToolCallPart$inboundSchema: z.ZodType<
-  ToolCallPart,
-  z.ZodTypeDef,
-  unknown
-> = z.object({
-  kind: InvokeAgentPartsAgentsResponseKind$inboundSchema,
-  tool_name: z.string(),
-  tool_call_id: z.string(),
-  arguments: z.record(z.any()),
-  metadata: z.record(z.any()).optional(),
-}).transform((v) => {
-  return remap$(v, {
-    "tool_name": "toolName",
-    "tool_call_id": "toolCallId",
-  });
-});
-/** @internal */
-export type ToolCallPart$Outbound = {
-  kind: string;
-  tool_name: string;
-  tool_call_id: string;
-  arguments: { [k: string]: any };
-  metadata?: { [k: string]: any } | undefined;
-};
-
-/** @internal */
-export const ToolCallPart$outboundSchema: z.ZodType<
-  ToolCallPart$Outbound,
-  z.ZodTypeDef,
-  ToolCallPart
-> = z.object({
-  kind: InvokeAgentPartsAgentsResponseKind$outboundSchema,
-  toolName: z.string(),
-  toolCallId: z.string(),
-  arguments: z.record(z.any()),
-  metadata: z.record(z.any()).optional(),
-}).transform((v) => {
-  return remap$(v, {
-    toolName: "tool_name",
-    toolCallId: "tool_call_id",
-  });
-});
-
-export function toolCallPartToJSON(toolCallPart: ToolCallPart): string {
-  return JSON.stringify(ToolCallPart$outboundSchema.parse(toolCallPart));
-}
-export function toolCallPartFromJSON(
-  jsonString: string,
-): SafeParseResult<ToolCallPart, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => ToolCallPart$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'ToolCallPart' from JSON`,
-  );
-}
-
-/** @internal */
-export const InvokeAgentPartsAgentsKind$inboundSchema: z.ZodNativeEnum<
-  typeof InvokeAgentPartsAgentsKind
-> = z.nativeEnum(InvokeAgentPartsAgentsKind);
-/** @internal */
-export const InvokeAgentPartsAgentsKind$outboundSchema: z.ZodNativeEnum<
-  typeof InvokeAgentPartsAgentsKind
-> = InvokeAgentPartsAgentsKind$inboundSchema;
-
-/** @internal */
-export const InvokeAgentFileFileInURIFormat$inboundSchema: z.ZodType<
-  InvokeAgentFileFileInURIFormat,
-  z.ZodTypeDef,
-  unknown
-> = z.object({
-  uri: z.string(),
-  mimeType: z.string().optional(),
-  name: z.string().optional(),
-});
-/** @internal */
-export type InvokeAgentFileFileInURIFormat$Outbound = {
-  uri: string;
-  mimeType?: string | undefined;
-  name?: string | undefined;
-};
-
-/** @internal */
-export const InvokeAgentFileFileInURIFormat$outboundSchema: z.ZodType<
-  InvokeAgentFileFileInURIFormat$Outbound,
-  z.ZodTypeDef,
-  InvokeAgentFileFileInURIFormat
-> = z.object({
-  uri: z.string(),
-  mimeType: z.string().optional(),
-  name: z.string().optional(),
-});
-
-export function invokeAgentFileFileInURIFormatToJSON(
-  invokeAgentFileFileInURIFormat: InvokeAgentFileFileInURIFormat,
-): string {
-  return JSON.stringify(
-    InvokeAgentFileFileInURIFormat$outboundSchema.parse(
-      invokeAgentFileFileInURIFormat,
-    ),
-  );
-}
-export function invokeAgentFileFileInURIFormatFromJSON(
-  jsonString: string,
-): SafeParseResult<InvokeAgentFileFileInURIFormat, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => InvokeAgentFileFileInURIFormat$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'InvokeAgentFileFileInURIFormat' from JSON`,
-  );
-}
-
-/** @internal */
-export const InvokeAgentFileBinaryFormat$inboundSchema: z.ZodType<
-  InvokeAgentFileBinaryFormat,
-  z.ZodTypeDef,
-  unknown
-> = z.object({
-  bytes: z.string(),
-  mimeType: z.string().optional(),
-  name: z.string().optional(),
-});
-/** @internal */
-export type InvokeAgentFileBinaryFormat$Outbound = {
-  bytes: string;
-  mimeType?: string | undefined;
-  name?: string | undefined;
-};
-
-/** @internal */
-export const InvokeAgentFileBinaryFormat$outboundSchema: z.ZodType<
-  InvokeAgentFileBinaryFormat$Outbound,
-  z.ZodTypeDef,
-  InvokeAgentFileBinaryFormat
-> = z.object({
-  bytes: z.string(),
-  mimeType: z.string().optional(),
-  name: z.string().optional(),
-});
-
-export function invokeAgentFileBinaryFormatToJSON(
-  invokeAgentFileBinaryFormat: InvokeAgentFileBinaryFormat,
-): string {
-  return JSON.stringify(
-    InvokeAgentFileBinaryFormat$outboundSchema.parse(
-      invokeAgentFileBinaryFormat,
-    ),
-  );
-}
-export function invokeAgentFileBinaryFormatFromJSON(
-  jsonString: string,
-): SafeParseResult<InvokeAgentFileBinaryFormat, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => InvokeAgentFileBinaryFormat$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'InvokeAgentFileBinaryFormat' from JSON`,
-  );
-}
-
-/** @internal */
-export const PartsFile$inboundSchema: z.ZodType<
-  PartsFile,
+export const InvokeAgentParts$inboundSchema: z.ZodType<
+  InvokeAgentParts,
   z.ZodTypeDef,
   unknown
 > = z.union([
-  z.lazy(() => InvokeAgentFileBinaryFormat$inboundSchema),
-  z.lazy(() => InvokeAgentFileFileInURIFormat$inboundSchema),
+  components.ToolCallPart$inboundSchema.and(
+    z.object({ kind: z.literal("tool_call") }).transform((v) => ({
+      kind: v.kind,
+    })),
+  ),
+  components.TextPart$inboundSchema.and(
+    z.object({ kind: z.literal("text") }).transform((v) => ({ kind: v.kind })),
+  ),
+  components.DataPart$inboundSchema.and(
+    z.object({ kind: z.literal("data") }).transform((v) => ({ kind: v.kind })),
+  ),
+  components.FilePart$inboundSchema.and(
+    z.object({ kind: z.literal("file") }).transform((v) => ({ kind: v.kind })),
+  ),
+  components.ToolResultPart$inboundSchema.and(
+    z.object({ kind: z.literal("tool_result") }).transform((v) => ({
+      kind: v.kind,
+    })),
+  ),
 ]);
 /** @internal */
-export type PartsFile$Outbound =
-  | InvokeAgentFileBinaryFormat$Outbound
-  | InvokeAgentFileFileInURIFormat$Outbound;
+export type InvokeAgentParts$Outbound =
+  | (components.ToolCallPart$Outbound & { kind: "tool_call" })
+  | (components.TextPart$Outbound & { kind: "text" })
+  | (components.DataPart$Outbound & { kind: "data" })
+  | (components.FilePart$Outbound & { kind: "file" })
+  | (components.ToolResultPart$Outbound & { kind: "tool_result" });
 
 /** @internal */
-export const PartsFile$outboundSchema: z.ZodType<
-  PartsFile$Outbound,
+export const InvokeAgentParts$outboundSchema: z.ZodType<
+  InvokeAgentParts$Outbound,
   z.ZodTypeDef,
-  PartsFile
+  InvokeAgentParts
 > = z.union([
-  z.lazy(() => InvokeAgentFileBinaryFormat$outboundSchema),
-  z.lazy(() => InvokeAgentFileFileInURIFormat$outboundSchema),
+  components.ToolCallPart$outboundSchema.and(
+    z.object({ kind: z.literal("tool_call") }).transform((v) => ({
+      kind: v.kind,
+    })),
+  ),
+  components.TextPart$outboundSchema.and(
+    z.object({ kind: z.literal("text") }).transform((v) => ({ kind: v.kind })),
+  ),
+  components.DataPart$outboundSchema.and(
+    z.object({ kind: z.literal("data") }).transform((v) => ({ kind: v.kind })),
+  ),
+  components.FilePart$outboundSchema.and(
+    z.object({ kind: z.literal("file") }).transform((v) => ({ kind: v.kind })),
+  ),
+  components.ToolResultPart$outboundSchema.and(
+    z.object({ kind: z.literal("tool_result") }).transform((v) => ({
+      kind: v.kind,
+    })),
+  ),
 ]);
 
-export function partsFileToJSON(partsFile: PartsFile): string {
-  return JSON.stringify(PartsFile$outboundSchema.parse(partsFile));
-}
-export function partsFileFromJSON(
-  jsonString: string,
-): SafeParseResult<PartsFile, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => PartsFile$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'PartsFile' from JSON`,
-  );
-}
-
-/** @internal */
-export const PartsFilePart$inboundSchema: z.ZodType<
-  PartsFilePart,
-  z.ZodTypeDef,
-  unknown
-> = z.object({
-  kind: InvokeAgentPartsAgentsKind$inboundSchema,
-  file: z.union([
-    z.lazy(() => InvokeAgentFileBinaryFormat$inboundSchema),
-    z.lazy(() => InvokeAgentFileFileInURIFormat$inboundSchema),
-  ]),
-  metadata: z.record(z.any()).optional(),
-});
-/** @internal */
-export type PartsFilePart$Outbound = {
-  kind: string;
-  file:
-    | InvokeAgentFileBinaryFormat$Outbound
-    | InvokeAgentFileFileInURIFormat$Outbound;
-  metadata?: { [k: string]: any } | undefined;
-};
-
-/** @internal */
-export const PartsFilePart$outboundSchema: z.ZodType<
-  PartsFilePart$Outbound,
-  z.ZodTypeDef,
-  PartsFilePart
-> = z.object({
-  kind: InvokeAgentPartsAgentsKind$outboundSchema,
-  file: z.union([
-    z.lazy(() => InvokeAgentFileBinaryFormat$outboundSchema),
-    z.lazy(() => InvokeAgentFileFileInURIFormat$outboundSchema),
-  ]),
-  metadata: z.record(z.any()).optional(),
-});
-
-export function partsFilePartToJSON(partsFilePart: PartsFilePart): string {
-  return JSON.stringify(PartsFilePart$outboundSchema.parse(partsFilePart));
-}
-export function partsFilePartFromJSON(
-  jsonString: string,
-): SafeParseResult<PartsFilePart, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => PartsFilePart$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'PartsFilePart' from JSON`,
-  );
-}
-
-/** @internal */
-export const InvokeAgentPartsKind$inboundSchema: z.ZodNativeEnum<
-  typeof InvokeAgentPartsKind
-> = z.nativeEnum(InvokeAgentPartsKind);
-/** @internal */
-export const InvokeAgentPartsKind$outboundSchema: z.ZodNativeEnum<
-  typeof InvokeAgentPartsKind
-> = InvokeAgentPartsKind$inboundSchema;
-
-/** @internal */
-export const DataPart$inboundSchema: z.ZodType<
-  DataPart,
-  z.ZodTypeDef,
-  unknown
-> = z.object({
-  kind: InvokeAgentPartsKind$inboundSchema,
-  data: z.record(z.any()),
-  metadata: z.record(z.any()).optional(),
-});
-/** @internal */
-export type DataPart$Outbound = {
-  kind: string;
-  data: { [k: string]: any };
-  metadata?: { [k: string]: any } | undefined;
-};
-
-/** @internal */
-export const DataPart$outboundSchema: z.ZodType<
-  DataPart$Outbound,
-  z.ZodTypeDef,
-  DataPart
-> = z.object({
-  kind: InvokeAgentPartsKind$outboundSchema,
-  data: z.record(z.any()),
-  metadata: z.record(z.any()).optional(),
-});
-
-export function dataPartToJSON(dataPart: DataPart): string {
-  return JSON.stringify(DataPart$outboundSchema.parse(dataPart));
-}
-export function dataPartFromJSON(
-  jsonString: string,
-): SafeParseResult<DataPart, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => DataPart$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'DataPart' from JSON`,
-  );
-}
-
-/** @internal */
-export const PartsKind$inboundSchema: z.ZodNativeEnum<typeof PartsKind> = z
-  .nativeEnum(PartsKind);
-/** @internal */
-export const PartsKind$outboundSchema: z.ZodNativeEnum<typeof PartsKind> =
-  PartsKind$inboundSchema;
-
-/** @internal */
-export const PartsTextPart$inboundSchema: z.ZodType<
-  PartsTextPart,
-  z.ZodTypeDef,
-  unknown
-> = z.object({
-  kind: PartsKind$inboundSchema,
-  text: z.string(),
-});
-/** @internal */
-export type PartsTextPart$Outbound = {
-  kind: string;
-  text: string;
-};
-
-/** @internal */
-export const PartsTextPart$outboundSchema: z.ZodType<
-  PartsTextPart$Outbound,
-  z.ZodTypeDef,
-  PartsTextPart
-> = z.object({
-  kind: PartsKind$outboundSchema,
-  text: z.string(),
-});
-
-export function partsTextPartToJSON(partsTextPart: PartsTextPart): string {
-  return JSON.stringify(PartsTextPart$outboundSchema.parse(partsTextPart));
-}
-export function partsTextPartFromJSON(
-  jsonString: string,
-): SafeParseResult<PartsTextPart, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => PartsTextPart$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'PartsTextPart' from JSON`,
+export function invokeAgentPartsToJSON(
+  invokeAgentParts: InvokeAgentParts,
+): string {
+  return JSON.stringify(
+    InvokeAgentParts$outboundSchema.parse(invokeAgentParts),
   );
 }
-
-/** @internal */
-export const Parts$inboundSchema: z.ZodType<Parts, z.ZodTypeDef, unknown> = z
-  .union([
-    z.lazy(() => ToolCallPart$inboundSchema),
-    z.lazy(() => PartsTextPart$inboundSchema),
-    z.lazy(() => DataPart$inboundSchema),
-    z.lazy(() => PartsFilePart$inboundSchema),
-    z.lazy(() => PartsToolResultPart$inboundSchema),
-  ]);
-/** @internal */
-export type Parts$Outbound =
-  | ToolCallPart$Outbound
-  | PartsTextPart$Outbound
-  | DataPart$Outbound
-  | PartsFilePart$Outbound
-  | PartsToolResultPart$Outbound;
-
-/** @internal */
-export const Parts$outboundSchema: z.ZodType<
-  Parts$Outbound,
-  z.ZodTypeDef,
-  Parts
-> = z.union([
-  z.lazy(() => ToolCallPart$outboundSchema),
-  z.lazy(() => PartsTextPart$outboundSchema),
-  z.lazy(() => DataPart$outboundSchema),
-  z.lazy(() => PartsFilePart$outboundSchema),
-  z.lazy(() => PartsToolResultPart$outboundSchema),
-]);
-
-export function partsToJSON(parts: Parts): string {
-  return JSON.stringify(Parts$outboundSchema.parse(parts));
-}
-export function partsFromJSON(
+export function invokeAgentPartsFromJSON(
   jsonString: string,
-): SafeParseResult<Parts, SDKValidationError> {
+): SafeParseResult<InvokeAgentParts, SDKValidationError> {
   return safeParse(
     jsonString,
-    (x) => Parts$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'Parts' from JSON`,
+    (x) => InvokeAgentParts$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'InvokeAgentParts' from JSON`,
   );
 }
 
@@ -1736,16 +835,36 @@ export const TaskStatusMessage$inboundSchema: z.ZodType<
   z.ZodTypeDef,
   unknown
 > = z.object({
-  kind: InvokeAgentAgentsKind$inboundSchema,
+  kind: InvokeAgentKind$inboundSchema,
   messageId: z.string(),
   role: ExtendedMessageRole$inboundSchema,
   parts: z.array(
     z.union([
-      z.lazy(() => ToolCallPart$inboundSchema),
-      z.lazy(() => PartsTextPart$inboundSchema),
-      z.lazy(() => DataPart$inboundSchema),
-      z.lazy(() => PartsFilePart$inboundSchema),
-      z.lazy(() => PartsToolResultPart$inboundSchema),
+      components.ToolCallPart$inboundSchema.and(
+        z.object({ kind: z.literal("tool_call") }).transform((v) => ({
+          kind: v.kind,
+        })),
+      ),
+      components.TextPart$inboundSchema.and(
+        z.object({ kind: z.literal("text") }).transform((v) => ({
+          kind: v.kind,
+        })),
+      ),
+      components.DataPart$inboundSchema.and(
+        z.object({ kind: z.literal("data") }).transform((v) => ({
+          kind: v.kind,
+        })),
+      ),
+      components.FilePart$inboundSchema.and(
+        z.object({ kind: z.literal("file") }).transform((v) => ({
+          kind: v.kind,
+        })),
+      ),
+      components.ToolResultPart$inboundSchema.and(
+        z.object({ kind: z.literal("tool_result") }).transform((v) => ({
+          kind: v.kind,
+        })),
+      ),
     ]),
   ),
 });
@@ -1755,11 +874,11 @@ export type TaskStatusMessage$Outbound = {
   messageId: string;
   role: string;
   parts: Array<
-    | ToolCallPart$Outbound
-    | PartsTextPart$Outbound
-    | DataPart$Outbound
-    | PartsFilePart$Outbound
-    | PartsToolResultPart$Outbound
+    | (components.ToolCallPart$Outbound & { kind: "tool_call" })
+    | (components.TextPart$Outbound & { kind: "text" })
+    | (components.DataPart$Outbound & { kind: "data" })
+    | (components.FilePart$Outbound & { kind: "file" })
+    | (components.ToolResultPart$Outbound & { kind: "tool_result" })
   >;
 };
 
@@ -1769,16 +888,36 @@ export const TaskStatusMessage$outboundSchema: z.ZodType<
   z.ZodTypeDef,
   TaskStatusMessage
 > = z.object({
-  kind: InvokeAgentAgentsKind$outboundSchema,
+  kind: InvokeAgentKind$outboundSchema,
   messageId: z.string(),
   role: ExtendedMessageRole$outboundSchema,
   parts: z.array(
     z.union([
-      z.lazy(() => ToolCallPart$outboundSchema),
-      z.lazy(() => PartsTextPart$outboundSchema),
-      z.lazy(() => DataPart$outboundSchema),
-      z.lazy(() => PartsFilePart$outboundSchema),
-      z.lazy(() => PartsToolResultPart$outboundSchema),
+      components.ToolCallPart$outboundSchema.and(
+        z.object({ kind: z.literal("tool_call") }).transform((v) => ({
+          kind: v.kind,
+        })),
+      ),
+      components.TextPart$outboundSchema.and(
+        z.object({ kind: z.literal("text") }).transform((v) => ({
+          kind: v.kind,
+        })),
+      ),
+      components.DataPart$outboundSchema.and(
+        z.object({ kind: z.literal("data") }).transform((v) => ({
+          kind: v.kind,
+        })),
+      ),
+      components.FilePart$outboundSchema.and(
+        z.object({ kind: z.literal("file") }).transform((v) => ({
+          kind: v.kind,
+        })),
+      ),
+      components.ToolResultPart$outboundSchema.and(
+        z.object({ kind: z.literal("tool_result") }).transform((v) => ({
+          kind: v.kind,
+        })),
+      ),
     ]),
   ),
 });
@@ -1849,7 +988,7 @@ export const InvokeAgentA2ATaskResponse$inboundSchema: z.ZodType<
 > = z.object({
   id: z.string(),
   contextId: z.string(),
-  kind: InvokeAgentKind$inboundSchema,
+  kind: Kind$inboundSchema,
   status: z.lazy(() => TaskStatus$inboundSchema),
   metadata: z.record(z.any()).optional(),
 });
@@ -1870,7 +1009,7 @@ export const InvokeAgentA2ATaskResponse$outboundSchema: z.ZodType<
 > = z.object({
   id: z.string(),
   contextId: z.string(),
-  kind: InvokeAgentKind$outboundSchema,
+  kind: Kind$outboundSchema,
   status: z.lazy(() => TaskStatus$outboundSchema),
   metadata: z.record(z.any()).optional(),
 });