feat(parser): enhance model metadata extraction in Automatic1111 parser

- Add MODEL_NAME_PATTERN regex to extract model names from parameters
- Extract model hash from parsed hashes when available in metadata
- Add checkpoint model hash and name extraction from parameters section
- Implement checkpoint resource processing from Civitai metadata
- Improve model information completeness for better recipe tracking

tests/services/test_automatic_metadata_parser.py

@@ -0,0 +1,120 @@
+import pytest
+
+from py.recipes.parsers.automatic import AutomaticMetadataParser
+
+
+@pytest.mark.asyncio
+async def test_parse_metadata_extracts_checkpoint_from_civitai_resources(monkeypatch):
+    checkpoint_info = {
+        "id": 2442439,
+        "modelId": 123456,
+        "model": {"name": "Z Image", "type": "checkpoint"},
+        "name": "Turbo",
+        "images": [{"url": "https://image.civitai.com/checkpoints/original=true"}],
+        "baseModel": "sdxl",
+        "downloadUrl": "https://civitai.com/api/download/checkpoint",
+        "files": [
+            {
+                "type": "Model",
+                "primary": True,
+                "sizeKB": 2048,
+                "name": "Z_Image_Turbo.safetensors",
+                "hashes": {"SHA256": "ABC123FF"},
+            }
+        ],
+    }
+
+    async def fake_metadata_provider():
+        class Provider:
+            async def get_model_version_info(self, version_id):
+                assert version_id == "2442439"
+                return checkpoint_info, None
+
+        return Provider()
+
+    monkeypatch.setattr(
+        "py.recipes.parsers.automatic.get_default_metadata_provider",
+        fake_metadata_provider,
+    )
+
+    parser = AutomaticMetadataParser()
+
+    metadata_text = (
+        "Negative space, fog, BLACK blue color GRADIENT BACKGROUND, a vintage car in the middle, "
+        "FOG, and a silhouetted figure near the car, in the style of the Blade Runner movie "
+        "Negative prompt: Steps: 23, Sampler: Undefined, CFG scale: 3.5, Seed: 1760020955, "
+        "Size: 832x1216, Clip skip: 2, Created Date: 2025-11-28T09:18:43.5269343Z, "
+        'Civitai resources: [{"type":"checkpoint","modelVersionId":2442439,"modelName":"Z Image","modelVersionName":"Turbo"}], '
+        "Civitai metadata: {}"
+    )
+
+    result = await parser.parse_metadata(metadata_text)
+
+    checkpoint = result.get("checkpoint")
+    assert checkpoint is not None
+    assert checkpoint["name"] == "Z Image"
+    assert checkpoint["version"] == "Turbo"
+    assert checkpoint["type"] == "checkpoint"
+    assert checkpoint["modelId"] == 123456
+    assert checkpoint["hash"] == "abc123ff"
+    assert checkpoint["file_name"] == "Z_Image_Turbo"
+    assert checkpoint["thumbnailUrl"].endswith("width=450,optimized=true")
+    assert result["model"] == checkpoint
+    assert result["base_model"] == "sdxl"
+    assert result["loras"] == []
+
+
+@pytest.mark.asyncio
+async def test_parse_metadata_extracts_checkpoint_from_model_hash(monkeypatch):
+    checkpoint_info = {
+        "id": 98765,
+        "modelId": 654321,
+        "model": {"name": "Flux Illustrious", "type": "checkpoint"},
+        "name": "v1",
+        "images": [{"url": "https://image.civitai.com/checkpoints/original=true"}],
+        "baseModel": "flux",
+        "downloadUrl": "https://civitai.com/api/download/checkpoint",
+        "files": [
+            {
+                "type": "Model",
+                "primary": True,
+                "sizeKB": 1024,
+                "name": "FluxIllustrious_v1.safetensors",
+                "hashes": {"SHA256": "C3688EE04C"},
+            }
+        ],
+    }
+
+    async def fake_metadata_provider():
+        class Provider:
+            async def get_model_by_hash(self, model_hash):
+                assert model_hash == "c3688ee04c"
+                return checkpoint_info, None
+
+        return Provider()
+
+    monkeypatch.setattr(
+        "py.recipes.parsers.automatic.get_default_metadata_provider",
+        fake_metadata_provider,
+    )
+
+    parser = AutomaticMetadataParser()
+
+    metadata_text = (
+        "A cyberpunk portrait with neon highlights.\n"
+        "Negative prompt: low quality\n"
+        "Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 123456, Size: 832x1216, "
+        "Model hash: c3688ee04c, Model: models/waiNSFWIllustrious_v110.safetensors"
+    )
+
+    result = await parser.parse_metadata(metadata_text)
+
+    checkpoint = result.get("checkpoint")
+    assert checkpoint is not None
+    assert checkpoint["hash"] == "c3688ee04c"
+    assert checkpoint["name"] == "Flux Illustrious"
+    assert checkpoint["version"] == "v1"
+    assert checkpoint["file_name"] == "FluxIllustrious_v1"
+    assert result["model"] == checkpoint
+    assert result["base_model"] == "flux"
+    assert result["loras"] == []

