From 054d0839acff84edf992e84c7d3359a0471a376b Mon Sep 17 00:00:00 2001 From: RockChinQ <1010553892@qq.com> Date: Fri, 4 Aug 2023 19:08:48 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E6=9C=AA=E5=BA=8F=E5=88=97=E5=8C=96?= =?UTF-8?q?=E7=9A=84function=5Fcall=E5=B1=9E=E6=80=A7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/openai/modelmgr.py | 1 + pkg/openai/session.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/pkg/openai/modelmgr.py b/pkg/openai/modelmgr.py index 0b287752..cdc51297 100644 --- a/pkg/openai/modelmgr.py +++ b/pkg/openai/modelmgr.py @@ -112,6 +112,7 @@ def count_completion_tokens(messages: list, model: str) -> int: def count_tokens(messages: list, model: str): + if model in CHAT_COMPLETION_MODELS: return count_chat_completion_tokens(messages, model) elif model in COMPLETION_MODELS: diff --git a/pkg/openai/session.py b/pkg/openai/session.py index 2b8a145b..1400a8c6 100644 --- a/pkg/openai/session.py +++ b/pkg/openai/session.py @@ -271,7 +271,7 @@ class Session: } if 'function_call' in resp['choices'][0]['message']: - msg['function_call'] = resp['choices'][0]['message']['function_call'] + msg['function_call'] = json.dumps(resp['choices'][0]['message']['function_call']) pending_msgs.append(msg)