mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-08-23 05:37:54 +08:00
use auth_kwargs instead of auth_token/comfy_api_key
This commit is contained in:
parent
cf7a45fb67
commit
563a070cbc
@ -157,7 +157,6 @@ class VeoVideoGenerationNode(comfy_io.ComfyNode):
|
|||||||
image=None,
|
image=None,
|
||||||
model="veo-2.0-generate-001",
|
model="veo-2.0-generate-001",
|
||||||
generate_audio=False,
|
generate_audio=False,
|
||||||
**kwargs,
|
|
||||||
):
|
):
|
||||||
# Prepare the instances for the request
|
# Prepare the instances for the request
|
||||||
instances = []
|
instances = []
|
||||||
@ -194,6 +193,10 @@ class VeoVideoGenerationNode(comfy_io.ComfyNode):
|
|||||||
if "veo-3.0" in model:
|
if "veo-3.0" in model:
|
||||||
parameters["generateAudio"] = generate_audio
|
parameters["generateAudio"] = generate_audio
|
||||||
|
|
||||||
|
auth = {
|
||||||
|
"auth_token": cls.hidden.auth_token_comfy_org,
|
||||||
|
"comfy_api_key": cls.hidden.api_key_comfy_org,
|
||||||
|
}
|
||||||
# Initial request to start video generation
|
# Initial request to start video generation
|
||||||
initial_operation = SynchronousOperation(
|
initial_operation = SynchronousOperation(
|
||||||
endpoint=ApiEndpoint(
|
endpoint=ApiEndpoint(
|
||||||
@ -206,8 +209,7 @@ class VeoVideoGenerationNode(comfy_io.ComfyNode):
|
|||||||
instances=instances,
|
instances=instances,
|
||||||
parameters=parameters
|
parameters=parameters
|
||||||
),
|
),
|
||||||
auth_token=cls.hidden.auth_token_comfy_org,
|
auth_kwargs=auth,
|
||||||
comfy_api_key=cls.hidden.api_key_comfy_org,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
initial_response = await initial_operation.execute()
|
initial_response = await initial_operation.execute()
|
||||||
@ -241,8 +243,7 @@ class VeoVideoGenerationNode(comfy_io.ComfyNode):
|
|||||||
request=VeoGenVidPollRequest(
|
request=VeoGenVidPollRequest(
|
||||||
operationName=operation_name
|
operationName=operation_name
|
||||||
),
|
),
|
||||||
auth_token=cls.hidden.auth_token_comfy_org,
|
auth_kwargs=auth,
|
||||||
comfy_api_key=cls.hidden.api_key_comfy_org,
|
|
||||||
poll_interval=5.0,
|
poll_interval=5.0,
|
||||||
result_url_extractor=get_video_url_from_response,
|
result_url_extractor=get_video_url_from_response,
|
||||||
node_id=cls.hidden.unique_id,
|
node_id=cls.hidden.unique_id,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user