o
    iV                     @  s   d dl mZ d dlmZ d dlmZmZ d dlmZm	Z	 G dd deZ
G dd deZG d	d
 d
eZG dd deeZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZdS )    )annotations)Enum)OptionalList)	BaseModelFieldc                   @  s   e Zd ZU edddZded< edddZded< edd	dZded
< edddZded< edddZ	ded< edddZ
ded< dS )Rodin3DGenerateRequest.seed_descriptionintseedzTier of generation.strtierzThe material type.materialzThe poly count of the mesh.quality_overridez2It controls the type of faces of generated models.	mesh_modeN zOptional[bool]TAPose)__name__
__module____qualname__r   r   __annotations__r   r   r   r   r    r   r   8/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/apis/rodin.pyr      s   
 r   c                   @  s6   e Zd ZU edddZded< edddZded< d	S )
GenerateJobsData.zstr LISTr
   z	List[str]uuidszsubscription keyr   subscription_keyN)r   r   r   r   r   r   r   r   r   r   r   r         
 r   c                   @  sr   e Zd ZU edddZded< edddZded< edddZded	< edd
dZded< edddZ	ded< dS )Rodin3DGenerateResponseNzReturn message.r
   zOptional[str]messagezGenerated Prompt from image.promptzSubmit Timesubmit_timeTask struuidzDetails of jobszOptional[GenerateJobsData]jobs)
r   r   r   r   r    r   r!   r"   r$   r%   r   r   r   r   r      s   
 r   c                   @  s    e Zd ZdZdZdZdZdZdS )	JobStatusz
    Status for jobs
    DoneFailed
GeneratingWaitingN)r   r   r   __doc__r'   r(   r)   r*   r   r   r   r   r&      s    r&   c                   @  "   e Zd ZU edddZded< dS )Rodin3DCheckStatusRequest.z#subscription from generate endpointr
   r   r   N)r   r   r   r   r   r   r   r   r   r   r-   $      
 r-   c                   @  s6   e Zd ZU edddZded< edddZded< dS )	JobItem.r$   r
   r   zStatus Currentlyr&   statusN)r   r   r   r   r$   r   r0   r   r   r   r   r/   '   r   r/   c                   @  r,   )Rodin3DCheckStatusResponse.zJob status Listr
   zList[JobItem]r%   N)r   r   r   r   r%   r   r   r   r   r   r1   +   r.   r1   c                   @  r,   )Rodin3DDownloadRequest.r#   r
   r   	task_uuidN)r   r   r   r   r3   r   r   r   r   r   r2   .   r.   r2   c                   @  s6   e Zd ZU edddZded< edddZded< dS )	RodinResourceItem.zDownload Urlr
   r   urlzFile name with extnameN)r   r   r   r   r5   r   r6   r   r   r   r   r4   1   r   r4   c                   @  r,   )Rodin3DDownloadResponse.zSource Listr
   zList[RodinResourceItem]listN)r   r   r   r   r8   r   r   r   r   r   r7   5   r.   r7   N)
__future__r   enumr   typingr   r   pydanticr   r   r   r   r   r   r&   r-   r/   r1   r2   r4   r7   r   r   r   r   <module>   s    	