ollama_chat 0.0.55 → 0.0.57

data/lib/ollama_chat/think_control.rb

@@ -5,52 +5,24 @@
 # thinking modes, checking the current state, and displaying the current
 # think mode status.
 module OllamaChat::ThinkControl
-  # The think method returns the current state of the think mode.
+  # The think method returns the current think mode selection.
   #
-  # @return [ true, false, String ] the think mode
-  attr_reader :think
-
-  # The choose_think_mode method presents a menu to select a think mode.
-  #
-  # This method displays available think modes to the user and sets the
-  # selected mode as the current think mode for the chat session.
-  def choose_think_mode
-    think_modes = %w[ off on low medium high [EXIT] ]
-    case chosen = OllamaChat::Utils::Chooser.choose(think_modes)
-    when '[EXIT]', nil
-      STDOUT.puts "Exiting chooser."
-    when 'off'
-      @think = false
-    when 'on'
-      @think = true
-    when 'low', 'medium', 'high'
-      @think = chosen
+  # @return [ String ] the selected think mode value
+  def think
+    if think_mode.off?
+      false
+    elsif think_mode.selected == 'enabled'
+      true
+    else
+      think_mode.selected
     end
   end
 
   # The think? method checks if the think mode is enabled.
   #
-  # @return [ TrueClass, FalseClass ] true if think mode is enabled, false
-  #   otherwise
+  # @return [TrueClass, FalseClass] true if think mode is enabled, false otherwise
   def think?
-    !!think
-  end
-
-  # The think_mode method returns the current think mode status as a string.
-  #
-  # @return [ String ] returns 'enabled' if think mode is true, the think mode
-  #   value if it's a string, or 'disabled' if think mode is false or nil
-  def think_mode
-    think == true ? 'enabled' : think || 'disabled'
-  end
-
-  # The think_show method displays the current think mode status.
-  #
-  # This method checks the current think mode setting and outputs a message
-  # indicating whether think mode is enabled, disabled, or set to a specific
-  # mode level (low, medium, high).
-  def think_show
-    STDOUT.puts "Think mode is #{bold(think_mode)}."
+    think_mode.on?
   end
 
   # The think_loud? method checks if both think mode and think loud mode are

data/lib/ollama_chat/version.rb

@@ -1,6 +1,6 @@
 module OllamaChat
   # OllamaChat version
-  VERSION         = '0.0.55'
+  VERSION         = '0.0.57'
   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

@@ -20,6 +20,7 @@ require 'ollama_chat/message_format'
 require 'ollama_chat/ollama_chat_config'
 require 'ollama_chat/follow_chat'
 require 'ollama_chat/switches'
+require 'ollama_chat/state_selectors'
 require 'ollama_chat/message_list'
 require 'ollama_chat/model_handling'
 require 'ollama_chat/parsing'

data/ollama_chat.gemspec

@@ -1,9 +1,9 @@
 # -*- encoding: utf-8 -*-
-# stub: ollama_chat 0.0.55 ruby lib
+# stub: ollama_chat 0.0.57 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "ollama_chat".freeze
-  s.version = "0.0.55".freeze
+  s.version = "0.0.57".freeze
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0".freeze) if s.respond_to? :required_rubygems_version=
   s.require_paths = ["lib".freeze]
