Skip to content

Commit 6bc6f18

Browse files
authored
chore: remove unused handleData (#108)
1 parent 8c2093c commit 6bc6f18

File tree

2 files changed

+1
-21
lines changed

2 files changed

+1
-21
lines changed

gateway/src/providers/anthropic.ts

Lines changed: 1 addition & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
1-
import { extractUsage } from '@pydantic/genai-prices'
21
import type { ModelAPI } from '../api'
32
import { AnthropicAPI } from '../api/anthropic'
4-
import { DefaultProviderProxy, type JsonData } from './default'
3+
import { DefaultProviderProxy } from './default'
54

65
export 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')

gateway/src/providers/default.ts

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff 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
}

0 commit comments

Comments
 (0)