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

package/dist/agent-worker.js

@@ -2688,6 +2688,7 @@ function replaceLastAssistantInteraction(input) {
     ];
 }
 export async function runAgentWorkerRequest(input) {
+    const requestStartedAt = performance.now();
     const sessionChatId = input.inboxExecution || input.sessionExecution
         ? runRequestSchema.parse(await input.request.clone().json()).chatId
         : undefined;
@@ -2708,6 +2709,7 @@ export async function runAgentWorkerRequest(input) {
     }, async (signal, waitUntil) => {
         const invocationTraceContext = captureTraceContext();
         const invocationSpan = trace.getActiveSpan();
+        invocationSpan?.setAttribute("gea.startup.pre_trace_ms", Math.round(performance.now() - requestStartedAt));
         const invocationTiming = {};
         const cancellation = new AbortController();
         let stopComputer;
@@ -2919,6 +2921,8 @@ export async function runAgentWorkerRequest(input) {
             await finish();
             return response;
         }
+        invocationSpan?.setAttribute("gea.startup.response_headers_ms", Math.round(performance.now() - requestStartedAt));
+        let receivedFirstChunk = false;
         const reader = response.body.getReader();
         const cancel = async (reason) => {
             signal.removeEventListener("abort", onAbort);
@@ -2942,6 +2946,10 @@ export async function runAgentWorkerRequest(input) {
                 try {
                     const next = await reader.read();
                     if (!next.done) {
+                        if (!receivedFirstChunk && next.value.byteLength > 0) {
+                            receivedFirstChunk = true;
+                            invocationSpan?.setAttribute("gea.startup.first_chunk_ms", Math.round(performance.now() - requestStartedAt));
+                        }
                         controller.enqueue(next.value);
                         return;
                     }
@@ -2972,6 +2980,7 @@ export async function runAgentWorkerRequest(input) {
 }
 async function executeAgentWorkerRequest(input, waitUntil, ownComputer, ownSessionOutcome, invocationTiming) {
     const invocationTrace = activeTraceRef();
+    const startupTracing = createAgentTracing(input.request.signal);
     const host = input.bindings
         ? createAgentWorkerHostAdapter(input.bindings)
         : input.host;
@@ -3010,14 +3019,14 @@ async function executeAgentWorkerRequest(input, waitUntil, ownComputer, ownSessi
     if (sessionExecution) {
         if (sessionExecution.runId !== payload.runId)
             throw new Error("Session Run identity mismatch.");
-        const admission = await requestSessionJson({
+        const admission = await startupTracing.withSpan("agent.startup.session_renew", () => requestSessionJson({
             body: { ...sessionExecution, leaseOwner: "worker" },
             chatId: payload.chatId,
             method: "POST",
             namespace: input.sessionNamespace,
             path: "/v2/turns/renew",
             schema: agentSessionExecutionStartResultSchema,
-        });
+        }));
         // Renewal atomically validates the attempt and returns cancellation intent;
         // a separate metadata read before it only adds a network round trip.
         if (admission.turn.cancelRequestedAt)
@@ -3106,29 +3115,6 @@ async function executeAgentWorkerRequest(input, waitUntil, ownComputer, ownSessi
         });
         return result;
     };
-    const initializationPromise = input.initialization
-        ? Promise.resolve(input.initialization)
-        : input.parentRunId !== undefined ||
-            !!input.snapshot.subagents?.length ||
-            !!Object.keys(input.snapshot.agentReferences ?? {}).length ||
-            requiresToolInitialization(input.snapshot, input.hookRuntime)
-            ? requireHostJson(host, "/tools/initialize", {
-                headers: { "x-gea-agent-runs": "1" },
-            }).then((value) => agentWorkerToolInitializationSchema.parse(value))
-            : Promise.resolve(agentWorkerToolInitializationSchema.parse({
-                connectors: [],
-                conversation: {
-                    canListHistory: false,
-                    canRead: false,
-                    canSend: false,
-                    canStartChild: false,
-                },
-                runtime: {
-                    auth: { current: null },
-                    environment: "production",
-                    variables: {},
-                },
-            }));
     const configuredModel = input.snapshot.model ?? payload.model;
     if (!configuredModel) {
         throw new Error("Agent Package does not declare a public model id.");
@@ -3144,9 +3130,54 @@ async function executeAgentWorkerRequest(input, waitUntil, ownComputer, ownSessi
             };
         }
     }
-    const [identityValue, initialization] = await Promise.all([
-        requireHostJson(host, "/context"),
-        initializationPromise,
+    // These reads depend only on the admitted request, not each other or the
+    // Computer. Resolve them together before allocating resources or writing input.
+    const [identityValue, initialization, engine, initialCoreConnection, model, storedContext,] = await Promise.all([
+        startupTracing.withSpan("agent.startup.identity", () => requireHostJson(host, "/context")),
+        input.initialization
+            ? Promise.resolve(input.initialization)
+            : input.parentRunId !== undefined ||
+                !!input.snapshot.subagents?.length ||
+                !!Object.keys(input.snapshot.agentReferences ?? {}).length ||
+                requiresToolInitialization(input.snapshot, input.hookRuntime)
+                ? startupTracing.withSpan("agent.startup.tools", () => requireHostJson(host, "/tools/initialize", {
+                    headers: { "x-gea-agent-runs": "1" },
+                }).then((value) => agentWorkerToolInitializationSchema.parse(value)))
+                : Promise.resolve(agentWorkerToolInitializationSchema.parse({
+                    connectors: [],
+                    conversation: {
+                        canListHistory: false,
+                        canRead: false,
+                        canSend: false,
+                        canStartChild: false,
+                    },
+                    runtime: {
+                        auth: { current: null },
+                        environment: "production",
+                        variables: {},
+                    },
+                })),
+        input.engine === "ai-sdk"
+            ? undefined
+            : input.engine === undefined || input.engine === "agent-core"
+                ? startupTracing.withSpan("agent.startup.engine", async () => (await import("./agent-core.js")).agentCore())
+                : input.engine,
+        input.engine !== "ai-sdk"
+            ? startupTracing.withSpan("agent.startup.model", () => createGeaCoreConnection(host, configuredModel, requestOptions, input.request.signal, input.snapshot.maxOutputTokens))
+            : undefined,
+        input.engine !== "ai-sdk"
+            ? undefined
+            : startupTracing.withSpan("agent.startup.model", () => createGeaAI(host, configuredModel, requestOptions.extraBody, {
+                kind: "agent-dev",
+                packageName: input.snapshot.name,
+            }, requestOptions.requestBody)),
+        startupTracing.withSpan("agent.startup.model_context", () => requestSessionJson({
+            chatId: payload.chatId,
+            method: "GET",
+            namespace: input.sessionNamespace,
+            path: sessionExecution ? "/v2/model-context" : "/v1/model-context",
+            schema: agentSessionGetModelContextResultSchema,
+        })),
     ]);
     const identity = agentToolIdentitySchema.parse(identityValue);
     if (input.snapshot.artifacts?.scope === "custom" &&
@@ -3211,7 +3242,7 @@ async function executeAgentWorkerRequest(input, waitUntil, ownComputer, ownSessi
     let computerRoots = defaultComputerRoots;
     let mountedSkillRoot;
     if (input.bindings && input.snapshot.computer.enabled) {
-        const scope = await resolveAgentComputerScope({
+        const scope = await startupTracing.withSpan("agent.startup.computer_scope", () => resolveAgentComputerScope({
             namespace: input.sessionNamespace,
             chatId: payload.chatId,
             principal: initialization.runtime.auth.current,
@@ -3219,14 +3250,14 @@ async function executeAgentWorkerRequest(input, waitUntil, ownComputer, ownSessi
             policy: input.snapshot.computer.filesystem?.scope ?? "chat",
             resolver: input.scopeResolver,
             signal: input.request.signal,
-        });
+        }));
         const scopedBindings = input.bindings;
         let computer = bindAgentComputerScope(scopedBindings.computer, scope);
         if (input.snapshot.computer.provider) {
-            const retained = await readSessionComputer(input.sessionNamespace, payload.chatId);
+            const retained = await startupTracing.withSpan("agent.startup.computer_reference", () => readSessionComputer(input.sessionNamespace, payload.chatId));
             if (retained && !retained.reference)
                 throw new Error("Session Computer preparation has not completed. Prepare it before starting a Run.");
-            const opened = await createAgentComputerSession({
+            const opened = await startupTracing.withSpan("agent.startup.computer_open", () => createAgentComputerSession({
                 computer,
                 retain: Boolean(retained),
                 reference: retained?.reference ?? undefined,
@@ -3241,7 +3272,7 @@ async function executeAgentWorkerRequest(input, waitUntil, ownComputer, ownSessi
                     metadata: payload.metadata,
                     scope,
                 },
-            });
+            }));
             ownComputer(opened.stop);
             computer = opened.computer;
             computerRoots = opened.roots;
@@ -3587,30 +3618,9 @@ async function executeAgentWorkerRequest(input, waitUntil, ownComputer, ownSessi
             throw new Error(`Agent Worker Tool runtime is missing: ${toolName}`);
         }
     }
-    const engine = input.engine === "ai-sdk"
-        ? undefined
-        : input.engine === undefined || input.engine === "agent-core"
-            ? (await import("./agent-core.js")).agentCore()
-            : input.engine;
-    const coreModule = engine?.module();
-    const [initialCoreConnection, model, storedContext] = await Promise.all([
-        engine
-            ? createGeaCoreConnection(host, configuredModel, requestOptions, input.request.signal, input.snapshot.maxOutputTokens)
-            : undefined,
-        engine
-            ? undefined
-            : createGeaAI(host, configuredModel, requestOptions.extraBody, {
-                kind: "agent-dev",
-                packageName: input.snapshot.name,
-            }, requestOptions.requestBody),
-        requestSessionJson({
-            chatId: payload.chatId,
-            method: "GET",
-            namespace: input.sessionNamespace,
-            path: sessionExecution ? "/v2/model-context" : "/v1/model-context",
-            schema: agentSessionGetModelContextResultSchema,
-        }),
-    ]);
+    const coreModule = engine
+        ? await startupTracing.withSpan("agent.startup.engine_module", () => engine.module())
+        : undefined;
     let coreConnection = initialCoreConnection;
     // Model file parts can contain URL and byte values that ordinary JSON cannot preserve.
     const runtimeContext = decodeAgentModelProxyValue(payload.runtimeContext);
@@ -3945,7 +3955,7 @@ async function executeAgentWorkerRequest(input, waitUntil, ownComputer, ownSessi
         return pending;
     };
     if (startsNewTurn && findLatestUserMessage(currentTurn)) {
-        await persistModelContext();
+        await startupTracing.withSpan("agent.startup.initial_checkpoint", persistModelContext);
     }
     const writeContextArtifact = async (artifact) => {
         const output = { id: artifact.sha256, content: artifact.content };
@@ -4097,7 +4107,7 @@ async function executeAgentWorkerRequest(input, waitUntil, ownComputer, ownSessi
             messages: prepared.slice(instructionCount),
         };
     };
-    const initialPrompt = await preparePrompt(withTurnImages(assembleAgentWorkerContext({ context, runtimeContext }), turnImages));
+    const initialPrompt = await startupTracing.withSpan("agent.startup.prompt", () => preparePrompt(withTurnImages(assembleAgentWorkerContext({ context, runtimeContext }), turnImages)));
     const modelCancellation = new AbortController();
     let failure;
     let pendingMainCall;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gea-ai/agent-sdk",
-  "version": "0.1.260923-alpha.2",
+  "version": "0.1.260923-alpha.3",
   "private": false,
   "homepage": "https://musegea.com/developers",
   "license": "Apache-2.0",
@@ -99,7 +99,7 @@
     "@ai-sdk/otel": "1.0.9",
     "@ai-sdk/provider": "4.0.1",
     "@ai-sdk/provider-utils": "5.0.2",
-    "@gea-ai/contract": "0.1.260923-alpha.2",
+    "@gea-ai/contract": "0.1.260923-alpha.3",
     "@opentelemetry/api": "1.9.1",
     "@opentelemetry/context-async-hooks": "2.10.0",
     "@opentelemetry/core": "2.10.0",