From 3aaa357e7897773cbbe4900dcf0982eca44a96d5 Mon Sep 17 00:00:00 2001
From: Christian Tzolov <ctzolov@vmware.com>
Date: Wed, 18 Oct 2023 16:35:45 +0200
Subject: [PATCH] Bump open-ai-client.version from 0.12.0 to 0.16.0

---
 pom.xml                                           |  2 +-
 .../ai/openai/client/OpenAiClient.java            | 15 ++++++++-------
 2 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/pom.xml b/pom.xml
index 83e981991..3c948346b 100644
--- a/pom.xml
+++ b/pom.xml
@@ -69,7 +69,7 @@
 		<!-- production dependencies -->
 		<spring-boot.version>3.1.2</spring-boot.version>
 		<stringtemplate.version>4.0.2</stringtemplate.version>
-		<open-ai-client.version>0.12.0</open-ai-client.version>
+		<open-ai-client.version>0.16.0</open-ai-client.version>
 		<azure-open-ai-client.version>1.0.0-beta.3</azure-open-ai-client.version>
 		<jtokkit.version>0.6.1</jtokkit.version>
 		<victools.version>4.31.1</victools.version>
diff --git a/spring-ai-openai/src/main/java/org/springframework/ai/openai/client/OpenAiClient.java b/spring-ai-openai/src/main/java/org/springframework/ai/openai/client/OpenAiClient.java
index 1db724628..d76923d5f 100644
--- a/spring-ai-openai/src/main/java/org/springframework/ai/openai/client/OpenAiClient.java
+++ b/spring-ai-openai/src/main/java/org/springframework/ai/openai/client/OpenAiClient.java
@@ -32,9 +32,10 @@ import org.springframework.util.Assert;
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
 
 /**
- * Implementation of {@link AiClient} backed by an OpenAiService
+ * Implementation of {@link AiClient} backed by an {@link OpenAiService}.
  */
 public class OpenAiClient implements AiClient {
 
@@ -78,11 +79,11 @@ public class OpenAiClient implements AiClient {
 	@Override
 	public AiResponse generate(Prompt prompt) {
 		List<Message> messages = prompt.getMessages();
-		List<ChatMessage> theoMessages = new ArrayList<>();
-		for (Message message : messages) {
-			String messageType = message.getMessageTypeValue();
-			theoMessages.add(new ChatMessage(messageType, message.getContent()));
-		}
+
+		List<ChatMessage> theoMessages = messages.stream()
+			.map(message -> new ChatMessage(message.getMessageTypeValue(), message.getContent()))
+			.toList();
+
 		ChatCompletionRequest chatCompletionRequest = ChatCompletionRequest.builder()
 			.model(this.model)
 			.temperature(this.temperature)
@@ -114,7 +115,7 @@ public class OpenAiClient implements AiClient {
 			ChatMessage chatMessage = chatCompletionChoice.getMessage();
 			// TODO investigate mapping of additional metadata/runtime info to the
 			// general model.
-			Generation generation = new Generation(chatMessage.getContent());
+			Generation generation = new Generation(chatMessage.getContent(), Map.of("role", chatMessage.getRole()));
 			generations.add(generation);
 		}
 		return new AiResponse(generations);