Update src/aibom_generator/generator.py

src/aibom_generator/generator.py

@@ -381,13 +381,31 @@ class AIBOMGenerator:
             return model_card_section
             
         try:
-            # Safely extract model parameters
-            model_parameters = {}
+            # Create a CycloneDX-compliant modelParameters structure
+            # Instead of adding custom properties directly, use a standardized structure
+            model_parameters = {
+                "description": "Model parameters and configuration"
+            }
+            
+            # Add properties array for custom parameters to ensure schema compliance
+            parameter_properties = []
+            
             for k in ["base_model", "library_name", "pipeline_tag"]:
-                if k in metadata:
-                    model_parameters[k] = metadata[k]
-                    
-            if model_parameters:
+                if k in metadata and metadata[k] is not None:
+                    try:
+                        if isinstance(metadata[k], (list, dict)):
+                            value = json.dumps(metadata[k])
+                        else:
+                            value = str(metadata[k])
+                        parameter_properties.append({
+                            "name": k,
+                            "value": value
+                        })
+                    except Exception as e:
+                        print(f"Error processing model parameter {k}: {e}")
+            
+            if parameter_properties:
+                model_parameters["properties"] = parameter_properties
                 model_card_section["modelParameters"] = model_parameters
 
             # Safely extract evaluation results