File tree Expand file tree Collapse file tree 2 files changed +1
-21
lines changed Expand file tree Collapse file tree 2 files changed +1
-21
lines changed Original file line number Diff line number Diff line change 1- import { extractUsage } from '@pydantic/genai-prices'
21import type { ModelAPI } from '../api'
32import { AnthropicAPI } from '../api/anthropic'
4- import { DefaultProviderProxy , type JsonData } from './default'
3+ import { DefaultProviderProxy } from './default'
54
65export class AnthropicProvider extends DefaultProviderProxy {
76 protected isWhitelistedEndpoint ( ) : boolean {
@@ -18,23 +17,6 @@ export class AnthropicProvider extends DefaultProviderProxy {
1817 headers . set ( 'x-api-key' , this . providerProxy . credentials )
1918 }
2019
21- // TODO(Marcelo): This should be moved to the `ModelAPI` class, and we need to improve the typing here!
22- protected handleData ( data : JsonData ) {
23- if ( 'type' in data && data . type === 'message_start' ) {
24- const message = 'message' in data ? ( data . message as JsonData ) : undefined
25- if ( message ) {
26- const model = 'model' in message ? ( message . model as string ) : undefined
27- if ( model ) {
28- this . responseModel = model
29- }
30- }
31- }
32- if ( 'usage' in data ) {
33- const { usage } = extractUsage ( this . usageProvider ( ) ! , data , this . apiFlavor ( ) )
34- this . usage = usage
35- }
36- }
37-
3820 protected responseHeaders ( headers : Headers ) : Headers {
3921 const newHeaders = super . responseHeaders ( headers )
4022 newHeaders . delete ( 'anthropic-organization-id' )
Original file line number Diff line number Diff line change @@ -468,8 +468,6 @@ export class DefaultProviderProxy {
468468 }
469469 }
470470
471- protected handleData ( _data : JsonData ) : void { }
472-
473471 protected isWhitelistedEndpoint ( ) : boolean {
474472 return false
475473 }
You can’t perform that action at this time.
0 commit comments