ollama_chat 0.0.25 → 0.0.26

data/lib/ollama_chat/model_handling.rb

@@ -1,3 +1,20 @@
+# A module that provides functionality for managing Ollama models, including
+# checking model availability, pulling models from remote servers, and handling
+# model presence verification.
+#
+# This module encapsulates the logic for interacting with Ollama models,
+# ensuring that required models are available locally before attempting to use
+# them in chat sessions. It handles both local model verification and remote
+# model retrieval when necessary.
+#
+# @example Checking if a model is present
+#   chat.model_present?('llama3.1')
+#
+# @example Pulling a model from a remote server
+#   chat.pull_model_from_remote('mistral')
+#
+# @example Ensuring a model is available locally
+#   chat.pull_model_unless_present('phi3', {})
 module OllamaChat::ModelHandling
   # The model_present? method checks if the specified Ollama model is available.
   #

data/lib/ollama_chat/ollama_chat_config.rb

@@ -1,12 +1,31 @@
 require 'pathname'
 
+# A configuration class for managing OllamaChat settings and file paths.
+#
+# This class handles the initialization and management of configuration files
+# for the OllamaChat application. It provides methods for setting up default
+# configurations, determining appropriate file paths for config and cache
+# directories, and managing the loading and creation of configuration files
+# based on XDG standards.
+#
+# @example Initializing with a custom configuration file
+#   config = OllamaChat::OllamaChatConfig.new('/path/to/custom/config.yml')
+#
+# @example Accessing default configuration paths
+#   config = OllamaChat::OllamaChatConfig.new
+#   config.default_config_path # => Path to the default configuration file
+#   config.config_dir_path     # => Path to the configuration directory
+#   config.cache_dir_path      # => Path to the cache directory
+#   config.database_path       # => Path to the documents database file
 class OllamaChat::OllamaChatConfig
   include ComplexConfig
   include FileUtils
 
+  # Path to the default config
   DEFAULT_CONFIG_PATH = Pathname.new(__FILE__).dirname.
     join('ollama_chat_config/default_config.yml')
 
+  # Content of the default config
   DEFAULT_CONFIG = File.read(DEFAULT_CONFIG_PATH)
 
   # The initialize method sets up the configuration file path and ensures the

data/lib/ollama_chat/parsing.rb

@@ -1,3 +1,16 @@
+# A module that provides content parsing functionality for OllamaChat.
+#
+# The Parsing module encapsulates methods for processing various types of input
+# sources including HTML, XML, CSV, RSS, Atom, PDF, and Postscript documents.
+# It handles content extraction and conversion into standardized text formats
+# suitable for chat interactions. The module supports different document
+# policies for handling imported or embedded content and provides utilities for
+# parsing structured data from multiple source types.
+#
+# @example Processing different document types
+#   chat.parse_source(html_io)    # Converts HTML to markdown
+#   chat.parse_source(pdf_io)     # Extracts text from PDF files
+#   chat.parse_source(csv_io)     # Formats CSV data into readable strings
 module OllamaChat::Parsing
   # The parse_source method processes different types of input sources and
   # converts them into a standardized text representation.

data/lib/ollama_chat/server_socket.rb

@@ -1,3 +1,25 @@
+# A module that provides server socket functionality for OllamaChat
+#
+# The ServerSocket module encapsulates the logic for creating and managing Unix
+# domain socket servers that enable external processes to send input to running
+# ollama_chat sessions. It supports both simple message transmission and
+# bidirectional communication with response handling, allowing for integration
+# with tools like ollama_chat_send.
+#
+# @example Sending a message to a running chat session
+#   OllamaChat::ServerSocket.send_to_server_socket(
+#     "Hello from external process",
+#     config: chat_config,
+#     type: :socket_input
+#   )
+#
+# @example Sending a message and waiting for a response
+#   response = OllamaChat::ServerSocket.send_to_server_socket(
+#     "What is the answer?",
+#     config: chat_config,
+#     type: :socket_input_with_response,
+#     parse: true
+#   )
 module OllamaChat::ServerSocket
   class << self
     # The send_to_server_socket method transmits a message to a Unix domain
@@ -56,6 +78,11 @@ module OllamaChat::ServerSocket
     end
   end
 
+  # The server_socket_message accessor method provides read and write access to
+  # the server socket message instance variable.
+  #
+  # @return [ Object, nil ] the current server socket message object or nil if
+  # not set
   attr_accessor :server_socket_message
 
   # Initializes the server socket to receive messages from the Ollama Chat

data/lib/ollama_chat/source_fetching.rb

@@ -1,3 +1,28 @@
+# A module that provides functionality for fetching and processing various
+# types of content sources.
+#
+# The SourceFetching module encapsulates methods for retrieving content from
+# different source types including URLs, file paths, and shell commands. It
+# handles the logic for determining the appropriate fetching method based on
+# the source identifier and processes the retrieved content through specialized
+# parsers depending on the content type. The module also manages image
+# handling, document importing, summarizing, and embedding operations while
+# providing error handling and debugging capabilities.
+#
+# @example Fetching content from a URL
+#   chat.fetch_source('https://example.com/document.html') do |source_io|
+#     # Process the fetched content
+#   end
+#
+# @example Importing a local file
+#   chat.fetch_source('/path/to/local/file.txt') do |source_io|
+#     # Process the imported file content
+#   end
+#
+# @example Executing a shell command
+#   chat.fetch_source('!ls -la') do |source_io|
+#     # Process the command output
+#   end
 module OllamaChat::SourceFetching
   # The http_options method prepares HTTP options for requests based on
   # configuration settings.

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.
   #

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,9 @@ 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)
+      infobar.puts "…hit, found#{result.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.26'
   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.26 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "ollama_chat".freeze
-  s.version = "0.0.25".freeze
+  s.version = "0.0.26".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 = [".all_images.yml".freeze, ".contexts/code_comment.rb".freeze, ".contexts/full.rb".freeze, ".contexts/info.rb".freeze, ".contexts/lib.rb".freeze, ".contexts/yard.md".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.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.1".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