@gea-ai/agent-sdk 0.1.261009-alpha.1 → 0.1.261012-alpha.0

package/dist/agent-call.d.ts

@@ -1,6 +1,32 @@
+import { z } from "zod";
 import { type AgentWorkerToolInitialization } from "@gea-ai/contract/agent-worker-tools";
 import type { AgentHostBinding } from "./agent-worker.js";
 import type { StudioArtifactScope } from "@gea-ai/contract/studio-artifacts";
+/** Fresh Sessions keep the declared-target rule. A resume only needs sessionId. */
+export declare function workerAgentToolInputSchema(input: {
+    agents: NonNullable<AgentWorkerToolInitialization["agents"]>;
+    canCopy?: boolean;
+    artifactScope?: StudioArtifactScope;
+}): z.ZodObject<{
+    sessionId: z.ZodOptional<z.ZodUUID>;
+    target: z.ZodOptional<z.ZodString>;
+    message: z.ZodUnion<readonly [z.ZodString, z.ZodObject<{
+        role: z.ZodLiteral<"user">;
+        parts: z.ZodArray<z.ZodObject<{
+            type: z.ZodLiteral<"text">;
+            text: z.ZodString;
+        }, z.core.$strict>>;
+    }, z.core.$strict>]> | z.ZodUnion<readonly [z.ZodString, z.ZodObject<{
+        role: z.ZodLiteral<"user">;
+        parts: z.ZodArray<z.ZodUnion<readonly [z.ZodObject<{
+            type: z.ZodLiteral<"text">;
+            text: z.ZodString;
+        }, z.core.$strict>, z.ZodObject<{
+            type: z.ZodLiteral<"file">;
+            file_id: z.ZodUUID;
+        }, z.core.$strict>]>>;
+    }, z.core.$strict>]>;
+}, z.core.$strip>;
 /** The host opens a fresh Worker invocation; only its visible answer returns to the caller. */
 export declare function createWorkerAgentTool(input: {
     agents: NonNullable<AgentWorkerToolInitialization["agents"]>;
@@ -14,7 +40,7 @@ export declare function createWorkerAgentTool(input: {
     metadata?: import("@ai-sdk/provider").JSONObject;
     inputSchema: import("ai").FlexibleSchema<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -35,7 +61,7 @@ export declare function createWorkerAgentTool(input: {
     contextSchema?: import("ai").FlexibleSchema<import("@ai-sdk/provider-utils").Context> | undefined;
     needsApproval?: boolean | import("@ai-sdk/provider-utils").ToolNeedsApprovalFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -60,7 +86,7 @@ export declare function createWorkerAgentTool(input: {
     onInputAvailable?: ((options: {
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -83,7 +109,7 @@ export declare function createWorkerAgentTool(input: {
         toolCallId: string;
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -145,7 +171,7 @@ export declare function createWorkerAgentTool(input: {
     }> | undefined;
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -191,7 +217,7 @@ export declare function createWorkerAgentTool(input: {
     inputExamples?: {
         input: NoInfer<{
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -219,7 +245,7 @@ export declare function createWorkerAgentTool(input: {
 } & {
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -263,7 +289,7 @@ export declare function createWorkerAgentTool(input: {
     metadata?: import("@ai-sdk/provider").JSONObject;
     inputSchema: import("ai").FlexibleSchema<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -284,7 +310,7 @@ export declare function createWorkerAgentTool(input: {
     contextSchema?: import("ai").FlexibleSchema<import("@ai-sdk/provider-utils").Context> | undefined;
     needsApproval?: boolean | import("@ai-sdk/provider-utils").ToolNeedsApprovalFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -309,7 +335,7 @@ export declare function createWorkerAgentTool(input: {
     onInputAvailable?: ((options: {
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -332,7 +358,7 @@ export declare function createWorkerAgentTool(input: {
         toolCallId: string;
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -394,7 +420,7 @@ export declare function createWorkerAgentTool(input: {
     }> | undefined;
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -440,7 +466,7 @@ export declare function createWorkerAgentTool(input: {
     inputExamples?: {
         input: NoInfer<{
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -468,7 +494,7 @@ export declare function createWorkerAgentTool(input: {
 } & {
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -512,7 +538,7 @@ export declare function createWorkerAgentTool(input: {
     metadata?: import("@ai-sdk/provider").JSONObject;
     inputSchema: import("ai").FlexibleSchema<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -533,7 +559,7 @@ export declare function createWorkerAgentTool(input: {
     contextSchema?: import("ai").FlexibleSchema<import("@ai-sdk/provider-utils").Context> | undefined;
     needsApproval?: boolean | import("@ai-sdk/provider-utils").ToolNeedsApprovalFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -558,7 +584,7 @@ export declare function createWorkerAgentTool(input: {
     onInputAvailable?: ((options: {
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -581,7 +607,7 @@ export declare function createWorkerAgentTool(input: {
         toolCallId: string;
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -643,7 +669,7 @@ export declare function createWorkerAgentTool(input: {
     }> | undefined;
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -693,7 +719,7 @@ export declare function createWorkerAgentTool(input: {
 } & {
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -737,7 +763,7 @@ export declare function createWorkerAgentTool(input: {
     metadata?: import("@ai-sdk/provider").JSONObject;
     inputSchema: import("ai").FlexibleSchema<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -758,7 +784,7 @@ export declare function createWorkerAgentTool(input: {
     contextSchema?: import("ai").FlexibleSchema<import("@ai-sdk/provider-utils").Context> | undefined;
     needsApproval?: boolean | import("@ai-sdk/provider-utils").ToolNeedsApprovalFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -783,7 +809,7 @@ export declare function createWorkerAgentTool(input: {
     onInputAvailable?: ((options: {
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -806,7 +832,7 @@ export declare function createWorkerAgentTool(input: {
         toolCallId: string;
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -868,7 +894,7 @@ export declare function createWorkerAgentTool(input: {
     }> | undefined;
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -918,7 +944,7 @@ export declare function createWorkerAgentTool(input: {
 } & {
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -1225,7 +1251,9 @@ export declare function createWorkerRunCancelTool({ host, }: {
         status: "abort_requested" | "not_running";
     }, NoInfer<import("@ai-sdk/provider-utils").Context>>;
 });
-export declare function readWorkerAgentResponse(response: Response): Promise<{
+export declare function readWorkerAgentResponse(response: Response, options?: {
+    persistedSessionWaiting?: () => Promise<boolean>;
+}): Promise<{
     status: "completed" | "suspended";
     text: string;
 }>;

package/dist/agent-call.js

@@ -5,20 +5,45 @@ import { parseJsonEventStream } from "@ai-sdk/provider-utils";
 import { tool, uiMessageChunkSchema } from "ai";
 import { agentSubagentConcurrencyErrorSchema, agentRemoteAdmissionErrorSchema, agentCallInputSchema, agentCallTextMessageSchema, agentRunCallResultSchema, getAgentCallFileIds, } from "@gea-ai/contract/agent-worker-tools";
 import { captureTraceContext } from "./trace-context.js";
+/** Fresh Sessions keep the declared-target rule. A resume only needs sessionId. */
+export function workerAgentToolInputSchema(input) {
+    const message = input.artifactScope
+        ? agentCallInputSchema.shape.message
+        : agentCallTextMessageSchema;
+    const keys = new Set(input.agents.map((agent) => agent.key));
+    return z
+        .object({
+        sessionId: z.uuid().optional(),
+        target: z.string().trim().min(1).max(80).optional(),
+        message,
+    })
+        .superRefine((value, context) => {
+        if (value.sessionId)
+            return;
+        if (value.target === undefined) {
+            if (keys.size > 0 && !input.canCopy) {
+                context.addIssue({
+                    code: "custom",
+                    path: ["target"],
+                    message: "A new Session requires a declared target.",
+                });
+            }
+            return;
+        }
+        if (!keys.has(value.target)) {
+            context.addIssue({
+                code: "custom",
+                path: ["target"],
+                message: "Select a declared target for a new Session.",
+            });
+        }
+    });
+}
 /** The host opens a fresh Worker invocation; only its visible answer returns to the caller. */
 export function createWorkerAgentTool(input) {
     return tool({
-        description: `Start an Agent Run in a new or idle child Session. Use runWait with returned runIds and runCancel to cancel. Use sessionId to continue the same Session. ${input.canCopy ? "Omit target to create a copy of yourself." : "Select a declared target."} Available targets: ${JSON.stringify(input.agents)}. Supply context in message.${input.artifactScope ? " To send selected files, use a user message with file parts containing immutable Artifact file_id values. Refer to attachments by filename or order in task text; remote transfer may assign new IDs, exposed to the child in Attached Artifact metadata. IDs hardcoded in text are not rewritten." : ""}`,
-        inputSchema: agentCallInputSchema.extend({
-            target: input.agents.length
-                ? input.canCopy
-                    ? z.enum(input.agents.map((agent) => agent.key)).optional()
-                    : z.enum(input.agents.map((agent) => agent.key))
-                : z.never().optional(),
-            message: input.artifactScope
-                ? agentCallInputSchema.shape.message
-                : agentCallTextMessageSchema,
-        }),
+        description: `Start an Agent Run in a new or idle child Session. Use runWait with returned runIds and runCancel to cancel. Use sessionId to continue the same Session; target may be omitted or name the recorded Agent. ${input.canCopy ? "Omit target to create a copy of yourself." : "Select a declared target for a new Session."} Available targets: ${JSON.stringify(input.agents)}. Supply context in message.${input.artifactScope ? " To send selected files, use a user message with file parts containing immutable Artifact file_id values. Refer to attachments by filename or order in task text; remote transfer may assign new IDs, exposed to the child in Attached Artifact metadata. IDs hardcoded in text are not rewritten." : ""}`,
+        inputSchema: workerAgentToolInputSchema(input),
         execute: async (value, options) => {
             const fileIds = getAgentCallFileIds(value.message);
             const response = await input.host.fetch("/internal/agent-worker-host/v1/agents/runs", {
@@ -70,6 +95,23 @@ export function createWorkerAgentTool(input) {
                             error: "child_busy",
                             message: "This child Session has unfinished work. Wait for its Run before continuing the Session.",
                         };
+                    if (response.status === 403) {
+                        const rejection = z
+                            .object({
+                            error: z
+                                .string()
+                                .max(80)
+                                .regex(/^[A-Za-z0-9_.-]+$/),
+                        })
+                            .safeParse(value);
+                        if (rejection.success) {
+                            console.error("Child run rejected", {
+                                status: 403,
+                                error: rejection.data.error,
+                            });
+                            throw new Error(`Agent invocation was rejected (403 ${rejection.data.error}).`);
+                        }
+                    }
                 }
                 throw new Error(`Agent invocation was rejected (${response.status}).`);
             }
@@ -99,7 +141,15 @@ export function createWorkerRunCancelTool({ host, }) {
 }
 class AgentChildRunResponseError extends TaggedError("AgentChildRunResponseError")() {
 }
-export async function readWorkerAgentResponse(response) {
+const childDiagnosticIdentifierPattern = /^[A-Za-z0-9_.-]+$/;
+function childDiagnosticIdentifier(value) {
+    return typeof value === "string" &&
+        value.length <= 80 &&
+        childDiagnosticIdentifierPattern.test(value)
+        ? value
+        : undefined;
+}
+export async function readWorkerAgentResponse(response, options) {
     if (response.status === 409 &&
         response.headers.get("x-gea-session-execution") === "already-started") {
         await response.body?.cancel();
@@ -117,6 +167,9 @@ export async function readWorkerAgentResponse(response) {
     let completed = false;
     let suspended = false;
     let failure;
+    let legacyToolCalls = false;
+    let otherUnacceptableFinish = false;
+    let sawStreamFailure = false;
     const reader = parseJsonEventStream({
         stream: response.body,
         schema: uiMessageChunkSchema,
@@ -133,26 +186,46 @@ export async function readWorkerAgentResponse(response) {
             const chunk = event.value;
             if (chunk.type === "text-delta")
                 text += chunk.delta;
-            if (chunk.type === "error")
+            if (chunk.type === "error") {
+                sawStreamFailure = true;
                 failure ??= new AgentChildRunResponseError({
                     code: "child_stream_error",
                     message: chunk.errorText,
                 });
-            if (chunk.type === "abort")
+            }
+            if (chunk.type === "abort") {
+                sawStreamFailure = true;
                 failure ??= new AgentChildRunResponseError({
                     code: "child_stream_aborted",
                     message: "Child Agent was cancelled.",
                 });
+            }
             if (chunk.type === "finish") {
                 suspended = z
                     .object({ sessionStatus: z.literal("waiting") })
                     .safeParse(chunk.messageMetadata).success;
-                if (chunk.finishReason !== "stop" &&
-                    !(suspended && chunk.finishReason === "tool-calls"))
+                const unacceptable = chunk.finishReason !== "stop" &&
+                    !(suspended && chunk.finishReason === "tool-calls");
+                if (unacceptable) {
+                    if (chunk.finishReason === "tool-calls" && !suspended)
+                        legacyToolCalls = true;
+                    else
+                        otherUnacceptableFinish = true;
+                    const parsedSession = z
+                        .object({ sessionStatus: z.unknown() })
+                        .safeParse(chunk.messageMetadata);
+                    const rawSessionStatus = parsedSession.success
+                        ? parsedSession.data.sessionStatus
+                        : undefined;
+                    const sessionStatus = childDiagnosticIdentifier(rawSessionStatus);
                     failure ??= new AgentChildRunResponseError({
                         code: "child_finish_reason",
                         message: `Child Agent did not complete (${chunk.finishReason ?? "unknown"}).`,
+                        finishReason: childDiagnosticIdentifier(chunk.finishReason) ?? "unknown",
+                        hasSessionStatus: rawSessionStatus !== undefined,
+                        ...(sessionStatus ? { sessionStatus } : {}),
                     });
+                }
                 completed = true;
             }
         }
@@ -161,6 +234,18 @@ export async function readWorkerAgentResponse(response) {
         await reader.cancel();
         reader.releaseLock();
     }
+    if (options?.persistedSessionWaiting &&
+        legacyToolCalls &&
+        !otherUnacceptableFinish &&
+        !sawStreamFailure &&
+        failure instanceof AgentChildRunResponseError &&
+        failure.code === "child_finish_reason") {
+        const waiting = await Result.tryPromise(() => options.persistedSessionWaiting());
+        if (waiting.isOk() && waiting.value) {
+            failure = undefined;
+            suspended = true;
+        }
+    }
     if (failure)
         throw failure;
     if (!completed)

package/dist/agent-channel-api.js

@@ -40,7 +40,7 @@ export function extractChannelReply(messageValue, responseMessageId) {
 }
 /** Channel owns delivery; ordinary Agents API owns admission, execution and history. */
 export async function executeChannelAgentRun(input) {
-    const fetchApi = async (path, body) => {
+    const fetchApi = async (path, body, headers) => {
         input.signal.throwIfAborted();
         const response = await input.binding.fetch(`/api/v1${path}`, {
             method: body === undefined ? "GET" : "POST",
@@ -48,7 +48,7 @@ export async function executeChannelAgentRun(input) {
             ...(body === undefined
                 ? {}
                 : {
-                    headers: { "content-type": "application/json" },
+                    headers: { "content-type": "application/json", ...headers },
                     body: JSON.stringify(body),
                 }),
         });
@@ -67,7 +67,7 @@ export async function executeChannelAgentRun(input) {
         id: input.runId,
         input: input.message,
         stream: false,
-    })).body?.cancel();
+    }, { "idempotency-key": input.runId })).body?.cancel();
     let stream = true;
     for (;;) {
         const current = runSchema.parse(await (await fetchApi(`/runs/${input.runId}`)).json());

package/dist/agent-channel-author-runtime.d.ts

@@ -19,19 +19,6 @@ export declare function executeChannelBehavior(input: {
     environment: AgentChannelFetchEnvironment;
     signal: AbortSignal;
 }): Promise<{
-    sequence: number;
-    duplicate: boolean;
-} | {
-    state: AgentChannelJson;
-    intents: {
-        key: string;
-        payload: AgentChannelJson;
-        eventId: string;
-        operationId: string;
-    }[];
-    cursor: number;
-    idempotencyWindowMs?: number | undefined;
-} | {
     status: "sent";
     receipt?: AgentChannelJson | undefined;
 } | {
@@ -56,6 +43,19 @@ export declare function executeChannelBehavior(input: {
         message: string;
         retryable: boolean;
     };
+} | {
+    sequence: number;
+    duplicate: boolean;
+} | {
+    state: AgentChannelJson;
+    intents: {
+        key: string;
+        payload: AgentChannelJson;
+        eventId: string;
+        operationId: string;
+    }[];
+    cursor: number;
+    idempotencyWindowMs?: number | undefined;
 } | {
     events: {
         id: string;

package/dist/agent-remote-run.js

@@ -62,13 +62,42 @@ export async function submitRemoteRun(fetch, input, options = {}) {
             }))).id;
             signal.throwIfAborted();
             submitting = true;
-            // No automatic replay: a lost response does not prove Run admission failed.
-            const response = await fetch(`/sessions/${sessionId}/runs`, {
-                method: "POST",
-                headers: { "content-type": "application/json" },
-                signal,
-                body: JSON.stringify({ input: message, stream: false }),
+            // One key per logical submission. Retries reuse it so a lost 202 cannot
+            // start a second Run. A later call to submitRemoteRun is a new request.
+            const idempotencyKey = crypto.randomUUID();
+            const runBody = JSON.stringify({
+                id: idempotencyKey,
+                input: message,
+                stream: false,
             });
+            const runHeaders = {
+                "content-type": "application/json",
+                "idempotency-key": idempotencyKey,
+            };
+            let response;
+            for (let attempt = 0; attempt < 3; attempt++) {
+                signal.throwIfAborted();
+                try {
+                    response = await fetch(`/sessions/${sessionId}/runs`, {
+                        method: "POST",
+                        headers: runHeaders,
+                        signal,
+                        body: runBody,
+                    });
+                }
+                catch (error) {
+                    if (signal.aborted || attempt === 2)
+                        throw error;
+                    continue;
+                }
+                const retryable = [502, 503, 504].includes(response.status);
+                const admitted = response.headers.has("x-gea-agent-run-id");
+                if (!retryable || admitted || attempt === 2)
+                    break;
+                await response.body?.cancel();
+            }
+            if (!response)
+                throw new Error("Remote Run was not submitted.");
             rejected = [400, 401, 403, 404, 405, 409, 413, 415, 422, 429].includes(response.status);
             const run = agentHttpRunSchema.parse(await responseJson(response));
             if (run.session_id !== sessionId)

package/dist/agent-run-host.d.ts

@@ -7,21 +7,33 @@ export type { RemoteRunFetch, RemoteRunDestination, RemoteInputFiles, RemoteFile
 export { readWorkerAgentResponse } from "./agent-call.js";
 export { streamManagedAgentSessionRun } from "./agent-session-stream.js";
 export type AgentChildRunSession = (operation: string, body: unknown, traceContext?: TraceContext) => Promise<Response>;
+/**
+ * Renders child Run updates for a parent model turn. Upstream dropped the task-era
+ * helper without a replacement; App Server adapters still need one summary shape.
+ * JSON escaping can expand each character sixfold; leave room for Run identities.
+ */
+export declare function formatAgentChildRunUpdates(runs: Pick<AgentChildRun, "id" | "sessionId" | "agentKey" | "target" | "outcome">[]): string;
 /** Invocation transport only. SDK Sessions own Run state, waits, completion and wakeups. */
 export declare class AgentRunHost {
     private readonly host;
     private readonly executions;
+    private readonly executionDone;
     private readonly work;
     private closed;
     private readonly shutdown;
     constructor(host: {
         session: AgentChildRunSession;
+        /** Native App Server owns Worker handles; this callback only admits the resolved target. */
+        dispatch?(run: AgentChildRun, execution?: AgentSessionInboxExecution): Promise<void>;
+        /** Authorize against the root Session and return its retained child binding. */
+        resolveSession?(sessionId: string): Promise<Response>;
         remoteFetch?(destination: RemoteRunDestination, path: string, init?: RequestInit): Promise<Response>;
         transferRemoteInputs?(destination: RemoteRunDestination, files: RemoteInputFiles, scope: StudioArtifactScope | undefined, signal: AbortSignal): Promise<RemoteFileMapping>;
         register(run: AgentChildRun): Promise<string>;
         invoke(run: AgentChildRun, signal: AbortSignal, continuation?: TraceContinuation, inboxExecution?: AgentSessionInboxExecution): Promise<{
             response: Response;
             dispose?: () => void;
+            persistedSessionWaiting?: () => Promise<boolean>;
         }>;
         cancelChild(run: AgentChildRun): Promise<boolean | void>;
         onError(error: unknown, identity?: {