@@ -12,19 +12,19 @@ Gem::Specification.new do |s|
   s.description = "The app provides a command-line interface (CLI) to an Ollama AI model,\nallowing users to engage in text-based conversations and generate\nhuman-like responses. Users can import data from local files or web pages,\nwhich are then processed through three different modes: fully importing the\ncontent into the conversation context, summarizing the information for\nconcise reference, or storing it in an embedding vector database for later\nretrieval based on the conversation.\n".freeze
   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/conversation.rb".freeze, "lib/ollama_chat/dialog.rb".freeze, "lib/ollama_chat/document_cache.rb".freeze, "lib/ollama_chat/env_config.rb".freeze, "lib/ollama_chat/follow_chat.rb".freeze, "lib/ollama_chat/history.rb".freeze, "lib/ollama_chat/information.rb".freeze, "lib/ollama_chat/input_content.rb".freeze, "lib/ollama_chat/kramdown_ansi.rb".freeze, "lib/ollama_chat/message_editing.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/redis_cache.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/think_control.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 = [".utilsrc".freeze, "CHANGES.md".freeze, "Gemfile".freeze, "README.md".freeze, "Rakefile".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/conversation.rb".freeze, "lib/ollama_chat/dialog.rb".freeze, "lib/ollama_chat/document_cache.rb".freeze, "lib/ollama_chat/env_config.rb".freeze, "lib/ollama_chat/follow_chat.rb".freeze, "lib/ollama_chat/history.rb".freeze, "lib/ollama_chat/information.rb".freeze, "lib/ollama_chat/input_content.rb".freeze, "lib/ollama_chat/kramdown_ansi.rb".freeze, "lib/ollama_chat/message_editing.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/redis_cache.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/think_control.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/example_with_quote.html".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/input_content_spec.rb".freeze, "spec/ollama_chat/kramdown_ansi_spec.rb".freeze, "spec/ollama_chat/message_editing_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/redis_cache_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/think_control_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/vim_spec.rb".freeze, "spec/ollama_chat/web_searching_spec.rb".freeze, "spec/spec_helper.rb".freeze, "tmp/.keep".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/conversation.rb".freeze, "lib/ollama_chat/dialog.rb".freeze, "lib/ollama_chat/document_cache.rb".freeze, "lib/ollama_chat/env_config.rb".freeze, "lib/ollama_chat/follow_chat.rb".freeze, "lib/ollama_chat/history.rb".freeze, "lib/ollama_chat/information.rb".freeze, "lib/ollama_chat/input_content.rb".freeze, "lib/ollama_chat/kramdown_ansi.rb".freeze, "lib/ollama_chat/message_editing.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/redis_cache.rb".freeze, "lib/ollama_chat/server_socket.rb".freeze, "lib/ollama_chat/source_fetching.rb".freeze, "lib/ollama_chat/state_selectors.rb".freeze, "lib/ollama_chat/switches.rb".freeze, "lib/ollama_chat/think_control.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 = [".utilsrc".freeze, "CHANGES.md".freeze, "Gemfile".freeze, "README.md".freeze, "Rakefile".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/conversation.rb".freeze, "lib/ollama_chat/dialog.rb".freeze, "lib/ollama_chat/document_cache.rb".freeze, "lib/ollama_chat/env_config.rb".freeze, "lib/ollama_chat/follow_chat.rb".freeze, "lib/ollama_chat/history.rb".freeze, "lib/ollama_chat/information.rb".freeze, "lib/ollama_chat/input_content.rb".freeze, "lib/ollama_chat/kramdown_ansi.rb".freeze, "lib/ollama_chat/message_editing.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/redis_cache.rb".freeze, "lib/ollama_chat/server_socket.rb".freeze, "lib/ollama_chat/source_fetching.rb".freeze, "lib/ollama_chat/state_selectors.rb".freeze, "lib/ollama_chat/switches.rb".freeze, "lib/ollama_chat/think_control.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/example_with_quote.html".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/input_content_spec.rb".freeze, "spec/ollama_chat/kramdown_ansi_spec.rb".freeze, "spec/ollama_chat/message_editing_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/redis_cache_spec.rb".freeze, "spec/ollama_chat/server_socket_spec.rb".freeze, "spec/ollama_chat/source_fetching_spec.rb".freeze, "spec/ollama_chat/state_selectors_spec.rb".freeze, "spec/ollama_chat/switches_spec.rb".freeze, "spec/ollama_chat/think_control_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/vim_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.2".freeze)
-  s.rubygems_version = "4.0.2".freeze
+  s.rubygems_version = "4.0.3".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/input_content_spec.rb".freeze, "spec/ollama_chat/kramdown_ansi_spec.rb".freeze, "spec/ollama_chat/message_editing_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/redis_cache_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/think_control_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/vim_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/input_content_spec.rb".freeze, "spec/ollama_chat/kramdown_ansi_spec.rb".freeze, "spec/ollama_chat/message_editing_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/redis_cache_spec.rb".freeze, "spec/ollama_chat/server_socket_spec.rb".freeze, "spec/ollama_chat/source_fetching_spec.rb".freeze, "spec/ollama_chat/state_selectors_spec.rb".freeze, "spec/ollama_chat/switches_spec.rb".freeze, "spec/ollama_chat/think_control_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/vim_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.16.3".freeze])
+  s.add_development_dependency(%q<gem_hadar>.freeze, [">= 2.17.0".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])
@@ -50,5 +50,5 @@ Gem::Specification.new do |s|
   s.add_runtime_dependency(%q<bigdecimal>.freeze, ["~> 3.1".freeze])
   s.add_runtime_dependency(%q<csv>.freeze, ["~> 3.0".freeze])
   s.add_runtime_dependency(%q<const_conf>.freeze, ["~> 0.3".freeze])
-  s.add_runtime_dependency(%q<context_spook>.freeze, ["~> 1.1".freeze])
+  s.add_runtime_dependency(%q<context_spook>.freeze, ["~> 1.5".freeze])
 end

data/spec/ollama_chat/chat_spec.rb

@@ -1,12 +1,8 @@
 require 'spec_helper'
 
 describe OllamaChat::Chat, protect_env: true do
-  use_default_config = -> a {
-    a << '-f' << 'lib/ollama_chat/ollama_chat_config/default_config.yml'
-  }
-
   let :argv do
-    use_default_config.(%w[ -C test ])
+    chat_default_config(%w[ -C test ])
   end
 
   before do
@@ -28,6 +24,11 @@ describe OllamaChat::Chat, protect_env: true do
   describe 'handle_input' do
     connect_to_ollama_server
 
+    it 'returns :next when input is "/reconnect"' do
+      expect(chat).to receive(:connect_ollama).and_return double('ollama')
+      expect(chat.handle_input("/reconnect")).to eq :next
+    end
+
     it 'returns :next when input is "/copy"' do
       expect(chat).to receive(:copy_to_clipboard)
       expect(chat.handle_input("/copy")).to eq :next
@@ -129,7 +130,7 @@ describe OllamaChat::Chat, protect_env: true do
     end
 
     it 'returns :next when input is "/document_policy"' do
-      expect(chat).to receive(:choose_document_policy)
+      expect_any_instance_of(OllamaChat::StateSelectors::StateSelector).to receive(:choose)
       expect(chat.handle_input("/document_policy")).to eq :next
     end
 
@@ -230,7 +231,7 @@ describe OllamaChat::Chat, protect_env: true do
     connect_to_ollama_server(instantiate: false)
 
     let :argv do
-      use_default_config.(%w[ -C test -c ] << asset('conversation.json'))
+      chat_default_config(%w[ -C test -c ] << asset('conversation.json'))
     end
 
     it 'dispays the last exchange of the converstation' do
@@ -247,7 +248,7 @@ describe OllamaChat::Chat, protect_env: true do
     context 'with MemoryCache' do
 
       let :argv do
-        use_default_config.(%w[ -M ])
+        chat_default_config(%w[ -M ])
       end
 
       it 'can use MemoryCache' do
@@ -269,7 +270,7 @@ describe OllamaChat::Chat, protect_env: true do
     connect_to_ollama_server(instantiate: false)
 
       let :argv do
-        use_default_config.(%w[ -C test -D ] << asset('example.html'))
+        chat_default_config(%w[ -C test -D ] << asset('example.html'))
       end
 
       it 'Adds documents passed to app via -D option' do

data/spec/ollama_chat/clipboard_spec.rb

@@ -2,7 +2,7 @@ require 'spec_helper'
 
 describe OllamaChat::Clipboard do
   let :chat do
-    OllamaChat::Chat.new
+    OllamaChat::Chat.new argv: chat_default_config
   end
 
   connect_to_ollama_server

data/spec/ollama_chat/information_spec.rb

@@ -2,7 +2,7 @@ require 'spec_helper'
 
 describe OllamaChat::Information do
   let :chat do
-    OllamaChat::Chat.new
+    OllamaChat::Chat.new argv: chat_default_config
   end
 
   connect_to_ollama_server

data/spec/ollama_chat/input_content_spec.rb

@@ -2,16 +2,19 @@ require 'spec_helper'
 
 describe OllamaChat::InputContent do
   let :chat do
-    OllamaChat::Chat.new
+    OllamaChat::Chat.new argv: chat_default_config
   end
 
   connect_to_ollama_server
 
   describe '#input' do
     it 'can read content from a selected file' do
+      selected_filename = 'spec/assets/example.rb'
       # Mock the file selection process
-      expect(chat).to receive(:choose_filename).with('**/*').
-        and_return('spec/assets/example.rb')
+      expect(chat).to receive(:choose_filename).with('**/*', chosen: Set[]).
+        and_return(selected_filename)
+      expect(chat).to receive(:choose_filename).with('**/*', chosen: Set[selected_filename]).
+        and_return nil
 
       # Test that it returns the file content
       result = chat.input(nil)
@@ -19,13 +22,19 @@ describe OllamaChat::InputContent do
     end
 
     it 'returns nil when no file is selected' do
-      expect(chat).to receive(:choose_filename).with('**/*').and_return(nil)
+      expect(chat).to receive(:choose_filename).with('**/*', chosen: Set[]).
+        and_return(nil)
       expect(chat.input(nil)).to be_nil
     end
 
     it 'can read content with specific pattern' do
-      expect(chat).to receive(:choose_filename).with('spec/assets/*').
-        and_return('spec/assets/example.rb')
+      selected_filename = 'spec/assets/example.rb'
+      expect(chat).to receive(:choose_filename).
+        with('spec/assets/*', chosen: Set[]).
+        and_return(selected_filename)
+      expect(chat).to receive(:choose_filename).
+        with('spec/assets/*', chosen: Set[selected_filename]).
+        and_return nil
       result = chat.input('spec/assets/*')
       expect(result).to include('puts "Hello World!"')
     end

data/spec/ollama_chat/message_editing_spec.rb

@@ -2,7 +2,7 @@ require 'spec_helper'
 
 describe OllamaChat::MessageEditing do
   let :chat do
-    OllamaChat::Chat.new
+    OllamaChat::Chat.new argv: chat_default_config
   end
 
   connect_to_ollama_server

data/spec/ollama_chat/message_output_spec.rb

@@ -2,7 +2,7 @@ require 'spec_helper'
 
 describe OllamaChat::MessageOutput do
   let :chat do
-    OllamaChat::Chat.new
+    OllamaChat::Chat.new argv: chat_default_config
   end
 
   connect_to_ollama_server

data/spec/ollama_chat/model_handling_spec.rb

@@ -2,7 +2,7 @@ require 'spec_helper'
 
 describe OllamaChat::ModelHandling do
   let :chat do
-    OllamaChat::Chat.new
+    OllamaChat::Chat.new argv: chat_default_config
   end
 
   connect_to_ollama_server

data/spec/ollama_chat/parsing_spec.rb

@@ -3,8 +3,8 @@ require 'pathname'
 
 describe OllamaChat::Parsing do
   let :chat do
-    OllamaChat::Chat.new.tap do |chat|
-      chat.document_policy = 'importing'
+    OllamaChat::Chat.new(argv: chat_default_config).tap do |chat|
+      chat.document_policy.selected = 'importing'
     end
   end
 
@@ -49,24 +49,7 @@ describe OllamaChat::Parsing do
         def io.content_type
           'text/csv'
         end
-        expect(chat.parse_source(io)).to eq(<<EOT)
-  name: John Doe
-  age: 32
-  occupation: Software Engineer
-
-  name: Jane Smith
-  age: 28
-  occupation: Marketing Manager
-
-  name: Bob Johnson
-  age: 45
-  occupation: Retired
-
-  name: Alice Brown
-  age: 25
-  occupation: Student
-
-EOT
+        expect(chat.parse_source(io)).to eq(asset_content('example.csv'))
       end
     end
 
@@ -250,14 +233,14 @@ EOT
 
     context 'document_policy' do
       it 'can be ignoring' do
-        chat.document_policy = 'ignoring'
+        chat.document_policy.selected = 'ignoring'
         c = "see #{Dir.pwd}/spec/assets/example.html"
         content, = chat.parse_content(c, [])
         expect(content).to eq(c)
       end
 
       it 'can be importing' do
-        chat.document_policy = 'importing'
+        chat.document_policy.selected = 'importing'
         c = "see #{Dir.pwd}/spec/assets/example.html"
         content, = chat.parse_content(c, [])
         expect(content).to include(<<~EOT)
@@ -270,7 +253,7 @@ EOT
       end
 
       it 'can be embedding' do
-        chat.document_policy = 'embedding'
+        chat.document_policy.selected = 'embedding'
         c = "see #{Dir.pwd}/spec/assets/example.html"
         expect(chat).to receive(:embed_source).with(
           kind_of(IO),
@@ -281,7 +264,7 @@ EOT
       end
 
       it 'can be summarizing' do
-        chat.document_policy = 'summarizing'
+        chat.document_policy.selected = 'summarizing'
         c = "see #{Dir.pwd}/spec/assets/example.html"
         content, = chat.parse_content(c, [])
         expect(content).to start_with(<<~EOT)

data/spec/ollama_chat/source_fetching_spec.rb

@@ -2,9 +2,7 @@ require 'spec_helper'
 
 describe OllamaChat::SourceFetching do
   let :chat do
-    OllamaChat::Chat.new(
-      argv: %w[ -f lib/ollama_chat/ollama_chat_config/default_config.yml ]
-    )
+    OllamaChat::Chat.new(argv: chat_default_config)
   end
 
   connect_to_ollama_server

data/spec/ollama_chat/state_selectors_spec.rb

@@ -0,0 +1,193 @@
+require 'spec_helper'
+
+describe OllamaChat::StateSelectors::StateSelector do
+  let(:name) { 'Test Selector' }
+  let(:states) { %w[ enabled disabled low high ] }
+  let(:default) { 'enabled' }
+  let(:off) { %w[ disabled ] }
+
+  let(:selector) do
+    described_class.new(
+      name: name,
+      states: states,
+      default: default,
+      off: off
+    )
+  end
+
+  describe '#initialize' do
+    it 'creates a new StateSelector with provided parameters' do
+      expect(selector).to be_a described_class
+    end
+
+    it 'sets the name correctly' do
+      expect(selector.instance_variable_get(:@name)).to eq name
+    end
+
+    it 'sets the states correctly' do
+      expect(selector.instance_variable_get(:@states)).to eq Set.new(states)
+    end
+
+    it 'sets the default state' do
+      expect(selector.instance_variable_get(:@default)).to eq default
+    end
+
+    it 'sets the off states correctly' do
+      expect(selector.instance_variable_get(:@off)).to eq off
+    end
+
+    it 'sets the selected state to default' do
+      expect(selector.selected).to eq default
+    end
+
+    context 'with empty states' do
+      it 'raises ArgumentError when states are empty' do
+        expect {
+          described_class.new(name: 'Test', states: [])
+        }.to raise_error(ArgumentError, 'states cannot be empty')
+      end
+    end
+
+    context 'with invalid default' do
+      it 'raises ArgumentError when default is not in states' do
+        expect {
+          described_class.new(name: 'Test', states: %w[ a b ], default: 'c')
+        }.to raise_error(ArgumentError, 'default has to be one of a, b.')
+      end
+    end
+  end
+
+  describe '#selected' do
+    it 'returns the currently selected state' do
+      expect(selector.selected).to eq default
+    end
+  end
+
+  describe '#selected=' do
+    it 'sets the selected state to a valid value' do
+      selector.selected = 'low'
+      expect(selector.selected).to eq 'low'
+    end
+
+    it 'raises ArgumentError when setting invalid state' do
+      expect {
+        selector.selected = 'invalid'
+      }.to raise_error(ArgumentError, 'value has to be one of enabled, disabled, low, high.')
+    end
+  end
+
+  describe '#allow_empty?' do
+    it 'returns false by default' do
+      expect(selector.allow_empty?).to be false
+    end
+
+    context 'when allow_empty is true' do
+      let(:selector) do
+        described_class.new(
+          name: name,
+          states: states,
+          default: nil,
+          allow_empty: true
+        )
+      end
+
+      it 'returns true when allow_empty is set' do
+        expect(selector.allow_empty?).to be true
+      end
+    end
+  end
+
+  describe '#off?' do
+    it 'returns true when selected state is in off set' do
+      selector.selected = 'disabled'
+      expect(selector.off?).to be true
+    end
+
+    it 'returns false when selected state is not in off set' do
+      selector.selected = 'enabled'
+      expect(selector.off?).to be false
+    end
+  end
+
+  describe '#on?' do
+    it 'returns true when selected state is not in off set' do
+      selector.selected = 'enabled'
+      expect(selector.on?).to be true
+    end
+
+    it 'returns false when selected state is in off set' do
+      selector.selected = 'disabled'
+      expect(selector.on?).to be false
+    end
+  end
+
+  describe '#choose' do
+    it 'allows user to select a state from available options' do
+      # Mock the chooser to return a specific choice
+      expect(OllamaChat::Utils::Chooser).to receive(:choose).with(
+        %w[ enabled disabled low high [EXIT] ]
+      ).and_return('low')
+
+      selector.choose
+      expect(selector.selected).to eq 'low'
+    end
+
+    it 'exits when user selects [EXIT]' do
+      expect(OllamaChat::Utils::Chooser).to receive(:choose).and_return('[EXIT]')
+
+      selector.choose
+      expect(selector.selected).to eq default
+    end
+
+    it 'exits when user cancels selection' do
+      expect(OllamaChat::Utils::Chooser).to receive(:choose).and_return(nil)
+
+      selector.choose
+      expect(selector.selected).to eq default
+    end
+  end
+
+  describe '#show' do
+    it 'outputs the current state to stdout' do
+      expect(STDOUT).to receive(:puts).with(/Test Selector is .*?enabled.*?\./)
+      selector.show
+    end
+  end
+
+  describe '#to_s' do
+    it 'returns the string representation of the selected state' do
+      expect(selector.to_s).to eq default
+    end
+
+    it 'returns the string representation after changing state' do
+      selector.selected = 'low'
+      expect(selector.to_s).to eq 'low'
+    end
+  end
+
+  describe 'with allow_empty true' do
+    let(:selector) do
+      described_class.new(
+        name: 'Empty Selector',
+        states: %w[ a b c ],
+        default: nil,
+        allow_empty: true
+      )
+    end
+
+    it 'allows setting nil as selected state' do
+      selector.selected = nil
+      expect(selector.selected).to eq ""
+    end
+
+    it 'allows empty state when allow_empty is true' do
+      expect(selector.allow_empty?).to be true
+    end
+
+    it 'raises no error when setting invalid state' do
+      expect {
+        selector.selected = 'invalid'
+      }.not_to raise_error
+    end
+  end
+end