upstream asis

This commit is contained in:
2023-09-16 10:52:33 +08:00
parent 1fcdc59b63
commit 043fb3db85
2 changed files with 8 additions and 3 deletions

View File

@@ -20,7 +20,8 @@ func handleAuth(c *gin.Context) error {
authorization = strings.Trim(authorization[len("Bearer"):], " ") authorization = strings.Trim(authorization[len("Bearer"):], " ")
log.Println("Received authorization", authorization) log.Println("Received authorization", authorization)
if authorization != authConfig.Value {
if authConfig.Value != "asis" && authorization != authConfig.Value {
err = errors.New("wrong authorization header") err = errors.New("wrong authorization header")
c.AbortWithError(403, err) c.AbortWithError(403, err)
return err return err

View File

@@ -185,7 +185,11 @@ func main() {
out.URL.Path = in.URL.Path out.URL.Path = in.URL.Path
out.Header = http.Header{} out.Header = http.Header{}
out.Header.Set("Host", remote.Host) out.Header.Set("Host", remote.Host)
out.Header.Set("Authorization", "Bearer "+upstream.SK) if upstream.SK == "asis" {
out.Header.Set("Authorization", c.Request.Header.Get("Authorization"))
} else {
out.Header.Set("Authorization", "Bearer "+upstream.SK)
}
out.Header.Set("Content-Type", c.Request.Header.Get("Content-Type")) out.Header.Set("Content-Type", c.Request.Header.Get("Content-Type"))
} }
var buf bytes.Buffer var buf bytes.Buffer
@@ -230,7 +234,7 @@ func main() {
} }
content := fmt.Sprintf("[%s] OpenAI 转发出错 ID: %d... 密钥: [%s] 上游: [%s] 错误: %s\n---\n%s", content := fmt.Sprintf("[%s] OpenAI 转发出错 ID: %d... 密钥: [%s] 上游: [%s] 错误: %s\n---\n%s",
c.ClientIP(), c.ClientIP(),
upstream.ID, upstream.SK[:10], upstream.Endpoint, err.Error(), upstream.ID, upstream.SK, upstream.Endpoint, err.Error(),
strings.Join(upstreamDescriptions, "\n"), strings.Join(upstreamDescriptions, "\n"),
) )
go sendMatrixMessage(content) go sendMatrixMessage(content)