chore: fix ali image implementation

This commit is contained in:
JustSong
2024-04-05 18:25:57 +08:00
parent d4347e7a35
commit eec41849ec
5 changed files with 197 additions and 179 deletions

View File

@@ -92,7 +92,7 @@ func (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *util.Rel
} else {
switch meta.Mode {
case constant.RelayModeImagesGenerations:
err, usage = ImageHandler(c, resp)
err, _ = ImageHandler(c, resp)
default:
err, usage = Handler(c, resp, meta.PromptTokens, meta.ActualModelName)
}

View File

@@ -181,5 +181,5 @@ func ImageHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCo
if err != nil {
return ErrorWrapper(err, "close_response_body_failed", http.StatusInternalServerError), nil
}
return nil, &imageResponse.Usage
return nil, nil
}

View File

@@ -117,9 +117,9 @@ type ImageData struct {
}
type ImageResponse struct {
Created int `json:"created"`
Data []ImageData `json:"data"`
model.Usage `json:"usage"`
Created int64 `json:"created"`
Data []ImageData `json:"data"`
//model.Usage `json:"usage"`
}
type ChatCompletionsStreamResponseChoice struct {