)j΃UdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZmZddlmZ ddlZdZn#e$r d ZddlZddlZYnwxYwd Zd Zd Zd ZdZdZdZhdZ dZ!hdZ"hdZ#de$d<ddgddgddgddgddgdZ%de$d<dd$Z&id%d&gd'd&gd(d&gd)d&gd*d&gd+d,gd-d,gd.d,gd/d0gd1d2gd3d2gd4d2gd5d6gd7d8d9gd:gd;dd?gd@gd?gdAgdBgdCgdDgdEgdFgdGgdGgdGgdHggdI Z'dJe$dK<gdLZ(dMe$dN<hdOZ)ej*dPej+Z,dQZ-dRhZ.ddUZ/ddVdd[Z0dd\Z1dd]dd_Z2ddbZ3e GdcddZ4eefddjZ5dddddeded ddk dd}Z6d~Z7erGddej8Z9ddZ:ddZ;ddZddZ?ddZ@ddZAddZBddZCddZDddZEdZFdZGddZHddZIddZJddZKddddZLddZMdS)u _common.py — Shared logic for ComfyUI skill scripts. Single source of truth for: - HTTP transport (with retry/backoff, streaming, timeout handling) - Cloud detection and endpoint mapping (local ComfyUI vs Comfy Cloud) - Workflow node-type catalogs (param patterns, model loaders, output nodes) - API-format validation - Path-traversal-safe file writes - API-key loading from env / CLI Stdlib-only by design (with optional `requests` upgrade if installed). Python 3.10+. ) annotationsN) dataclass)Path)AnyIterator)urlparseTFzhttp://127.0.0.1:8188zhttps://cloud.comfy.orgCOMFY_CLOUD_API_KEY<g?g>@>  i> SaveAudio SaveVideoCogVideoSamplerLTXVideoSamplerSaveAnimatedPNGWanVideoSamplerSaveAnimatedWEBPVHS_VideoCombineHunyuanVideoSamplerSaveAnimateDiffVideoSVD_img2vid_Conditioning> Image Saveeasy imageSaveShowText|pysssssPreviewImage|pysssssImage Save With MetadataSave3DSaveGLB ImageSaver SaveImager SaveLatent PreviewAudio PreviewImage PreviewVideorrrset[str] OUTPUT_NODESunetdiffusion_modelsclip text_encoders controlnet control_net)r.r/r0r1r2zdict[str, list[str]]FOLDER_ALIASESfolderstrreturn list[str]c:t||gS)z8Return the search order of folder names (primary first).)r4get)r5s A/home/wildlama/.hermes/skills/creative/comfyui/scripts/_common.pyfolder_aliases_forr<fs   fvh / //CheckpointLoaderSimple) ckpt_name checkpointsCheckpointLoaderzCheckpointLoader (Simple)ImageOnlyCheckpointLoaderunCLIPCheckpointLoader LoraLoader) lora_namelorasLoraLoaderModelOnlyLoraLoaderTagsQuery VAELoader)vae_namevaeControlNetLoader)control_net_namer2DiffControlNetLoaderControlNetLoaderAdvanced CLIPLoader) clip_namer0DualCLIPLoader clip_name1r0 clip_name2r0TripleCLIPLoader)rSrU) clip_name3r0CLIPVisionLoader)rQ clip_vision UNETLoader) unet_namer.) model_namer/)r]upscale_models)style_model_name style_models) gligen_namegligen)hypernetwork_name hypernetworks)ipadapter_file ipadapter)instantid_file instantid)r]animatediff_models)photomaker_model_name photomaker) DiffusionModelLoaderUNETLoaderGGUFUpscaleModelLoaderStyleModelLoader GLIGENLoaderHypernetworkLoaderIPAdapterModelLoaderInstantIDModelLoaderADE_LoadAnimateDiffModel ADE_AnimateDiffLoaderWithContextADE_AnimateDiffLoaderGen1PhotoMakerLoaderModelSamplingFluxz dict[str, list[tuple[str, str]]] MODEL_LOADERS)u)CLIPTextEncodetextprompt)CLIPTextEncodeSDXLtext_gr|)r}text_lprompt_l)CLIPTextEncodeSDXLRefinerr{refiner_prompt)CLIPTextEncodeFluxclip_lr)rt5xxlr|)rguidancer)zsmZ CLIPTextEncoder{r|)BNK_CLIPTextEncodeAdvancedr{r|)KSamplerseedr)rstepsr)rcfgr)r sampler_namer)r schedulerr)rdenoiser)KSamplerAdvanced noise_seedr)rrr)rrr)rrr)rrr)r start_at_stepr)r end_at_stepr) RandomNoiserr)BasicSchedulerrr)rrr)rrr)KSamplerSelectrr) CFGGuiderrr) DualCFGGuider cfg_condsr)rcfg_cond2_negative cfg_negative)rx max_shiftr)rx base_shiftr)rxwidth model_width)rxheight model_height)ModelSamplingSD3shiftr)ModelSamplingDiscretesamplingr)SDTurboSchedulerrr)rrr) SamplerCustomrr)rrr)EmptyLatentImagerr)rrr)r batch_sizer)EmptySD3LatentImagerr)rrr)rrr)EmptyHunyuanLatentVideorr)rrr)rlengthr)rrr)EmptyMochiLatentVideorr)rrr)rrr)EmptyLTXVLatentVideorr)rrr)rrr) LatentUpscaler upscale_width)rrupscale_height)LatentUpscaleByscale_byr) ImageScalerr)rrr) LoadImageimager) LoadImageMaskr mask_image)LoadImageOutputrr) VHS_LoadVideovideor) VHS_LoadAudioaudior)r>r?r?)rAr?r?)rBr?r?)rIrJrJ)r[r\r\)rlr]diffusion_model_name)rnr]upscale_model_name)rPrQrQ)rRrTrT)rRrVrV)rLrMcontrolnet_name)rDrErE)rDstrength_model lora_strength)rD strength_cliplora_strength_clip)rGrErE)rGrr)ControlNetApplystrengthcontrolnet_strength)ControlNetApplyAdvancedrr)r start_percentcontrolnet_start)r end_percentcontrolnet_end)IPAdapterAdvancedweightipadapter_weight)rstart_atipadapter_start)rend_at ipadapter_end) IPAdapterrr)ImageUpscaleWithModelupscale_methodr)ru motion_scaler)rvrr)r frame_rater)rformat video_format)rfilename_prefixr)r'rr)rrr)rrr)rrr)rrr)rrr)rrr) LTXVSchedulerrr)rrr)zSeed (rgthree)rr)zImage Comparer (rgthree)image_ar)zPower Lora Loader (rgthree)PowerLoraLoaderHeaderWidget _lora_header) PrimitiveNodevalueprimitive_value)z easy seedrr)z easy positivepositiver|)z easy negativenegativenegative_prompt)easy fullLoaderr?r?)rrJrJ)rrErE)rrr|)rrrzlist[tuple[str, str, str]]PARAM_PATTERNS>r{rrr~rrrzc(?:^|[\s,(\[])embedding\s*:\s*([A-Za-z0-9_\-\./\\]+?)(?:\.(?:pt|safetensors|bin))?(?=[\s:,)\(\]]|$))z .comfy.orgzcloud.comfy.orghostboolctd|vr|nd|}|jpdtvrdSt fdt DS)zFTrue if the host points at Comfy Cloud (or staging/preview subdomain).z://zhttp://Tc3BK|]}|VdSN)endswith).0shostnames r; z is_cloud_host..Us1CCx  ##CCCCCCr=)rrlowerCLOUD_DOMAIN_EXACTanyCLOUD_DOMAIN_SUFFIXES)rparsedrs @r; is_cloud_hostrOsr etmmdd1A41A1A B BF%2,,..H%%%t CCCC-BCCC C CCr= force_cloudbasepathr bool | Nonec|d}|t|n|}|dsd|z}|r|dsd|z}||zS)aBuild a URL that adds /api prefix when targeting Comfy Cloud. Local ComfyUI accepts both `/foo` and `/api/foo` for many endpoints. Cloud requires `/api/foo`. `path` should be a path component (e.g. "/prompt") or full path with query (e.g. "/view?filename=x"). /Nz/api/z/api)rstripr startswith)rrrclouds r;build_cloud_aware_urlrXsw ;;s  D#.#6M$   KE ??3  Tz T__W--} $;r=c|dr/|dsd|tddzS|drd|tddzS|dkrdS|S)zMap a cloud endpoint path to its current canonical form. Handles known renames documented in the Comfy Cloud API: /history -> /history_v2 /models/ -> /experiment/models/ /models -> /experiment/models z/historyz /history_v2Nz/models/z/experiment/models/z/modelsz/experiment/models)rlen)rs r;cloud_endpointrjs z""64??=+I+I6tC OO$4$4555 z"">$tC OO,<,<'=== y## Kr=)is_cloudrcn|t|n|}|rt|}t|||S)zETop-level URL resolver. Applies cloud rename + /api prefix as needed.Nr)rrr)rrrrs r; resolve_urlr{sC#+#3M$   E $d## t ? ? ??r=explicit str | Nonec|r|n#tjt}|dS|d}|pdS)uHLook up API key from CLI flag → env var. Strips whitespace and quotes.Nz'")osenvironr: ENV_API_KEYstrip)r vals r;resolve_api_keyrsM ?((BJNN;$?$?C {t ))++  E " "C ;$r=cJeZdZUded<ded<ded<ded<ddd ZddZdS) HTTPResponseintstatusdict[str, str]headersbytesbodyr6urlutf-8encodingr7c:|j|dS)Nreplaceerrors)rdecode)selfrs r;r{zHTTPResponse.textsy;;;r=rc^tj|jddS)Nrrr)jsonloadsrr!)r"s r;r$zHTTPResponse.jsons'z$)**79*EEFFFr=N)r)rr6r7r6)r7r)__name__ __module__ __qualname____annotations__r{r$r=r;rrspKKKKKK HHH<<<<<GGGGGGr=rattemptrfloatcapNonect||d|zz}tjd|}tj|dS)z+Sleep with full-jitter exponential backoff.rN)minrandomuniformtimesleep)r+rr-delays r;_sleep_backoffr7sA TQ'\* + +E N1e $ $EJur=) r json_bodydatafilesformtimeoutfollow_redirectsretriesstreamsinkmethodrrdict[str, str] | Noner8rr9 bytes | Noner: dict | Noner;r<r=r>r?r@ Path | Nonec |i}t|}|dd|s|tstdd} t | D]} t |||||||||| |  }|jtvr| dz| krt| D|cS#tttf$r%}|} | dz| krt| Yd}~~d}~wwxYw| r| td)aSingle entry point for all HTTP traffic. Behavior: - Retries on connection errors and on HTTP statuses in RETRY_STATUS_CODES, with exponential backoff + jitter. - For cross-host redirects, drops Authorization-style headers (so signed URLs don't leak the API key to S3/CloudFront). - When `stream=True` and `sink` is a Path, streams the response body to disk in 64 KiB chunks instead of buffering. Either `json_body`, `data`, or `files`+`form` may be supplied (mutually exclusive). Nz User-Agentzhermes-comfyui-skill/5.0zTMultipart upload requires the `requests` package. Install with: pip install requests) rArrr8r9r:r;r<r=r?r@z0http_request: retries exhausted with no response) dict setdefault HAS_REQUESTS RuntimeErrorrange _http_oncerRETRY_STATUS_CODESr7 TimeoutErrorConnectionErrorOSError)rArrr8r9r:r;r<r=r>r?r@last_excr+respes r; http_requestrUsZ67mmG |%?@@@   5  "&H>> 3#$e$2BD D {000Wq[75J5Jw'''KKKow7   H{W$$w'''    I J JJs%ABBC4CCC) x-api-key authorizationcookiec"eZdZdZfdZxZS) _StripSensitiveOnRedirectSessionuSession that drops sensitive headers on cross-host redirects. `requests` already strips `Authorization` cross-host (rebuild_auth), but it does NOT strip custom headers like `X-API-Key`. We override `rebuild_auth` to additionally strip every header in `_SENSITIVE_HEADERS` when the destination is a different host — critical when ComfyUI Cloud's `/api/view` redirects to a signed S3 URL. ct|| |jj}|j}t |jpd}t |jpd}|rQ|rQ||krM|j}t| D]&}|tvr||=!dSdSdSdS#t$rYdSwxYw)Nr) super rebuild_authrequestrrrrrlistkeys_SENSITIVE_HEADERS Exception) r"prepared_requestresponseold_urlnew_urlold_hostnew_hostrkey __class__s r;r]z-_StripSensitiveOnRedirectSession.rebuild_auths GG !18 < < < "*.*.$W--6<"CCEE$W--6<"CCEE--X-A-A.6G#GLLNN33--99;;*<<< ' -----A-A--    sB6C## C10C1)r&r'r(__doc__r] __classcell__rjs@r;rZrZsB           r=rZrc tr|||||d} ||| d<n||| d<n|| || d<|| d<| rd| d<t5} | jdi| } | r| | jdd| d 5}| tD]}|r|| dddn #1swxYwYd }n| j }t| j d | j D|| j cdddS#tjj$r^}t%|tjjrt)t+||t-t+||d}~wwxYw#1swxYwY|>t/j|d }|ddn|}t6j||||}Gddt6jj}t=|jpd }t6j!|||} |||}nw#t6j"j#$r`}t|j$|j rtK|j ni|&pd tO|d| cYd}~Sd}~wwxYw|(}|j)}tK|j }| r| | jdd| d 5} |&t}|sn||3 dddn #1swxYwYt||d | St|||&| S)zOne HTTP attempt. No retry.)rArrr<allow_redirectsNr$r9r:Tr?)parentsexist_okwbr=ci|]\}}|| Sr*r*rkvs r; z_http_once..0s@@@daQ@@@r=)rrrrrz Content-Typezapplication/json)r9rrAc&eZdZddZfdZxZS)$_http_once.._RedirectHandler original_hostr6followrc"||_||_dSr)rzr{)r"rzr{s r;__init__z-_http_once.._RedirectHandler.__init__Gs!.D  DKKKr=cL|jsdSt|jpd}||jkrBd|D}t j||d} | St ||||||S)NrcFi|]\}}|dv||S)>rXrWrV)rrts r;rwzI_http_once.._RedirectHandler.redirect_request..Qs<!!!!Qwwyy(PPPqPPPr=GET)rrA) r{rrrrz header_itemsurllibr^Requestr\redirect_request) r"req2fpcodemsghdrsnewurlrh clean_headersnew_reqrjs r;rz5_http_once.._RedirectHandler.redirect_requestKs; t ((17R>>@@H4---!!%)%6%6%8%8!!! !.00W\0]]77++D"dCvNN Nr=)rzr6r{r)r&r'r(r}rrlrms@r;_RedirectHandlerryFsU ! ! ! ! O O O O O O O O Or=rr)r<rr*)*rJrZr^parentmkdiropen iter_contentDOWNLOAD_CHUNK_SIZEwritecontentr status_coderitemsrrequests exceptionsRequestException isinstanceTimeoutrOr6rPr$dumpsencoderIrrHTTPRedirectHandlerrrr build_openererror HTTPErrorrrHreadgetattrgeturlr)rArrr8r9r:r;r<r=r?r@kwargsrrfchunkrrT body_bytesreqrrzopenerrS final_url final_status final_headerss r;rMrM s!'5SW3C" "   &F6NN  !F6NN  $"2#F7O!F6N  $#F8 . / / 51 5AI''''%d.K%%dT%BBB4/A%&^^4G%H%H//E$/ !////////////////DD9D#=@@aioo.?.?@@@  5 5 5 5 5 5 5 5$&7 5 5 5a!4!<==6&s1vv..A5%c!ff--14  5% 5 5 5 5 5 5 5 54Z **11':: >+=>>>> . :wv V VC OOOOO6>=OOO&c]]+1r88::M ^ ( ()9)9-IY)Z)Z [ [F {{3{00 < !   6'(y8DOOObS5#&&            I;L&&M a$" $666 YYt__   "566                   <SV_```` |]Zc d d ddsFAD)5C > D) C D)C A D))F=AFFFF"%F"=JL )AL>L L 05N22N69N6rc td|fi|S)NrrUrrs r;http_getrws s - -f - --r=c td|fi|S)NPOSTrrs r; http_postr{s  . .v . ..r=workflowct|tsdSd|vrd|vrdS|D]}t|trd|vrdSdS)z>API format = top-level dict where each value has `class_type`.Fnodeslinks class_typeT)rrHvalues)rrvs r; is_api_formatrsu h % %u(w(22u __   a   <1#4#444 5r=payloadrHc0t|trt|r|St|tr!d|vrt|dr|dSt|trd|vrd|vrtdtd)zQUnwrap common wrapper variants. Returns API-format workflow or raises ValueError.r|rruWorkflow is in editor format (has top-level 'nodes' and 'links' arrays). Re-export from ComfyUI using 'Workflow → Export (API)' (newer UI) or 'Save (API Format)' (older UI).zSWorkflow is not in API format. Each top-level entry must have a 'class_type' field.)rrHr ValueError)rs r;unwrap_workflowrs'4  ]7%;%;'4  !X%8%8]7S[K\=]=]%8x  '4   W%7%7Gw.avi.gif.mkv.mov.mp4.webm.webpr>.m4a.mp3.ogg.wav.flacr>.glb.obj.ply.gltf3d>.md.txt.jsonr{r)rsuffixr)rexts r;media_type_from_filenamerse x..  % % ' 'C HHHw 777w ///t &&&v 7r=ct|D]G\}}|dtvrdS|ddrdSHdS)z1Used to bump default timeout for video workflows.rT) animatediffade_wanvideo hunyuanvideoltxvideocogvideoF)rSLOW_OUTPUT_NODESrr)r_rs r;looks_like_video_workflowrsnh''4  !2 2 244   # # % % 0 01| } } 44  5r=lcD|tjttSt |t r7|dkrtjttS|dkrtjttSt|S)zConvert -1 or None to a fresh random seed; otherwise return int(value). Accepts numeric -1 OR string "-1" (both treated as "randomize"). Other parse failures raise TypeError/ValueError for the caller to surface. Nz-1)r2randintSEED_MINSEED_MAXrr6rrrs r; coerce_seedrsx  }~h111%2%++--4"7"7~h111 {{~h111 u::r=ct|tstSt}|D]}t|tr||-t|t ri|dp)|dp|d}t|tr|||S)zNormalize model-list responses from local ComfyUI vs Comfy Cloud. Local: `["a.safetensors", "b.safetensors"]` Cloud: `[{"name": "a.safetensors", "pathIndex": 0}, ...]` namerr)rr_setr6addrHr:)routitemr s r;parse_model_listrs gt $ $uu EEC dC   GGDMMMM d # # 88F##Otxx ';';Otxx?O?OD$$$  Jr=cBttjSr)r6uuiduuid4r*r=r; new_client_idr3s tz||  r=dcddd|DS)zPretty key=value for log lines. c3*K|]\}}|d|VdS)=Nr*rts r;rzfmt_kv..9s077TQqLL1LL777777r=)joinr)rs r;fmt_kvr7s+ 8877QWWYY777 7 77r=r0)indentobjrcXttj||tdS)zKPrint JSON to stdout. Centralised so behavior can be tweaked (e.g., --raw).)rdefaultN)printr$rr6)rrs r; emit_jsonr <s' $*S 5 5 566666r=rcBtd|tjdS)z?stderr log with consistent prefix (so JSON stdout stays clean).z[comfyui-skill] )fileN)rsysstderr)rs r;logr%As& "S " "444444r=)r5r6r7r8)rr6r7r)rr6rr6rrr7r6)rr6r7r6)rr6rr6rrr7r6)r r r7r )r+rrr,r-r,r7r.)rAr6rr6rrBr8rr9rCr:rDr;rDr<r,r=rr>rr?rr@rEr7r)rAr6rr6rrr8rr9rCr:rDr;rDr<r,r=rr?rr@rEr7r)rr6rrr7r)rrr7r)rrr7rH)rrr7r)rrHr7r)rrHr7r)rrHr7r)rrrr6r7r)rr6r7r6)rrHr7r)rrr7r)rrr7r,)r7r6)rrHr7r6)rrrrr7r.)rr6r7r.)Nrk __future__rr$r r2rer#r4r dataclassesrpathlibrtypingrr urllib.parserrrJ ImportError urllib.errorrurllib.requestDEFAULT_LOCAL_HOSTDEFAULT_CLOUD_HOSTrDEFAULT_HTTP_TIMEOUTDEFAULT_RETRIESRETRY_BASE_DELAYRETRY_MAX_DELAYrNrrr-r)r4r<ryrrcompile IGNORECASErrrrrrrrrr7rUraSessionrZrMrrrrrrrrrrrrrrrrrr r%r*r=r;r8s   #""""" !!!!!! !!!!!!OOOLLL-.# === 2' (+V4 _ %%v./ ((0000.3!= >.3!= >.3 ">!? .3  ">!? .3 != > .3!7 8.3!7 8.3!7 8.3!4 5.3!C D.3!C D.3 !C D!.3$!6 7%.3&!79O P'.3( h h h).3*!= >+.3.!6 7/.30"D D!6 7!A B!E F!: ;!G H "A A!@ A!E F)M(N"F!G!H I "[.3.3.3 ....hU.U.U.UUUUpXWW "*jM('(DDDDOS$"BF@@@@@@  G G G G G G G  G0@o&*)!"@K@K@K@K@K@KF>8+;6ieieieieX....////    &    & * * * *""          (.8888 *+777777 555555sAAA