ollama_chat 0.0.71 → 0.0.72

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e83451e14cc4c204b237950b58832b231440487cf490e9434ee127654e849e22
-  data.tar.gz: 8fb94fb04d963e05d778edc13af7c0988f3d7f180ea17af90d51a6d5bf07a74c
+  metadata.gz: aba37f4f59280b1d21b76df79cb20b1e6d4f1b7e05fd91dfd6940368ff8a168f
+  data.tar.gz: 88d8d50a503c98973159f2874c0f04a4139e27469363c77b66560e0a965cf890
 SHA512:
-  metadata.gz: b2cca9e66e8063fd891ffafeac1c32c7a18467056d4b64c04476cb31154be6f23b6fd13ad3b38299c63e73bdaa73813048cf7e89150d398f4dd21de39a4c3d3a
-  data.tar.gz: cd134c0ff8544014765e792e104c76a47c1511d4f219665364eda23e52c7b6dcd382058d935c54658ef93407a0a697227cc2cef88510b8555c4a294ac4102d9e
+  metadata.gz: 9bc913077bc16ded309925188841e5f803521df0582a47c747fd20c31b03b4a15cfdabf324f683a910d3f9de4f2003b8b31867a4da052e84c072736033f613a2
+  data.tar.gz: 9df32e077b73b959ddff9754241fe10a507b8b861e917c59dd6a6032ed4eb4d520ce8bf3047d4401f06e528c17de55d2cba12d26137d52b8fc69eeddf0d17045

data/CHANGES.md

@@ -1,5 +1,33 @@
 # Changes
 
+## 2026-03-04 v0.0.72
+
+- Introduced `OllamaChat::PersonaeManagement` module and `/persona` command
+  with subcommands `add`, `delete`, `edit`, `list`, `play`, `file`, `info`,
+  `load`.  
+- Added `/config reload` subcommand to `OllamaChat::Chat`; handler
+  `reload_config` in `OllamaChat::ConfigHandling` prompts for confirmation,
+  calls `save_conversation`, then restarts with `exec($0,*ARGV)`.  
+- Added `-p` command‑line option (`-p PERSONA`) to load personas at startup.  
+- Created `OllamaChat::Pager` module for pager functionality.  
+- Created `OllamaChat::FileEditing` module; refactored `edit_config`,
+  `compose`, `revise_last` to use `edit_file`; added `vim(server_name = nil)`
+  returning `OllamaChat::Vim` and `perform_insert(text:, content:)`.  
+- Added LLM‑driven tool `insert_into_editor`
+  (`lib/ollama_chat/tools/insert_into_editor.rb`) with RSpec tests; default
+  config (`default_config.yml`) now enables confirmation flag for this tool.  
+- Enhanced `copy_to_clipboard` tool: accepts optional `text` parameter,
+  includes helper `last_message_content`, returns `true` on success, updated
+  error messages, added test for custom text.  
+- Added `bytes_written` tracking to `write_file` tool; fixed return hash key.  
+- Added visual separator for tool output.  
+- Renamed `env_config.rb` to `oc.rb`; updated requires accordingly.  
+- Added `save_history` call after conversation backup to persist history before
+  restart.  
+- Updated documentation and help messages to reflect new commands and options.  
+- Standardised type hints across codebase, replacing generic Boolean comments
+  with explicit `[ true, false ]`.  
+
 ## 2026-02-26 v0.0.71
 
 - Add new clipboard utilities: `tools/copy_to_clipboard.rb` and

data/README.md

@@ -58,6 +58,7 @@ Usage: ollama_chat [OPTIONS]
   -u URL         the ollama base url, OLLAMA_URL
   -m MODEL       the ollama model to chat with, OLLAMA_CHAT_MODEL, ?selector
   -s SYSTEM      the system prompt to use as a file, OLLAMA_CHAT_SYSTEM, ?selector
+  -p PERSONA     load a persona via name/file for roleplay at startup
   -c CHAT        a saved chat conversation to load
   -C COLLECTION  name of the collection used in this conversation
   -D DOCUMENT    load document and add to embeddings collection (multiple)
