From fc3a40f0fe08943b0515e6b4fe87bae76d5fe71e Mon Sep 17 00:00:00 2001
From: Vinlic <vinlic@vinlic.com>
Date: Wed, 27 Mar 2024 10:44:00 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=AF=E6=8C=81=E6=8C=87=E5=AE=9A=E7=BB=98?=
 =?UTF-8?q?=E5=9B=BE=E7=9A=84=E6=99=BA=E8=83=BD=E4=BD=93?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/api/controllers/chat.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts
index b29fec6..a743853 100644
--- 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)
     );