Skip to content

Commit 53ac65f

Browse files
committed
Stop using deprecated assistant message constructor
Signed-off-by: Nicolas Krier <[email protected]>
1 parent f337ee0 commit 53ac65f

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

models/spring-ai-mistral-ai/src/test/java/org/springframework/ai/mistralai/MistralAiChatCompletionRequestTests.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,12 @@ void createChatCompletionMessagesWithAssistantMessage() {
165165
var toolCall1 = createToolCall(1);
166166
var toolCall2 = createToolCall(2);
167167
var toolCall3 = createToolCall(3);
168-
var assistantMessage = new AssistantMessage(TEXT_CONTENT, Map.of(), List.of(toolCall1, toolCall2, toolCall3));
168+
// @formatter:off
169+
var assistantMessage = AssistantMessage.builder()
170+
.content(TEXT_CONTENT)
171+
.toolCalls(List.of(toolCall1, toolCall2, toolCall3))
172+
.build();
173+
// @formatter:on
169174
var prompt = createPrompt(assistantMessage);
170175
var chatCompletionRequest = this.chatModel.createRequest(prompt, false);
171176
var chatCompletionMessages = chatCompletionRequest.messages();

0 commit comments

Comments
 (0)