ollama_chat 0.0.25 → 0.0.27

data/lib/ollama_chat/switches.rb

@@ -1,4 +1,28 @@
+# A module that provides switch functionality for configuring application
+# behavior.
+#
+# The Switches module encapsulates various toggle switches used throughout the
+# OllamaChat application to control different features and settings such as
+# streaming, thinking, markdown output, voice output, embedding, and location
+# information. These switches allow users to dynamically enable or disable
+# specific functionalities during a chat session.
+#
+# @example Toggling a switch on/off
+#   switch = OllamaChat::Switches::Switch.new(value: false, msg: { true => 'Enabled', false => 'Disabled' })
+#   switch.toggle  # Turns the switch on
+#   switch.toggle  # Turns the switch off
 module OllamaChat::Switches
+  # A module that provides switch state checking functionality.
+  #
+  # The CheckSwitch module adds methods for checking the boolean state of
+  # switches and displaying their current status. It's designed to be included
+  # in switch classes to provide consistent behavior for querying switch states
+  # and outputting status messages.
+  #
+  # @example Checking switch states
+  #   switch = OllamaChat::Switches::Switch.new(value: true, msg: { true => 'On', false => 'Off' })
+  #   switch.on?   # Returns true
+  #   switch.off?  # Returns false
   module CheckSwitch
     extend Tins::Concern
 
@@ -23,6 +47,19 @@ module OllamaChat::Switches
     end
   end
 
+  # A switch class that manages boolean state with toggle and set
+  # functionality.
+  #
+  # The Switch class provides a simple way to manage boolean configuration
+  # options with methods to toggle, set, and query the current state. It
+  # includes messaging capabilities to provide feedback when the state changes.
+  #
+  # @example Creating and using a switch
+  #   switch = Switch.new(value: false, msg: { true => 'Enabled', false => 'Disabled' })
+  #   switch.toggle  # Turns the switch on
+  #   switch.value   # Returns true
+  #   switch.off?    # Returns false
+  #   switch.on?     # Returns true
   class Switch
     # The initialize method sets up the switch with a default value and
     # message.
@@ -64,6 +101,18 @@ module OllamaChat::Switches
     include CheckSwitch
   end
 
+  # A switch class that manages a boolean state based on a proc value.
+  #
+  # The CombinedSwitch class provides a way to manage a boolean configuration
+  # option where the state is determined by evaluating a stored proc. This is
+  # useful for complex conditions that depend on multiple factors or dynamic
+  # values, such as combining multiple switch states into a single effective
+  # state.
+  #
+  # @example Checking if embedding is currently performed
+  #   # When embedding_enabled is true and embedding_paused is false,
+  #   # the combined switch will return true
+  #   combined_switch.value # => true
   class CombinedSwitch
     # The initialize method sets up the switch with a value and message.
     #

data/lib/ollama_chat/utils/cache_fetcher.rb

@@ -1,5 +1,20 @@
 require 'digest/md5'
 
+# A cache fetcher implementation that handles caching of HTTP responses with
+# content type metadata.
+#
+# This class provides a mechanism to store and retrieve cached HTTP responses,
+# including their content types, using a key-based system. It is designed to
+# work with various cache backends and ensures that both the response body and
+# metadata are properly cached and retrieved for efficient subsequent requests.
+#
+# @example Using the CacheFetcher to cache and retrieve HTTP responses
+#   cache = Redis.new
+#   fetcher = OllamaChat::Utils::CacheFetcher.new(cache)
+#   fetcher.put('https://example.com', io)
+#   fetcher.get('https://example.com') do |cached_io|
+#     # Process cached content
+#   end
 class OllamaChat::Utils::CacheFetcher
   # The initialize method sets up the cache instance variable for the object.
   #
@@ -30,6 +45,7 @@ class OllamaChat::Utils::CacheFetcher
       io.extend(OllamaChat::Utils::Fetcher::HeaderExtension)
       io.content_type = content_type
       block.(io)
+      io
     end
   end
 

data/lib/ollama_chat/utils/fetcher.rb

@@ -5,7 +5,34 @@ require 'mime-types'
 require 'stringio'
 require 'ollama_chat/utils/cache_fetcher'
 
