Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Bump open-ai-client.version from 0.12.0 to 0.16.0

Browse Source
This commit is contained in:
Christian Tzolov
2023-10-18 16:35:45 +02:00
parent e5693f7e60
commit 3aaa357e78
2 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pom.xml
View File

@@ -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>

15
spring-ai-openai/src/main/java/org/springframework/ai/openai/client/OpenAiClient.java
View File

@@ -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);