Quellcode durchsuchen

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

creamlike1024 vor 6 Monaten
Ursprung
Commit
e02e1e8d4a
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      relay/channel/vertex/adaptor.go

+ 1 - 1
relay/channel/vertex/adaptor.go

@@ -215,7 +215,7 @@ func (a *Adaptor) ConvertOpenAIRequest(c *gin.Context, info *relaycommon.RelayIn
 		if len(request.ExtraBody) > 0 {
 			var extra map[string]any
 			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)
 				}
 				if size, ok := extra["size"].(string); ok {