veryfront 0.1.150 → 0.1.152

package/esm/src/internal-agents/schema.js

@@ -1,8 +1,6 @@
 import { z } from "zod";
+import { AgUiRuntimeContextItemSchema, AgUiRuntimeContextSchema, AgUiRuntimeInjectedToolSchema, AgUiRuntimeMessageSchema, AgUiRuntimeRequestSchema, AgUiRuntimeRunIdSchema, } from "../agent/runtime-ag-ui-contract.js";
 const AGENT_ID_PATTERN = /^[a-zA-Z0-9_-]+$/;
-const MAX_TOOL_PARAMETERS_BYTES = 16_384;
-const MAX_CONTEXT_ITEM_BYTES = 16_384;
-const MAX_CONTEXT_TOTAL_BYTES = 65_536;
 const MAX_FORWARDED_PROPS_BYTES = 65_536;
 const MAX_TOOL_RESULT_BYTES = 65_536;
 const MAX_RUNTIME_MESSAGES = 100;
@@ -15,37 +13,8 @@ function isWithinJsonSizeLimit(value, maxBytes) {
         return false;
     }
 }
-export const RunIdSchema = z.string().min(1).max(128).regex(AGENT_ID_PATTERN);
+export const RunIdSchema = AgUiRuntimeRunIdSchema;
 export const AgentIdSchema = z.string().min(1).max(128).regex(AGENT_ID_PATTERN);
