From c9620504377b47df1e7a0bd49b347f0045590684 Mon Sep 17 00:00:00 2001 From: Daniel Riccio Date: Mon, 21 Jul 2025 08:32:17 -0500 Subject: [PATCH] fix: add bedrock to ANTHROPIC_STYLE_PROVIDERS for correct token counting - Added 'bedrock' to ANTHROPIC_STYLE_PROVIDERS array to ensure Bedrock provider uses Anthropic-style API protocol - This fixes incorrect token counting for AWS Bedrock when using Claude models - Resolves issue #6004 where token count was showing unrealistic values --- packages/types/src/provider-settings.ts | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/packages/types/src/provider-settings.ts b/packages/types/src/provider-settings.ts index 3b53627295..70635d7b8a 100644 --- a/packages/types/src/provider-settings.ts +++ b/packages/types/src/provider-settings.ts @@ -301,9 +301,14 @@ export const getModelId = (settings: ProviderSettings): string | undefined => { } // Providers that use Anthropic-style API protocol -export const ANTHROPIC_STYLE_PROVIDERS: ProviderName[] = ["anthropic", "claude-code"] +export const ANTHROPIC_STYLE_PROVIDERS: ProviderName[] = ["anthropic", "claude-code", "bedrock"] // Helper function to determine API protocol for a provider export const getApiProtocol = (provider: ProviderName | undefined): "anthropic" | "openai" => { - return provider && ANTHROPIC_STYLE_PROVIDERS.includes(provider) ? "anthropic" : "openai" + // Check if provider uses Anthropic-style API by default + if (provider && ANTHROPIC_STYLE_PROVIDERS.includes(provider)) { + return "anthropic" + } + // Default to OpenAI style + return "openai" }