ollama_chat 0.0.14 → 0.0.16

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9c7760df8454e8d8b522c8d37448ca4b872f27b44ef484e842f8c35a96fec50b
-  data.tar.gz: abbc3fe1a7d53ded7ed878dfbbd6a584a99e8b687e16f1386e6e4e058cda1f46
+  metadata.gz: 1cddb901277529210fcc39b4c086217515e98aa75c1569044a47e1cf7149acbe
+  data.tar.gz: f7a40ef2722c750bb71ff2b42d3c2cf0c04816bfd11753e22fa7a09f2d13c9f8
 SHA512:
-  metadata.gz: 28056039df80caf298aafb8445c0d3b6662e9fee8d28b6756fb13b844c006092754e68e33fc8947f8326ee11e4248206558c522b170463e4f58da88792cadfde
-  data.tar.gz: 795e2bf8fefb75b53c1108637affadb796c24b86c1cec853b16f62053d80c0319a239a8053676434a6c22197411557d91ad995b025e00bb735b83dccbb0d6ef2
+  metadata.gz: 7cd5baab3a073464bbf28d3cd9e6ebbcfdcd938c55b1a17ca5d6f2a6b3949d1a2d0a138d9aacf192f6d2f2de1751b8d3e10368c2c15eba5b955ee5454fbd6ff1
+  data.tar.gz: 3ed3442aa4962666de9a56ce5a10e816c9b6f21d85e1281c65962d70c41c60e0c9c8f47fc4e380703799d67e006b2aef36fbe4e2208632e7c3d02c4cc3120c1a

data/CHANGES.md

@@ -1,5 +1,38 @@
 # Changes
 
+## 2025-07-10 v0.0.16
+
+- **New Features**
+  - Added `-f CONFIG` option to `ollama_chat_send` for specifying configuration files.
+  - Introduced `server_socket_runtime_dir` setting in the default config, and
+  make it default to the current directory, allowing for a per directory chat
+  to receive server socket messages.
+
+- **Enhancements**
+  - Improved logging with debug output for received server socket messages.
+  - Refactored server socket handling:
+    - Created `create_socket_server` method for UnixSocks setup with configurable runtime directories.
+    - Updated `send_to_server_socket` and `init_server_socket` methods to use the new helper.
+  - Changed evaluation rate metrics from 'c/s' to 't/s' for better clarity.
+
+- **Documentation**
+  - Added additional documentation for key classes and methods in `FollowChat`.
+
+## 2025-07-02 v0.0.15
+
+- **Enhanced `ollama_chat_send` and Unix Domain Socket Support:**
+  - Added support for advanced parameters:
+    - `-t`: Sends input as terminal commands.
+    - `-r`: Enables two-way communication by waiting for and returning the server's response.
+    - `-h` or `--help`: Displays usage information and available options.
+  - Improved socket management using the `unix_socks` gem.
+  - Enhanced message processing logic to handle different types of messages
+    (`:socket_input`, `:terminal_input`, `:socket_input_with_response`).
+- **Selector Support for Model and System Prompt Selection:**
+  - Introduced `?selector` syntax to filter models and prompts.
+  - Updated documentation to reflect this new feature.
+  - Added a chooser dialog when multiple options match the selector.
+
 ## 2025-06-07 v0.0.14
 
 * **Message List Improvements**:

data/README.md

@@ -24,8 +24,8 @@ Usage: ollama_chat [OPTIONS]
 
   -f CONFIG      config file to read
   -u URL         the ollama base url, OLLAMA_URL
-  -m MODEL       the ollama model to chat with, OLLAMA_CHAT_MODEL
-  -s SYSTEM      the system prompt to use as a file, OLLAMA_CHAT_SYSTEM
+  -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
   -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)
@@ -34,6 +34,9 @@ Usage: ollama_chat [OPTIONS]
   -S             open a socket to receive input from ollama_chat_send
   -V             display the current version number and quit
   -h             this help
