Просмотр исходного кода

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

CaIon 9 месяцев назад
Родитель
Сommit
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
 	}
 }