使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

View File

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