veryfront 0.1.223 → 0.1.225

package/esm/deno.js

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

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

@@ -55,6 +55,12 @@ 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 ExecuteAgUiDetachedStartInput {
+    request: AgUiDetachedStartRequest;
+    rawRequest?: Request;
+    requestOrCtx?: unknown;
+    context?: Record<string, unknown>;
+}
 interface AgUiDetachedStartExecutionInput {
     request: AgUiDetachedStartRequest;
     requestOrCtx: unknown;
@@ -93,6 +99,7 @@ export type AgUiDetachedStartHandlerOptions = (AgUiDetachedStartHandlerOptionsBa
     agent?: undefined;
     startDetachedExecution: AgUiDetachedExecutionStarter;
 });
+export declare function executeAgUiDetachedStart(options: AgUiDetachedStartHandlerOptions, input: ExecuteAgUiDetachedStartInput): Promise<Response>;
 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;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"}
+{"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;AAmIvF,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,6BAA6B;IAC5C,OAAO,EAAE,wBAAwB,CAAC;IAClC,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,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;AAqEL,wBAAsB,wBAAwB,CAC5C,OAAO,EAAE,+BAA+B,EACxC,KAAK,EAAE,6BAA6B,GACnC,OAAO,CAAC,QAAQ,CAAC,CAwFnB;AAED,wBAAgB,8BAA8B,CAC5C,OAAO,EAAE,+BAA+B,GACvC,CAAC,YAAY,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAuC9C"}

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

@@ -84,12 +84,6 @@ function buildMergedTools(agent, request, sessionManager) {
     }
     return { ...agent.config.tools, ...injectedTools };
 }
