From bf589917e733dd50a1694be7ad532d9cbcbfc901 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=91=A3=E6=98=AD=E6=B4=8B?= <1241877130@qq.com>
Date: Thu, 23 Apr 2026 12:12:00 +0800
Subject: [PATCH] =?UTF-8?q?Feat:=E5=A2=9E=E5=8A=A0gpt-image-2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 vector_engine_node.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/vector_engine_node.py b/vector_engine_node.py
index 0ae0cdc..2806323 100644
--- a/vector_engine_node.py
+++ b/vector_engine_node.py
@@ -338,12 +338,15 @@ class VectorEngineGPT(VectorEngineBase):
     def INPUT_TYPES(cls):
         return {
             "required": {
+                "model": (["gpt-image-2", "gpt-image-1.5", "gpt-image-1"], {
+                    "default": "gpt-image-2"
+                }),
                 "prompt": ("STRING", {
                     "multiline": True,
                     "default": "A beautiful landscape with mountains and a lake."
                 }),
-                "size": (["1024x1024", "1536x1024", "1024x1536", "auto"], {
-                    "default": "1024x1024"
+                "size": (["auto", "1024x1024", "1536x1024", "1024x1536"], {
+                    "default": "auto"
                 }),
                 "seed": ("INT", {
                     "default": 0,
@@ -365,12 +368,11 @@ class VectorEngineGPT(VectorEngineBase):
     FUNCTION = "generate_image"
     CATEGORY = "VectorEngine"
     
-    def generate_image(self, prompt, size, seed,
+    def generate_image(self, model, prompt, size, seed,
                       image_1=None, image_2=None, image_3=None, image_4=None, image_5=None):
         """
         Generate or edit image using GPT API via Vector Engine
         """
-        model = "gpt-image-1.5"
         
         # Collect input images
         images = [image_1, image_2, image_3, image_4, image_5]