@@ -151,46 +152,47 @@ subject - the young, blue-eyed cat.
 The following commands can be given inside the chat, if prefixed by a `/`:
 
 ```
-/reconnect                      reconnect to current ollama server
-/copy                           to copy last response to clipboard
-/paste                          to paste content
-/markdown                       toggle markdown output
-/stream                         toggle stream output
-/location                       toggle location submission
-/voice [change]                 toggle voice output or change the voice
-/last [n]                       show the last n / 1 system/assistant message
-/list [n]                       list the last n / all conversation exchanges
-/clear [what]                   clear what=messages|links|history|tags|all
-/clobber                        clear the conversation, links, and collection
-/drop [n]                       drop the last n exchanges, defaults to 1
-/model                          change the model
-/system [show]                  change/show system prompt
-/prompt                         prefill user prompt with preset prompts
-/regenerate                     the last answer message
-/collection [clear|change]      change (default) collection or clear
-/info                           show information for current session
-/config                         output current configuration ("/Users/flori/.config/ollama_chat/config.yml")
-/document_policy                pick a scan policy for document references
-/think                          choose ollama think mode setting for models
-/think_loud                     enable to think out loud instead of silently
-/import source                  import the source's content
-/summarize [n] source           summarize the source's content in n words
-/embedding                      toggle embedding paused or not
-/embed source                   embed the source's content
-/web [n] query                  query web & for n(=1) results (policy: importing)
-/links [clear]                  display (or clear) links used in the chat
-/save filename                  store conversation messages
-/load filename                  load conversation messages
-/compose                        compose content using an EDITOR
-/input [pattern]                select and read content from a file (default: **/*)
-/context [pattern...]           collect context with glob patterns
-/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|on|off]  list enabled, enable/disable tools, support on/off
-/vim                            insert the last message into a vim server
-/quit                           to quit
-/help [me]                      to view this help (me = interactive ai help)
+/reconnect                                         reconnect to current ollama server
+/copy                                              to copy last response to clipboard
+/paste                                             to paste content
+/markdown                                          toggle markdown output
+/stream                                            toggle stream output
+/location                                          toggle location submission
+/voice [change]                                    toggle voice output or change the voice
+/last [n]                                          show the last n / 1 system/assistant message
+/list [n]                                          list the last n / all conversation exchanges
+/clear [what]                                      clear what=messages|links|history|tags|all
+/clobber                                           clear the conversation, links, and collection
+/drop [n]                                          drop the last n exchanges, defaults to 1
+/model                                             change the model
+/system [show]                                     change/show system prompt
+/prompt                                            prefill user prompt with preset prompts
+/persona add|delete|edit|file|info|list|load|play  manage and load/play personas for roleplay
+/regenerate                                        the last answer message
+/collection [clear|change]                         change (default) collection or clear
+/info                                              show information for current session
+/config [edit|reload]                              output/edit/reload current configuration ("/Users/flori/.config/ollama_chat/config.yml")
+/document_policy                                   pick a scan policy for document references
+/think                                             choose ollama think mode setting for models
+/think_loud                                        enable to think out loud instead of silently
+/import source                                     import the source's content
+/summarize [n] source                              summarize the source's content in n words
+/embedding                                         toggle embedding paused or not
+/embed source                                      embed the source's content
+/web [n] query                                     query web & for n(=1) results (policy: importing)
+/links [clear]                                     display (or clear) links used in the chat
+/save filename                                     store conversation messages
+/load filename                                     load conversation messages
+/compose                                           compose content using an EDITOR
+/input [pattern]                                   select and read content from a file (default: **/*)
+/context [pattern...]                              collect context with glob patterns
+/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|on|off]                     list enabled, enable/disable tools, support on/off
+/vim                                               insert the last message into a vim server
+/quit                                              to quit
+/help [me]                                         to view this help (me = interactive ai help)
 ```
 
 ### Using `ollama_chat_send` to send input to a running `ollama_chat`

data/lib/ollama_chat/chat.rb

@@ -47,15 +47,18 @@ class OllamaChat::Chat
   include OllamaChat::MessageOutput
   include OllamaChat::Clipboard
   include OllamaChat::MessageFormat
+  include OllamaChat::Pager
   include OllamaChat::History
   include OllamaChat::ServerSocket
   include OllamaChat::KramdownANSI
+  include OllamaChat::FileEditing
   include OllamaChat::Conversation
   include OllamaChat::InputContent
   include OllamaChat::MessageEditing
   include OllamaChat::LocationHandling
   include OllamaChat::ToolCalling
   include OllamaChat::ConfigHandling
