fix(agents): align stale error follow-up with branch types

altaywtf · altaywtf · commit 05179362b439 · 2026-03-11T21:58:18.000+03:00
diff --git a/src/agents/pi-embedded-runner/run/payloads.errors.test.ts b/src/agents/pi-embedded-runner/run/payloads.errors.test.ts
@@ -104,7 +104,7 @@ describe("buildEmbeddedRunPayloads", () => {
   it("does not emit a synthetic billing error for successful turns with stale errorMessage", () => {
     const payloads = buildPayloads({
       lastAssistant: makeAssistant({
-        stopReason: "end_turn",
+        stopReason: "stop",
         errorMessage: "insufficient credits for embedding model",
         content: [{ type: "text", text: "Handle payment required errors in your API." }],
       }),
diff --git a/src/agents/pi-embedded-runner/run/payloads.ts b/src/agents/pi-embedded-runner/run/payloads.ts
@@ -133,11 +133,11 @@ export function buildEmbeddedRunPayloads(params: {
       ? suppressAssistantArtifacts
         ? undefined
         : formatAssistantErrorText(params.lastAssistant, {
-          cfg: params.config,
-          sessionKey: params.sessionKey,
-          provider: params.provider,
-          model: params.model,
-        })
+            cfg: params.config,
+            sessionKey: params.sessionKey,
+            provider: params.provider,
+            model: params.model,
+          })
       : undefined;
   const rawErrorMessage = lastAssistantErrored
     ? params.lastAssistant?.errorMessage?.trim() || undefined
diff --git a/src/agents/pi-embedded-utils.test.ts b/src/agents/pi-embedded-utils.test.ts
@@ -139,7 +139,7 @@ describe("extractAssistantText", () => {
     const msg = makeAssistantMessage({
       role: "assistant",
       errorMessage: "insufficient credits for embedding model",
-      stopReason: "end_turn",
+      stopReason: "stop",
       content: [{ type: "text", text: responseText }],
       timestamp: Date.now(),
     });
