wit_bot 0.4.1 → 0.4.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 1e387a08441a8c5f9c9b1fa391cd8ac4b5a1edb9
-  data.tar.gz: 6d49997d4fff960c07f92ed6200cbb72814d1dc9
+  metadata.gz: c1b7c52f2b1b3dcb990efec623c7b8937d070863
+  data.tar.gz: c4fac73cffe1125374cfd876f4cbae198b2fcd00
 SHA512:
-  metadata.gz: fa5a966a259be1ad435f06600a0b15a17b18749047625ce923192e3cceebbbb32607323df8091f20e5e069859eb79870dff82c29dfe824628bc62905600bb6a7
-  data.tar.gz: 7d5ae281e07a6382491a7a830f9001197712576dc21152f859d57bc6b0828e6efdfbd7d713c1e16bfa4575a77bd9d053ad503e3a572960b7383f3da6b10e0717
+  metadata.gz: 658ca8a99ce362a44beb1d233b1df2c6ee30e1ea0fbe27d002526af4f12c8b287fd258ba060c7b2db7ef8972dede09be7b9a16457df1bc285c24da2114d8961b
+  data.tar.gz: 2fe302edf51fe1bde5c5e1f5d6b719b11314b83fb58771b7575a873a200729ef0ab7c0494f92f088e2a74b0605ce08c5ff07b49faa80a4cfcc1f4d4d99ee38f8

data/lib/wit_bot/models/wit/entity/entity_model.rb

@@ -5,17 +5,19 @@ module WitBot
     def initialize(raw_entities)
       @raw = raw_entities
 
-      super Hash[*raw_entities.map do |(role, all)|
-        all = all.map { |data| WitModel::Entity.find(data[:entity], create: false).model role, data }
+      super Hash[raw_entities.map do |(role, all)|
+        all = all.map do |data|
+          entity_name = data[:entity] || role
+          entity = WitModel::Entity.find(entity_name, create: false)
+          entity.model role, data
+        end
 
-        data = all.first
-
-        data.all = all
+        data = all.length == 1 ? all.first : all
 
         create_getter role
 
         [role, data]
-      end.flatten]
+      end]
     end
 
     def method_missing(*_)

data/lib/wit_bot/version.rb

@@ -1,3 +1,3 @@
 module WitBot
-  VERSION = '0.4.1'
+  VERSION = '0.4.2'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: wit_bot
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.4.2
 platform: ruby
 authors:
 - Ben (@penne12_)