+
+  Use `?selector` with `-m` or `-s` to filter options. Multiple matches
+  will open a chooser dialog.
 ```
 
 The base URL can be either set by the environment variable `OLLAMA_URL` or it
@@ -167,6 +170,32 @@ function! OllamaChatSend(input)
 endfunction
 ```
 
+#### Advanced Parameters for `ollama_chat_send`
+
+The `ollama_chat_send` command now supports additional parameters to enhance functionality:
+
+- **Terminal Input (`-t`)**: Sends input as terminal commands, enabling special commands like `/import`.
+
+  ```bash
+  $ echo "/import https://example.com/some-content" | ollama_chat_send -t
+  ```
+
+- **Wait for Response (`-r`)**: Enables two-way communication by waiting for and returning the server's response.
+
+  ```bash
+  $ response=$(echo "Tell me a joke." | ollama_chat_send -r)
+  $ echo "$response"
+  ```
+
+- **Help (`-h` or `--help`)**: Displays usage information and available options.
+
+  ```bash
+  $ ollama_chat_send -h
+  ```
+
+These parameters provide greater flexibility in how you interact with
+`ollama_chat`, whether from the command line or integrated tools like `vim`.
+
 ## Download
 
 The homepage of this app is located at

data/Rakefile

@@ -32,6 +32,7 @@ GemHadar do
   dependency             'excon',                 '~> 1.0'
   dependency             'ollama-ruby',           '~> 1.2'
   dependency             'documentrix',           '~> 0.0', '>= 0.0.2'
+  dependency             'unix_socks'
   dependency             'rss',                   '~> 0.3'
   dependency             'term-ansicolor',        '~> 1.11'
   dependency             'redis',                 '~> 5.0'

data/VERSION

@@ -1 +1 @@
-0.0.14
+0.0.16

data/bin/ollama_chat_send

@@ -1,10 +1,37 @@
 #!/usr/bin/env ruby
 
 require 'ollama_chat'
+require 'tins/go'
+include Tins::GO
 
+
+opts = go 'f:rth', ARGV
+
+def usage(rc = 0)
+  puts <<~EOT
+    Usage: #{File.basename($0)} [OPTIONS]
+
+    Options:
+      -r         Wait for the response from Ollama Chat and output it
+      -t         Send input as terminal input including commands, e. g. /import
+      -f CONFIG  file to read
+      -h         Show this help message
+
+    Send data to a running Ollame Chat client via standard input.
+  EOT
+  exit rc
+end
+
+config = OllamaChat::OllamaChatConfig.new(opts[?f]).config
+opts[?h] and usage
 begin
-  type = (ARGV.shift || 'socket_input').to_sym
-  OllamaChat::ServerSocket.send_to_server_socket(STDIN.read, type:)
+  type = if opts[?t]
+           :terminal_input
+         else
+           opts[?r] ? :socket_input_with_response : :socket_input
+         end
+  response = OllamaChat::ServerSocket.send_to_server_socket(STDIN.read, type:, config:)
+  type == :socket_input_with_response and puts response.content
 rescue => e
   warn "Caught #{e.class}: #{e}"
   exit 1

data/lib/ollama_chat/chat.rb

@@ -1,6 +1,7 @@
 require 'tins'
 require 'tins/secure_write'
 require 'tins/xt/string_version'
+require 'tins/xt/full'
 require 'json'
 require 'term/ansicolor'
 require 'reline'
@@ -352,15 +353,15 @@ class OllamaChat::Chat
         content = Reline.readline(input_prompt, true)&.chomp
       rescue Interrupt
         if message = server_socket_message
-          self.server_socket_message = nil
-          type    = message.fetch('type', 'socket_input').to_sym
-          content = message['content']
+          type    = message.type.full?(:to_sym) || :socket_input
+          content = message.content
+          STDOUT.puts color(112) { "Received a server socket message. Processing now…" }
         else
           raise
         end
       end
 
-      unless type == :socket_input
+      if type == :terminal_input
         case next_action = handle_input(content)
         when :next
           next
