🤖 [feat] Implement model creation for multiple providers

- Add getOpenAIModel function for OpenAI models
- Implement getBedrockModel function for AWS Bedrock
- Update imports to include new SDK components
This commit is contained in:
Maki 2024-10-17 00:21:03 +09:00
parent aa4c5dc676
commit f544c0c654

View File

@ -1,4 +1,6 @@
import { createAnthropic } from '@ai-sdk/anthropic';
import { createOpenAI } from '@ai-sdk/openai';
import { createAmazonBedrock } from '@ai-sdk/amazon-bedrock';
export function getAnthropicModel(apiKey: string) {
const anthropic = createAnthropic({
@ -7,3 +9,27 @@ export function getAnthropicModel(apiKey: string) {
return anthropic('claude-3-5-sonnet-20240620');
}
export function getOpenAIModel(model: string, apiKey: string) {
const openai = createOpenAI({
apiKey,
});
return openai(model);
}
interface AWSCredentials {
accessKeyId: string;
secretAccessKey: string;
region: string;
}
export function getBedrockModel(modelId: string, credentials: AWSCredentials) {
const bedrock = createAmazonBedrock({
region: credentials.region,
accessKeyId: credentials.accessKeyId,
secretAccessKey: credentials.secretAccessKey,
});
return bedrock(modelId);
}