lib/search_kit/models/document.rb in search-kit-0.0.6 vs lib/search_kit/models/document.rb in search-kit-0.0.7

@@ -8,15 +8,13 @@
       attribute :id
       attribute :source
       attribute :score
 
       def initialize(document_data = {})
-        attributes = document_data.fetch(:attributes, {})
-
         super(
-          source: attributes,
-          id:     attributes.fetch(:id, nil),
-          score:  attributes.fetch(:score, nil)
+          source: document_data.fetch(:source, {}),
+          id:     document_data.fetch(:id, nil),
+          score:  document_data.fetch(:score, nil)
         )
       end
 
       def get(field)
         source.fetch(field.to_sym)