@@ -421,10 +422,22 @@ class OllamaChat::Chat
         }.uniq.map { |l, t| hyperlink(l, t) }.join(' ')
         config.debug and jj messages.to_ary
       end
+
+      case type
+      when :socket_input
+        server_socket_message&.disconnect
+      when :socket_input_with_response
+        if message = handler.messages.last
+          server_socket_message.respond({ role: message.role, content: message.content })
+        end
+        server_socket_message&.disconnect
+      end
     rescue Ollama::Errors::TimeoutError
       STDOUT.puts "#{bold('Error')}: Currently lost connection to ollama server and cannot send command."
     rescue Interrupt
       STDOUT.puts "Type /quit to quit."
+    ensure
+        self.server_socket_message = nil
     end
     0
   rescue ComplexConfig::AttributeMissing, ComplexConfig::ConfigurationSyntaxError => e

data/lib/ollama_chat/dialog.rb

@@ -1,6 +1,11 @@
 module OllamaChat::Dialog
   def choose_model(cli_model, current_model)
+    selector = if cli_model =~ /\A\?+(.*)\z/
+                 cli_model = ''
+                 Regexp.new($1)
+               end
     models = ollama.tags.models.map(&:name).sort
+    selector and models = models.grep(selector)
     model = if cli_model == ''
               OllamaChat::Utils::Chooser.choose(models) || current_model
             else
@@ -58,8 +63,12 @@ module OllamaChat::Dialog
   end
 
   def change_system_prompt(default, system: nil)
-    selector = Regexp.new(system.to_s[1..-1].to_s)
-    prompts  = config.system_prompts.attribute_names.compact.grep(selector)
+    selector = if system =~ /\A\?(.+)\z/
+                 Regexp.new($1)
+               else
+                 Regexp.new(system.to_s)
+               end
+    prompts = config.system_prompts.attribute_names.compact.grep(selector)
     if prompts.size == 1
       system = config.system_prompts.send(prompts.first)
     else

data/lib/ollama_chat/follow_chat.rb

@@ -4,6 +4,15 @@ class OllamaChat::FollowChat
   include Term::ANSIColor
   include OllamaChat::MessageFormat
 
+
+  # Initializes a new instance of OllamaChat::FollowChat.
+  #
+  # @param [OllamaChat::Chat] chat The chat object, which represents the conversation context.
+  # @param [#to_a] messages A collection of message objects, representing the conversation history.
+  # @param [String] voice (optional) to speek with if any.
+  # @param [IO] output (optional) The output stream where terminal output should be printed. Defaults to STDOUT.
+  #
+  # @return [OllamaChat::FollowChat] A new instance of OllamaChat::FollowChat.
   def initialize(chat:, messages:, voice: nil, output: STDOUT)
     super(output:)
     @chat        = chat
@@ -13,6 +22,28 @@ class OllamaChat::FollowChat
     @user        = nil
   end
 
+  # Returns the conversation history (an array of message objects).
+  #
+  # @return [OllamaChat::MessageList<Ollama::Message>] The array of messages in the conversation.
+  attr_reader :messages
+
+  # Invokes the chat flow based on the provided Ollama server response.
+  #
+  # The response is expected to be a parsed JSON object containing information
+  # about the user input and the assistant's response.
+  #
+  # If the response indicates an assistant message, this method:
+  #   1. Ensures that an assistant response exists in the message history (if not already present).
+  #   2. Updates the last message with the new content and thinking (if applicable).
+  #   3. Displays the formatted terminal output for the user.
+  #   4. Outputs the voice response (if configured).
+  #
+  # Regardless of whether an assistant message is present, this method also
+  # outputs evaluation statistics (if applicable).
+  #
+  # @param [Ollama::Response] response The parsed JSON response from the Ollama server.
+  #
+  # @return [OllamaChat::FollowChat] The current instance for method chaining.
   def call(response)
     debug_output(response)
 
