fix bedrock model empty bug

chuan-edutrigger · Dec 19, 2023 · 7952ef8 · 7952ef8
1 parent a1fb5b8
commit 7952ef8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/packages/components/nodes/chatmodels/AWSBedrock/AWSChatBedrock.ts b/packages/components/nodes/chatmodels/AWSBedrock/AWSChatBedrock.ts
@@ -143,7 +143,7 @@ class AWSChatBedrock_ChatModels implements INode {
 
         const obj: BaseBedrockInput & BaseChatModelParams = {
             region: iRegion,
-            model: customModel ?? iModel,
+            model: customModel ? customModel : iModel,
             maxTokens: parseInt(iMax_tokens_to_sample, 10),
             temperature: parseFloat(iTemperature),
             streaming: streaming ?? true

diff --git a/packages/components/nodes/embeddings/AWSBedrockEmbedding/AWSBedrockEmbedding.ts b/packages/components/nodes/embeddings/AWSBedrockEmbedding/AWSBedrockEmbedding.ts
@@ -103,7 +103,7 @@ class AWSBedrockEmbedding_Embeddings implements INode {
         const customModel = nodeData.inputs?.customModel as string
 
         const obj: BedrockEmbeddingsParams = {
-            model: customModel ?? iModel,
+            model: customModel ? customModel : iModel,
             region: iRegion
         }
 

diff --git a/packages/components/nodes/llms/AWSBedrock/AWSBedrock.ts b/packages/components/nodes/llms/AWSBedrock/AWSBedrock.ts
@@ -143,7 +143,7 @@ class AWSBedrock_LLMs implements INode {
         const iMax_tokens_to_sample = nodeData.inputs?.max_tokens_to_sample as string
         const cache = nodeData.inputs?.cache as BaseCache
         const obj: Partial<BaseBedrockInput> & BaseLLMParams = {
-            model: customModel ?? iModel,
+            model: customModel ? customModel : iModel,
             region: iRegion,
             temperature: parseFloat(iTemperature),
             maxTokens: parseInt(iMax_tokens_to_sample, 10)