Skip to content

Commit a7b5652

Browse files
garethjevansmarkpollack
authored andcommitted
Upgrade azure-openai library to beta.12
- Correct merge utils after lib update
1 parent de53e64 commit a7b5652

File tree

3 files changed

+8
-30
lines changed

3 files changed

+8
-30
lines changed

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

Lines changed: 3 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -30,29 +30,7 @@
3030
import com.azure.ai.openai.OpenAIAsyncClient;
3131
import com.azure.ai.openai.OpenAIClient;
3232
import com.azure.ai.openai.OpenAIClientBuilder;
33-
import com.azure.ai.openai.models.ChatChoice;
34-
import com.azure.ai.openai.models.ChatCompletions;
35-
import com.azure.ai.openai.models.ChatCompletionsFunctionToolCall;
36-
import com.azure.ai.openai.models.ChatCompletionsFunctionToolDefinition;
37-
import com.azure.ai.openai.models.ChatCompletionsJsonResponseFormat;
38-
import com.azure.ai.openai.models.ChatCompletionsOptions;
39-
import com.azure.ai.openai.models.ChatCompletionsResponseFormat;
40-
import com.azure.ai.openai.models.ChatCompletionsTextResponseFormat;
41-
import com.azure.ai.openai.models.ChatCompletionsToolCall;
42-
import com.azure.ai.openai.models.ChatCompletionsToolDefinition;
43-
import com.azure.ai.openai.models.ChatMessageContentItem;
44-
import com.azure.ai.openai.models.ChatMessageImageContentItem;
45-
import com.azure.ai.openai.models.ChatMessageImageUrl;
46-
import com.azure.ai.openai.models.ChatMessageTextContentItem;
47-
import com.azure.ai.openai.models.ChatRequestAssistantMessage;
48-
import com.azure.ai.openai.models.ChatRequestMessage;
49-
import com.azure.ai.openai.models.ChatRequestSystemMessage;
50-
import com.azure.ai.openai.models.ChatRequestToolMessage;
51-
import com.azure.ai.openai.models.ChatRequestUserMessage;
52-
import com.azure.ai.openai.models.CompletionsFinishReason;
53-
import com.azure.ai.openai.models.ContentFilterResultsForPrompt;
54-
import com.azure.ai.openai.models.FunctionCall;
55-
import com.azure.ai.openai.models.FunctionDefinition;
33+
import com.azure.ai.openai.models.*;
5634
import com.azure.core.util.BinaryData;
5735
import io.micrometer.observation.Observation;
5836
import io.micrometer.observation.ObservationRegistry;
@@ -387,7 +365,8 @@ ChatCompletionsOptions toAzureChatCompletionsOptions(Prompt prompt) {
387365
private List<ChatCompletionsFunctionToolDefinition> getFunctionTools(Set<String> functionNames) {
388366
return this.resolveFunctionCallbacks(functionNames).stream().map(functionCallback -> {
389367

390-
FunctionDefinition functionDefinition = new FunctionDefinition(functionCallback.getName());
368+
ChatCompletionsFunctionToolDefinitionFunction functionDefinition = new ChatCompletionsFunctionToolDefinitionFunction(
369+
functionCallback.getName());
391370
functionDefinition.setDescription(functionCallback.getDescription());
392371
BinaryData parameters = BinaryData
393372
.fromObject(ModelOptionsUtils.jsonToMap(functionCallback.getInputTypeSchema()));

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

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ public final class MergeUtils {
5757
ChatChoiceLogProbabilityInfo.class, int.class, CompletionsFinishReason.class };
5858

5959
private static final Class<?>[] chatResponseMessageConstructorArgumentTypes = new Class<?>[] { ChatRole.class,
60-
String.class };
60+
String.class, String.class };
6161

6262
private MergeUtils() {
6363

@@ -210,9 +210,6 @@ private static ChatChoice mergeChatChoice(ChatChoice left, ChatChoice right) {
210210
? left.getContentFilterResults() : right.getContentFilterResults();
211211
setField(instance, "contentFilterResults", contentFilterResults);
212212

213-
var finishDetails = left.getFinishDetails() != null ? left.getFinishDetails() : right.getFinishDetails();
214-
setField(instance, "finishDetails", finishDetails);
215-
216213
var enhancements = left.getEnhancements() != null ? left.getEnhancements() : right.getEnhancements();
217214
setField(instance, "enhancements", enhancements);
218215

@@ -239,8 +236,10 @@ else if (left.getContent() == null) {
239236
content = left.getContent();
240237
}
241238

239+
String refusal = left.getRefusal() != null ? left.getRefusal() : right.getRefusal();
240+
242241
ChatResponseMessage instance = newInstance(chatResponseMessageConstructorArgumentTypes,
243-
ChatResponseMessage.class, role, content);
242+
ChatResponseMessage.class, role, refusal, content);
244243

245244
List<ChatCompletionsToolCall> toolCalls = new ArrayList<>();
246245
if (left.getToolCalls() == null) {

pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,7 @@
172172
<spring-cloud-function-context.version>4.1.3</spring-cloud-function-context.version>
173173
<spring-boot.version>3.3.4</spring-boot.version>
174174
<ST4.version>4.3.4</ST4.version>
175-
<azure-open-ai-client.version>1.0.0-beta.10</azure-open-ai-client.version>
175+
<azure-open-ai-client.version>1.0.0-beta.12</azure-open-ai-client.version>
176176
<jtokkit.version>1.1.0</jtokkit.version>
177177
<victools.version>4.31.1</victools.version>
178178
<kotlin.version>1.9.25</kotlin.version>

0 commit comments

Comments
 (0)