GithubProxy/ChatGPT-Next-Web
1
0
Fork 1
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-05-23 06:00:17 +09:00
Code Issues Packages Projects Releases Wiki Activity

Update app/client/platforms/bedrock.ts

Browse Source 
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
This commit is contained in:
glay 2024-12-08 23:58:05 +08:00 committed by GitHub
parent 4b2f447474
commit 93337b2b92
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
app/client/platforms/bedrock.ts
View File

@ -51,6 +51,11 @@ async function getBedrockHeaders(
shouldStream: boolean,
): Promise<Record<string, string>> {
const accessStore = useAccessStore.getState();
// Validate credentials
if (!accessStore.awsAccessKey || !accessStore.awsSecretKey || !accessStore.awsRegion) {
throw new Error("Missing required AWS credentials");
}
const bedrockHeaders = isApp
? await sign({
method: "POST",
@ -66,19 +71,24 @@ async function getBedrockHeaders(
: getHeaders();
if (!isApp) {
const { awsRegion, awsAccessKey, awsSecretKey, encryptionKey } =
accessStore;
const { awsRegion, awsAccessKey, awsSecretKey, encryptionKey } = accessStore;
if (!encryptionKey) {
throw new Error("Missing encryption key");
}
const bedrockHeadersConfig = {
XModelID: modelId,
XEncryptionKey: encryptionKey,
ShouldStream: String(shouldStream),
Authorization: await createAuthHeader(
awsRegion,
awsAccessKey,
awsSecretKey,
Authorization: await createAuthHeader({
region: awsRegion,
accessKey: awsAccessKey,
secretKey: awsSecretKey,
encryptionKey,
),
}).catch(error => {
console.error("[Bedrock] Failed to create auth header:", error);
throw new Error("Failed to create authorization header");
}),
};
Object.assign(bedrockHeaders, bedrockHeadersConfig);