fix: update Init method to correctly set RequestMode based on upstream model name prefixes

94455e34 · CaIon · adb2c395 · 94455e34
Commit 94455e34 authored May 23, 2025 by CaIon
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

relay/channel/claude/adaptor.go
+3 -3

No files found.
--- 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") {
+	if strings.HasPrefix(info.UpstreamModelName, "claude-2") || strings.HasPrefix(info.UpstreamModelName, "claude-instant") {
-		a.RequestMode = RequestModeMessage
-	} else {
 		a.RequestMode = RequestModeCompletion
+	} else {
+		a.RequestMode = RequestModeMessage
 	}
 }