ollama_chat 0.0.68 → 0.0.69

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 135734ab31efeba4bc0ba5a5268bb739269de2e30ed9bed5e92897e2bcf91007
-  data.tar.gz: 7b331fa27f44318d73524b532573a4e2a60d1d33a234400794ee30dfc9cb3210
+  metadata.gz: b4c1e0e4af70239c191a9b7fb36bafb5e813bebbe99b3884a19ed1350074937c
+  data.tar.gz: 39ff2ea149df672fdf021fc367681d578e701fe9d8fe7b4a4b202e9e73575cee
 SHA512:
-  metadata.gz: 646671b9819e6f531e170edb969ce21d60bc7e32abbb1872eaa22da107895deaf0c5d46f816189d724a92a10e5cdcf21850e12ba80fa532e8e99d2f68265cfd7
-  data.tar.gz: b2f2a6e619d599d435025dfe715d7fab6af0d416356672a4c95721cff0bedb534868217281db916f9b26af3d1886e75c566d0695d712a341227fafd4b6e38ef4
+  metadata.gz: 7bb2eb1885554be1140c8748adad8168c4eec9cfc85b267ea3af8f046643489c5611787a43da878a41fe3382828930caeb80be10b24f0059bae865f56d8f1e8b
+  data.tar.gz: a1c629f1678eb8b763b04ba0def7666e00f0d76ca346d5a285efc9aa8ad3b71dd13dba04482a778e2b7b178e1cbe43e10895f0cac69ad01145d276771789e574

data/CHANGES.md

@@ -1,5 +1,104 @@
 # Changes
 
