From 735fb0c7bd75a2b3ff6bd56405219eed73a8b0c5 Mon Sep 17 00:00:00 2001
From: quzard <1191890118@qq.com>
Date: Sat, 17 Jun 2023 10:03:26 +0800
Subject: [PATCH] =?UTF-8?q?=E5=BC=BA=E5=88=B6=E4=BD=BF=E7=94=A80613?=
 =?UTF-8?q?=E6=A8=A1=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 controller/relay.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/controller/relay.go b/controller/relay.go
index 1b7cee5a..efd3168e 100644
--- a/controller/relay.go
+++ b/controller/relay.go
@@ -182,6 +182,12 @@ func relayHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
 	} else if channelType == common.ChannelTypePaLM {
 		err := relayPaLM(textRequest, c)
 		return err
+	} else {
+		// 强制使用0613模型
+		textRequest.Model = strings.TrimSuffix(textRequest.Model, "-0301")
+		textRequest.Model = strings.TrimSuffix(textRequest.Model, "-0314")
+		textRequest.Model = strings.TrimSuffix(textRequest.Model, "-0613")
+		textRequest.Model = textRequest.Model + "-0613"
 	}
 	var promptTokens int
 	switch relayMode {