diff --git a/erniebot/src/erniebot/resources/chat_completion.py b/erniebot/src/erniebot/resources/chat_completion.py index 4b861721..e4092ff7 100644 --- a/erniebot/src/erniebot/resources/chat_completion.py +++ b/erniebot/src/erniebot/resources/chat_completion.py @@ -97,12 +97,8 @@ class ChatCompletion(EBResource, CreatableWithStreaming): "ernie-char-8k": { "model_id": "ernie-char-8k", }, - "ernie-func-8k": { - "model_id": "ernie-func-8k" - }, - "ernie-function-8k": { - "model_id": "ernie-func-8k" - } + "ernie-func-8k": {"model_id": "ernie-func-8k"}, + "ernie-function-8k": {"model_id": "ernie-func-8k"}, }, }, APIType.AISTUDIO: { @@ -135,12 +131,8 @@ class ChatCompletion(EBResource, CreatableWithStreaming): "ernie-char-8k": { "model_id": "ernie-char-8k", }, - "ernie-func-8k": { - "model_id": "ernie-func-8k" - }, - "ernie-function-8k": { - "model_id": "ernie-func-8k" - } + "ernie-func-8k": {"model_id": "ernie-func-8k"}, + "ernie-function-8k": {"model_id": "ernie-func-8k"}, }, }, APIType.CUSTOM: {