+## 2026-02-23 v0.0.69
+
+- Reorganized access modifiers in `module OllamaChat::Dialog`, moving `ask?` to
+  be public before the `private` keyword.  
+- Moved `links`, `debug`, `config=`, and `config` from public to private in
+  `lib/ollama_chat/chat.rb`.  
+- Added a `class_methods` block in `lib/ollama_chat/config_handling.rb`
+  declaring a class‑level `config` accessor via `Tins::Concern`.  
+- Added a private `config=` method in `config_handling.rb` to write to the
+  class‑level `config`.  
+- Kept a public `config` method that returns `self.class.config`.  
+- Added `private` to modules: `OllamaChat::Clipboard`,
+  `OllamaChat::Conversation`, `OllamaChat::Dialog`,
+  `OllamaChat::DocumentCache`, `OllamaChat::History`,
+  `OllamaChat::InputContent`, `OllamaChat::MessageEditing`,
+  `OllamaChat::ModelHandling`, `OllamaChat::ServerSocket` and removed the
+  `private` declaration from `OllamaChat::ToolCalling`.  
+- Updated spec files to call `expose` on `OllamaChat::Chat.new(argv:
+  chat_default_config)` and on `Object.extend(described_class)` so that private
+  helpers can be exercised during testing.  
+- Ensured the command‑handling logic in `handle_input` continues to invoke the
+  newly private methods correctly.  
+- Added `tool_function(name)` helper to `OllamaChat::ToolCalling` for
+  centralised tool config lookup.  
+- Updated `lib/ollama_chat/follow_chat.rb` to use `@chat.tool_function(name)`
+  instead of `config.tools.functions[name]`.  
+- Modified `lib/ollama_chat/switches.rb` switch messages to lower‑case
+  “support” (`"Tools support enabled."` / `"Tools support disabled."`).  
+- In `lib/ollama_chat/tool_calling.rb`, replaced `config.tools.functions[tool]`
+  with `tool_function(tool)` when printing tools and checking
+  `require_confirmation?`.  
+- Added a `private` section header in `lib/ollama_chat/tool_calling.rb` for
+  clarity.  
+- Adjusted spec regex in `spec/ollama_chat/chat_spec.rb` to match updated
+  message (`Tools support enabled`).  
+- Added `OllamaChat::ConfigHandling` module in
+  `lib/ollama_chat/config_handling.rb` with `display_config`, `fix_config`,
+  `edit_config`.  
+- Required `ollama_chat/config_handling` in `lib/ollama_chat.rb`.  
+- Included `OllamaChat::ConfigHandling` in `OllamaChat::Chat`.  
+- Updated `/config` command regex to `^/config(?:\\s+(edit))?$` and dispatched
+  to `edit_config` or `display_config`.  
+- Removed old `display_config`, `fix_config`, `edit_config` methods from
+  `chat.rb`.  
+- Added helper methods `tool_configured?`, `tool_registered?`, `tool_enabled?`
+  to `OllamaChat::ToolCalling` for cleaner tool state checks.  
+- Updated `FollowChat` to use the new helpers instead of direct config/registry
+  lookups (`@chat.config.tools.functions.attribute_set?(name)` →
+  `@chat.tool_configured?(name)`, `OllamaChat::Tools.registered?(name)` →
+  `@chat.tool_registered?(name)`, `@chat.enabled_tools.member?(name)` →
+  `@chat.tool_enabled?(name)`).  
+- Refactored `default_enabled_tools` in `OllamaChat::ToolCalling` to call
+  `tool_registered?` and adjusted `list_tools` to use `tool_enabled?` for
+  checkbox rendering.  
+- Simplified `Fetcher#fetch` signature from `fetch(url, headers: {}, options:
+  {})` to `fetch(url, opts = {})`; updated documentation to reflect the new
+  `opts` hash and optional `:headers` key.  
+- Added guard in `lib/ollama_chat/follow_chat.rb` to skip execution of disabled
+  tools and record an error in `@chat.tool_call_results`.  
+- Added `attr_reader :enabled_tools` to `OllamaChat::ToolCalling` and replaced
+  all `@enabled_tools` references with the reader.  
+- Updated `tools`, `list_tools`, `enable_tool`, and `disable_tool` to use
+  `enabled_tools` instead of the instance variable.  
+- Updated `/tools` command to accept `on`/`off` and handle them.  
+- Added `tools_support` switch in `switches.rb` reflecting
+  `config.tools.enabled`.  
+- Refactored config to use `enabled:` and `functions:` under `tools:`.  
+- Updated all tool references to `config.tools.functions`.  
+- Guard tool calls with `@tools_support.off?` in `tool_calling.rb`.  
+- Updated `default_enabled_tools` and `list_tools` to use
+  `config.tools.functions`.  
+- Added `tools_support.show` to `information.rb`.  
+- Updated tool implementations (`directory_structure`, `execute_grep`,
+  `file_context`, `get_current_weather`, `get_cve`, `get_endoflife`,
+  `import_url`, `read_file`, `search_web`, `write_file`) to use
+  `config.tools.functions`.  
+- Adjusted specs to reference `functions` and modify `gem_path_lookup` spec to
+  return a symbol.  
+- Added `OLLAMA_CHAT_TOOLS_RUN_TESTS_TEST_RUNNER` to `.envrc` for the
+  `run_tests` tool.  
+- Removed gem system updates from Dockerfile.  
+- Removed `model_with_size` and `choose_model` from
+  `lib/ollama_chat/dialog.rb`.  
+- Added `model_with_size` and `choose_model` to
+  `lib/ollama_chat/model_handling.rb`.  
+- In `lib/ollama_chat/env_config.rb`, changed `decode` lambda to use
+  `Pathname.new(_1).join('ollama_chat').expand_path` for absolute paths.  
+- In `lib/ollama_chat/ollama_chat_config.rb`, replaced
+  `File.directory?(cache_dir_path)` with `cache_dir_path.directory?`.  
+- Updated `default_path`, `config_dir_path`, `cache_dir_path`, and
+  `database_path` to consistently return `Pathname` objects.  
+- Added `and_call_original` to source fetching test.  
+- Removed `bundle update --all` from `.all_images.yml`.  
+- Added `before:` block to `.all_images.yml` that echoes 'Preparing…' and
+  deletes `Gemfile.lock`.  
+- Added `after:` block to `.all_images.yml` that checks `$RESULT` and prints
+  success/failure messages, then deletes `Gemfile.lock`.  
+- Updated dependency `all_images` to **~>0.12**.  
+
 ## 2026-02-19 v0.0.68
 
 - Added JSON serialization for directory structure in `OllamaChat::Parsing`

