ollama_chat 0.0.24 → 0.0.26

data/lib/ollama_chat/utils/file_argument.rb

@@ -1,3 +1,22 @@
+# A module that provides functionality for handling file arguments and content
+# retrieval.
+#
+# The FileArgument module offers methods to process either file paths or direct
+# string content, determining whether the input represents a file that should
+# be read or if it's already a string of content to be used directly. It also
+# includes logic to handle default values when no valid input is provided.
+#
+# @example Retrieving file contents or using direct content
+#   FileArgument.get_file_argument('path/to/file.txt')
+#   # Returns the contents of the file if it exists
+#
+# @example Using a string as content
+#   FileArgument.get_file_argument('direct content string')
+#   # Returns the string itself
+#
+# @example Providing a default value
+#   FileArgument.get_file_argument(nil, default: 'fallback content')
+#   # Returns 'fallback content' when no valid input is given
 module OllamaChat::Utils::FileArgument
   module_function
 

data/lib/ollama_chat/utils.rb

@@ -1,3 +1,11 @@
+# A module that provides utility classes and methods for the OllamaChat
+# application.
+#
+# The Utils module serves as a namespace for various helper components that
+# support the core functionality of OllamaChat. It contains implementations for
+# caching, interactive selection, content fetching, and file argument handling
+# that are used throughout the application to provide robust and user-friendly
+# features.
 module OllamaChat::Utils
 end
 

data/lib/ollama_chat/version.rb

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

data/lib/ollama_chat/vim.rb

@@ -1,6 +1,12 @@
 require 'tempfile'
 require 'pathname'
 
+# A class that provides functionality for inserting text into Vim buffers via
+# remote communication.
+#
+# @example
+#   vim = OllamaChat::Vim.new("MY_SERVER")
+#   vim.insert("Hello, Vim!")
 class OllamaChat::Vim
   # Initializes a new Vim server connection
   #

data/lib/ollama_chat/web_searching.rb

@@ -1,3 +1,14 @@
+# A module that provides web search functionality for OllamaChat.
+#
+# The WebSearching module encapsulates the logic for performing web searches
+# using configured search engines. It handles query construction, location
+# information integration, and delegates to engine-specific implementations for
+# retrieving search results. The module supports multiple search engines
+# including SearxNG and DuckDuckGo, making it flexible for different deployment
+# scenarios and privacy preferences.
+#
+# @example Performing a web search
+#   chat.search_web('ruby programming tutorials', 5)
 module OllamaChat::WebSearching
   # The search_web method performs a web search using the configured search
   # engine.

data/lib/ollama_chat.rb

@@ -1,3 +1,12 @@
+# The main module namespace for the OllamaChat application.
+#
+# This module serves as the root namespace for all components of the OllamaChat
+# Ruby gem, providing access to core classes, utilities, and configuration
+# management for interacting with Ollama language models through a terminal
+# interface.
+#
+# @example Accessing the main module
+#   OllamaChat::VERSION # => "0.0.25"
 module OllamaChat
 end
 

data/ollama_chat.gemspec

@@ -1,9 +1,9 @@
 # -*- encoding: utf-8 -*-
-# stub: ollama_chat 0.0.24 ruby lib
+# stub: ollama_chat 0.0.26 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "ollama_chat".freeze
-  s.version = "0.0.24".freeze
+  s.version = "0.0.26".freeze
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0".freeze) if s.respond_to? :required_rubygems_version=
   s.require_paths = ["lib".freeze]
