@agent-native/core 0.77.16 → 0.77.18

package/dist/agent/production-agent.js

@@ -2975,6 +2975,20 @@ export function createProductionAgentHandler(options) {
             if (bgRunId)
                 void recordRunDiagnostic(bgRunId, RUN_DIAG_STAGE.workerSetupStep, `${s}=${Date.now() - setupT0}ms`).catch(() => { });
         };
+        // DIAGNOSTIC-ONLY: AWAITED milestone write. Fire-and-forget `workerStep`
+        // writes never land once the worker freezes (a blocked event loop can't
+        // flush async I/O), so for the key post-model_done milestones we AWAIT the
+        // write — it lands while the loop is still running, so the LAST milestone
+        // visible in /runs/active is exactly the one right before the freeze (a sync
+        // block, or a hung await).
+        const workerStepAwait = async (s) => {
+            if (!bgRunId)
+                return;
+            try {
+                await recordRunDiagnostic(bgRunId, RUN_DIAG_STAGE.workerSetupStep, `${s}=${Date.now() - setupT0}ms`);
+            }
+            catch { }
+        };
         // DIAGNOSTIC-ONLY: non-DB breadcrumb to the function log drain. `workerStep`
         // writes to the DB, which is exactly what stalls after `model_done` in the bg
         // worker (no diag write lands), so this logs to stdout (Netlify function
@@ -3247,7 +3261,7 @@ export function createProductionAgentHandler(options) {
         // DIAGNOSTIC-ONLY: engine/model/api-key resolution finished. A worker that
         // reached db_request_ctx but not env_config hung in attachment upload or
         // engine/model resolution.
-        workerStep("env_config");
+        await workerStepAwait("env_config");
         bgLog("env_config");
         bgLog("presend:creating");
         // Run all independent pre-send steps in parallel. Each of these hits
@@ -3497,7 +3511,7 @@ export function createProductionAgentHandler(options) {
         setupMark("ctxAll");
         // DIAGNOSTIC-ONLY: all parallel context gathering (system prompt, screen,
         // files, loop settings, enriched message) resolved.
-        workerStep("context_all");
+        await workerStepAwait("context_all");
         bgLog("context_all");
         if (systemPromptError) {
             setResponseHeader(event, "Content-Type", "text/event-stream");
@@ -3519,7 +3533,7 @@ export function createProductionAgentHandler(options) {
         const requestTools = filterInitialEngineTools(availableRequestTools, options.initialToolNames);
         setupMark("actions");
         // DIAGNOSTIC-ONLY: action/tool resolution + engine-tool filtering finished.
-        workerStep("action_tool_setup");
+        await workerStepAwait("action_tool_setup");
         bgLog("action_tool_setup");
         const requestSystemPrompt = requestMode === "plan"
             ? `${systemPrompt}\n\n${PLAN_MODE_SYSTEM_PROMPT}`
@@ -3619,7 +3633,7 @@ export function createProductionAgentHandler(options) {
         setupMark("depsThread");
         // DIAGNOSTIC-ONLY: owner/thread resolution + runId/effectiveThreadId +
         // chained-continuation thread fetch finished.
-        workerStep("owner_thread");
+        await workerStepAwait("owner_thread");
         bgLog("owner_thread");
         // Persist the user's turn exactly once. The foreground POST does this
         // before dispatching; the background worker must NOT repeat it (it re-enters
@@ -3978,7 +3992,7 @@ export function createProductionAgentHandler(options) {
         setupMark("preStart");
         // DIAGNOSTIC-ONLY: last stage before startRun fires. A worker that reaches
         // prestart but never workerStarted is hanging inside startRun itself.
-        workerStep("prestart");
+        await workerStepAwait("prestart");
         bgLog("prestart");
         const setupDetail = Object.entries(setupMarks)
             .map(([k, v]) => `${k}=${v}`)