From 05179362b439ff3b0330df3eb6a28aa05396ea9d Mon Sep 17 00:00:00 2001
From: Altay <altay@uinaf.dev>
Date: Wed, 11 Mar 2026 21:34:53 +0300
Subject: [PATCH] fix(agents): align stale error follow-up with branch types

---
 .../pi-embedded-runner/run/payloads.errors.test.ts     |  2 +-
 src/agents/pi-embedded-runner/run/payloads.ts          | 10 +++++-----
 src/agents/pi-embedded-utils.test.ts                   |  2 +-
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/agents/pi-embedded-runner/run/payloads.errors.test.ts b/src/agents/pi-embedded-runner/run/payloads.errors.test.ts
index 764ee8330fd..a2e7873aedf 100644
--- 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
index 542bc44564b..c0e0ded136e 100644
--- 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
index 5b5529e2da6..ab84a375d94 100644
--- 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(),
     });