fix: route Codex OpenAI runtime through Codex provider

PsiClawOps · PsiClawOps · commit c46f0e524989 · 2026-05-16T19:42:45.000-07:00
diff --git a/src/agents/openai-codex-routing.test.ts b/src/agents/openai-codex-routing.test.ts
@@ -150,4 +150,22 @@ describe("OpenAI Codex routing policy", () => {
       }),
     ).toEqual(["openai-codex"]);
   });
+
+  it("routes openai provider to openai-codex when harness runtime is codex", () => {
+    expect(
+      resolveSelectedOpenAIPiRuntimeProvider({
+        provider: "openai",
+        harnessRuntime: "codex",
+      }),
+    ).toBe("openai-codex");
+  });
+
+  it("does not route non-OpenAI providers when runtime is codex", () => {
+    expect(
+      resolveSelectedOpenAIPiRuntimeProvider({
+        provider: "anthropic",
+        harnessRuntime: "codex",
+      }),
+    ).toBe("anthropic");
+  });
 });
diff --git a/src/agents/openai-codex-routing.ts b/src/agents/openai-codex-routing.ts
@@ -181,8 +181,13 @@ export function resolveSelectedOpenAIPiRuntimeProvider(params: {
     return OPENAI_CODEX_PROVIDER_ID;
   }
   const runtime = normalizeEmbeddedAgentRuntime(params.agentHarnessId ?? params.harnessRuntime);
-  return isOpenAIProvider(params.provider) &&
-    runtime === "pi" &&
+  if (!isOpenAIProvider(params.provider)) {
+    return params.provider;
+  }
+  if (runtime === "codex") {
+    return OPENAI_CODEX_PROVIDER_ID;
+  }
+  return runtime === "pi" &&
     !params.authProfileId?.trim() &&
     configuredOpenAIAuthOrderStartsWithCodexProfile(params.config)
     ? OPENAI_CODEX_PROVIDER_ID
