88 lines
13 KiB
Plaintext
88 lines
13 KiB
Plaintext
|
ó
|
|||
|
|
ó0aj, ã óx • S r SSKJr SSKrSSKrSSKrSSKJr SSKJ r \R R S\ " \" \
|
|||
|
|
5 R 5 R 5 5 SSKJrJrJrJrJrJrJrJr 1 SkrSS jrS S
|
|||
|
|
.SS jjrSS jr SS
|
|||
|
|
jr SS jr SSS jjr \ S:X a \RB " \ " 5 5 gg)uÅ
|
|||
|
|
extract_schema.py — Analyze a ComfyUI API-format workflow and extract
|
|||
|
|
controllable parameters.
|
|||
|
|
|
|||
|
|
Improvements over v1:
|
|||
|
|
- Catalogs live in `_common.py`, shared with `check_deps.py`
|
|||
|
|
- Coverage expanded for Flux / SD3 / Wan / Hunyuan / LTX / IPAdapter / rgthree
|
|||
|
|
- Symmetric duplicate-name resolution: ALL duplicates get a node-id suffix
|
|||
|
|
(instead of "first wins, second renamed"), so callers see consistent names
|
|||
|
|
- Negative prompt detected by tracing `KSampler.negative` connections back to
|
|||
|
|
the source CLIPTextEncode (more reliable than meta-title heuristic)
|
|||
|
|
- Embedding references in prompt text are extracted as model dependencies
|
|||
|
|
- Detects Primitive nodes that drive other nodes' inputs (and surfaces them
|
|||
|
|
as the user-facing parameter)
|
|||
|
|
- Reroutes are followed when tracing connections
|
|||
|
|
|
|||
|
|
Usage:
|
|||
|
|
python3 extract_schema.py workflow_api.json
|
|||
|
|
python3 extract_schema.py workflow_api.json --output schema.json
|
|||
|
|
|
|||
|
|
Stdlib-only. Python 3.10+.
|
|||
|
|
é )Ú annotationsN)ÚPath)ÚAny)Ú OUTPUT_NODESÚPARAM_PATTERNSÚ
|
|||
|
|
PROMPT_FIELDSÚis_linkÚiter_embedding_refsÚiter_model_depsÚ
|
|||
|
|
iter_nodesÚunwrap_workflow> ÚKSamplerÚ CFGGuiderÚ BasicGuiderÚ
|
|||
|
|
DualCFGGuiderÚ
|
|||
|
|
SamplerCustomÚKSamplerAdvancedÚSamplerCustomAdvancedc ó • [ U [ 5 ( a g[ U [ 5 ( a g[ U [ 5 ( a g[ U [ 5 ( a g[ U [
|
|||
|
|
5 ( a g[ U [ 5 ( a gg)NÚboolÚintÚfloatÚstringÚlinkÚobjectÚunknown)Ú
|
|||
|
|
isinstancer r r ÚstrÚlistÚdict)Úvalues ÚT/home/n8n/.hermes/profiles/general/skills/creative/comfyui/scripts/extract_schema.pyÚ
|
|||
|
|
infer_typer# 0 se € Ü�%œ× Ñ ØÜ�%œ× Ñ ØÜ�%œ× Ñ ØÜ�%œ× Ñ ØÜ�%œ× Ñ ØÜ�%œ× Ñ ØØ ó é )Úmax_hopsc ó¼ • [ U5 ( d gUS n[ 5 n[ U5 H¬ nUb X4; a Us $ UR U5 U R U5 n[ U[ 5 ( d gUR SS5 nUS; aJ UR S0 5 =( d 0 n[ S UR 5 5 S5 n U c Us $ U S nMª Us $ U$ )zöFollow a [node_id, slot] link, hopping through Reroute / Primitive nodes
|
|||
|
|
if needed, to find the *upstream* node id that holds the actual value/input.
|
|||
|
|
|
|||
|
|
Bounded by both `max_hops` AND a visited-set to prevent infinite loops on
|
|||
|
|
pathological graphs.
|
|||
|
|
Nr Ú
|
|||
|
|
class_typeÚ )ÚRerouteÚ
|
|||
|
|
PrimitiveNodeÚNotezeasy showAnythingÚinputsc 3 óJ # • U H n[ U5 ( d M Uv • M g 7f©N)r )Ú.0Úvs r" Ú <genexpr>Ú trace_to_node.<locals>.<genexpr>W s é € Ð Gª A¼GÀA¿JŸa™aª ùs ‚#š #) r ÚsetÚrangeÚaddÚgetr r ÚnextÚvalues)
|
|||
|
|
Úworkflowr r& ÚnidÚvisitedÚ_ÚnodeÚclsr- Ú next_links
|
|||
|
|
r" Ú
|
|||
|
|
trace_to_noderA @ sÏ € ô �4�=‰=Øؘ1‘g€CÜ›€GÜ
|
|||
|
|
�8Ž_ˆØ ‰;˜#›.ØŠJØ� ‰ �CÔØ�|‰|˜CÓ ˆÜ˜$¤×%Ñ%ÙØ�h‰h�| RÓ(ˆà ÐKÓ KØ—X‘X˜h¨Ó+×1¨rˆFä Ñ G¨¯ © ¬ Ó GÈÓNˆIØÑ Ø’
|
|||
|
|
ؘA‘,ˆCÙ ØŠ
|
|||
|
|
ñ# ð$ €Jr$ c ó¢ • [ U 5 HÀ u pUS [ ; a M UR S0 5 =( d 0 nUR S5 n[ U5 ( d MR [ X5 nU( d Mf [ U R U5 [ 5 ( d MŒ X R SS5 nUR S5 ( d US; d M¾ Us $ g)zDTrace `negative` input of a sampler back to the source text encoder.r( r- Únegativer) ÚCLIPTextEncode©zsmZ CLIPTextEncodeÚBNK_CLIPTextEncodeAdvancedN©r ÚSAMPLER_NODE_FAMILYr7 r rA r r Ú
|
|||
|
|
startswith)r: r; r> r- ÚnegÚsrcr? s r" Úfind_negative_prompt_noderL ` s € ä Ö)‰ ˆØ � Ñ Ô%8Ó 8Ù Ø—‘˜( BÓ'×-¨2ˆØ�j‰j˜Ó$ˆÜ�s�|‰|٠ܘHÓ*ˆß ˆ3”:˜hŸl™l¨3Ó /´×6Ó6Ø‘-×#Ñ# L°"Ó5ˆCØ�~‰~Ð .×/Ñ/°3Ð:nÕ3nØ’
|
|||
|
|
ñ *ð r$ c ó¢ • [ U 5 HÀ u pUS [ ; a M UR S0 5 =( d 0 nUR S5 n[ U5 ( d MR [ X5 nU( d Mf [ U R U5 [ 5 ( d MŒ X R SS5 nUR S5 ( d US; d M¾ Us $ g )Nr( r- Úpositiver) rD rE rG )r: r; r> r- ÚposrK r? s r" Úfind_positive_prompt_noderP q s € Ü Ö)‰ ˆØ � Ñ Ô%8Ó 8Ù Ø—‘˜( BÓ'×-¨2ˆØ�j‰j˜Ó$ˆÜ�s�|‰|٠ܘHÓ*ˆß ˆ3”:˜hŸl™l¨3Ó /´×6Ó6Ø‘-×#Ñ# L°"Ó5ˆCØ�~‰~Ð .×/Ñ/°3Ð:nÕ3nØ’
|
|||
|
|
ñ *ð r$ c ód ^ ^ • / n[ T 5 n[ T 5 n/ n[ T 5 Hÿ u pVUS nUR S0 5 =( d 0 nU[ ; a UR U5 [ H´ u pšn Xy:w a M
|
|||
|
|
X¨; a M XŠ n [ U 5 n
|
|||
|
|
U
|
|||
|
|
S:X a M+ U nU S:X ai XS:X a X#:w a SnO\XR:X a SnOTUR S5 =( d 0 R SS5 R 5 m [ U 4S jS
|
|||
|
|
5 5 ( a SnUR UUU
|
|||
|
|
U
|
|||
|
|
U US .5 M¶ GM 0 nU H' nUR US / 5 R U5 M) 0 nUR 5 Ht u nn[ U5 S
|
|||
|
|
:X a! US nUS US US US US S.UU' M6 UR S S9 U H( nU SUS 3nUS US US US US US.UU' M* Mv [ [ T 5 5 n/ n[! 5 n[# T 5 H· u nnUU4nUU; a M UR% U5 T R U0 5 nUR S0 5 =( d 0 nSnSn UR 5 H8 u n n [' U [( 5 ( d M U [* ; d M) UU ; d M1 U nU SS n O UR UUUU SS.5 M¹ [ U5 [ U5 [ U5 [ U5 SU; S U; =( d [ S U 5 5 [ U 4S jU 5 5 S .n UUUUU S .$ )!av Extract controllable parameters from a workflow.
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
{
|
|||
|
|
"parameters": { friendly_name: {node_id, field, type, value, ...} },
|
|||
|
|
"output_nodes": [node_id, ...],
|
|||
|
|
"model_dependencies": [{node_id, class_type, field, value, folder}],
|
|||
|
|
"embedding_dependencies": [{node_id, embedding_name, found_in_field, value_excerpt}],
|
|||
|
|
"summary": {...}
|
|||
|
|
}
|
|||
|
|
r( r- r ÚpromptÚnegative_promptÚ_metaÚtitler) c 3 ó, ># • U H oT; v • M g 7fr/ © )r0 Út_Ú
|
|||
|
|
meta_titles €r" r2 Ú!extract_schema.<locals>.<genexpr>µ s øé € Ð]Ò6\° Ö+Ò6\ùs ƒ)rC rJ z-promptÚanti)Ú name_hintÚnode_idÚfieldÚtyper! r( r\ é r r] r^ r_ r! )r] r^ r_ r! r( c óF • [ U S 5 R S5 U S 4$ )Nr] r% r^ )r Úzfill)Úxs r" Ú<lambda>Ú extract_schema.<locals>.<lambda>Ó s! € ¬¨A¨i©LÓ(9×(?Ñ(?ÀÓ(BÀAÀgÁJÑ'Or$ )Úkeyr= )r] r^ r_ r! r( Úalias_ofNéx Ú
|
|||
|
|
embeddings)r] Úembedding_namer^ Ú
|
|||
|
|
value_excerptÚfolderÚseedc 3 óB # • U H oR S 5 v • M g7f)Úseed_N)rI )r0 Úps r" r2 rZ s é € Ð/ZÊzÈ!· ± ¸W×0EÐ0EÊzùs ‚ c 3 ón ># • U H* nTR U0 5 R S S5 S; v • M, g7f)r( r) > Ú SaveVideoÚSaveAnimatedPNGÚSaveAnimatedWEBPÚVHS_VideoCombineN)r7 )r0 Únr: s €r" r2 rZ s< øé € ð !
|
|||
|
|
ò $�!ð
|
|||
|
|
�L‰L˜˜BÓ × #Ñ # L°"Ó 5ð :ö
|
|||
|
|
â#ùs ƒ25)Úparameter_countÚoutput_node_countÚmodel_dep_countÚembedding_dep_countÚhas_negative_promptÚhas_seedÚis_video_workflow)Ú
|
|||
|
|
parametersÚ output_nodesÚmodel_dependenciesÚembedding_dependenciesÚsummary)rP rL r r7 r Úappendr r# ÚlowerÚanyÚ
|
|||
|
|
setdefaultÚitemsÚlenÚsortr r r4 r
|
|||
|
|
r6 r r r )!r: r Úpos_nodeÚneg_nodeÚ
|
|||
|
|
raw_paramsr] r> r? r- Úp_classÚp_fieldÚfriendlyr! ÚtÚ actual_nameÚby_nameÚrr~ ÚnameÚentriesÚ full_nameÚ
|
|||
|
|
model_depsÚembedding_depsÚseen_embr; Úemb_namerf Ú found_fieldÚexcerptÚfnameÚfvalr‚ rY s! ` @r" Úextract_schemarŸ � sˆ ù€ ð !€Lô )¨Ó2€HÜ(¨Ó2€Hð
|
|||
|
|
€Jä# HÖ-‰
|
|||
|
|
ˆØ�<Ñ ˆØ—‘˜( BÓ'×-¨2ˆà ”,Ó Ø × Ñ Ô (÷ +9Ñ &ˆG˜hØ‹~ÙØÓ$ÙØ‘OˆEܘ5Ó!ˆAØ�F‹{Ùà"ˆKð ˜8Ó#ØÓ&¨8Ó+?Ø"3‘KØ Ó(Ø"*‘Kð #'§(¡(¨7Ó"3×"9°r×!>Ñ!>¸wÈÓ!K×!QÑ!QÓ!S�JÜÔ]Ñ6\Ó]×]Ñ]Ø&7˜ à × Ñ Ø (Ø"Ø ØØ Ø !ñ
|
|||
|
|
ö
|
|||
|
|
ô1 +9ñ .ðX &(€GÛ
|
|||
|
|
ˆØ×ј1˜[™>¨2Ó.×5Ñ5°aÖ8ñ ð #%€JØ Ÿ ™ ž ‰
|
|||
|
|
ˆˆgÜ ˆw‹<˜1Ó Ø˜‘
|
|||
|
|
ˆAà ˜Y™<°!°G±*ؘ&™ ¨A¨g©JØ ™oñ ˆJ�tÓ ð
|
|||
|
|
�L‰LÑ OˆLÑ PÛ �Ø #˜f A a¨ ¡l ^Ð 4� à ™|°a¸±jØ ˜f™I°°'±
|
|||
|
|
Ø"# L¡/Ø $ñ )�
|
|||
|
|
˜9Ó%ó ñ )ô* ”o hÓ/Ó0€Jð "$€NÜ%(£U€HÜ,¨XÖ6‰
|
|||
|
|
ˆˆXØ�HˆoˆØ �(‹?Ù Ø� ‰ �SÔà�|‰|˜C Ó$ˆØ—‘˜( BÓ'×-¨2ˆØˆ ؈Ø!Ÿ<™<ž>‰KˆE�4ܘ$¤×$Ó$¨´-Õ)?ÀHÐPTÕDTØ #� Ø ˜t ˜*�Ùñ *ð
|