From f7830b5e9d539c50ded95f823fe350c5a4edaa64 Mon Sep 17 00:00:00 2001
From: Rock Chin <1010553892@qq.com>
Date: Thu, 2 Mar 2023 17:57:39 +0800
Subject: [PATCH] =?UTF-8?q?feat(modelmgr.py):=20=E5=AE=8C=E5=96=84?=
 =?UTF-8?q?=E5=8F=AF=E9=80=89=E6=A8=A1=E5=9E=8B=E5=88=97=E8=A1=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/openai/modelmgr.py | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/pkg/openai/modelmgr.py b/pkg/openai/modelmgr.py
index 6360f40b..dbd44054 100644
--- a/pkg/openai/modelmgr.py
+++ b/pkg/openai/modelmgr.py
@@ -1,12 +1,26 @@
 # 提供与模型交互的抽象接口
 import openai, logging
 
-CHAT_COMPLETION_MODELS = {
-    'gpt-3.5-turbo'
+COMPLETION_MODELS = {
+    'text-davinci-003',
+    'text-davinci-002',
+    'code-davinci-002',
+    'code-cushman-001',
+    'text-curie-001',
+    'text-babbage-001',
+    'text-ada-001',
 }
 
-COMPLETION_MODELS = {
-    'text-davinci-003'
+CHAT_COMPLETION_MODELS = {
+    'gpt-3.5-turbo',
+    'gpt-3.5-turbo-0301',
+    'text-davinci-003',
+    'text-davinci-002',
+    'code-davinci-002',
+    'code-cushman-001',
+    'text-curie-001',
+    'text-babbage-001',
+    'text-ada-001',
 }
 
 EDIT_MODELS = {