xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-12 03:13:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

增加vertex embedding的支持，修改vertex的模型adapter匹配逻辑

Browse Source
This commit is contained in:
RandyZhang
2025-05-10 15:09:05 +08:00
parent 8df4a2670b
commit 209a14c26f
6 changed files with 217 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
relay/adaptor/gemini/main.go
View File

@@ -435,3 +435,17 @@ func EmbeddingHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStat
_, err = c.Writer.Write(jsonResponse)
return nil, &fullTextResponse.Usage
}
func EmbeddingResponseHandler(c *gin.Context, statusCode int, resp *openai.EmbeddingResponse) (*model.ErrorWithStatusCode, *model.Usage) {
jsonResponse, err := json.Marshal(resp)
if err != nil {
return openai.ErrorWrapper(err, "marshal_response_body_failed", http.StatusInternalServerError), nil
}
c.Writer.Header().Set("Content-Type", "application/json")
c.Writer.WriteHeader(statusCode)
_, err = c.Writer.Write(jsonResponse)
if err != nil {
return openai.ErrorWrapper(err, "write_response_body_failed", http.StatusInternalServerError), nil
}
return nil, &resp.Usage
}