tests+api: use smaller model gpt-image-1-mini instead of dall-e-2; update requests (size=1024x1024, quality=low)

Signed-off-by: Hrushikesh Patil <[email protected]>

Author: nutanix-Hrushikesh

internal/apischema/openai/openai.go

@@ -58,10 +58,9 @@ const (
 	// ModelTextEmbedding3Small is the cheapest model usable with /embeddings.
 	ModelTextEmbedding3Small = "text-embedding-3-small"
 
-	// ModelDALLE2 is the DALL-E 2 model usable with /v1/images/generations.
-	ModelDALLE2 = "dall-e-2"
-	// ModelDALLE3 is the DALL-E 3 model usable with /v1/images/generations.
-	ModelDALLE3 = "dall-e-3"
+	// ModelGPTImage1Mini is the smallest/cheapest Images model usable with
+	// /v1/images/generations. Use with size "1024x1024" and quality "low".
+	ModelGPTImage1Mini = "gpt-image-1-mini"
 )
 
 // ChatCompletionContentPartRefusalType The type of the content part.

internal/extproc/imagegeneration_processor_test.go

@@ -343,7 +343,7 @@ func Test_imageGenerationProcessorUpstreamFilter_SetBackend(t *testing.T) {
 	rp := &imageGenerationProcessorRouterFilter{originalRequestBody: &openaisdk.ImageGenerateParams{}}
 	p2 := &imageGenerationProcessorUpstreamFilter{
 		config:         &processorConfig{},
-		requestHeaders: map[string]string{internalapi.ModelNameHeaderKeyDefault: "dall-e-2"},
+		requestHeaders: map[string]string{internalapi.ModelNameHeaderKeyDefault: "gpt-image-1-mini"},
 		logger:         slog.Default(),
 		metrics:        &mockImageGenerationMetrics{},
 	}
@@ -358,12 +358,12 @@ func Test_imageGenerationProcessorUpstreamFilter_SetBackend(t *testing.T) {
 
 func TestImageGeneration_ParseBody(t *testing.T) {
 	t.Run("ok", func(t *testing.T) {
-		jsonBody := `{"model":"dall-e-2","prompt":"a cat"}`
+		jsonBody := `{"model":"gpt-image-1-mini","prompt":"a cat","size":"1024x1024","quality":"low"}`
 		modelName, rb, err := parseOpenAIImageGenerationBody(&extprocv3.HttpBody{Body: []byte(jsonBody)})
 		require.NoError(t, err)
-		require.Equal(t, "dall-e-2", modelName)
+		require.Equal(t, "gpt-image-1-mini", modelName)
 		require.NotNil(t, rb)
-		require.Equal(t, "dall-e-2", rb.Model)
+		require.Equal(t, "gpt-image-1-mini", rb.Model)
 		require.Equal(t, "a cat", rb.Prompt)
 	})
 	t.Run("error", func(t *testing.T) {

tests/extproc/testupstream_test.go

@@ -142,11 +142,11 @@ func TestWithTestUpstream(t *testing.T) {
 			backend:         "openai",
 			path:            "/v1/images/generations",
 			method:          http.MethodPost,
-			requestBody:     `{"model":"dall-e-2","prompt":"a cat wearing sunglasses"}`,
+			requestBody:     `{"model":"gpt-image-1-mini","prompt":"a cat wearing sunglasses","size":"1024x1024","quality":"low"}`,
 			expPath:         "/v1/images/generations",
-			responseBody:    `{"created":1736890000,"data":[{"url":"https://example.com/image1.png"}],"model":"dall-e-2","usage":{"prompt_tokens":0,"completion_tokens":0,"total_tokens":0}}`,
+			responseBody:    `{"created":1736890000,"data":[{"url":"https://example.com/image1.png"}],"model":"gpt-image-1-mini","usage":{"prompt_tokens":0,"completion_tokens":0,"total_tokens":0}}`,
 			expStatus:       http.StatusOK,
-			expResponseBody: `{"created":1736890000,"data":[{"url":"https://example.com/image1.png"}],"model":"dall-e-2","usage":{"prompt_tokens":0,"completion_tokens":0,"total_tokens":0}}`,
+			expResponseBody: `{"created":1736890000,"data":[{"url":"https://example.com/image1.png"}],"model":"gpt-image-1-mini","usage":{"prompt_tokens":0,"completion_tokens":0,"total_tokens":0}}`,
 		},
 		{
 			name:            "openai - /v1/images/generations - non json upstream error mapped to OpenAI",

tests/internal/testopenai/image_requests.go

@@ -16,12 +16,16 @@ type imageGenerationRequest struct {
 	Model  string `json:"model"`
 	Prompt string `json:"prompt"`
 	// Optional fields like size/quality/response_format can be added later if needed.
+	Size    string `json:"size,omitempty"`
+	Quality string `json:"quality,omitempty"`
 }
 
 // imageRequests contains the actual request body for each image generation cassette.
 var imageRequests = map[Cassette]*imageGenerationRequest{
 	CassetteImageGenerationBasic: {
-		Model:  "dall-e-2",
-		Prompt: "A simple black-and-white line drawing of a cat playing with yarn",
+		Model:   "gpt-image-1-mini",
+		Prompt:  "A simple black-and-white line drawing of a cat playing with yarn",
+		Size:    "1024x1024",
+		Quality: "low",
 	},
 }