Update src/aibom_generator/generator.py

src/aibom_generator/generator.py

@@ -133,6 +133,8 @@ class AIBOMGenerator:
         return {}
 
     def _create_metadata_section(self, model_id: str, metadata: Dict[str, Any]) -> Dict[str, Any]:
+        from utils import calculate_completeness_score  # Assumes utils.py is in same module
+        completeness_score = calculate_completeness_score({"metadata": metadata})
         timestamp = datetime.datetime.utcnow().isoformat() + "Z"
         tools = [{
             "vendor": "Aetheris AI",
@@ -160,6 +162,8 @@ class AIBOMGenerator:
                     value = json.dumps(value)
                 properties.append({"name": key, "value": str(value)})
 
+                properties.append({"name": "aibom:quality-score", "value": str(completeness_score)})
+
         metadata_section = {
             "timestamp": timestamp,
             "tools": tools,