From 3e87530478e79b314b0ff7f0a698771391be7b96 Mon Sep 17 00:00:00 2001 From: liunux4odoo Date: Tue, 5 Sep 2023 12:43:51 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BD=BFzhipuai=E5=8F=98=E4=B8=BA=E9=9D=9E?= =?UTF-8?q?=E5=BF=85=E8=A6=81=E4=BE=9D=E8=B5=96(close=20#1369)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/model_workers/zhipu.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server/model_workers/zhipu.py b/server/model_workers/zhipu.py index 4e4e15e..82eab5d 100644 --- a/server/model_workers/zhipu.py +++ b/server/model_workers/zhipu.py @@ -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")