Skip to content

Commit 06079b2

Browse files
committed
Fix BedrockChatOptions usage in BedrockProxyChatModel
Signed-off-by: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>
1 parent a110c8b commit 06079b2

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

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

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -33,11 +33,6 @@
3333
import io.micrometer.observation.contextpropagation.ObservationThreadLocalAccessor;
3434
import org.slf4j.Logger;
3535
import org.slf4j.LoggerFactory;
36-
import org.springframework.ai.model.tool.DefaultToolExecutionEligibilityPredicate;
37-
import org.springframework.ai.model.tool.ToolCallingChatOptions;
38-
import org.springframework.ai.model.tool.ToolCallingManager;
39-
import org.springframework.ai.model.tool.ToolExecutionEligibilityPredicate;
40-
import org.springframework.ai.model.tool.ToolExecutionResult;
4136
import reactor.core.publisher.Flux;
4237
import reactor.core.publisher.Sinks;
4338
import reactor.core.publisher.Sinks.EmitFailureHandler;
@@ -101,7 +96,6 @@
10196
import org.springframework.ai.chat.prompt.Prompt;
10297
import org.springframework.ai.content.Media;
10398
import org.springframework.ai.model.ModelOptionsUtils;
104-
import org.springframework.ai.model.tool.internal.ToolCallReactiveContextHolder;
10599
import org.springframework.ai.model.tool.DefaultToolExecutionEligibilityPredicate;
106100
import org.springframework.ai.model.tool.ToolCallingChatOptions;
107101
import org.springframework.ai.model.tool.ToolCallingManager;
@@ -786,7 +780,7 @@ public static final class Builder {
786780

787781
private ToolExecutionEligibilityPredicate toolExecutionEligibilityPredicate = new DefaultToolExecutionEligibilityPredicate();
788782

789-
private ToolCallingChatOptions defaultOptions = ToolCallingChatOptions.builder().build();
783+
private BedrockChatOptions defaultOptions = BedrockChatOptions.builder().build();
790784

791785
private ObservationRegistry observationRegistry = ObservationRegistry.NOOP;
792786

@@ -834,7 +828,7 @@ public Builder timeout(Duration timeout) {
834828
return this;
835829
}
836830

837-
public Builder defaultOptions(ToolCallingChatOptions defaultOptions) {
831+
public Builder defaultOptions(BedrockChatOptions defaultOptions) {
838832
Assert.notNull(defaultOptions, "'defaultOptions' must not be null.");
839833
this.defaultOptions = defaultOptions;
840834
return this;

0 commit comments

Comments
 (0)