chunzhi/chatgpt_academic
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

welcome glm4 from 智谱!

Browse Source
This commit is contained in:
binary-husky 2024-01-16 21:51:14 +08:00
parent 96832a8228
commit 2c740fc641
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.py
View File

@ -195,7 +195,7 @@ XFYUN_API_KEY = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
# 接入智谱大模型 # 接入智谱大模型
ZHIPUAI_API_KEY = "" ZHIPUAI_API_KEY = ""
ZHIPUAI_MODEL = "chatglm_turbo" ZHIPUAI_MODEL = "glm-4" # 可选 "glm-3-turbo" "glm-4"
# Claude API KEY # Claude API KEY

2
request_llms/com_zhipuapi.py
View File

@ -26,6 +26,8 @@ class ZhipuRequestInstance():
) )
for event in response.events(): for event in response.events():
if event.event == "add": if event.event == "add":
# if self.result_buf == "" and event.data.startswith(" "):
# event.data = event.data.lstrip(" ") # 每次智谱为啥都要带个空格开头呢？
self.result_buf += event.data self.result_buf += event.data
yield self.result_buf yield self.result_buf
elif event.event == "error" or event.event == "interrupted": elif event.event == "error" or event.event == "interrupted":