-async function resolveContextValue(value, request) {
-    if (typeof value === "function") {
-        return await value(request);
-    }
-    return value ?? {};
-}
 function scheduleDetachedTask(requestOrCtx, task) {
     if (typeof requestOrCtx === "object" &&
         requestOrCtx !== null &&
@@ -123,6 +117,111 @@ async function startDefaultDetachedExecution(input) {
     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);
 }
+async function resolveDetachedStartContext(options, input) {
+    if (input.context) {
+        return input.context;
+    }
+    if (!options.context) {
+        return {};
+    }
+    if (typeof options.context === "function") {
+        if (!input.rawRequest) {
+            throw INVALID_ARGUMENT.create({
+                detail: "executeAgUiDetachedStart requires rawRequest when options.context is a function.",
+            });
+        }
+        return await options.context(input.rawRequest);
+    }
+    return options.context;
+}
+function assertDetachedStartRawRequest(options, input) {
+    if (!options.startDetachedExecution) {
+        return input.rawRequest;
+    }
+    if (input.rawRequest) {
+        return input.rawRequest;
+    }
+    throw INVALID_ARGUMENT.create({
+        detail: "executeAgUiDetachedStart requires rawRequest when options.startDetachedExecution is used.",
+    });
+}
+export async function executeAgUiDetachedStart(options, input) {
+    const rawRequest = assertDetachedStartRawRequest(options, input);
+    const context = await resolveDetachedStartContext(options, input);
+    try {
+        const abortSignal = options.sessionManager.startRun({
+            runId: input.request.runId,
+            threadId: input.request.threadId,
+        });
+        await options.onAccepted?.({
+            request: input.request,
+            runId: input.request.runId,
+            threadId: input.request.threadId,
+        });
+        const detachedTask = (async () => {
+            try {
+                if (options.startDetachedExecution) {
+                    await options.startDetachedExecution({
+                        request: input.request,
+                        requestOrCtx: input.requestOrCtx,
+                        rawRequest: rawRequest,
+                        context,
+                        abortSignal,
+                    });
+                }
+                else if (options.agent) {
+                    await startDefaultDetachedExecution({
+                        agent: options.agent,
+                        request: input.request,
+                        context,
+                        abortSignal,
+                        sessionManager: options.sessionManager,
+                    });
+                }
+                else {
+                    throw new Error("Detached AG-UI start configuration became invalid during execution.");
+                }
+                options.sessionManager.completeRun(input.request.runId);
+                await options.onFinish?.({
+                    runId: input.request.runId,
+                    threadId: input.request.threadId,
+                });
+            }
+            catch (error) {
+                options.sessionManager.failRun(input.request.runId);
+                await options.onError?.({
+                    runId: input.request.runId,
+                    threadId: input.request.threadId,
+                    error,
+                });
+            }
+        })().catch(() => undefined);
+        scheduleDetachedTask(input.requestOrCtx, detachedTask);
+        return Response.json({
+            accepted: true,
+            duplicate: false,
+            runId: input.request.runId,
+            threadId: input.request.threadId,
+        }, { status: 202 });
+    }
+    catch (error) {
+        if (error instanceof RunAlreadyExistsError) {
+            await options.onDuplicate?.({
+                request: input.request,
+                runId: input.request.runId,
+                threadId: input.request.threadId,
+            });
+            return Response.json({
+                accepted: true,
+                duplicate: true,
+                runId: input.request.runId,
+                threadId: input.request.threadId,
+            }, { status: 202 });
+        }
+        options.sessionManager.failRun(input.request.runId);
+        throw error;
+    }
+}
 export function createAgUiDetachedStartHandler(options) {
     if (!options.agent && !options.startDetachedExecution) {
         throw new Error("Detached AG-UI start requires either an agent or startDetachedExecution handler.");
@@ -131,80 +230,11 @@ export function createAgUiDetachedStartHandler(options) {
         const request = extractRequest(requestOrCtx);
         try {
             const parsed = AgUiDetachedStartRequestSchema.parse(await request.json());
-            const context = await resolveContextValue(options.context, request);
-            try {
-                const abortSignal = options.sessionManager.startRun({
-                    runId: parsed.runId,
-                    threadId: parsed.threadId,
-                });
-                await options.onAccepted?.({
-                    request: parsed,
-                    runId: parsed.runId,
-                    threadId: parsed.threadId,
-                });
-                const detachedTask = (async () => {
-                    try {
-                        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,
-                            threadId: parsed.threadId,
-                        });
-                    }
-                    catch (error) {
-                        options.sessionManager.failRun(parsed.runId);
-                        await options.onError?.({
-                            runId: parsed.runId,
-                            threadId: parsed.threadId,
-                            error,
-                        });
-                    }
-                })().catch(() => undefined);
-                scheduleDetachedTask(requestOrCtx, detachedTask);
-                return Response.json({
-                    accepted: true,
-                    duplicate: false,
-                    runId: parsed.runId,
-                    threadId: parsed.threadId,
-                }, { status: 202 });
-            }
-            catch (error) {
-                if (error instanceof RunAlreadyExistsError) {
-                    await options.onDuplicate?.({
-                        request: parsed,
-                        runId: parsed.runId,
-                        threadId: parsed.threadId,
-                    });
-                    return Response.json({
-                        accepted: true,
-                        duplicate: true,
-                        runId: parsed.runId,
-                        threadId: parsed.threadId,
-                    }, { status: 202 });
-                }
-                options.sessionManager.failRun(parsed.runId);
-                throw error;
-            }
+            return await executeAgUiDetachedStart(options, {
+                request: parsed,
+                rawRequest: request,
+                requestOrCtx,
+            });
         }
         catch (error) {
             if (error instanceof z.ZodError) {

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

@@ -5,6 +5,11 @@ type AgUiResumeValue = {
     result: unknown;
     isError: boolean;
 };
+export interface AgUiRuntimeLifecycleContext {
+    request: AgUiRuntimeRequest;
+    toolCallId?: string;
+    error?: unknown;
+}
 export interface AgUiRuntimeHandlerExecuteInput {
     request: Request;
     agUiInput: AgUiRuntimeRequest;
@@ -16,6 +21,9 @@ export interface AgUiRuntimeHandlerOptions {
     context?: Record<string, unknown> | ((request: Request) => Record<string, unknown> | Promise<Record<string, unknown>>);
     sessionManager?: RunResumeSessionManager<AgUiResumeValue>;
     execute?: AgUiRuntimeHandlerExecute;
+    onToolCallSeen?: (context: AgUiRuntimeLifecycleContext) => Promise<void> | void;
+    onFinish?: (context: AgUiRuntimeLifecycleContext) => Promise<void> | void;
+    onError?: (context: AgUiRuntimeLifecycleContext) => Promise<void> | void;
 }
 export interface AgUiRuntimeHandlerConfigWithAgent extends AgUiRuntimeHandlerOptions {
     agent: Agent;

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

@@ -1 +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"}
+{"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;AAG7D,MAAM,WAAW,2BAA2B;IAC1C,OAAO,EAAE,kBAAkB,CAAC;IAC5B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,OAAO,CAAC;CACjB;AAqWD,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;IACpC,cAAc,CAAC,EAAE,CAAC,OAAO,EAAE,2BAA2B,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAChF,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,2BAA2B,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC1E,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,2BAA2B,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;CAC1E;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,CAqI9C"}

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

@@ -12,6 +12,27 @@ const AG_UI_HEADERS = {
     "Cache-Control": "no-cache",
     Connection: "keep-alive",
 };
+function invokeLifecycleCallback(callback, context) {
+    if (!callback)
+        return;
+    try {
+        const result = callback(context);
+        void Promise.resolve(result).catch(() => undefined);
+    }
+    catch {
+        return;
+    }
+}
+async function invokeLifecycleCallbackAndWait(callback, context) {
+    if (!callback)
+        return;
+    try {
+        await callback(context);
+    }
+    catch {
+        return;
+    }
+}
 function isRequest(obj) {
     return (typeof obj === "object" &&
         obj !== null &&
@@ -140,7 +161,7 @@ async function createAgUiRuntimeStreamResponse(options) {
         headers: { ...AG_UI_HEADERS },
     });
 }
-async function createAgUiRuntimeDirectStreamResponse(agent, request, baseContext) {
+async function createAgUiRuntimeDirectStreamResponse(agent, request, baseContext, lifecycle) {
     await agent.clearMemory();
     const result = await agent.stream({
         messages: normalizeAgUiRuntimeMessages(request.messages),
@@ -153,6 +174,9 @@ async function createAgUiRuntimeDirectStreamResponse(agent, request, baseContext
         upstreamBody: upstream.body,
         upstreamStatus: upstream.status,
         upstreamStatusText: upstream.statusText,
+        onFinish: lifecycle?.onFinish,
+        onError: lifecycle?.onError,
+        onToolCallSeen: lifecycle?.onToolCallSeen,
     });
 }
 function createInjectedAgUiTool(runId, tool, sessionManager) {
@@ -199,7 +223,7 @@ function buildMergedTools(agent, request, sessionManager) {
     }
     return { ...agent.config.tools, ...injectedTools };
 }
-async function createAgUiRuntimeInjectedToolsStreamResponse(agent, request, baseContext, sessionManager) {
+async function createAgUiRuntimeInjectedToolsStreamResponse(agent, request, baseContext, sessionManager, lifecycle) {
     try {
         sessionManager.startRun({ runId: request.runId, threadId: request.threadId });
     }
@@ -228,12 +252,15 @@ async function createAgUiRuntimeInjectedToolsStreamResponse(agent, request, base
         upstreamStatus: 200,
         onFinish: () => {
             sessionManager.completeRun(request.runId);
+            void lifecycle?.onFinish?.();
         },
-        onError: () => {
+        onError: (error) => {
             sessionManager.failRun(request.runId);
+            void lifecycle?.onError?.(error);
         },
         onToolCallSeen: (toolCallId) => {
             sessionManager.prepareForSignal(request.runId, toolCallId);
+            void lifecycle?.onToolCallSeen?.(toolCallId);
         },
     });
 }
@@ -260,16 +287,53 @@ export function createAgUiRuntimeHandler(config) {
                         }, { status: 501 }))
                     : createAgUiRuntimeDirectStreamResponse(config.agent, parsed, context)
                 : undefined;
+            const invokeLifecycle = (type, extra = {}) => {
+                invokeLifecycleCallback(config[type], {
+                    request: parsed,
+                    ...extra,
+                });
+            };
+            const createDefaultResponseWithLifecycle = createDefaultResponse
+                ? async () => {
+                    try {
+                        if (!config.agent) {
+                            return await createDefaultResponse();
+                        }
+                        if (parsed.tools.length > 0) {
+                            if (!config.sessionManager) {
+                                return await createDefaultResponse();
+                            }
+                            return await createAgUiRuntimeInjectedToolsStreamResponse(config.agent, parsed, context, config.sessionManager, {
+                                onFinish: () => invokeLifecycle("onFinish"),
+                                onError: (error) => invokeLifecycle("onError", { error }),
+                                onToolCallSeen: (toolCallId) => invokeLifecycle("onToolCallSeen", { toolCallId }),
+                            });
+                        }
+                        return await createAgUiRuntimeDirectStreamResponse(config.agent, parsed, context, {
+                            onFinish: () => invokeLifecycle("onFinish"),
+                            onError: (error) => invokeLifecycle("onError", { error }),
+                            onToolCallSeen: (toolCallId) => invokeLifecycle("onToolCallSeen", { toolCallId }),
+                        });
+                    }
+                    catch (error) {
+                        await invokeLifecycleCallbackAndWait(config.onError, {
+                            request: parsed,
+                            error,
+                        });
+                        throw error;
+                    }
+                }
+                : undefined;
             if (config.execute) {
                 return await config.execute({
                     request,
                     agUiInput: parsed,
                     context,
-                    createDefaultResponse,
+                    createDefaultResponse: createDefaultResponseWithLifecycle,
                 });
             }
-            if (createDefaultResponse) {
-                return await createDefaultResponse();
+            if (createDefaultResponseWithLifecycle) {
+                return await createDefaultResponseWithLifecycle();
             }
             throw new Error("createAgUiRuntimeHandler configuration became invalid during execution.");
         }

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

@@ -84,13 +84,13 @@ 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 AgUiRuntimeHandlerConfig, type AgUiRuntimeHandlerConfigWithAgent, type AgUiRuntimeHandlerExecute, type AgUiRuntimeHandlerExecuteInput, type AgUiRuntimeHandlerOptions, type AgUiRuntimeLifecycleContext, 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";
-export { type AgUiDetachedStartAccepted, AgUiDetachedStartAcceptedSchema, type AgUiDetachedStartHandlerOptions, type AgUiDetachedStartRequest, AgUiDetachedStartRequestSchema, createAgUiDetachedStartHandler, } from "./ag-ui-detached-start.js";
+export { type AgUiDetachedStartAccepted, AgUiDetachedStartAcceptedSchema, type AgUiDetachedStartHandlerOptions, type AgUiDetachedStartRequest, AgUiDetachedStartRequestSchema, createAgUiDetachedStartHandler, executeAgUiDetachedStart, type ExecuteAgUiDetachedStartInput, } 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";

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,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"}
+{"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,KAAK,2BAA2B,EAChC,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,EAC9B,wBAAwB,EACxB,KAAK,6BAA6B,GACnC,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,7 +89,7 @@ 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";
-export { AgUiDetachedStartAcceptedSchema, AgUiDetachedStartRequestSchema, createAgUiDetachedStartHandler, } from "./ag-ui-detached-start.js";
+export { AgUiDetachedStartAcceptedSchema, AgUiDetachedStartRequestSchema, createAgUiDetachedStartHandler, executeAgUiDetachedStart, } 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";

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

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

package/package.json

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

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

@@ -133,17 +133,6 @@ function buildMergedTools(
   return { ...agent.config.tools, ...injectedTools };
 }
 
-async function resolveContextValue(
-  value: AgUiContextValue | undefined,
-  request: Request,
-): Promise<Record<string, unknown>> {
-  if (typeof value === "function") {
-    return await value(request);
-  }
-
-  return value ?? {};
-}
-
 function scheduleDetachedTask(requestOrCtx: unknown, task: Promise<void>): void {
   if (
     typeof requestOrCtx === "object" &&
@@ -181,6 +170,13 @@ export const AgUiDetachedStartAcceptedSchema = z.object({
 export type AgUiDetachedStartRequest = z.infer<typeof AgUiDetachedStartRequestSchema>;
 export type AgUiDetachedStartAccepted = z.infer<typeof AgUiDetachedStartAcceptedSchema>;
 
+export interface ExecuteAgUiDetachedStartInput {
+  request: AgUiDetachedStartRequest;
+  rawRequest?: Request;
+  requestOrCtx?: unknown;
+  context?: Record<string, unknown>;
+}
+
 interface AgUiDetachedStartExecutionInput {
   request: AgUiDetachedStartRequest;
   requestOrCtx: unknown;
@@ -242,6 +238,142 @@ async function startDefaultDetachedExecution(input: {
   await drainRuntimeStream(runtimeStream);
 }
 
+async function resolveDetachedStartContext(
+  options: AgUiDetachedStartHandlerOptions,
+  input: ExecuteAgUiDetachedStartInput,
+): Promise<Record<string, unknown>> {
+  if (input.context) {
+    return input.context;
+  }
+
+  if (!options.context) {
+    return {};
+  }
+
+  if (typeof options.context === "function") {
+    if (!input.rawRequest) {
+      throw INVALID_ARGUMENT.create({
+        detail: "executeAgUiDetachedStart requires rawRequest when options.context is a function.",
+      });
+    }
+
+    return await options.context(input.rawRequest);
+  }
+
+  return options.context;
+}
+
+function assertDetachedStartRawRequest(
+  options: AgUiDetachedStartHandlerOptions,
+  input: ExecuteAgUiDetachedStartInput,
+): Request | undefined {
+  if (!options.startDetachedExecution) {
+    return input.rawRequest;
+  }
+
+  if (input.rawRequest) {
+    return input.rawRequest;
+  }
+
+  throw INVALID_ARGUMENT.create({
+    detail:
+      "executeAgUiDetachedStart requires rawRequest when options.startDetachedExecution is used.",
+  });
+}
+
+export async function executeAgUiDetachedStart(
+  options: AgUiDetachedStartHandlerOptions,
+  input: ExecuteAgUiDetachedStartInput,
+): Promise<Response> {
+  const rawRequest = assertDetachedStartRawRequest(options, input);
+  const context = await resolveDetachedStartContext(options, input);
+
+  try {
+    const abortSignal = options.sessionManager.startRun({
+      runId: input.request.runId,
+      threadId: input.request.threadId,
+    });
+
+    await options.onAccepted?.({
+      request: input.request,
+      runId: input.request.runId,
+      threadId: input.request.threadId,
+    });
+
+    const detachedTask = (async () => {
+      try {
+        if (options.startDetachedExecution) {
+          await options.startDetachedExecution({
+            request: input.request,
+            requestOrCtx: input.requestOrCtx,
+            rawRequest: rawRequest!,
+            context,
+            abortSignal,
+          });
+        } else if (options.agent) {
+          await startDefaultDetachedExecution({
+            agent: options.agent,
+            request: input.request,
+            context,
+            abortSignal,
+            sessionManager: options.sessionManager,
+          });
+        } else {
+          throw new Error(
+            "Detached AG-UI start configuration became invalid during execution.",
+          );
+        }
+
+        options.sessionManager.completeRun(input.request.runId);
+        await options.onFinish?.({
+          runId: input.request.runId,
+          threadId: input.request.threadId,
+        });
+      } catch (error) {
+        options.sessionManager.failRun(input.request.runId);
+        await options.onError?.({
+          runId: input.request.runId,
+          threadId: input.request.threadId,
+          error,
+        });
+      }
+    })().catch(() => undefined);
+
+    scheduleDetachedTask(input.requestOrCtx, detachedTask);
+
+    return Response.json(
+      {
+        accepted: true,
+        duplicate: false,
+        runId: input.request.runId,
+        threadId: input.request.threadId,
+      } satisfies AgUiDetachedStartAccepted,
+      { status: 202 },
+    );
+  } catch (error) {
+    if (error instanceof RunAlreadyExistsError) {
+      await options.onDuplicate?.({
+        request: input.request,
+        runId: input.request.runId,
+        threadId: input.request.threadId,
+      });
+
+      return Response.json(
+        {
+          accepted: true,
+          duplicate: true,
+          runId: input.request.runId,
+          threadId: input.request.threadId,
+        } satisfies AgUiDetachedStartAccepted,
+        { status: 202 },
+      );
+    }
+
+    options.sessionManager.failRun(input.request.runId);
+    throw error;
+  }
+}
+
 export function createAgUiDetachedStartHandler(
   options: AgUiDetachedStartHandlerOptions,
 ): (requestOrCtx: unknown) => Promise<Response> {
@@ -256,92 +388,11 @@ export function createAgUiDetachedStartHandler(
 
     try {
       const parsed = AgUiDetachedStartRequestSchema.parse(await request.json());
-      const context = await resolveContextValue(options.context, request);
-
-      try {
-        const abortSignal = options.sessionManager.startRun({
-          runId: parsed.runId,
-          threadId: parsed.threadId,
-        });
-
-        await options.onAccepted?.({
-          request: parsed,
-          runId: parsed.runId,
-          threadId: parsed.threadId,
-        });
-
-        const detachedTask = (async () => {
-          try {
-            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,
-              threadId: parsed.threadId,
-            });
-          } catch (error) {
-            options.sessionManager.failRun(parsed.runId);
-            await options.onError?.({
-              runId: parsed.runId,
-              threadId: parsed.threadId,
-              error,
-            });
-          }
-        })().catch(() => undefined);
-
-        scheduleDetachedTask(requestOrCtx, detachedTask);
-
-        return Response.json(
-          {
-            accepted: true,
-            duplicate: false,
-            runId: parsed.runId,
-            threadId: parsed.threadId,
-          } satisfies AgUiDetachedStartAccepted,
-          { status: 202 },
-        );
-      } catch (error) {
-        if (error instanceof RunAlreadyExistsError) {
-          await options.onDuplicate?.({
-            request: parsed,
-            runId: parsed.runId,
-            threadId: parsed.threadId,
-          });
-
-          return Response.json(
-            {
-              accepted: true,
-              duplicate: true,
-              runId: parsed.runId,
-              threadId: parsed.threadId,
-            } satisfies AgUiDetachedStartAccepted,
-            { status: 202 },
-          );
-        }
-
-        options.sessionManager.failRun(parsed.runId);
-        throw error;
-      }
+      return await executeAgUiDetachedStart(options, {
+        request: parsed,
+        rawRequest: request,
+        requestOrCtx,
+      });
     } catch (error) {
       if (error instanceof z.ZodError) {
         return Response.json(

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

@@ -31,6 +31,39 @@ const AG_UI_HEADERS: Record<string, string> = {
 type AgUiResumeValue = { result: unknown; isError: boolean };
 type AgUiRuntimePart = Record<string, unknown> & { type: string };
 
+export interface AgUiRuntimeLifecycleContext {
+  request: AgUiRuntimeRequest;
+  toolCallId?: string;
+  error?: unknown;
+}
+
+function invokeLifecycleCallback(
+  callback: ((context: AgUiRuntimeLifecycleContext) => Promise<void> | void) | undefined,
+  context: AgUiRuntimeLifecycleContext,
+): void {
+  if (!callback) return;
+
+  try {
+    const result = callback(context);
+    void Promise.resolve(result).catch(() => undefined);
+  } catch {
+    return;
+  }
+}
+
+async function invokeLifecycleCallbackAndWait(
+  callback: ((context: AgUiRuntimeLifecycleContext) => Promise<void> | void) | undefined,
+  context: AgUiRuntimeLifecycleContext,
+): Promise<void> {
+  if (!callback) return;
+
+  try {
+    await callback(context);
+  } catch {
+    return;
+  }
+}
+
 function isRequest(obj: unknown): obj is Request {
   return (
     typeof obj === "object" &&
@@ -212,6 +245,11 @@ async function createAgUiRuntimeDirectStreamResponse(
   agent: Agent,
   request: AgUiRuntimeRequest,
   baseContext: Record<string, unknown>,
+  lifecycle?: {
+    onFinish?: () => Promise<void> | void;
+    onError?: (error: unknown) => Promise<void> | void;
+    onToolCallSeen?: (toolCallId: string) => Promise<void> | void;
+  },
 ): Promise<Response> {
   await agent.clearMemory();
 
@@ -227,6 +265,9 @@ async function createAgUiRuntimeDirectStreamResponse(
     upstreamBody: upstream.body,
     upstreamStatus: upstream.status,
     upstreamStatusText: upstream.statusText,
+    onFinish: lifecycle?.onFinish,
+    onError: lifecycle?.onError,
+    onToolCallSeen: lifecycle?.onToolCallSeen,
   });
 }
 
@@ -297,6 +338,11 @@ async function createAgUiRuntimeInjectedToolsStreamResponse(
   request: AgUiRuntimeRequest,
   baseContext: Record<string, unknown>,
   sessionManager: RunResumeSessionManager<AgUiResumeValue>,
+  lifecycle?: {
+    onFinish?: () => Promise<void> | void;
+    onError?: (error: unknown) => Promise<void> | void;
+    onToolCallSeen?: (toolCallId: string) => Promise<void> | void;
+  },
 ): Promise<Response> {
   try {
     sessionManager.startRun({ runId: request.runId, threadId: request.threadId });
@@ -333,12 +379,15 @@ async function createAgUiRuntimeInjectedToolsStreamResponse(
     upstreamStatus: 200,
     onFinish: () => {
       sessionManager.completeRun(request.runId);
+      void lifecycle?.onFinish?.();
     },
-    onError: () => {
+    onError: (error) => {
       sessionManager.failRun(request.runId);
+      void lifecycle?.onError?.(error);
     },
     onToolCallSeen: (toolCallId) => {
       sessionManager.prepareForSignal(request.runId, toolCallId);
+      void lifecycle?.onToolCallSeen?.(toolCallId);
     },
   });
 }
@@ -360,6 +409,9 @@ export interface AgUiRuntimeHandlerOptions {
     | ((request: Request) => Record<string, unknown> | Promise<Record<string, unknown>>);
   sessionManager?: RunResumeSessionManager<AgUiResumeValue>;
   execute?: AgUiRuntimeHandlerExecute;
+  onToolCallSeen?: (context: AgUiRuntimeLifecycleContext) => Promise<void> | void;
+  onFinish?: (context: AgUiRuntimeLifecycleContext) => Promise<void> | void;
+  onError?: (context: AgUiRuntimeLifecycleContext) => Promise<void> | void;
 }
 
 export interface AgUiRuntimeHandlerConfigWithAgent extends AgUiRuntimeHandlerOptions {
@@ -414,17 +466,72 @@ export function createAgUiRuntimeHandler(
             : createAgUiRuntimeDirectStreamResponse(config.agent, parsed, context)
         : undefined;
 
+      const invokeLifecycle = (
+        type: "onFinish" | "onError" | "onToolCallSeen",
+        extra: Partial<AgUiRuntimeLifecycleContext> = {},
+      ): void => {
+        invokeLifecycleCallback(config[type], {
+          request: parsed,
+          ...extra,
+        });
+      };
+
+      const createDefaultResponseWithLifecycle = createDefaultResponse
+        ? async () => {
+          try {
+            if (!config.agent) {
+              return await createDefaultResponse();
+            }
+
+            if (parsed.tools.length > 0) {
+              if (!config.sessionManager) {
+                return await createDefaultResponse();
+              }
+
+              return await createAgUiRuntimeInjectedToolsStreamResponse(
+                config.agent,
+                parsed,
+                context,
+                config.sessionManager,
+                {
+                  onFinish: () => invokeLifecycle("onFinish"),
+                  onError: (error) => invokeLifecycle("onError", { error }),
+                  onToolCallSeen: (toolCallId) => invokeLifecycle("onToolCallSeen", { toolCallId }),
+                },
+              );
+            }
+
+            return await createAgUiRuntimeDirectStreamResponse(
+              config.agent,
+              parsed,
+              context,
+              {
+                onFinish: () => invokeLifecycle("onFinish"),
+                onError: (error) => invokeLifecycle("onError", { error }),
+                onToolCallSeen: (toolCallId) => invokeLifecycle("onToolCallSeen", { toolCallId }),
+              },
+            );
+          } catch (error) {
+            await invokeLifecycleCallbackAndWait(config.onError, {
+              request: parsed,
+              error,
+            });
+            throw error;
+          }
+        }
+        : undefined;
+
       if (config.execute) {
         return await config.execute({
           request,
           agUiInput: parsed,
           context,
-          createDefaultResponse,
+          createDefaultResponse: createDefaultResponseWithLifecycle,
         });
       }
 
-      if (createDefaultResponse) {
-        return await createDefaultResponse();
+      if (createDefaultResponseWithLifecycle) {
+        return await createDefaultResponseWithLifecycle();
       }
 
       throw new Error("createAgUiRuntimeHandler configuration became invalid during execution.");

package/src/src/agent/index.ts

@@ -147,6 +147,7 @@ export {
   type AgUiRuntimeHandlerExecute,
   type AgUiRuntimeHandlerExecuteInput,
   type AgUiRuntimeHandlerOptions,
+  type AgUiRuntimeLifecycleContext,
   createAgUiRuntimeHandler,
 } from "./ag-ui-runtime-handler.js";
 export {
@@ -191,6 +192,8 @@ export {
   type AgUiDetachedStartRequest,
   AgUiDetachedStartRequestSchema,
   createAgUiDetachedStartHandler,
+  executeAgUiDetachedStart,
+  type ExecuteAgUiDetachedStartInput,
 } from "./ag-ui-detached-start.js";
 export {
   type AgUiCancelHandlerOptions,

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.223";
+export const VERSION = "0.1.225";