From fc41c8cbf8e5b7cddf6ce5f45bf55c89bf70b874 Mon Sep 17 00:00:00 2001
From: Yang Luo <hsluoyz@qq.com>
Date: Wed, 13 Mar 2024 09:57:09 +0800
Subject: [PATCH] Move "custom-model" to chat model

---
 model/openai_util.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model/openai_util.go b/model/openai_util.go
index 68f42ff5f..8485c0e7e 100644
--- a/model/openai_util.go
+++ b/model/openai_util.go
@@ -62,13 +62,13 @@ func getOpenAiModelType(model string) string {
 		"dall-e-3", "gpt-4-vision-preview", "gpt-4-32k-0613", "gpt-4-32k-0314", "gpt-4-32k",
 		"gpt-4-0613", "gpt-4-0314", "gpt-4", "gpt-3.5-turbo-0613",
 		"gpt-3.5-turbo-0301", "gpt-3.5-turbo-16k", "gpt-3.5-turbo-16k-0613",
-		"gpt-3.5-turbo",
+		"gpt-3.5-turbo", "custom-model",
 	}
 
 	completionModels := []string{
 		"text-davinci-003", "text-davinci-002", "text-curie-001",
 		"text-babbage-001", "text-ada-001", "text-davinci-001",
-		"davinci-instruct-beta", "davinci", "curie-instruct-beta", "curie", "ada", "babbage", "custom-model",
+		"davinci-instruct-beta", "davinci", "curie-instruct-beta", "curie", "ada", "babbage",
 	}
 
 	for _, chatModel := range chatModels {