+j&UdZddlmZddlmZmZmZddlmZm Z m Z m Z m Z m Z mZddlmZmZmZmZmZmZdZdZdZd d d d Zeeefed <ddddZeeeeeeffed<eefde de fdZ!dedededzfdZ"dedefdZ#de$ej%de&ej'de&efdZ(de$ej%de&ej)de&e fd Z*Gd!d"ej%Z+Gd#d$eZ,de,fd%Z-dS)&zAPI Nodes for ByteDance Seed LLM via the BytePlus ModelArk Responses API. See: https://docs.byteplus.com/en/docs/ModelArk/1585128 )override)IOComfyExtensionInput)BytePlusInputImageBytePlusInputMessageBytePlusInputTextBytePlusInputVideoBytePlusMessageContentBytePlusResponseCreateRequestBytePlusResponseObject) ApiEndpointget_number_of_imagessync_opupload_images_to_comfyapiupload_video_to_comfyapivalidate_stringz /proxy/byteplus/api/v3/responsesseed-2-0-pro-260328seed-2-0-lite-260228seed-2-0-mini-260215)z Seed 2.0 Proz Seed 2.0 Litez Seed 2.0 Mini SEED_MODELS)g?皙?g@)g?g?@)rg{Gz?g?)rrr_SEED_PRICES_PER_MILLION max_images max_videosc Dtjdtjtjddt d|dzDdd|d tjd tjtjd d t d|dzDdd |d tjdddddddgS)Nimagesimagecg|]}d|S)image_.0is C/home/wildlama/comfy/ComfyUI/comfy_api_nodes/nodes_bytedance_llm.py z&_seed_model_inputs..3FFF|||FFFr)namesminz9Optional image(s) to use as context for the model. Up to z images.)templatetooltipvideosvideocg|]}d|S)video_r$r%s r(r)z&_seed_model_inputs..<r*r+z9Optional video(s) to use as context for the model. Up to z videos. temperatureg?grg{Gz?zIControls randomness. 0.0 is deterministic, higher values are more random.T)defaultr.maxstepr0advanced)rAutogrowr TemplateNamesImagerangeVideoFloat)rrs r(_seed_model_inputsr@-s5  [..w''FFU1j1n-E-EFFF/ ePZddd     [..w''FFU1j1n-E-EFFF/ ePZddd     _   ' r+model_idresponsereturnNc(|jsdSt|}|dS|\}}}|jjpd}|jjpd}d}|jjr|jjjpd}td||z } | |z||zz||zz} | dz S)z2Compute approximate USD price from response usage.Nrg.A)usagerget input_tokens output_tokensinput_tokens_details cached_tokensr7) rArBrates input_ratecache_hit_rate output_raterGrHcached fresh_inputtotals r(_calculate_pricerRMs >t $ ( ( 2 2E }t.3+J >.3!LN05AM F~*H4BGaa.//K * $v'> >Q\A\ \E ; r+c<|jsdSg}|jD]u}|jdks|js|jD]X}|jdkr"|jr||j/|jdkr|jrt d|jYvd|S)zHExtract concatenated text from all assistant message output_text blocks.message output_textrefusalzModel refused to respond:  )outputtypecontenttextappendrW ValueErrorjoin)rBchunksitemblocks r(_get_text_from_responserc_s ?rFOO 9 ! ! ! \ O OEz]**uz* ej))))y((U]( !Mem!M!MNNN  O 99V  r+cls image_tensorscZKt||tdd{V}d|DS)NzUploading reference images)r wait_labelc.g|]}t|S)) image_url)r)r&urls r(r)z/_build_image_content_blocks..ys$ > > >#  - - - > > >r+)rSEED_MAX_IMAGES)rdreurlss r(_build_image_content_blocksrmos\+ "/       D ? > > > >>r+r1cKg}t|}t|D]V\}}d}|dkr|d|dzd|d}t|||d{V}|t |W|S)NzUploading reference videor,z (/))rg) video_url)len enumeraterr]r )rdr1blocksrQidxr2labelrjs r(_build_video_content_blocksrw|s(*F KKE''99 U+ 19922a22%222E,S%EJJJJJJJJJ (37778888 Mr+c beZdZdZedZe d dedededede j f d Z d S) ByteDanceSeedNodez8Generate text responses from a ByteDance Seed 2.0 model.c\tjdddddtjdddd tjd d t Dd tjddddddtjddddddgtjgtjj tjj tjj gdtj tj d gd S)NryzByteDance Seedzpartner/text/ByteDancezText GenerationzGenerate text responses with ByteDance's Seed 2.0 models. Provide a text prompt and optionally one or more images or videos for multimodal context.promptTrTzText input to the model.) multiliner6r0modelcfg|].}tj|t/Sr$)r DynamicComboOptionr@)r&rvs r(r)z3ByteDanceSeedNode.define_schema..s2jjjUZR_33E;M;O;OPPjjjr+z-The Seed model used to generate the response.)optionsr0seedriz_Seed controls whether the node should re-run; results are non-deterministic regardless of seed.)r6r.r7control_after_generater0 system_promptz.s+m+m+m!_`_lA_l_l_lr+r c34K|]}t|VdSr)rrs r( z,ByteDanceSeedNode.execute..s+>>1#A&&>>>>>>r+zUp to z" images are supported per request.cg|]}||Srr$)r&vs r(r)z-ByteDanceSeedNode.execute..s*l*l*l^_^k1^k^k^kr+r1z" videos are supported per request.)r\POST)pathmethoduser)roler[F)r}input instructionsr5storestreamc$t|Sr)rR)rrAs r(z+ByteDanceSeedNode.execute..s&6x&C&Cr+)response_modeldataprice_extractorzSeed API error (z): zEmpty response from Seed model.)rrrFvaluessumrkr^rrSEED_MAX_VIDEOSextendrmrwr]r rrBYTEPLUS_RESPONSES_ENDPOINTr r rerrorcoderUrcr NodeOutput) rdr{r}rr model_labelr5re video_inputsr[rBresultrAs @r(executezByteDanceSeedNode.executes !DDDDGn M* {++m+m (8K8K8Qr7Y7Y7[7[+m+m+m >> >>> > > P PYoYYYZZ Z*l*luyy7J7J7Pb6X6X6Z6Z*l*l*l |   . .YoYYYZZ Z02  R NN! b`0C``H^``aa a(22 @>?? ?}V$$$r+N)rT) __name__ __module__ __qualname____doc__ classmethodrstrdictintrrrr$r+r(ryrysBBE E [E N  .%.%.%.% .%  .%  .%.%.%[.%.%.%r+rycLeZdZedeeejfdZdS)ByteDanceLLMExtensionrCcKtgSr)ry)selfs r( get_node_listz#ByteDanceLLMExtension.get_node_list s!""r+N) rrrrlistrZr ComfyNoderr$r+r(rrsE #T$r|*<%=###X###r+rc"KtSr)rr$r+r(comfy_entrypointrs " ""r+).rtyping_extensionsrcomfy_api.latestrrr"comfy_api_nodes.apis.bytedance_llmrrr r r r r comfy_api_nodes.utilrrrrrrrrkrrrr__annotations__rtuplefloatrr@rRrcrZrrr<rmr>rwryrrr$r+r(rs '&&&&&6666666666A*++ T#s(^...CC$sE%*=$>>?*9O3c@s.DQU$ &<      ? bl  ? $ ?  ? ? ? ?  bl          z%z%z%z%z% z%z%z%z#####N### # 5######r+