6a8a68e1a3
* fix: claude-4 not having the right max_tokens * feat: add bedrock support * chore: fix package-lock.json * fix: rename baseUrl to baseURL * feat: add azure support * fix: final touches of azure integration * feat: add google vertex provider * chore: fix tests and refactor task-manager.test.js * chore: move task 92 to 94
42 lines
1.1 KiB
JavaScript
42 lines
1.1 KiB
JavaScript
import { createAmazonBedrock } from '@ai-sdk/amazon-bedrock';
|
|
import { fromNodeProviderChain } from '@aws-sdk/credential-providers';
|
|
import { BaseAIProvider } from './base-provider.js';
|
|
|
|
export class BedrockAIProvider extends BaseAIProvider {
|
|
constructor() {
|
|
super();
|
|
this.name = 'Bedrock';
|
|
}
|
|
|
|
/**
|
|
* Override auth validation - Bedrock uses AWS credentials instead of API keys
|
|
* @param {object} params - Parameters to validate
|
|
*/
|
|
validateAuth(params) {}
|
|
|
|
/**
|
|
* Creates and returns a Bedrock client instance.
|
|
* See https://docs.aws.amazon.com/cli/latest/userguide/cli-configure-envvars.html
|
|
* for AWS SDK environment variables and configuration options.
|
|
*/
|
|
getClient(params) {
|
|
try {
|
|
const {
|
|
profile = process.env.AWS_PROFILE || 'default',
|
|
region = process.env.AWS_DEFAULT_REGION || 'us-east-1',
|
|
baseURL
|
|
} = params;
|
|
|
|
const credentialProvider = fromNodeProviderChain({ profile });
|
|
|
|
return createAmazonBedrock({
|
|
region,
|
|
credentialProvider,
|
|
...(baseURL && { baseURL })
|
|
});
|
|
} catch (error) {
|
|
this.handleError('client initialization', error);
|
|
}
|
|
}
|
|
}
|