From c5f1a0c7129d7e474fec9c65312c83a679c7ab8c Mon Sep 17 00:00:00 2001
From: datehoer <datehoer@gmail.com>
Date: Wed, 9 Apr 2025 15:05:00 +0800
Subject: [PATCH] Add support for grok-2-image. Currently, grok-2-image doesn't
 support the size, quality, or style parameters. Set 'size'='empty' to use
 grok-2-image

---
 relay/relay-image.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/relay/relay-image.go b/relay/relay-image.go
index f9f542a7..5e9929a0 100644
--- a/relay/relay-image.go
+++ b/relay/relay-image.go
@@ -5,7 +5,6 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
-	"github.com/gin-gonic/gin"
 	"io"
 	"net/http"
 	"one-api/common"
@@ -16,6 +15,8 @@ import (
 	"one-api/service"
 	"one-api/setting"
 	"strings"
+
+	"github.com/gin-gonic/gin"
 )
 
 func getAndValidImageRequest(c *gin.Context, info *relaycommon.RelayInfo) (*dto.ImageRequest, error) {
@@ -39,6 +40,10 @@ func getAndValidImageRequest(c *gin.Context, info *relaycommon.RelayInfo) (*dto.
 	if imageRequest.Model == "" {
 		imageRequest.Model = "dall-e-2"
 	}
+	// x.ai grok-2-image not support size, quality or style
+	if imageRequest.Size == "empty" {
+		imageRequest.Size = ""
+	}
 
 	// Not "256x256", "512x512", or "1024x1024"
 	if imageRequest.Model == "dall-e-2" || imageRequest.Model == "dall-e" {