Skip to content

Commit fb83eda

Browse files
authored
Revert "Add support for Veo3 API node." (Comfy-Org#8322)
This reverts commit 592d056.
1 parent 5e5e46d commit fb83eda

File tree

1 file changed

+3
-16
lines changed

1 file changed

+3
-16
lines changed

comfy_api_nodes/nodes_veo2.py

Lines changed: 3 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -54,10 +54,6 @@ class VeoVideoGenerationNode(ComfyNodeABC):
5454
"""
5555
Generates videos from text prompts using Google's Veo API.
5656
57-
Supported models:
58-
- veo-2.0-generate-001
59-
- veo-3.0-generate-preview
60-
6157
This node can create videos from text descriptions and optional image inputs,
6258
with control over parameters like aspect ratio, duration, and more.
6359
"""
@@ -134,14 +130,6 @@ def INPUT_TYPES(s):
134130
"default": None,
135131
"tooltip": "Optional reference image to guide video generation",
136132
}),
137-
"model": (
138-
IO.COMBO,
139-
{
140-
"options": ["veo-2.0-generate-001", "veo-3.0-generate-preview"],
141-
"default": "veo-2.0-generate-001",
142-
"tooltip": "Model to use for video generation. Defaults to veo 2.0",
143-
},
144-
),
145133
},
146134
"hidden": {
147135
"auth_token": "AUTH_TOKEN_COMFY_ORG",
@@ -166,7 +154,6 @@ def generate_video(
166154
person_generation="ALLOW",
167155
seed=0,
168156
image=None,
169-
model="veo-2.0-generate-001",
170157
unique_id: Optional[str] = None,
171158
**kwargs,
172159
):
@@ -205,7 +192,7 @@ def generate_video(
205192
# Initial request to start video generation
206193
initial_operation = SynchronousOperation(
207194
endpoint=ApiEndpoint(
208-
path=f"/proxy/veo/{model}/generate",
195+
path="/proxy/veo/generate",
209196
method=HttpMethod.POST,
210197
request_model=Veo2GenVidRequest,
211198
response_model=Veo2GenVidResponse
@@ -236,7 +223,7 @@ def progress_extractor(response):
236223
# Define the polling operation
237224
poll_operation = PollingOperation(
238225
poll_endpoint=ApiEndpoint(
239-
path=f"/proxy/veo/{model}/poll",
226+
path="/proxy/veo/poll",
240227
method=HttpMethod.POST,
241228
request_model=Veo2GenVidPollRequest,
242229
response_model=Veo2GenVidPollResponse
@@ -317,5 +304,5 @@ def progress_extractor(response):
317304
}
318305

319306
NODE_DISPLAY_NAME_MAPPINGS = {
320-
"VeoVideoGenerationNode": "Google Veo Video Generation",
307+
"VeoVideoGenerationNode": "Google Veo2 Video Generation",
321308
}

0 commit comments

Comments
 (0)