+j-UdZddlmZddlmZmZmZddlmZm Z m Z m Z m Z m Z mZmZmZmZddlmZmZmZmZmZdZdZdZd d d d d dZeeefed<dhZhdZ ddddZ!eee"fed<gdZ#defdZ$dede%e&e&fdzfdZ'de de&dzfdZ(de defd Z)d!e*ej+d"e,ej-de,efd#Z.Gd$d%ej+Z/Gd&d'eZ0de0fd(Z1dS))z^API Nodes for Anthropic Claude (Messages API). See: https://docs.anthropic.com/en/api/messages)override)IOComfyExtensionInput) AnthropicImageContentAnthropicImageSourceUrlAnthropicMessageAnthropicMessagesRequestAnthropicMessagesResponseAnthropicOutputConfigAnthropicResponseTextBlock AnthropicRoleAnthropicTextContentAnthropicThinkingConfig) ApiEndpointget_number_of_imagessync_opupload_images_to_comfyapivalidate_stringz/proxy/anthropic/v1/messagesi%zclaude-opus-4-7zclaude-opus-4-6zclaude-sonnet-4-6zclaude-sonnet-4-5-20250929zclaude-haiku-4-5-20251001)Opus 4.7Opus 4.6 Sonnet 4.6z Sonnet 4.5 Haiku 4.5 CLAUDE_MODELSr>rrrii i@)lowmediumhigh_REASONING_BUDGET)offrrr model_labelc &tjddddddtjdd d d d d d g}|tvr<|tjdtddd|S)N max_tokensiizNMaximum number of tokens to generate (includes reasoning tokens when enabled).T)defaultminmaxtooltipadvanced temperature?gg{Gz?zControls randomness. 0.0 is deterministic, 1.0 is most random. Ignored for Opus 4.7 and any model when reasoning_effort is set.)r%r&r'stepr(r)reasoning_effortr z3Extended thinking effort. 'off' disables reasoning.)optionsr%r(r))rIntrFloat_THINKING_UNSUPPORTEDappendCombo_REASONING_EFFORTS)r!inputss ?/home/wildlama/comfy/ComfyUI/comfy_api_nodes/nodes_anthropic.py_claude_model_inputsr75s  d     S  F,/// HNN"*M       MmodelreturnNc:d|vsd|vsd|vrdSd|vrdSd|vrdSd S) zPReturn (input_per_1M, output_per_1M) USD for a Claude model, or None if unknown.zopus-4-7zopus-4-6zopus-4-5)@g9@zsonnet-4)g@g.@z haiku-4-5)r+r<N)r9s r6_model_price_per_millionr>YsKUjE11Z55H5HyUyex 4r8responsecr|jr|jsdSt|j}|dS|\}}|jjpd}|jjpd}|jjpd}d}d}|jjr&|jjjpd}|jjjpd}||z||zz||zdzz||zdzz||zdzz} | dz S)zXCompute approximate USD price from response usage. Server-side billing is authoritative.Nrg?g?g@g.A) usager9r> input_tokens output_tokenscache_read_input_tokenscache_creationephemeral_5m_input_tokensephemeral_1h_input_tokens) r?rates input_rate output_raterBrC cache_readcache_5mcache_1htotals r6calculate_tokens_pricerOds >t $X^ 4 4E }t#J >.3!LN05AM7<1JHH~$P>0JOa>0JOaz! + % & z !C ' ( Z $ & ' Z # %  &  ; r8c\|jsdSdd|jDS)N c3\K|]'}t|t|j|jV(dSN) isinstancer text).0blocks r6 z*_get_text_from_response..sS e7 8 8=BZ r8)contentjoin)r?s r6_get_text_from_responser\~sG  r 99 ( 0  r8cls image_tensorscfKt||ttdd{V}d|DS)NzUploading reference images) max_images total_pixels wait_labelcJg|] }tt|!S))url)source)rr)rWrds r6 z/_build_image_content_blocks..s0 [ [ [s !)@S)I)I)I J J J [ [ [r8)rCLAUDE_MAX_IMAGESANTHROPIC_IMAGE_MAX_PIXELS)r]r^urlss r6_build_image_content_blocksrjsa+ $//       D \ [VZ [ [ [[r8cneZdZdZedZe d dededededzd ed e j f d Z dS) ClaudeNodez7Generate text responses from an Anthropic Claude model.cjtjdddddtjdddd tjd d t Dd tjddddddtjdtjtj ddtdtdzDddtdtjddddddgtj gtj jtj jtj jgdtjtjd gd !" S)#NrlzAnthropic Claudezpartner/text/AnthropiczText GenerationzGenerate text responses with Anthropic's Claude models. Provide a text prompt and optionally one or more images for multimodal context.promptTrQzText input to the model.) multiliner%r(r9chg|]/}tj|t|0Sr=)r DynamicComboOptionr7)rWlabels r6rfz,ClaudeNode.define_schema..sC!..u6J56Q6QRRr8z/The Claude model used to generate the response.)r.r(seedriz_Seed controls whether the node should re-run; results are non-deterministic regardless of seed.)r%r&r'control_after_generater(imagesimagecg|]}d|S)image_r=)rWis r6rfz,ClaudeNode.define_schema..sUUU|||UUUr8)namesr&z9Optional image(s) to use as context for the model. Up to z images.)templater( system_promptzORS>S5T5TUUU7 tXisss" #"!!Z  C)TY%%''( . + #  / BBBqP P P P r8NrQrnr9rtrvr~r:cLKt|dd|d}|dd}|dd}|d vo|tv} | s|d krd} n|d d } d} d} | ri|tvr!t d } t |} n?t |} t| td|dz } t d| } d|pi D}td|Dtkrtdtdg}|r)| t||d{V|t!|t#|t%t&dt(t+t,||t/t0j|g|pd| | | t4d{V}t7jt;|pdS)NTr{)strip_whitespace min_lengthr9r#r$r-r )r Nrr*r+adaptive)type)effortienabled)r budget_tokenscg|]}||SrTr=rWts r6rfz&ClaudeNode.execute..s+`+`+`!RSR_AR_R_R_r8c34K|]}t|VdSrT)rrs r6rYz%ClaudeNode.execute..s+>>1#A&&>>>>>>r8zUp to z" images are supported per request.)rVPOST)pathmethod)rolerZ)r9r#messagessystemr*thinking output_config)response_modeldataprice_extractorz!Empty response from Claude model.)rgetr1_ADAPTIVE_THINKING_MODELSrr rr&r'valuessumrg ValueErrorextendrjr2rrrANTHROPIC_MESSAGES_ENDPOINTr r rr ruserrOr NodeOutputr\)r]rnr9rtrvr~r!r#r-thinking_enabledr* thinking_cfg output_cfgbudgetr^rZr?s r6executezClaudeNode.executes !DDDDGn YY|U33  99%7??+=@m[XmEm  8{j88KK))M377K7; 37  ]7776JGGG 2:JKKK ++;<VSzD/@%A%ABB6IU[\\\ +`+` "7L7L7N7N+`+`+` >> >>> > >AR R R[&7[[[\\ \FH  R NN!>eBefffr8)NrQ) __name__ __module__ __qualname____doc__ classmethodrstrdictintrrrr=r8r6rlrlsAAQ Q [Q f # :g:g:g:g :g t :g  :g :g:g:g[:g:g:gr8rlcLeZdZedeeejfdZdS)AnthropicExtensionr:cKtgSrT)rl)selfs r6 get_node_listz AnthropicExtension.get_node_list,s|r8N) rrrrlistrr ComfyNoderr=r8r6rr+sE T$r|*<%=Xr8rc"KtSrT)rr=r8r6comfy_entrypointr1s   r8)2rtyping_extensionsrcomfy_api.latestrrrcomfy_api_nodes.apis.anthropicrrr r r r r rrrcomfy_api_nodes.utilrrrrrrrhrgrrr__annotations__r1rrrr4r7tuplefloatr>rOr\rrrrrjrlrrr=r8r6rs ddd&&&&&&6666666666                        =("!%., !! tCH~% CBB  %%4S> 655!c!!!!HCE%,,?$,F%>54<4&?C \ bl  \ $ \   \ \ \ \RgRgRgRgRgRgRgRgj  2      r8