
    +jG                     h   S r SSKJr  SSKJrJr   " S S\5      r " S S\5      r " S S	\5      r\\-  \-  r	 " S
 S\5      r
 " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      rg )!zPydantic models for BytePlus ModelArk Responses API.

See: https://docs.byteplus.com/en/docs/ModelArk/1585128 (request)
     https://docs.byteplus.com/en/docs/ModelArk/1783703 (response)
    )Literal)	BaseModelFieldc                   D    \ rS rSr% Sr\S   \S'   \" S5      r\	\S'   Sr
g)BytePlusInputText   
input_texttype.text N__name__
__module____qualname____firstlineno__r
   r   __annotations__r   r   str__static_attributes__r       B/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/bytedance_llm.pyr   r      s    ".D',
.c
D#r   r   c                   Z    \ rS rSr% Sr\S   \S'   \" SSS9r\	\S'   \" SS	S9r
\	\S
'   Srg)BytePlusInputImage   input_imager
   .z0Image URL or `data:image/...;base64,...` payloaddescription	image_urlautozOne of high, low, autodetailr   N)r   r   r   r   r
   r   r   r   r   r   r   r   r   r   r   r   r      s5    #0D'-
 03,^_Is_,DEFCEr   r   c                   b    \ rS rSr% Sr\S   \S'   \" SSS9r\	\S'   \" SS	S
S9r
\S-  \S'   Srg)BytePlusInputVideo   input_videor
   .z0Video URL or `data:video/...;base64,...` payloadr   	video_urlNg?g      @gelefpsr   )r   r   r   r   r
   r   r   r   r$   r   r(   floatr   r   r   r   r!   r!      s:    #0D'-
 03,^_Is_dss3C3r   r!   c                   b    \ rS rSr% Sr\S   \S'   \" SSS9r\	\S'   \" S5      r
\\   \S'   S	rg
)BytePlusInputMessage    messager
   .z)One of user, system, assistant, developerr   rolecontentr   N)r   r   r   r   r
   r   r   r   r.   r   r/   listBytePlusMessageContentr   r   r   r   r+   r+       s7    (D')
(c'RSD#S,1#JGT()6r   r+   c                       \ rS rSr% \" S5      r\\S'   \" S5      r\	\
   \S'   \" S5      r\S-  \S'   \" SSS9r\S-  \S	'   \" SS
SS9r\S-  \S'   \" S5      r\S-  \S'   \" S5      r\S-  \S'   Srg)BytePlusResponseCreateRequest&   .modelinputNinstructions   )r&   max_output_tokensg        g       @r%   temperatureFstorestreamr   )r   r   r   r   r   r5   r   r   r6   r0   r+   r7   r9   intr:   r)   r;   boolr<   r   r   r   r   r3   r3   &   s    sE3(-c
E4$%2$T{L#**$)$1$5sTz5 %dss ;K;uE4$;%,FD4K&r   r3   c                   D    \ rS rSr% Sr\S   \S'   \" S5      r\	\S'   Sr
g)BytePlusOutputText0   output_textr
   .r   r   Nr   r   r   r   r@   r@   0   s    #0D'-
 0c
D#r   r@   c                   D    \ rS rSr% Sr\S   \S'   \" S5      r\	\S'   Sr
g)BytePlusOutputRefusal5   refusalr
   .r   N)r   r   r   r   r
   r   r   r   rF   r   r   r   r   r   rD   rD   5   s    (D')
(:GSr   rD   c                   p    \ rS rSr% \" S5      r\\S'   \" S5      r\S-  \S'   \" S5      r	\S-  \S'   Sr
g)BytePlusOutputContent:   .r
   Nr   rF   r   )r   r   r   r   r   r
   r   r   r   rF   r   r   r   r   rH   rH   :   s4    c
D#T{D#*"+GS4Z%r   rH   c                       \ rS rSr% \" S5      r\\S'   \" S5      r\S-  \S'   \" S5      r	\S-  \S'   \" S5      r
\S-  \S'   \" S5      r\\   S-  \S'   S	rg)
BytePlusOutputMessage@   .r
   Nidr.   statusr/   r   )r   r   r   r   r   r
   r   r   rM   r.   rN   r/   r0   rH   r   r   r   r   rK   rK   @   s_    c
D#4[Bd
 T{D#*"tFC$J$27+GT'(4/=r   rK   c                   6    \ rS rSr% \" S5      r\S-  \S'   Srg)BytePlusInputTokensDetailsH   Ncached_tokensr   )	r   r   r   r   r   rR   r=   r   r   r   r   r   rP   rP   H   s     %dM3:+r   rP   c                   6    \ rS rSr% \" S5      r\S-  \S'   Srg)BytePlusOutputTokensDetailsL   Nreasoning_tokensr   )	r   r   r   r   r   rV   r=   r   r   r   r   r   rT   rT   L   s    #(;cDj.r   rT   c                       \ rS rSr% \" S5      r\S-  \S'   \" S5      r\S-  \S'   \" S5      r	\S-  \S'   \" S5      r
\S-  \S'   \" S5      r\S-  \S'   Srg)	BytePlusResponseUsageP   Ninput_tokensoutput_tokenstotal_tokensinput_tokens_detailsoutput_tokens_detailsr   )r   r   r   r   r   rZ   r=   r   r[   r\   r]   rP   r^   rT   r   r   r   r   rX   rX   P   sb    $T{L#** %dM3:+$T{L#**>CDk4t;I@Ed6=Kr   rX   c                   J    \ rS rSr% \" S5      r\\S'   \" S5      r\\S'   Sr	g)BytePlusResponseErrorX   .coder-   r   N)
r   r   r   r   r   rb   r   r   r-   r   r   r   r   r`   r`   X   s    c
D#:GSr   r`   c                      \ rS rSr% \" S5      r\S-  \S'   \" S5      r\S-  \S'   \" S5      r	\
S-  \S'   \" S5      r\S-  \S'   \" S5      r\S-  \S'   \" S5      r\S-  \S'   \" S5      r\\   S-  \S	'   \" S5      r\S-  \S
'   Srg)BytePlusResponseObject]   NrM   object
created_atr5   rN   erroroutputusager   )r   r   r   r   r   rM   r   r   rf   rg   r=   r5   rN   rh   r`   ri   r0   rK   rj   rX   r   r   r   r   rd   rd   ]   s    4[Bd
 tFC$J$"4[Jd
(dE3:#tFC$J$*/+E 4'516tFD&'$.<*/+E 4'5r   rd   N)__doc__typingr   pydanticr   r   r   r   r!   r1   r+   r3   r@   rD   rH   rK   rP   rT   rX   r`   rd   r   r   r   <module>rn      s     %	 
F F4 4 +-??BTT 79 7'I ' 
I 
&I &>I >, ,/) /LI LI 
6Y 6r   