data/README.md

@@ -187,7 +187,7 @@ The following commands can be given inside the chat, if prefixed by a `/`:
 /revise_last                    edit the last response in an external editor
 /output filename                save last response to filename
 /pipe command                   write last response to command's stdin
-/tools [enable|disable]         list enabled, enable, or disable tools
+/tools [enable|disable|on|off]  list enabled, enable/disable tools, support on/off
 /vim                            insert the last message into a vim server
 /quit                           to quit
 /help                           to view this help

data/Rakefile

@@ -61,7 +61,7 @@ GemHadar do
   dependency             'context_spook',         '~> 1.5'
   dependency             'infobar',               '>= 0.13.1'
   dependency             'rubyzip',               '~> 3.0'
-  development_dependency 'all_images',            '~> 0.6'
+  development_dependency 'all_images',            '~> 0.12'
   development_dependency 'rspec',                 '~> 3.2'
   development_dependency 'kramdown',              '~> 2.0'
   development_dependency 'webmock'

data/lib/ollama_chat/chat.rb

@@ -55,6 +55,7 @@ class OllamaChat::Chat
   include OllamaChat::MessageEditing
   include OllamaChat::LocationHandling
   include OllamaChat::ToolCalling
+  include OllamaChat::ConfigHandling
 
   # Initializes a new OllamaChat::Chat instance with the given command-line
   # arguments.
@@ -126,13 +127,6 @@ class OllamaChat::Chat
   #   that manages the thinking mode setting for the Ollama model interactions
   attr_reader :think_mode
 
-  # The debug method accesses the debug configuration setting.
-  #
-  # @return [TrueClass, FalseClass] the current debug mode status
-  def debug
-    OllamaChat::EnvConfig::OLLAMA::CHAT::DEBUG
-  end
-
   # The ollama reader returns the Ollama API client instance.
   #
   # @return [Ollama::Client] the configured Ollama API client
@@ -158,37 +152,6 @@ class OllamaChat::Chat
   #   messages associated with this instance
   attr_reader :messages
 
-  # Returns the links set for this object, initializing it lazily if needed.
-  #
-  # The links set is memoized, meaning it will only be created once per object
-  # instance and subsequent calls will return the same Set instance.
-  #
-  # @return [Set] A Set object containing all links associated with this instance
-  def links
-    @links ||= Set.new
-  end
-
-  class << self
-    # The config attribute accessor provides read and write access to the
-    # configuration object associated with this instance.
-    attr_accessor :config
-  end
-
-  # The config= method assigns a new configuration object to the class.
-  #
-  # @param config [ ComplexConfig::Settings ] the configuration object to be set
-  def config=(config)
-    self.class.config = config
-  end
-
-  # The config method returns the configuration object associated with the
-  # class.
-  #
-  # @return [ ComplexConfig::Settings ] the configuration instance
-  def config
-    self.class.config
-  end
-
   # The start method initializes the chat session by displaying information and
   # conversation history, then prompts the user for input to begin interacting
   # with the chat.
@@ -221,8 +184,25 @@ class OllamaChat::Chat
     OllamaChat::Vim.new(server_name, clientserver:)
   end
 
+  # The debug method accesses the debug configuration setting.
+  #
+  # @return [TrueClass, FalseClass] the current debug mode status
+  def debug
+    OllamaChat::EnvConfig::OLLAMA::CHAT::DEBUG
+  end
+
   private
 
+  # Returns the links set for this object, initializing it lazily if needed.
+  #
+  # The links set is memoized, meaning it will only be created once per object
+  # instance and subsequent calls will return the same Set instance.
+  #
+  # @return [Set] A Set object containing all links associated with this instance
+  def links
+    @links ||= Set.new
+  end
+
   # Handles user input commands and processes chat interactions.
   #
   # @param content [String] The input content to process
@@ -388,7 +368,7 @@ class OllamaChat::Chat
         STDERR.puts "Warning: No message found to insert into Vim"
       end
       :next
