diff --git a/gateway/src/providers/anthropic.ts b/gateway/src/providers/anthropic.ts index 6c08c08..8492578 100644 --- a/gateway/src/providers/anthropic.ts +++ b/gateway/src/providers/anthropic.ts @@ -1,7 +1,6 @@ -import { extractUsage } from '@pydantic/genai-prices' import type { ModelAPI } from '../api' import { AnthropicAPI } from '../api/anthropic' -import { DefaultProviderProxy, type JsonData } from './default' +import { DefaultProviderProxy } from './default' export class AnthropicProvider extends DefaultProviderProxy { protected isWhitelistedEndpoint(): boolean { @@ -18,23 +17,6 @@ export class AnthropicProvider extends DefaultProviderProxy { headers.set('x-api-key', this.providerProxy.credentials) } - // TODO(Marcelo): This should be moved to the `ModelAPI` class, and we need to improve the typing here! - protected handleData(data: JsonData) { - if ('type' in data && data.type === 'message_start') { - const message = 'message' in data ? (data.message as JsonData) : undefined - if (message) { - const model = 'model' in message ? (message.model as string) : undefined - if (model) { - this.responseModel = model - } - } - } - if ('usage' in data) { - const { usage } = extractUsage(this.usageProvider()!, data, this.apiFlavor()) - this.usage = usage - } - } - protected responseHeaders(headers: Headers): Headers { const newHeaders = super.responseHeaders(headers) newHeaders.delete('anthropic-organization-id') diff --git a/gateway/src/providers/default.ts b/gateway/src/providers/default.ts index 0b08067..2032648 100644 --- a/gateway/src/providers/default.ts +++ b/gateway/src/providers/default.ts @@ -468,8 +468,6 @@ export class DefaultProviderProxy { } } - protected handleData(_data: JsonData): void {} - protected isWhitelistedEndpoint(): boolean { return false }