+# A fetcher implementation that handles retrieval and caching of HTTP
+# resources.
+#
+# This class provides functionality to fetch content from URLs, with support
+# for caching responses and their metadata. It handles various content types
+# and integrates with different cache backends to improve performance by
+# avoiding redundant network requests.
+#
+# @example Fetching content from a URL with caching
+#   fetcher = OllamaChat::Utils::Fetcher.new(cache: redis_cache)
+#   fetcher.get('https://example.com/data.json') do |tmp|
+#     # Process the fetched content
+#   end
 class OllamaChat::Utils::Fetcher
+  # A module that extends IO objects with content type metadata and expiration
+  # tracking.
+  #
+  # This module provides a way to attach MIME content type information and
+  # cache expiration details to IO objects, enabling them to carry metadata
+  # about their source and caching behavior. It is primarily used by fetcher
+  # implementations to decorate response objects with additional context for
+  # processing and caching decisions.
+  #
+  # @example Extending an IO object with header metadata
+  #   io = StringIO.new("content")
+  #   io.extend(OllamaChat::Utils::Fetcher::HeaderExtension)
+  #   io.content_type = MIME::Types['text/plain'].first
+  #   io.ex = 3600
   module HeaderExtension
     # The content_type method accesses the content type attribute of the object.
     #
@@ -30,6 +57,18 @@ class OllamaChat::Utils::Fetcher
     end
   end
 
+  # A custom error class raised when retrying HTTP requests without streaming.
+  #
+  # This exception is specifically used in the Fetcher class to indicate that
+  # an HTTP request should be retried using a non-streaming approach when a
+  # streaming attempt fails or is not supported.
+  #
+  # @example Handling the RetryWithoutStreaming error
+  #   begin
+  #     fetcher.get('https://example.com')
+  #   rescue RetryWithoutStreaming
+  #     # Handle retry with non-streaming method
+  #   end
   class RetryWithoutStreaming < StandardError; end
 
   # The get method retrieves content from a URL, using caching when available.
@@ -51,8 +90,10 @@ class OllamaChat::Utils::Fetcher
   def self.get(url, headers: {}, **options, &block)
     cache = options.delete(:cache) and
       cache = OllamaChat::Utils::CacheFetcher.new(cache)
-    cache and infobar.puts "Getting #{url.to_s.inspect} from cache."
+    cache and infobar.puts "Getting #{url.to_s.inspect} via cache…"
     if result = cache&.get(url, &block)
+      content_type = result&.content_type || 'unknown'
+      infobar.puts "…hit, found #{content_type} content in cache."
       return result
     else
       new(**options).send(:get, url, headers:) do |tmp|

data/lib/ollama_chat/utils.rb

@@ -1,3 +1,11 @@
+# A module that provides utility classes and methods for the OllamaChat
+# application.
+#
+# The Utils module serves as a namespace for various helper components that
+# support the core functionality of OllamaChat. It contains implementations for
+# caching, interactive selection, content fetching, and file argument handling
+# that are used throughout the application to provide robust and user-friendly
+# features.
 module OllamaChat::Utils
 end
 

data/lib/ollama_chat/version.rb

@@ -1,6 +1,6 @@
 module OllamaChat
   # OllamaChat version
-  VERSION         = '0.0.25'
+  VERSION         = '0.0.27'
   VERSION_ARRAY   = VERSION.split('.').map(&:to_i) # :nodoc:
   VERSION_MAJOR   = VERSION_ARRAY[0] # :nodoc:
   VERSION_MINOR   = VERSION_ARRAY[1] # :nodoc:

data/lib/ollama_chat/vim.rb

@@ -1,6 +1,12 @@
 require 'tempfile'
 require 'pathname'
 
+# A class that provides functionality for inserting text into Vim buffers via
+# remote communication.
+#
+# @example
+#   vim = OllamaChat::Vim.new("MY_SERVER")
+#   vim.insert("Hello, Vim!")
 class OllamaChat::Vim
   # Initializes a new Vim server connection
   #

data/lib/ollama_chat/web_searching.rb

@@ -1,3 +1,14 @@
+# A module that provides web search functionality for OllamaChat.
+#
+# The WebSearching module encapsulates the logic for performing web searches
+# using configured search engines. It handles query construction, location
+# information integration, and delegates to engine-specific implementations for
+# retrieving search results. The module supports multiple search engines
+# including SearxNG and DuckDuckGo, making it flexible for different deployment
+# scenarios and privacy preferences.
+#
+# @example Performing a web search
+#   chat.search_web('ruby programming tutorials', 5)
 module OllamaChat::WebSearching
   # The search_web method performs a web search using the configured search
   # engine.