-    when %r(^/tools(?:\s+(enable|disable))?$)
+    when %r(^/tools(?:\s+(enable|disable|on|off))?$)
       case $1
       when nil
         list_tools
@@ -396,10 +376,19 @@ class OllamaChat::Chat
         enable_tool
       when 'disable'
         disable_tool
+      when 'on'
+        tools_support.set(true, show: true)
+      when 'off'
+        tools_support.set(false, show: true)
       end
       :next
-    when %r(^/config$)
-      display_config
+    when %r(^/config(?:\s+(edit))?$)
+      case $1
+      when 'edit'
+        edit_config
+      else
+        display_config
+      end
       :next
     when %r(^/quit$), nil
       STDOUT.puts "Goodbye."
@@ -555,22 +544,6 @@ class OllamaChat::Chat
     end
   end
 
-  # The display_config method renders the configuration and displays it using a
-  # pager.
-  # It determines an appropriate pager command based on environment variables
-  # and available system commands, then uses Kramdown::ANSI::Pager to show the
-  # formatted configuration output.
-  def display_config
-    command  = OllamaChat::EnvConfig::PAGER?
-    rendered = config.to_s
-    Kramdown::ANSI::Pager.pager(
-      lines: rendered.count(?\n),
-      command:
-    ) do |output|
-      output.puts rendered
-    end
-  end
-
   # The interact_with_user method manages the interactive loop for user input
   # and chat processing.
   # It handles reading user input, processing commands, managing messages, and
@@ -756,8 +729,6 @@ class OllamaChat::Chat
   # change_system_prompt method to handle the selection. Otherwise, it
   # retrieves the system prompt from a file or uses the default value, then
   # sets it in the message history.
-  #
-  # @return [ void ] this method returns nil after setting up the system prompt
   def setup_system_prompt
     default = config.system_prompts.default? || @model_system
     if @opts[?s] =~ /\A\?/
@@ -866,33 +837,6 @@ class OllamaChat::Chat
     end
   end
 
-  # The fix_config method handles configuration file errors by informing the
-  # user about the exception and prompting them to fix it.
-  # It then executes a diff tool to compare the current config file with the
-  # default one.
-  # This method exits the program after handling the configuration error
-  #
-  # @param exception [ Exception ] the exception that occurred while reading
-  #   the config file
-  def fix_config(exception)
-    save_conversation('backup.json')
-    STDOUT.puts "When reading the config file, a #{exception.class} "\
-      "exception was caught: #{exception.message.inspect}"
-    unless diff_tool = OllamaChat::EnvConfig::DIFF_TOOL?
-      exit 1
-    end
-    if ask?(prompt: 'Do you want to fix the config? (y/n) ') =~ /\Ay/i
-      system Shellwords.join([
-        diff_tool,
-        @ollama_chat_config.filename,
-        @ollama_chat_config.default_config_path,
-      ])
-      exit 0
-    else
-      exit 1
-    end
-  end
-
   # Enables tab completion for chat commands within the interactive session
   #
   # Temporarily replaces the current Reline completion procedure with a custom

data/lib/ollama_chat/clipboard.rb

@@ -11,6 +11,8 @@
 # @example Pasting content from input
 #   content = chat.paste_from_input
 module OllamaChat::Clipboard
+  private
+
   # Copy the last assistant's message to the system clipboard.
   #
   # This method checks if there is a last message from an assistant in the

data/lib/ollama_chat/config_handling.rb

