From df9f29caefee9245ce4f9ca4a867291e2b50694a Mon Sep 17 00:00:00 2001
From: Vincent Koc <vincentkoc@ieee.org>
Date: Sun, 17 May 2026 01:54:36 +0800
Subject: [PATCH] test(agents): stabilize embedded runner release checks

---
 src/agents/pi-embedded-runner.e2e.test.ts             | 11 ++++++++++-
 .../auto-reply/trigger-handling-test-harness.ts       |  2 ++
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/agents/pi-embedded-runner.e2e.test.ts b/src/agents/pi-embedded-runner.e2e.test.ts
index e73abe8e89b..9edddd3e86d 100644
--- a/src/agents/pi-embedded-runner.e2e.test.ts
+++ b/src/agents/pi-embedded-runner.e2e.test.ts
@@ -345,8 +345,17 @@ describe("runEmbeddedPiAgent", () => {
 
   it("resolves explicit OpenAI PI runs through Codex when auth order starts with Codex OAuth", async () => {
     const sessionFile = nextSessionFile();
+    const baseConfig = createEmbeddedPiRunnerOpenAiConfig(["mock-1"]);
     const cfg = {
-      ...createEmbeddedPiRunnerOpenAiConfig(["mock-1"]),
+      ...baseConfig,
+      models: {
+        providers: {
+          openai: {
+            ...baseConfig.models?.providers?.openai,
+            baseUrl: "https://api.openai.com/v1",
+          },
+        },
+      },
       agents: {
         defaults: {
           models: {
diff --git a/test/helpers/auto-reply/trigger-handling-test-harness.ts b/test/helpers/auto-reply/trigger-handling-test-harness.ts
index 54ca47cac9e..52112ff96a4 100644
--- a/test/helpers/auto-reply/trigger-handling-test-harness.ts
+++ b/test/helpers/auto-reply/trigger-handling-test-harness.ts
@@ -76,6 +76,8 @@ vi.doMock("../../../src/agents/pi-embedded-runner/runs.js", () => ({
       : undefined,
   queueEmbeddedPiMessageWithOutcome: (sessionId: string, text: string, options?: unknown) =>
     piEmbeddedMocks.queueEmbeddedPiMessageWithOutcome(sessionId, text, options),
+  resolveActiveEmbeddedRunSessionId: (...args: unknown[]) =>
+    piEmbeddedMocks.resolveActiveEmbeddedRunSessionId(...args),
 }));
 
 const providerUsageMocks = vi.hoisted(() => ({