-export const ClientToolNameSchema = z
-    .string()
-    .min(1)
-    .max(128)
-    .regex(/^[a-zA-Z][a-zA-Z0-9._:-]*$/, "Tool names must start with a letter and use a valid client-tool format");
-export const RuntimeInjectedToolSchema = z.object({
-    name: ClientToolNameSchema,
-    description: z.string().max(1024).optional(),
-    parameters: z.record(z.string(), z.unknown()).optional().refine((value) => value === undefined || isWithinJsonSizeLimit(value, MAX_TOOL_PARAMETERS_BYTES), { message: "Tool parameters must be less than 16 KB" }),
-});
-export const RuntimeContextItemSchema = z.discriminatedUnion("type", [
-    z.object({
-        type: z.literal("text"),
-        title: z.string().max(256).optional(),
-        text: z.string().max(MAX_CONTEXT_ITEM_BYTES),
-    }),
-    z.object({
-        type: z.literal("json"),
-        title: z.string().max(256).optional(),
-        data: z.record(z.string(), z.unknown()).refine((value) => isWithinJsonSizeLimit(value, MAX_CONTEXT_ITEM_BYTES), { message: "JSON context item must be less than 16 KB" }),
-    }),
-    z.object({
-        type: z.literal("resource"),
-        title: z.string().max(256).optional(),
-        uri: z.string().max(2048),
-        mimeType: z.string().max(256).optional(),
-        text: z.string().max(MAX_CONTEXT_ITEM_BYTES).optional(),
-    }),
-]);
 export const RuntimeAgentSourceContextSchema = z.discriminatedUnion("type", [
     z.object({
         type: z.literal("branch"),
@@ -61,70 +30,11 @@ export const RuntimeAgentSourceContextSchema = z.discriminatedUnion("type", [
         releaseId: z.string().min(1).max(255),
     }),
 ]);
-const RuntimeMessageExtensionFieldsSchema = {
-    name: z.string().max(256).optional(),
-    metadata: z.record(z.string(), z.unknown()).optional(),
-    createdAt: z.string().optional(),
-};
-export const RuntimeToolFunctionCallSchema = z.object({
-    name: ClientToolNameSchema,
-    arguments: z.string().max(MAX_TOOL_PARAMETERS_BYTES),
-}).strict();
-export const RuntimeToolCallSchema = z.object({
-    id: z.string().min(1).max(128),
-    type: z.literal("function"),
-    function: RuntimeToolFunctionCallSchema,
-}).strict();
-export const RuntimeSystemMessageSchema = z.object({
-    id: z.string().min(1),
-    role: z.literal("system"),
-    content: z.string(),
-    ...RuntimeMessageExtensionFieldsSchema,
-}).strict();
-export const RuntimeUserMessageSchema = z.object({
-    id: z.string().min(1),
-    role: z.literal("user"),
-    content: z.string(),
-    ...RuntimeMessageExtensionFieldsSchema,
-}).strict();
-export const RuntimeAssistantMessageSchema = z.object({
-    id: z.string().min(1),
-    role: z.literal("assistant"),
-    content: z.string().optional(),
-    toolCalls: z.array(RuntimeToolCallSchema).optional(),
-    ...RuntimeMessageExtensionFieldsSchema,
-}).strict();
-export const RuntimeToolMessageSchema = z.object({
-    id: z.string().min(1),
-    role: z.literal("tool"),
-    toolCallId: z.string().min(1).max(128),
-    content: z.string(),
-    error: z.string().optional(),
-    ...RuntimeMessageExtensionFieldsSchema,
-}).strict();
-export const RuntimeMessageSchema = z.discriminatedUnion("role", [
-    RuntimeSystemMessageSchema,
-    RuntimeUserMessageSchema,
-    RuntimeAssistantMessageSchema,
-    RuntimeToolMessageSchema,
-]);
-export const RuntimeContextSchema = z.union([
-    z.object({
-        description: z.string().max(1024),
-        value: z.string().max(MAX_CONTEXT_ITEM_BYTES),
-    }),
-    RuntimeContextItemSchema,
-]);
-export const RuntimeRunAgentInputSchema = z.object({
-    threadId: z.string().uuid(),
-    runId: RunIdSchema,
-    parentRunId: RunIdSchema.optional(),
-    state: z.unknown().optional(),
-    messages: z.array(RuntimeMessageSchema).max(MAX_RUNTIME_MESSAGES),
-    tools: z.array(RuntimeInjectedToolSchema).max(50).default([]),
-    context: z.array(RuntimeContextSchema).max(10).default([]).refine((value) => isWithinJsonSizeLimit(value, MAX_CONTEXT_TOTAL_BYTES), { message: "context must be less than 64 KB total" }),
-    forwardedProps: z.record(z.string(), z.unknown()).optional().refine((value) => value === undefined || isWithinJsonSizeLimit(value, MAX_FORWARDED_PROPS_BYTES), { message: "forwardedProps must be less than 64 KB" }),
-});
+export const RuntimeInjectedToolSchema = AgUiRuntimeInjectedToolSchema;
+export const RuntimeContextItemSchema = AgUiRuntimeContextItemSchema;
+export const RuntimeMessageSchema = AgUiRuntimeMessageSchema;
+export const RuntimeContextSchema = AgUiRuntimeContextSchema;
+export const RuntimeRunAgentInputSchema = AgUiRuntimeRequestSchema;
 export const InternalAgentCompatibilityMessageSchema = z.object({
     id: z.string().min(1),
     role: z.enum(["user", "assistant", "system", "tool"]),
@@ -140,7 +50,7 @@ export const InternalAgentStreamRequestSchema = z.object({
     state: z.unknown().optional(),
     messages: z.array(z.union([RuntimeMessageSchema, InternalAgentCompatibilityMessageSchema])).max(MAX_RUNTIME_MESSAGES),
     tools: z.array(RuntimeInjectedToolSchema).max(50).default([]),
-    context: z.array(RuntimeContextSchema).max(10).default([]).refine((value) => isWithinJsonSizeLimit(value, MAX_CONTEXT_TOTAL_BYTES), { message: "context must be less than 64 KB total" }),
+    context: z.array(RuntimeContextSchema).max(10).default([]).refine((value) => isWithinJsonSizeLimit(value, 65_536), { message: "context must be less than 64 KB total" }),
     agentSource: RuntimeAgentSourceContextSchema.optional(),
     forwardedProps: z.record(z.string(), z.unknown()).optional().refine((value) => value === undefined || isWithinJsonSizeLimit(value, MAX_FORWARDED_PROPS_BYTES), { message: "forwardedProps must be less than 64 KB" }),
 });

package/esm/src/provider/runtime-loader.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runtime-loader.d.ts","sourceRoot":"","sources":["../../../src/src/provider/runtime-loader.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,OAAO,MAAM,qBAAqB,CAAC;AAC/C,OAAO,KAAK,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAEjE,MAAM,WAAW,mBAAmB;IAClC,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,OAAO,OAAO,CAAC,KAAK,CAAC;CAC9B;AAED,MAAM,WAAW,sBAAsB;IACrC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,OAAO,OAAO,CAAC,KAAK,CAAC;CAC9B;AAED,MAAM,WAAW,mBAAmB;IAClC,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,OAAO,OAAO,CAAC,KAAK,CAAC;CAC9B;AAqhDD,wBAAgB,wBAAwB,CACtC,MAAM,EAAE,mBAAmB,EAC3B,OAAO,EAAE,MAAM,GACd,YAAY,CAkDd;AAED,wBAAgB,2BAA2B,CACzC,MAAM,EAAE,sBAAsB,EAC9B,OAAO,EAAE,MAAM,GACd,YAAY,CA4Dd;AAED,wBAAgB,wBAAwB,CACtC,MAAM,EAAE,mBAAmB,EAC3B,OAAO,EAAE,MAAM,GACd,YAAY,CAkDd;AAED,wBAAgB,4BAA4B,CAC1C,MAAM,EAAE,mBAAmB,EAC3B,OAAO,EAAE,MAAM,GACd,gBAAgB,CA0ClB;AAED,wBAAgB,4BAA4B,CAC1C,MAAM,EAAE,mBAAmB,EAC3B,OAAO,EAAE,MAAM,GACd,gBAAgB,CAgDlB"}
+{"version":3,"file":"runtime-loader.d.ts","sourceRoot":"","sources":["../../../src/src/provider/runtime-loader.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,OAAO,MAAM,qBAAqB,CAAC;AAC/C,OAAO,KAAK,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAEjE,MAAM,WAAW,mBAAmB;IAClC,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,OAAO,OAAO,CAAC,KAAK,CAAC;CAC9B;AAED,MAAM,WAAW,sBAAsB;IACrC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,OAAO,OAAO,CAAC,KAAK,CAAC;CAC9B;AAED,MAAM,WAAW,mBAAmB;IAClC,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,OAAO,OAAO,CAAC,KAAK,CAAC;CAC9B;AAu/CD,wBAAgB,wBAAwB,CACtC,MAAM,EAAE,mBAAmB,EAC3B,OAAO,EAAE,MAAM,GACd,YAAY,CAkDd;AAED,wBAAgB,2BAA2B,CACzC,MAAM,EAAE,sBAAsB,EAC9B,OAAO,EAAE,MAAM,GACd,YAAY,CA4Dd;AAED,wBAAgB,wBAAwB,CACtC,MAAM,EAAE,mBAAmB,EAC3B,OAAO,EAAE,MAAM,GACd,YAAY,CAkDd;AAED,wBAAgB,4BAA4B,CAC1C,MAAM,EAAE,mBAAmB,EAC3B,OAAO,EAAE,MAAM,GACd,gBAAgB,CA0ClB;AAED,wBAAgB,4BAA4B,CAC1C,MAAM,EAAE,mBAAmB,EAC3B,OAAO,EAAE,MAAM,GACd,gBAAgB,CAgDlB"}

package/esm/src/provider/runtime-loader.js

@@ -404,11 +404,11 @@ function buildAnthropicGenerateResult(payload) {
         usage: extractAnthropicUsage(payload),
     };
 }
-function parseAnthropicSseChunk(chunk) {
-    const blocks = chunk.split("\n\n");
+function parseSseChunk(chunk) {
+    const blocks = chunk.split(/\r?\n\r?\n/);
     const remainder = blocks.pop() ?? "";
     const events = blocks.flatMap((block) => {
-        const dataLines = block.split("\n")
+        const dataLines = block.split(/\r?\n/)
             .filter((line) => line.startsWith("data:"))
             .map((line) => line.slice(5).trimStart());
         if (!dataLines.length) {
@@ -435,7 +435,7 @@ async function* streamAnthropicCompatibleParts(stream) {
     let usage;
     for await (const chunk of stream) {
         buffer += decoder.decode(chunk, { stream: true });
-        const parsed = parseAnthropicSseChunk(buffer);
+        const parsed = parseSseChunk(buffer);
         buffer = parsed.remainder;
         for (const event of parsed.events) {
             if (event === "[DONE]") {
@@ -547,7 +547,7 @@ async function* streamAnthropicCompatibleParts(stream) {
         }
     }
     if (buffer.trim().length > 0) {
-        const parsed = parseAnthropicSseChunk(`${buffer}\n\n`);
+        const parsed = parseSseChunk(`${buffer}\n\n`);
         for (const event of parsed.events) {
             if (event === "[DONE]") {
                 continue;
@@ -855,7 +855,7 @@ async function* streamGoogleCompatibleParts(stream) {
     let usage;
     for await (const chunk of stream) {
         buffer += decoder.decode(chunk, { stream: true });
-        const parsed = parseOpenAISseChunk(buffer);
+        const parsed = parseSseChunk(buffer);
         buffer = parsed.remainder;
         for (const event of parsed.events) {
             if (event === "[DONE]") {
@@ -902,7 +902,7 @@ async function* streamGoogleCompatibleParts(stream) {
         }
     }
     if (buffer.trim().length > 0) {
-        const parsed = parseOpenAISseChunk(`${buffer}\n\n`);
+        const parsed = parseSseChunk(`${buffer}\n\n`);
         for (const event of parsed.events) {
             if (event === "[DONE]") {
                 continue;
@@ -947,29 +947,6 @@ function buildOpenAIGenerateResult(payload) {
         usage: extractOpenAIUsage(payload),
     };
 }
-function parseOpenAISseChunk(chunk) {
-    const blocks = chunk.split("\n\n");
-    const remainder = blocks.pop() ?? "";
-    const events = blocks.flatMap((block) => {
-        const dataLines = block.split("\n")
-            .filter((line) => line.startsWith("data:"))
-            .map((line) => line.slice(5).trimStart());
-        if (!dataLines.length) {
-            return [];
-        }
-        const payload = dataLines.join("\n").trim();
-        if (payload === "[DONE]") {
-            return ["[DONE]"];
-        }
-        try {
-            return [JSON.parse(payload)];
-        }
-        catch {
-            return [];
-        }
-    });
-    return { events, remainder };
-}
 async function* streamOpenAICompatibleParts(stream) {
     const decoder = new TextDecoder();
     let buffer = "";
@@ -978,7 +955,7 @@ async function* streamOpenAICompatibleParts(stream) {
     let usage;
     for await (const chunk of stream) {
         buffer += decoder.decode(chunk, { stream: true });
-        const parsed = parseOpenAISseChunk(buffer);
+        const parsed = parseSseChunk(buffer);
         buffer = parsed.remainder;
         for (const event of parsed.events) {
             if (event === "[DONE]") {
@@ -1040,7 +1017,7 @@ async function* streamOpenAICompatibleParts(stream) {
         }
     }
     if (buffer.trim().length > 0) {
-        const parsed = parseOpenAISseChunk(`${buffer}\n\n`);
+        const parsed = parseSseChunk(`${buffer}\n\n`);
         for (const event of parsed.events) {
             if (event === "[DONE]") {
                 continue;

package/esm/src/utils/version-constant.d.ts

@@ -1,2 +1,2 @@
-export declare const VERSION = "0.1.150";
+export declare const VERSION = "0.1.152";
 //# sourceMappingURL=version-constant.d.ts.map

package/esm/src/utils/version-constant.js

@@ -1,3 +1,3 @@
 // Keep in sync with deno.json version.
 // scripts/release.ts updates this constant during releases.
-export const VERSION = "0.1.150";
+export const VERSION = "0.1.152";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "veryfront",
-  "version": "0.1.150",
+  "version": "0.1.152",
   "description": "The simplest way to build AI-powered apps",
   "keywords": [
     "react",

package/src/deno.js

@@ -1,6 +1,6 @@
 export default {
   "name": "veryfront",
-  "version": "0.1.150",
+  "version": "0.1.152",
   "license": "Apache-2.0",
   "nodeModulesDir": "auto",
   "exclude": [

package/src/src/agent/ag-ui-handler.ts

@@ -288,6 +288,12 @@ async function createAgUiStreamResponse(
       if (!enqueueEvent(controller, "RunStarted", { runId, threadId, agentId: agent.id })) {
         return;
       }
+      if (!enqueueEvent(controller, "StateSnapshot", { snapshot: {} })) {
+        return;
+      }
+      if (!enqueueEvent(controller, "MessagesSnapshot", { messages: request.messages })) {
+        return;
+      }
 
       try {
         if (!upstreamBody) {

package/src/src/agent/index.ts

@@ -135,6 +135,16 @@ export {
 } from "./composition/index.js";
 
 export { agent } from "./factory.js";
+export {
+  type AgUiRuntimeContextItem,
+  AgUiRuntimeContextItemSchema,
+  type AgUiRuntimeInjectedTool,
+  AgUiRuntimeInjectedToolSchema,
+  type AgUiRuntimeMessage,
+  AgUiRuntimeMessageSchema,
+  type AgUiRuntimeRequest,
+  AgUiRuntimeRequestSchema,
+} from "./runtime-ag-ui-contract.js";
 export {
   type AgUiContextItem,
   type AgUiHandlerConfigWithAgent,

package/src/src/agent/runtime-ag-ui-contract.ts

@@ -0,0 +1,144 @@
+import { z } from "zod";
+
+const AGENT_ID_PATTERN = /^[a-zA-Z0-9_-]+$/;
+const MAX_TOOL_PARAMETERS_BYTES = 16_384;
+const MAX_CONTEXT_ITEM_BYTES = 16_384;
+const MAX_CONTEXT_TOTAL_BYTES = 65_536;
+const MAX_FORWARDED_PROPS_BYTES = 65_536;
+const MAX_RUNTIME_MESSAGES = 100;
+
+const encoder = new TextEncoder();
+
+function isWithinJsonSizeLimit(value: unknown, maxBytes: number): boolean {
+  try {
+    return encoder.encode(JSON.stringify(value)).byteLength <= maxBytes;
+  } catch {
+    return false;
+  }
+}
+
+export const AgUiRuntimeRunIdSchema = z.string().min(1).max(128).regex(AGENT_ID_PATTERN);
+
+export const AgUiRuntimeInjectedToolSchema = z.object({
+  name: z
+    .string()
+    .min(1)
+    .max(128)
+    .regex(
+      /^[a-zA-Z][a-zA-Z0-9._:-]*$/,
+      "Tool names must start with a letter and use a valid client-tool format",
+    ),
+  description: z.string().max(1024).optional(),
+  parameters: z.record(z.string(), z.unknown()).optional().refine(
+    (value) => value === undefined || isWithinJsonSizeLimit(value, MAX_TOOL_PARAMETERS_BYTES),
+    { message: "Tool parameters must be less than 16 KB" },
+  ),
+});
+
+export const AgUiRuntimeContextItemSchema = z.discriminatedUnion("type", [
+  z.object({
+    type: z.literal("text"),
+    title: z.string().max(256).optional(),
+    text: z.string().max(MAX_CONTEXT_ITEM_BYTES),
+  }),
+  z.object({
+    type: z.literal("json"),
+    title: z.string().max(256).optional(),
+    data: z.record(z.string(), z.unknown()).refine(
+      (value) => isWithinJsonSizeLimit(value, MAX_CONTEXT_ITEM_BYTES),
+      { message: "JSON context item must be less than 16 KB" },
+    ),
+  }),
+  z.object({
+    type: z.literal("resource"),
+    title: z.string().max(256).optional(),
+    uri: z.string().max(2048),
+    mimeType: z.string().max(256).optional(),
+    text: z.string().max(MAX_CONTEXT_ITEM_BYTES).optional(),
+  }),
+]);
+
+const RuntimeMessageExtensionFieldsSchema = {
+  name: z.string().max(256).optional(),
+  metadata: z.record(z.string(), z.unknown()).optional(),
+  createdAt: z.string().optional(),
+} as const;
+
+export const AgUiRuntimeToolFunctionCallSchema = z.object({
+  name: AgUiRuntimeInjectedToolSchema.shape.name,
+  arguments: z.string().max(MAX_TOOL_PARAMETERS_BYTES),
+}).strict();
+
+export const AgUiRuntimeToolCallSchema = z.object({
+  id: z.string().min(1).max(128),
+  type: z.literal("function"),
+  function: AgUiRuntimeToolFunctionCallSchema,
+}).strict();
+
+export const AgUiRuntimeSystemMessageSchema = z.object({
+  id: z.string().min(1),
+  role: z.literal("system"),
+  content: z.string(),
+  ...RuntimeMessageExtensionFieldsSchema,
+}).strict();
+
+export const AgUiRuntimeUserMessageSchema = z.object({
+  id: z.string().min(1),
+  role: z.literal("user"),
+  content: z.string(),
+  ...RuntimeMessageExtensionFieldsSchema,
+}).strict();
+
+export const AgUiRuntimeAssistantMessageSchema = z.object({
+  id: z.string().min(1),
+  role: z.literal("assistant"),
+  content: z.string().optional(),
+  toolCalls: z.array(AgUiRuntimeToolCallSchema).optional(),
+  ...RuntimeMessageExtensionFieldsSchema,
+}).strict();
+
+export const AgUiRuntimeToolMessageSchema = z.object({
+  id: z.string().min(1),
+  role: z.literal("tool"),
+  toolCallId: z.string().min(1).max(128),
+  content: z.string(),
+  error: z.string().optional(),
+  ...RuntimeMessageExtensionFieldsSchema,
+}).strict();
+
+export const AgUiRuntimeMessageSchema = z.discriminatedUnion("role", [
+  AgUiRuntimeSystemMessageSchema,
+  AgUiRuntimeUserMessageSchema,
+  AgUiRuntimeAssistantMessageSchema,
+  AgUiRuntimeToolMessageSchema,
+]);
+
+export const AgUiRuntimeContextSchema = z.union([
+  z.object({
+    description: z.string().max(1024),
+    value: z.string().max(MAX_CONTEXT_ITEM_BYTES),
+  }),
+  AgUiRuntimeContextItemSchema,
+]);
+
+export const AgUiRuntimeRequestSchema = z.object({
+  threadId: z.string().uuid(),
+  runId: AgUiRuntimeRunIdSchema,
+  parentRunId: AgUiRuntimeRunIdSchema.optional(),
+  state: z.unknown().optional(),
+  messages: z.array(AgUiRuntimeMessageSchema).max(MAX_RUNTIME_MESSAGES),
+  tools: z.array(AgUiRuntimeInjectedToolSchema).max(50).default([]),
+  context: z.array(AgUiRuntimeContextSchema).max(10).default([]).refine(
+    (value) => isWithinJsonSizeLimit(value, MAX_CONTEXT_TOTAL_BYTES),
+    { message: "context must be less than 64 KB total" },
+  ),
+  forwardedProps: z.record(z.string(), z.unknown()).optional().refine(
+    (value) => value === undefined || isWithinJsonSizeLimit(value, MAX_FORWARDED_PROPS_BYTES),
+    { message: "forwardedProps must be less than 64 KB" },
+  ),
+});
+
+export type AgUiRuntimeInjectedTool = z.infer<typeof AgUiRuntimeInjectedToolSchema>;
+export type AgUiRuntimeContextItem = z.infer<typeof AgUiRuntimeContextItemSchema>;
+export type AgUiRuntimeMessage = z.infer<typeof AgUiRuntimeMessageSchema>;
+export type AgUiRuntimeRequest = z.infer<typeof AgUiRuntimeRequestSchema>;

package/src/src/internal-agents/ag-ui-sse.ts

@@ -41,6 +41,18 @@ const agUiEventPayloadSchemas = {
     threadId: z.string().min(1),
     agentId: z.string().min(1),
   }),
+  StateSnapshot: z.object({
+    snapshot: z.record(z.string(), z.unknown()),
+  }),
+  MessagesSnapshot: z.object({
+    messages: z.array(z.object({
+      id: z.string().min(1),
+      role: z.enum(["user", "assistant", "system", "tool"]),
+      parts: z.array(z.record(z.string(), z.unknown())),
+      metadata: z.record(z.string(), z.unknown()).optional(),
+      createdAt: z.string().optional(),
+    })),
+  }),
   TextMessageStart: z.object({
     messageId: z.string().min(1),
     role: z.literal("assistant"),

package/src/src/internal-agents/schema.ts

@@ -1,9 +1,17 @@
 import { z } from "zod";
+import {
+  AgUiRuntimeContextItemSchema,
+  AgUiRuntimeContextSchema,
+  AgUiRuntimeInjectedToolSchema,
+  type AgUiRuntimeMessage,
+  AgUiRuntimeMessageSchema,
+  type AgUiRuntimeRequest,
+  AgUiRuntimeRequestSchema,
+  AgUiRuntimeRunIdSchema,
+  AgUiRuntimeToolCallSchema,
+} from "../agent/runtime-ag-ui-contract.js";
 
 const AGENT_ID_PATTERN = /^[a-zA-Z0-9_-]+$/;
-const MAX_TOOL_PARAMETERS_BYTES = 16_384;
-const MAX_CONTEXT_ITEM_BYTES = 16_384;
-const MAX_CONTEXT_TOTAL_BYTES = 65_536;
 const MAX_FORWARDED_PROPS_BYTES = 65_536;
 const MAX_TOOL_RESULT_BYTES = 65_536;
 const MAX_RUNTIME_MESSAGES = 100;
@@ -18,51 +26,10 @@ function isWithinJsonSizeLimit(value: unknown, maxBytes: number): boolean {
   }
 }
 
-export const RunIdSchema = z.string().min(1).max(128).regex(AGENT_ID_PATTERN);
+export const RunIdSchema = AgUiRuntimeRunIdSchema;
 
 export const AgentIdSchema = z.string().min(1).max(128).regex(AGENT_ID_PATTERN);
 
-export const ClientToolNameSchema = z
-  .string()
-  .min(1)
-  .max(128)
-  .regex(
-    /^[a-zA-Z][a-zA-Z0-9._:-]*$/,
-    "Tool names must start with a letter and use a valid client-tool format",
-  );
-
-export const RuntimeInjectedToolSchema = z.object({
-  name: ClientToolNameSchema,
-  description: z.string().max(1024).optional(),
-  parameters: z.record(z.string(), z.unknown()).optional().refine(
-    (value) => value === undefined || isWithinJsonSizeLimit(value, MAX_TOOL_PARAMETERS_BYTES),
-    { message: "Tool parameters must be less than 16 KB" },
-  ),
-});
-
-export const RuntimeContextItemSchema = z.discriminatedUnion("type", [
-  z.object({
-    type: z.literal("text"),
-    title: z.string().max(256).optional(),
-    text: z.string().max(MAX_CONTEXT_ITEM_BYTES),
-  }),
-  z.object({
-    type: z.literal("json"),
-    title: z.string().max(256).optional(),
-    data: z.record(z.string(), z.unknown()).refine(
-      (value) => isWithinJsonSizeLimit(value, MAX_CONTEXT_ITEM_BYTES),
-      { message: "JSON context item must be less than 16 KB" },
-    ),
-  }),
-  z.object({
-    type: z.literal("resource"),
-    title: z.string().max(256).optional(),
-    uri: z.string().max(2048),
-    mimeType: z.string().max(256).optional(),
-    text: z.string().max(MAX_CONTEXT_ITEM_BYTES).optional(),
-  }),
-]);
-
 export const RuntimeAgentSourceContextSchema = z.discriminatedUnion("type", [
   z.object({
     type: z.literal("branch"),
@@ -79,85 +46,11 @@ export const RuntimeAgentSourceContextSchema = z.discriminatedUnion("type", [
   }),
 ]);
 
-const RuntimeMessageExtensionFieldsSchema = {
-  name: z.string().max(256).optional(),
-  metadata: z.record(z.string(), z.unknown()).optional(),
-  createdAt: z.string().optional(),
-} as const;
-
-export const RuntimeToolFunctionCallSchema = z.object({
-  name: ClientToolNameSchema,
-  arguments: z.string().max(MAX_TOOL_PARAMETERS_BYTES),
-}).strict();
-
-export const RuntimeToolCallSchema = z.object({
-  id: z.string().min(1).max(128),
-  type: z.literal("function"),
-  function: RuntimeToolFunctionCallSchema,
-}).strict();
-
-export const RuntimeSystemMessageSchema = z.object({
-  id: z.string().min(1),
-  role: z.literal("system"),
-  content: z.string(),
-  ...RuntimeMessageExtensionFieldsSchema,
-}).strict();
-
-export const RuntimeUserMessageSchema = z.object({
-  id: z.string().min(1),
-  role: z.literal("user"),
-  content: z.string(),
-  ...RuntimeMessageExtensionFieldsSchema,
-}).strict();
-
-export const RuntimeAssistantMessageSchema = z.object({
-  id: z.string().min(1),
-  role: z.literal("assistant"),
-  content: z.string().optional(),
-  toolCalls: z.array(RuntimeToolCallSchema).optional(),
-  ...RuntimeMessageExtensionFieldsSchema,
-}).strict();
-
-export const RuntimeToolMessageSchema = z.object({
-  id: z.string().min(1),
-  role: z.literal("tool"),
-  toolCallId: z.string().min(1).max(128),
-  content: z.string(),
-  error: z.string().optional(),
-  ...RuntimeMessageExtensionFieldsSchema,
-}).strict();
-
-export const RuntimeMessageSchema = z.discriminatedUnion("role", [
-  RuntimeSystemMessageSchema,
-  RuntimeUserMessageSchema,
-  RuntimeAssistantMessageSchema,
-  RuntimeToolMessageSchema,
-]);
-
-export const RuntimeContextSchema = z.union([
-  z.object({
-    description: z.string().max(1024),
-    value: z.string().max(MAX_CONTEXT_ITEM_BYTES),
-  }),
-  RuntimeContextItemSchema,
-]);
-
-export const RuntimeRunAgentInputSchema = z.object({
-  threadId: z.string().uuid(),
-  runId: RunIdSchema,
-  parentRunId: RunIdSchema.optional(),
-  state: z.unknown().optional(),
-  messages: z.array(RuntimeMessageSchema).max(MAX_RUNTIME_MESSAGES),
-  tools: z.array(RuntimeInjectedToolSchema).max(50).default([]),
-  context: z.array(RuntimeContextSchema).max(10).default([]).refine(
-    (value) => isWithinJsonSizeLimit(value, MAX_CONTEXT_TOTAL_BYTES),
-    { message: "context must be less than 64 KB total" },
-  ),
-  forwardedProps: z.record(z.string(), z.unknown()).optional().refine(
-    (value) => value === undefined || isWithinJsonSizeLimit(value, MAX_FORWARDED_PROPS_BYTES),
-    { message: "forwardedProps must be less than 64 KB" },
-  ),
-});
+export const RuntimeInjectedToolSchema = AgUiRuntimeInjectedToolSchema;
+export const RuntimeContextItemSchema = AgUiRuntimeContextItemSchema;
+export const RuntimeMessageSchema = AgUiRuntimeMessageSchema;
+export const RuntimeContextSchema = AgUiRuntimeContextSchema;
+export const RuntimeRunAgentInputSchema = AgUiRuntimeRequestSchema;
 
 export const InternalAgentCompatibilityMessageSchema = z.object({
   id: z.string().min(1),
@@ -178,7 +71,7 @@ export const InternalAgentStreamRequestSchema = z.object({
   ),
   tools: z.array(RuntimeInjectedToolSchema).max(50).default([]),
   context: z.array(RuntimeContextSchema).max(10).default([]).refine(
-    (value) => isWithinJsonSizeLimit(value, MAX_CONTEXT_TOTAL_BYTES),
+    (value) => isWithinJsonSizeLimit(value, 65_536),
     { message: "context must be less than 64 KB total" },
   ),
   agentSource: RuntimeAgentSourceContextSchema.optional(),
@@ -188,7 +81,7 @@ export const InternalAgentStreamRequestSchema = z.object({
   ),
 });
 
-type RuntimeMessage = z.infer<typeof RuntimeMessageSchema>;
+type RuntimeMessage = AgUiRuntimeMessage;
 type InternalAgentCompatibilityMessage = z.infer<typeof InternalAgentCompatibilityMessageSchema>;
 
 function extractToolArgs(
@@ -243,7 +136,7 @@ function isCanonicalToolCallPart(part: Record<string, unknown>): boolean {
 
 function getToolCallShape(
   part: Record<string, unknown>,
-): z.infer<typeof RuntimeToolCallSchema> | null {
+): z.infer<typeof AgUiRuntimeToolCallSchema> | null {
   const id = getPartString(part, "toolCallId", "tool_call_id", "id");
   const name = getPartString(part, "toolName", "tool_name", "name");
 
@@ -357,7 +250,7 @@ function toRuntimeMessage(
 
 export function toRuntimeRunAgentInput(
   input: z.infer<typeof InternalAgentStreamRequestSchema>,
-): z.infer<typeof RuntimeRunAgentInputSchema> {
+): AgUiRuntimeRequest {
   return {
     threadId: input.threadId,
     runId: input.runId,
@@ -385,6 +278,6 @@ export const ResumeSignalSchema = z.discriminatedUnion("type", [
 export type RuntimeInjectedTool = z.infer<typeof RuntimeInjectedToolSchema>;
 export type RuntimeContextItem = z.infer<typeof RuntimeContextItemSchema>;
 export type RuntimeAgentSourceContext = z.infer<typeof RuntimeAgentSourceContextSchema>;
-export type RuntimeRunAgentInput = z.infer<typeof RuntimeRunAgentInputSchema>;
+export type RuntimeRunAgentInput = AgUiRuntimeRequest;
 export type InternalAgentStreamRequest = z.infer<typeof InternalAgentStreamRequestSchema>;
 export type ResumeSignal = z.infer<typeof ResumeSignalSchema>;