weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

使zhipuai变为非必要依赖(close #1369)

This commit is contained in:
liunux4odoo 2023-09-05 12:43:51 +08:00
parent 464ccc6d53
commit 3e87530478
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/model_workers/zhipu.py
View File

@ -1,4 +1,3 @@
import zhipuai
from server.model_workers.base import ApiModelWorker from server.model_workers.base import ApiModelWorker
from fastchat import conversation as conv from fastchat import conversation as conv
import sys import sys
@ -37,6 +36,7 @@ class ChatGLMWorker(ApiModelWorker):
def generate_stream_gate(self, params): def generate_stream_gate(self, params):
# TODO: 支持stream参数，维护request_id，传过来的prompt也有问题 # TODO: 支持stream参数，维护request_id，传过来的prompt也有问题
from server.utils import get_model_worker_config from server.utils import get_model_worker_config
import zhipuai
super().generate_stream_gate(params) super().generate_stream_gate(params)
zhipuai.api_key = get_model_worker_config("chatglm-api").get("api_key") zhipuai.api_key = get_model_worker_config("chatglm-api").get("api_key")