44import com .fasterxml .jackson .annotation .JsonInclude ;
55import com .fasterxml .jackson .annotation .JsonProperty ;
66import com .plexpt .chatgpt .util .TokensUtil ;
7+ import lombok .*;
8+ import lombok .extern .slf4j .Slf4j ;
79
8- import java .io .Serializable ;
910import java .util .List ;
1011import java .util .Map ;
1112
12- import lombok .AllArgsConstructor ;
13- import lombok .Builder ;
14- import lombok .Data ;
15- import lombok .Getter ;
16- import lombok .NoArgsConstructor ;
17- import lombok .NonNull ;
18- import lombok .extern .slf4j .Slf4j ;
19-
2013/**
2114 * chat
2215 *
2922@ NoArgsConstructor (force = true )
3023@ JsonInclude (JsonInclude .Include .NON_NULL )
3124@ JsonIgnoreProperties (ignoreUnknown = true )
32- public class ChatCompletion implements Serializable {
25+ public class ChatCompletion {
3326
3427 @ NonNull
3528 @ Builder .Default
36- private String model = Model . GPT_3_5_TURBO_0613 . getName () ;
29+ private String model = "gpt-3.5-turbo" ;
3730
3831 @ NonNull
3932 private List <Message > messages ;
@@ -60,6 +53,11 @@ public class ChatCompletion implements Serializable {
6053 */
6154 String function_call ;
6255
56+ @ JsonProperty ("tool_choice" )
57+ String toolChoice ;
58+
59+ List <ChatTool > tools ;
60+
6361 List <ChatFunction > functions ;
6462
6563 /**
@@ -124,13 +122,16 @@ public enum Model {
124122 */
125123 GPT_3_5_TURBO_0301 ("gpt-3.5-turbo-0301" ),
126124 GPT_3_5_TURBO_1106 ("gpt-3.5-turbo-1106" ),
125+ GPT_3_5_TURBO_0125 ("gpt-3.5-turbo-0125" ),
127126 GPT_3_5_TURBO_INSTRUCT ("gpt-3.5-turbo-instruct" ),
128127 /**
129128 * GPT4.0
130129 */
131130 GPT_4 ("gpt-4" ),
132131 GPT4Turbo ("gpt-4-1106-preview" ),
132+ GPT4Turbo0125 ("gpt-4-0125-preview" ),
133133 GPT_4VP ("gpt-4-vision-preview" ),
134+ GPT_4V ("gpt-4-vision-preview" ),
134135 /**
135136 * 临时模型,不建议使用
136137 */
0 commit comments