data/lib/ollama_chat.rb

@@ -1,3 +1,12 @@
+# The main module namespace for the OllamaChat application.
+#
+# This module serves as the root namespace for all components of the OllamaChat
+# Ruby gem, providing access to core classes, utilities, and configuration
+# management for interacting with Ollama language models through a terminal
+# interface.
+#
+# @example Accessing the main module
+#   OllamaChat::VERSION # => "0.0.25"
 module OllamaChat
 end
 

data/ollama_chat.gemspec

@@ -1,9 +1,9 @@
 # -*- encoding: utf-8 -*-
-# stub: ollama_chat 0.0.25 ruby lib
+# stub: ollama_chat 0.0.27 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "ollama_chat".freeze
-  s.version = "0.0.25".freeze
+  s.version = "0.0.27".freeze
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0".freeze) if s.respond_to? :required_rubygems_version=
   s.require_paths = ["lib".freeze]
@@ -13,7 +13,7 @@ Gem::Specification.new do |s|
   s.email = "flori@ping.de".freeze
   s.executables = ["ollama_chat".freeze, "ollama_chat_send".freeze]
   s.extra_rdoc_files = ["README.md".freeze, "lib/ollama_chat.rb".freeze, "lib/ollama_chat/chat.rb".freeze, "lib/ollama_chat/clipboard.rb".freeze, "lib/ollama_chat/dialog.rb".freeze, "lib/ollama_chat/document_cache.rb".freeze, "lib/ollama_chat/follow_chat.rb".freeze, "lib/ollama_chat/history.rb".freeze, "lib/ollama_chat/information.rb".freeze, "lib/ollama_chat/kramdown_ansi.rb".freeze, "lib/ollama_chat/message_format.rb".freeze, "lib/ollama_chat/message_list.rb".freeze, "lib/ollama_chat/message_output.rb".freeze, "lib/ollama_chat/model_handling.rb".freeze, "lib/ollama_chat/ollama_chat_config.rb".freeze, "lib/ollama_chat/parsing.rb".freeze, "lib/ollama_chat/server_socket.rb".freeze, "lib/ollama_chat/source_fetching.rb".freeze, "lib/ollama_chat/switches.rb".freeze, "lib/ollama_chat/utils.rb".freeze, "lib/ollama_chat/utils/cache_fetcher.rb".freeze, "lib/ollama_chat/utils/chooser.rb".freeze, "lib/ollama_chat/utils/fetcher.rb".freeze, "lib/ollama_chat/utils/file_argument.rb".freeze, "lib/ollama_chat/version.rb".freeze, "lib/ollama_chat/vim.rb".freeze, "lib/ollama_chat/web_searching.rb".freeze]
