Skip to content

Commit 216e165

Browse files
committed
fix formatting
1 parent 760a466 commit 216e165

File tree

2 files changed

+37
-41
lines changed

2 files changed

+37
-41
lines changed

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

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -174,9 +174,9 @@ public Builder withDeploymentName(String deploymentName) {
174174
}
175175

176176
public Builder withFrequencyPenalty(Float frequencyPenalty) {
177-
if(frequencyPenalty != null) {
178-
this.options.frequencyPenalty = frequencyPenalty.doubleValue();
179-
}
177+
if (frequencyPenalty != null) {
178+
this.options.frequencyPenalty = frequencyPenalty.doubleValue();
179+
}
180180
return this;
181181
}
182182

@@ -196,9 +196,9 @@ public Builder withN(Integer n) {
196196
}
197197

198198
public Builder withPresencePenalty(Float presencePenalty) {
199-
if(presencePenalty != null) {
200-
this.options.presencePenalty = presencePenalty.doubleValue();
201-
}
199+
if (presencePenalty != null) {
200+
this.options.presencePenalty = presencePenalty.doubleValue();
201+
}
202202
return this;
203203
}
204204

models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/AzureChatCompletionsOptionsTests.java

Lines changed: 31 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -56,39 +56,35 @@ public void createRequestWithChatOptions() {
5656
assertThat(requestOptions.getTemperature()).isEqualTo(99.9f);
5757
}
5858

59-
private static Stream<Arguments> providePresencePenaltyAndFrequencyPenaltyTest() {
60-
return Stream.of(
61-
Arguments.of(0.0f, 0.0f),
62-
Arguments.of(0.0f, 1.0f),
63-
Arguments.of(1.0f, 0.0f),
64-
Arguments.of(1.0f, 1.0f),
65-
Arguments.of(1.0f, null),
66-
Arguments.of(null, 1.0f),
67-
Arguments.of(null, null)
68-
);
69-
}
70-
71-
@ParameterizedTest
72-
@MethodSource("providePresencePenaltyAndFrequencyPenaltyTest")
73-
public void createChatOptionsWithPresencePenaltyAndFrequencyPenalty(Float presencePenalty, Float frequencyPenalty) {
74-
var options = AzureOpenAiChatOptions.builder()
75-
.withMaxTokens(800)
76-
.withTemperature(0.7F)
77-
.withTopP(0.95F)
78-
.withPresencePenalty(presencePenalty)
79-
.withFrequencyPenalty(frequencyPenalty)
80-
.build();
81-
82-
if (presencePenalty == null) {
83-
assertThat(options.getPresencePenalty()).isEqualTo(null);
84-
} else {
85-
assertThat(options.getPresencePenalty().floatValue()).isEqualTo(presencePenalty);
86-
}
87-
88-
if (frequencyPenalty == null) {
89-
assertThat(options.getFrequencyPenalty()).isEqualTo(null);
90-
} else {
91-
assertThat(options.getFrequencyPenalty().floatValue()).isEqualTo(frequencyPenalty);
92-
}
93-
}
59+
private static Stream<Arguments> providePresencePenaltyAndFrequencyPenaltyTest() {
60+
return Stream.of(Arguments.of(0.0f, 0.0f), Arguments.of(0.0f, 1.0f), Arguments.of(1.0f, 0.0f),
61+
Arguments.of(1.0f, 1.0f), Arguments.of(1.0f, null), Arguments.of(null, 1.0f), Arguments.of(null, null));
62+
}
63+
64+
@ParameterizedTest
65+
@MethodSource("providePresencePenaltyAndFrequencyPenaltyTest")
66+
public void createChatOptionsWithPresencePenaltyAndFrequencyPenalty(Float presencePenalty, Float frequencyPenalty) {
67+
var options = AzureOpenAiChatOptions.builder()
68+
.withMaxTokens(800)
69+
.withTemperature(0.7F)
70+
.withTopP(0.95F)
71+
.withPresencePenalty(presencePenalty)
72+
.withFrequencyPenalty(frequencyPenalty)
73+
.build();
74+
75+
if (presencePenalty == null) {
76+
assertThat(options.getPresencePenalty()).isEqualTo(null);
77+
}
78+
else {
79+
assertThat(options.getPresencePenalty().floatValue()).isEqualTo(presencePenalty);
80+
}
81+
82+
if (frequencyPenalty == null) {
83+
assertThat(options.getFrequencyPenalty()).isEqualTo(null);
84+
}
85+
else {
86+
assertThat(options.getFrequencyPenalty().floatValue()).isEqualTo(frequencyPenalty);
87+
}
88+
}
89+
9490
}

0 commit comments

Comments
 (0)