lutaml 0.10.10 → 0.10.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f6bf7ec9ab56a8d2d17b70b191ee22dda71459f5c0e00f6066aa499a26427cde
-  data.tar.gz: a5b2f9ed9ac2f89b8bb76b65e0b648e41c42fd20fd94e422cc8a90403683c430
+  metadata.gz: e6c2a2164de4ea6ec27e076a4724d35d573af91829fa0103fc3e129707597de8
+  data.tar.gz: 811651ff02e1b7001d056f70b3f9e96bd52c227a67d9e943134816b3408dc321
 SHA512:
-  metadata.gz: 112cf5de38571225fd17e354dc1c4bd2debd878841f2681423c7e616176e6bd93a8935f7d504c5be181b9f4d3a6ad5220e3ff102def9753796c821e899576e65
-  data.tar.gz: e7afc11d364ebfc3be4f651bb2fd389da30256504877209ee76154483be9b0525ca6a2dbd4fe2242a61d56447818030e304c31da02fdbd2b0851a7ff13d4d788
+  metadata.gz: '086fe280c4b5555a9e03d3718c5976bfbaaa7dbc9a7d4722fd6557e7779b6a673e372c5ea5e26cf9272c768067a8663672e06a2b3914bc6bbba459568a85faa5'
+  data.tar.gz: 49c1c37825b6e3390b696a30584998c29433dd182bb2dfbf733985fc960306913d7ed2706b38959ba9766ec2d6458f2e552180a904fef22d655c409acb276183

data/.rubocop_todo.yml

@@ -1,30 +1,18 @@
 # This configuration was generated by
 # `rubocop --auto-gen-config`
-# on 2026-05-11 04:25:12 UTC using RuboCop version 1.86.1.
+# on 2026-05-11 22:46:59 UTC using RuboCop version 1.86.1.
 # The point is for the user to remove these configuration records
 # one by one as the offenses are removed from the code base.
 # Note that changes in the inspected code, or installation of new
 # versions of RuboCop, may require this file to be generated again.
 
-# Offense count: 1
-# This cop supports safe autocorrection (--autocorrect).
-# Configuration parameters: IndentationWidth.
-Layout/AssignmentIndentation:
-  Exclude:
-    - 'lib/lutaml/ea/diagram/style_resolver.rb'
-
-# Offense count: 1
-# This cop supports safe autocorrection (--autocorrect).
-# Configuration parameters: EmptyLineBetweenMethodDefs, EmptyLineBetweenClassDefs, EmptyLineBetweenModuleDefs, DefLikeMacros, AllowAdjacentOneLineDefs, NumberOfEmptyLines.
-Layout/EmptyLineBetweenDefs:
-  Exclude:
-    - 'lib/lutaml/ea/diagram/style_parser.rb'
-
-# Offense count: 1
+# Offense count: 5
 # This cop supports safe autocorrection (--autocorrect).
-Layout/EmptyLines:
+# Configuration parameters: EnforcedStyle, IndentationWidth.
+# SupportedStyles: with_first_argument, with_fixed_indentation
+Layout/ArgumentAlignment:
   Exclude:
-    - 'lib/lutaml/ea/diagram/style_parser.rb'
+    - 'spec/lutaml/xmi/liquid_drops/nil_safety_spec.rb'
 
 # Offense count: 159
 # This cop supports safe autocorrection (--autocorrect).
@@ -33,12 +21,12 @@ Layout/EmptyLines:
 Layout/LineLength:
   Enabled: false
 
-# Offense count: 1
+# Offense count: 5
 # This cop supports safe autocorrection (--autocorrect).
 # Configuration parameters: AllowInHeredoc.
 Layout/TrailingWhitespace:
   Exclude:
-    - 'lib/lutaml/ea/diagram/style_resolver.rb'
+    - 'spec/lutaml/xmi/liquid_drops/nil_safety_spec.rb'
 
 # Offense count: 1
 Lint/BinaryOperatorWithIdenticalOperands:
@@ -115,7 +103,7 @@ Metrics/AbcSize:
 Metrics/BlockLength:
   Max: 53
 
-# Offense count: 33
+# Offense count: 35
 # Configuration parameters: AllowedMethods, AllowedPatterns, Max.
 Metrics/CyclomaticComplexity:
   Enabled: false
@@ -125,7 +113,7 @@ Metrics/CyclomaticComplexity:
 Metrics/MethodLength:
   Max: 62
 
-# Offense count: 25
+# Offense count: 26
 # Configuration parameters: AllowedMethods, AllowedPatterns, Max.
 Metrics/PerceivedComplexity:
   Enabled: false
@@ -139,13 +127,14 @@ Naming/MethodParameterName:
     - 'lib/lutaml/uml_repository/static_site/generator.rb'
     - 'spec/lutaml/parsers/serialize_xmi_to_liquid_spec.rb'
 
-# Offense count: 1
+# Offense count: 3
 # Configuration parameters: Mode, AllowedMethods, AllowedPatterns, AllowBangMethods, WaywardPredicates.
 # AllowedMethods: call
 # WaywardPredicates: infinite?, nonzero?
 Naming/PredicateMethod:
   Exclude:
     - 'lib/lutaml/qea/factory/document_builder.rb'
+    - 'spec/lutaml/xmi/liquid_drops/nil_safety_spec.rb'
 
 # Offense count: 3
 # Configuration parameters: MinSize.
@@ -190,7 +179,7 @@ RSpec/ContextWording:
     - 'spec/lutaml/qea/integration/full_parsing_spec.rb'
     - 'spec/lutaml/xml/parsers/xml_spec.rb'
 
-# Offense count: 21
+# Offense count: 22
 # Configuration parameters: IgnoredMetadata.
 RSpec/DescribeClass:
   Enabled: false
@@ -253,12 +242,12 @@ RSpec/MessageSpies:
 RSpec/MultipleExpectations:
   Max: 7
 
-# Offense count: 151
+# Offense count: 149
 # Configuration parameters: AllowSubject.
 RSpec/MultipleMemoizedHelpers:
   Max: 11
 
-# Offense count: 35
+# Offense count: 33
 # Configuration parameters: AllowedGroups.
 RSpec/NestedGroups:
   Max: 4
@@ -334,10 +323,3 @@ Style/SafeNavigationChainLength:
   Exclude:
     - 'lib/lutaml/qea/validation/attribute_validator.rb'
     - 'lib/lutaml/qea/validation/operation_validator.rb'
-
-# Offense count: 3
-# This cop supports unsafe autocorrection (--autocorrect-all).
-# Configuration parameters: AllowSend.
-Style/SendWithLiteralMethodName:
-  Exclude:
-    - 'lib/lutaml/ea/diagram/style_resolver.rb'

data/lib/lutaml/cli/enhanced_formatter.rb

@@ -125,7 +125,7 @@ module Lutaml
       # @param klass [Object] Class object to display
       # @param path_formatter [Proc] Formatter for package paths
       # @return [String] Enhanced class details
-      def self.format_class_details_enhanced(klass, path_formatter = nil) # rubocop:disable Metrics/AbcSize,Metrics/CyclomaticComplexity,Metrics/MethodLength,Metrics/PerceivedComplexity
+      def self.format_class_details_enhanced(klass, _path_formatter = nil) # rubocop:disable Metrics/AbcSize,Metrics/CyclomaticComplexity,Metrics/MethodLength,Metrics/PerceivedComplexity
         lines = []
 
         # Header box
@@ -137,25 +137,21 @@ module Lutaml
 
         # Basic information
         lines << colorize("Basic Information:", :cyan)
-        if klass.respond_to?(:xmi_id)
+        if klass.is_a?(Lutaml::Uml::TopElement) && klass.xmi_id
           lines << "  XMI ID:      #{klass.xmi_id}"
         end