@@ -0,0 +1,99 @@
+# Provides functionality for handling configuration files and settings for
+# OllamaChat. It loads configuration from YAML files, supports environment
+# variable overrides, and offers methods to read and write configuration data.
+# The module also ensures default values are set and validates configuration
+# structure.
+module OllamaChat::ConfigHandling
+  extend Tins::Concern
+
+  class_methods do
+    # The config attribute accessor provides read and write access to the
+    # configuration object associated with this instance.
+    attr_accessor :config
+  end
+
+  # The config method returns the configuration object associated with the
+  # class.
+  #
+  # @return [ ComplexConfig::Settings ] the configuration instance
+  def config
+    self.class.config
+  end
+
+  private
+
+  # The config= method assigns a new configuration object to the class.
+  #
+  # @param config [ ComplexConfig::Settings ] the configuration object to be set
+  def config=(config)
+    self.class.config = config
+  end
+
+  # The display_config method renders the configuration and displays it using a
+  # pager. It determines an appropriate pager command based on environment
+  # variables and available system commands, then uses Kramdown::ANSI::Pager to
+  # show the formatted configuration output.
+  def display_config
+    command  = OllamaChat::EnvConfig::PAGER?
+    rendered = config.to_s
+    Kramdown::ANSI::Pager.pager(
+      lines: rendered.count(?\n),
+      command:
+    ) do |output|
+      output.puts rendered
+    end
+  end
+
+  # The fix_config method handles configuration file errors by informing the
+  # user about the exception and prompting them to fix it. It then executes a
+  # diff tool to compare the current config file with the default one.
+  # This method exits the program after handling the configuration error.
+  #
+  # @param exception [Exception] the exception that occurred while reading
+  #   the config file
+  def fix_config(exception)
+    save_conversation('backup.json')
+    STDOUT.puts "When reading the config file, a #{exception.class} "\
+      "exception was caught: #{exception.message.inspect}"
+    unless diff_tool = OllamaChat::EnvConfig::DIFF_TOOL?
+      exit 1
+    end
+    if ask?(prompt: 'Do you want to fix the config? (y/n) ') =~ /\Ay/i
+      system Shellwords.join([
+        diff_tool,
+        @ollama_chat_config.filename,
+        @ollama_chat_config.default_config_path,
+      ])
+      exit 0
+    else
+      exit 1
+    end
+  end
+
+  # Edit the current configuration file in the editor defined by the
+  # environment variable `EDITOR`.
+  #
+  # 1. Looks up the editor command via `OllamaChat::EnvConfig::EDITOR`.
+  #    If the value is `nil` or empty, it prints an error message to
+  #    STDERR and returns immediately.
+  # 2. Invokes the editor with the path to the active configuration
+  #    file (`@ollama_chat_config.filename`). The editor is launched via
+  #    `system` so that the process inherits the current terminal,
+  #    allowing in‑place editing.
+  # 3. If editing was successful, prompts the user to restart
+  #    `ollama_chat` if desired.
+  def edit_config
+    unless editor = OllamaChat::EnvConfig::EDITOR?
+      STDERR.puts "Need the environment variable var EDITOR defined to use an editor"
+      return
+    end
+    result = system Shellwords.join([ editor, @ollama_chat_config.filename ])
+    if result
+      if ask?(prompt: "Do you want to restart #{progname}? (y/n) ") =~ /\Ay/i
+        exec($0, *ARGV)
+      end
+    else
+      STDERR.puts "Editor returned a non-zero status!"
+    end
+  end
+end

data/lib/ollama_chat/conversation.rb

@@ -12,6 +12,8 @@
 # @example Load a conversation
 #   chat.load_conversation('my_chat.json')
 module OllamaChat::Conversation
+  private
+
   # Saves the current conversation to a JSON file.
   #
   # This method delegates to the `messages` object's `save_conversation`

data/lib/ollama_chat/dialog.rb

@@ -13,50 +13,17 @@
 # @example Changing the system prompt
 #   chat.change_system_prompt('default_prompt', system: '?sherlock')
 module OllamaChat::Dialog
-  # The model_with_size method formats a model's size for display
-  # by creating a formatted string that includes the model name and its size
-  # in a human-readable format with appropriate units.
+  # The ask? method prompts the user with a question and returns their input.
   #
-  # @param model [ Object ] the model object that has name and size attributes
+  # @param prompt [ String ] the message to display to the user
   #
-  # @return [ Object ] a result object with an overridden to_s method
-  #                     that combines the model name and formatted size
-  private def model_with_size(model)
-    result = model.name
-    formatted_size = Term::ANSIColor.bold {
-      Tins::Unit.format(model.size, unit: ?B, prefix: 1024, format: '%.1f %U')
-    }
-    result.singleton_class.class_eval do
-      define_method(:to_s) { "%s %s" % [ model.name, formatted_size ] }
-    end
-    result
+  # @return [ String ] the user's response with trailing newline removed
+  def ask?(prompt:)
+    print prompt
+    STDIN.gets.to_s.chomp
   end
 
