Improved datamodel and deterministic generation

src/repo_registry/core/models.py

@@ -161,6 +161,8 @@ class CandidateFeature:
     status: str
     source_refs: list[SourceReference]
     confidence_label: str = ""
+    primary_class: str = ""
+    attributes: list[str] = field(default_factory=list)
 
 
 @dataclass(frozen=True)
@@ -174,6 +176,8 @@ class CandidateCapability:
     status: str
     source_refs: list[SourceReference]
     confidence_label: str = ""
+    primary_class: str = "capability"
+    attributes: list[str] = field(default_factory=list)
     features: list[CandidateFeature] = field(default_factory=list)
     evidence: list[CandidateEvidence] = field(default_factory=list)
 
@@ -187,6 +191,8 @@ class CandidateAbility:
     status: str
     source_refs: list[SourceReference]
     confidence_label: str = ""
+    primary_class: str = "ability"
+    attributes: list[str] = field(default_factory=list)
     capabilities: list[CandidateCapability] = field(default_factory=list)
 
 
@@ -228,6 +234,8 @@ class Feature:
     confidence: float
     confidence_label: str = ""
     source_refs: list[SourceReference] = field(default_factory=list)
+    primary_class: str = ""
+    attributes: list[str] = field(default_factory=list)
 
 
 @dataclass(frozen=True)
@@ -239,6 +247,8 @@ class Capability:
     outputs: list[str]
     confidence: float
     confidence_label: str = ""
+    primary_class: str = "capability"
+    attributes: list[str] = field(default_factory=list)
     features: list[Feature] = field(default_factory=list)
     evidence: list[Evidence] = field(default_factory=list)
 
@@ -250,6 +260,8 @@ class Ability:
     description: str
     confidence: float
     confidence_label: str = ""
+    primary_class: str = "ability"
+    attributes: list[str] = field(default_factory=list)
     capabilities: list[Capability] = field(default_factory=list)