Make consistent sync/stream AssistantMessage properties for OpenAI and Mistral AI

Author: tzolov

models/spring-ai-mistral-ai/src/main/java/org/springframework/ai/mistralai/MistralAiChatClient.java

@@ -15,6 +15,7 @@
  */
 package org.springframework.ai.mistralai;
 
+import java.util.HashMap;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
@@ -113,15 +114,28 @@ public ChatResponse call(Prompt prompt) {
 
 			List<Generation> generations = chatCompletion.choices()
 				.stream()
-				.map(choice -> new Generation(choice.message().content(),
-						Map.of("role", choice.message().role().name()))
+				.map(choice -> new Generation(choice.message().content(), toMap(chatCompletion.id(), choice))
 					.withGenerationMetadata(ChatGenerationMetadata.from(choice.finishReason().name(), null)))
 				.toList();
 
 			return new ChatResponse(generations);
 		});
 	}
 
+	private Map<String, Object> toMap(String id, ChatCompletion.Choice choice) {
+		Map<String, Object> map = new HashMap<>();
+
+		var message = choice.message();
+		if (message.role() != null) {
+			map.put("role", message.role().name());
+		}
+		if (choice.finishReason() != null) {
+			map.put("finishReason", choice.finishReason().name());
+		}
+		map.put("id", id);
+		return map;
+	}
+
 	@Override
 	public Flux<ChatResponse> stream(Prompt prompt) {
 		var request = createRequest(prompt, true);

models/spring-ai-openai/src/main/java/org/springframework/ai/openai/OpenAiChatClient.java

@@ -147,7 +147,7 @@ public ChatResponse call(Prompt prompt) {
 			RateLimit rateLimits = OpenAiResponseHeaderExtractor.extractAiResponseHeaders(completionEntity);
 
 			List<Generation> generations = chatCompletion.choices().stream().map(choice -> {
-				return new Generation(choice.message().content(), toMap(choice.message()))
+				return new Generation(choice.message().content(), toMap(chatCompletion.id(), choice))
 					.withGenerationMetadata(ChatGenerationMetadata.from(choice.finishReason().name(), null));
 			}).toList();
 
@@ -156,6 +156,20 @@ public ChatResponse call(Prompt prompt) {
 		});
 	}
 
+	private Map<String, Object> toMap(String id, ChatCompletion.Choice choice) {
+		Map<String, Object> map = new HashMap<>();
+
+		var message = choice.message();
+		if (message.role() != null) {
+			map.put("role", message.role().name());
+		}
+		if (choice.finishReason() != null) {
+			map.put("finishReason", choice.finishReason().name());
+		}
+		map.put("id", id);
+		return map;
+	}
+
 	@Override
 	public Flux<ChatResponse> stream(Prompt prompt) {
 
@@ -280,15 +294,6 @@ private List<OpenAiApi.FunctionTool> getFunctionTools(Set<String> functionNames)
 		}).toList();
 	}
 
-	private Map<String, Object> toMap(ChatCompletionMessage message) {
-		Map<String, Object> map = new HashMap<>();
-
-		if (message.role() != null) {
-			map.put("role", message.role().name());
-		}
-		return map;
-	}
-
 	@Override
 	protected ChatCompletionRequest doCreateToolResponseRequest(ChatCompletionRequest previousRequest,
 			ChatCompletionMessage responseMessage, List<ChatCompletionMessage> conversationHistory) {