Skip to content

Commit bdaf93f

Browse files
committed
fix: modify ChatRequest
1 parent 8bc8564 commit bdaf93f

File tree

2 files changed

+65
-2
lines changed

2 files changed

+65
-2
lines changed

base/src/main/java/com/tinyengine/it/model/dto/ChatRequest.java

Lines changed: 29 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212

1313
package com.tinyengine.it.model.dto;
1414

15+
import com.fasterxml.jackson.annotation.JsonProperty;
1516
import lombok.Data;
1617

1718
/**
@@ -26,7 +27,34 @@ public class ChatRequest {
2627
private String baseUrl;
2728
private Object messages;
2829
private Object tools;
29-
private Double temperature = 0.7;
30+
private Float temperature;
3031
private boolean stream = false;
32+
@JsonProperty("max_tokens")
3133
private Integer maxTokens;
34+
@JsonProperty("stream_options")
35+
private Object streamOptions;
36+
@JsonProperty("presence_penalty")
37+
private Float presencePenalty;
38+
@JsonProperty("response_format")
39+
private Object responseFormat;
40+
@JsonProperty("max_input_tokens")
41+
private Integer maxInputTokens;
42+
@JsonProperty("vl_high_resolution_images")
43+
private Boolean vlHighResolutionImages = false;
44+
@JsonProperty("enable_thinking")
45+
private Boolean enableThinking;
46+
@JsonProperty("thinking_budget")
47+
private Integer thinkingBudget;
48+
private String stop;
49+
@JsonProperty("tool_choice")
50+
private Object toolChoice;
51+
@JsonProperty("parallel_tool_calls")
52+
private Boolean parallelToolCalls;
53+
@JsonProperty("enable_search")
54+
private Boolean enableSearch;
55+
@JsonProperty("search_options")
56+
private Object searchOptions;
57+
@JsonProperty("frequency_penalty")
58+
private Float frequencyPenalty;
59+
3260
}

base/src/main/java/com/tinyengine/it/service/app/impl/v1/AiChatV1ServiceImpl.java

Lines changed: 36 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -285,9 +285,44 @@ private String buildRequestBody(ChatRequest request) {
285285
if (request.getMaxTokens() != null) {
286286
body.put("max_tokens", request.getMaxTokens());
287287
}
288-
if (request.getTemperature() != null) {
288+
body.put("temperature", request.getTemperature());
289+
if(request.getTemperature() != null) {
289290
body.put("temperature", request.getTemperature());
290291
}
292+
if(request.getSearchOptions() != null) {
293+
body.put("stream_options", request.getSearchOptions());
294+
}
295+
if(request.getPresencePenalty() != null) {
296+
body.put("presence_penalty", request.getPresencePenalty());
297+
}
298+
if(request.getResponseFormat() != null) {
299+
body.put("response_format", request.getResponseFormat());
300+
}
301+
if(request.getMaxInputTokens() != null) {
302+
body.put("max_input_tokens", request.getMaxInputTokens());
303+
}
304+
if(request.getMaxInputTokens() != null) {
305+
body.put("vl_high_resolution_images", request.getVlHighResolutionImages());
306+
}
307+
if(request.getEnableThinking() != null) {
308+
body.put("enable_thinking", request.getEnableThinking());
309+
}
310+
if(request.getToolChoice() != null) {
311+
body.put("tool_choice", request.getToolChoice());
312+
}
313+
if(request.getStop() != null) {
314+
body.put("stop", request.getStop());
315+
}
316+
if(request.getParallelToolCalls() != null) {
317+
body.put("parallel_tool_calls", request.getParallelToolCalls());
318+
}
319+
if(request.getEnableSearch() != null) {
320+
body.put("enable_search", request.getEnableSearch());
321+
}
322+
if(request.getFrequencyPenalty() != null) {
323+
body.put("frequency_penalty", request.getFrequencyPenalty());
324+
}
325+
291326
return JsonUtils.encode(body);
292327
}
293328

0 commit comments

Comments
 (0)