Commit 47ddf76a by creamlike1024

fix: Guard against negative or zero n from ExtraBody to prevent uint underflow

parent 8a1f6534
...@@ -215,7 +215,7 @@ func (a *Adaptor) ConvertOpenAIRequest(c *gin.Context, info *relaycommon.RelayIn ...@@ -215,7 +215,7 @@ func (a *Adaptor) ConvertOpenAIRequest(c *gin.Context, info *relaycommon.RelayIn
if len(request.ExtraBody) > 0 { if len(request.ExtraBody) > 0 {
var extra map[string]any var extra map[string]any
if err := json.Unmarshal(request.ExtraBody, &extra); err == nil { if err := json.Unmarshal(request.ExtraBody, &extra); err == nil {
if n, ok := extra["n"].(float64); ok { if n, ok := extra["n"].(float64); ok && n > 0 {
imgReq.N = uint(n) imgReq.N = uint(n)
} }
if size, ok := extra["size"].(string); ok { if size, ok := extra["size"].(string); ok {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment