From a1c167fb7f0996b12c0ef15d0b0f1e5822a00366 Mon Sep 17 00:00:00 2001 From: Rock Chin <1010553892@qq.com> Date: Fri, 3 Mar 2023 00:21:16 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=8A=9F=E8=83=BD=E5=AE=8C=E6=88=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/openai/modelmgr.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pkg/openai/modelmgr.py b/pkg/openai/modelmgr.py index 5f36f7c3..0a68fac4 100644 --- a/pkg/openai/modelmgr.py +++ b/pkg/openai/modelmgr.py @@ -65,7 +65,7 @@ class ChatCompletionModel(ModelRequest): super().__init__(model_name, user_name, request_fun) def request(self, prompts, **kwargs): - ret = self.request_fun(messages = self.__msg_handle__(prompts), **kwargs, user=self.user_name) + self.ret = self.request_fun(messages = self.__msg_handle__(prompts), **kwargs, user=self.user_name) self.ret_handle() self.message = self.ret["choices"][0]["message"]['content'] @@ -87,7 +87,7 @@ class CompletionModel(ModelRequest): super().__init__(model_name, user_name, request_fun) def request(self, prompts, **kwargs): - ret = self.request_fun(prompt = self.__msg_handle__(prompts), **kwargs) + self.ret = self.request_fun(prompt = self.__msg_handle__(prompts), **kwargs) self.ret_handle() self.message = self.ret["choices"][0]["text"]