veryfront 0.1.222 → 0.1.223

package/esm/deno.js

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

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

@@ -0,0 +1,29 @@
+import { type RunResumeSessionManager } from "./runtime/index.js";
+import type { Agent } from "./types.js";
+import { type AgUiRuntimeRequest } from "./runtime-ag-ui-contract.js";
+type AgUiResumeValue = {
+    result: unknown;
+    isError: boolean;
+};
+export interface AgUiRuntimeHandlerExecuteInput {
+    request: Request;
+    agUiInput: AgUiRuntimeRequest;
+    context: Record<string, unknown>;
+    createDefaultResponse?: () => Promise<Response>;
+}
+export type AgUiRuntimeHandlerExecute = (input: AgUiRuntimeHandlerExecuteInput) => Promise<Response> | Response;
+export interface AgUiRuntimeHandlerOptions {
+    context?: Record<string, unknown> | ((request: Request) => Record<string, unknown> | Promise<Record<string, unknown>>);
+    sessionManager?: RunResumeSessionManager<AgUiResumeValue>;
+    execute?: AgUiRuntimeHandlerExecute;
+}
+export interface AgUiRuntimeHandlerConfigWithAgent extends AgUiRuntimeHandlerOptions {
+    agent: Agent;
+}
+export type AgUiRuntimeHandlerConfig = AgUiRuntimeHandlerConfigWithAgent | (AgUiRuntimeHandlerOptions & {
+    agent?: undefined;
+    execute: AgUiRuntimeHandlerExecute;
+});
+export declare function createAgUiRuntimeHandler(config: AgUiRuntimeHandlerConfig): (requestOrCtx: unknown) => Promise<Response>;
+export {};
+//# sourceMappingURL=ag-ui-runtime-handler.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ag-ui-runtime-handler.d.ts","sourceRoot":"","sources":["../../../src/src/agent/ag-ui-runtime-handler.ts"],"names":[],"mappings":"AAIA,OAAO,EAGL,KAAK,uBAAuB,EAC7B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,EAEL,KAAK,kBAAkB,EAExB,MAAM,6BAA6B,CAAC;AAgBrC,KAAK,eAAe,GAAG;IAAE,MAAM,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,OAAO,CAAA;CAAE,CAAC;AA2T7D,MAAM,WAAW,8BAA8B;IAC7C,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,kBAAkB,CAAC;IAC9B,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC,qBAAqB,CAAC,EAAE,MAAM,OAAO,CAAC,QAAQ,CAAC,CAAC;CACjD;AAED,MAAM,MAAM,yBAAyB,GAAG,CACtC,KAAK,EAAE,8BAA8B,KAClC,OAAO,CAAC,QAAQ,CAAC,GAAG,QAAQ,CAAC;AAElC,MAAM,WAAW,yBAAyB;IACxC,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;IAC1D,OAAO,CAAC,EAAE,yBAAyB,CAAC;CACrC;AAED,MAAM,WAAW,iCAAkC,SAAQ,yBAAyB;IAClF,KAAK,EAAE,KAAK,CAAC;CACd;AAED,MAAM,MAAM,wBAAwB,GAChC,iCAAiC,GACjC,CAAC,yBAAyB,GAAG;IAAE,KAAK,CAAC,EAAE,SAAS,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAA;CAAE,CAAC,CAAC;AAE5F,wBAAgB,wBAAwB,CACtC,MAAM,EAAE,wBAAwB,GAC/B,CAAC,YAAY,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CA8E9C"}

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

