bolt.diy/app/lib/modules/llm/providers/anthropic.ts
2025-03-16 14:04:23 +00:00

134 lines
3.7 KiB
TypeScript

import { BaseProvider } from '~/lib/modules/llm/base-provider';
import type { ModelInfo } from '~/lib/modules/llm/types';
import type { LanguageModelV1 } from 'ai';
import type { IProviderSetting } from '~/types/model';
import { createAnthropic } from '@ai-sdk/anthropic';
export default class AnthropicProvider extends BaseProvider {
name = 'Anthropic';
getApiKeyLink = 'https://console.anthropic.com/settings/keys';
config = {
apiTokenKey: 'ANTHROPIC_API_KEY',
};
staticModels: ModelInfo[] = [
{
name: 'claude-3-7-sonnet-20250219',
label: 'Claude 3.7 Sonnet',
provider: 'Anthropic',
maxTokenAllowed: 8000,
createdAt: '2025-02-24T00:00:00Z',
},
{
name: 'claude-3-5-sonnet-latest',
label: 'Claude 3.5 Sonnet (Latest)',
provider: 'Anthropic',
maxTokenAllowed: 8000,
createdAt: '2024-10-22T00:00:00Z',
},
{
name: 'claude-3-5-sonnet-20240620',
label: 'Claude 3.5 Sonnet (Old)',
provider: 'Anthropic',
maxTokenAllowed: 8000,
createdAt: '2024-06-20T00:00:00Z',
},
{
name: 'claude-3-5-haiku-latest',
label: 'Claude 3.5 Haiku (Latest)',
provider: 'Anthropic',
maxTokenAllowed: 8000,
createdAt: '2024-10-22T00:00:00Z',
},
{
name: 'claude-3-opus-latest',
label: 'Claude 3 Opus',
provider: 'Anthropic',
maxTokenAllowed: 8000,
createdAt: '2024-02-29T00:00:00Z',
},
{
name: 'claude-3-sonnet-20240229',
label: 'Claude 3 Sonnet',
provider: 'Anthropic',
maxTokenAllowed: 8000,
createdAt: '2024-02-29T00:00:00Z',
},
{
name: 'claude-3-haiku-20240307',
label: 'Claude 3 Haiku',
provider: 'Anthropic',
maxTokenAllowed: 8000,
createdAt: '2024-03-07T00:00:00Z',
},
];
async getDynamicModels(
apiKeys?: Record<string, string>,
settings?: IProviderSetting,
serverEnv?: Record<string, string>,
): Promise<ModelInfo[]> {
const { apiKey } = this.getProviderBaseUrlAndKey({
apiKeys,
providerSettings: settings,
serverEnv: serverEnv as any,
defaultBaseUrlKey: '',
defaultApiTokenKey: 'ANTHROPIC_API_KEY',
});
if (!apiKey) {
throw `Missing Api Key configuration for ${this.name} provider`;
}
const response = await fetch(`https://api.anthropic.com/v1/models`, {
headers: {
'x-api-key': `${apiKey}`,
'anthropic-version': '2023-06-01',
},
});
const res = (await response.json()) as any;
const staticModelIds = this.staticModels.map((m) => m.name);
const data = res.data.filter((model: any) => model.type === 'model' && !staticModelIds.includes(model.id));
return data.map((m: any) => ({
name: m.id,
label: `${m.display_name}`,
provider: this.name,
maxTokenAllowed: 32000,
createdAt: m.created_at ?? '0000-00-00T00:00:00Z',
}));
}
getModelInstance: (options: {
model: string;
serverEnv: Env;
apiKeys?: Record<string, string>;
providerSettings?: Record<string, IProviderSetting>;
}) => LanguageModelV1 = (options) => {
const { apiKeys, providerSettings, serverEnv, model } = options;
const { apiKey } = this.getProviderBaseUrlAndKey({
apiKeys,
providerSettings,
serverEnv: serverEnv as any,
defaultBaseUrlKey: '',
defaultApiTokenKey: 'ANTHROPIC_API_KEY',
});
const anthropic = createAnthropic({
apiKey,
});
return anthropic(model);
};
sort(models: ModelInfo[]): ModelInfo[] {
return models.sort((a, b) => {
const compare = b.createdAt!.localeCompare(a.createdAt!);
return compare === 0 ? b.name.localeCompare(a.name) : compare;
});
}
}