feat: enhance metadata handling with generation type and source image extraction

2026-03-25 15:18:46 -03:00
parent 6e2db46174
commit 7aa4ffdef2
7 changed files with 140 additions and 0 deletions
@@ -309,6 +309,8 @@ class MetaHubSaveNode:
                "project_name": project_name,
                "lora_list": lora_list,
                "lora_hashes": lora_hashes,
+                "generation_type": extracted.get("generation_type"),
+                "source_image": extracted.get("source_image"),
                # Performance metrics (Tier 1, 2, 3)
                "vram_peak_mb": vram_peak_mb if vram_peak_mb is not None else gpu_metrics.get("vram_peak_mb"),
                "gpu_device": gpu_device_override if gpu_device_override else gpu_metrics.get("gpu_device"),