Merge 703859214b into 8df4a2670b

relay/adaptor/vertexai/embedding/adapter.go

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