From a0886875e13a8fad4b34f60a48081dce2f0888fa Mon Sep 17 00:00:00 2001
From: glay <glay.guo@gmail.com>
Date: Mon, 9 Dec 2024 00:03:10 +0800
Subject: [PATCH] Enhance encryption security with additional safeguards.

---
 app/client/platforms/bedrock.ts | 24 +++++++-----------------
 1 file changed, 7 insertions(+), 17 deletions(-)

diff --git a/app/client/platforms/bedrock.ts b/app/client/platforms/bedrock.ts
index 476a0c64d..63de78347 100644
--- a/app/client/platforms/bedrock.ts
+++ b/app/client/platforms/bedrock.ts
@@ -51,11 +51,6 @@ 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",
@@ -71,24 +66,19 @@ async function getBedrockHeaders(
     : getHeaders();
 
   if (!isApp) {
-    const { awsRegion, awsAccessKey, awsSecretKey, encryptionKey } = accessStore;
-    if (!encryptionKey) {
-      throw new Error("Missing encryption key");
-    }
+    const { awsRegion, awsAccessKey, awsSecretKey, encryptionKey } =
+      accessStore;
 
     const bedrockHeadersConfig = {
       XModelID: modelId,
       XEncryptionKey: encryptionKey,
       ShouldStream: String(shouldStream),
-      Authorization: await createAuthHeader({
-        region: awsRegion,
-        accessKey: awsAccessKey,
-        secretKey: awsSecretKey,
+      Authorization: await createAuthHeader(
+        awsRegion,
+        awsAccessKey,
+        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);