xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-02 03:55:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 未序列化的function_call属性

Browse Source
This commit is contained in:
RockChinQ
2023-08-04 19:08:48 +08:00
parent dd75f98d85
commit 054d0839ac
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pkg/openai/modelmgr.py
View File

@@ -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:

2
pkg/openai/session.py
View File

@@ -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)