o
    i                     @  sX   d dl mZ d dlmZ d dlmZmZ G dd dejZG dd deZ	dd	d
Z
dS )    )annotations)override)ComfyExtensionioc                   @  s&   e Zd Zedd Zed
ddZd	S )
CreateListc                 C  sb   t jd}t jjt jjd|ddd}t jddddg d	t jjd
|dgt jj|dddgdS )Ntypeinput)template)r   prefixr   zCreate ListlogicT)zImage IteratorzText IteratorIteratorinputslist)r	   is_output_listdisplay_name)node_idr   categoryis_input_listsearch_aliasesr   outputs)r   	MatchTypeTemplateAutogrowTemplatePrefixInputSchemaOutput)clstemplate_matchtypetemplate_autogrow r    8/mnt/c/Users/fbmor/ComfyUI/comfy_extras/nodes_toolkit.pydefine_schema   s&   zCreateList.define_schemar   io.Autogrow.Typereturnio.NodeOutputc                 C  s$   g }|  D ]}||7 }qt|S N)valuesr   
NodeOutput)r   r   output_listr   r    r    r!   execute   s   

zCreateList.executeN)r   r#   r$   r%   )__name__
__module____qualname__classmethodr"   r*   r    r    r    r!   r      s
    
r   c                   @  s   e Zd ZedddZdS )ToolkitExtensionr$   list[type[io.ComfyNode]]c                   s   t gS r&   )r   )selfr    r    r!   get_node_list'   s   zToolkitExtension.get_node_listN)r$   r0   )r+   r,   r-   r   r2   r    r    r    r!   r/   &   s    r/   r$   c                     s   t  S r&   )r/   r    r    r    r!   comfy_entrypoint.   s   r3   N)r$   r/   )
__future__r   typing_extensionsr   comfy_api.latestr   r   	ComfyNoder   r/   r3   r    r    r    r!   <module>   s     