mirror of
https://wget.la/https://github.com/leookun/cursor-byok
synced 2026-08-17 03:27:02 +08:00
上玄
c50e7d0606
- Introduced a new endpoint option for OpenAI integrations, allowing users to specify a custom path. - Updated relevant components and validation logic to accommodate the new endpoint. - Enhanced documentation and error messages to reflect the addition of the custom endpoint option.
102 lines
3.0 KiB
Go
102 lines
3.0 KiB
Go
package modelchannel
|
||
|
||
import (
|
||
"crypto/sha256"
|
||
"encoding/hex"
|
||
"fmt"
|
||
"net/url"
|
||
"strings"
|
||
)
|
||
|
||
const ChannelIDHexLength = 16
|
||
|
||
const (
|
||
OpenAIEndpointResponses = "/v1/responses"
|
||
OpenAIEndpointChatCompletions = "/v1/chat/completions"
|
||
OpenAIEndpointCustom = "/custom"
|
||
)
|
||
|
||
func NormalizeBaseURL(raw string) (string, error) {
|
||
trimmed := strings.TrimSpace(raw)
|
||
if trimmed == "" {
|
||
return "", fmt.Errorf("模型适配器 baseURL 不能为空")
|
||
}
|
||
parsed, err := url.Parse(trimmed)
|
||
if err != nil {
|
||
return "", fmt.Errorf("模型适配器 baseURL 不是合法 URL")
|
||
}
|
||
if parsed.Scheme != "http" && parsed.Scheme != "https" {
|
||
return "", fmt.Errorf("模型适配器 baseURL 仅支持 http 或 https")
|
||
}
|
||
if strings.TrimSpace(parsed.Host) == "" {
|
||
return "", fmt.Errorf("模型适配器 baseURL 缺少主机名")
|
||
}
|
||
parsed.Scheme = strings.ToLower(strings.TrimSpace(parsed.Scheme))
|
||
parsed.Host = strings.ToLower(strings.TrimSpace(parsed.Host))
|
||
normalized := strings.TrimRight(parsed.String(), "/")
|
||
if normalized == "" {
|
||
normalized = parsed.String()
|
||
}
|
||
return normalized, nil
|
||
}
|
||
|
||
// NormalizeOpenAIEndpoint 归一化 OpenAI endpoint 路径。
|
||
// 支持三个预设值：/v1/responses、/v1/chat/completions、/custom（自定义路径）。
|
||
// 选 /custom 时，用户需在接口地址栏填写完整请求 URL。
|
||
func NormalizeOpenAIEndpoint(providerType string, endpoint string) string {
|
||
if strings.TrimSpace(strings.ToLower(providerType)) != "openai" {
|
||
return ""
|
||
}
|
||
normalized := strings.TrimSpace(endpoint)
|
||
switch normalized {
|
||
case "":
|
||
return OpenAIEndpointResponses
|
||
case OpenAIEndpointResponses, OpenAIEndpointChatCompletions, OpenAIEndpointCustom:
|
||
return normalized
|
||
default:
|
||
return ""
|
||
}
|
||
}
|
||
|
||
// OpenAIEndpointShape 根据 endpoint 路径末段推断协议形态。
|
||
// 返回 "responses"（Responses API）或 "chat/completions"（Chat Completions API）。
|
||
// 这样 /v1/chat/completions、/v4/chat/completions、/chat/completions 都走同一协议分支。
|
||
func OpenAIEndpointShape(endpoint string) string {
|
||
lower := strings.ToLower(strings.TrimSpace(endpoint))
|
||
switch {
|
||
case strings.HasSuffix(lower, "/responses"):
|
||
return "responses"
|
||
default:
|
||
return "chat/completions"
|
||
}
|
||
}
|
||
|
||
func BuildLegacyChannelID(baseURL string, modelID string, apiKey string, name string) string {
|
||
return buildChannelID([]string{
|
||
strings.TrimSpace(baseURL),
|
||
strings.TrimSpace(modelID),
|
||
strings.TrimSpace(apiKey),
|
||
strings.TrimSpace(name),
|
||
})
|
||
}
|
||
|
||
func BuildChannelID(baseURL string, modelID string, apiKey string, name string, openAIEndpoint string) string {
|
||
endpoint := strings.TrimSpace(openAIEndpoint)
|
||
if endpoint == "" {
|
||
return BuildLegacyChannelID(baseURL, modelID, apiKey, name)
|
||
}
|
||
return buildChannelID([]string{
|
||
strings.TrimSpace(baseURL),
|
||
strings.TrimSpace(modelID),
|
||
strings.TrimSpace(apiKey),
|
||
strings.TrimSpace(name),
|
||
endpoint,
|
||
})
|
||
}
|
||
|
||
func buildChannelID(parts []string) string {
|
||
payload := strings.Join(parts, "\n")
|
||
hashBytes := sha256.Sum256([]byte(payload))
|
||
return hex.EncodeToString(hashBytes[:])[:ChannelIDHexLength]
|
||
}
|