lib/kugiru/builder.rb in kugiru-0.1.0 vs lib/kugiru/builder.rb in kugiru-0.1.1

@@ -27,10 +27,15 @@
       self.new(args).build
     end
 
     def self.build_enumerator(**args)
       Enumerator.new do |y|
-        self.new(y, args).build
+        begin
+          self.new(y, args).build
+        rescue => error
+          Kugiru.configuration.after_streaming_error.try(:call, error)
+          raise error
+        end
       end
     end
   end
 end