@@ -13,26 +13,27 @@ Gem::Specification.new do |s|
   s.email = "flori@ping.de".freeze
   s.executables = ["ollama_chat".freeze, "ollama_chat_send".freeze]
   s.extra_rdoc_files = ["README.md".freeze, "lib/ollama_chat.rb".freeze, "lib/ollama_chat/chat.rb".freeze, "lib/ollama_chat/clipboard.rb".freeze, "lib/ollama_chat/dialog.rb".freeze, "lib/ollama_chat/document_cache.rb".freeze, "lib/ollama_chat/follow_chat.rb".freeze, "lib/ollama_chat/history.rb".freeze, "lib/ollama_chat/information.rb".freeze, "lib/ollama_chat/kramdown_ansi.rb".freeze, "lib/ollama_chat/message_format.rb".freeze, "lib/ollama_chat/message_list.rb".freeze, "lib/ollama_chat/message_output.rb".freeze, "lib/ollama_chat/model_handling.rb".freeze, "lib/ollama_chat/ollama_chat_config.rb".freeze, "lib/ollama_chat/parsing.rb".freeze, "lib/ollama_chat/server_socket.rb".freeze, "lib/ollama_chat/source_fetching.rb".freeze, "lib/ollama_chat/switches.rb".freeze, "lib/ollama_chat/utils.rb".freeze, "lib/ollama_chat/utils/cache_fetcher.rb".freeze, "lib/ollama_chat/utils/chooser.rb".freeze, "lib/ollama_chat/utils/fetcher.rb".freeze, "lib/ollama_chat/utils/file_argument.rb".freeze, "lib/ollama_chat/version.rb".freeze, "lib/ollama_chat/vim.rb".freeze, "lib/ollama_chat/web_searching.rb".freeze]
