feat: pass through OpenRouter provider routing params

Mitsuyuki Osabe · Mitsuyuki Osabe · commit 57187f8e9b4b · 2026-02-15T13:27:36.000Z
extraParams.provider was silently dropped by createStreamFnWithExtraParams(). This change injects it into model.compat.openRouterRouting so pi-ai's buildParams includes params.provider in the API request body. Enables OpenRouter provider routing options (only, order, allow_fallbacks, data_collection, ignore, sort, quantizations) via model config: ```jsonc "openrouter/model-name": { "params": { "provider": { "only": ["deepinfra", "fireworks"], "allow_fallbacks": false } } } ``` Closes #10869 ✍️ Author: Claude Code with @carrotRakko (AI-written, human-approved)
diff --git a/src/agents/pi-embedded-runner/extra-params.ts b/src/agents/pi-embedded-runner/extra-params.ts
@@ -89,18 +89,43 @@ function createStreamFnWithExtraParams(
     streamParams.cacheRetention = cacheRetention;
   }
 
-  if (Object.keys(streamParams).length === 0) {
+  // Extract OpenRouter provider routing preferences from extraParams.provider.
+  // Injected into model.compat.openRouterRouting so pi-ai's buildParams sets
+  // params.provider in the API request body (openai-completions.js L359-362).
+  // pi-ai's OpenRouterRouting type only declares { only?, order? }, but at
+  // runtime the full object is forwarded — enabling allow_fallbacks,
+  // data_collection, ignore, sort, quantizations, etc.
+  const providerRouting =
+    provider === "openrouter" &&
+    extraParams.provider != null &&
+    typeof extraParams.provider === "object"
+      ? (extraParams.provider as Record<string, unknown>)
+      : undefined;
+
+  if (Object.keys(streamParams).length === 0 && !providerRouting) {
     return undefined;
   }
 
   log.debug(`creating streamFn wrapper with params: ${JSON.stringify(streamParams)}`);
+  if (providerRouting) {
+    log.debug(`OpenRouter provider routing: ${JSON.stringify(providerRouting)}`);
+  }
 
   const underlying = baseStreamFn ?? streamSimple;
-  const wrappedStreamFn: StreamFn = (model, context, options) =>
-    underlying(model, context, {
+  const wrappedStreamFn: StreamFn = (model, context, options) => {
+    // When provider routing is configured, inject it into model.compat so
+    // pi-ai picks it up via model.compat.openRouterRouting.
+    const effectiveModel = providerRouting
+      ? ({
+          ...model,
+          compat: { ...model.compat, openRouterRouting: providerRouting },
+        } as unknown as typeof model)
+      : model;
+    return underlying(effectiveModel, context, {
       ...streamParams,
       ...options,
     });
+  };
 
   return wrappedStreamFn;
 }