+  include OllamaChat::PersonaeManagement
 
   # Initializes a new OllamaChat::Chat instance with the given command-line
   # arguments.
@@ -85,7 +88,7 @@ class OllamaChat::Chat
   # @raise [RuntimeError] If the Ollama API version is less than 0.9.0, indicating
   #   incompatibility with required API features
   def initialize(argv: ARGV.dup)
-    @opts               = go 'f:u:m:s:c:C:D:MESVh', argv
+    @opts               = go 'f:u:m:s:p:c:C:D:MESVh', argv
     @opts[?h] and exit usage
     @opts[?V] and exit version
     @messages           = OllamaChat::MessageList.new(self)
@@ -113,6 +116,7 @@ class OllamaChat::Chat
     @enabled_tools        = default_enabled_tools
     @tool_call_results    = {}
     init_chat_history
+    setup_personae_directory
     @opts[?S] and init_server_socket
   rescue ComplexConfig::AttributeMissing, ComplexConfig::ConfigurationSyntaxError => e
     fix_config(e)
@@ -155,38 +159,15 @@ class OllamaChat::Chat
   #   messages associated with this instance
   attr_reader :messages
 
-  # 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.
+  # The start method initializes the chat session by displaying information,
+  # then prompts the user for input to begin interacting with the chat.
   def start
     info
-    if messages.size > 1
-      messages.list_conversation(2)
-    end
     STDOUT.puts "\nType /help to display the chat help."
 
     interact_with_user
   end
 
-  # The vim method creates and returns a new Vim instance for interacting with
-  # a Vim server.
-  #
-  # This method initializes a Vim client that can be used to insert text into
-  # Vim buffers or open files in a running Vim server. It derives the server
-  # name from the provided argument or uses a default server name based on the
-  # current working directory.
-  #
-  # @param server_name [ String, nil ] the name of the Vim server to connect to
-  #   If nil or empty, a default server name is derived from the current
-  #   working directory
-  #
-  # @return [ OllamaChat::Vim ] a new Vim instance configured with the
-  #   specified server name
-  def vim(server_name = nil)
-    clientserver = config.vim?&.clientserver
-    OllamaChat::Vim.new(server_name, clientserver:)
-  end
-
   # The debug method accesses the debug configuration setting.
   #
   # @return [TrueClass, FalseClass] the current debug mode status
@@ -222,6 +203,7 @@ class OllamaChat::Chat
       copy_to_clipboard
       :next
     when %r(^/paste$)
+      @parse_content = false
       paste_from_clipboard
     when %r(^/markdown$)
       markdown.toggle
@@ -385,18 +367,73 @@ class OllamaChat::Chat
         tools_support.set(false, show: true)
       end
       :next
-    when %r(^/config(?:\s+(edit))?$)
+    when %r(^/config(?:\s+(edit|reload))?$)
       case $1
       when 'edit'
         edit_config
+      when 'reload'
+        reload_config
       else
         display_config
       end
       :next
+    when %r(^/persona(?:\s+(add|delete|edit|file|info|list|load|play))?$)
+      @parse_content = false
+      case $1
+      when 'add'
+        if result = add_persona
+          result
+        else
+          :next
+        end
+      when 'delete'
+        if result = delete_persona
+          result
+        else
+          :next
+        end
+      when 'edit'
+        if result = edit_persona
+          result
+        else
+          :next
+        end
+      when 'file'
+        if pathname = choose_filename('**/*.md')
+          pathname.read
+        else
+          :next
+        end
+      when 'info'
+        info_persona
+        :next
+      when 'list'
+        list_personae
+        :next
+      when 'load'
+        if result = load_personae
+          result
+        else
+          :next
+        end
+      when 'play'
+        if pathname = choose_filename('**/*.md')
+          play_persona_file(pathname)
+        else
+          :next
+        end
+      else
+        if result = play_persona.full?
+          result
+        else
+          :next
+        end
+      end
     when %r(^/quit$), nil
       STDOUT.puts "Goodbye."
       :return
     when %r(^/help me$)
+      @parse_content = false
       config.prompts.help % { commands: display_chat_help_message }
     when %r(^/)
       display_chat_help
@@ -563,46 +600,51 @@ class OllamaChat::Chat
   # interaction.
   def interact_with_user
     loop do