-  s.files = [".all_images.yml".freeze, ".envrc".freeze, ".gitignore".freeze, "CHANGES.md".freeze, "Gemfile".freeze, "README.md".freeze, "Rakefile".freeze, "VERSION".freeze, "bin/ollama_chat".freeze, "bin/ollama_chat_send".freeze, "config/searxng/settings.yml".freeze, "docker-compose.yml".freeze, "lib/ollama_chat.rb".freeze, "lib/ollama_chat/chat.rb".freeze, "lib/ollama_chat/clipboard.rb".freeze, "lib/ollama_chat/dialog.rb".freeze, "lib/ollama_chat/document_cache.rb".freeze, "lib/ollama_chat/follow_chat.rb".freeze, "lib/ollama_chat/history.rb".freeze, "lib/ollama_chat/information.rb".freeze, "lib/ollama_chat/kramdown_ansi.rb".freeze, "lib/ollama_chat/message_format.rb".freeze, "lib/ollama_chat/message_list.rb".freeze, "lib/ollama_chat/message_output.rb".freeze, "lib/ollama_chat/model_handling.rb".freeze, "lib/ollama_chat/ollama_chat_config.rb".freeze, "lib/ollama_chat/ollama_chat_config/default_config.yml".freeze, "lib/ollama_chat/parsing.rb".freeze, "lib/ollama_chat/server_socket.rb".freeze, "lib/ollama_chat/source_fetching.rb".freeze, "lib/ollama_chat/switches.rb".freeze, "lib/ollama_chat/utils.rb".freeze, "lib/ollama_chat/utils/cache_fetcher.rb".freeze, "lib/ollama_chat/utils/chooser.rb".freeze, "lib/ollama_chat/utils/fetcher.rb".freeze, "lib/ollama_chat/utils/file_argument.rb".freeze, "lib/ollama_chat/version.rb".freeze, "lib/ollama_chat/vim.rb".freeze, "lib/ollama_chat/web_searching.rb".freeze, "ollama_chat.gemspec".freeze, "redis/redis.conf".freeze, "spec/assets/api_show.json".freeze, "spec/assets/api_tags.json".freeze, "spec/assets/api_version.json".freeze, "spec/assets/conversation.json".freeze, "spec/assets/duckduckgo.html".freeze, "spec/assets/example.atom".freeze, "spec/assets/example.csv".freeze, "spec/assets/example.html".freeze, "spec/assets/example.pdf".freeze, "spec/assets/example.ps".freeze, "spec/assets/example.rb".freeze, "spec/assets/example.rss".freeze, "spec/assets/example.xml".freeze, "spec/assets/kitten.jpg".freeze, "spec/assets/prompt.txt".freeze, "spec/assets/searxng.json".freeze, "spec/ollama_chat/chat_spec.rb".freeze, "spec/ollama_chat/clipboard_spec.rb".freeze, "spec/ollama_chat/follow_chat_spec.rb".freeze, "spec/ollama_chat/information_spec.rb".freeze, "spec/ollama_chat/message_list_spec.rb".freeze, "spec/ollama_chat/message_output_spec.rb".freeze, "spec/ollama_chat/model_handling_spec.rb".freeze, "spec/ollama_chat/parsing_spec.rb".freeze, "spec/ollama_chat/server_socket_spec.rb".freeze, "spec/ollama_chat/source_fetching_spec.rb".freeze, "spec/ollama_chat/switches_spec.rb".freeze, "spec/ollama_chat/utils/cache_fetcher_spec.rb".freeze, "spec/ollama_chat/utils/fetcher_spec.rb".freeze, "spec/ollama_chat/utils/file_argument_spec.rb".freeze, "spec/ollama_chat/web_searching_spec.rb".freeze, "spec/spec_helper.rb".freeze, "tmp/.keep".freeze]
+  s.files = [".all_images.yml".freeze, ".contexts/code_comment.rb".freeze, ".contexts/full.rb".freeze, ".contexts/info.rb".freeze, ".contexts/lib.rb".freeze, ".contexts/yard.md".freeze, ".envrc".freeze, ".gitignore".freeze, "CHANGES.md".freeze, "Gemfile".freeze, "README.md".freeze, "Rakefile".freeze, "VERSION".freeze, "bin/ollama_chat".freeze, "bin/ollama_chat_send".freeze, "config/searxng/settings.yml".freeze, "docker-compose.yml".freeze, "lib/ollama_chat.rb".freeze, "lib/ollama_chat/chat.rb".freeze, "lib/ollama_chat/clipboard.rb".freeze, "lib/ollama_chat/dialog.rb".freeze, "lib/ollama_chat/document_cache.rb".freeze, "lib/ollama_chat/follow_chat.rb".freeze, "lib/ollama_chat/history.rb".freeze, "lib/ollama_chat/information.rb".freeze, "lib/ollama_chat/kramdown_ansi.rb".freeze, "lib/ollama_chat/message_format.rb".freeze, "lib/ollama_chat/message_list.rb".freeze, "lib/ollama_chat/message_output.rb".freeze, "lib/ollama_chat/model_handling.rb".freeze, "lib/ollama_chat/ollama_chat_config.rb".freeze, "lib/ollama_chat/ollama_chat_config/default_config.yml".freeze, "lib/ollama_chat/parsing.rb".freeze, "lib/ollama_chat/server_socket.rb".freeze, "lib/ollama_chat/source_fetching.rb".freeze, "lib/ollama_chat/switches.rb".freeze, "lib/ollama_chat/utils.rb".freeze, "lib/ollama_chat/utils/cache_fetcher.rb".freeze, "lib/ollama_chat/utils/chooser.rb".freeze, "lib/ollama_chat/utils/fetcher.rb".freeze, "lib/ollama_chat/utils/file_argument.rb".freeze, "lib/ollama_chat/version.rb".freeze, "lib/ollama_chat/vim.rb".freeze, "lib/ollama_chat/web_searching.rb".freeze, "ollama_chat.gemspec".freeze, "redis/redis.conf".freeze, "spec/assets/api_show.json".freeze, "spec/assets/api_tags.json".freeze, "spec/assets/api_version.json".freeze, "spec/assets/conversation.json".freeze, "spec/assets/duckduckgo.html".freeze, "spec/assets/example.atom".freeze, "spec/assets/example.csv".freeze, "spec/assets/example.html".freeze, "spec/assets/example.pdf".freeze, "spec/assets/example.ps".freeze, "spec/assets/example.rb".freeze, "spec/assets/example.rss".freeze, "spec/assets/example.xml".freeze, "spec/assets/kitten.jpg".freeze, "spec/assets/prompt.txt".freeze, "spec/assets/searxng.json".freeze, "spec/ollama_chat/chat_spec.rb".freeze, "spec/ollama_chat/clipboard_spec.rb".freeze, "spec/ollama_chat/follow_chat_spec.rb".freeze, "spec/ollama_chat/information_spec.rb".freeze, "spec/ollama_chat/kramdown_ansi_spec.rb".freeze, "spec/ollama_chat/message_list_spec.rb".freeze, "spec/ollama_chat/message_output_spec.rb".freeze, "spec/ollama_chat/model_handling_spec.rb".freeze, "spec/ollama_chat/parsing_spec.rb".freeze, "spec/ollama_chat/server_socket_spec.rb".freeze, "spec/ollama_chat/source_fetching_spec.rb".freeze, "spec/ollama_chat/switches_spec.rb".freeze, "spec/ollama_chat/utils/cache_fetcher_spec.rb".freeze, "spec/ollama_chat/utils/fetcher_spec.rb".freeze, "spec/ollama_chat/utils/file_argument_spec.rb".freeze, "spec/ollama_chat/web_searching_spec.rb".freeze, "spec/spec_helper.rb".freeze, "tmp/.keep".freeze]
   s.homepage = "https://github.com/flori/ollama_chat".freeze
   s.licenses = ["MIT".freeze]
   s.rdoc_options = ["--title".freeze, "OllamaChat - A command-line interface (CLI) for interacting with an Ollama AI model.".freeze, "--main".freeze, "README.md".freeze]
   s.required_ruby_version = Gem::Requirement.new("~> 3.1".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/message_output_spec.rb".freeze, "spec/ollama_chat/model_handling_spec.rb".freeze, "spec/ollama_chat/parsing_spec.rb".freeze, "spec/ollama_chat/server_socket_spec.rb".freeze, "spec/ollama_chat/source_fetching_spec.rb".freeze, "spec/ollama_chat/switches_spec.rb".freeze, "spec/ollama_chat/utils/cache_fetcher_spec.rb".freeze, "spec/ollama_chat/utils/fetcher_spec.rb".freeze, "spec/ollama_chat/utils/file_argument_spec.rb".freeze, "spec/ollama_chat/web_searching_spec.rb".freeze, "spec/spec_helper.rb".freeze]
+  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/kramdown_ansi_spec.rb".freeze, "spec/ollama_chat/message_list_spec.rb".freeze, "spec/ollama_chat/message_output_spec.rb".freeze, "spec/ollama_chat/model_handling_spec.rb".freeze, "spec/ollama_chat/parsing_spec.rb".freeze, "spec/ollama_chat/server_socket_spec.rb".freeze, "spec/ollama_chat/source_fetching_spec.rb".freeze, "spec/ollama_chat/switches_spec.rb".freeze, "spec/ollama_chat/utils/cache_fetcher_spec.rb".freeze, "spec/ollama_chat/utils/fetcher_spec.rb".freeze, "spec/ollama_chat/utils/file_argument_spec.rb".freeze, "spec/ollama_chat/web_searching_spec.rb".freeze, "spec/spec_helper.rb".freeze]
 
   s.specification_version = 4
 
-  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 2.0".freeze])
+  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 2.1".freeze])
   s.add_development_dependency(%q<all_images>.freeze, ["~> 0.6".freeze])
   s.add_development_dependency(%q<rspec>.freeze, ["~> 3.2".freeze])
   s.add_development_dependency(%q<kramdown>.freeze, ["~> 2.0".freeze])
   s.add_development_dependency(%q<webmock>.freeze, [">= 0".freeze])
   s.add_development_dependency(%q<debug>.freeze, [">= 0".freeze])
   s.add_development_dependency(%q<simplecov>.freeze, [">= 0".freeze])
+  s.add_development_dependency(%q<context_spook>.freeze, [">= 0".freeze])
   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<ollama-ruby>.freeze, ["~> 1.6".freeze])
   s.add_runtime_dependency(%q<documentrix>.freeze, ["~> 0.0".freeze, ">= 0.0.2".freeze])
   s.add_runtime_dependency(%q<unix_socks>.freeze, [">= 0.0.1".freeze])
   s.add_runtime_dependency(%q<rss>.freeze, ["~> 0.3".freeze])
