From fb59b5c461a3960df868ba0b450bd6a16ff93ca4 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Sat, 4 Apr 2026 14:27:42 +0100
Subject: [PATCH] fix(ci): sync openrouter stream hook seams

---
 extensions/openrouter/index.test.ts | 11 +++++------
 extensions/openrouter/index.ts      |  2 +-
 src/plugin-sdk/provider-stream.ts   |  8 ++++++--
 3 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/extensions/openrouter/index.test.ts b/extensions/openrouter/index.test.ts
index 7d64c3d06b7..8f13f27d8a6 100644
--- a/extensions/openrouter/index.test.ts
+++ b/extensions/openrouter/index.test.ts
@@ -17,11 +17,10 @@ describe("openrouter provider hooks", () => {
 
   it("injects provider routing into compat before applying stream wrappers", async () => {
     const provider = await registerSingleProviderPlugin(openrouterPlugin);
-    let capturedModel: Record<string, unknown> | undefined;
-    const baseStreamFn = vi.fn((model) => {
-      capturedModel = model as Record<string, unknown>;
-      return { async *[Symbol.asyncIterator]() {} } as never;
-    });
+    const baseStreamFn = vi.fn(
+      (..._args: Parameters<import("@mariozechner/pi-agent-core").StreamFn>) =>
+        ({ async *[Symbol.asyncIterator]() {} }) as never,
+    );
 
     const wrapped = provider.wrapStreamFn?.({
       provider: "openrouter",
@@ -47,7 +46,7 @@ describe("openrouter provider hooks", () => {
     );
 
     expect(baseStreamFn).toHaveBeenCalledOnce();
-    expect(capturedModel).toMatchObject({
+    expect(baseStreamFn.mock.calls[0]?.[0]).toMatchObject({
       compat: {
         openRouterRouting: {
           order: ["moonshot"],
diff --git a/extensions/openrouter/index.ts b/extensions/openrouter/index.ts
index eaa5029703d..22f5d4dad1b 100644
--- a/extensions/openrouter/index.ts
+++ b/extensions/openrouter/index.ts
@@ -83,7 +83,7 @@ export default definePluginEntry({
 
     function wrapOpenRouterProviderStream(
       ctx: ProviderWrapStreamFnContext,
-    ): StreamFn | undefined {
+    ): StreamFn | null | undefined {
       const providerRouting =
         ctx.extraParams?.provider != null && typeof ctx.extraParams.provider === "object"
           ? (ctx.extraParams.provider as Record<string, unknown>)
diff --git a/src/plugin-sdk/provider-stream.ts b/src/plugin-sdk/provider-stream.ts
index da43086eaef..c05f01fce44 100644
--- a/src/plugin-sdk/provider-stream.ts
+++ b/src/plugin-sdk/provider-stream.ts
@@ -163,14 +163,18 @@ export {
 export {
   createGoogleThinkingPayloadWrapper,
   sanitizeGoogleThinkingPayload,
-  createMinimaxFastModeWrapper,
+} from "../agents/pi-embedded-runner/google-stream-wrappers.js";
+export {
   createKilocodeWrapper,
   createOpenRouterSystemCacheWrapper,
   createOpenRouterWrapper,
   isProxyReasoningUnsupported,
+} from "../agents/pi-embedded-runner/proxy-stream-wrappers.js";
+export { createMinimaxFastModeWrapper } from "../agents/pi-embedded-runner/minimax-stream-wrappers.js";
+export {
   createMoonshotThinkingWrapper,
   resolveMoonshotThinkingType,
-};
+} from "../agents/pi-embedded-runner/moonshot-thinking-stream-wrappers.js";
 export {
   createOpenAIAttributionHeadersWrapper,
   createCodexNativeWebSearchWrapper,