From 3c894fe70e5950e9227f2cde515f07790ad6ff17 Mon Sep 17 00:00:00 2001 From: RockChinQ <1010553892@qq.com> Date: Sat, 29 Jul 2023 16:29:16 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20chat=5Fcompletion=E7=9A=84=E5=87=BD?= =?UTF-8?q?=E6=95=B0=E5=BC=80=E5=85=B3=E6=94=AF=E6=8C=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/openai/api/chat_completion.py | 20 ++++++++++++++------ 1 file changed, 14 insertions(+), 6 deletions(-) diff --git a/pkg/openai/api/chat_completion.py b/pkg/openai/api/chat_completion.py index c33c19ae..f9cab135 100644 --- a/pkg/openai/api/chat_completion.py +++ b/pkg/openai/api/chat_completion.py @@ -69,12 +69,20 @@ class ChatCompletionRequest(RequestBase): if self.pending_func_call is None: # 没有待处理的函数调用请求 - resp = self._req( - model=self.model, - messages=self.messages, - functions=get_func_schema_list(), - **self.kwargs - ) + args = { + "model": self.model, + "messages": self.messages, + } + + funcs = get_func_schema_list() + + if len(funcs) > 0: + args['functions'] = funcs + + # 拼接kwargs + args = {**args, **self.kwargs} + + resp = self._req(**args) choice0 = resp["choices"][0]