-  s.files = [".all_images.yml".freeze, ".contexts/full.rb".freeze, ".contexts/info.rb".freeze, ".contexts/lib.rb".freeze, ".envrc".freeze, ".gitignore".freeze, "CHANGES.md".freeze, "Gemfile".freeze, "README.md".freeze, "Rakefile".freeze, "VERSION".freeze, "bin/ollama_chat".freeze, "bin/ollama_chat_send".freeze, "config/searxng/settings.yml".freeze, "docker-compose.yml".freeze, "lib/ollama_chat.rb".freeze, "lib/ollama_chat/chat.rb".freeze, "lib/ollama_chat/clipboard.rb".freeze, "lib/ollama_chat/dialog.rb".freeze, "lib/ollama_chat/document_cache.rb".freeze, "lib/ollama_chat/follow_chat.rb".freeze, "lib/ollama_chat/history.rb".freeze, "lib/ollama_chat/information.rb".freeze, "lib/ollama_chat/kramdown_ansi.rb".freeze, "lib/ollama_chat/message_format.rb".freeze, "lib/ollama_chat/message_list.rb".freeze, "lib/ollama_chat/message_output.rb".freeze, "lib/ollama_chat/model_handling.rb".freeze, "lib/ollama_chat/ollama_chat_config.rb".freeze, "lib/ollama_chat/ollama_chat_config/default_config.yml".freeze, "lib/ollama_chat/parsing.rb".freeze, "lib/ollama_chat/server_socket.rb".freeze, "lib/ollama_chat/source_fetching.rb".freeze, "lib/ollama_chat/switches.rb".freeze, "lib/ollama_chat/utils.rb".freeze, "lib/ollama_chat/utils/cache_fetcher.rb".freeze, "lib/ollama_chat/utils/chooser.rb".freeze, "lib/ollama_chat/utils/fetcher.rb".freeze, "lib/ollama_chat/utils/file_argument.rb".freeze, "lib/ollama_chat/version.rb".freeze, "lib/ollama_chat/vim.rb".freeze, "lib/ollama_chat/web_searching.rb".freeze, "ollama_chat.gemspec".freeze, "redis/redis.conf".freeze, "spec/assets/api_show.json".freeze, "spec/assets/api_tags.json".freeze, "spec/assets/api_version.json".freeze, "spec/assets/conversation.json".freeze, "spec/assets/duckduckgo.html".freeze, "spec/assets/example.atom".freeze, "spec/assets/example.csv".freeze, "spec/assets/example.html".freeze, "spec/assets/example.pdf".freeze, "spec/assets/example.ps".freeze, "spec/assets/example.rb".freeze, "spec/assets/example.rss".freeze, "spec/assets/example.xml".freeze, "spec/assets/kitten.jpg".freeze, "spec/assets/prompt.txt".freeze, "spec/assets/searxng.json".freeze, "spec/ollama_chat/chat_spec.rb".freeze, "spec/ollama_chat/clipboard_spec.rb".freeze, "spec/ollama_chat/follow_chat_spec.rb".freeze, "spec/ollama_chat/information_spec.rb".freeze, "spec/ollama_chat/kramdown_ansi_spec.rb".freeze, "spec/ollama_chat/message_list_spec.rb".freeze, "spec/ollama_chat/message_output_spec.rb".freeze, "spec/ollama_chat/model_handling_spec.rb".freeze, "spec/ollama_chat/parsing_spec.rb".freeze, "spec/ollama_chat/server_socket_spec.rb".freeze, "spec/ollama_chat/source_fetching_spec.rb".freeze, "spec/ollama_chat/switches_spec.rb".freeze, "spec/ollama_chat/utils/cache_fetcher_spec.rb".freeze, "spec/ollama_chat/utils/fetcher_spec.rb".freeze, "spec/ollama_chat/utils/file_argument_spec.rb".freeze, "spec/ollama_chat/web_searching_spec.rb".freeze, "spec/spec_helper.rb".freeze, "tmp/.keep".freeze]
+  s.files = ["CHANGES.md".freeze, "Gemfile".freeze, "README.md".freeze, "Rakefile".freeze, "bin/ollama_chat".freeze, "bin/ollama_chat_send".freeze, "config/searxng/settings.yml".freeze, "docker-compose.yml".freeze, "lib/ollama_chat.rb".freeze, "lib/ollama_chat/chat.rb".freeze, "lib/ollama_chat/clipboard.rb".freeze, "lib/ollama_chat/dialog.rb".freeze, "lib/ollama_chat/document_cache.rb".freeze, "lib/ollama_chat/follow_chat.rb".freeze, "lib/ollama_chat/history.rb".freeze, "lib/ollama_chat/information.rb".freeze, "lib/ollama_chat/kramdown_ansi.rb".freeze, "lib/ollama_chat/message_format.rb".freeze, "lib/ollama_chat/message_list.rb".freeze, "lib/ollama_chat/message_output.rb".freeze, "lib/ollama_chat/model_handling.rb".freeze, "lib/ollama_chat/ollama_chat_config.rb".freeze, "lib/ollama_chat/ollama_chat_config/default_config.yml".freeze, "lib/ollama_chat/parsing.rb".freeze, "lib/ollama_chat/server_socket.rb".freeze, "lib/ollama_chat/source_fetching.rb".freeze, "lib/ollama_chat/switches.rb".freeze, "lib/ollama_chat/utils.rb".freeze, "lib/ollama_chat/utils/cache_fetcher.rb".freeze, "lib/ollama_chat/utils/chooser.rb".freeze, "lib/ollama_chat/utils/fetcher.rb".freeze, "lib/ollama_chat/utils/file_argument.rb".freeze, "lib/ollama_chat/version.rb".freeze, "lib/ollama_chat/vim.rb".freeze, "lib/ollama_chat/web_searching.rb".freeze, "ollama_chat.gemspec".freeze, "redis/redis.conf".freeze, "spec/assets/api_show.json".freeze, "spec/assets/api_tags.json".freeze, "spec/assets/api_version.json".freeze, "spec/assets/conversation.json".freeze, "spec/assets/duckduckgo.html".freeze, "spec/assets/example.atom".freeze, "spec/assets/example.csv".freeze, "spec/assets/example.html".freeze, "spec/assets/example.pdf".freeze, "spec/assets/example.ps".freeze, "spec/assets/example.rb".freeze, "spec/assets/example.rss".freeze, "spec/assets/example.xml".freeze, "spec/assets/kitten.jpg".freeze, "spec/assets/prompt.txt".freeze, "spec/assets/searxng.json".freeze, "spec/ollama_chat/chat_spec.rb".freeze, "spec/ollama_chat/clipboard_spec.rb".freeze, "spec/ollama_chat/follow_chat_spec.rb".freeze, "spec/ollama_chat/information_spec.rb".freeze, "spec/ollama_chat/kramdown_ansi_spec.rb".freeze, "spec/ollama_chat/message_list_spec.rb".freeze, "spec/ollama_chat/message_output_spec.rb".freeze, "spec/ollama_chat/model_handling_spec.rb".freeze, "spec/ollama_chat/parsing_spec.rb".freeze, "spec/ollama_chat/server_socket_spec.rb".freeze, "spec/ollama_chat/source_fetching_spec.rb".freeze, "spec/ollama_chat/switches_spec.rb".freeze, "spec/ollama_chat/utils/cache_fetcher_spec.rb".freeze, "spec/ollama_chat/utils/fetcher_spec.rb".freeze, "spec/ollama_chat/utils/file_argument_spec.rb".freeze, "spec/ollama_chat/web_searching_spec.rb".freeze, "spec/spec_helper.rb".freeze, "tmp/.keep".freeze]
   s.homepage = "https://github.com/flori/ollama_chat".freeze
   s.licenses = ["MIT".freeze]
   s.rdoc_options = ["--title".freeze, "OllamaChat - A command-line interface (CLI) for interacting with an Ollama AI model.".freeze, "--main".freeze, "README.md".freeze]
