veryfront 0.1.219 → 0.1.221

package/esm/deno.js

@@ -1,6 +1,6 @@
 export default {
     "name": "veryfront",
-    "version": "0.1.219",
+    "version": "0.1.221",
     "license": "Apache-2.0",
     "nodeModulesDir": "auto",
     "workspace": [

package/esm/src/agent/ag-ui-detached-start.d.ts

@@ -55,10 +55,18 @@ export declare const AgUiDetachedStartAcceptedSchema: z.ZodObject<{
 }, z.core.$strip>;
 export type AgUiDetachedStartRequest = z.infer<typeof AgUiDetachedStartRequestSchema>;
 export type AgUiDetachedStartAccepted = z.infer<typeof AgUiDetachedStartAcceptedSchema>;
-export interface AgUiDetachedStartHandlerOptions {
-    agent: Agent;
+interface AgUiDetachedStartExecutionInput {
+    request: AgUiDetachedStartRequest;
+    requestOrCtx: unknown;
+    rawRequest: Request;
+    context: Record<string, unknown>;
+    abortSignal: AbortSignal;
+}
+type AgUiDetachedExecutionStarter = (input: AgUiDetachedStartExecutionInput) => Promise<void> | void;
+interface AgUiDetachedStartHandlerOptionsBase {
     sessionManager: RunResumeSessionManager<AgUiResumeValue>;
     context?: AgUiContextValue;
+    startDetachedExecution?: AgUiDetachedExecutionStarter;
     onAccepted?: (input: {
         request: AgUiDetachedStartRequest;
         runId: string;
@@ -79,6 +87,12 @@ export interface AgUiDetachedStartHandlerOptions {
         error: unknown;
     }) => Promise<void> | void;
 }
+export type AgUiDetachedStartHandlerOptions = (AgUiDetachedStartHandlerOptionsBase & {
+    agent: Agent;
+}) | (AgUiDetachedStartHandlerOptionsBase & {
+    agent?: undefined;
+    startDetachedExecution: AgUiDetachedExecutionStarter;
+});
 export declare function createAgUiDetachedStartHandler(options: AgUiDetachedStartHandlerOptions): (requestOrCtx: unknown) => Promise<Response>;
 export {};
 //# sourceMappingURL=ag-ui-detached-start.d.ts.map

package/esm/src/agent/ag-ui-detached-start.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ag-ui-detached-start.d.ts","sourceRoot":"","sources":["../../../src/src/agent/ag-ui-detached-start.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAMxB,OAAO,EAGL,KAAK,uBAAuB,EAC7B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,KAAK,EAAW,MAAM,YAAY,CAAC;AAMjD,KAAK,eAAe,GAAG;IACrB,MAAM,EAAE,OAAO,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;CAClB,CAAC;AAEF,KAAK,gBAAgB,GACjB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACvB,CAAC,CAAC,OAAO,EAAE,OAAO,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;AAsOvF,eAAO,MAAM,8BAA8B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAGzC,CAAC;AAEH,eAAO,MAAM,+BAA+B;;;;;iBAK1C,CAAC;AAEH,MAAM,MAAM,wBAAwB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,8BAA8B,CAAC,CAAC;AACtF,MAAM,MAAM,yBAAyB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,+BAA+B,CAAC,CAAC;AAExF,MAAM,WAAW,+BAA+B;IAC9C,KAAK,EAAE,KAAK,CAAC;IACb,cAAc,EAAE,uBAAuB,CAAC,eAAe,CAAC,CAAC;IACzD,OAAO,CAAC,EAAE,gBAAgB,CAAC;IAC3B,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE;QACnB,OAAO,EAAE,wBAAwB,CAAC;QAClC,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC;KAClB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC3B,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE;QACpB,OAAO,EAAE,wBAAwB,CAAC;QAClC,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC;KAClB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC3B,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAChF,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,OAAO,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;CAChG;AAED,wBAAgB,8BAA8B,CAC5C,OAAO,EAAE,+BAA+B,GACvC,CAAC,YAAY,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CA2G9C"}
+{"version":3,"file":"ag-ui-detached-start.d.ts","sourceRoot":"","sources":["../../../src/src/agent/ag-ui-detached-start.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAUxB,OAAO,EAGL,KAAK,uBAAuB,EAC7B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAIxC,KAAK,eAAe,GAAG;IACrB,MAAM,EAAE,OAAO,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;CAClB,CAAC;AAEF,KAAK,gBAAgB,GACjB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACvB,CAAC,CAAC,OAAO,EAAE,OAAO,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;AA8IvF,eAAO,MAAM,8BAA8B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAGzC,CAAC;AAEH,eAAO,MAAM,+BAA+B;;;;;iBAK1C,CAAC;AAEH,MAAM,MAAM,wBAAwB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,8BAA8B,CAAC,CAAC;AACtF,MAAM,MAAM,yBAAyB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,+BAA+B,CAAC,CAAC;AAExF,UAAU,+BAA+B;IACvC,OAAO,EAAE,wBAAwB,CAAC;IAClC,YAAY,EAAE,OAAO,CAAC;IACtB,UAAU,EAAE,OAAO,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC,WAAW,EAAE,WAAW,CAAC;CAC1B;AAED,KAAK,4BAA4B,GAAG,CAClC,KAAK,EAAE,+BAA+B,KACnC,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AAE1B,UAAU,mCAAmC;IAC3C,cAAc,EAAE,uBAAuB,CAAC,eAAe,CAAC,CAAC;IACzD,OAAO,CAAC,EAAE,gBAAgB,CAAC;IAC3B,sBAAsB,CAAC,EAAE,4BAA4B,CAAC;IACtD,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE;QACnB,OAAO,EAAE,wBAAwB,CAAC;QAClC,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC;KAClB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC3B,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE;QACpB,OAAO,EAAE,wBAAwB,CAAC;QAClC,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC;KAClB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC3B,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAChF,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,OAAO,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;CAChG;AAED,MAAM,MAAM,+BAA+B,GACvC,CAAC,mCAAmC,GAAG;IAAE,KAAK,EAAE,KAAK,CAAA;CAAE,CAAC,GACxD,CAAC,mCAAmC,GAAG;IACvC,KAAK,CAAC,EAAE,SAAS,CAAC;IAClB,sBAAsB,EAAE,4BAA4B,CAAC;CACtD,CAAC,CAAC;AA0BL,wBAAgB,8BAA8B,CAC5C,OAAO,EAAE,+BAA+B,GACvC,CAAC,YAAY,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAwH9C"}

package/esm/src/agent/ag-ui-detached-start.js

@@ -3,85 +3,10 @@ import { INVALID_ARGUMENT } from "../errors/index.js";
 import { SKILL_TOOL_IDS } from "../skill/types.js";
 import { toolRegistry } from "../tool/index.js";
 import { streamDataStreamEvents } from "./data-stream.js";
-import { AgUiRequestSchema } from "./ag-ui-handler.js";
+import { AgUiRequestSchema, normalizeAgUiMessages, } from "./ag-ui-host-support.js";
 import { AgentRuntime, RunAlreadyExistsError, } from "./runtime/index.js";
 const AGENT_ID_PATTERN = /^[a-zA-Z0-9_-]+$/;
 const AG_UI_DETACHED_RUN_ID_SCHEMA = z.string().min(1).max(128).regex(AGENT_ID_PATTERN);
-const MAX_TEXT_PART_LENGTH = 10_000;
-function isRecord(value) {
-    return typeof value === "object" && value !== null && !Array.isArray(value);
-}
-function normalizeToolArgs(part) {
-    if (isRecord(part.args))
-        return part.args;
-    if (isRecord(part.input))
-        return part.input;
-    return {};
-}
-function normalizeMessagePart(part) {
-    if (part.type === "text" && typeof part.text === "string" &&
-        part.text.length <= MAX_TEXT_PART_LENGTH) {
-        return { type: "text", text: part.text };
-    }
-    if (part.type === "tool_call" && typeof part.id === "string" && typeof part.name === "string") {
-        return {
-            type: "tool-call",
-            toolCallId: part.id,
-            toolName: part.name,
-            args: normalizeToolArgs(part),
-        };
-    }
-    if (part.type === "tool-call" &&
-        typeof part.toolCallId === "string" &&
-        typeof part.toolName === "string") {
-        return {
-            type: "tool-call",
-            toolCallId: part.toolCallId,
-            toolName: part.toolName,
-            args: normalizeToolArgs(part),
-        };
-    }
-    if (typeof part.type === "string" &&
-        part.type.startsWith("tool-") &&
-        part.type !== "tool-result" &&
-        typeof part.toolCallId === "string" &&
-        typeof part.toolName === "string") {
-        return {
-            type: part.type,
-            toolCallId: part.toolCallId,
-            toolName: part.toolName,
-            args: normalizeToolArgs(part),
-        };
-    }
-    if (part.type === "tool_result" && typeof part.tool_call_id === "string") {
-        return {
-            type: "tool-result",
-            toolCallId: part.tool_call_id,
-            toolName: typeof part.tool_name === "string" ? part.tool_name : "unknown",
-            result: "output" in part ? part.output : undefined,
-        };
-    }
-    if (part.type === "tool-result" && typeof part.toolCallId === "string") {
-        return {
-            type: "tool-result",
-            toolCallId: part.toolCallId,
-            toolName: typeof part.toolName === "string" ? part.toolName : "unknown",
-            result: "result" in part ? part.result : undefined,
-        };
-    }
-    return null;
-}
-function normalizeMessages(messages) {
-    return messages.map((message) => ({
-        id: message.id,
-        role: message.role,
-        parts: message.parts
-            .map((part) => normalizeMessagePart(part))
-            .filter((part) => part !== null),
-        ...(message.createdAt ? { timestamp: Date.parse(message.createdAt) || undefined } : {}),
-        ...(message.metadata ? { metadata: message.metadata } : {}),
-    }));
-}
 function isRequest(obj) {
     return (typeof obj === "object" &&
         obj !== null &&
@@ -190,7 +115,18 @@ export const AgUiDetachedStartAcceptedSchema = z.object({
     runId: AG_UI_DETACHED_RUN_ID_SCHEMA,
     threadId: z.string().uuid(),
 });
+async function startDefaultDetachedExecution(input) {
+    const runtime = new AgentRuntime(input.agent.id, {
+        ...input.agent.config,
+        tools: buildMergedTools(input.agent, input.request, input.sessionManager),
+    });
+    const runtimeStream = await runtime.stream(normalizeAgUiMessages(input.request.messages), buildStreamContext(input.request, input.context, input.request.threadId, input.request.runId), undefined, input.request.model, input.request.maxOutputTokens, input.abortSignal);
+    await drainRuntimeStream(runtimeStream);
+}
 export function createAgUiDetachedStartHandler(options) {
+    if (!options.agent && !options.startDetachedExecution) {
+        throw new Error("Detached AG-UI start requires either an agent or startDetachedExecution handler.");
+    }
     return async function POST(requestOrCtx) {
         const request = extractRequest(requestOrCtx);
         try {
@@ -201,11 +137,6 @@ export function createAgUiDetachedStartHandler(options) {
                     runId: parsed.runId,
                     threadId: parsed.threadId,
                 });
-                const runtime = new AgentRuntime(options.agent.id, {
-                    ...options.agent.config,
-                    tools: buildMergedTools(options.agent, parsed, options.sessionManager),
-                });
-                const runtimeStream = await runtime.stream(normalizeMessages(parsed.messages), buildStreamContext(parsed, context, parsed.threadId, parsed.runId), undefined, parsed.model, parsed.maxOutputTokens, abortSignal);
                 await options.onAccepted?.({
                     request: parsed,
                     runId: parsed.runId,
@@ -213,7 +144,27 @@ export function createAgUiDetachedStartHandler(options) {
                 });
                 const detachedTask = (async () => {
                     try {
-                        await drainRuntimeStream(runtimeStream);
+                        if (options.startDetachedExecution) {
+                            await options.startDetachedExecution({
+                                request: parsed,
+                                requestOrCtx,
+                                rawRequest: request,
+                                context,
+                                abortSignal,
+                            });
+                        }
+                        else if (options.agent) {
+                            await startDefaultDetachedExecution({
+                                agent: options.agent,
+                                request: parsed,
+                                context,
+                                abortSignal,
+                                sessionManager: options.sessionManager,
+                            });
+                        }
+                        else {
+                            throw new Error("Detached AG-UI start configuration became invalid during execution.");
+                        }
                         options.sessionManager.completeRun(parsed.runId);
                         await options.onFinish?.({
                             runId: parsed.runId,

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

@@ -1,70 +1,6 @@
-import { z } from "zod";
 import type { Agent } from "./types.js";
 import { type RunResumeSessionManager } from "./runtime/index.js";
-declare const AgUiInjectedToolSchema: z.ZodObject<{
-    name: z.ZodString;
-    description: z.ZodOptional<z.ZodString>;
-    parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-}, z.core.$strip>;
-declare const AgUiContextItemSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
-    type: z.ZodLiteral<"text">;
-    title: z.ZodOptional<z.ZodString>;
-    text: z.ZodString;
-}, z.core.$strip>, z.ZodObject<{
-    type: z.ZodLiteral<"json">;
-    title: z.ZodOptional<z.ZodString>;
-    data: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-}, z.core.$strip>, z.ZodObject<{
-    type: z.ZodLiteral<"resource">;
-    title: z.ZodOptional<z.ZodString>;
-    uri: z.ZodString;
-    mimeType: z.ZodOptional<z.ZodString>;
-    text: z.ZodOptional<z.ZodString>;
-}, z.core.$strip>], "type">;
-export declare const AgUiRequestSchema: z.ZodObject<{
-    threadId: z.ZodOptional<z.ZodString>;
-    runId: z.ZodOptional<z.ZodString>;
-    messages: z.ZodArray<z.ZodObject<{
-        id: z.ZodString;
-        role: z.ZodEnum<{
-            tool: "tool";
-            user: "user";
-            assistant: "assistant";
-            system: "system";
-        }>;
-        parts: z.ZodDefault<z.ZodArray<z.ZodObject<{
-            type: z.ZodString;
-        }, z.core.$loose>>>;
-        metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-        createdAt: z.ZodOptional<z.ZodString>;
-    }, z.core.$strip>>;
-    tools: z.ZodDefault<z.ZodArray<z.ZodObject<{
-        name: z.ZodString;
-        description: z.ZodOptional<z.ZodString>;
-        parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-    }, z.core.$strip>>>;
-    context: z.ZodDefault<z.ZodArray<z.ZodDiscriminatedUnion<[z.ZodObject<{
-        type: z.ZodLiteral<"text">;
-        title: z.ZodOptional<z.ZodString>;
-        text: z.ZodString;
-    }, z.core.$strip>, z.ZodObject<{
-        type: z.ZodLiteral<"json">;
-        title: z.ZodOptional<z.ZodString>;
-        data: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-    }, z.core.$strip>, z.ZodObject<{
-        type: z.ZodLiteral<"resource">;
-        title: z.ZodOptional<z.ZodString>;
-        uri: z.ZodString;
-        mimeType: z.ZodOptional<z.ZodString>;
-        text: z.ZodOptional<z.ZodString>;
-    }, z.core.$strip>], "type">>>;
-    forwardedProps: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-    model: z.ZodOptional<z.ZodString>;
-    maxOutputTokens: z.ZodOptional<z.ZodNumber>;
-}, z.core.$strip>;
-export type AgUiInjectedTool = z.infer<typeof AgUiInjectedToolSchema>;
-export type AgUiContextItem = z.infer<typeof AgUiContextItemSchema>;
-export type AgUiRequest = z.infer<typeof AgUiRequestSchema>;
+export { type AgUiContextItem, AgUiContextItemSchema, type AgUiInjectedTool, AgUiInjectedToolSchema, type AgUiRequest, AgUiRequestSchema, } from "./ag-ui-host-support.js";
 type AgUiResumeValue = {
     result: unknown;
     isError: boolean;
@@ -78,5 +14,4 @@ export interface AgUiHandlerConfigWithAgent extends AgUiHandlerOptions {
 }
 export declare function createAgUiHandler(agentId: string, options?: AgUiHandlerOptions): (requestOrCtx: unknown) => Promise<Response>;
 export declare function createAgUiHandler(config: AgUiHandlerConfigWithAgent, options?: AgUiHandlerOptions): (requestOrCtx: unknown) => Promise<Response>;
-export {};
 //# sourceMappingURL=ag-ui-handler.d.ts.map

package/esm/src/agent/ag-ui-handler.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ag-ui-handler.d.ts","sourceRoot":"","sources":["../../../src/src/agent/ag-ui-handler.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,KAAK,EAAE,KAAK,EAAW,MAAM,YAAY,CAAC;AACjD,OAAO,EAGL,KAAK,uBAAuB,EAC7B,MAAM,oBAAoB,CAAC;AAsC5B,QAAA,MAAM,sBAAsB;;;;iBAO1B,CAAC;AAEH,QAAA,MAAM,qBAAqB;;;;;;;;;;;;;;2BAqBzB,CAAC;AAYH,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAe5B,CAAC;AAEH,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AACtE,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AACpE,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAG5D,KAAK,eAAe,GAAG;IAAE,MAAM,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,OAAO,CAAA;CAAE,CAAC;AAyY7D,MAAM,WAAW,kBAAkB;IACjC,OAAO,CAAC,EACJ,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACvB,CAAC,CAAC,OAAO,EAAE,OAAO,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IACvF,cAAc,CAAC,EAAE,uBAAuB,CAAC,eAAe,CAAC,CAAC;CAC3D;AAED,MAAM,WAAW,0BAA2B,SAAQ,kBAAkB;IACpE,KAAK,EAAE,KAAK,CAAC;CACd;AAUD,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,kBAAkB,GAC3B,CAAC,YAAY,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC;AAChD,wBAAgB,iBAAiB,CAC/B,MAAM,EAAE,0BAA0B,EAClC,OAAO,CAAC,EAAE,kBAAkB,GAC3B,CAAC,YAAY,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC"}
+{"version":3,"file":"ag-ui-handler.d.ts","sourceRoot":"","sources":["../../../src/src/agent/ag-ui-handler.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,EAGL,KAAK,uBAAuB,EAC7B,MAAM,oBAAoB,CAAC;AAkB5B,OAAO,EACL,KAAK,eAAe,EACpB,qBAAqB,EACrB,KAAK,gBAAgB,EACrB,sBAAsB,EACtB,KAAK,WAAW,EAChB,iBAAiB,GAClB,MAAM,yBAAyB,CAAC;AAQjC,KAAK,eAAe,GAAG;IAAE,MAAM,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,OAAO,CAAA;CAAE,CAAC;AAkT7D,MAAM,WAAW,kBAAkB;IACjC,OAAO,CAAC,EACJ,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACvB,CAAC,CAAC,OAAO,EAAE,OAAO,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IACvF,cAAc,CAAC,EAAE,uBAAuB,CAAC,eAAe,CAAC,CAAC;CAC3D;AAED,MAAM,WAAW,0BAA2B,SAAQ,kBAAkB;IACpE,KAAK,EAAE,KAAK,CAAC;CACd;AAUD,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,kBAAkB,GAC3B,CAAC,YAAY,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC;AAChD,wBAAgB,iBAAiB,CAC/B,MAAM,EAAE,0BAA0B,EAClC,OAAO,CAAC,EAAE,kBAAkB,GAC3B,CAAC,YAAY,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC"}

package/esm/src/agent/ag-ui-handler.js

@@ -7,144 +7,13 @@ import { SKILL_TOOL_IDS } from "../skill/types.js";
 import { toolRegistry } from "../tool/index.js";
 import { createStreamTransformState, finalizeRunEvents, formatAgUiEvent, mapRuntimeEventToAgUi, } from "../internal-agents/ag-ui-sse.js";
 import { streamDataStreamEvents } from "./data-stream.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_TEXT_PART_LENGTH = 10_000;
-const MAX_MESSAGES_PER_REQUEST = 100;
-const encoder = new TextEncoder();
+import { normalizeAgUiMessages, parseAgUiRequestOrError, } from "./ag-ui-host-support.js";
+export { AgUiContextItemSchema, AgUiInjectedToolSchema, AgUiRequestSchema, } from "./ag-ui-host-support.js";
 const AG_UI_HEADERS = {
     "Content-Type": "text/event-stream",
     "Cache-Control": "no-cache",
     Connection: "keep-alive",
 };
-function isWithinJsonSizeLimit(value, maxBytes) {
-    try {
-        return encoder.encode(JSON.stringify(value)).byteLength <= maxBytes;
-    }
-    catch {
-        return false;
-    }
-}
-const AgUiRunIdSchema = z.string().min(1).max(128).regex(AGENT_ID_PATTERN);
-const AgUiInjectedToolSchema = z.object({
-    name: z.string().min(1).max(128),
-    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" }),
-});
-const AgUiContextItemSchema = 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 AgUiMessagePartSchema = z.object({ type: z.string().min(1) }).passthrough();
-const AgUiMessageSchema = z.object({
-    id: z.string().min(1),
-    role: z.enum(["user", "assistant", "system", "tool"]),
-    parts: z.array(AgUiMessagePartSchema).default([]),
-    metadata: z.record(z.string(), z.unknown()).optional(),
-    createdAt: z.string().optional(),
-});
-export const AgUiRequestSchema = z.object({
-    threadId: z.string().uuid().optional(),
-    runId: AgUiRunIdSchema.optional(),
-    messages: z.array(AgUiMessageSchema).min(1).max(MAX_MESSAGES_PER_REQUEST),
-    tools: z.array(AgUiInjectedToolSchema).max(50).default([]),
-    context: z.array(AgUiContextItemSchema).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" }),
-    model: z.string().optional(),
-    maxOutputTokens: z.number().int().positive().optional(),
-});
-function isRecord(value) {
-    return typeof value === "object" && value !== null && !Array.isArray(value);
-}
-function normalizeToolArgs(part) {
-    if (isRecord(part.args))
-        return part.args;
-    if (isRecord(part.input))
-        return part.input;
-    return {};
-}
-function normalizeMessagePart(part) {
-    if (part.type === "text" && typeof part.text === "string" &&
-        part.text.length <= MAX_TEXT_PART_LENGTH) {
-        return { type: "text", text: part.text };
-    }
-    if (part.type === "tool_call" && typeof part.id === "string" && typeof part.name === "string") {
-        return {
-            type: "tool-call",
-            toolCallId: part.id,
-            toolName: part.name,
-            args: normalizeToolArgs(part),
-        };
-    }
-    if (part.type === "tool-call" &&
-        typeof part.toolCallId === "string" &&
-        typeof part.toolName === "string") {
-        return {
-            type: "tool-call",
-            toolCallId: part.toolCallId,
-            toolName: part.toolName,
-            args: normalizeToolArgs(part),
-        };
-    }
-    if (typeof part.type === "string" &&
-        part.type.startsWith("tool-") &&
-        part.type !== "tool-result" &&
-        typeof part.toolCallId === "string" &&
-        typeof part.toolName === "string") {
-        return {
-            type: part.type,
-            toolCallId: part.toolCallId,
-            toolName: part.toolName,
-            args: normalizeToolArgs(part),
-        };
-    }
-    if (part.type === "tool_result" && typeof part.tool_call_id === "string") {
-        return {
-            type: "tool-result",
-            toolCallId: part.tool_call_id,
-            toolName: typeof part.tool_name === "string" ? part.tool_name : "unknown",
-            result: "output" in part ? part.output : undefined,
-        };
-    }
-    if (part.type === "tool-result" && typeof part.toolCallId === "string") {
-        return {
-            type: "tool-result",
-            toolCallId: part.toolCallId,
-            toolName: typeof part.toolName === "string" ? part.toolName : "unknown",
-            result: "result" in part ? part.result : undefined,
-        };
-    }
-    return null;
-}
-function normalizeMessages(messages) {
-    return messages.map((message) => ({
-        id: message.id,
-        role: message.role,
-        parts: message.parts
-            .map((part) => normalizeMessagePart(part))
-            .filter((part) => part !== null),
-        ...(message.createdAt ? { timestamp: Date.parse(message.createdAt) || undefined } : {}),
-        ...(message.metadata ? { metadata: message.metadata } : {}),
-    }));
-}
 function isRequest(obj) {
     return (typeof obj === "object" &&
         obj !== null &&
@@ -270,7 +139,7 @@ async function createAgUiDirectStreamResponse(agent, request, baseContext) {
     const runId = request.runId ?? generateRunId();
     await agent.clearMemory();
     const result = await agent.stream({
-        messages: normalizeMessages(request.messages),
+        messages: normalizeAgUiMessages(request.messages),
         context: buildStreamContext(request, baseContext, threadId, runId),
         ...(request.model ? { model: request.model } : {}),
         ...(request.maxOutputTokens ? { maxOutputTokens: request.maxOutputTokens } : {}),
@@ -339,7 +208,7 @@ async function createAgUiInjectedToolsStreamResponse(agent, request, baseContext
     });
     let upstreamBody;
     try {
-        upstreamBody = await runtime.stream(normalizeMessages(request.messages), buildStreamContext(request, baseContext, threadId, runId), undefined, request.model, request.maxOutputTokens);
+        upstreamBody = await runtime.stream(normalizeAgUiMessages(request.messages), buildStreamContext(request, baseContext, threadId, runId), undefined, request.model, request.maxOutputTokens);
     }
     catch (error) {
         sessionManager.failRun(runId);
@@ -392,7 +261,10 @@ export function createAgUiHandler(agentIdOrConfig, options) {
             return Response.json({ error: "Agent not found" }, { status: 404 });
         }
         try {
-            const parsed = AgUiRequestSchema.parse(await request.json());
+            const parsed = await parseAgUiRequestOrError(request);
+            if (parsed instanceof Response) {
+                return parsed;
+            }
             if (parsed.tools.length > 0) {
                 if (!options?.sessionManager) {
                     return Response.json({

package/esm/src/agent/ag-ui-host-support.d.ts

@@ -0,0 +1,76 @@
+import { z } from "zod";
+import type { Message } from "./types.js";
+export interface AgUiSseEvent {
+    event: string;
+    payload: Record<string, unknown>;
+}
+export declare const AgUiInjectedToolSchema: z.ZodObject<{
+    name: z.ZodString;
+    description: z.ZodOptional<z.ZodString>;
+    parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+}, z.core.$strip>;
+export declare const AgUiContextItemSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
+    type: z.ZodLiteral<"text">;
+    title: z.ZodOptional<z.ZodString>;
+    text: z.ZodString;
+}, z.core.$strip>, z.ZodObject<{
+    type: z.ZodLiteral<"json">;
+    title: z.ZodOptional<z.ZodString>;
+    data: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+}, z.core.$strip>, z.ZodObject<{
+    type: z.ZodLiteral<"resource">;
+    title: z.ZodOptional<z.ZodString>;
+    uri: z.ZodString;
+    mimeType: z.ZodOptional<z.ZodString>;
+    text: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>], "type">;
+export declare const AgUiRequestSchema: z.ZodObject<{
+    threadId: z.ZodOptional<z.ZodString>;
+    runId: z.ZodOptional<z.ZodString>;
+    messages: z.ZodArray<z.ZodObject<{
+        id: z.ZodString;
+        role: z.ZodEnum<{
+            tool: "tool";
+            user: "user";
+            assistant: "assistant";
+            system: "system";
+        }>;
+        parts: z.ZodDefault<z.ZodArray<z.ZodObject<{
+            type: z.ZodString;
+        }, z.core.$loose>>>;
+        metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+        createdAt: z.ZodOptional<z.ZodString>;
+    }, z.core.$strip>>;
+    tools: z.ZodDefault<z.ZodArray<z.ZodObject<{
+        name: z.ZodString;
+        description: z.ZodOptional<z.ZodString>;
+        parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+    }, z.core.$strip>>>;
+    context: z.ZodDefault<z.ZodArray<z.ZodDiscriminatedUnion<[z.ZodObject<{
+        type: z.ZodLiteral<"text">;
+        title: z.ZodOptional<z.ZodString>;
+        text: z.ZodString;
+    }, z.core.$strip>, z.ZodObject<{
+        type: z.ZodLiteral<"json">;
+        title: z.ZodOptional<z.ZodString>;
+        data: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+    }, z.core.$strip>, z.ZodObject<{
+        type: z.ZodLiteral<"resource">;
+        title: z.ZodOptional<z.ZodString>;
+        uri: z.ZodString;
+        mimeType: z.ZodOptional<z.ZodString>;
+        text: z.ZodOptional<z.ZodString>;
+    }, z.core.$strip>], "type">>>;
+    forwardedProps: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+    model: z.ZodOptional<z.ZodString>;
+    maxOutputTokens: z.ZodOptional<z.ZodNumber>;
+}, z.core.$strip>;
+export type AgUiInjectedTool = z.infer<typeof AgUiInjectedToolSchema>;
+export type AgUiContextItem = z.infer<typeof AgUiContextItemSchema>;
+export type AgUiRequest = z.infer<typeof AgUiRequestSchema>;
+export declare function parseAgUiRequest(request: Request): Promise<AgUiRequest>;
+export declare function parseAgUiRequestOrError(request: Request): Promise<AgUiRequest | Response>;
+export declare function normalizeAgUiMessages(messages: AgUiRequest["messages"]): Message[];
+export declare function createAgUiRunErrorEvent(message: string, code?: string): AgUiSseEvent;
+export declare function createAgUiSseErrorResponse(event: AgUiSseEvent, status: number): Response;
+//# sourceMappingURL=ag-ui-host-support.d.ts.map

package/esm/src/agent/ag-ui-host-support.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ag-ui-host-support.d.ts","sourceRoot":"","sources":["../../../src/src/agent/ag-ui-host-support.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAY1C,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAClC;AAgBD,eAAO,MAAM,sBAAsB;;;;iBAOjC,CAAC;AAEH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;2BAqBhC,CAAC;AAYH,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAe5B,CAAC;AAEH,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AACtE,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AACpE,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAC;AA0E5D,wBAAsB,gBAAgB,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,CAE7E;AAED,wBAAsB,uBAAuB,CAC3C,OAAO,EAAE,OAAO,GACf,OAAO,CAAC,WAAW,GAAG,QAAQ,CAAC,CA6BjC;AAED,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,WAAW,CAAC,UAAU,CAAC,GAAG,OAAO,EAAE,CAUlF;AAED,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,YAAY,CAQpF;AAED,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,MAAM,GAAG,QAAQ,CAUxF"}