Files

T

Dariusz L e8939d4bf1 Add URN (ComfyUI-AIR) resolution via CivitAI

Detect ComfyUI-AIR URNs in workflows, resolve them to expected filenames via the CivitAI API, and prefer those filenames for matching.

Key changes:
- core/workflow_analyzer.py: added URN_REGEX, URN_TYPE_MAP, is_model_filename() now recognizes URNs, and get_node_model_info() emits structured URN info (is_urn, urn fields, category, exists=False).
- core/sources/civitai.py: added resolve_urn(model_id, version_id) with caching, API fetch of model/version and primary file, and result shaping (model_name, version_name, expected_filename, files).
- core/linker.py: import resolve_urn and call it for missing URNs, attach civitai_info/expected_filename, and prefer expected_filename as matching target; cleaned up deduplication logic.
- web/linker.js: UI shows URN original plus resolved CivitAI model/version labels where available.
- TODO.md: added plan/notes for URN detection and next steps.

This improves detection and fuzzy matching for models referenced by URNs so missing model resolution is more accurate. Further matcher and updater enhancements remain TODOs.

2026-04-17 18:55:55 +02:00

1.9 KiB

Raw Blame History

ComfyUI Model Linker - Fix URN Detection Plan

Overview

Fix plugin to detect ComfyUI-AIR URNs as missing models, resolve to expected filenames via CivitAI API, fuzzy match locals.

Status: In Progress

Steps

1. [x] URN Parsing in workflow_analyzer.py

Added URN_REGEX, URN_TYPE_MAP
Updated is_model_filename() to detect URNs
Modified get_node_model_info(): parse URN → 'urn' dict, category mapping, is_urn=True, exists=False
Add URN regex to is_model_filename()
Parse components: base, type→category, provider, model_id, version_id
Treat URNs as always "missing" (no folder_paths check)

2. [x] CivitAI URN Resolver in sources/civitai.py

Added _urn_cache, resolve_urn(model_id, version_id) → model_name, expected_filename (primary file)
API: /models/{model_id}?modelVersionId={version_id}
Full files list, logging, error handling
resolve_urn(model_id, version_id) → API https://civitai.com/api/v1/models/{model_id}?modelVersionId={version_id}
Extract modelVersions[0].files[primary].name as expected_filename
Cache results

3. [x] Integrate URN Handling in linker.py

Added import .sources.civitai.resolve_urn
In analyze_and_find_matches(): resolve_urn() for is_urn=True → civitai_info, expected_filename
Matching: use expected_filename for URNs (better fuzzy), else original_path
In analyze_and_find_matches(): For URN refs, fetch expected_filename
Add to missing_models: 'urn_original', 'expected_filename'

4. [ ] Enhance matcher.py

Prioritize matching on expected_filename if present

5. [ ] Workflow Updater Support (Optional)

Replace URN → relative_path in update_model_path()

6. [ ] Testing

Create sample workflow.json with URNs
Run plugin, verify detection/matches
Restart ComfyUI

7. [ ] Polish

UI updates (web/linker.js)
Error handling (API fails)
Rate limit/cache

Next: Step 2 - Add resolver to core/sources/civitai.py

1.9 KiB Raw Blame History