@@ -43,7 +44,7 @@ Gem::Specification.new do |s|
   s.add_runtime_dependency(%q<xdg>.freeze, [">= 0".freeze])
   s.add_runtime_dependency(%q<kramdown-ansi>.freeze, ["~> 0.1".freeze])
   s.add_runtime_dependency(%q<complex_config>.freeze, ["~> 0.22".freeze, ">= 0.22.2".freeze])
-  s.add_runtime_dependency(%q<tins>.freeze, ["~> 1.34".freeze])
+  s.add_runtime_dependency(%q<tins>.freeze, ["~> 1.41".freeze])
   s.add_runtime_dependency(%q<search_ui>.freeze, ["~> 0.0".freeze])
   s.add_runtime_dependency(%q<amatch>.freeze, ["~> 0.4.1".freeze])
   s.add_runtime_dependency(%q<pdf-reader>.freeze, ["~> 2.0".freeze])

data/spec/ollama_chat/chat_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-describe OllamaChat::Chat do
+describe OllamaChat::Chat, protect_env: true do
   let :argv do
     %w[ -C test ]
   end
@@ -77,6 +77,16 @@ describe OllamaChat::Chat do
       expect(chat.handle_input("/clobber")).to eq :next
     end
 
+    it 'returns :next when input is "/last"' do
+      expect(chat.messages).to receive(:show_last)
+      expect(chat.handle_input("/last")).to eq :next
+    end
+
+    it 'returns :next when input is "/last\s+(\d+)"' do
+      expect(chat.messages).to receive(:show_last).with(2)
+      expect(chat.handle_input("/last 2")).to eq :next
+    end
+
     it 'returns :next when input is "/drop(?:\s+(\d*))?"' do
       expect(chat.messages).to receive(:drop).with(?2)
       expect(chat.messages).to receive(:show_last)