@@ -24,7 +24,7 @@ Gem::Specification.new do |s|
 
   s.specification_version = 4
 
-  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 2.0".freeze])
+  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 2.2".freeze])
   s.add_development_dependency(%q<all_images>.freeze, ["~> 0.6".freeze])
   s.add_development_dependency(%q<rspec>.freeze, ["~> 3.2".freeze])
   s.add_development_dependency(%q<kramdown>.freeze, ["~> 2.0".freeze])

data/spec/ollama_chat/chat_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-describe OllamaChat::Chat do
+describe OllamaChat::Chat, protect_env: true do
   let :argv do
     %w[ -C test ]
   end
@@ -77,6 +77,16 @@ describe OllamaChat::Chat do
       expect(chat.handle_input("/clobber")).to eq :next
     end
 
+    it 'returns :next when input is "/last"' do
+      expect(chat.messages).to receive(:show_last)
+      expect(chat.handle_input("/last")).to eq :next
+    end
+
+    it 'returns :next when input is "/last\s+(\d+)"' do
+      expect(chat.messages).to receive(:show_last).with(2)
+      expect(chat.handle_input("/last 2")).to eq :next
+    end
+
     it 'returns :next when input is "/drop(?:\s+(\d*))?"' do
       expect(chat.messages).to receive(:drop).with(?2)
       expect(chat.messages).to receive(:show_last)

data/spec/ollama_chat/kramdown_ansi_spec.rb

@@ -5,10 +5,10 @@ describe OllamaChat::KramdownANSI do
     double('Chat').extend(described_class)
   end
 
-  describe '#configure_kramdown_ansi_styles' do
+  describe '#configure_kramdown_ansi_styles', protect_env: true do
     it 'can be configured via env var' do
-      allow(ENV).to receive(:key?).with('KRAMDOWN_ANSI_OLLAMA_CHAT_STYLES').and_return(true)
-      allow(ENV).to receive(:key?).with('KRAMDOWN_ANSI_STYLES').and_return(false)
+      ENV['KRAMDOWN_ANSI_OLLAMA_CHAT_STYLES'] = '{}'
+      ENV.delete('KRAMDOWN_ANSI_STYLES')
 
       styles = { bold: '1' }
       expect(Kramdown::ANSI::Styles).to receive(:from_env_var).
