fix: ali embedding support base64
This commit is contained in:
@@ -48,6 +48,19 @@ type OpenAIEmbeddingResponse struct {
|
|||||||
Usage `json:"usage"`
|
Usage `json:"usage"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type FlexibleEmbeddingResponseItem struct {
|
||||||
|
Object string `json:"object"`
|
||||||
|
Index int `json:"index"`
|
||||||
|
Embedding any `json:"embedding"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type FlexibleEmbeddingResponse struct {
|
||||||
|
Object string `json:"object"`
|
||||||
|
Data []FlexibleEmbeddingResponseItem `json:"data"`
|
||||||
|
Model string `json:"model"`
|
||||||
|
Usage `json:"usage"`
|
||||||
|
}
|
||||||
|
|
||||||
type ChatCompletionsStreamResponseChoice struct {
|
type ChatCompletionsStreamResponseChoice struct {
|
||||||
Delta ChatCompletionsStreamResponseChoiceDelta `json:"delta,omitempty"`
|
Delta ChatCompletionsStreamResponseChoiceDelta `json:"delta,omitempty"`
|
||||||
Logprobs *any `json:"logprobs"`
|
Logprobs *any `json:"logprobs"`
|
||||||
|
|||||||
@@ -40,7 +40,7 @@ func embeddingRequestOpenAI2Ali(request dto.EmbeddingRequest) *AliEmbeddingReque
|
|||||||
}
|
}
|
||||||
|
|
||||||
func aliEmbeddingHandler(c *gin.Context, resp *http.Response) (*types.NewAPIError, *dto.Usage) {
|
func aliEmbeddingHandler(c *gin.Context, resp *http.Response) (*types.NewAPIError, *dto.Usage) {
|
||||||
var fullTextResponse dto.OpenAIEmbeddingResponse
|
var fullTextResponse dto.FlexibleEmbeddingResponse
|
||||||
err := json.NewDecoder(resp.Body).Decode(&fullTextResponse)
|
err := json.NewDecoder(resp.Body).Decode(&fullTextResponse)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return types.NewError(err, types.ErrorCodeBadResponseBody), nil
|
return types.NewError(err, types.ErrorCodeBadResponseBody), nil
|
||||||
|
|||||||
Reference in New Issue
Block a user