ollama_chat 0.0.107 → 0.0.109

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e2686defd34fe5cf4e0f61c980480578f6b2589bc429d7b724ed4f3cb478d7b8
-  data.tar.gz: 8bd9136e0e6f3e202c93f8c4aec0a4b482a447bfd1e29163b0dff1973e64f2fa
+  metadata.gz: e6c85b95f4c5051b6b0f257dd8f63ee2a3997146c527fd8af251458fee641039
+  data.tar.gz: 45a66e25a1a92d8cbd86253a991ad28eeb48d1a8ca68c1f8bbb8df90bbbba37f
 SHA512:
-  metadata.gz: 1a177889fb72f197a05895b60c7e73f1493a119fcc7832354772097f121d53b150ea500fdf50ae81bdeb05161f0a3121af204496d54a708a13c427096e126159
-  data.tar.gz: 55c0270902a86e39e1704d62a4478daf0d04526312d47eb35f560b3b95218045d6699dd428b874fa9059ce3af8585bed97a5c2bd662751ef05885baa452a9c6b
+  metadata.gz: a401173a7deb626af01ad8c9adf358259a6ffd7ea292644bb1a5a4553df2f70e9c74bb4ff0807d04914a7aa47e8140e81e0bd46745e33900962d8559950ea914
+  data.tar.gz: 6b04fcce611684e73bbe54cdadb01859fc4219f57960604a64c3037b414e56a02c7389a5bfef2341d0a2aa2a44d8a723761469f9c677ec8784c2c9d9e8b6fae0

data/CHANGES.md

@@ -1,5 +1,123 @@
 # Changes
 