-      @parse_content = true
-      type           = :terminal_input
-      input_prompt   = bold { color(172) { message_type(@images) + " user" } } + bold { "> " }
-      begin
-        if content = handle_tool_call_results?
-          @parse_content = false
-        else
-          content = enable_command_completion do
-            if prefill_prompt = @prefill_prompt.full?
-              Reline.pre_input_hook = -> {
-                Reline.insert_text prefill_prompt.gsub(/\n*\z/, '')
-                @prefill_prompt = nil
-              }
-            else
-              Reline.pre_input_hook = nil
+      if persona_result = setup_persona_from_opts
+        @parse_content = false
+        content = persona_result
+      else
+        @parse_content = true
+        type           = :terminal_input
+        input_prompt   = bold { color(172) { message_type(@images) + " user" } } + bold { "> " }
+        begin
+          if content = handle_tool_call_results?
+            @parse_content = false
+          else
+            content = enable_command_completion do
+              if prefill_prompt = @prefill_prompt.full?
+                Reline.pre_input_hook = -> {
+                  Reline.insert_text prefill_prompt.gsub(/\n*\z/, '')
+                  @prefill_prompt = nil
+                }
+              else
+                Reline.pre_input_hook = nil
+              end
+              Reline.readline(input_prompt, true)&.chomp
             end
-            Reline.readline(input_prompt, true)&.chomp
+          end
+        rescue Interrupt
+          if message = server_socket_message
+            type           = message.type.full?(:to_sym) || :socket_input
+            content        = message.content
+            @parse_content = message.parse
+            STDOUT.puts color(112) { "Received a server socket message. Processing now…" }
+          else
+            raise
           end
         end
-      rescue Interrupt
-        if message = server_socket_message
-          type           = message.type.full?(:to_sym) || :socket_input
-          content        = message.content
-          @parse_content = message.parse
-          STDOUT.puts color(112) { "Received a server socket message. Processing now…" }
-        else
-          raise
-        end
-      end
 
-      if type == :terminal_input
-        case next_action = handle_input(content)
-        when :next
-          next
-        when :redo
-          redo
-        when :return
-          return
-        when String
-          content = next_action
+        if type == :terminal_input
+          case next_action = handle_input(content)
+          when :next
+            next
+          when :redo
+            redo
+          when :return
+            return
+          when String
+            content = next_action
+          end
         end
       end
 

data/lib/ollama_chat/clipboard.rb

@@ -11,25 +11,29 @@ module OllamaChat::Clipboard
   # its content to the system clipboard using the command specified in
   # the configuration (config.copy).
   #
-  # @param content [Boolean] If true, copies the content of the message;
+  # @param content [true, false] If true, copies the content of the message;
   #   if false, copies the entire message object (default: false)
   #
   # @raise [OllamaChat::OllamaChatError] if the clipboard command specified
   #   in config.copy is not found in the system's PATH
   # @raise [OllamaChat::OllamaChatError] if no assistant message is available
   #   to copy to the clipboard
-  def perform_copy_to_clipboard(content: false)
-    if message = @messages.find_last(content:) { _1.role == 'assistant' }
+  def perform_copy_to_clipboard(text: nil, content: false)
+    text ||= last_message_content(content:)
+    if text
       copy = `which #{config.copy}`.chomp
       if copy.present?
         IO.popen(copy, 'w') do |clipboard|
-          clipboard.write(message.content)
+          clipboard.write(text)
         end
+        true
       else
-        raise OllamaChat::OllamaChatError, "#{config.copy.inspect} command not found in system's path!"
+        raise OllamaChat::OllamaChatError,
+          "#{config.copy.inspect} command not found in system's path!"
       end
     else
-      raise OllamaChat::OllamaChatError, "No response available to copy to the system clipboard."
+      raise OllamaChat::OllamaChatError,
+        "No text available to copy to the system clipboard."
     end
   end
 
@@ -67,6 +71,26 @@ module OllamaChat::Clipboard
 
   private
 
+  # Returns the content of the last assistant message.
+  #
+  # This private helper method finds the most recent message from the assistant
+  # in the messages array and returns its content. It is used by
+  # `perform_copy_to_clipboard` when no custom text is provided.
+  #
+  # @param content [true, false] If true, returns the content of the message;
+  #   if false, returns nil if no assistant message is found (default: false)
+  #
+  # @return [String, nil] The content of the last assistant message, or nil if
+  #   no assistant message is found
+  #
+  # @example
+  #   # Assuming @messages contains assistant messages
+  #   last_message_content
+  #   # => "This is the last assistant response"
+  def last_message_content(content: false)
+    @messages.find_last(content:) { _1.role == 'assistant' }&.content
+  end
+
   # Copies the last assistant message to the system clipboard.
   #
   # This method is the interface for copying assistant messages to the
