Spaces:
Paused
Paused
add missing body fields for z.ai
Browse files- 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 |
-
|
| 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":
|
| 201 |
-
"web_search":
|
| 202 |
-
"auto_web_search":
|
| 203 |
-
"preview_mode":
|
| 204 |
-
"enable_thinking":
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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(),
|