+## 2026-08-18 v0.0.109
+
+### Added
+
+*   Support for exporting and importing model options via the `/model` command:
+    *   New `export_model_options` method in `model_handling.rb` to export
+        stored model profiles to a JSON file.
+    *   New `import_model_options` method in `model_handling.rb` to import
+        profiles from a JSON file, featuring selective model picking, diff
+        display, and override confirmation.
+    *   Extended `/model` command with `options export` and `options import`
+        subcommands.
+*   `-p` pattern flag for `/model options import` to narrow file search, with a
+    fallback to `**/*.json`.
+*   `-i` flag for `/paste` command to read input from stdin:
+    *   New `paste_from_stdin` method in `clipboard.rb`.
+    *   Updated `/paste` command regex and parsing to support both `-e` and
+        `-i` flags.
+*   `options` metadata to command definitions for `/copy`, `/paste`, `/toggle`,
+    and `/model`.
+*   Comprehensive test coverage for previously untested or under-tested
+    modules:
+    *   `spec/ollama_chat/conversation_spec.rb` (9 examples).
+    *   `spec/ollama_chat/favourites_management_spec.rb` (15 examples).
+    *   `spec/ollama_chat/rag_handling_spec.rb` covering
+        `OllamaChat::RAGHandling`.
+    *   `spec/ollama_chat/prompt_handling_spec.rb` (19 examples).
+    *   `spec/ollama_chat/prompt_management_spec.rb` (47 examples).
+    *   `spec/ollama_chat/system_prompt_management_spec.rb` (20 examples).
+    *   `spec/ollama_chat/config_handling_spec.rb` (15 examples).
+    *   `spec/ollama_chat/web_searching_spec.rb` (8 examples).
+    *   `spec/ollama_chat/information_spec.rb` (5 examples).
+
+### Changed
+
+*   Refactored session management to extract and reuse
+    `set_previous_session_on_change` as a shared private method in
+    `lib/ollama_chat/session_management.rb`.
+    *   `set_new_session`, `duplicate_session`, and `change_session` now
+        delegate to this method for consistent `@previous_session_id` handling.
+*   Migrated test stubs from `allow` to `expect` across multiple spec files to
+    eliminate false-green tests where stubs were never invoked.
+*   Normalized `context` and `name` arguments in `prompt_handling.rb` to handle
+    `Symbol` inputs safely in SQL queries.
+*   Updated `s.files` and `s.test_files` in `ollama_chat.gemspec` to include
+    new configuration, asset, and test files.
+
+### Fixed
+
+*   Infinite retry loop in `kramdown_ansi.rb` by using `retried ||= false` to
+    persist the guard across `retry` calls.
+*   Incorrect YARD `@return` documentation for `#info_prompt` in
+    `prompt_management.rb`.
+
+## 2026-08-15 v0.0.108
+
+### Added
+
+- Added `require 'io/console'` to support `IO#raw` for key reads.
+- Added `max` to `THINK_MODE_STATES` in `think_control.rb` to support newer
+  Ollama releases (e.g., **0.32.13**).
+- Added migration `008_drop_think_mode_constraint.rb` to drop the `think_mode`
+  `CHECK` constraint in SQLite.
+- Added `.obsidian` to `.gitignore` and the `GemHadar` `ignore` list in
+  `Rakefile`.
+- Added `require_confirmation: true` to `get_current_weather`, `get_cve`,
+  `get_endoflife`, `browse`, `get_jira_issue`, `get_rfc`, and `get_ghr`.
+
+### Changed
+
+- Refactored `confirm?` key reads to use `STDIN.raw(min:, time:, intr: true)`
+  instead of `stty` commands.
+- Updated `confirm?` to print prompts to `output` instead of `STDOUT`.
+- Updated glyphs for confirmation: `\u2328\uFE0F` to `\u2328`, `☑️` to `\u2611`,
+  and `🚫` to `\u{1F6AB}`.
+- Scoped `Reline` histories for various interactive prompts:
+  - `duplicate_session` name prompt now scoped to `:session_name`.
+  - `determine_valid_new_name_for_persona` now scoped to `:persona_name`.
+  - `add_persona` `[FILES]` patterns prompt now scoped to `:patterns`.
+  - `add_new_prompt` `[FILES]` patterns prompt now scoped to `:patterns`.
+  - `determine_valid_output_filename` loop now scoped to `:filename`.
+  - `/session summarize -f` filename prompt now scoped to `:filename`.
+  - `choose_profile_for_model` profile name prompt now scoped to
+    `:profile_name`.
+- Scoped `Reline` histories for RAG prompts:
+  - `rename_collection` now scopes to `:collection_name`.
+  - `create_collection` scopes to `:collection_name`,
+    `:collection_description`, and `:patterns`.
+  - `edit_collection` description prompt now uses `:collection_description`.
+- Refactored terminal truncation by moving `truncate_for_terminal` from
+  `prepare_last_message` to `last_message_with_user`.
+- Added `height:` keyword argument to `last_message_with_user` and
+  `display_formatted_terminal_output`.
+- Refactored `display_formatted_terminal_output` to use `move_home` and
+  `erase_in_display(nil)` instead of `clear_screen` to prevent `tmux` flicker.
+- Updated `require_confirmation` to `false` for `open_file_in_editor`,
+  `paste_into_editor`, and `generate_image`.
+- Removed `jq` parsing from `OLLAMA_CHAT_TOOLS_TEST_RUNNER` in `.envrc`.
+- Introduced `OLLAMA_CHAT_COLLECTION_NAME_REGEXP` in `lib/ollama_chat.rb` to
+  enforce safe identifiers.
+- Updated `OLLAMA_CHAT::Chat#initial_collection` to validate names against the
+  new regex.
+- Wrapped web search link list output in `use_pager` within
+  `OLLAMA_CHAT::WebSearching`.
+
+### Fixed
+- Added `rescue` for `Interrupt` and `IRB::Abort` during key reads to return
+  `nil` and print `"\u274C\uFE0F"`.
+- Fixed `tmux` flicker in vertically split panes by avoiding full-pane redraws.
+- Fixed `OLLAMA_CHAT::Tools::PatchFile` to explicitly warn against redundant
+  tool calls.
+
+### Refactored
+- Updated YARD `@param output` and `@return` documentation.
+- Simplified `last_message_with_user` to return a single joined string.
+- Simplified `OLLAMA_CHAT::Tools::SearchKnowledge` to raise
+  `OLLAMA_CHAT::ToolFunctionArgumentError` on invalid input.
+
 ## 2026-08-09 v0.0.107
 
 ### Added

