From 6477c0ccfc028c0bf62858ed9b53dd0196bc8241 Mon Sep 17 00:00:00 2001 From: jitokim Date: Sat, 14 Dec 2024 20:16:50 +0900 Subject: [PATCH] Fix missing `inferenceConfig` in ConverseStreamRequest builder Signed-off-by: jitokim --- .../ai/bedrock/converse/BedrockProxyChatModel.java | 2 ++ 1 file changed, 2 insertions(+) diff --git a/models/spring-ai-bedrock-converse/src/main/java/org/springframework/ai/bedrock/converse/BedrockProxyChatModel.java b/models/spring-ai-bedrock-converse/src/main/java/org/springframework/ai/bedrock/converse/BedrockProxyChatModel.java index bec449ee9..8648ec917 100644 --- a/models/spring-ai-bedrock-converse/src/main/java/org/springframework/ai/bedrock/converse/BedrockProxyChatModel.java +++ b/models/spring-ai-bedrock-converse/src/main/java/org/springframework/ai/bedrock/converse/BedrockProxyChatModel.java @@ -129,6 +129,7 @@ import org.springframework.util.StringUtils; * @author Christian Tzolov * @author Wei Jiang * @author Alexandros Pappas + * @author Jihoon Kim * @since 1.0.0 */ public class BedrockProxyChatModel extends AbstractToolCallSupport implements ChatModel { @@ -606,6 +607,7 @@ public class BedrockProxyChatModel extends AbstractToolCallSupport implements Ch ConverseStreamRequest converseStreamRequest = ConverseStreamRequest.builder() .modelId(converseRequest.modelId()) + .inferenceConfig(converseRequest.inferenceConfig()) .messages(converseRequest.messages()) .system(converseRequest.system()) .additionalModelRequestFields(converseRequest.additionalModelRequestFields())