Преглед изворни кода

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

CaIon пре 9 месеци
родитељ
комит
f796c3b216
1 измењених фајлова са 3 додато и 3 уклоњено
  1. 3 3
      relay/channel/claude/adaptor.go

+ 3 - 3
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
 	}
 }