Skip to content

Commit 3449656

Browse files
committed
resolve comments
Signed-off-by: Alexandros Pappas <[email protected]>
1 parent 93535f9 commit 3449656

File tree

2 files changed

+17
-17
lines changed

2 files changed

+17
-17
lines changed

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

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -805,7 +805,6 @@ public record ChatCompletionRequest(// @formatter:off
805805
@JsonProperty("messages") List<ChatCompletionMessage> messages,
806806
@JsonProperty("model") String model,
807807
@JsonProperty("store") Boolean store,
808-
@JsonProperty("reasoning_effort") String reasoningEffort,
809808
@JsonProperty("metadata") Map<String, String> metadata,
810809
@JsonProperty("frequency_penalty") Double frequencyPenalty,
811810
@JsonProperty("logit_bias") Map<String, Integer> logitBias,
@@ -828,7 +827,8 @@ public record ChatCompletionRequest(// @formatter:off
828827
@JsonProperty("tools") List<FunctionTool> tools,
829828
@JsonProperty("tool_choice") Object toolChoice,
830829
@JsonProperty("parallel_tool_calls") Boolean parallelToolCalls,
831-
@JsonProperty("user") String user) {
830+
@JsonProperty("user") String user,
831+
@JsonProperty("reasoning_effort") String reasoningEffort) {
832832

833833
/**
834834
* Shortcut constructor for a chat completion request with the given messages, model and temperature.
@@ -838,9 +838,9 @@ public record ChatCompletionRequest(// @formatter:off
838838
* @param temperature What sampling temperature to use, between 0 and 1.
839839
*/
840840
public ChatCompletionRequest(List<ChatCompletionMessage> messages, String model, Double temperature) {
841-
this(messages, model, null, null, null, null, null, null, null, null, null, null, null, null, null, null,
841+
this(messages, model, null, null, null, null, null, null, null, null, null, null, null, null, null,
842842
null, null, null, false, null, temperature, null,
843-
null, null, null, null);
843+
null, null, null, null, null);
844844
}
845845

846846
/**
@@ -851,10 +851,10 @@ public ChatCompletionRequest(List<ChatCompletionMessage> messages, String model,
851851
* @param audio Parameters for audio output. Required when audio output is requested with outputModalities: ["audio"].
852852
*/
853853
public ChatCompletionRequest(List<ChatCompletionMessage> messages, String model, AudioParameters audio, boolean stream) {
854-
this(messages, model, null, null, null, null, null, null, null,
854+
this(messages, model, null, null, null, null, null, null,
855855
null, null, null, List.of(OutputModality.AUDIO, OutputModality.TEXT), audio, null, null,
856856
null, null, null, stream, null, null, null,
857-
null, null, null, null);
857+
null, null, null, null, null);
858858
}
859859

860860
/**
@@ -867,9 +867,9 @@ public ChatCompletionRequest(List<ChatCompletionMessage> messages, String model,
867867
* as they become available, with the stream terminated by a data: [DONE] message.
868868
*/
869869
public ChatCompletionRequest(List<ChatCompletionMessage> messages, String model, Double temperature, boolean stream) {
870-
this(messages, model, null, null, null, null, null, null, null, null, null, null,
870+
this(messages, model, null, null, null, null, null, null, null, null, null,
871871
null, null, null, null, null, null, null, stream, null, temperature, null,
872-
null, null, null, null);
872+
null, null, null, null, null);
873873
}
874874

875875
/**
@@ -883,9 +883,9 @@ public ChatCompletionRequest(List<ChatCompletionMessage> messages, String model,
883883
*/
884884
public ChatCompletionRequest(List<ChatCompletionMessage> messages, String model,
885885
List<FunctionTool> tools, Object toolChoice) {
886-
this(messages, model, null, null, null, null, null, null, null, null, null, null,
886+
this(messages, model, null, null, null, null, null, null, null, null, null,
887887
null, null, null, null, null, null, null, false, null, 0.8, null,
888-
tools, toolChoice, null, null);
888+
tools, toolChoice, null, null, null);
889889
}
890890

891891
/**
@@ -896,9 +896,9 @@ public ChatCompletionRequest(List<ChatCompletionMessage> messages, String model,
896896
* as they become available, with the stream terminated by a data: [DONE] message.
897897
*/
898898
public ChatCompletionRequest(List<ChatCompletionMessage> messages, Boolean stream) {
899-
this(messages, null, null, null, null, null, null, null, null, null, null, null,
899+
this(messages, null, null, null, null, null, null, null, null, null, null,
900900
null, null, null, null, null, null, null, stream, null, null, null,
901-
null, null, null, null);
901+
null, null, null, null, null);
902902
}
903903

904904
/**
@@ -908,10 +908,10 @@ public ChatCompletionRequest(List<ChatCompletionMessage> messages, Boolean strea
908908
* @return A new {@link ChatCompletionRequest} with the specified stream options.
909909
*/
910910
public ChatCompletionRequest streamOptions(StreamOptions streamOptions) {
911-
return new ChatCompletionRequest(this.messages, this.model, this.store, this.reasoningEffort, this.metadata, this.frequencyPenalty, this.logitBias, this.logprobs,
911+
return new ChatCompletionRequest(this.messages, this.model, this.store, this.metadata, this.frequencyPenalty, this.logitBias, this.logprobs,
912912
this.topLogprobs, this.maxTokens, this.maxCompletionTokens, this.n, this.outputModalities, this.audioParameters, this.presencePenalty,
913913
this.responseFormat, this.seed, this.serviceTier, this.stop, this.stream, streamOptions, this.temperature, this.topP,
914-
this.tools, this.toolChoice, this.parallelToolCalls, this.user);
914+
this.tools, this.toolChoice, this.parallelToolCalls, this.user, this.reasoningEffort);
915915
}
916916

917917
/**

models/spring-ai-openai/src/test/java/org/springframework/ai/openai/api/OpenAiApiIT.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,9 +71,9 @@ void chatCompletionStream() {
7171
void validateReasoningTokens() {
7272
ChatCompletionMessage userMessage = new ChatCompletionMessage(
7373
"If a train travels 100 miles in 2 hours, what is its average speed?", ChatCompletionMessage.Role.USER);
74-
ChatCompletionRequest request = new ChatCompletionRequest(List.of(userMessage), "o1", null, "low", null, null,
75-
null, null, null, null, null, null, null, null, null, null, null, null, null, false, null, null, null,
76-
null, null, null, null);
74+
ChatCompletionRequest request = new ChatCompletionRequest(List.of(userMessage), "o1", null, null, null, null,
75+
null, null, null, null, null, null, null, null, null, null, null, null, false, null, null, null, null,
76+
null, null, null, "low");
7777
ResponseEntity<ChatCompletion> response = this.openAiApi.chatCompletionEntity(request);
7878

7979
assertThat(response).isNotNull();

0 commit comments

Comments
 (0)