@@ -71,10 +102,10 @@ class OllamaChat::FollowChat
     stats_text = {
       eval_duration:        Tins::Duration.new(eval_duration),
       eval_count:           response.eval_count.to_i,
-      eval_rate:            bold { "%.2f c/s" % (response.eval_count.to_i / eval_duration) } + color(111),
+      eval_rate:            bold { "%.2f t/s" % (response.eval_count.to_i / eval_duration) } + color(111),
       prompt_eval_duration: Tins::Duration.new(prompt_eval_duration),
       prompt_eval_count:    response.prompt_eval_count.to_i,
-      prompt_eval_rate:     bold { "%.2f c/s" % (response.prompt_eval_count.to_i / prompt_eval_duration) } + color(111),
+      prompt_eval_rate:     bold { "%.2f t/s" % (response.prompt_eval_count.to_i / prompt_eval_duration) } + color(111),
       total_duration:       Tins::Duration.new(response.total_duration / 1e9),
       load_duration:        Tins::Duration.new(response.load_duration / 1e9),
     }.map { _1 * ?= } * ' '

data/lib/ollama_chat/information.rb

@@ -29,7 +29,7 @@ module OllamaChat::Information
 
   def info
     STDOUT.puts "Running ollama_chat version: #{bold(OllamaChat::VERSION)}"
-    STDOUT.puts "Connected to ollama server version: #{bold(server_version)}"
+    STDOUT.puts "Connected to ollama server version: #{bold(server_version)} on: #{bold(server_url)}"
     STDOUT.puts "Current conversation model is #{bold{@model}}."
     if @model_options.present?
       STDOUT.puts "  Options: #{JSON.pretty_generate(@model_options).gsub(/(?<!\A)^/, '  ')}"
@@ -98,8 +98,8 @@ module OllamaChat::Information
 
         -f CONFIG      config file to read
         -u URL         the ollama base url, OLLAMA_URL
-        -m MODEL       the ollama model to chat with, OLLAMA_CHAT_MODEL
-        -s SYSTEM      the system prompt to use as a file, OLLAMA_CHAT_SYSTEM
+        -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
         -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)
@@ -109,6 +109,8 @@ module OllamaChat::Information
         -V             display the current version number and quit
         -h             this help
 
+        Use `?selector` with `-m` or `-s` to filter options. Multiple matches
+        will open a chooser dialog.
     EOT
     0
   end
@@ -121,4 +123,8 @@ module OllamaChat::Information
   def server_version
     @server_version ||= ollama.version.version
   end
+
+  def server_url
+    @server_url ||= ollama.base_url
+  end
 end

data/lib/ollama_chat/message_list.rb

@@ -128,22 +128,42 @@ class OllamaChat::MessageList
     self
   end
 
-  # The drop method removes the last n exchanges from the message list and returns the number of removed exchanges.
+  # Removes the last `n` exchanges from the message list. An exchange consists
+  # of a user and an assistant message. If only a single user message is
+  # present at the end, it will be removed first before proceeding with
+  # complete exchanges.
   #
-  # @param n [ Integer ] the number of exchanges to remove
+  # @param n [Integer] The number of exchanges to remove.
+  # @return [Integer] The actual number of complete exchanges removed.
+  #                   This may be less than `n` if there are not enough messages.
   #
-  # @return [ Integer ] the number of removed exchanges, or 0 if there are no more exchanges to pop
+  # @note
+  #   - System messages are preserved and not considered part of an exchange.
+  #   - If only one incomplete exchange (a single user message) exists, it will
+  #     be dropped first before removing complete exchanges.
   def drop(n)
