lib/vite_ruby/cli/clobber.rb in vite_ruby-1.2.0 vs lib/vite_ruby/cli/clobber.rb in vite_ruby-1.2.1

@@ -5,9 +5,10 @@
 
   current_env = ENV['RACK_ENV'] || ENV['RAILS_ENV'] || 'development'
 
   option(:mode, default: current_env, values: %w[development production test], aliases: ['m'], desc: 'The mode to use')
 
-  def call(**)
+  def call(mode:, **)
+    ViteRuby.env['VITE_RUBY_MODE'] = mode
     ViteRuby.commands.clobber
   end
 end