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

feat(adaptor): Add support for Claude-specific headers in SetupRequestHeader

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

+ 3 - 0
relay/channel/vertex/adaptor.go

@@ -222,6 +222,9 @@ func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Header, info *rel
 	if a.AccountCredentials.ProjectID != "" {
 		req.Set("x-goog-user-project", a.AccountCredentials.ProjectID)
 	}
+	if strings.Contains(info.UpstreamModelName, "claude") {
+		claude.CommonClaudeHeadersOperation(c, req, info)
+	}
 	return nil
 }