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

feat: Enhance ConvertClaudeRequest method to set request model and handle vertex-specific request conversion

1808837298@qq.com 11 месяцев назад
Родитель
Сommit
19935ee8ac
2 измененных файлов с 10 добавлено и 1 удалено
  1. 2 0
      relay/channel/aws/adaptor.go
  2. 8 1
      relay/channel/vertex/adaptor.go

+ 2 - 0
relay/channel/aws/adaptor.go

@@ -21,6 +21,8 @@ type Adaptor struct {
 }
 
 func (a *Adaptor) ConvertClaudeRequest(c *gin.Context, info *relaycommon.RelayInfo, request *dto.ClaudeRequest) (any, error) {
+	c.Set("request_model", request.Model)
+	c.Set("converted_request", request)
 	return request, nil
 }
 

+ 8 - 1
relay/channel/vertex/adaptor.go

@@ -39,8 +39,15 @@ type Adaptor struct {
 }
 
 func (a *Adaptor) ConvertClaudeRequest(c *gin.Context, info *relaycommon.RelayInfo, request *dto.ClaudeRequest) (any, error) {
-	return request, nil
+	if v, ok := claudeModelMap[info.UpstreamModelName]; ok {
+		c.Set("request_model", v)
+	} else {
+		c.Set("request_model", request.Model)
+	}
+	vertexClaudeReq := copyRequest(request, anthropicVersion)
+	return vertexClaudeReq, nil
 }
+
 func (a *Adaptor) ConvertAudioRequest(c *gin.Context, info *relaycommon.RelayInfo, request dto.AudioRequest) (io.Reader, error) {
 	//TODO implement me
 	return nil, errors.New("not implemented")