Feat:增加gpt-image-2

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]