veryfront 0.1.219 → 0.1.221

package/esm/src/agent/ag-ui-host-support.js

@@ -0,0 +1,182 @@
+import { z } from "zod";
+import { formatAgUiEvent } from "../internal-agents/ag-ui-sse.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();
+const decoder = new TextDecoder();
+function isWithinJsonSizeLimit(value, maxBytes) {
+    try {
+        return encoder.encode(JSON.stringify(value)).byteLength <= maxBytes;
+    }
+    catch {
+        return false;
+    }
+}
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+const AgUiRunIdSchema = z.string().min(1).max(128).regex(AGENT_ID_PATTERN);
+export 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" }),
+});
+export 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 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;
+}
+export async function parseAgUiRequest(request) {
+    return AgUiRequestSchema.parse(await request.json());
+}
+export async function parseAgUiRequestOrError(request) {
+    try {
+        return await parseAgUiRequest(request);
+    }
+    catch (error) {
+        if (error instanceof z.ZodError) {
+            return Response.json({
+                error: "Invalid AG-UI request",
+                details: error.issues.map((issue) => ({
+                    path: issue.path,
+                    message: issue.message,
+                })),
+            }, { status: 400 });
+        }
+        if (error instanceof SyntaxError || error instanceof TypeError) {
+            return Response.json({
+                error: "Invalid AG-UI request",
+                details: [{ path: [], message: "Malformed JSON request body" }],
+            }, { status: 400 });
+        }
+        throw error;
+    }
+}
+export function normalizeAgUiMessages(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 } : {}),
+    }));
+}
+export function createAgUiRunErrorEvent(message, code) {
+    return {
+        event: "RunError",
+        payload: {
+            message,
+            ...(code ? { code } : {}),
+        },
+    };
+}
+export function createAgUiSseErrorResponse(event, status) {
+    return new Response(decoder.decode(formatAgUiEvent(event.event, event.payload)), {
+        status,
+        headers: {
+            "Content-Type": "text/event-stream; charset=utf-8",
+            "Cache-Control": "no-cache, no-transform",
+            "Connection": "keep-alive",
+            "X-Accel-Buffering": "no",
+        },
+    });
+}

package/esm/src/agent/index.d.ts

