63 lines
2.5 KiB
Go
63 lines
2.5 KiB
Go
package handler
|
|
|
|
import (
|
|
"github.com/qianfree/team-api/relay/common"
|
|
"github.com/qianfree/team-api/relay/constant"
|
|
"github.com/qianfree/team-api/relay/helper"
|
|
)
|
|
|
|
// canPassThrough 判断当前请求是否可以直连转发（跳过协议转换和参数改写）
|
|
func canPassThrough(info *common.RelayInfo) bool {
|
|
settings := info.ChannelMeta.Settings
|
|
|
|
// 阿里云同步 multimodal 图片（qwen-image-2.x）：入站 OpenAI Images 格式与上游 DashScope
|
|
// multimodal-generation 的 messages 结构不兼容，必须经 ConvertRequest 转换。即使运营者
|
|
// 显式开启直连、或未配置模型映射，也不能原样透传，否则上游因请求体格式错误报错。
|
|
if constant.RelayMode(info.RelayMode) == constant.RelayModeImagesGenerations &&
|
|
constant.IsAliSyncMultimodalImageModel(info.ChannelMeta.UpstreamModelName) {
|
|
return false
|
|
}
|
|
|
|
// 显式开启：运营者明确配置直连，不做额外检查
|
|
if settings.PassThroughBodyEnabled {
|
|
return true
|
|
}
|
|
|
|
// 自动检测：入站格式必须匹配渠道原生格式才可原样直连转发。
|
|
// 多协议原生透传渠道（New API / Sub2API）的上游同时支持 OpenAI/Claude/Gemini，三者均匹配。
|
|
if !inboundMatchesChannelNative(info) {
|
|
return false
|
|
}
|
|
// 需要模型名映射 → 必须经过转换来替换模型名
|
|
if info.ChannelMeta.IsModelMapped {
|
|
return false
|
|
}
|
|
// 有参数改写规则 → 必须经过转换来应用改写
|
|
if settings.ParamOverride != nil {
|
|
return false
|
|
}
|
|
// 有系统提示词注入 → 必须经过转换
|
|
if settings.SystemPrompt != "" {
|
|
return false
|
|
}
|
|
// 有 thinking 后缀 → 必须经过转换来注入 thinking 参数
|
|
if info.ThinkingEnabled || info.ThinkingDisabled || info.ReasoningEffort != "" {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
// inboundMatchesChannelNative 判断入站格式是否与渠道原生格式匹配（匹配则可原样直连转发）。
|
|
// - 普通渠道：入站格式须等于该渠道的唯一原生格式（helper.ProviderNativeFormat）。
|
|
// - 多协议原生透传渠道（New API / Sub2API）：OpenAI/Claude/Gemini 三种格式均视为匹配。
|
|
func inboundMatchesChannelNative(info *common.RelayInfo) bool {
|
|
if constant.IsMultiNativeProvider(info.ChannelMeta.ChannelType) {
|
|
switch info.InboundFormat {
|
|
case constant.RelayFormatOpenAI, constant.RelayFormatClaude, constant.RelayFormatGemini:
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
return helper.ProviderNativeFormat(info.ChannelMeta.ChannelType) == info.InboundFormat
|
|
}
|