@@ -0,0 +1,291 @@
+import { z } from "zod";
+import { INVALID_ARGUMENT } from "../errors/index.js";
+import { SKILL_TOOL_IDS } from "../skill/types.js";
+import { toolRegistry } from "../tool/index.js";
+import { AgentRuntime, RunAlreadyExistsError, } from "./runtime/index.js";
+import { parseAgUiRuntimeRequestOrError, } from "./runtime-ag-ui-contract.js";
+import { normalizeAgUiRuntimeMessages } from "./ag-ui-runtime-support.js";
+import { createStreamTransformState, finalizeRunEvents, formatAgUiEvent, mapRuntimeEventToAgUi, } from "../internal-agents/ag-ui-sse.js";
+import { streamDataStreamEvents } from "./data-stream.js";
+const AG_UI_HEADERS = {
+    "Content-Type": "text/event-stream",
+    "Cache-Control": "no-cache",
+    Connection: "keep-alive",
+};
+function isRequest(obj) {
+    return (typeof obj === "object" &&
+        obj !== null &&
+        "json" in obj &&
+        typeof obj.json === "function" &&
+        "url" in obj &&
+        typeof obj.url === "string" &&
+        "method" in obj &&
+        typeof obj.method === "string");
+}
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function extractRequest(requestOrCtx) {
+    if (isRequest(requestOrCtx))
+        return requestOrCtx;
+    if (typeof requestOrCtx === "object" && requestOrCtx !== null && "request" in requestOrCtx) {
+        const candidate = requestOrCtx.request;
+        if (isRequest(candidate))
+            return candidate;
+    }
+    throw INVALID_ARGUMENT.create({
+        detail: "Invalid handler argument: expected Request or APIContext",
+    });
+}
+function buildStreamContext(request, baseContext) {
+    return {
+        ...baseContext,
+        threadId: request.threadId,
+        runId: request.runId,
+        agUi: {
+            context: request.context,
+            forwardedProps: request.forwardedProps,
+        },
+    };
+}
+function closeController(controller) {
+    try {
+        controller.close();
+    }
+    catch {
+        return;
+    }
+}
+function enqueueEvent(controller, event, payload) {
+    try {
+        controller.enqueue(formatAgUiEvent(event, payload));
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+async function createAgUiRuntimeStreamResponse(options) {
+    const { agentId, request, upstreamBody, upstreamStatus, upstreamStatusText, onFinish, onError, onToolCallSeen, } = options;
+    const stream = new ReadableStream({
+        start: async (controller) => {
+            const state = createStreamTransformState();
+            const prepareToolResultIfNeeded = (event, payload) => {
+                if (event !== "ToolCallStart" && event !== "ToolCallArgs" &&
+                    event !== "ToolCallEnd") {
+                    return;
+                }
+                const toolCallId = typeof payload.toolCallId === "string" ? payload.toolCallId : null;
+                if (toolCallId) {
+                    onToolCallSeen?.(toolCallId);
+                }
+            };
+            if (!enqueueEvent(controller, "RunStarted", {
+                runId: request.runId,
+                threadId: request.threadId,
+                agentId,
+            })) {
+                return;
+            }
+            if (!enqueueEvent(controller, "StateSnapshot", {
+                snapshot: isRecord(request.state) ? request.state : {},
+            })) {
+                return;
+            }
+            if (!enqueueEvent(controller, "MessagesSnapshot", {
+                messages: normalizeAgUiRuntimeMessages(request.messages),
+            })) {
+                return;
+            }
+            try {
+                if (!upstreamBody) {
+                    for (const event of finalizeRunEvents(state, null)) {
+                        if (!enqueueEvent(controller, event.event, event.payload)) {
+                            return;
+                        }
+                    }
+                    onFinish?.();
+                    closeController(controller);
+                    return;
+                }
+                for await (const event of streamDataStreamEvents(upstreamBody)) {
+                    for (const mapped of mapRuntimeEventToAgUi(state, event)) {
+                        prepareToolResultIfNeeded(mapped.event, mapped.payload);
+                        if (!enqueueEvent(controller, mapped.event, mapped.payload)) {
+                            return;
+                        }
+                    }
+                }
+                for (const event of finalizeRunEvents(state, null)) {
+                    if (!enqueueEvent(controller, event.event, event.payload)) {
+                        return;
+                    }
+                }
+                onFinish?.();
+            }
+            catch (error) {
+                onError?.(error);
+                enqueueEvent(controller, "RunError", {
+                    message: error instanceof Error ? error.message : "Agent run failed",
+                });
+            }
+            finally {
+                closeController(controller);
+            }
+        },
+    });
+    return new Response(stream, {
+        status: upstreamStatus,
+        statusText: upstreamStatusText,
+        headers: { ...AG_UI_HEADERS },
+    });
+}
+async function createAgUiRuntimeDirectStreamResponse(agent, request, baseContext) {
+    await agent.clearMemory();
+    const result = await agent.stream({
+        messages: normalizeAgUiRuntimeMessages(request.messages),
+        context: buildStreamContext(request, baseContext),
+    });
+    const upstream = result.toDataStreamResponse();
+    return await createAgUiRuntimeStreamResponse({
+        agentId: agent.id,
+        request,
+        upstreamBody: upstream.body,
+        upstreamStatus: upstream.status,
+        upstreamStatusText: upstream.statusText,
+    });
+}
+function createInjectedAgUiTool(runId, tool, sessionManager) {
+    return {
+        id: tool.name,
+        type: "function",
+        description: tool.description ?? tool.name,
+        inputSchema: z.record(z.string(), z.unknown()),
+        inputSchemaJson: (tool.parameters ??
+            { type: "object", properties: {}, additionalProperties: true }),
+        execute: async (_input, context) => {
+            const toolCallId = typeof context?.toolCallId === "string" ? context.toolCallId : null;
+            if (!toolCallId) {
+                throw new Error(`Missing toolCallId for injected tool "${tool.name}"`);
+            }
+            sessionManager.prepareForSignal(runId, toolCallId);
+            const submitted = await sessionManager.waitForSignal(runId, toolCallId);
+            if (submitted.isError) {
+                throw new Error(typeof submitted.result === "string"
+                    ? submitted.result
+                    : JSON.stringify(submitted.result));
+            }
+            return submitted.result;
+        },
+    };
+}
+function buildMergedTools(agent, request, sessionManager) {
+    const injectedTools = Object.fromEntries(request.tools.map((tool) => [
+        tool.name,
+        createInjectedAgUiTool(request.runId, tool, sessionManager),
+    ]));
+    if (!agent.config.tools) {
+        return Object.keys(injectedTools).length > 0 ? injectedTools : undefined;
+    }
+    if (agent.config.tools === true) {
+        const merged = {};
+        for (const [toolId] of toolRegistry.getAll()) {
+            if (!agent.config.skills && SKILL_TOOL_IDS.has(toolId)) {
+                continue;
+            }
+            merged[toolId] = true;
+        }
+        return { ...merged, ...injectedTools };
+    }
+    return { ...agent.config.tools, ...injectedTools };
+}
+async function createAgUiRuntimeInjectedToolsStreamResponse(agent, request, baseContext, sessionManager) {
+    try {
+        sessionManager.startRun({ runId: request.runId, threadId: request.threadId });
+    }
+    catch (error) {
+        if (error instanceof RunAlreadyExistsError) {
+            return Response.json({ error: "Run already active" }, { status: 409 });
+        }
+        throw error;
+    }
+    const runtime = new AgentRuntime(agent.id, {
+        ...agent.config,
+        tools: buildMergedTools(agent, request, sessionManager),
+    });
+    let upstreamBody;
+    try {
+        upstreamBody = await runtime.stream(normalizeAgUiRuntimeMessages(request.messages), buildStreamContext(request, baseContext), undefined, undefined, undefined);
+    }
+    catch (error) {
+        sessionManager.failRun(request.runId);
+        throw error;
+    }
+    return await createAgUiRuntimeStreamResponse({
+        agentId: agent.id,
+        request,
+        upstreamBody,
+        upstreamStatus: 200,
+        onFinish: () => {
+            sessionManager.completeRun(request.runId);
+        },
+        onError: () => {
+            sessionManager.failRun(request.runId);
+        },
+        onToolCallSeen: (toolCallId) => {
+            sessionManager.prepareForSignal(request.runId, toolCallId);
+        },
+    });
+}
+export function createAgUiRuntimeHandler(config) {
+    if (!config.agent && !config.execute) {
+        throw new Error("createAgUiRuntimeHandler requires either an agent or an execute handler.");
+    }
+    return async function POST(requestOrCtx) {
+        const request = extractRequest(requestOrCtx);
+        try {
+            const parsed = await parseAgUiRuntimeRequestOrError(request);
+            if (parsed instanceof Response) {
+                return parsed;
+            }
+            const context = typeof config.context === "function"
+                ? await config.context(request)
+                : config.context ?? {};
+            const createDefaultResponse = config.agent
+                ? () => parsed.tools.length > 0
+                    ? config.sessionManager
+                        ? createAgUiRuntimeInjectedToolsStreamResponse(config.agent, parsed, context, config.sessionManager)
+                        : Promise.resolve(Response.json({
+                            error: "Injected AG-UI tools require a public RunResumeSessionManager on createAgUiRuntimeHandler().",
+                        }, { status: 501 }))
+                    : createAgUiRuntimeDirectStreamResponse(config.agent, parsed, context)
+                : undefined;
+            if (config.execute) {
+                return await config.execute({
+                    request,
+                    agUiInput: parsed,
+                    context,
+                    createDefaultResponse,
+                });
+            }
+            if (createDefaultResponse) {
+                return await createDefaultResponse();
+            }
+            throw new Error("createAgUiRuntimeHandler configuration became invalid during execution.");
+        }
+        catch (error) {
+            if (error instanceof z.ZodError) {
+                return Response.json({
+                    error: "Invalid AG-UI runtime request",
+                    details: error.issues.map((issue) => ({
+                        path: issue.path,
+                        message: issue.message,
+                    })),
+                }, { status: 400 });
+            }
+            return Response.json({
+                error: error instanceof Error ? error.message : "Internal server error",
+            }, { status: 500 });
+        }
+    };
+}

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

@@ -0,0 +1,4 @@
+import type { Message } from "./types.js";
+import type { AgUiRuntimeRequest } from "./runtime-ag-ui-contract.js";
+export declare function normalizeAgUiRuntimeMessages(messages: AgUiRuntimeRequest["messages"]): Message[];
+//# sourceMappingURL=ag-ui-runtime-support.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ag-ui-runtime-support.d.ts","sourceRoot":"","sources":["../../../src/src/agent/ag-ui-runtime-support.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAC1C,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAetE,wBAAgB,4BAA4B,CAC1C,QAAQ,EAAE,kBAAkB,CAAC,UAAU,CAAC,GACvC,OAAO,EAAE,CA6CX"}

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

@@ -0,0 +1,56 @@
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function parseToolArguments(serializedArguments) {
+    try {
+        const parsed = JSON.parse(serializedArguments);
+        return isRecord(parsed) ? parsed : {};
+    }
+    catch {
+        return {};
+    }
+}
+export function normalizeAgUiRuntimeMessages(messages) {
+    return messages.map((message) => {
+        const parts = [];
+        switch (message.role) {
+            case "system":
+            case "user":
+                parts.push({ type: "text", text: message.content });
+                break;
+            case "assistant":
+                if (typeof message.content === "string" && message.content.length > 0) {
+                    parts.push({ type: "text", text: message.content });
+                }
+                for (const toolCall of message.toolCalls ?? []) {
+                    parts.push({
+                        type: "tool-call",
+                        toolCallId: toolCall.id,
+                        toolName: toolCall.function.name,
+                        args: parseToolArguments(toolCall.function.arguments),
+                    });
+                }
+                break;
+            case "tool":
+                parts.push({
+                    type: "tool-result",
+                    toolCallId: message.toolCallId,
+                    toolName: "unknown",
+                    result: message.error
+                        ? {
+                            content: message.content,
+                            error: message.error,
+                        }
+                        : message.content,
+                });
+                break;
+        }
+        return {
+            id: message.id,
+            role: message.role,
+            parts,
+            ...(message.createdAt ? { timestamp: Date.parse(message.createdAt) || undefined } : {}),
+            ...(message.metadata ? { metadata: message.metadata } : {}),
+        };
+    });
+}

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

@@ -84,7 +84,9 @@ export { getTextFromParts, getToolArguments, hasArgs, hasInput } from "./types.j
 export { BufferMemory, ConversationMemory, createMemory, createRedisMemory, type Memory, type MemoryPersistence, type MemoryStats, type RedisClient, RedisMemory, type RedisMemoryConfig, SummaryMemory, } from "./memory/index.js";
 export { agentAsTool, createWorkflow, getAgent, getAgentsAsTools, getAllAgentIds, registerAgent, type WorkflowConfig, type WorkflowResult, type WorkflowStep, } from "./composition/index.js";
 export { agent } from "./factory.js";
+export { type AgUiRuntimeHandlerConfig, type AgUiRuntimeHandlerConfigWithAgent, type AgUiRuntimeHandlerExecute, type AgUiRuntimeHandlerExecuteInput, type AgUiRuntimeHandlerOptions, createAgUiRuntimeHandler, } from "./ag-ui-runtime-handler.js";
 export { type AgUiRuntimeContextItem, AgUiRuntimeContextItemSchema, type AgUiRuntimeInjectedTool, AgUiRuntimeInjectedToolSchema, type AgUiRuntimeMessage, AgUiRuntimeMessageSchema, type AgUiRuntimeRequest, AgUiRuntimeRequestSchema, parseAgUiRuntimeRequest, parseAgUiRuntimeRequestOrError, } from "./runtime-ag-ui-contract.js";
+export { normalizeAgUiRuntimeMessages } from "./ag-ui-runtime-support.js";
 export { type AgUiBrowserEncodedEvent, type AgUiBrowserEncoderState, type AgUiBrowserRunFinishedMetadata, type AgUiRuntimeStreamEvent, createAgUiBrowserEncoderState, finalizeAgUiBrowserEvents, mapRuntimeStreamEventToAgUiBrowserEvents, } from "./ag-ui-browser-encoder.js";
 export { mergeToolCallInput, mergeToolInputDelta, parseDataStreamSseEvents, parseToolInputObject, streamDataStreamEvents, stripLeadingEmptyObjectPlaceholder, } from "./data-stream.js";
 export { expandAllowedRemoteToolNames, getProviderNativeToolNames, type ProviderNativeToolInventoryOptions, } from "./provider-native-tool-inventory.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,EACxB,uBAAuB,EACvB,8BAA8B,GAC/B,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"}
+{"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,wBAAwB,EAC7B,KAAK,iCAAiC,EACtC,KAAK,yBAAyB,EAC9B,KAAK,8BAA8B,EACnC,KAAK,yBAAyB,EAC9B,wBAAwB,GACzB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,KAAK,sBAAsB,EAC3B,4BAA4B,EAC5B,KAAK,uBAAuB,EAC5B,6BAA6B,EAC7B,KAAK,kBAAkB,EACvB,wBAAwB,EACxB,KAAK,kBAAkB,EACvB,wBAAwB,EACxB,uBAAuB,EACvB,8BAA8B,GAC/B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,4BAA4B,EAAE,MAAM,4BAA4B,CAAC;AAC1E,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

@@ -83,7 +83,9 @@ export { getTextFromParts, getToolArguments, hasArgs, hasInput } from "./types.j
 export { BufferMemory, ConversationMemory, createMemory, createRedisMemory, RedisMemory, SummaryMemory, } from "./memory/index.js";
 export { agentAsTool, createWorkflow, getAgent, getAgentsAsTools, getAllAgentIds, registerAgent, } from "./composition/index.js";
 export { agent } from "./factory.js";
+export { createAgUiRuntimeHandler, } from "./ag-ui-runtime-handler.js";
 export { AgUiRuntimeContextItemSchema, AgUiRuntimeInjectedToolSchema, AgUiRuntimeMessageSchema, AgUiRuntimeRequestSchema, parseAgUiRuntimeRequest, parseAgUiRuntimeRequestOrError, } from "./runtime-ag-ui-contract.js";
+export { normalizeAgUiRuntimeMessages } from "./ag-ui-runtime-support.js";
 export { createAgUiBrowserEncoderState, finalizeAgUiBrowserEvents, mapRuntimeStreamEventToAgUiBrowserEvents, } from "./ag-ui-browser-encoder.js";
 export { mergeToolCallInput, mergeToolInputDelta, parseDataStreamSseEvents, parseToolInputObject, streamDataStreamEvents, stripLeadingEmptyObjectPlaceholder, } from "./data-stream.js";
 export { expandAllowedRemoteToolNames, getProviderNativeToolNames, } from "./provider-native-tool-inventory.js";

package/esm/src/internal-agents/run-stream.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"run-stream.d.ts","sourceRoot":"","sources":["../../../src/src/internal-agents/run-stream.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,KAAK,EACV,KAAK,YAAY,IAAI,OAAO,EAC5B,KAAK,aAAa,EAEnB,MAAM,mBAAmB,CAAC;AAW3B,OAAO,EAA0B,KAAK,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC3F,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAYxD,MAAM,WAAW,+BAA+B;IAC9C,cAAc,EAAE,sBAAsB,CAAC;IACvC,aAAa,CAAC,EAAE,CACd,KAAK,EAAE,KAAK,EACZ,WAAW,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,KAClC;QACH,MAAM,EAAE,CACN,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACjC,SAAS,CAAC,EAAE;YACV,QAAQ,CAAC,EAAE,CAAC,QAAQ,EAAE,aAAa,KAAK,IAAI,CAAC;SAC9C,EACD,aAAa,CAAC,EAAE,MAAM,EACtB,uBAAuB,CAAC,EAAE,MAAM,EAChC,WAAW,CAAC,EAAE,WAAW,KACtB,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC;KAC1C,CAAC;CACH;AAoJD,wBAAsB,gCAAgC,CACpD,KAAK,EAAE,oBAAoB,EAC3B,KAAK,EAAE,KAAK,EACZ,IAAI,EAAE,+BAA+B,GACpC,OAAO,CAAC,QAAQ,CAAC,CA8OnB"}
+{"version":3,"file":"run-stream.d.ts","sourceRoot":"","sources":["../../../src/src/internal-agents/run-stream.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,KAAK,EACV,KAAK,YAAY,IAAI,OAAO,EAC5B,KAAK,aAAa,EAEnB,MAAM,mBAAmB,CAAC;AAY3B,OAAO,EAA0B,KAAK,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC3F,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAYxD,MAAM,WAAW,+BAA+B;IAC9C,cAAc,EAAE,sBAAsB,CAAC;IACvC,aAAa,CAAC,EAAE,CACd,KAAK,EAAE,KAAK,EACZ,WAAW,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,KAClC;QACH,MAAM,EAAE,CACN,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACjC,SAAS,CAAC,EAAE;YACV,QAAQ,CAAC,EAAE,CAAC,QAAQ,EAAE,aAAa,KAAK,IAAI,CAAC;SAC9C,EACD,aAAa,CAAC,EAAE,MAAM,EACtB,uBAAuB,CAAC,EAAE,MAAM,EAChC,WAAW,CAAC,EAAE,WAAW,KACtB,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC;KAC1C,CAAC;CACH;AA4FD,wBAAsB,gCAAgC,CACpD,KAAK,EAAE,oBAAoB,EAC3B,KAAK,EAAE,KAAK,EACZ,IAAI,EAAE,+BAA+B,GACpC,OAAO,CAAC,QAAQ,CAAC,CA8OnB"}

package/esm/src/internal-agents/run-stream.js

@@ -1,4 +1,5 @@
 import { AgentRuntime, } from "../agent/index.js";
+import { normalizeAgUiRuntimeMessages } from "../agent/ag-ui-runtime-support.js";
 import { SKILL_TOOL_IDS } from "../skill/types.js";
 import { toolRegistry } from "../tool/index.js";
 import { z } from "zod";
@@ -54,59 +55,6 @@ function buildMergedTools(agent, input, sessionManager) {
 function isRecord(value) {
     return typeof value === "object" && value !== null && !Array.isArray(value);
 }
-function parseToolArguments(serializedArguments) {
-    try {
-        const parsed = JSON.parse(serializedArguments);
-        return isRecord(parsed) ? parsed : {};
-    }
-    catch {
-        return {};
-    }
-}
-function normalizeRuntimeMessages(messages) {
-    return messages.map((message) => {
-        const parts = [];
-        switch (message.role) {
-            case "system":
-            case "user":
-                parts.push({ type: "text", text: message.content });
-                break;
-            case "assistant":
-                if (typeof message.content === "string" && message.content.length > 0) {
-                    parts.push({ type: "text", text: message.content });
-                }
-                for (const toolCall of message.toolCalls ?? []) {
-                    parts.push({
-                        type: "tool-call",
-                        toolCallId: toolCall.id,
-                        toolName: toolCall.function.name,
-                        args: parseToolArguments(toolCall.function.arguments),
-                    });
-                }
-                break;
-            case "tool":
-                parts.push({
-                    type: "tool-result",
-                    toolCallId: message.toolCallId,
-                    toolName: "unknown",
-                    result: message.error
-                        ? {
-                            content: message.content,
-                            error: message.error,
-                        }
-                        : message.content,
-                });
-                break;
-        }
-        return {
-            id: message.id,
-            role: message.role,
-            parts,
-            ...(message.createdAt ? { timestamp: Date.parse(message.createdAt) || undefined } : {}),
-            ...(message.metadata ? { metadata: message.metadata } : {}),
-        };
-    });
-}
 function getAllowedRemoteToolNames(forwardedProps) {
     const runtimeOverrides = isRecord(forwardedProps?.runtimeOverrides)
         ? forwardedProps.runtimeOverrides
@@ -148,7 +96,7 @@ export async function createRuntimeAgentStreamResponse(input, agent, deps) {
     const runtime = deps.createRuntime?.(runtimeAgent, mergedTools) ??
         new AgentRuntime(runtimeAgent.id, runtimeAgent.config);
     let completedResponse = null;
-    const runtimeMessages = normalizeRuntimeMessages(input.messages);
+    const runtimeMessages = normalizeAgUiRuntimeMessages(input.messages);
     let runtimeStream;
     let clientAttached = true;
     try {

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

@@ -1,2 +1,2 @@
-export declare const VERSION = "0.1.222";
+export declare const VERSION = "0.1.223";
 //# 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.222";
+export const VERSION = "0.1.223";

package/package.json

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

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

@@ -0,0 +1,453 @@
+import { z } from "zod";
+import { INVALID_ARGUMENT } from "../errors/index.js";
+import { SKILL_TOOL_IDS } from "../skill/types.js";
+import { type Tool, toolRegistry } from "../tool/index.js";
+import {
+  AgentRuntime,
+  RunAlreadyExistsError,
+  type RunResumeSessionManager,
+} from "./runtime/index.js";
+import type { Agent } from "./types.js";
+import {
+  type AgUiRuntimeInjectedTool,
+  type AgUiRuntimeRequest,
+  parseAgUiRuntimeRequestOrError,
+} from "./runtime-ag-ui-contract.js";
+import { normalizeAgUiRuntimeMessages } from "./ag-ui-runtime-support.js";
+import {
+  createStreamTransformState,
+  finalizeRunEvents,
+  formatAgUiEvent,
+  mapRuntimeEventToAgUi,
+} from "../internal-agents/ag-ui-sse.js";
+import { streamDataStreamEvents } from "./data-stream.js";
+
+const AG_UI_HEADERS: Record<string, string> = {
+  "Content-Type": "text/event-stream",
+  "Cache-Control": "no-cache",
+  Connection: "keep-alive",
+};
+
+type AgUiResumeValue = { result: unknown; isError: boolean };
+type AgUiRuntimePart = Record<string, unknown> & { type: string };
+
+function isRequest(obj: unknown): obj is Request {
+  return (
+    typeof obj === "object" &&
+    obj !== null &&
+    "json" in obj &&
+    typeof obj.json === "function" &&
+    "url" in obj &&
+    typeof obj.url === "string" &&
+    "method" in obj &&
+    typeof obj.method === "string"
+  );
+}
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+
+function extractRequest(requestOrCtx: unknown): Request {
+  if (isRequest(requestOrCtx)) return requestOrCtx;
+
+  if (typeof requestOrCtx === "object" && requestOrCtx !== null && "request" in requestOrCtx) {
+    const candidate = (requestOrCtx as Record<string, unknown>).request;
+    if (isRequest(candidate)) return candidate;
+  }
+
+  throw INVALID_ARGUMENT.create({
+    detail: "Invalid handler argument: expected Request or APIContext",
+  });
+}
+
+function buildStreamContext(
+  request: AgUiRuntimeRequest,
+  baseContext: Record<string, unknown>,
+): Record<string, unknown> {
+  return {
+    ...baseContext,
+    threadId: request.threadId,
+    runId: request.runId,
+    agUi: {
+      context: request.context,
+      forwardedProps: request.forwardedProps,
+    },
+  };
+}
+
+function closeController(controller: ReadableStreamDefaultController<Uint8Array>): void {
+  try {
+    controller.close();
+  } catch {
+    return;
+  }
+}
+
+function enqueueEvent(
+  controller: ReadableStreamDefaultController<Uint8Array>,
+  event: string,
+  payload: Record<string, unknown>,
+): boolean {
+  try {
+    controller.enqueue(formatAgUiEvent(event, payload));
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+async function createAgUiRuntimeStreamResponse(
+  options: {
+    agentId: string;
+    request: AgUiRuntimeRequest;
+    upstreamBody: ReadableStream<Uint8Array> | null;
+    upstreamStatus: number;
+    upstreamStatusText?: string;
+    onFinish?: () => void;
+    onError?: (error: unknown) => void;
+    onToolCallSeen?: (toolCallId: string) => void;
+  },
+): Promise<Response> {
+  const {
+    agentId,
+    request,
+    upstreamBody,
+    upstreamStatus,
+    upstreamStatusText,
+    onFinish,
+    onError,
+    onToolCallSeen,
+  } = options;
+
+  const stream = new ReadableStream<Uint8Array>({
+    start: async (controller) => {
+      const state = createStreamTransformState();
+      const prepareToolResultIfNeeded = (event: string, payload: Record<string, unknown>) => {
+        if (
+          event !== "ToolCallStart" && event !== "ToolCallArgs" &&
+          event !== "ToolCallEnd"
+        ) {
+          return;
+        }
+
+        const toolCallId = typeof payload.toolCallId === "string" ? payload.toolCallId : null;
+        if (toolCallId) {
+          onToolCallSeen?.(toolCallId);
+        }
+      };
+
+      if (
+        !enqueueEvent(controller, "RunStarted", {
+          runId: request.runId,
+          threadId: request.threadId,
+          agentId,
+        })
+      ) {
+        return;
+      }
+      if (
+        !enqueueEvent(controller, "StateSnapshot", {
+          snapshot: isRecord(request.state) ? request.state : {},
+        })
+      ) {
+        return;
+      }
+      if (
+        !enqueueEvent(controller, "MessagesSnapshot", {
+          messages: normalizeAgUiRuntimeMessages(request.messages),
+        })
+      ) {
+        return;
+      }
+
+      try {
+        if (!upstreamBody) {
+          for (const event of finalizeRunEvents(state, null)) {
+            if (!enqueueEvent(controller, event.event, event.payload)) {
+              return;
+            }
+          }
+          onFinish?.();
+          closeController(controller);
+          return;
+        }
+
+        for await (
+          const event of streamDataStreamEvents(upstreamBody) as AsyncIterable<AgUiRuntimePart>
+        ) {
+          for (const mapped of mapRuntimeEventToAgUi(state, event)) {
+            prepareToolResultIfNeeded(mapped.event, mapped.payload);
+            if (!enqueueEvent(controller, mapped.event, mapped.payload)) {
+              return;
+            }
+          }
+        }
+
+        for (const event of finalizeRunEvents(state, null)) {
+          if (!enqueueEvent(controller, event.event, event.payload)) {
+            return;
+          }
+        }
+        onFinish?.();
+      } catch (error) {
+        onError?.(error);
+        enqueueEvent(controller, "RunError", {
+          message: error instanceof Error ? error.message : "Agent run failed",
+        });
+      } finally {
+        closeController(controller);
+      }
+    },
+  });
+
+  return new Response(stream, {
+    status: upstreamStatus,
+    statusText: upstreamStatusText,
+    headers: { ...AG_UI_HEADERS },
+  });
+}
+
+async function createAgUiRuntimeDirectStreamResponse(
+  agent: Agent,
+  request: AgUiRuntimeRequest,
+  baseContext: Record<string, unknown>,
+): Promise<Response> {
+  await agent.clearMemory();
+
+  const result = await agent.stream({
+    messages: normalizeAgUiRuntimeMessages(request.messages),
+    context: buildStreamContext(request, baseContext),
+  });
+
+  const upstream = result.toDataStreamResponse();
+  return await createAgUiRuntimeStreamResponse({
+    agentId: agent.id,
+    request,
+    upstreamBody: upstream.body,
+    upstreamStatus: upstream.status,
+    upstreamStatusText: upstream.statusText,
+  });
+}
+
+function createInjectedAgUiTool(
+  runId: string,
+  tool: AgUiRuntimeInjectedTool,
+  sessionManager: RunResumeSessionManager<AgUiResumeValue>,
+): Tool {
+  return {
+    id: tool.name,
+    type: "function",
+    description: tool.description ?? tool.name,
+    inputSchema: z.record(z.string(), z.unknown()),
+    inputSchemaJson: (tool.parameters ??
+      { type: "object", properties: {}, additionalProperties: true }) as Tool["inputSchemaJson"],
+    execute: async (_input, context) => {
+      const toolCallId = typeof context?.toolCallId === "string" ? context.toolCallId : null;
+      if (!toolCallId) {
+        throw new Error(`Missing toolCallId for injected tool "${tool.name}"`);
+      }
+
+      sessionManager.prepareForSignal(runId, toolCallId);
+      const submitted = await sessionManager.waitForSignal(runId, toolCallId);
+      if (submitted.isError) {
+        throw new Error(
+          typeof submitted.result === "string"
+            ? submitted.result
+            : JSON.stringify(submitted.result),
+        );
+      }
+      return submitted.result;
+    },
+  };
+}
+
+function buildMergedTools(
+  agent: Agent,
+  request: AgUiRuntimeRequest,
+  sessionManager: RunResumeSessionManager<AgUiResumeValue>,
+): Agent["config"]["tools"] {
+  const injectedTools = Object.fromEntries(
+    request.tools.map((tool) => [
+      tool.name,
+      createInjectedAgUiTool(request.runId, tool, sessionManager),
+    ]),
+  );
+
+  if (!agent.config.tools) {
+    return Object.keys(injectedTools).length > 0 ? injectedTools : undefined;
+  }
+
+  if (agent.config.tools === true) {
+    const merged: Record<string, Tool | boolean> = {};
+    for (const [toolId] of toolRegistry.getAll()) {
+      if (!agent.config.skills && SKILL_TOOL_IDS.has(toolId)) {
+        continue;
+      }
+      merged[toolId] = true;
+    }
+    return { ...merged, ...injectedTools };
+  }
+
+  return { ...agent.config.tools, ...injectedTools };
+}
+
+async function createAgUiRuntimeInjectedToolsStreamResponse(
+  agent: Agent,
+  request: AgUiRuntimeRequest,
+  baseContext: Record<string, unknown>,
+  sessionManager: RunResumeSessionManager<AgUiResumeValue>,
+): Promise<Response> {
+  try {
+    sessionManager.startRun({ runId: request.runId, threadId: request.threadId });
+  } catch (error) {
+    if (error instanceof RunAlreadyExistsError) {
+      return Response.json({ error: "Run already active" }, { status: 409 });
+    }
+    throw error;
+  }
+
+  const runtime = new AgentRuntime(agent.id, {
+    ...agent.config,
+    tools: buildMergedTools(agent, request, sessionManager),
+  });
+
+  let upstreamBody: ReadableStream<Uint8Array>;
+  try {
+    upstreamBody = await runtime.stream(
+      normalizeAgUiRuntimeMessages(request.messages),
+      buildStreamContext(request, baseContext),
+      undefined,
+      undefined,
+      undefined,
+    );
+  } catch (error) {
+    sessionManager.failRun(request.runId);
+    throw error;
+  }
+
+  return await createAgUiRuntimeStreamResponse({
+    agentId: agent.id,
+    request,
+    upstreamBody,
+    upstreamStatus: 200,
+    onFinish: () => {
+      sessionManager.completeRun(request.runId);
+    },
+    onError: () => {
+      sessionManager.failRun(request.runId);
+    },
+    onToolCallSeen: (toolCallId) => {
+      sessionManager.prepareForSignal(request.runId, toolCallId);
+    },
+  });
+}
+
+export interface AgUiRuntimeHandlerExecuteInput {
+  request: Request;
+  agUiInput: AgUiRuntimeRequest;
+  context: Record<string, unknown>;
+  createDefaultResponse?: () => Promise<Response>;
+}
+
+export type AgUiRuntimeHandlerExecute = (
+  input: AgUiRuntimeHandlerExecuteInput,
+) => Promise<Response> | Response;
+
+export interface AgUiRuntimeHandlerOptions {
+  context?:
+    | Record<string, unknown>
+    | ((request: Request) => Record<string, unknown> | Promise<Record<string, unknown>>);
+  sessionManager?: RunResumeSessionManager<AgUiResumeValue>;
+  execute?: AgUiRuntimeHandlerExecute;
+}
+
+export interface AgUiRuntimeHandlerConfigWithAgent extends AgUiRuntimeHandlerOptions {
+  agent: Agent;
+}
+
+export type AgUiRuntimeHandlerConfig =
+  | AgUiRuntimeHandlerConfigWithAgent
+  | (AgUiRuntimeHandlerOptions & { agent?: undefined; execute: AgUiRuntimeHandlerExecute });
+
+export function createAgUiRuntimeHandler(
+  config: AgUiRuntimeHandlerConfig,
+): (requestOrCtx: unknown) => Promise<Response> {
+  if (!config.agent && !config.execute) {
+    throw new Error(
+      "createAgUiRuntimeHandler requires either an agent or an execute handler.",
+    );
+  }
+
+  return async function POST(requestOrCtx: unknown): Promise<Response> {
+    const request = extractRequest(requestOrCtx);
+
+    try {
+      const parsed = await parseAgUiRuntimeRequestOrError(request);
+      if (parsed instanceof Response) {
+        return parsed;
+      }
+
+      const context = typeof config.context === "function"
+        ? await config.context(request)
+        : config.context ?? {};
+
+      const createDefaultResponse = config.agent
+        ? () =>
+          parsed.tools.length > 0
+            ? config.sessionManager
+              ? createAgUiRuntimeInjectedToolsStreamResponse(
+                config.agent,
+                parsed,
+                context,
+                config.sessionManager,
+              )
+              : Promise.resolve(
+                Response.json(
+                  {
+                    error:
+                      "Injected AG-UI tools require a public RunResumeSessionManager on createAgUiRuntimeHandler().",
+                  },
+                  { status: 501 },
+                ),
+              )
+            : createAgUiRuntimeDirectStreamResponse(config.agent, parsed, context)
+        : undefined;
+
+      if (config.execute) {
+        return await config.execute({
+          request,
+          agUiInput: parsed,
+          context,
+          createDefaultResponse,
+        });
+      }
+
+      if (createDefaultResponse) {
+        return await createDefaultResponse();
+      }
+
+      throw new Error("createAgUiRuntimeHandler configuration became invalid during execution.");
+    } catch (error) {
+      if (error instanceof z.ZodError) {
+        return Response.json(
+          {
+            error: "Invalid AG-UI runtime request",
+            details: error.issues.map((issue) => ({
+              path: issue.path,
+              message: issue.message,
+            })),
+          },
+          { status: 400 },
+        );
+      }
+
+      return Response.json(
+        {
+          error: error instanceof Error ? error.message : "Internal server error",
+        },
+        { status: 500 },
+      );
+    }
+  };
+}

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

@@ -0,0 +1,64 @@
+import type { Message } from "./types.js";
+import type { AgUiRuntimeRequest } from "./runtime-ag-ui-contract.js";
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+
+function parseToolArguments(serializedArguments: string): Record<string, unknown> {
+  try {
+    const parsed = JSON.parse(serializedArguments);
+    return isRecord(parsed) ? parsed : {};
+  } catch {
+    return {};
+  }
+}
+
+export function normalizeAgUiRuntimeMessages(
+  messages: AgUiRuntimeRequest["messages"],
+): Message[] {
+  return messages.map((message) => {
+    const parts: Message["parts"] = [];
+
+    switch (message.role) {
+      case "system":
+      case "user":
+        parts.push({ type: "text", text: message.content });
+        break;
+      case "assistant":
+        if (typeof message.content === "string" && message.content.length > 0) {
+          parts.push({ type: "text", text: message.content });
+        }
+        for (const toolCall of message.toolCalls ?? []) {
+          parts.push({
+            type: "tool-call",
+            toolCallId: toolCall.id,
+            toolName: toolCall.function.name,
+            args: parseToolArguments(toolCall.function.arguments),
+          });
+        }
+        break;
+      case "tool":
+        parts.push({
+          type: "tool-result",
+          toolCallId: message.toolCallId,
+          toolName: "unknown",
+          result: message.error
+            ? {
+              content: message.content,
+              error: message.error,
+            }
+            : message.content,
+        });
+        break;
+    }
+
+    return {
+      id: message.id,
+      role: message.role,
+      parts,
+      ...(message.createdAt ? { timestamp: Date.parse(message.createdAt) || undefined } : {}),
+      ...(message.metadata ? { metadata: message.metadata } : {}),
+    };
+  });
+}

package/src/src/agent/index.ts

@@ -141,6 +141,14 @@ export {
 } from "./composition/index.js";
 
 export { agent } from "./factory.js";
+export {
+  type AgUiRuntimeHandlerConfig,
+  type AgUiRuntimeHandlerConfigWithAgent,
+  type AgUiRuntimeHandlerExecute,
+  type AgUiRuntimeHandlerExecuteInput,
+  type AgUiRuntimeHandlerOptions,
+  createAgUiRuntimeHandler,
+} from "./ag-ui-runtime-handler.js";
 export {
   type AgUiRuntimeContextItem,
   AgUiRuntimeContextItemSchema,
@@ -153,6 +161,7 @@ export {
   parseAgUiRuntimeRequest,
   parseAgUiRuntimeRequestOrError,
 } from "./runtime-ag-ui-contract.js";
+export { normalizeAgUiRuntimeMessages } from "./ag-ui-runtime-support.js";
 export {
   type AgUiBrowserEncodedEvent,
   type AgUiBrowserEncoderState,

package/src/src/internal-agents/run-stream.ts

@@ -4,6 +4,7 @@ import {
   type AgentResponse,
   AgentRuntime,
 } from "../agent/index.js";
+import { normalizeAgUiRuntimeMessages } from "../agent/ag-ui-runtime-support.js";
 import { SKILL_TOOL_IDS } from "../skill/types.js";
 import { type Tool, toolRegistry } from "../tool/index.js";
 import { z } from "zod";
@@ -120,62 +121,6 @@ function isRecord(value: unknown): value is Record<string, unknown> {
   return typeof value === "object" && value !== null && !Array.isArray(value);
 }
 
-function parseToolArguments(serializedArguments: string): Record<string, unknown> {
-  try {
-    const parsed = JSON.parse(serializedArguments);
-    return isRecord(parsed) ? parsed : {};
-  } catch {
-    return {};
-  }
-}
-
-function normalizeRuntimeMessages(messages: RuntimeRunAgentInput["messages"]): Message[] {
-  return messages.map((message) => {
-    const parts: Message["parts"] = [];
-
-    switch (message.role) {
-      case "system":
-      case "user":
-        parts.push({ type: "text", text: message.content });
-        break;
-      case "assistant":
-        if (typeof message.content === "string" && message.content.length > 0) {
-          parts.push({ type: "text", text: message.content });
-        }
-        for (const toolCall of message.toolCalls ?? []) {
-          parts.push({
-            type: "tool-call",
-            toolCallId: toolCall.id,
-            toolName: toolCall.function.name,
-            args: parseToolArguments(toolCall.function.arguments),
-          });
-        }
-        break;
-      case "tool":
-        parts.push({
-          type: "tool-result",
-          toolCallId: message.toolCallId,
-          toolName: "unknown",
-          result: message.error
-            ? {
-              content: message.content,
-              error: message.error,
-            }
-            : message.content,
-        });
-        break;
-    }
-
-    return {
-      id: message.id,
-      role: message.role,
-      parts,
-      ...(message.createdAt ? { timestamp: Date.parse(message.createdAt) || undefined } : {}),
-      ...(message.metadata ? { metadata: message.metadata } : {}),
-    };
-  });
-}
-
 function getAllowedRemoteToolNames(
   forwardedProps: RuntimeRunAgentInput["forwardedProps"],
 ): string[] | undefined {
@@ -226,7 +171,7 @@ export async function createRuntimeAgentStreamResponse(
     new AgentRuntime(runtimeAgent.id, runtimeAgent.config);
 
   let completedResponse: AgentResponse | null = null;
-  const runtimeMessages = normalizeRuntimeMessages(input.messages);
+  const runtimeMessages = normalizeAgUiRuntimeMessages(input.messages);
   let runtimeStream: ReadableStream<Uint8Array>;
   let clientAttached = true;
   try {

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

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