From 2c740fc641194c655bd216a174a2d068e5c360cc Mon Sep 17 00:00:00 2001
From: binary-husky <qingxu.fu@outlook.com>
Date: Tue, 16 Jan 2024 21:51:14 +0800
Subject: [PATCH] =?UTF-8?q?welcome=20glm4=20from=20=E6=99=BA=E8=B0=B1!?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 config.py                    | 2 +-
 request_llms/com_zhipuapi.py | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/config.py b/config.py
index c202ca0d..0bdba130 100644
--- a/config.py
+++ b/config.py
@@ -195,7 +195,7 @@ XFYUN_API_KEY = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
 
 # 接入智谱大模型
 ZHIPUAI_API_KEY = ""
-ZHIPUAI_MODEL = "chatglm_turbo"
+ZHIPUAI_MODEL = "glm-4" # 可选 "glm-3-turbo" "glm-4"
 
 
 # Claude API KEY
diff --git a/request_llms/com_zhipuapi.py b/request_llms/com_zhipuapi.py
index 445720d8..ba826d20 100644
--- a/request_llms/com_zhipuapi.py
+++ b/request_llms/com_zhipuapi.py
@@ -26,6 +26,8 @@ class ZhipuRequestInstance():
         )
         for event in response.events():
             if event.event == "add":
+                # if self.result_buf == "" and event.data.startswith(" "):
+                #     event.data = event.data.lstrip(" ") # 每次智谱为啥都要带个空格开头呢？
                 self.result_buf += event.data
                 yield self.result_buf
             elif event.event == "error" or event.event == "interrupted":