-  # The choose_model method selects a model from the available list based on
-  # CLI input or user interaction.
-  # It processes the provided CLI model parameter to determine if a regex
-  # selector is used, filters the models accordingly, and prompts the user to
-  # choose from the filtered list if needed.
-  # The method ensures that a model is selected and displays a connection
-  # message with the chosen model and base URL.
-  def choose_model(cli_model, current_model)
-    selector = if cli_model =~ /\A\?+(.*)\z/
-                 cli_model = ''
-                 Regexp.new($1)
-               end
-    models = ollama.tags.models.sort_by(&:name).map { |m| model_with_size(m) }
-    selector and models = models.grep(selector)
-    model =
-      if models.size == 1
-        models.first
-      elsif cli_model == ''
-        OllamaChat::Utils::Chooser.choose(models) || current_model
-      else
-        cli_model || current_model
-      end
-  ensure
-    connect_message(model, ollama.base_url)
-  end
+  private
 
   # The connect_message method displays a connection status message.
   #
@@ -66,16 +33,6 @@ module OllamaChat::Dialog
     STDOUT.puts green { "Connecting to #{model}@#{base_url} now…" }
   end
 
-  # The ask? method prompts the user with a question and returns their input.
-  #
-  # @param prompt [ String ] the message to display to the user
-  #
-  # @return [ String ] the user's response with trailing newline removed
-  def ask?(prompt:)
-    print prompt
-    STDIN.gets.to_s.chomp
-  end
-
   # The choose_collection method presents a menu to select or create a document
   # collection. It displays existing collections along with options to create a
   # new one or exit.

data/lib/ollama_chat/document_cache.rb

@@ -6,6 +6,8 @@
 # document management system to enable efficient storage, retrieval, and
 # similarity-based searching of documents using vector embeddings.
 module OllamaChat::DocumentCache
+  private
+
   # Retrieves the cache class specified in the configuration.
   #
   # This method resolves the cache class name from the application's

data/lib/ollama_chat/env_config.rb

@@ -20,13 +20,13 @@ module OllamaChat
     XDG_CONFIG_HOME = set do
       description 'XDG Configuration directory path'
       default { '~/.config' }
-      decode  { Pathname.new(_1) + 'ollama_chat' }
+      decode  { Pathname.new(_1).join('ollama_chat').expand_path }
     end
 
     XDG_CACHE_HOME = set do
       description 'XDG Cache directory path'
       default { '~/.cache' }
-      decode  { Pathname.new(_1) + 'ollama_chat' }
+      decode  { Pathname.new(_1).join('ollama_chat').expand_path }
     end
 
     PAGER = set do

data/lib/ollama_chat/follow_chat.rb

@@ -98,20 +98,25 @@ class OllamaChat::FollowChat
 
     response.message.tool_calls.each do |tool_call|
       name = tool_call.function.name
-      unless @chat.config.tools.attribute_set?(name)
+      unless @chat.tool_configured?(name)
         @chat.tool_call_results[name] =
           "Error: Unconfigured tool named %s ignored => Skip.\n" % name
         next
       end
-      unless OllamaChat::Tools.registered?(name)
+      unless @chat.tool_registered?(name)
         @chat.tool_call_results[name] =
           "Error: Unregistered tool named %s ignored => Skip.\n" % name
         next
       end
+      unless @chat.tool_enabled?(name)
+        @chat.tool_call_results[name] =
+          "Error: Disabled tool named %s ignored => Skip.\n" % name
+        next
+      end
       STDOUT.puts
       confirmed = true
       args = JSON.pretty_generate(tool_call.function.arguments)
