hjjjj
c734c541b2
refactor: 重构Gemini适配器以支持多模态输入 fix: 修复React Hooks依赖警告 style: 清理未使用的导入和代码 docs: 更新用户界面文本和提示 perf: 优化图像和视频URL处理性能 test: 添加数据迁移工具和测试 build: 更新依赖项和.gitignore chore: 同步Zenmux模型和价格比例
65 lines
1.8 KiB
Go
65 lines
1.8 KiB
Go
package vertexai
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
claude "github.com/songquanpeng/one-api/relay/adaptor/vertexai/claude"
|
|
gemini "github.com/songquanpeng/one-api/relay/adaptor/vertexai/gemini"
|
|
"github.com/songquanpeng/one-api/relay/meta"
|
|
"github.com/songquanpeng/one-api/relay/model"
|
|
)
|
|
|
|
type VertexAIModelType int
|
|
|
|
const (
|
|
VerterAIClaude VertexAIModelType = iota + 1
|
|
VerterAIGemini
|
|
)
|
|
|
|
var modelMapping = map[string]VertexAIModelType{}
|
|
var modelList = []string{}
|
|
|
|
func init() {
|
|
modelList = append(modelList, claude.ModelList...)
|
|
for _, model := range claude.ModelList {
|
|
modelMapping[model] = VerterAIClaude
|
|
}
|
|
|
|
modelList = append(modelList, gemini.ModelList...)
|
|
for _, model := range gemini.ModelList {
|
|
modelMapping[model] = VerterAIGemini
|
|
}
|
|
}
|
|
|
|
type innerAIAdapter interface {
|
|
ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error)
|
|
DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode)
|
|
}
|
|
|
|
func GetAdaptor(model string) innerAIAdapter {
|
|
adaptorType := modelMapping[model]
|
|
// Handle "google/" prefixed model names (e.g. "google/gemini-2.5-flash-image")
|
|
// by stripping the prefix and looking up the bare model name.
|
|
if adaptorType == 0 {
|
|
bare := model
|
|
if idx := len("google/"); len(model) > idx && model[:idx] == "google/" {
|
|
bare = model[idx:]
|
|
}
|
|
adaptorType = modelMapping[bare]
|
|
}
|
|
// If still not found, default to Gemini adaptor for any unrecognized model
|
|
// (Zenmux proxies all Gemini variants through the same generateContent endpoint).
|
|
if adaptorType == 0 {
|
|
adaptorType = VerterAIGemini
|
|
}
|
|
switch adaptorType {
|
|
case VerterAIClaude:
|
|
return &claude.Adaptor{}
|
|
case VerterAIGemini:
|
|
return &gemini.Adaptor{}
|
|
default:
|
|
return nil
|
|
}
|
|
}
|