xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-09-17 01:06:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修改服务名，防止返回的模型名为空

Browse Source
This commit is contained in:
RandyZhang 2025-05-20 17:15:10 +08:00
parent 703859214b
commit 3e9f5ad0c7
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
relay/adaptor/vertexai/embedding/adapter.go
View File

@ -23,7 +23,6 @@ var ModelList = []string{
} }
type Adaptor struct { type Adaptor struct {
model string
} }
var _ model2.InnerAIAdapter = (*Adaptor)(nil) var _ model2.InnerAIAdapter = (*Adaptor)(nil)
@ -52,8 +51,7 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.G
if len(inputs) == 0 { if len(inputs) == 0 {
return nil, errors.New("request is nil") return nil, errors.New("request is nil")
} }
modelName, modelTaskType := a.parseEmbeddingTaskType(request.Model) _, modelTaskType := a.parseEmbeddingTaskType(request.Model)
a.model = modelName
instances := make([]EmbeddingInstance, len(inputs)) instances := make([]EmbeddingInstance, len(inputs))
for i, input := range inputs { for i, input := range inputs {
instances[i] = EmbeddingInstance{ instances[i] = EmbeddingInstance{
@ -73,7 +71,11 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.G
} }
func (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) { func (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {
err, usage = EmbeddingHandler(c, a.model, resp) modelName := ""
if meta != nil {
modelName = meta.ActualModelName
}
err, usage = EmbeddingHandler(c, modelName, resp)
return return
} }