@@ -74,8 +98,10 @@ module OllamaChat::Clipboard
   # handles any OllamaChat::OllamaChatError exceptions by printing the error
   # message to standard error and does not re-raise the exception.
   def copy_to_clipboard
+
     perform_copy_to_clipboard
     STDOUT.puts "The last response has been successfully copied to the system clipboard."
+    true
   rescue OllamaChat::OllamaChatError => e
     STDERR.puts e.message
   end

data/lib/ollama_chat/config_handling.rb

@@ -83,18 +83,30 @@ module OllamaChat::ConfigHandling
   # 3. If editing was successful, prompts the user to restart
   #    `ollama_chat` if desired.
   def edit_config
-    unless editor = OC::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 result = edit_file(@ollama_chat_config.filename)
       if ask?(prompt: "Do you want to restart #{progname}? (y/n) ") =~ /\Ay/i
         save_conversation(OC::XDG_CACHE_HOME + 'backup.json')
+        save_history
         exec($0, *ARGV)
+      else
+        STDOUT.puts "Skipped reloading the config."
       end
     else
       STDERR.puts "Editor returned a non-zero status!"
     end
   end
+
+  # Reloads the current configuration by restarting the process.
+  #
+  # @example Restarting the app after confirmation
+  #   config.reload_config  # => restarts if user answers "y"
+  def reload_config
+    if ask?(prompt: "Do you want to restart #{progname}? (y/n) ") =~ /\Ay/i
+      save_conversation(OC::XDG_CACHE_HOME + 'backup.json')
+      save_history
+      exec($0, *ARGV)
+    else
+      STDOUT.puts "Skipped reloading the config."
+    end
+  end
 end

data/lib/ollama_chat/file_editing.rb

@@ -0,0 +1,61 @@
+# Module for editing files using the configured editor
+module OllamaChat::FileEditing
+  # Opens a file in the configured editor for editing.
+  #
+  # @param filename [String, Pathname] Path to the file to edit
+  # @return [true, false, nil] Exit status if successful, nil if editor not
+  #   configured
+  def edit_file(filename)
+    unless editor = OC::EDITOR?
+      STDERR.puts "Need the environment variable var EDITOR defined to use an editor"
+      return
+    end
+    system Shellwords.join([ editor, filename ])
+  end
+
+  # The vim method creates and returns a new Vim instance for interacting with
+  # a Vim server.
+  #
+  # This method initializes a Vim client that can be used to insert text into
+  # Vim buffers or open files in a running Vim server. It derives the server
+  # name from the provided argument or uses a default server name based on the
+  # current working directory.
+  #
+  # @param server_name [ String, nil ] the name of the Vim server to connect to
+  #   If nil or empty, a default server name is derived from the current
+  #   working directory
+  #
+  # @return [ OllamaChat::Vim ] a new Vim instance configured with the
+  #   specified server name
+  def vim(server_name = nil)
+    clientserver = config.vim?&.clientserver
+    OllamaChat::Vim.new(server_name, clientserver:)
+  end
+
+  # Inserts provided **text** into Vim using the configured editor client.
+  #
+  # @param [String, nil] text    – Text to insert; if `nil`, defaults to the
+  #   last response.
+  #
+  # @param [true, false]     content  – Treat input as content‑mode (`true`) or
+  #   plain text (`false`).
+  #
+  # @return [true, false] true if insertion succeeded; otherwise raises
+  #   OllamaChat::OllamaChatError.
+  #
+  # @raise [OllamaChat::OllamaChatError]
+  #   * Raised when no text is available and insertion cannot proceed.
+  #   * Also raised if the underlying Vim client reports a failure.
+  def perform_insert(text: nil, content: false)
+    text ||= last_message_content(content:)
+    if text
+      unless vim.insert(text)
+        raise OllamaChat::OllamaChatError, "Inserting text into editor failed!"
+      end
+      true
+    else
+      raise OllamaChat::OllamaChatError,
+        "No text available to copy to the system clipboard."
+    end
+  end
+end