From b4e3bbc57b645e0b5658afd75beed13de8a6f724 Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Mon, 20 Apr 2026 15:50:05 +0100 Subject: [PATCH] refactor: reuse anthropic service tier wrapper --- extensions/anthropic/stream-wrappers.ts | 22 +--------------------- 1 file changed, 1 insertion(+), 21 deletions(-) diff --git a/extensions/anthropic/stream-wrappers.ts b/extensions/anthropic/stream-wrappers.ts index 034a27b4835..1283484b1c6 100644 --- a/extensions/anthropic/stream-wrappers.ts +++ b/extensions/anthropic/stream-wrappers.ts @@ -152,27 +152,7 @@ export function createAnthropicFastModeWrapper( baseStreamFn: StreamFn | undefined, enabled: boolean, ): StreamFn { - const underlying = baseStreamFn ?? streamSimple; - const serviceTier = resolveAnthropicFastServiceTier(enabled); - return (model, context, options) => { - if (isAnthropicOAuthApiKey(options?.apiKey)) { - return underlying(model, context, options); - } - - const payloadPolicy = resolveAnthropicPayloadPolicy({ - provider: readStringValue(model.provider), - api: readStringValue(model.api), - baseUrl: readStringValue(model.baseUrl), - serviceTier, - }); - if (!payloadPolicy.allowsServiceTier) { - return underlying(model, context, options); - } - - return streamWithPayloadPatch(underlying, model, context, options, (payloadObj) => - applyAnthropicPayloadPolicyToParams(payloadObj, payloadPolicy), - ); - }; + return createAnthropicServiceTierWrapper(baseStreamFn, resolveAnthropicFastServiceTier(enabled)); } export function createAnthropicServiceTierWrapper(