From f796c3b216cf3a748d58abea65d2d0acaf8d68e6 Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Fri, 23 May 2025 01:34:53 +0800
Subject: [PATCH] fix: update Init method to correctly set RequestMode based on
 upstream model name prefixes

---
 relay/channel/claude/adaptor.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/relay/channel/claude/adaptor.go b/relay/channel/claude/adaptor.go
index 4b071712..8389b9f1 100644
--- a/relay/channel/claude/adaptor.go
+++ b/relay/channel/claude/adaptor.go
@@ -38,10 +38,10 @@ func (a *Adaptor) ConvertImageRequest(c *gin.Context, info *relaycommon.RelayInf
 }
 
 func (a *Adaptor) Init(info *relaycommon.RelayInfo) {
-	if strings.HasPrefix(info.UpstreamModelName, "claude-3") {
-		a.RequestMode = RequestModeMessage
-	} else {
+	if strings.HasPrefix(info.UpstreamModelName, "claude-2") || strings.HasPrefix(info.UpstreamModelName, "claude-instant") {
 		a.RequestMode = RequestModeCompletion
+	} else {
+		a.RequestMode = RequestModeMessage
 	}
 }