From 21eb1c99214380f20af1fc5fda6de4b4be9aa35b 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 a1bbd10c0c4..11c1bc3dba7 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 @@ * * @author Christian Tzolov * @author Wei Jiang + * @author Jihoon Kim * @since 1.0.0 */ public class BedrockProxyChatModel extends AbstractToolCallSupport implements ChatModel { @@ -606,6 +607,7 @@ private Flux internalStream(Prompt prompt, ChatResponse perviousCh ConverseStreamRequest converseStreamRequest = ConverseStreamRequest.builder() .modelId(converseRequest.modelId()) + .inferenceConfig(converseRequest.inferenceConfig()) .messages(converseRequest.messages()) .system(converseRequest.system()) .additionalModelRequestFields(converseRequest.additionalModelRequestFields())