@@ -90,6 +90,7 @@ export { mergeToolCallInput, mergeToolInputDelta, parseDataStreamSseEvents, pars
 export { expandAllowedRemoteToolNames, getProviderNativeToolNames, type ProviderNativeToolInventoryOptions, } from "./provider-native-tool-inventory.js";
 export { type AgUiDetachedStartAccepted, AgUiDetachedStartAcceptedSchema, type AgUiDetachedStartHandlerOptions, type AgUiDetachedStartRequest, AgUiDetachedStartRequestSchema, createAgUiDetachedStartHandler, } from "./ag-ui-detached-start.js";
 export { type AgUiCancelHandlerOptions, type AgUiResumeHandlerOptions, type AgUiResumeSignal, AgUiResumeSignalSchema, createAgUiCancelHandler, createAgUiResumeHandler, } from "./ag-ui-run-control.js";
+export { type AgUiSseEvent, createAgUiRunErrorEvent, createAgUiSseErrorResponse, normalizeAgUiMessages, parseAgUiRequest, parseAgUiRequestOrError, } from "./ag-ui-host-support.js";
 export { type AgUiContextItem, type AgUiHandlerConfigWithAgent, type AgUiHandlerOptions, type AgUiInjectedTool, type AgUiRequest, AgUiRequestSchema, createAgUiHandler, } from "./ag-ui-handler.js";
 export { type HumanInputField, type HumanInputFieldInput, HumanInputFieldSchema, type HumanInputOption, HumanInputOptionSchema, type HumanInputPendingRequest, HumanInputPendingRequestSchema, type HumanInputRequest, type HumanInputRequestInput, HumanInputRequestSchema, type HumanInputResult, HumanInputResultSchema, HumanInputResumeError, InvalidHumanInputResultError, waitForHumanInput, type WaitForHumanInputOptions, } from "./human-input.js";
 export { type ChatHandlerBeforeStream, type ChatHandlerBeforeStreamContext, type ChatHandlerBeforeStreamResult, type ChatHandlerConfigWithAgent, type ChatHandlerMessageInput, type ChatHandlerOptions, createChatHandler, } from "./chat-handler.js";

package/esm/src/agent/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/src/agent/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+EG;AACH,OAAO,yBAAyB,CAAC;AAGjC,YAAY,EACV,KAAK,EACL,WAAW,EACX,YAAY,EACZ,eAAe,EACf,aAAa,EACb,WAAW,EACX,iBAAiB,EACjB,eAAe,EACf,gBAAgB,EAChB,UAAU,EACV,YAAY,EACZ,OAAO,IAAI,YAAY,EACvB,WAAW,EACX,aAAa,EACb,WAAW,EACX,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EACnB,sBAAsB,EACtB,oBAAoB,EACpB,mBAAmB,EACnB,oBAAoB,EACpB,cAAc,EACd,QAAQ,EACR,YAAY,EACZ,oBAAoB,EACpB,qBAAqB,EACrB,cAAc,GACf,MAAM,YAAY,CAAC;AAEpB,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAEnF,OAAO,EACL,YAAY,EACZ,kBAAkB,EAClB,YAAY,EACZ,iBAAiB,EACjB,KAAK,MAAM,EACX,KAAK,iBAAiB,EACtB,KAAK,WAAW,EAChB,KAAK,WAAW,EAChB,WAAW,EACX,KAAK,iBAAiB,EACtB,aAAa,GACd,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EACL,WAAW,EACX,cAAc,EACd,QAAQ,EACR,gBAAgB,EAChB,cAAc,EACd,aAAa,EACb,KAAK,cAAc,EACnB,KAAK,cAAc,EACnB,KAAK,YAAY,GAClB,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAAE,KAAK,EAAE,MAAM,cAAc,CAAC;AACrC,OAAO,EACL,KAAK,sBAAsB,EAC3B,4BAA4B,EAC5B,KAAK,uBAAuB,EAC5B,6BAA6B,EAC7B,KAAK,kBAAkB,EACvB,wBAAwB,EACxB,KAAK,kBAAkB,EACvB,wBAAwB,GACzB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,KAAK,uBAAuB,EAC5B,KAAK,uBAAuB,EAC5B,KAAK,8BAA8B,EACnC,KAAK,sBAAsB,EAC3B,6BAA6B,EAC7B,yBAAyB,EACzB,wCAAwC,GACzC,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,kBAAkB,EAClB,mBAAmB,EACnB,wBAAwB,EACxB,oBAAoB,EACpB,sBAAsB,EACtB,kCAAkC,GACnC,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,4BAA4B,EAC5B,0BAA0B,EAC1B,KAAK,kCAAkC,GACxC,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EACL,KAAK,yBAAyB,EAC9B,+BAA+B,EAC/B,KAAK,+BAA+B,EACpC,KAAK,wBAAwB,EAC7B,8BAA8B,EAC9B,8BAA8B,GAC/B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,KAAK,wBAAwB,EAC7B,KAAK,wBAAwB,EAC7B,KAAK,gBAAgB,EACrB,sBAAsB,EACtB,uBAAuB,EACvB,uBAAuB,GACxB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,KAAK,eAAe,EACpB,KAAK,0BAA0B,EAC/B,KAAK,kBAAkB,EACvB,KAAK,gBAAgB,EACrB,KAAK,WAAW,EAChB,iBAAiB,EACjB,iBAAiB,GAClB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,KAAK,eAAe,EACpB,KAAK,oBAAoB,EACzB,qBAAqB,EACrB,KAAK,gBAAgB,EACrB,sBAAsB,EACtB,KAAK,wBAAwB,EAC7B,8BAA8B,EAC9B,KAAK,iBAAiB,EACtB,KAAK,sBAAsB,EAC3B,uBAAuB,EACvB,KAAK,gBAAgB,EACrB,sBAAsB,EACtB,qBAAqB,EACrB,4BAA4B,EAC5B,iBAAiB,EACjB,KAAK,wBAAwB,GAC9B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,KAAK,uBAAuB,EAC5B,KAAK,8BAA8B,EACnC,KAAK,6BAA6B,EAClC,KAAK,0BAA0B,EAC/B,KAAK,uBAAuB,EAC5B,KAAK,kBAAkB,EACvB,iBAAiB,GAClB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,YAAY,EACZ,qBAAqB,EACrB,iBAAiB,EACjB,iBAAiB,EACjB,uBAAuB,EACvB,KAAK,8BAA8B,EACnC,KAAK,gBAAgB,EACrB,KAAK,wBAAwB,EAC7B,iBAAiB,EACjB,mBAAmB,GACpB,MAAM,oBAAoB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/src/agent/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+EG;AACH,OAAO,yBAAyB,CAAC;AAGjC,YAAY,EACV,KAAK,EACL,WAAW,EACX,YAAY,EACZ,eAAe,EACf,aAAa,EACb,WAAW,EACX,iBAAiB,EACjB,eAAe,EACf,gBAAgB,EAChB,UAAU,EACV,YAAY,EACZ,OAAO,IAAI,YAAY,EACvB,WAAW,EACX,aAAa,EACb,WAAW,EACX,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EACnB,sBAAsB,EACtB,oBAAoB,EACpB,mBAAmB,EACnB,oBAAoB,EACpB,cAAc,EACd,QAAQ,EACR,YAAY,EACZ,oBAAoB,EACpB,qBAAqB,EACrB,cAAc,GACf,MAAM,YAAY,CAAC;AAEpB,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAEnF,OAAO,EACL,YAAY,EACZ,kBAAkB,EAClB,YAAY,EACZ,iBAAiB,EACjB,KAAK,MAAM,EACX,KAAK,iBAAiB,EACtB,KAAK,WAAW,EAChB,KAAK,WAAW,EAChB,WAAW,EACX,KAAK,iBAAiB,EACtB,aAAa,GACd,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EACL,WAAW,EACX,cAAc,EACd,QAAQ,EACR,gBAAgB,EAChB,cAAc,EACd,aAAa,EACb,KAAK,cAAc,EACnB,KAAK,cAAc,EACnB,KAAK,YAAY,GAClB,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAAE,KAAK,EAAE,MAAM,cAAc,CAAC;AACrC,OAAO,EACL,KAAK,sBAAsB,EAC3B,4BAA4B,EAC5B,KAAK,uBAAuB,EAC5B,6BAA6B,EAC7B,KAAK,kBAAkB,EACvB,wBAAwB,EACxB,KAAK,kBAAkB,EACvB,wBAAwB,GACzB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,KAAK,uBAAuB,EAC5B,KAAK,uBAAuB,EAC5B,KAAK,8BAA8B,EACnC,KAAK,sBAAsB,EAC3B,6BAA6B,EAC7B,yBAAyB,EACzB,wCAAwC,GACzC,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,kBAAkB,EAClB,mBAAmB,EACnB,wBAAwB,EACxB,oBAAoB,EACpB,sBAAsB,EACtB,kCAAkC,GACnC,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,4BAA4B,EAC5B,0BAA0B,EAC1B,KAAK,kCAAkC,GACxC,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EACL,KAAK,yBAAyB,EAC9B,+BAA+B,EAC/B,KAAK,+BAA+B,EACpC,KAAK,wBAAwB,EAC7B,8BAA8B,EAC9B,8BAA8B,GAC/B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,KAAK,wBAAwB,EAC7B,KAAK,wBAAwB,EAC7B,KAAK,gBAAgB,EACrB,sBAAsB,EACtB,uBAAuB,EACvB,uBAAuB,GACxB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,KAAK,YAAY,EACjB,uBAAuB,EACvB,0BAA0B,EAC1B,qBAAqB,EACrB,gBAAgB,EAChB,uBAAuB,GACxB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,KAAK,eAAe,EACpB,KAAK,0BAA0B,EAC/B,KAAK,kBAAkB,EACvB,KAAK,gBAAgB,EACrB,KAAK,WAAW,EAChB,iBAAiB,EACjB,iBAAiB,GAClB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,KAAK,eAAe,EACpB,KAAK,oBAAoB,EACzB,qBAAqB,EACrB,KAAK,gBAAgB,EACrB,sBAAsB,EACtB,KAAK,wBAAwB,EAC7B,8BAA8B,EAC9B,KAAK,iBAAiB,EACtB,KAAK,sBAAsB,EAC3B,uBAAuB,EACvB,KAAK,gBAAgB,EACrB,sBAAsB,EACtB,qBAAqB,EACrB,4BAA4B,EAC5B,iBAAiB,EACjB,KAAK,wBAAwB,GAC9B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,KAAK,uBAAuB,EAC5B,KAAK,8BAA8B,EACnC,KAAK,6BAA6B,EAClC,KAAK,0BAA0B,EAC/B,KAAK,uBAAuB,EAC5B,KAAK,kBAAkB,EACvB,iBAAiB,GAClB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,YAAY,EACZ,qBAAqB,EACrB,iBAAiB,EACjB,iBAAiB,EACjB,uBAAuB,EACvB,KAAK,8BAA8B,EACnC,KAAK,gBAAgB,EACrB,KAAK,wBAAwB,EAC7B,iBAAiB,EACjB,mBAAmB,GACpB,MAAM,oBAAoB,CAAC"}

package/esm/src/agent/index.js

@@ -89,6 +89,7 @@ export { mergeToolCallInput, mergeToolInputDelta, parseDataStreamSseEvents, pars
 export { expandAllowedRemoteToolNames, getProviderNativeToolNames, } from "./provider-native-tool-inventory.js";
 export { AgUiDetachedStartAcceptedSchema, AgUiDetachedStartRequestSchema, createAgUiDetachedStartHandler, } from "./ag-ui-detached-start.js";
 export { AgUiResumeSignalSchema, createAgUiCancelHandler, createAgUiResumeHandler, } from "./ag-ui-run-control.js";
+export { createAgUiRunErrorEvent, createAgUiSseErrorResponse, normalizeAgUiMessages, parseAgUiRequest, parseAgUiRequestOrError, } from "./ag-ui-host-support.js";
 export { AgUiRequestSchema, createAgUiHandler, } from "./ag-ui-handler.js";
 export { HumanInputFieldSchema, HumanInputOptionSchema, HumanInputPendingRequestSchema, HumanInputRequestSchema, HumanInputResultSchema, HumanInputResumeError, InvalidHumanInputResultError, waitForHumanInput, } from "./human-input.js";
 export { createChatHandler, } from "./chat-handler.js";

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

@@ -1,2 +1,2 @@
-export declare const VERSION = "0.1.219";
+export declare const VERSION = "0.1.221";
 //# 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.219";
+export const VERSION = "0.1.221";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "veryfront",
-  "version": "0.1.219",
+  "version": "0.1.221",
   "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.219",
+  "version": "0.1.221",
   "license": "Apache-2.0",
   "nodeModulesDir": "auto",
   "workspace": [

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

@@ -3,18 +3,20 @@ import { INVALID_ARGUMENT } from "../errors/index.js";
 import { SKILL_TOOL_IDS } from "../skill/types.js";
 import { type Tool, toolRegistry } from "../tool/index.js";
 import { streamDataStreamEvents } from "./data-stream.js";
-import { type AgUiInjectedTool, type AgUiRequest, AgUiRequestSchema } from "./ag-ui-handler.js";
+import {
+  type AgUiInjectedTool,
+  AgUiRequestSchema,
+  normalizeAgUiMessages,
+} from "./ag-ui-host-support.js";
 import {
   AgentRuntime,
   RunAlreadyExistsError,
   type RunResumeSessionManager,
 } from "./runtime/index.js";
-import type { Agent, Message } from "./types.js";
+import type { Agent } from "./types.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;
-
 type AgUiResumeValue = {
   result: unknown;
   isError: boolean;
@@ -26,94 +28,6 @@ type AgUiContextValue =
 
 type AgUiRuntimePart = Record<string, unknown> & { type: string };
 
-function isRecord(value: unknown): value is Record<string, unknown> {
-  return typeof value === "object" && value !== null && !Array.isArray(value);
-}
-
-function normalizeToolArgs(part: Record<string, unknown>): Record<string, unknown> {
-  if (isRecord(part.args)) return part.args;
-  if (isRecord(part.input)) return part.input;
-  return {};
-}
-
-function normalizeMessagePart(part: Record<string, unknown>): Message["parts"][number] | null {
-  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: AgUiRequest["messages"]): Message[] {
-  return messages.map((message) => ({
-    id: message.id,
-    role: message.role,
-    parts: message.parts
-      .map((part) => normalizeMessagePart(part))
-      .filter((part): part is Message["parts"][number] => part !== null),
-    ...(message.createdAt ? { timestamp: Date.parse(message.createdAt) || undefined } : {}),
-    ...(message.metadata ? { metadata: message.metadata } : {}),
-  }));
-}
-
 function isRequest(obj: unknown): obj is Request {
   return (
     typeof obj === "object" &&
@@ -267,10 +181,22 @@ export const AgUiDetachedStartAcceptedSchema = z.object({
 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;
@@ -285,9 +211,46 @@ export interface AgUiDetachedStartHandlerOptions {
   onError?: (input: { runId: string; threadId: string; error: unknown }) => Promise<void> | void;
 }
 
+export type AgUiDetachedStartHandlerOptions =
+  | (AgUiDetachedStartHandlerOptionsBase & { agent: Agent })
+  | (AgUiDetachedStartHandlerOptionsBase & {
+    agent?: undefined;
+    startDetachedExecution: AgUiDetachedExecutionStarter;
+  });
+
+async function startDefaultDetachedExecution(input: {
+  agent: Agent;
+  request: AgUiDetachedStartRequest;
+  context: Record<string, unknown>;
+  abortSignal: AbortSignal;
+  sessionManager: RunResumeSessionManager<AgUiResumeValue>;
+}): Promise<void> {
+  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: AgUiDetachedStartHandlerOptions,
 ): (requestOrCtx: unknown) => Promise<Response> {
+  if (!options.agent && !options.startDetachedExecution) {
+    throw new Error(
+      "Detached AG-UI start requires either an agent or startDetachedExecution handler.",
+    );
+  }
+
   return async function POST(requestOrCtx: unknown): Promise<Response> {
     const request = extractRequest(requestOrCtx);
 
@@ -301,20 +264,6 @@ export function createAgUiDetachedStartHandler(
           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,
@@ -323,7 +272,28 @@ export function createAgUiDetachedStartHandler(
 
         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,