@@ -19,8 +19,8 @@ describe OllamaChat::KramdownANSI do
     end
 
     it 'has a default configuration' do
-      allow(ENV).to receive(:key?).with('KRAMDOWN_ANSI_OLLAMA_CHAT_STYLES').and_return(false)
-      allow(ENV).to receive(:key?).with('KRAMDOWN_ANSI_STYLES').and_return(false)
+      ENV.delete('KRAMDOWN_ANSI_OLLAMA_CHAT_STYLES')
+      ENV.delete('KRAMDOWN_ANSI_STYLES')
 
       expect(chat.configure_kramdown_ansi_styles).to be_a(Hash)
     end

data/spec/ollama_chat/message_list_spec.rb

@@ -39,7 +39,7 @@ describe OllamaChat::MessageList do
     expect(list.size).to eq 1
     list.clear
     expect(list.size).to eq 1
-    list <<  Ollama::Message.new(role: 'user', content: 'world')
+    list << Ollama::Message.new(role: 'user', content: 'world')
     expect(list.size).to eq 2
     list.clear
     expect(list.size).to eq 1
@@ -47,7 +47,7 @@ describe OllamaChat::MessageList do
 
   it 'can be added to' do
     expect(list.size).to eq 1
-    list <<  Ollama::Message.new(role: 'user', content: 'world')
+    list << Ollama::Message.new(role: 'user', content: 'world')
     expect(list.size).to eq 2
   end
 
@@ -71,8 +71,31 @@ describe OllamaChat::MessageList do
     FileUtils.rm_f 'tmp/test-conversation.json'
   end
 
+  describe "#last" do
+    it "returns the last message when there are multiple messages" do
+      list = described_class.new(chat)
+      list << Ollama::Message.new(role: 'system', content: 'hello')
+      list << Ollama::Message.new(role: 'user', content: 'First message')
+      list << Ollama::Message.new(role: 'assistant', content: 'Second message')
+
+      expect(list.last.content).to eq('Second message')
+    end
+
+    it "returns the last message when there is only one message" do
+      list = described_class.new(chat)
+      list << Ollama::Message.new(role: 'system', content: 'hello')
+
+      expect(list.last.content).to eq('hello')
+    end
+
+    it "returns nil when there are no messages" do
+      list = described_class.new(chat)
+
+      expect(list.last).to be_nil
+    end
+  end
 
-  describe '.show_last' do
+  describe '#show_last' do
     it 'shows nothing when there are no messages' do
       empty_list = described_class.new(chat)
       expect { empty_list.show_last }.not_to raise_error
@@ -83,7 +106,7 @@ describe OllamaChat::MessageList do
       list = described_class.new(chat)
       allow(chat).to receive(:think).and_return(double(on?: false))
       allow(chat).to receive(:markdown).and_return(double(on?: false))
-      list <<  Ollama::Message.new(role: 'assistant', content: 'hello')
+      list << Ollama::Message.new(role: 'assistant', content: 'hello')
       expect(STDOUT).to receive(:puts).
         with("📨 \e[1m\e[38;5;111massistant\e[0m\e[0m:\nhello\n")
       expect(list.show_last).to be_a described_class
@@ -91,10 +114,22 @@ describe OllamaChat::MessageList do
 
     it 'shows nothing when the last message is by the user' do
       list = described_class.new(chat)
-      list <<  Ollama::Message.new(role: 'user', content: 'world')
+      list << Ollama::Message.new(role: 'user', content: 'world')
       expect { list.show_last }.not_to raise_error
       expect(list.show_last).to be nil
     end
+
+    it "shows last N messages when N is larger than available messages" do
+      allow(chat).to receive(:think).and_return(double(on?: false))
+      allow(chat).to receive(:markdown).and_return(double(on?: false))
+      list = described_class.new(chat)
+      list << Ollama::Message.new(role: 'system', content: 'hello')
+      list << Ollama::Message.new(role: 'user', content: 'First message')
+      list << Ollama::Message.new(role: 'assistant', content: 'Second message')
+
+      expect(STDOUT).to receive(:puts).with(/Second message/)
+      expect(list.show_last(23)).to eq(list)
+    end
   end
 
   context 'without pager' do