-        if klass.respond_to?(:stereotype) && klass.stereotype && !klass.stereotype.empty?
+        if klass.is_a?(Lutaml::Uml::TopElement) && klass.stereotype && !klass.stereotype.empty?
           st = klass.stereotype
           st_str = st.is_a?(Array) ? st.join(", ") : st
           lines << "  Stereotype:  #{st_str}"
         end
-        if klass.respond_to?(:is_abstract)
+        if klass.is_a?(Lutaml::Uml::Classifier)
           lines << "  Abstract:    #{klass.is_abstract ? 'Yes' : 'No'}"
         end
-        if path_formatter && klass.respond_to?(:package)
-          package_path = path_formatter.call(klass.package)
-          lines << "  Package:     #{ICONS[:package]} #{package_path}"
-        end
         lines << ""
 
         # Attributes
-        if klass.respond_to?(:attributes) && klass.attributes &&
+        if klass.is_a?(Lutaml::Uml::Classifier) && klass.attributes &&
             !klass.attributes.empty?
           lines << colorize(
             "#{ICONS[:attribute]} Attributes (#{klass.attributes.size}):",
@@ -173,21 +169,21 @@ module Lutaml
         end
 
         # Operations
-        if klass.respond_to?(:operations) && klass.operations &&
+        if klass.is_a?(Lutaml::Uml::Classifier) && klass.operations &&
             !klass.operations.empty?
           lines << colorize(
             "#{ICONS[:operation]} Operations (#{klass.operations.size}):",
             :yellow,
           )
           klass.operations.each do |op|
-            params = if op.respond_to?(:parameters) && op.parameters
-                       op.parameters.map do |p|
+            params = if op.owned_parameter
+                       op.owned_parameter.map do |p|
                          "#{p.name}: #{p.type}"
                        end.join(", ")
                      else
                        ""
                      end
-            return_type = if op.respond_to?(:return_type) && op.return_type
+            return_type = if op.return_type
                             " : #{op.return_type}"
                           else
                             ""
@@ -486,18 +482,13 @@ module Lutaml
       #
       # @param attr [Object] Attribute with cardinality
       # @return [String] Formatted cardinality
-      def self.format_cardinality(attr) # rubocop:disable Metrics/CyclomaticComplexity,Metrics/PerceivedComplexity
-        return "" unless attr.respond_to?(:cardinality)
+      def self.format_cardinality(attr)
         return "" unless attr.cardinality
 
         card = attr.cardinality
-        if card.respond_to?(:min) && card.respond_to?(:max)
-          min = card.min || "0"
-          max = card.max || "*"
-          "[#{min}..#{max}]"
-        else
-          ""
-        end
+        min = card.min || "0"
+        max = card.max || "*"
+        "[#{min}..#{max}]"
       end
     end
   end

data/lib/lutaml/cli/interactive_shell/query_commands.rb

@@ -99,7 +99,7 @@ module Lutaml
             puts ""
             puts "Name: #{cls.name}"
 
-            if cls.respond_to?(:attributes) && cls.attributes && !cls.attributes.empty?
+            if cls.is_a?(Lutaml::Uml::Classifier) && cls.attributes && !cls.attributes.empty?
               puts ""
               puts OutputFormatter.colorize("Attributes:", :yellow)
               cls.attributes.each do |attr|

data/lib/lutaml/cli/output_formatter.rb

@@ -69,7 +69,7 @@ module Lutaml
           data
         else
           # For complex objects, convert to string representation
-          data.respond_to?(:name) ? data.name : data.to_s
+          data.is_a?(Lutaml::Uml::TopElement) ? data.name : data.to_s
         end
       end
 
@@ -271,7 +271,7 @@ module Lutaml
           stats[:most_complex_classes].first(5).each do |item|
             klass = item[:class]
             complexity = item[:complexity]
-            class_name = klass.respond_to?(:name) ? klass.name : klass.to_s
+            class_name = klass&.name || klass.to_s
 
             lines << "  #{class_name} (complexity: #{complexity})"
           end

data/lib/lutaml/cli/tree_view_formatter.rb

@@ -163,7 +163,7 @@ module Lutaml
         children = []
 
         # Add attributes
-        if @show_attributes && klass.respond_to?(:attributes) &&
+        if @show_attributes && klass.is_a?(Lutaml::Uml::Classifier) &&
             klass.attributes
           children.concat(klass.attributes.map do |a|
             { type: :attribute, obj: a }
@@ -171,7 +171,7 @@ module Lutaml
         end
 
         # Add operations
-        if @show_operations && klass.respond_to?(:operations) &&
+        if @show_operations && klass.is_a?(Lutaml::Uml::Classifier) &&
             klass.operations
           children.concat(klass.operations.map do |o|
             { type: :operation, obj: o }
@@ -239,11 +239,7 @@ module Lutaml
         return "" if @max_depth && depth >= @max_depth
 
         assoc_name = assoc.name || "(unnamed)"
-        target = if assoc.respond_to?(:member_end) && assoc.member_end&.first
-                   assoc.member_end.first.xmi_type || "Unknown"
-                 else
-                   "Unknown"
-                 end
+        target = assoc.member_end || "Unknown"
 
         connector = is_last ? TREE_CHARS[:last_branch] : TREE_CHARS[:branch]
         icon = ICONS[:association]
@@ -280,7 +276,7 @@ module Lutaml
       # Determine class type (class, interface, enumeration)
       def determine_class_type(klass)
         return :enumeration if klass.class.name&.include?("Enum")
-        return :interface if klass.respond_to?(:stereotype) &&
+        return :interface if klass.is_a?(Lutaml::Uml::TopElement) &&
           Array(klass.stereotype).any? { |s| s&.downcase == "interface" }
 
         :class

data/lib/lutaml/cli/uml/build_command.rb

@@ -141,7 +141,7 @@ module Lutaml
             OutputFormatter.progress_done
 
             # Display verbose validation if requested
-            if options[:verbose] && result.respond_to?(:validation_details)
+            if options[:verbose] && result.validation_details
               display_verbose_validation(result.validation_details)
             end
 
@@ -149,8 +149,7 @@ module Lutaml
               handle_validation_result(result)
 
               # Display unique unresolved types if present
-              if result.respond_to?(:external_references) &&
-                  result.external_references.any?
+              if result.external_references&.any?
                 display_unresolved_types(result.external_references)
               end
 

data/lib/lutaml/cli/uml/diagram_command.rb

@@ -83,12 +83,12 @@ module Lutaml
             }
 
             # Add stereotype if available
-            if uml_element.respond_to?(:stereotype) && uml_element.stereotype && !uml_element.stereotype.empty?
+            if uml_element.is_a?(Lutaml::Uml::TopElement) && uml_element.stereotype && !uml_element.stereotype.empty?
               element_data[:stereotype] = uml_element.stereotype.first
             end
 
             # Add attributes and operations for classes
-            if uml_element.respond_to?(:attributes) && uml_element.attributes
+            if uml_element.is_a?(Lutaml::Uml::Classifier) && uml_element.attributes
               element_data[:attributes] = uml_element.attributes.map do |attr|
                 {
                   name: attr.name,
@@ -98,7 +98,7 @@ module Lutaml
               end
             end
 
-            if uml_element.respond_to?(:operations) && uml_element.operations
+            if uml_element.is_a?(Lutaml::Uml::Classifier) && uml_element.operations
               element_data[:operations] = uml_element.operations.map do |op|
                 {
                   name: op.name,
@@ -131,30 +131,30 @@ module Lutaml
             }
 
             # Add role and multiplicity information if available
-            if connector.respond_to?(:owner_end_attribute_name) &&
+            if connector.is_a?(Lutaml::Uml::Association) &&
                 connector.owner_end_attribute_name
               connector_data[:source_role] = connector.owner_end_attribute_name
             end
 
-            if connector.respond_to?(:member_end_attribute_name) &&
+            if connector.is_a?(Lutaml::Uml::Association) &&
                 connector.member_end_attribute_name
               connector_data[:target_role] = connector.member_end_attribute_name
             end
 
-            if connector.respond_to?(:owner_end_cardinality) &&
+            if connector.is_a?(Lutaml::Uml::Association) &&
                 connector.owner_end_cardinality
               connector_data[:source_multiplicity] =
                 format_cardinality(connector.owner_end_cardinality)
             end
 
-            if connector.respond_to?(:member_end_cardinality) &&
+            if connector.is_a?(Lutaml::Uml::Association) &&
                 connector.member_end_cardinality
               connector_data[:target_multiplicity] =
                 format_cardinality(connector.member_end_cardinality)
             end
 
             # Add source and target information if available
-            if connector.respond_to?(:source) && connector.source
+            if connector.is_a?(Lutaml::Uml::Association) && connector.owner_end
               source_obj = find_diagram_object_for_element(
                 connector.source.xmi_id, diagram
               )
@@ -166,7 +166,7 @@ module Lutaml
               end
             end
 
-            if connector.respond_to?(:target) && connector.target
+            if connector.is_a?(Lutaml::Uml::Association) && connector.member_end
               target_obj = find_diagram_object_for_element(
                 connector.target.xmi_id, diagram
               )
@@ -405,11 +405,7 @@ module Lutaml
         def format_cardinality(cardinality)
           return "" unless cardinality
 
-          if cardinality.respond_to?(:to_s)
-            cardinality.to_s
-          else
-            ""
-          end
+          cardinality.to_s
         end
       end
     end

data/lib/lutaml/cli/uml/find_command.rb

@@ -84,7 +84,7 @@ module Lutaml
           end
 
           output = classes.map do |cls|
-            cls.respond_to?(:name) ? (cls.name || cls.to_s) : cls.to_s
+            cls.is_a?(Lutaml::Uml::TopElement) ? cls.name : cls.to_s
           end
 
           # output result based on the format option

data/lib/lutaml/ea/diagram/configuration.rb

@@ -47,14 +47,14 @@ module Lutaml
           value = nil
 
           # 1. Try class-specific override (highest priority)
-          if element.respond_to?(:name) && element.name
+          if element.is_a?(Lutaml::Uml::TopElement) && element.name
             class_config = dig_config("classes.#{element.name}.#{property}")
             value = class_config if class_config
           end
 
           # 2. Try package-based styling
           if !value &&
-              element.respond_to?(:package_name) && element.package_name
+              element.is_a?(Lutaml::Uml::Diagram) && element.package_name
             # Support wildcards: "CityGML::*"
             package_configs = config_data["packages"] || {}
             package_configs.each do |pattern, pkg_config|
@@ -67,7 +67,7 @@ module Lutaml
           end
 
           # 3. Try stereotype-based styling
-          if !value && element.respond_to?(:stereotype) && element.stereotype
+          if !value && element.is_a?(Lutaml::Uml::TopElement) && element.stereotype
             stereotypes = Array(element.stereotype)
             stereotypes.each do |stereo|
               stereo_value = dig_config("stereotypes.#{stereo}.#{property}")

data/lib/lutaml/ea/diagram/style_resolver.rb

@@ -264,23 +264,15 @@ module Lutaml
         #
         # @param connector [Object] Association connector
         # @return [String] Specific association type
-        def determine_association_type(connector) # rubocop:disable Metrics/CyclomaticComplexity,Metrics/MethodLength
-          return "association" unless connector.respond_to?(:member_end)
-          return "association" unless connector.member_end
-
-          # Ensure member_end is an array (handle legacy data
-          # where it might be a string)
-          member_ends = Array(connector.member_end)
-
-          # Check for aggregation or composition
-          member_ends.each do |end_point|
-            if end_point.respond_to?(:aggregation)
-              case end_point.aggregation&.downcase
-              when "shared"
-                return "aggregation"
-              when "composite"
-                return "composition"
-              end
+        def determine_association_type(connector)
+          return "association" unless connector.is_a?(Lutaml::Uml::Association)
+
+          [connector.owner_end_type, connector.member_end_type].each do |type|
+            case type&.downcase
+            when "aggregation"
+              return "aggregation"
+            when "composition"
+              return "composition"
             end
           end
 

data/lib/lutaml/model_transformations/format_registry.rb

@@ -63,7 +63,7 @@ module Lutaml
       # @return [void]
       def auto_register_from_parser(parser_class)
         supported_extensions = ""
-        if parser_class.new.respond_to?(:supported_extensions)
+        if parser_class.method_defined?(:supported_extensions)
           supported_extensions = parser_class.new.supported_extensions
         end
         register(supported_extensions, parser_class)
@@ -255,7 +255,7 @@ module Lutaml
 
           # Check header match for content patterns
           @parsers.each do |ext, parser_class|
-            if parser_class.new.respond_to? :content_patterns
+            if parser_class.method_defined?(:content_patterns)
               parser_klass = parser_class.new
               parser_klass.content_patterns.each do |pattern|
                 if header.match?(pattern)

data/lib/lutaml/model_transformations/parsers/base_parser.rb

@@ -107,7 +107,7 @@ module Lutaml
           extension = File.extname(file_path).downcase
           return true if supported_extensions.include?(extension)
 
-          if respond_to?(:content_patterns) && File.exist?(file_path)
+          if self.class.method_defined?(:content_patterns) && File.exist?(file_path)
             File.open(file_path, "rb") do |file|
               header = file.read(1024) # Read first 1KB
               return false if header.nil? || header.empty?

data/lib/lutaml/model_transformations/parsers/qea_parser.rb

@@ -269,16 +269,16 @@ module Lutaml
         # @return [void]
         def post_process_qea_document(document, file_path)
           # Set QEA-specific source information
-          if document.respond_to?(:source_file=)
+          if document.class.method_defined?(:source_file=)
             document.source_file = file_path
           end
 
-          if document.respond_to?(:source_format=)
+          if document.class.method_defined?(:source_format=)
             document.source_format = "QEA"
           end
 
           # Store database statistics
-          if document.respond_to?(:database_stats=)
+          if document.class.method_defined?(:database_stats=)
             document.database_stats = @database_stats
           end
         end

data/lib/lutaml/model_transformations/parsers/xmi_parser.rb

@@ -138,12 +138,12 @@ module Lutaml
         # @return [void]
         def post_process_xmi_document(document, file_path)
           # Set source file information
-          if document.respond_to?(:source_file=)
+          if document.class.method_defined?(:source_file=)
             document.source_file = file_path
           end
 
           # Add timestamp
-          if document.respond_to?(:parsed_at=)
+          if document.class.method_defined?(:parsed_at=)
             document.parsed_at = Time.now
           end
 

data/lib/lutaml/model_transformations/transformation_engine.rb

@@ -256,7 +256,7 @@ module Lutaml
         @format_registry.all_parsers.each_value do |parser_class|
           # Try to create instance to validate
           parser = parser_class.new(configuration: @configuration)
-          if parser.respond_to?(:parse)
+          if parser.class.method_defined?(:parse)
             results[:parsers_loaded] += 1
           else
             results[:parser_errors] << "Parser #{parser_class} does not " \

data/lib/lutaml/qea/factory/ea_to_uml_factory.rb

@@ -244,7 +244,8 @@ module Lutaml
         def collect_package_associations(package, associations) # rubocop:disable Metrics/CyclomaticComplexity
           # Collect from classes in this package
           package.classes&.each do |klass|
-            if klass.respond_to?(:associations) && klass.associations
+            if (klass.is_a?(Lutaml::Uml::Class) ||
+                klass.is_a?(Lutaml::Uml::DataType)) && klass.associations
               associations.concat(klass.associations)
             end
           end

data/lib/lutaml/qea/validation/base_validator.rb

@@ -209,7 +209,10 @@ module Lutaml
         # @param value [Object] Value to check
         # @return [Boolean]
         def present?(value)
-          !value.nil? && (value.respond_to?(:empty?) ? !value.empty? : true)
+          return false if value.nil?
+          return !value.empty? if value.is_a?(String) || value.is_a?(Array)
+
+          true
         end
 
         # Finds entity name from database

data/lib/lutaml/qea/validation/validation_engine.rb

@@ -223,11 +223,7 @@ module Lutaml
           # Note: Use document entities for primary validation
           if @database
             context[:db_packages] = @database.packages || []
-            context[:db_objects] = if @database.objects.respond_to?(:all)
-                                     @database.objects.all
-                                   else
-                                     @database.objects || []
-                                   end
+            context[:db_objects] = @database.objects.all
             context[:attributes] = @database.attributes || []
             context[:operations] = @database.operations || []
             context[:connectors] = @database.connectors || []

data/lib/lutaml/qea/verification/document_normalizer.rb

@@ -73,7 +73,7 @@ module Lutaml
         # Process packages recursively to remove XMI IDs
         def process_packages(packages) # rubocop:disable Metrics/AbcSize,Metrics/CyclomaticComplexity
           packages.each do |package|
-            package.xmi_id = nil if package.respond_to?(:xmi_id=)
+            package.xmi_id = nil
             process_classes(package.classes) if package.classes
             process_enums(package.enums) if package.enums
             process_data_types(package.data_types) if package.data_types
@@ -84,23 +84,23 @@ module Lutaml
         # Process classes to remove XMI IDs
         def process_classes(classes) # rubocop:disable Metrics/AbcSize,Metrics/CyclomaticComplexity,Metrics/MethodLength,Metrics/PerceivedComplexity
           classes.each do |klass|
-            klass.xmi_id = nil if klass.respond_to?(:xmi_id=)
+            klass.xmi_id = nil
 
             # Remove XMI IDs from attributes
             klass.attributes&.each do |attr|
-              attr.xmi_id = nil if attr.respond_to?(:xmi_id=)
+              attr.xmi_id = nil
             end
 
             # Remove XMI IDs from operations
             klass.operations&.each do |op|
-              op.xmi_id = nil if op.respond_to?(:xmi_id=)
+              op.xmi_id = nil
               op.parameters&.each do |param|
-                param.xmi_id = nil if param.respond_to?(:xmi_id=)
+                param.xmi_id = nil if param.class.attributes.key?(:xmi_id)
               end
             end
 
             # Process nested classes
-            if klass.respond_to?(:classes) && klass.classes
+            if klass.is_a?(Lutaml::Uml::Package) && klass.classes
               process_classes(klass.classes)
             end
           end
@@ -109,11 +109,11 @@ module Lutaml
         # Process associations to remove XMI IDs
         def process_associations(associations)
           associations.each do |assoc|
-            assoc.xmi_id = nil if assoc.respond_to?(:xmi_id=)
-            if assoc.respond_to?(:owner_end_xmi_id=)
+            assoc.xmi_id = nil
+            if assoc.owner_end_xmi_id
               assoc.owner_end_xmi_id = nil
             end
-            if assoc.respond_to?(:member_end_xmi_id=)
+            if assoc.member_end_xmi_id
               assoc.member_end_xmi_id = nil
             end
           end
@@ -122,14 +122,14 @@ module Lutaml
         # Process enums to remove XMI IDs
         def process_enums(enums)
           enums.each do |enum|
-            enum.xmi_id = nil if enum.respond_to?(:xmi_id=)
+            enum.xmi_id = nil
           end
         end
 
         # Process data types to remove XMI IDs
         def process_data_types(data_types)
           data_types.each do |dt|
-            dt.xmi_id = nil if dt.respond_to?(:xmi_id=)
+            dt.xmi_id = nil
           end
         end