data/Rakefile

@@ -20,7 +20,7 @@ GemHadar do
   test_dir    'spec'
   ignore      '.*.sw[pon]', 'pkg', 'Gemfile.lock', '.AppleDouble', '.bundle',
     '.yardoc', 'doc', 'tags', 'corpus', 'coverage', '/config/searxng/*',
-    '.starscope.db', 'cscope.out', 'errors.lst', 'tmp/*'
+    '.starscope.db', 'cscope.out', 'errors.lst', 'tmp/*', '.obsidian'
   package_ignore '.all_images.yml', '.tool-versions', '.gitignore', 'VERSION',
     '.rspec', '.github', '.contexts', '.envrc', '.yardopts'
 

data/lib/ollama_chat/chat.rb

@@ -229,11 +229,14 @@ class OllamaChat::Chat
   #
   # @return [ Symbol ] the collection name symbol
   def initial_collection
-    (
-      session&.current_collection.full? ||
+    c = session&.current_collection.full? ||
       config.embedding.collection.full? ||
-      :default
-    ).to_sym
+      'default'
+    if c =~ /\A#{OllamaChat::COLLECTION_NAME_REGEXP.source}\z/
+      c
+    else
+      'default'
+    end.to_sym
   end
 
   # The initial_system_prompt method returns the system prompt for the initial

data/lib/ollama_chat/clipboard.rb

@@ -143,4 +143,15 @@ module OllamaChat::Clipboard
     log(:error, e, data: { method: __method__ })
     STDERR.puts e.message
   end
+
+  # Read the full text from STDIN and return it.
+  #
+  # @param edit [Boolean] If true, the text is passed through
+  #   {#edit_text} before being returned.
+  # @return [String] The raw (or edited) content read from STDIN.
+  def paste_from_stdin(edit: false)
+    text = STDIN.read
+    edit and text = edit_text(text)
+    text
+  end
 end

data/lib/ollama_chat/commands.rb

