diff --git a/extensions/fireworks/stream.test.ts b/extensions/fireworks/stream.test.ts
index 11e543847dca..33b8ba0e0277 100644
--- a/extensions/fireworks/stream.test.ts
+++ b/extensions/fireworks/stream.test.ts
@@ -37,7 +37,7 @@ function capturePayload(params: {
   return captured;
 }
 
-describe("createFireworksKimiThinkingDisabledWrapper", () => {
+describe("wrapFireworksProviderStream", () => {
   it("forces thinking disabled for Fireworks Kimi models", () => {
     expect(
       capturePayload({
diff --git a/extensions/fireworks/stream.ts b/extensions/fireworks/stream.ts
index abe7453ff0eb..c6430e68f36a 100644
--- a/extensions/fireworks/stream.ts
+++ b/extensions/fireworks/stream.ts
@@ -10,17 +10,6 @@ function isFireworksProviderId(providerId: string): boolean {
   return normalized === "fireworks" || normalized === "fireworks-ai";
 }
 
-function createFireworksKimiThinkingDisabledWrapper(baseStreamFn: StreamFn | undefined): StreamFn {
-  return createPayloadPatchStreamWrapper(baseStreamFn, ({ payload }) => {
-    // Fireworks Kimi can emit chain-of-thought in visible `content` unless
-    // the Anthropic-style thinking toggle is explicitly disabled.
-    payload.thinking = { type: "disabled" };
-    delete payload.reasoning;
-    delete payload.reasoning_effort;
-    delete payload.reasoningEffort;
-  });
-}
-
 export function wrapFireworksProviderStream(
   ctx: ProviderWrapStreamFnContext,
 ): StreamFn | undefined {
@@ -31,5 +20,12 @@ export function wrapFireworksProviderStream(
   ) {
     return undefined;
   }
-  return createFireworksKimiThinkingDisabledWrapper(ctx.streamFn);
+  return createPayloadPatchStreamWrapper(ctx.streamFn, ({ payload }) => {
+    // Fireworks Kimi can emit chain-of-thought in visible `content` unless
+    // the Anthropic-style thinking toggle is explicitly disabled.
+    payload.thinking = { type: "disabled" };
+    delete payload.reasoning;
+    delete payload.reasoning_effort;
+    delete payload.reasoningEffort;
+  });
 }