Sub2API v1.0 - AI API 网关(二开初始版本,基于上游 Wei-Shaw/sub2api)
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
This commit is contained in:
@@ -0,0 +1,32 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
func setOpenAIChatGPTAccountHeaders(headers http.Header, account *Account) {
|
||||
if headers == nil || account == nil || !account.IsOpenAIOAuth() {
|
||||
return
|
||||
}
|
||||
if chatgptAccountID := account.GetChatGPTAccountID(); chatgptAccountID != "" {
|
||||
headers.Set("chatgpt-account-id", chatgptAccountID)
|
||||
}
|
||||
if account.IsChatGPTAccountFedRAMP() {
|
||||
headers.Set("x-openai-fedramp", "true")
|
||||
} else {
|
||||
headers.Del("x-openai-fedramp")
|
||||
}
|
||||
}
|
||||
|
||||
// resolveAndSetOpenAIChatGPTAccountHeaders 解析 spark 影子账号至其母账号(凭据透传),
|
||||
// 再调用 setOpenAIChatGPTAccountHeaders 写入 chatgpt-account-id / x-openai-fedramp 头。
|
||||
// 普通账号(非影子)为直通,行为与直接调用 setOpenAIChatGPTAccountHeaders 一致。
|
||||
func resolveAndSetOpenAIChatGPTAccountHeaders(ctx context.Context, repo AccountRepository, headers http.Header, account *Account) error {
|
||||
credAccount, err := resolveCredentialAccount(ctx, repo, account)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
setOpenAIChatGPTAccountHeaders(headers, credAccount)
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user