
    +jN                        d dl mZ d dlmZ d dlmZmZ  G d dee          Z G d de          Z	 G d d	e          Z
 G d
 de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Zeez  Z G d de          Z G d de          Z G d de          Zd S )!    )Enum)Literal)	BaseModelFieldc                       e Zd ZdZdZdS )AnthropicRoleuser	assistantN)__name__
__module____qualname__r	   r
        >/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/anthropic.pyr   r      s        DIIIr   r   c                   J    e Zd ZU dZed         ed<    ed          Zeed<   dS )AnthropicTextContenttexttype.N	r   r   r   r   r   __annotations__r   r   strr   r   r   r   r      >         "D'&/"""c

D#r   r   c                   r    e Zd ZU dZed         ed<    edd          Zeed<    edd          Z	eed<   d	S )
AnthropicImageSourceBase64base64r   .z2MIME type of the image, e.g. image/png, image/jpegdescription
media_typezBase64-encoded image datadataN)
r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      sg         &D'(
&&&eC-abbbJbbbc'BCCCD#CCCCCr   r   c                   J    e Zd ZU dZed         ed<    ed          Zeed<   dS )AnthropicImageSourceUrlurlr   .N)	r   r   r   r   r   r   r   r"   r   r   r   r   r!   r!      s>          D'%.   uSzzCr   r!   c                   P    e Zd ZU dZed         ed<    ed          Zee	z  ed<   dS )AnthropicImageContentimager   .sourceN)
r   r   r   r   r   r   r   r&   r   r!   r   r   r   r$   r$      sF         $D''
$$$CH5::F&)@@MMMMMr   r$   c                   b    e Zd ZU  ed          Zeed<    ed          Zee	e
z           ed<   dS )AnthropicMessage.rolecontentN)r   r   r   r   r)   r   r   r*   listr   r$   r   r   r   r(   r(   !   sN         %**D-$$$BG%**GT&)>>?LLLLLr   r(   c                   h    e Zd ZU  ed          Zed         ed<    eddd          Zedz  ed<   dS )	AnthropicThinkingConfig.)enableddisabledadaptiver   Ni   zVReasoning budget in tokens. Used when type is 'enabled'. Must be less than max_tokens.)ger   budget_tokens)	r   r   r   r   r   r   r   r2   intr   r   r   r-   r-   &   sg         7<uSzzD'3
4AAA %l! ! !M3:     r   r-   c                   F    e Zd ZU dZ ed          Zed         dz  ed<   dS )AnthropicOutputConfigz=Used with `thinking.type='adaptive'` on models like Opus 4.7.N)lowmediumhigheffort)r   r   r   __doc__r   r9   r   r   r   r   r   r5   r5   .   s;         GG6;eDkkFG+,t3AAAAAr   r5   c                      e Zd ZU  ed          Zeed<    ed          Zee	         ed<    edd          Z
eed<    edd	          Zedz  ed
<    eddd          Zedz  ed<    eddd          Zedz  ed<    edd          Zedz  ed<    ed          Zee         dz  ed<    ed          Zedz  ed<    ed          Zedz  ed<   dS )AnthropicMessagesRequest.modelmessages   )r1   
max_tokensNzTop-level system promptr   systemg        g      ?)r1   letemperaturetop_pr   top_kstop_sequencesthinkingoutput_config)r   r   r   r   r=   r   r   r>   r+   r(   r@   r3   rA   rC   floatrD   rE   rF   rG   r-   rH   r5   r   r   r   r<   r<   3   sU        sE3',uSzzHd#$111eCA&&&J&&&t1JKKKFC$JKKK %dss ; ; ;K;;;%555E54<555dq)))E3:)))',uT{{NDI$222/4uT{{H%,:::27%++M(4/=====r   r<   c                   J    e Zd ZU dZed         ed<    ed          Zeed<   dS )AnthropicResponseTextBlockr   r   .Nr   r   r   r   rK   rK   @   r   r   rK   c                   J    e Zd ZU dZed         ed<    ed          Zeed<   dS )AnthropicResponseThinkingBlockrG   r   .N)	r   r   r   r   r   r   r   rG   r   r   r   r   rM   rM   E   s?          *D'*
***E#JJHcr   rM   c                   \    e Zd ZU  ed          Zedz  ed<    ed          Zedz  ed<   dS )AnthropicCacheCreationUsageNephemeral_5m_input_tokensephemeral_1h_input_tokens)r   r   r   r   rP   r3   r   rQ   r   r   r   rO   rO   M   sL         ,1E$KKsTz777,1E$KKsTz77777r   rO   c                       e Zd ZU  ed          Zedz  ed<    ed          Zedz  ed<    ed          Zedz  ed<    ed          Z	edz  ed<    ed          Z
edz  ed<   dS )AnthropicMessagesUsageNinput_tokensoutput_tokenscache_creation_input_tokenscache_read_input_tokenscache_creation)r   r   r   r   rT   r3   r   rU   rV   rW   rX   rO   r   r   r   rS   rS   R   s         $uT{{L#**** %dM3:+++.3eDkkt999*/%++S4Z5559>tN/$6DDDDDr   rS   c                   L   e Zd ZU  ed          Zedz  ed<    ed          Zedz  ed<    ed          Zedz  ed<    ed          Z	edz  ed<    ed          Z
ee         dz  ed<    ed          Zedz  ed<    ed          Zedz  ed<    ed          Zedz  ed	<   dS )
AnthropicMessagesResponseNidr   r)   r=   r*   stop_reasonstop_sequenceusage)r   r   r   r   r[   r   r   r   r)   r=   r*   r+   AnthropicResponseBlockr\   r]   r^   rS   r   r   r   rZ   rZ   Z   s         U4[[Bd
   uT{{D#*"""uT{{D#*"""dE3:###385;;GT()D0>>>#eDkkKt))) %dM3:++++05;;E!D(66666r   rZ   N)enumr   typingr   pydanticr   r   r   r   r   r   r!   r$   r(   r-   r5   r<   rK   rM   r_   rO   rS   rZ   r   r   r   <module>rc      s               % % % % % % % %    C   
    9   
D D D D D D D D    i   
N N N N NI N N N
M M M M My M M M
    i   B B B B BI B B B

> 
> 
> 
> 
>y 
> 
> 
>       
    Y   
 46TT 8 8 8 8 8) 8 8 8
E E E E EY E E E7 7 7 7 7	 7 7 7 7 7r   