From 5035cd054a66de6b792093c585f4fccb047a9349 Mon Sep 17 00:00:00 2001 From: "1808837298@qq.com" <1808837298@qq.com> Date: Wed, 9 Oct 2024 00:42:36 +0800 Subject: [PATCH] feat: update aws claude --- relay/channel/aws/adaptor.go | 14 +++----------- 1 file changed, 3 insertions(+), 11 deletions(-) diff --git a/relay/channel/aws/adaptor.go b/relay/channel/aws/adaptor.go index 44a870d..875d3dd 100644 --- a/relay/channel/aws/adaptor.go +++ b/relay/channel/aws/adaptor.go @@ -8,7 +8,6 @@ import ( "one-api/dto" "one-api/relay/channel/claude" relaycommon "one-api/relay/common" - "strings" ) const ( @@ -31,11 +30,7 @@ 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 { - a.RequestMode = RequestModeCompletion - } + a.RequestMode = RequestModeMessage } func (a *Adaptor) GetRequestURL(info *relaycommon.RelayInfo) (string, error) { @@ -53,11 +48,8 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, info *relaycommon.RelayInfo, re var claudeReq *claude.ClaudeRequest var err error - if a.RequestMode == RequestModeCompletion { - claudeReq = claude.RequestOpenAI2ClaudeComplete(*request) - } else { - claudeReq, err = claude.RequestOpenAI2ClaudeMessage(*request) - } + claudeReq, err = claude.RequestOpenAI2ClaudeMessage(*request) + c.Set("request_model", request.Model) c.Set("converted_request", claudeReq) return claudeReq, err