use auth_kwargs instead of auth_token/comfy_api_key

This commit is contained in:
bigcat88 2025-08-16 16:22:52 +03:00
parent cf7a45fb67
commit 563a070cbc
No known key found for this signature in database
GPG Key ID: 1F0BF0EC3CF22721

View File

@ -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,