data/spec/ollama_chat/kramdown_ansi_spec.rb

@@ -0,0 +1,45 @@
+require 'spec_helper'
+
+describe OllamaChat::KramdownANSI do
+  let :chat do
+    double('Chat').extend(described_class)
+  end
+
+  describe '#configure_kramdown_ansi_styles', protect_env: true do
+    it 'can be configured via env var' do
+      ENV['KRAMDOWN_ANSI_OLLAMA_CHAT_STYLES'] = '{}'
+      ENV.delete('KRAMDOWN_ANSI_STYLES')
+
+      styles = { bold: '1' }
+      expect(Kramdown::ANSI::Styles).to receive(:from_env_var).
+        with('KRAMDOWN_ANSI_OLLAMA_CHAT_STYLES').
+        and_return(double(ansi_styles: styles))
+
+      expect(chat.configure_kramdown_ansi_styles).to eq(styles)
+    end
+
+    it 'has a default configuration' do
+      ENV.delete('KRAMDOWN_ANSI_OLLAMA_CHAT_STYLES')
+      ENV.delete('KRAMDOWN_ANSI_STYLES')
+
+      expect(chat.configure_kramdown_ansi_styles).to be_a(Hash)
+    end
+  end
+
+  describe '#kramdown_ansi_parse' do
+    it 'can parse markdown' do
+      content = "# Header\n\nParagraph text"
+      result = chat.kramdown_ansi_parse(content)
+
+      expect(result).to be_a(String)
+    end
+
+    it 'handles empty string correctly' do
+      expect(chat.kramdown_ansi_parse('')).to eq ''
+    end
+
+    it 'handles nil correctly' do
+      expect(chat.kramdown_ansi_parse(nil)).to eq ''
+    end
+  end
+end

