File tree Expand file tree Collapse file tree 1 file changed +13
-6
lines changed Expand file tree Collapse file tree 1 file changed +13
-6
lines changed Original file line number Diff line number Diff line change @@ -77,17 +77,24 @@ export class ErnieApi implements LLMApi {
77
77
78
78
async chat ( options : ChatOptions ) {
79
79
const messages = options . messages . map ( ( v ) => ( {
80
- // "error_code": 336006, "error_msg": "the role of message with odd index in the messages must be assistant ",
81
- role : v . role === "system" ? "assistant " : v . role ,
80
+ // "error_code": 336006, "error_msg": "the role of message with even index in the messages must be user or function ",
81
+ role : v . role === "system" ? "user " : v . role ,
82
82
content : getMessageTextContent ( v ) ,
83
83
} ) ) ;
84
84
85
85
// "error_code": 336006, "error_msg": "the length of messages must be an odd number",
86
86
if ( messages . length % 2 === 0 ) {
87
- messages . unshift ( {
88
- role : "user" ,
89
- content : " " ,
90
- } ) ;
87
+ if ( messages . at ( 0 ) ?. role === "user" ) {
88
+ messages . splice ( 1 , 0 , {
89
+ role : "assistant" ,
90
+ content : " " ,
91
+ } ) ;
92
+ } else {
93
+ messages . unshift ( {
94
+ role : "user" ,
95
+ content : " " ,
96
+ } ) ;
97
+ }
91
98
}
92
99
93
100
const modelConfig = {
You can’t perform that action at this time.
0 commit comments