tests/services/test_meta_format_parser.py

@@ -0,0 +1,61 @@
+import pytest
+
+from py.recipes.parsers.meta_format import MetaFormatParser
+
+
+@pytest.mark.asyncio
+async def test_meta_format_parser_extracts_checkpoint_from_model_hash(monkeypatch):
+    checkpoint_info = {
+        "id": 222333,
+        "modelId": 999888,
+        "model": {"name": "Fluxmania V5P", "type": "checkpoint"},
+        "name": "v5p",
+        "images": [{"url": "https://image.civitai.com/checkpoints/original=true"}],
+        "baseModel": "flux",
+        "downloadUrl": "https://civitai.com/api/download/checkpoint",
+        "files": [
+            {
+                "type": "Model",
+                "primary": True,
+                "sizeKB": 1024,
+                "name": "Fluxmania_V5P.safetensors",
+                "hashes": {"SHA256": "8AE0583B06"},
+            }
+        ],
+    }
+
+    async def fake_metadata_provider():
+        class Provider:
+            async def get_model_by_hash(self, model_hash):
+                assert model_hash == "8ae0583b06"
+                return checkpoint_info, None
+
+        return Provider()
+
+    monkeypatch.setattr(
+        "py.recipes.parsers.meta_format.get_default_metadata_provider",
+        fake_metadata_provider,
+    )
+
+    parser = MetaFormatParser()
+
+    metadata_text = (
+        "Shimmering metal forms\n"
+        "Negative prompt: flat color\n"
+        "Steps: 25, Sampler: dpmpp_2m_sgm_uniform, Seed: 471889513588087, "
+        "Model: Fluxmania V5P.safetensors, Model hash: 8ae0583b06, VAE: ae.sft, "
+        "Lora_0 Model name: ArtVador I.safetensors, Lora_0 Model hash: 08f7133a58, "
+        "Lora_0 Strength model: 0.65, Lora_0 Strength clip: 0.65"
+    )
+
+    result = await parser.parse_metadata(metadata_text)
+
+    checkpoint = result.get("checkpoint")
+    assert checkpoint is not None
+    assert checkpoint["hash"] == "8ae0583b06"
+    assert checkpoint["name"] == "Fluxmania V5P"
+    assert checkpoint["version"] == "v5p"
+    assert checkpoint["file_name"] == "Fluxmania_V5P"
+    assert result["model"] == checkpoint
+    assert result["base_model"] == "flux"
+    assert len(result["loras"]) == 1

tests/services/test_recipe_format_parser.py

@@ -0,0 +1,67 @@
+import json
+import pytest
+
+from py.recipes.parsers.recipe_format import RecipeFormatParser
+
+
+@pytest.mark.asyncio
+async def test_recipe_format_parser_populates_checkpoint(monkeypatch):
+    checkpoint_info = {
+        "id": 777111,
+        "modelId": 333222,
+        "model": {"name": "Z Image", "type": "checkpoint"},
+        "name": "Turbo",
+        "images": [{"url": "https://image.civitai.com/checkpoints/original=true"}],
+        "baseModel": "sdxl",
+        "downloadUrl": "https://civitai.com/api/download/checkpoint",
+        "files": [
+            {
+                "type": "Model",
+                "primary": True,
+                "sizeKB": 2048,
+                "name": "Z_Image_Turbo.safetensors",
+                "hashes": {"SHA256": "ABC123FF"},
+            }
+        ],
+    }
+
+    async def fake_metadata_provider():
+        class Provider:
+            async def get_model_version_info(self, version_id):
+                assert version_id == "777111"
+                return checkpoint_info, None
+
+        return Provider()
+
+    monkeypatch.setattr(
+        "py.recipes.parsers.recipe_format.get_default_metadata_provider",
+        fake_metadata_provider,
+    )
+
+    parser = RecipeFormatParser()
+
+    recipe_metadata = {
+        "title": "Z Recipe",
+        "base_model": "",
+        "loras": [],
+        "gen_params": {"steps": 20},
+        "tags": ["test"],
+        "checkpoint": {
+            "modelVersionId": 777111,
+            "modelId": 333222,
+            "name": "Z Image",
+            "version": "Turbo",
+        },
+    }
+
+    metadata_text = f"Recipe metadata: {json.dumps(recipe_metadata)}"
+    result = await parser.parse_metadata(metadata_text)
+
+    checkpoint = result.get("checkpoint")
+    assert checkpoint is not None
+    assert checkpoint["name"] == "Z Image"
+    assert checkpoint["version"] == "Turbo"
+    assert checkpoint["hash"] == "abc123ff"
+    assert checkpoint["file_name"] == "Z_Image_Turbo"
+    assert result["base_model"] == "sdxl"
+    assert result["model"] == checkpoint