@@ -27,6 +27,7 @@ module OllamaChat::Commands
   command(
     name: :copy,
     regexp: %r(^/copy(\s+-e)?\s*$),
+    options: '[-e]',
     help: <<~EOT
       📋 Copy the last response to the clipboard.
          Options: -e to edit before copying.
@@ -39,15 +40,22 @@ module OllamaChat::Commands
 
   command(
     name: :paste,
-    regexp: %r(^/paste(\s+-e)?\s*$),
+    regexp: %r(^/paste(\s+-[ie])*\s*$),
+    options: '[-e|-i]',
     help: <<~EOT
-      📋 Paste content from the clipboard.
-         Options: -e to edit after pasting.
+      📋 Paste content from the clipboard or stdin.
+         Options:
+           -e  Edit after pasting.
+           -i  Read from stdin instead of clipboard.
     EOT
   ) do |opts|
     disable_content_parsing
-    opts = go_command('e', opts)
-    paste_from_clipboard(edit: opts[?e])
+    opts = go_command('ie', opts)
+    if opts[?i]
+      paste_from_stdin(edit: opts[?e])
+    else
+      paste_from_clipboard(edit: opts[?e])
+    end
   end
 
   ## Settings
@@ -88,6 +96,7 @@ module OllamaChat::Commands
     name: :toggle,
     regexp: %r(^/toggle(?:\s+(markdown|stream|location|runtime_info|voice|think_loud|think_strip|embedding)(?:\s+(-[yn]))?)?$),
     complete: [ 'toggle', %w[ markdown stream location runtime_info voice think_loud think_strip embedding ] ],
+    options: '[-y|-n]',
     help: <<~EOT
       🎛️ Toggle feature switches
       (markdown, stream, location, runtime_info,
@@ -139,17 +148,21 @@ module OllamaChat::Commands
 
   command(
     name: :model,
-    regexp: %r(^/model(?:\s+(change|options(?: copy| delete)?|options from session|options to session))?((?:\s+(?:-m))*)$),
-    complete: [ 'model', %w[ change options options\ copy options\ delete options\ from\ session options\ to\ session ] ],
+    regexp: %r(^/model(?:\s+(change|options(?: copy| delete| export| import)?|options from session|options to session))?((?:\s+(?:-m|-p\s+[\S]+))*)$),
+    complete: [ 'model', %w[ change options options\ copy options\ delete options\ export options\ import options\ from\ session options\ to\ session ] ],
+    options: '[-m|-p pattern]',
     help: <<~EOT
       🤖 Manage AI models & profiles:
          - change: Switch active model
          - options: Edit saved profile config
          - options copy: Copy profile from another model
          - options delete: Delete a saved profile
+         - options export: Export ALL model profiles to JSON
+         - options import: Selectively import profiles from JSON
          - options from session: Save live → Saved
          - options to session: Apply Saved → Live
          -m interactively choose a model
+         -p PATTERN narrow file search for import
     EOT
   ) do |subcommand, opts|
     if subcommand == 'change'
@@ -163,7 +176,7 @@ module OllamaChat::Commands
       end
       next :next
     end
-    opts    = go_command('m', opts)
+    opts    = go_command('mp:', opts)
     model   = opts[?m] ? choose_model('', @model) : @model
     case subcommand
     when 'options'
@@ -179,6 +192,12 @@ module OllamaChat::Commands
     when 'options to session'
       profile = choose_profile_for_model(model) || 'default'
       copy_model_options_to_session(model, profile:)
+    when 'options export'
+      export_model_options
+    when 'options import'
+      pattern  = opts[?p] || '**/*.json'
+      filename = choose_filename(pattern) or next :next
+      import_model_options(filename)
     end
     :next
   end
@@ -296,8 +315,10 @@ module OllamaChat::Commands
     when 'summarize'
       opts = go_command('fs', opts)
       if opts[?f] and
-          filename = ask?(prompt: "❓ Enter filename: ").full? { Pathname.new(_1) }
-        then
+        filename = switch_history(:filename) {
+          ask?(prompt: "❓ Enter filename: ")
+        }.full? { Pathname.new(_1) }
+      then
         if filename.exist? && !confirm?(
             prompt: "🔔 File #{filename.to_s.inspect} already exists, overwrite? (y/n) ",
             yes: /\Ay/i
@@ -807,7 +828,7 @@ module OllamaChat::Commands
 
   command(
     name: :input,
-    regexp: %r(^/input(?:\s+(path|context|embedding|summary)(?:\s*(?=\z))?)?((?:\s+-(?:[apre]|c\s*\w+|w\s*\d+|t\s*[-\w\.]+(?:,[-\w\.]+)*))*)(?:\s+([^-].*))?$),
+    regexp: %r(^/input(?:\s+(path|context|embedding|summary)(?:\s*(?=\z))?)?((?:\s+-(?:[apre]|c\s*#{OllamaChat::COLLECTION_NAME_REGEXP.source}|w\s*\d+|t\s*[-\w\.]+(?:,[-\w\.]+)*))*)(?:\s+([^-].*))?$),
     optional: true,
     complete: [ 'input', %w[ path context embedding summary ] ],
     options: <<~EOT,

data/lib/ollama_chat/database/migrations/008_drop_think_mode_constraint.rb

@@ -0,0 +1,130 @@
+# Removes the `think_mode` CHECK constraint from the `sessions` table.
+#
+# The set of valid think modes is model-dependent and evolving upstream
+# (e.g., `max` was only added mid-2026, and numeric token budgets are
+# under discussion), so the schema should not pin a fixed vocabulary.
+# State validation is handled in Ruby via
+# `OllamaChat::ThinkControl::THINK_MODE_STATES` and the state selectors.
+#
+# SQLite cannot alter CHECK constraints in place, so the table is rebuilt
+# using the same shadow-table pattern as
+# `004_add_profile_to_model_options.rb`.
+Sequel.migration do
+  up do
+    create_table(:sessions_new) do
+      primary_key :id
+      String :name, null: false
+      String :current_model
+      String :current_collection
+      String :default_persona_name
+      String :current_system_prompt
+      Boolean :tools_enabled, null: false
+      Text :tools_default_enabled
+      String :think_mode, null: false
+      Bool :think_loud_enabled, null: false
+      Bool :embedding_enabled, null: false
+      String :document_policy, null: false
+      Bool :runtime_info_enabled, null: false
+      Bool :think_strip_enabled, null: false
+      Boolean :markdown_enabled, null: false
+      Boolean :stream_enabled, null: false
+      Boolean :location_enabled, null: false
+      Boolean :voice_enabled, null: false
+      String :current_voice, null: false
+      Text :working_directory
+      Integer :locked_by_pid
+      Text :model_options
+      Text :messages, null: false
+      Text :links, null: false, default: ''
+      Text :history, null: false, default: ''
+      String :context_format, null: false, default: 'JSON'
+      Time :created_at
+      Time :updated_at
+      unique [:name]
+
+      constraint(
+        :document_policy,
+        document_policy: %w[ ignoring embedding importing summarizing ]
+      )
+    end
+
+    source = from(:sessions).select(
+      :id, :name, :current_model, :current_collection,
+      :default_persona_name, :current_system_prompt,
+      :tools_enabled, :tools_default_enabled, :think_mode,
+      :think_loud_enabled, :embedding_enabled, :document_policy,
+      :runtime_info_enabled, :think_strip_enabled,
+      :markdown_enabled, :stream_enabled, :location_enabled,
+      :voice_enabled, :current_voice, :working_directory,
+      :locked_by_pid, :model_options, :messages, :links,
+      :history, :context_format, :created_at, :updated_at
+    )
+    from(:sessions_new).insert(source)
+
+    drop_table(:sessions)
+    rename_table(:sessions_new, :sessions)
+  end
+
+  down do
+    # Restores the original `think_mode` constraint. All rows that
+    # predate the up-migration satisfy it; any newer value (e.g. `max`)
+    # would fail the insert, which is the desired loud failure on
+    # rollback.
+    create_table(:sessions_old) do
+      primary_key :id
+      String :name, null: false
+      String :current_model
+      String :current_collection
+      String :default_persona_name
+      String :current_system_prompt
+      Boolean :tools_enabled, null: false
+      Text :tools_default_enabled
+      String :think_mode, null: false
+      Bool :think_loud_enabled, null: false
+      Bool :embedding_enabled, null: false
+      String :document_policy, null: false
+      Bool :runtime_info_enabled, null: false
+      Bool :think_strip_enabled, null: false
+      Boolean :markdown_enabled, null: false
+      Boolean :stream_enabled, null: false
+      Boolean :location_enabled, null: false
+      Boolean :voice_enabled, null: false
+      String :current_voice, null: false
+      Text :working_directory
+      Integer :locked_by_pid
+      Text :model_options
+      Text :messages, null: false
+      Text :links, null: false, default: ''
+      Text :history, null: false, default: ''
+      String :context_format, null: false, default: 'JSON'
+      Time :created_at
+      Time :updated_at
+      unique [:name]
+
+      constraint(
+        :think_mode,
+        think_mode: %w[ disabled enabled low medium high ]
+      )
+      constraint(
+        :document_policy,
+        document_policy: %w[ ignoring embedding importing summarizing ]
+      )
+    end
+
+    source = from(:sessions).select(
+      :id, :name, :current_model, :current_collection,
+      :default_persona_name, :current_system_prompt,
+      :tools_enabled, :tools_default_enabled, :think_mode,
+      :think_loud_enabled, :embedding_enabled, :document_policy,
+      :runtime_info_enabled, :think_strip_enabled,
+      :markdown_enabled, :stream_enabled, :location_enabled,
+      :voice_enabled, :current_voice, :working_directory,
+      :locked_by_pid, :model_options, :messages, :links,
+      :history, :context_format, :created_at, :updated_at
+    )
+    from(:sessions_old).insert(source)
+
+    drop_table(:sessions)
+    rename_table(:sessions_old, :sessions)
+  end
+end

data/lib/ollama_chat/dialog.rb

@@ -1,3 +1,5 @@
+require 'io/console'
+
 # A module that provides interactive selection and configuration functionality
 # for OllamaChat.
 #
@@ -35,27 +37,32 @@ module OllamaChat::Dialog
   # @param default  [Object, nil]  value returned when the timeout expires
   #   (defaults to `nil`)
   # @param yes      [Object, nil]  value that is considered a positive response
-  # @param output   [IO]  the IO object to write the prompt to
+  # @param output   [IO]  the IO object to write the prompt and the answer to
   #
-  # @return [Object] the character entered by the user, or the `default` value
-  #   if a timeout occurs
+  # @return [Object] the character entered by the user, the `default` value
+  #   if a timeout occurs, or `nil` if the read is interrupted (e.g. Ctrl-C)
   def confirm?(prompt:, timeout: nil, default: nil, yes: nil, output: STDOUT)
     return default if timeout&.zero?
     if prompt.include?('%s')
       prompt = prompt % (timeout ? ('timeout in %us' % timeout) : 'no timeout')
     end
-    print prompt
-    system 'stty raw'
-    keypress = nil
-    c = if timeout
-          keypress = !!IO.select([ STDIN ], nil, nil, timeout)
-          keypress ? STDIN.getc : nil
-        else
-          keypress = true
-          STDIN.getc
-        end
-    system 'stty cooked'
-    answer = c || default
+    output.print prompt
+    min    = 1
+    time   = 0
+    if timeout
+      min  = 0
+      time = timeout
+    end
+    exceptions = [ (IRB::Abort if defined? IRB), Interrupt ].compact
+    begin
+      keypress = STDIN.raw(min:, time:, intr: true) do |io|
+        io.getc
+      end
+    rescue *exceptions
+      output.puts "\u274C\uFE0F"
+      return
+    end
+    answer = keypress || default
     case
     when yes.nil?
       if keypress
@@ -68,12 +75,12 @@ module OllamaChat::Dialog
       if keypress
         output.puts "\u2705\uFE0F #{answer}"
       else
-        output.puts "☑️  #{answer}"
+        output.puts "\u2611\uFE0F #{answer}"
       end
       answer
     else
       if keypress
-        output.puts "🚫 #{answer}"
+        output.puts "\u{1F6AB} #{answer}"
       else
         output.puts "\u231B\uFE0F #{answer}"
       end

data/lib/ollama_chat/file_editing.rb

@@ -111,20 +111,22 @@ module OllamaChat::FileEditing
   # @return [Pathname, nil] the validated filename as a Pathname, or nil if the
   #   operation was cancelled
   def determine_valid_output_filename(action)
-    loop do
-      filename_str = ask?(
-        prompt: "❓ Enter filename #{action}, C-c ⇒ cancel: "
-      )
-      if filename_str.nil?
-        STDOUT.puts "Cancelled."
-        return nil
-      end
+    switch_history(:filename) do
+      loop do
+        filename_str = ask?(
+          prompt: "❓ Enter filename #{action}, C-c ⇒ cancel: "
+        )
+        if filename_str.nil?
+          STDOUT.puts "Cancelled."
+          return nil
+        end
 
-      filename = Pathname.new(filename_str)
-      if filename.exist?
-        STDERR.puts "File #{filename.to_path.inspect} already exists!"
-      else
-        return filename
+        filename = Pathname.new(filename_str)
+        if filename.exist?
+          STDERR.puts "File #{filename.to_path.inspect} already exists!"
+        else
+          return filename
+        end
       end
     end
   end

data/lib/ollama_chat/follow_chat.rb

@@ -39,6 +39,9 @@ class OllamaChat::FollowChat
                    end
   end
 
+  # Returns the chat object representing the conversation context.
+  #
+  # @return [OllamaChat::Chat] The chat object.
   attr_reader :chat
 
   # Returns the conversation history (an array of message objects).
@@ -74,7 +77,7 @@ class OllamaChat::FollowChat
       ensure_assistant_response_exists
       update_last_message(response)
       if chat.stream.on?
-        display_formatted_terminal_output
+        display_formatted_terminal_output(height: Tins::Terminal.lines - 1)
       else
         if display_output
           display_formatted_terminal_output
@@ -293,9 +296,9 @@ class OllamaChat::FollowChat
   def prepare_last_message
     content, thinking = @messages.last.content, @messages.last.thinking
     if chat.markdown.on?
-      content = talk_annotate { truncate_for_terminal chat.kramdown_ansi_parse(content) }
+      content = talk_annotate { chat.kramdown_ansi_parse(content) }
       if chat.think?
-        thinking = think_annotate { truncate_for_terminal chat.kramdown_ansi_parse(thinking) }
+        thinking = think_annotate { chat.kramdown_ansi_parse(thinking) }
       end
     else
       content = talk_annotate { content }
@@ -304,20 +307,28 @@ class OllamaChat::FollowChat
     return content&.chomp, thinking&.chomp
   end
 
-  # The last_message_with_user method constructs a formatted message array by
+  # The last_message_with_user method constructs a formatted string by
   # combining user information, newline characters, thinking annotations, and
   # content for display in the terminal output.
   #
-  # @return [ Array ] an array containing the user identifier, newline
-  #   character, thinking annotation (if present), and content formatted for
-  #   terminal display
-  def last_message_with_user
+  # @param height [Integer] the maximum number of lines to truncate the output
+  #   to. Defaults to 0 (no truncation).
+  #
+  # @return [String] the formatted message string containing the user
+  #   identifier, newline character, thinking annotation (if present), and
+  #   content formatted for terminal display
+  def last_message_with_user(height: 0)
     content, thinking = prepare_last_message
-    if thinking.present?
-      [ @sender + ?:, ?\n, thinking, ?\n, content ]
-    else
-      [ @sender + ?:, ?\n, content ]
+    content ||= ''
+    result = if thinking.present?
+               [ @sender + ":\n", thinking, ?\n, content ]
+             else
+               [ @sender + ":\n", content ]
+             end * ''
+    if height > 0
+      result = truncate_for_terminal(result, max_lines: height)
     end
+    result
   end
 
   # The display_formatted_terminal_output method formats and outputs the
@@ -327,9 +338,17 @@ class OllamaChat::FollowChat
   # move home commands. The method takes into account whether markdown and
   # thinking modes are enabled to determine how to process and display the
   # content.
-  def display_formatted_terminal_output(output = nil)
+  #
+  # @param output [IO, nil] the output stream to print to. Defaults to
+  #   `@output`.
+  # @param height [Integer] the maximum number of lines to truncate the output
+  #   to. Defaults to 0 (no truncation).
+  def display_formatted_terminal_output(output = nil, height: 0)
     output ||= @output
-    output.print(*([ clear_screen, move_home, *last_message_with_user ].compact))
+    output.print(
+      move_home, erase_in_display(nil),
+      last_message_with_user(height:)
+    )
   end
 
   # The display_output method shows the last message in the conversation.
@@ -345,7 +364,7 @@ class OllamaChat::FollowChat
       if chat.markdown.on?
         display_formatted_terminal_output(output)
       else
-        output.print(*last_message_with_user)
+        output.print(last_message_with_user)
       end
     end
   end

data/lib/ollama_chat/kramdown_ansi.rb

@@ -39,7 +39,7 @@ module OllamaChat::KramdownANSI
   # @return [ String ] the content formatted with ANSI escape sequences
   #   according to the configured styles
   def kramdown_ansi_parse(content)
-    retried = false
+    retried ||= false
     content.nil? and return ''
     Kramdown::ANSI.parse(content, ansi_styles: @kramdown_ansi_styles)
   rescue Encoding::UndefinedConversionError