mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-27 21:56:39 +08:00
d6536d2907
The GetAdaptor function in the Adaptor struct has been updated to use the actual model name instead of the origin model name. This change ensures that the correct adaptor is retrieved for processing the response.
51 lines
1.2 KiB
Go
51 lines
1.2 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]
|
|
switch adaptorType {
|
|
case VerterAIClaude:
|
|
return &claude.Adaptor{}
|
|
case VerterAIGemini:
|
|
return &gemini.Adaptor{}
|
|
default:
|
|
return nil
|
|
}
|
|
}
|