-    if @messages.reject { _1.role == 'system' }.size > 1
-      n = n.to_i.clamp(1, Float::INFINITY)
-      r = @messages.pop(2 * n)
-      m = r.size / 2
-      STDOUT.puts "Dropped the last #{m} exchanges."
-      m
-    else
-      STDOUT.puts "No more exchanges you can drop."
-      0
+    n = n.to_i.clamp(1, Float::INFINITY)
+    non_system_messages = @messages.reject { _1.role == 'system' }
+    if non_system_messages&.last&.role == 'user'
+      @messages.pop
+      n -= 1
+    end
+    if n == 0
+      STDOUT.puts "Dropped the last exchange."
+      return 1
+    end
+    if non_system_messages.empty?
+      STDOUT.puts "No more exchanges can be dropped."
+      return 0
+    end
+    m = 0
+    while @messages.size > 1 && n > 0
+      @messages.pop(2)
+      m += 1
+      n -= 1
     end
+    STDOUT.puts "Dropped the last #{m} exchanges."
+    m
   end
 
   # Sets the system prompt for the chat session.

data/lib/ollama_chat/model_handling.rb

@@ -1,15 +1,44 @@
 module OllamaChat::ModelHandling
+
+  # The model_present? method checks if the specified Ollama model is available.
+  #
+  # @param model [ String ] the name of the Ollama model
+  #
+  # @return [ String, FalseClass ] the system prompt if the model is present,
+  # false otherwise
   def model_present?(model)
     ollama.show(model:) { return _1.system.to_s }
   rescue Ollama::Errors::NotFoundError
     false
   end
 
+  # The pull_model_from_remote method attempts to retrieve a model from the
+  # remote server if it is not found locally.
+  #
+  # @param model [ String ] the name of the model to be pulled
+  #
+  # @return [ nil ]
   def pull_model_from_remote(model)
     STDOUT.puts "Model #{bold{model}} not found locally, attempting to pull it from remote now…"
     ollama.pull(model:)
   end
 
+  # The pull_model_unless_present method checks if the specified model is
+  # present on the system.
+  #
+  # If the model is already present, it returns the system prompt if it is
+  # present.
+  #
+  # Otherwise, it attempts to pull the model from the remote server using the
+  # pull_model_from_remote method. If the model is still not found after
+  # pulling, it exits the program with a message indicating that the model was
+  # not found remotely.
+  #
+  # @param model [ String ] The name of the model to check for presence.
+  # @param options [ Hash ] Options for the pull_model_from_remote method.
+  #
+  # @return [ String, FalseClass ] the system prompt if the model and it are
+  # present, false otherwise.
   def pull_model_unless_present(model, options)
     if system = model_present?(model)
       return system.full?

data/lib/ollama_chat/ollama_chat_config/default_config.yml

@@ -61,6 +61,7 @@ redis:
     url: <%= ENV.fetch('REDIS_EXPIRING_URL', 'null') %>
     ex: 86400
 chat_history_filename: <%= ENV.fetch('OLLAMA_CHAT_HISTORY', '~/.ollama_chat_history') %>
+server_socket_runtime_dir: .
 debug: <%= ENV['OLLAMA_CHAT_DEBUG'].to_i == 1 ? true : false %>
 request_headers:
   Accept: 'text/*,application/*,image/*'

data/lib/ollama_chat/server_socket.rb

@@ -1,65 +1,50 @@
 module OllamaChat::ServerSocket
   class << self
-    # Returns the path to the XDG runtime directory, or a default path if not set.
-    # @return [String] the expanded path to the XDG runtime directory
-    def runtime_dir
-      File.expand_path(ENV.fetch('XDG_RUNTIME_DIR',  '~/.local/run'))
-    end
-
-    # Constructs the full path to the server socket file.
-    # @return [String] the full path to the Unix socket
-    def server_socket_path
-      File.join(runtime_dir, 'ollama_chat.sock')
-    end
+    # The send_to_server_socket method sends content to the server socket and returns
+    # the response if type is :socket_input_with_response, otherwise it returns nil.
 
-    # Sends a message to the server socket.
+    # @param content [ String ] the message to be sent to the server
+    # @param type [ Symbol ] the type of message being sent (default: :socket_input)
     #