-      if @chat.config.tools[name].require_confirmation?
+      if @chat.tool_function(name).require_confirmation?
         prompt = "I want to execute tool %s\n%s\nConfirm? (y/n) " % [
           bold { name },
           italic { args  },

data/lib/ollama_chat/history.rb

@@ -15,6 +15,8 @@
 # @example Clearing chat history
 #   chat.clear_history
 module OllamaChat::History
+  private
+
   # The chat_history_filename method constructs and returns the full file path
   # for the chat history file.
   #

data/lib/ollama_chat/information.rb

@@ -104,6 +104,7 @@ module OllamaChat::Information
     location.show
     voice.show
     @voice.on? and @voices.show
+    tools_support.show
     STDOUT.puts "Documents database cache is #{@documents.nil? ? 'n/a' : bold{@documents.cache.class}}"
     STDOUT.puts "Document policy for references in user text: #{bold{document_policy}}"
     STDOUT.puts "Currently selected search engine is #{bold(search_engine)}."
@@ -152,7 +153,7 @@ module OllamaChat::Information
       /revise_last                    edit the last response in an external editor
       /output filename                save last response to filename
       /pipe command                   write last response to command's stdin
-      /tools [enable|disable]         list enabled, enable, or disable tools
+      /tools [enable|disable|on|off]  list enabled, enable/disable tools, support on/off
       /vim                            insert the last message into a vim server
       /quit                           to quit
       /help                           to view this help

data/lib/ollama_chat/input_content.rb

@@ -8,6 +8,8 @@ require 'tempfile'
 # interactive file selection and context collection for enhancing chat
 # interactions with local or remote content.
 module OllamaChat::InputContent
+  private
+
   # The input method selects and reads content from files matching a pattern.
   #
   # This method prompts the user to select files matching the given glob

data/lib/ollama_chat/message_editing.rb

@@ -5,6 +5,8 @@
 # in the conversation, whether it's a system prompt, user message, or assistant
 # response.
 module OllamaChat::MessageEditing
+  private
+
   # The revise_last method opens the last message in an external editor for
   # modification.
   #

data/lib/ollama_chat/model_handling.rb

@@ -16,6 +16,8 @@
 # @example Ensuring a model is available locally
 #   chat.pull_model_unless_present('phi3', {})
 module OllamaChat::ModelHandling
+  private
+
   # The model_present? method checks if the specified Ollama model is available.
   #
   # @param model [ String ] the name of the Ollama model
@@ -69,4 +71,49 @@ module OllamaChat::ModelHandling
     warn "Caught #{e.class} while pulling model: #{e} => Exiting."
     exit 1
   end
+
+  # The model_with_size method formats a model's size for display
+  # by creating a formatted string that includes the model name and its size
+  # in a human-readable format with appropriate units.
+  #
+  # @param model [ Object ] the model object that has name and size attributes
+  #
+  # @return [ Object ] a result object with an overridden to_s method
+  #                     that combines the model name and formatted size
+  private def model_with_size(model)
+    result = model.name
+    formatted_size = Term::ANSIColor.bold {
+      Tins::Unit.format(model.size, unit: ?B, prefix: 1024, format: '%.1f %U')
+    }
+    result.singleton_class.class_eval do
+      define_method(:to_s) { "%s %s" % [ model.name, formatted_size ] }
+    end
+    result
+  end
+
+  # The choose_model method selects a model from the available list based on
+  # CLI input or user interaction.
+  # It processes the provided CLI model parameter to determine if a regex
+  # selector is used, filters the models accordingly, and prompts the user to
+  # choose from the filtered list if needed.
+  # The method ensures that a model is selected and displays a connection
+  # message with the chosen model and base URL.
+  def choose_model(cli_model, current_model)
+    selector = if cli_model =~ /\A\?+(.*)\z/
+                 cli_model = ''
+                 Regexp.new($1)
+               end
+    models = ollama.tags.models.sort_by(&:name).map { |m| model_with_size(m) }
+    selector and models = models.grep(selector)
+    model =
+      if models.size == 1
+        models.first
+      elsif cli_model == ''
+        OllamaChat::Utils::Chooser.choose(models) || current_model
+      else
+        cli_model || current_model
+      end
+  ensure
+    connect_message(model, ollama.base_url)
+  end
 end