支持指定绘图的智能体

LLM-Red-Team · Mar 27, 2024 · fc3a40f · fc3a40f
1 parent 1b1b3dd
commit fc3a40f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts
@@ -394,7 +394,7 @@ async function generateImages(
       {
         headers: {
           Authorization: `Bearer ${token}`,
-          Referer: `https://chatglm.cn/main/gdetail/${assistantId}`,
+          Referer: `https://chatglm.cn/main/gdetail/${model}`,
           "X-Device-Id": util.uuid(false),
           "X-Request-Id": util.uuid(false),
           ...FAKE_HEADERS,
@@ -420,7 +420,7 @@ async function generateImages(
     );
 
     // 异步移除会话，如果消息不合规，此操作可能会抛出数据库错误异常，请忽略
-    removeConversation(convId, refreshToken, assistantId).catch((err) =>
+    removeConversation(convId, refreshToken, model).catch((err) =>
       console.error(err)
     );