UniqAI/bolt.diy
1
0
Fork 0
mirror of https://github.com/stackblitz-labs/bolt.diy synced 2025-06-26 18:26:38 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

#1561: Permit bedrockOptions in AWS_BEDROCK_CONFIG

Browse Source
This commit is contained in:
Joe Still 2025-03-27 12:56:28 -05:00
parent 1364d4a503
commit 023a6a39fb
1 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
app/lib/modules/llm/providers/amazon-bedrock.ts
View File

@ -2,14 +2,7 @@ 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 { createAmazonBedrock } from '@ai-sdk/amazon-bedrock';
interface AWSBedRockConfig {
region: string;
accessKeyId: string;
secretAccessKey: string;
sessionToken?: string;
}
import { createAmazonBedrock, type AmazonBedrockProviderSettings } from '@ai-sdk/amazon-bedrock';
export default class AmazonBedrockProvider extends BaseProvider {
name = 'AmazonBedrock';
@ -64,8 +57,8 @@ export default class AmazonBedrockProvider extends BaseProvider {
},
];
private _parseAndValidateConfig(apiKey: string): AWSBedRockConfig {
let parsedConfig: AWSBedRockConfig;
private _parseAndValidateConfig(apiKey: string): AmazonBedrockProviderSettings {
let parsedConfig: AmazonBedrockProviderSettings;
try {
parsedConfig = JSON.parse(apiKey);
@ -75,7 +68,7 @@ export default class AmazonBedrockProvider extends BaseProvider {
);
}
const { region, accessKeyId, secretAccessKey, sessionToken } = parsedConfig;
const { region, accessKeyId, secretAccessKey, sessionToken, bedrockOptions } = parsedConfig;
if (!region || !accessKeyId || !secretAccessKey) {
throw new Error(
@ -88,6 +81,7 @@ export default class AmazonBedrockProvider extends BaseProvider {
accessKeyId,
secretAccessKey,
...(sessionToken && { sessionToken }),
...(bedrockOptions && { bedrockOptions }),
};
}