diff --git a/erniebot/src/erniebot/resources/chat_completion.py b/erniebot/src/erniebot/resources/chat_completion.py index b5070ccc..e4092ff7 100644 --- a/erniebot/src/erniebot/resources/chat_completion.py +++ b/erniebot/src/erniebot/resources/chat_completion.py @@ -97,6 +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"}, }, }, APIType.AISTUDIO: { @@ -129,6 +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"}, }, }, APIType.CUSTOM: {