@@ -117,7 +152,7 @@ describe OllamaChat::MessageList do
         and_return(double(on?: true)).at_least(:once)
       expect(chat).to receive(:think).
         and_return(double(on?: false)).at_least(:once)
-      list <<  Ollama::Message.new(role: 'user', content: 'world')
+      list << Ollama::Message.new(role: 'user', content: 'world')
       expect(STDOUT).to receive(:puts).
         with(
           "📨 \e[1m\e[38;5;213msystem\e[0m\e[0m:\nhello\n" \
@@ -158,7 +193,7 @@ describe OllamaChat::MessageList do
         and_return(double(on?: true)).at_least(:once)
       expect(chat).to receive(:think).
         and_return(double(on?: false)).at_least(:once)
-      list <<  Ollama::Message.new(role: 'user', content: 'world')
+      list << Ollama::Message.new(role: 'user', content: 'world')
       list.list_conversation
     end
   end
@@ -195,9 +230,9 @@ describe OllamaChat::MessageList do
     expect(list.size).to eq 1
     expect(list.drop(1)).to eq 0
     expect(list.size).to eq 1
-    list <<  Ollama::Message.new(role: 'user', content: 'world')
+    list << Ollama::Message.new(role: 'user', content: 'world')
     expect(list.size).to eq 2
-    list <<  Ollama::Message.new(role: 'assistant', content: 'hi')
+    list << Ollama::Message.new(role: 'assistant', content: 'hi')
     expect(list.size).to eq 3
     expect(list.drop(1)).to eq 1
     expect(list.size).to eq 1
@@ -227,7 +262,7 @@ describe OllamaChat::MessageList do
 
   it 'can be converted int an Ollama::Message array' do
     expect(chat).to receive(:location).and_return(double(on?: false))
-    list <<  Ollama::Message.new(role: 'user', content: 'world')
+    list << Ollama::Message.new(role: 'user', content: 'world')
     expect(list.to_ary.map(&:as_json)).to eq [
       Ollama::Message.new(role: 'system', content: 'hello').as_json,
       Ollama::Message.new(role: 'user', content: 'world').as_json,
@@ -236,7 +271,7 @@ describe OllamaChat::MessageList do
 
   it 'can be converted int an Ollama::Message array with location' do
     expect(chat).to receive(:location).and_return(double(on?: true))
-    list <<  Ollama::Message.new(role: 'user', content: 'world')
+    list << Ollama::Message.new(role: 'user', content: 'world')
     first = list.to_ary.first
     expect(first.role).to eq 'system'
     expect(first.content).to match(
@@ -246,7 +281,7 @@ describe OllamaChat::MessageList do
   it 'can be converted int an Ollama::Message array with location without a system prompt' do
     expect(chat).to receive(:location).and_return(double(on?: true))
     list = described_class.new(chat).tap do |list|
-      list <<  Ollama::Message.new(role: 'user', content: 'hello')
+      list << Ollama::Message.new(role: 'user', content: 'hello')
       list << Ollama::Message.new(role: 'assistant', content: 'world')
     end
     first = list.to_ary.first

data/spec/spec_helper.rb

@@ -12,44 +12,138 @@ require 'ollama_chat'
 
 ComplexConfig::Provider.deep_freeze = false
 
-def asset(name)
-  File.join(__dir__, 'assets', name)
-end
+# A module that provides helper methods for asset management within the
+# application.
+#
+# The AssetHelpers module encapsulates functionality related to handling and
+# processing application assets, such as CSS, JavaScript, and image files. It
+# offers utilities for managing asset paths, generating URLs, and performing
+# operations on assets during the application's runtime.
+module AssetHelpers
+  # The asset method constructs and returns the full path to an asset file.
+  #
+  # This method takes a filename argument and combines it with the assets directory
+  # located within the same directory as the calling file, returning the
+  # complete path to that asset.
+  #
+  # @param name [String] the name of the asset file
+  #
+  # @return [String] the full path to the asset file
+  def asset(name)
+    File.join(__dir__, 'assets', name)
+  end
 
-def asset_content(name)
-  File.read(File.join(__dir__, 'assets', name))
-end
+  # Reads and returns the content of an asset file from the assets directory.
+  #
+  # @param name [String] the name of the asset file to read
+  #
+  # @return [String] the content of the asset file as a string
+  def asset_content(name)
+    File.read(File.join(__dir__, 'assets', name))
+  end
 
-def asset_io(name, &block)
-  io = File.new(File.join(__dir__, 'assets', name))
-  if block
-    begin
-      block.call(io)
-    ensure
-      io.close
+  # The asset_io method retrieves an IO object for a specified asset file.
+  #
+  # This method constructs the path to an asset file within the assets directory
+  # and returns an IO object representing that file. If a block is provided, it
+  # yields the IO object to the block and ensures the file is properly closed
+  # after the block executes.
+  #
+  # @param name [ String ] the name of the asset file to retrieve
+  #
+  # @yield [ io ] yields the IO object for the asset file to the provided block
+  #
+  # @return [ File, nil ] returns the IO object for the asset file, or nil if a
+  # block is provided and the block does not return a value
+  def asset_io(name, &block)
+    io = File.new(File.join(__dir__, 'assets', name))
+    if block
+      begin
+        block.call(io)
+      ensure
+        io.close
+      end
+    else
+      io
     end
-  else
-    io
+  end
+
+  # The asset_json method reads and parses a JSON asset file.
+  #
+  # This method retrieves an asset by name, reads its contents from the
+  # filesystem, and then parses the resulting string as JSON, returning the
+  # parsed data structure.
+  #
+  # @param name [String] the name of the asset to retrieve and parse
+  #
+  # @return [Object] the parsed JSON data structure, typically a Hash or Array
+  def asset_json(name)
+    JSON(JSON(File.read(asset(name))))
   end
 end
 
-def asset_json(name)
-  JSON(JSON(File.read(asset(name))))
+# A module that provides functionality for stubbing Ollama server responses.
+#
+# The StubOllamaServer module enables developers to simulate Ollama API
+# interactions in test environments by intercepting requests and returning
+# predefined responses. This allows for faster, more reliable testing without
+# requiring external service calls.
+module StubOllamaServer
+  # The connect_to_ollama_server method establishes a connection to an Ollama
+  # server.
+  #
+  # This method sets up stubbed HTTP requests to simulate responses from an
+  # Ollama server, including API tags, show, and version endpoints. It can
+  # optionally instantiate a chat session after setting up the stubs.
+  #
+  # @param instantiate [Boolean] whether to instantiate a chat session after setting up stubs
+  def connect_to_ollama_server(instantiate: true)
+    before do
+      stub_request(:get, %r(/api/tags\z)).
+        to_return(status: 200, body: asset_json('api_tags.json'))
+      stub_request(:post, %r(/api/show\z)).
+        to_return(status: 200, body: asset_json('api_show.json'))
+      stub_request(:get, %r(/api/version\z)).
+        to_return(status: 200, body: asset_json('api_version.json'))
+      instantiate and chat
+    end
+  end
 end
 
-def connect_to_ollama_server(instantiate: true)
-  before do
-    stub_request(:get, %r(/api/tags\z)).
-      to_return(status: 200, body: asset_json('api_tags.json'))
-    stub_request(:post, %r(/api/show\z)).
-      to_return(status: 200, body: asset_json('api_show.json'))
-    stub_request(:get, %r(/api/version\z)).
-      to_return(status: 200, body: asset_json('api_version.json'))
-    instantiate and chat
+# A module that provides functionality for protecting environment variables during tests.
+#
+# This module ensures that environment variable changes made during test execution
+# are automatically restored to their original values after the test completes.
+# It is designed to prevent side effects between tests that modify environment
+# variables, maintaining a clean testing environment.
+module ProtectEnvVars
+  # The apply method creates a lambda that protects environment variables
+  # during test execution.
+  #
+  # @return [Proc] a lambda that wraps test execution with environment variable
+  # preservation
+  def self.apply
+    -> example do
+      if example.metadata[:protect_env]
+        begin
+          stored_env = ENV.to_h
+          example.run
+        ensure
+          ENV.replace(stored_env)
+        end
+      else
+        example.run
+      end
+    end
   end
 end
 
 RSpec.configure do |config|
+  config.include AssetHelpers
+  config.extend StubOllamaServer
+
+  config.around(&ProtectEnvVars.apply)
+
   config.before(:suite) do
     infobar.show = nil
   end