-    # @param content [String] the content to send
-    # @param type [Symbol] the type of message (default: :socket_input)
-    # @raise [Errno::ENOENT] if the socket file does not exist
-    # @raise [Errno::ECONNREFUSED] if the socket is not listening (server no running)
-    def send_to_server_socket(content, type: :socket_input)
-      FileUtils.mkdir_p runtime_dir
+    # @return [ String, NilClass ] the response from the server if type is
+    #   :socket_input_with_response, otherwise nil.
+    def send_to_server_socket(content, config:, type: :socket_input)
+      server  = create_socket_server(config:)
       message = { content:, type: }
-      socket = UNIXSocket.new(server_socket_path)
-      socket.puts JSON(message)
-      socket.close
+      if type.to_sym == :socket_input_with_response
+         return server.transmit_with_response(message)
+      else
+         server.transmit(message)
+         nil
+      end
+    end
+
+    def create_socket_server(config:)
+      if runtime_dir = config.server_socket_runtime_dir
+        UnixSocks::Server.new(socket_name: 'ollama_chat.sock', runtime_dir:)
+      else
+        UnixSocks::Server.new(socket_name: 'ollama_chat.sock')
+      end
     end
   end
 
-  # Accessor for the server socket message.
-  # Holds the last message received from the Unix socket.
-  # @return [String, nil] the message content, or nil if not set
-  # @see OllamaChat::ServerSocket#init_server_socket
-  # @see OllamaChat::ServerSocket#send_to_server_socket
   attr_accessor :server_socket_message
 
-  # Initializes a Unix domain socket server for OllamaChat.
+  # Initializes the server socket to receive messages from the Ollama Chat
+  # Client.
   #
-  # Creates the necessary runtime directory, checks for existing socket file,
-  # and starts a server loop in a new thread. Listens for incoming connections,
-  # reads JSON data, and terminates the server upon receiving a message.
+  # This method sets up a Unix domain socket server that listens for incoming
+  # messages in the background. When a message is received, it updates the
+  # instance variable `server_socket_message` and sends an interrupt signal
+  # to the current process in order to handle the message.
   #
-  # Raises Errno::EEXIST if the socket path already exists.
+  # @return [ nil ] This method does not return any value, it only sets up the
+  # server socket and kills the process when a message is received.
   def init_server_socket
-    FileUtils.mkdir_p OllamaChat::ServerSocket.runtime_dir
-    if File.exist?(OllamaChat::ServerSocket.server_socket_path)
-      raise Errno::EEXIST, "Path already exists #{OllamaChat::ServerSocket.server_socket_path.inspect}"
-    end
-    Thread.new do
-      Socket.unix_server_loop(OllamaChat::ServerSocket.server_socket_path) do |sock, client_addrinfo|
-        begin
-          data = sock.readline.chomp
-          self.server_socket_message = JSON.load(data)
-          Process.kill :INT, $$
-        rescue JSON::ParserError
-        ensure
-          sock.close
-        end
-      end
-    rescue Errno::ENOENT
-    ensure
-      FileUtils.rm_f OllamaChat::ServerSocket.server_socket_path
+    server = OllamaChat::ServerSocket.create_socket_server(config:)
+    server.receive_in_background do |message|
+      self.server_socket_message = message
+      Process.kill :INT, $$
     end
   end
 end

data/lib/ollama_chat/version.rb

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

data/lib/ollama_chat.rb

@@ -3,6 +3,7 @@ end
 
 require 'ollama'
 require 'documentrix'
+require 'unix_socks'
 require 'ollama_chat/version'
 require 'ollama_chat/utils'
 require 'ollama_chat/message_format'

data/ollama_chat.gemspec

@@ -1,9 +1,9 @@
 # -*- encoding: utf-8 -*-
-# stub: ollama_chat 0.0.14 ruby lib
+# stub: ollama_chat 0.0.16 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "ollama_chat".freeze
-  s.version = "0.0.14".freeze
+  s.version = "0.0.16".freeze
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0".freeze) if s.respond_to? :required_rubygems_version=
   s.require_paths = ["lib".freeze]