data/spec/ollama_chat/message_list_spec.rb

@@ -39,7 +39,7 @@ describe OllamaChat::MessageList do
     expect(list.size).to eq 1
     list.clear
     expect(list.size).to eq 1
-    list <<  Ollama::Message.new(role: 'user', content: 'world')
+    list << Ollama::Message.new(role: 'user', content: 'world')
     expect(list.size).to eq 2
     list.clear
     expect(list.size).to eq 1
@@ -47,7 +47,7 @@ describe OllamaChat::MessageList do
 
   it 'can be added to' do
     expect(list.size).to eq 1
-    list <<  Ollama::Message.new(role: 'user', content: 'world')
+    list << Ollama::Message.new(role: 'user', content: 'world')
     expect(list.size).to eq 2
   end
 
@@ -71,6 +71,67 @@ describe OllamaChat::MessageList do
     FileUtils.rm_f 'tmp/test-conversation.json'
   end
 
+  describe "#last" do
+    it "returns the last message when there are multiple messages" do
+      list = described_class.new(chat)
+      list << Ollama::Message.new(role: 'system', content: 'hello')
+      list << Ollama::Message.new(role: 'user', content: 'First message')
+      list << Ollama::Message.new(role: 'assistant', content: 'Second message')
+
+      expect(list.last.content).to eq('Second message')
+    end
+
+    it "returns the last message when there is only one message" do
+      list = described_class.new(chat)
+      list << Ollama::Message.new(role: 'system', content: 'hello')
+
+      expect(list.last.content).to eq('hello')
+    end
+
+    it "returns nil when there are no messages" do
+      list = described_class.new(chat)
+
+      expect(list.last).to be_nil
+    end
+  end
+
+  describe '#show_last' do
+    it 'shows nothing when there are no messages' do
+      empty_list = described_class.new(chat)
+      expect { empty_list.show_last }.not_to raise_error
+      expect(empty_list.show_last).to be nil
+    end
+
+    it 'shows nothing when the last message is by the assistant' do
+      list = described_class.new(chat)
+      allow(chat).to receive(:think).and_return(double(on?: false))
+      allow(chat).to receive(:markdown).and_return(double(on?: false))
+      list << Ollama::Message.new(role: 'assistant', content: 'hello')
+      expect(STDOUT).to receive(:puts).
+        with("📨 \e[1m\e[38;5;111massistant\e[0m\e[0m:\nhello\n")
+      expect(list.show_last).to be_a described_class
+    end
+
+    it 'shows nothing when the last message is by the user' do
+      list = described_class.new(chat)
+      list << Ollama::Message.new(role: 'user', content: 'world')
+      expect { list.show_last }.not_to raise_error
+      expect(list.show_last).to be nil
+    end
+
+    it "shows last N messages when N is larger than available messages" do
+      allow(chat).to receive(:think).and_return(double(on?: false))
+      allow(chat).to receive(:markdown).and_return(double(on?: false))
+      list = described_class.new(chat)
+      list << Ollama::Message.new(role: 'system', content: 'hello')
+      list << Ollama::Message.new(role: 'user', content: 'First message')
+      list << Ollama::Message.new(role: 'assistant', content: 'Second message')
+
+      expect(STDOUT).to receive(:puts).with(/Second message/)
+      expect(list.show_last(23)).to eq(list)
+    end
+  end
+
   context 'without pager' do
     before do
       expect(list).to receive(:determine_pager_command).and_return nil
@@ -91,7 +152,7 @@ describe OllamaChat::MessageList do
         and_return(double(on?: true)).at_least(:once)
       expect(chat).to receive(:think).
         and_return(double(on?: false)).at_least(:once)
