Skip to content

Commit ca2a531

Browse files
Stuart LoxtonStuart Loxton
authored andcommitted
[Enhancement] - Change argument type in getRequestMetada function
1 parent bf1ecfd commit ca2a531

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

models/spring-ai-bedrock-converse/src/main/java/org/springframework/ai/bedrock/converse/BedrockProxyChatModel.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -425,7 +425,7 @@ else if (message.getMessageType() == MessageType.TOOL) {
425425
Document additionalModelRequestFields = ConverseApiUtils
426426
.getChatOptionsAdditionalModelRequestFields(this.defaultOptions, prompt.getOptions());
427427

428-
Map<String, String> requestMetadata = ConverseApiUtils.getRequestMetadata(prompt);
428+
Map<String, String> requestMetadata = ConverseApiUtils.getRequestMetadata(prompt.getUserMessage().getMetadata());
429429

430430
return ConverseRequest.builder()
431431
.modelId(updatedRuntimeOptions.getModel())

models/spring-ai-bedrock-converse/src/main/java/org/springframework/ai/bedrock/converse/api/ConverseApiUtils.java

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
import java.util.concurrent.atomic.AtomicBoolean;
2626
import java.util.stream.Collectors;
2727

28-
import org.springframework.ai.chat.prompt.Prompt;
2928
import reactor.core.publisher.Flux;
3029
import reactor.core.publisher.Mono;
3130
import software.amazon.awssdk.core.SdkField;
@@ -386,8 +385,7 @@ else if (value instanceof Map mapValue) {
386385
}
387386

388387
@SuppressWarnings("unchecked")
389-
public static Map<String, String> getRequestMetadata(Prompt prompt) {
390-
Map<String, Object> metadata = prompt.getUserMessage().getMetadata();
388+
public static Map<String, String> getRequestMetadata(Map<String, Object> metadata) {
391389

392390
if (metadata.isEmpty()) {
393391
return Map.of();

0 commit comments

Comments
 (0)