Update src/aibom_generator/generator.py

src/aibom_generator/generator.py

@@ -97,6 +97,7 @@ class AIBOMGenerator:
                 "downloads": model_info.downloads if hasattr(model_info, "downloads") else 0,
                 "last_modified": model_info.lastModified if hasattr(model_info, "lastModified") else None,
                 "commit": model_info.sha[:7] if hasattr(model_info, "sha") and model_info.sha else None,
+                "commit_url": f"https://huggingface.co/{model_id}/commit/{model_info.sha}" if hasattr(model_info, "sha") and model_info.sha else None,
             })
 
         if model_card and model_card.data:
@@ -189,10 +190,16 @@ class AIBOMGenerator:
         if "license" in metadata:
             component["licenses"] = [{"license": {"id": metadata["license"]}}]
 
-        component["externalReferences"] = [{
+        external_refs = [{
             "type": "website",
             "url": f"https://huggingface.co/{model_id}"
         }]
+        if "commit_url" in metadata:
+            external_refs.append({
+                "type": "vcs",
+                "url": metadata["commit_url"]
+            })
+        component["externalReferences"] = external_refs
 
         component["modelCard"] = self._create_model_card_section(metadata)