bin/ollama_console in ollama-ruby-0.2.0 vs bin/ollama_console in ollama-ruby-0.3.0

@@ -3,11 +3,16 @@
 require 'ollama'
 include Ollama
 require 'irb'
 require 'irb/history'
 
-base_url = ENV['OLLAMA_URL'] || 'http://%s' % ENV.fetch('OLLAMA_HOST')
-ollama = Client.new(base_url:)
+def base_url
+  ENV['OLLAMA_URL'] || 'http://%s' % ENV.fetch('OLLAMA_HOST')
+end
+
+def ollama
+  $ollama ||= Client.new(base_url:)
+end
 IRB.setup nil
 IRB.conf[:MAIN_CONTEXT] = IRB::Irb.new.context
 IRB.conf[:HISTORY_FILE] = File.join(ENV.fetch('HOME'), '.ollama_console-history')
 IRB.conf[:SAVE_HISTORY] = 1000
 require 'irb/ext/multi-irb'