-      list <<  Ollama::Message.new(role: 'user', content: 'world')
+      list << Ollama::Message.new(role: 'user', content: 'world')
       expect(STDOUT).to receive(:puts).
         with(
           "📨 \e[1m\e[38;5;213msystem\e[0m\e[0m:\nhello\n" \
@@ -132,7 +193,7 @@ describe OllamaChat::MessageList do
         and_return(double(on?: true)).at_least(:once)
       expect(chat).to receive(:think).
         and_return(double(on?: false)).at_least(:once)
-      list <<  Ollama::Message.new(role: 'user', content: 'world')
+      list << Ollama::Message.new(role: 'user', content: 'world')
       list.list_conversation
     end
   end
@@ -169,9 +230,9 @@ describe OllamaChat::MessageList do
     expect(list.size).to eq 1
     expect(list.drop(1)).to eq 0
     expect(list.size).to eq 1
-    list <<  Ollama::Message.new(role: 'user', content: 'world')
+    list << Ollama::Message.new(role: 'user', content: 'world')
     expect(list.size).to eq 2
-    list <<  Ollama::Message.new(role: 'assistant', content: 'hi')
+    list << Ollama::Message.new(role: 'assistant', content: 'hi')
     expect(list.size).to eq 3
     expect(list.drop(1)).to eq 1
     expect(list.size).to eq 1
@@ -201,7 +262,7 @@ describe OllamaChat::MessageList do
 
   it 'can be converted int an Ollama::Message array' do
     expect(chat).to receive(:location).and_return(double(on?: false))
-    list <<  Ollama::Message.new(role: 'user', content: 'world')
+    list << Ollama::Message.new(role: 'user', content: 'world')
     expect(list.to_ary.map(&:as_json)).to eq [
       Ollama::Message.new(role: 'system', content: 'hello').as_json,
       Ollama::Message.new(role: 'user', content: 'world').as_json,
@@ -210,7 +271,7 @@ describe OllamaChat::MessageList do
 
   it 'can be converted int an Ollama::Message array with location' do
     expect(chat).to receive(:location).and_return(double(on?: true))
-    list <<  Ollama::Message.new(role: 'user', content: 'world')
+    list << Ollama::Message.new(role: 'user', content: 'world')
     first = list.to_ary.first
     expect(first.role).to eq 'system'
     expect(first.content).to match(
@@ -220,7 +281,7 @@ describe OllamaChat::MessageList do
   it 'can be converted int an Ollama::Message array with location without a system prompt' do
     expect(chat).to receive(:location).and_return(double(on?: true))
     list = described_class.new(chat).tap do |list|
-      list <<  Ollama::Message.new(role: 'user', content: 'hello')
+      list << Ollama::Message.new(role: 'user', content: 'hello')
       list << Ollama::Message.new(role: 'assistant', content: 'world')
     end
     first = list.to_ary.first

data/spec/ollama_chat/server_socket_spec.rb

@@ -9,75 +9,96 @@ describe OllamaChat::ServerSocket do
     let(:config) { double('Config') }
     let(:server) { double('Server') }
 
-    before do
-      expect(OllamaChat::ServerSocket).to receive(:create_socket_server).with(config: config).and_return(server)
-    end
+    context 'without runtime_dir' do
 
-    context 'with default parameters' do
-      it 'uses correct defaults' do
-        message = { content: 'test', type: :socket_input, parse: false }
+      before do
+        expect(OllamaChat::ServerSocket).to receive(:create_socket_server).
+          with(config: config, runtime_dir: nil).and_return(server)
+      end
 
-        expect(server).to receive(:transmit).with(message).and_return(nil)
+      context 'with default parameters' do
+        it 'uses correct defaults' do
+          message = { content: 'test', type: :socket_input, parse: false }
 
-        result = OllamaChat::ServerSocket.send_to_server_socket('test', config: config)
+          expect(server).to receive(:transmit).with(message).and_return(nil)
 
-        expect(result).to be_nil
+          result = OllamaChat::ServerSocket.send_to_server_socket('test', config: config)
+
+          expect(result).to be_nil
+        end
       end
-    end
 
-    context 'with :socket_input type and parse: true' do
-      it 'sends message with parse flag and returns nil' do
-        message = { content: 'test', type: :socket_input, parse: true }
+      context 'with :socket_input type and parse: true' do
+        it 'sends message with parse flag and returns nil' do
+          message = { content: 'test', type: :socket_input, parse: true }
 
-        expect(server).to receive(:transmit).with(message).and_return(nil)
+          expect(server).to receive(:transmit).with(message).and_return(nil)
 
-        result = OllamaChat::ServerSocket.send_to_server_socket(
-          'test',
-          config: config,
-          type: :socket_input,
-          parse: true
-        )
+          result = OllamaChat::ServerSocket.send_to_server_socket(
+            'test',
+            config: config,
+            type: :socket_input,
+            parse: true
+          )
 
-        expect(result).to be_nil
+          expect(result).to be_nil
+        end
       end
-    end
 
-    context 'with :socket_input_with_response type and parse: false' do
-      it 'sends message and returns response with parse flag' do
-        message = { content: 'test', type: :socket_input_with_response, parse: false }
-        response = double('Response')
+      context 'with :socket_input_with_response type and parse: false' do
+        it 'sends message and returns response with parse flag' do
+          message = { content: 'test', type: :socket_input_with_response, parse: false }
+          response = double('Response')
 
-        expect(server).to receive(:transmit_with_response).with(message).and_return(response)
+          expect(server).to receive(:transmit_with_response).with(message).and_return(response)
 
-        result = OllamaChat::ServerSocket.send_to_server_socket(
-          'test',
-          config: config,
-          type: :socket_input_with_response,
-          parse: false
-        )
+          result = OllamaChat::ServerSocket.send_to_server_socket(
+            'test',
+            config: config,
+            type: :socket_input_with_response,
+            parse: false
+          )
 
-        expect(result).to eq(response)
+          expect(result).to eq(response)
+        end
       end
-    end
 
-    context 'with :socket_input_with_response type and parse: true' do
-      it 'sends message and returns response with parse flag' do
-        message = { content: 'test', type: :socket_input_with_response, parse: true }
-        response = double('Response')
+      context 'with :socket_input_with_response type and parse: true' do
+        it 'sends message and returns response with parse flag' do
+          message = { content: 'test', type: :socket_input_with_response, parse: true }
+          response = double('Response')
 
-        expect(server).to receive(:transmit_with_response).with(message).and_return(response)
+          expect(server).to receive(:transmit_with_response).with(message).and_return(response)
 
-        result = OllamaChat::ServerSocket.send_to_server_socket(
-          'test',
-          config: config,
-          type: :socket_input_with_response,
-          parse: true
-        )
+          result = OllamaChat::ServerSocket.send_to_server_socket(
+            'test',
+            config: config,
+            type: :socket_input_with_response,
+            parse: true
+          )
 
-        expect(result).to eq(response)
+          expect(result).to eq(response)
+        end
       end
     end
 
+    context 'with runtime_dir parameter' do
+      before do
+        expect(OllamaChat::ServerSocket).to receive(:create_socket_server).
+          with(config: config, runtime_dir: '/foo/bar').and_return(server)
+      end
+
+      it 'uses correct defaults' do
+        message = { content: 'test', type: :socket_input, parse: false }
+
+        expect(server).to receive(:transmit).with(message).and_return(nil)
+
+
+        result = OllamaChat::ServerSocket.send_to_server_socket('test', config: config, runtime_dir: '/foo/bar')
+
+        expect(result).to be_nil
+      end
+    end
   end
 
   describe '#create_socket_server' do