@gea-ai/agent-sdk 0.1.260923-alpha.2 → 0.1.260923-alpha.4

package/dist/agent-session-inbox.js

@@ -500,17 +500,31 @@ export async function handleAgentSessionInboxRequest(request, storage) {
     if (!storage.setAlarm || !storage.deleteAlarm)
         return Response.json({ error: "session_alarms_unavailable" }, { status: 503 });
     const body = await request.json();
+    const admission = path === "/v1/inbox/start" || path === "/v1/inbox/enter"
+        ? agentSessionBootstrapSchema
+            .extend({
+            executionId: z.uuid(),
+            runId: z.uuid().optional(),
+            message: agentSessionSendSchema.optional(),
+            continuation: agentApplicationAssistantMessageSchema.optional(),
+            traceContext: traceContextSchema.optional(),
+            metadata: studioChatMetadataSchema.optional(),
+        })
+            .refine((input) => Boolean(input.message) !== Boolean(input.continuation), "Enter requires one message or validated continuation.")
+            .parse(body)
+        : undefined;
     return storage.transaction(async (tx) => {
-        const state = await readState(tx);
-        if (path === "/v1/inbox/initialize") {
-            const identity = agentSessionIdentitySchema.parse(body);
+        let state = await readState(tx);
+        if (path === "/v1/inbox/initialize" ||
+            (!state && path === "/v1/inbox/start")) {
+            const identity = admission?.identity ?? agentSessionIdentitySchema.parse(body);
             if (state)
                 return canonicalAgentChannelJson(state.identity) ===
                     canonicalAgentChannelJson(identity)
                     ? Response.json({ initialized: true })
                     : Response.json({ error: "session_identity_conflict" }, { status: 409 });
             await readChildRunIndex(tx);
-            await writeState(tx, {
+            state = {
                 identity,
                 messages: [],
                 execution: null,
@@ -525,8 +539,11 @@ export async function handleAgentSessionInboxRequest(request, storage) {
                 draining: null,
                 terminalQueue: [],
                 pendingStarts: [],
-            });
-            return Response.json({ initialized: true });
+            };
+            if (path === "/v1/inbox/initialize") {
+                await writeState(tx, state);
+                return Response.json({ initialized: true });
+            }
         }
         if (!state)
             return Response.json({ error: "session_not_initialized" }, { status: 409 });
@@ -542,25 +559,26 @@ export async function handleAgentSessionInboxRequest(request, storage) {
                         .filter((entry) => entry.mode === "follow_up" && !entry.id.startsWith("run:"))
                         .map(({ id, message }) => ({ id, message })),
                 });
+            case "start":
             case "enter": {
                 if (await activeComputerOperation(tx))
                     return Response.json({ error: "computer_busy" }, { status: 409 });
-                const input = agentSessionBootstrapSchema
-                    .extend({
-                    executionId: z.uuid(),
-                    runId: z.uuid().optional(),
-                    message: agentSessionSendSchema.optional(),
-                    continuation: agentApplicationAssistantMessageSchema.optional(),
-                    traceContext: traceContextSchema.optional(),
-                    metadata: studioChatMetadataSchema.optional(),
-                })
-                    .refine((input) => Boolean(input.message) !== Boolean(input.continuation), "Enter requires one message or validated continuation.")
-                    .parse(body);
+                const input = admission;
                 if (canonicalAgentChannelJson(state.identity) !==
                     canonicalAgentChannelJson(input.identity))
                     return Response.json({ error: "session_identity_conflict" }, { status: 409 });
-                if (state.execution?.id === input.executionId)
+                if (state.execution?.id === input.executionId) {
+                    if (path === "/v1/inbox/start") {
+                        if (state.started)
+                            return Response.json({ error: "execution_already_started" }, {
+                                status: 409,
+                                headers: { "x-gea-session-execution": "already-started" },
+                            });
+                        state.started = true;
+                        await save(tx, state, runs);
+                    }
                     return Response.json({ execution: state.execution });
+                }
                 if (state.execution ||
                     state.draining ||
                     state.terminalQueue.length >= 8 ||
@@ -613,7 +631,7 @@ export async function handleAgentSessionInboxRequest(request, storage) {
                 state.resumeWait = null;
                 if (!state.draining)
                     state.stopped = false;
-                state.started = false;
+                state.started = path === "/v1/inbox/start";
                 if (input.message)
                     await tx.put(`inbox:message:${input.message.id}`, input.message);
                 await save(tx, state, runs);

package/dist/agent-worker.d.ts

@@ -102,6 +102,10 @@ type AgentApplicationDefinition = {
     skillMount?: AgentWorkerSkillMount;
     subagents?: Record<string, AgentApplicationDefinition>;
 };
+type AgentStartupTiming = {
+    startedAt: number;
+    phases: Record<string, number>;
+};
 export declare function composeFetch<TEnvironment, TContext>(...handlers: ReadonlyArray<ComposableFetch<TEnvironment, TContext>>): (request: Request, environment: TEnvironment, context: TContext) => Promise<Response>;
 export declare function createAgentFetch<TEnvironment extends AgentFetchEnvironment = AgentFetchEnvironment, TContext = unknown>(definition: DefinedAgentPackage): ComposableFetch<TEnvironment, TContext>;
 /**
@@ -192,6 +196,8 @@ export type AgentSessionNamespace = {
     idFromName(name: string): unknown;
 };
 export declare function runAgentWorkerRequest(input: {
+    startup?: AgentStartupTiming;
+    inboxStarted?: boolean;
     interactionPolicy?: AgentInteractionPolicy;
     initialization?: AgentWorkerToolInitialization;
     inboxExecution?: AgentSessionInboxExecution;