Seefs
57746fc972
* fix(i18n): clarify Go regex and field passthrough copy * feat(channel): support Codex upstream model discovery * Revert "fix(i18n): clarify Go regex and field passthrough copy" This reverts commit d63d7975db3e34ff44e189112d3c15ad8c24ad88.
91 lines
2.6 KiB
Go
91 lines
2.6 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/QuantumNous/new-api/constant"
|
|
"github.com/QuantumNous/new-api/model"
|
|
"github.com/QuantumNous/new-api/setting/ratio_setting"
|
|
)
|
|
|
|
func FetchCodexChannelModels(channel *model.Channel) ([]string, error) {
|
|
if channel == nil || channel.Type != constant.ChannelTypeCodex {
|
|
return nil, fmt.Errorf("channel type is not Codex")
|
|
}
|
|
if channel.ChannelInfo.IsMultiKey {
|
|
return nil, fmt.Errorf("codex channel does not support multi-key model discovery")
|
|
}
|
|
|
|
client, err := NewProxyHttpClient(channel.GetSetting().Proxy)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
|
|
defer cancel()
|
|
|
|
clientVersion, err := GetLatestCodexClientVersion(ctx, client)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get Codex client version: %w", err)
|
|
}
|
|
|
|
baseURL := channel.GetBaseURL()
|
|
if baseURL == "" {
|
|
baseURL = constant.ChannelBaseURLs[constant.ChannelTypeCodex]
|
|
}
|
|
return fetchCodexChannelModels(ctx, channel, baseURL, client, clientVersion)
|
|
}
|
|
|
|
func fetchCodexChannelModels(
|
|
ctx context.Context,
|
|
channel *model.Channel,
|
|
baseURL string,
|
|
client *http.Client,
|
|
clientVersion string,
|
|
) ([]string, error) {
|
|
oauthKey, err := parseCodexOAuthKey(strings.TrimSpace(channel.Key))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
statusCode, models, err := FetchCodexModels(ctx, client, baseURL, oauthKey, clientVersion)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if statusCode == http.StatusUnauthorized {
|
|
if channel.Id <= 0 {
|
|
return nil, fmt.Errorf("codex channel credential expired; save the channel before retrying model fetch")
|
|
}
|
|
refreshedKey, _, refreshErr := RefreshCodexChannelCredential(
|
|
ctx,
|
|
channel.Id,
|
|
CodexCredentialRefreshOptions{ResetCaches: true},
|
|
)
|
|
if refreshErr != nil {
|
|
return nil, fmt.Errorf("failed to refresh Codex channel credential: %w", refreshErr)
|
|
}
|
|
statusCode, models, err = FetchCodexModels(ctx, client, baseURL, &CodexOAuthKey{
|
|
AccessToken: refreshedKey.AccessToken,
|
|
AccountID: refreshedKey.AccountID,
|
|
}, clientVersion)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
if statusCode < http.StatusOK || statusCode >= http.StatusMultipleChoices {
|
|
return nil, fmt.Errorf("upstream status: %d", statusCode)
|
|
}
|
|
modelVariants := make([]string, 0, len(models)*2)
|
|
modelVariants = append(modelVariants, models...)
|
|
for _, modelName := range models {
|
|
if modelName == "codex-auto-review" {
|
|
continue
|
|
}
|
|
modelVariants = append(modelVariants, ratio_setting.WithCompactModelSuffix(modelName))
|
|
}
|
|
return modelVariants, nil
|
|
}
|