Moge-Row commited on
Commit
4fca881
·
1 Parent(s): b813260

add missing body fields for z.ai

Browse files
Files changed (1) hide show
  1. internal/upstream/client.go +16 -6
internal/upstream/client.go CHANGED
@@ -190,18 +190,28 @@ func MakeUpstreamRequest(token string, messages []model.Message, modelName strin
190
  upstreamMessages = append([]map[string]interface{}{systemUserMsg, systemAssistantMsg}, upstreamMessages...)
191
  }
192
 
193
- body := map[string]interface{}{
194
  "stream": true,
195
  "model": targetModel,
196
  "messages": upstreamMessages,
197
  "signature_prompt": latestUserContent,
198
  "params": map[string]interface{}{},
 
 
 
 
 
 
199
  "features": map[string]interface{}{
200
- "image_generation": false,
201
- "web_search": false,
202
- "auto_web_search": autoWebSearch,
203
- "preview_mode": true,
204
- "enable_thinking": enableThinking,
 
 
 
 
205
  },
206
  "chat_id": chatID,
207
  "id": uuid.New().String(),
 
190
  upstreamMessages = append([]map[string]interface{}{systemUserMsg, systemAssistantMsg}, upstreamMessages...)
191
  }
192
 
193
+ body := map[string]interface{}{
194
  "stream": true,
195
  "model": targetModel,
196
  "messages": upstreamMessages,
197
  "signature_prompt": latestUserContent,
198
  "params": map[string]interface{}{},
199
+ "extra": map[string]interface{}{},
200
+ "variables": map[string]interface{}{},
201
+ "background_tasks": map[string]interface{}{
202
+ "tags_generation": true,
203
+ "title_generation": true,
204
+ },
205
  "features": map[string]interface{}{
206
+ "image_generation": false,
207
+ "web_search": false,
208
+ "auto_web_search": autoWebSearch,
209
+ "preview_mode": true,
210
+ "enable_thinking": enableThinking,
211
+ "flags": []interface{}{},
212
+ "vlm_tools_enable": false,
213
+ "vlm_web_search_enable": false,
214
+ "vlm_website_mode": false,
215
  },
216
  "chat_id": chatID,
217
  "id": uuid.New().String(),