@@ -18,7 +18,7 @@ Gem::Specification.new do |s|
   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]
   s.required_ruby_version = Gem::Requirement.new("~> 3.1".freeze)
-  s.rubygems_version = "3.6.7".freeze
+  s.rubygems_version = "3.6.9".freeze
   s.summary = "A command-line interface (CLI) for interacting with an Ollama AI model.".freeze
   s.test_files = ["spec/assets/example.rb".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/message_list_spec.rb".freeze, "spec/ollama_chat/model_handling_spec.rb".freeze, "spec/ollama_chat/parsing_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]
 
@@ -34,6 +34,7 @@ Gem::Specification.new do |s|
   s.add_runtime_dependency(%q<excon>.freeze, ["~> 1.0".freeze])
   s.add_runtime_dependency(%q<ollama-ruby>.freeze, ["~> 1.2".freeze])
   s.add_runtime_dependency(%q<documentrix>.freeze, ["~> 0.0".freeze, ">= 0.0.2".freeze])
+  s.add_runtime_dependency(%q<unix_socks>.freeze, [">= 0".freeze])
   s.add_runtime_dependency(%q<rss>.freeze, ["~> 0.3".freeze])
   s.add_runtime_dependency(%q<term-ansicolor>.freeze, ["~> 1.11".freeze])
   s.add_runtime_dependency(%q<redis>.freeze, ["~> 5.0".freeze])

data/spec/ollama_chat/chat_spec.rb

@@ -5,6 +5,10 @@ RSpec.describe OllamaChat::Chat do
     %w[ -C test ]
   end
 
+  before do
+    ENV['OLLAMA_CHAT_MODEL'] = 'llama3.1'
+  end
+
   let :chat do
     OllamaChat::Chat.new(argv: argv).expose
   end

data/spec/ollama_chat/information_spec.rb

@@ -39,7 +39,16 @@ RSpec.describe OllamaChat::Information do
     expect(chat.usage).to eq 0
   end
 
-  it 'can show version' do
+  it 'can show  version' do
+    expect(STDOUT).to receive(:puts).with(/^ollama_chat \d+\.\d+\.\d+$/)
     expect(chat.version).to eq 0
   end
+
+  it 'can show server version' do
+    expect(chat.server_version).to eq '6.6.6'
+  end
+
+  it 'can show server URL' do
+    expect(chat.server_url).to be_a URI::HTTP
+  end
 end

data/spec/ollama_chat/message_list_spec.rb

@@ -128,11 +128,26 @@ RSpec.describe OllamaChat::MessageList do
     expect(list.size).to eq 1
     list <<  Ollama::Message.new(role: 'user', content: 'world')
     expect(list.size).to eq 2
-    expect(list.drop(1)).to eq 0
     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
+    expect(list.drop(1)).to eq 0
+    expect(list.size).to eq 1
+    expect(list.drop(1)).to eq 0
+    expect(list.size).to eq 1
+  end
+
+  it 'drops the last user message when there is no assistant response' do
+    expect(list.size).to eq 1
+    list << Ollama::Message.new(role: 'user', content: 'hello')
+    list << Ollama::Message.new(role: 'assistant', content: 'hi')
+    list << Ollama::Message.new(role: 'user', content: 'world')
+    expect(list.size).to eq 4
+    expect(list.drop(1)).to eq 1
+    expect(list.size).to eq 3
+    expect(list.drop(1)).to eq 1
+    expect(list.size).to eq 1
   end
 
   it 'can determine location for system prompt' do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ollama_chat
 version: !ruby/object:Gem::Version
-  version: 0.0.14
+  version: 0.0.16
 platform: ruby
 authors:
 - Florian Frank
@@ -155,6 +155,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 0.0.2
+- !ruby/object:Gem::Dependency
+  name: unix_socks
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rss
   requirement: !ruby/object:Gem::Requirement
@@ -479,7 +493,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.7
+rubygems_version: 3.6.9
 specification_version: 4
 summary: A command-line interface (CLI) for interacting with an Ollama AI model.
 test_files: