mirror of
https://github.com/stackblitz-labs/bolt.diy
synced 2025-01-23 11:17:02 +00:00
7295352a98
* refactor: Refactoring Providers to have providers as modules * updated package and lock file * added grok model back * updated registry system
59 lines
1.8 KiB
TypeScript
59 lines
1.8 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-5-sonnet-latest',
|
|
label: 'Claude 3.5 Sonnet (new)',
|
|
provider: 'Anthropic',
|
|
maxTokenAllowed: 8000,
|
|
},
|
|
{
|
|
name: 'claude-3-5-sonnet-20240620',
|
|
label: 'Claude 3.5 Sonnet (old)',
|
|
provider: 'Anthropic',
|
|
maxTokenAllowed: 8000,
|
|
},
|
|
{
|
|
name: 'claude-3-5-haiku-latest',
|
|
label: 'Claude 3.5 Haiku (new)',
|
|
provider: 'Anthropic',
|
|
maxTokenAllowed: 8000,
|
|
},
|
|
{ name: 'claude-3-opus-latest', label: 'Claude 3 Opus', provider: 'Anthropic', maxTokenAllowed: 8000 },
|
|
{ name: 'claude-3-sonnet-20240229', label: 'Claude 3 Sonnet', provider: 'Anthropic', maxTokenAllowed: 8000 },
|
|
{ name: 'claude-3-haiku-20240307', label: 'Claude 3 Haiku', provider: 'Anthropic', maxTokenAllowed: 8000 },
|
|
];
|
|
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);
|
|
};
|
|
}
|