ollama_chat 0.0.56 → 0.0.57

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 683c0bf5fdb8756030c0c5b69767d95f8ae852094a0048cb61b4cf7166f6a045
-  data.tar.gz: 8e7f9022cfd1e9c32f987b76196222d346100cd5ceaa6d59e6dc5c30257e5dbb
+  metadata.gz: '0812722ef6dcc9b968a058ad3e4d96a8944ba61aaf347e39b064881e67704c3c'
+  data.tar.gz: 03ceba63e04403a982eac23c0a1d5bbf9ff50512f83a721a2def8c5ed8ebec46
 SHA512:
-  metadata.gz: 854af0ea07ced05e3f2f36741dd98592a2f6968e3307662e10c14bcb9a10b2168a1dde962de32f44e73719d177547cbccd8eee76c0554878a0de7ce2e1811772
-  data.tar.gz: 86b71f7b0748be19d0f772213b1a21ea8c7f838296bcd48348076ef39372e0a0f07db20c59a1c1b56f8a7592d100993a7ffa6b6507b4febbbca2977f0375baff
+  metadata.gz: 14949a812f2040b5f4ec0fb767abcb8e41e8ff2f64af2502cc9278d8245f9794e1d7f9bcab0b1b99214c80aa135b1c9b68551a5d208356fa521f3fb36bb8e7f3
+  data.tar.gz: 44a2dfcccaa933286b5c0a0a93d6916e4bfd8e2af516d16d3524b4c26ffb76a59fd50557aef8262416fd00e7cb88e12d73fab653a6fda3c470f23f9be937d767

data/CHANGES.md

@@ -1,5 +1,38 @@
 # Changes
 
+## 2026-01-21 v0.0.57
+
+- Introduce `OllamaChat::StateSelectors` module with `StateSelector` class for
+  managing configurable states
+- Replace simple string-based document policy and think mode with
+  `StateSelector` objects
+- Add `allow_empty` parameter to `StateSelector#initialize` method to allow
+  empty states in voice output
+- Update `StateSelector#selected=` to conditionally validate states based on
+  `allow_empty?`
+- Refactor voice handling to use `StateSelector` by replacing `@current_voice`
+  with `@voices` `StateSelector`
+- Update `FollowChat` to use `@voices.selected` instead of `@current_voice` for
+  voice selection
+- Simplify `change_voice` method in `dialog.rb` to delegate to `@voices.choose`
+- Update voice display in `information.rb` to use `@voices.show` instead of raw
+  voice name
+- Update configuration format to support nested `think` settings with `mode`
+  and `loud` sub-properties
+- Modify command handlers to use `document_policy.choose` and
+  `think_mode.choose` instead of legacy methods
+- Update `OllamaChat::Chat` initialization to use `setup_state_selectors`
+  method
+- Refactor `OllamaChat::ThinkControl` to use new state selector system
+- Update `OllamaChat::Parsing` to reference `@document_policy.selected` instead
+  of `@document_policy`
+- Update default configuration file to use format with nested think settings
+- Add proper `attr_reader` for `document_policy` and `think_mode` state
+  selectors
+- Update help text to reference new state selector system
+- Update `OllamaChat::Switches` to handle nested think configuration
+- Add `OllamaChat::StateSelectors` to required files in `lib/ollama_chat.rb`
+
 ## 2026-01-17 v0.0.56
 
 - Updated `context_spook` dependency from version **1.4** to **1.5**

data/lib/ollama_chat/chat.rb

@@ -36,6 +36,7 @@ class OllamaChat::Chat
   include Term::ANSIColor
   include OllamaChat::DocumentCache
   include OllamaChat::Switches
+  include OllamaChat::StateSelectors
   include OllamaChat::ModelHandling
   include OllamaChat::Parsing
   include OllamaChat::SourceFetching
@@ -88,14 +89,13 @@ class OllamaChat::Chat
     @ollama_chat_config = OllamaChat::OllamaChatConfig.new(@opts[?f])
     self.config         = @ollama_chat_config.config
     setup_switches(config)
+    setup_state_selectors(config)
     @ollama = connect_ollama
     if server_version.version < '0.9.0'.version
       raise ArgumentError, 'require ollama API version 0.9.0 or higher'
     end
-    @document_policy = config.document_policy
     @model           = choose_model(@opts[?m], config.model.name)
     @model_options   = Ollama::Options[config.model.options]
-    @think           = config.think
     model_system     = pull_model_unless_present(@model, @model_options)
     embedding_enabled.set(config.embedding.enabled && !@opts[?E])
     if @opts[?c]
@@ -111,7 +111,6 @@ class OllamaChat::Chat
     end
     @documents            = setup_documents
     @cache                = setup_cache
-    @current_voice        = config.voice.default
     @images               = []
     @kramdown_ansi_styles = configure_kramdown_ansi_styles
     init_chat_history
@@ -120,6 +119,18 @@ class OllamaChat::Chat
     fix_config(e)
   end
 
+  # The document_policy reader returns the document policy selector for the chat session.
+  #
+  # @return [ OllamaChat::StateSelector ] the document policy selector object
+  #   that manages the policy for handling document references in user text
+  attr_reader :document_policy
+
+  # The think_mode reader returns the think mode selector for the chat session.
+  #
+  # @return [ OllamaChat::StateSelector ] the think mode selector object
+  #   that manages the thinking mode setting for the Ollama model interactions
+  attr_reader :think_mode
+
   # The debug method accesses the debug configuration setting.
   #
   # @return [TrueClass, FalseClass] the current debug mode status
@@ -304,10 +315,10 @@ class OllamaChat::Chat
       info
       :next
     when %r(^/document_policy$)
-      choose_document_policy
+      document_policy.choose
       :next
     when %r(^/think$)
-      choose_think_mode
+      think_mode.choose
       :next
     when %r(^/think_loud$)
       think_loud.toggle
@@ -385,9 +396,9 @@ class OllamaChat::Chat
   # the specified number of URLs. The processing approach varies based on the current
   # document policy and embedding status:
   #
-  # - **Embedding mode**: When `@document_policy == 'embedding'` AND `@embedding.on?` is true,
+  # - **Embedding mode**: When `document_policy.selected == 'embedding'` AND `@embedding.on?` is true,
   #   each result is embedded and the query is interpolated into the `web_embed` prompt.
-  # - **Summarizing mode**: When `@document_policy == 'summarizing'`,
+  # - **Summarizing mode**: When `document_policy.selected == 'summarizing'`,
   #   each result is summarized and both query and results are interpolated into the
   #   `web_summarize` prompt.
   # - **Importing mode**: For all other cases, each result is imported and both query and
@@ -403,11 +414,11 @@ class OllamaChat::Chat
   #   web('3', 'ruby programming tutorials')
   #
   # @example Web search with embedding policy
-  #   # With @document_policy == 'embedding' and @embedding.on?
+  #   # With document_policy.selected == 'embedding' and @embedding.on?
   #   # Processes results through embedding pipeline
   #
   # @example Web search with summarizing policy
-  #   # With @document_policy == 'summarizing'
+  #   # With document_policy.selected == 'summarizing'
   #   # Processes results through summarization pipeline
   #
   # @see #search_web
@@ -417,13 +428,13 @@ class OllamaChat::Chat
   # @see #summarize
   def web(count, query)
     urls = search_web(query, count.to_i) or return :next
-    if @document_policy == 'embedding' && @embedding.on?
+    if document_policy.selected == 'embedding' && @embedding.on?
       prompt = config.prompts.web_embed
       urls.each do |url|
         fetch_source(url) { |url_io| embed_source(url_io, url) }
       end
       prompt.named_placeholders_interpolate({query:})
-    elsif @document_policy == 'summarizing'
+    elsif document_policy.selected == 'summarizing'
       prompt = config.prompts.web_import
       results = urls.each_with_object('') do |url, content|
         summarize(url).full? do |c|
@@ -617,7 +628,7 @@ class OllamaChat::Chat
       handler = OllamaChat::FollowChat.new(
         chat:     self,
         messages:,
-        voice:    (@current_voice if voice.on?)
+        voice:    (@voices.selected if voice.on?)
       )
       begin
         retried = false
@@ -633,7 +644,7 @@ class OllamaChat::Chat
         if think? && !retried
           STDOUT.puts "#{bold('Error')}: in think mode, switch thinking off and retry."
           sleep 1
-          @think  = false
+          think_mode.selected  = 'disabled'
           retried = true
           retry
         else

data/lib/ollama_chat/dialog.rb

@@ -12,9 +12,6 @@
 #
 # @example Changing the system prompt
 #   chat.change_system_prompt('default_prompt', system: '?sherlock')
-#
-# @example Choosing a document policy
-#   chat.choose_document_policy
 module OllamaChat::Dialog
   # The model_with_size method formats a model's size for display
   # by creating a formatted string that includes the model name and its size
@@ -104,41 +101,6 @@ module OllamaChat::Dialog
     info
   end
 
-  # The document_policy method sets the policy for handling document imports.
-  #
-  # @param value [ String ] the document policy to be set
-  attr_writer :document_policy
-
-  # The choose_document_policy method presents a menu to select a document policy.
-  # It allows the user to choose from importing, embedding, summarizing, or
-  # ignoring documents.
-  # The method displays available policies and sets the selected policy as the
-  # current document policy.
-  # If no valid policy is found, it defaults to the first option.
-  # After selection, it outputs the chosen policy and displays the current
-  # configuration information.
-  def choose_document_policy
-    policies = %w[ importing embedding summarizing ignoring ].sort
-    current  = if policies.index(@document_policy)
-                 @document_policy
-               elsif policies.index(config.document_policy)
-                 config.document_policy
-               else
-                 policies.first
-               end
-    policies.unshift('[EXIT]')
-    policy = OllamaChat::Utils::Chooser.choose(policies)
-    case policy
-    when nil, '[EXIT]'
-      STDOUT.puts "Exiting chooser."
-      policy = current
-    end
-    self.document_policy = policy
-  ensure
-    STDOUT.puts "Using document policy #{bold{@document_policy}}."
-    info
-  end
-
   # The change_system_prompt method allows the user to select or enter a new
   # system prompt for the chat session.
   # It provides an interactive chooser when multiple prompts match the given
@@ -204,8 +166,7 @@ module OllamaChat::Dialog
   #
   # @return [ String ] the full name of the chosen voice
   def change_voice
-    chosen  = OllamaChat::Utils::Chooser.choose(config.voice.list)
-    @current_voice = chosen.full? || config.voice.default
+    @voices.choose
   end
 
   # The message_list method creates and returns a new MessageList instance

data/lib/ollama_chat/env_config.rb

@@ -2,6 +2,15 @@ require 'const_conf'
 require 'pathname'
 
 module OllamaChat
+  # Environment configuration module for OllamaChat
+  #
+  # This module provides a structured way to manage environment variables and
+  # configuration settings for the OllamaChat application. It uses the
+  # ConstConf library to define and manage configuration parameters with
+  # default values, descriptions, and decoding logic.
+  #
+  # The module organizes configuration into logical sections including general
+  # settings, Ollama-specific configurations, and chat-specific options.
   module EnvConfig
     include ConstConf
 

data/lib/ollama_chat/information.rb

@@ -99,15 +99,13 @@ module OllamaChat::Information
     end
     markdown.show
     stream.show
-    think_show
+    think_mode.show
     think_loud.show
     location.show
     voice.show
-    if @voice.on?
-      STDOUT.puts "  Using voice #{bold{@current_voice}} to speak."
-    end
+    @voice.on? and @voices.show
     STDOUT.puts "Documents database cache is #{@documents.nil? ? 'n/a' : bold{@documents.cache.class}}"
-    STDOUT.puts "Document policy for references in user text: #{bold{@document_policy}}"
+    STDOUT.puts "Document policy for references in user text: #{bold{document_policy}}"
     STDOUT.puts "Currently selected search engine is #{bold(search_engine)}."
     STDOUT.puts "Conversation length: #{bold(@messages.size.to_s)} message(s)."
     nil
@@ -144,7 +142,7 @@ module OllamaChat::Information
       /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: #@document_policy)
+      /web [n] query                  query web & for n(=1) results (policy: #{document_policy})
       /links [clear]                  display (or clear) links used in the chat
       /save filename                  store conversation messages
       /load filename                  load conversation messages

data/lib/ollama_chat/ollama_chat_config/default_config.yml

@@ -41,8 +41,9 @@ voice:
 markdown: true
 stream: true
 document_policy: importing
-think: false
-think_loud: true
+think:
+  mode: disabled
+  loud: true
 context:
   format: JSON
 embedding:

data/lib/ollama_chat/parsing.rb

@@ -245,7 +245,7 @@ module OllamaChat::Parsing
         when 'image'
           add_image(images, source_io, source)
         when 'text', 'application', nil
-          case @document_policy
+          case document_policy.selected
           when 'ignoring'
             nil
           when 'importing'

data/lib/ollama_chat/state_selectors.rb

@@ -0,0 +1,146 @@
+# A module that provides state selection functionality for OllamaChat.
+#
+# The StateSelectors module encapsulates the StateSelector class, which manages
+# configurable states with selection and display capabilities. It is used to
+# handle various settings in the chat application such as document policies and
+# think modes, allowing users to dynamically configure
+# different aspects of the chat session behavior.
+module OllamaChat::StateSelectors
+  # A state selector that manages configurable states with selection and
+  # display capabilities.
+  class StateSelector
+    include Term::ANSIColor
+
+    # Initializes a new StateSelector with the given configuration.
+    #
+    # @param name [String] The name of the state selector for display purposes
+    # @param states [Array<String>] The list of valid states this selector can have
+    # @param default [String, nil] The default state to select (must be one of +states+)
+    # @param off [Array<String>, nil] The list of states that should be considered "off"
+    # @raise [ArgumentError] If +states+ is empty or +default+ is not in +states+
+    def initialize(name:, states:, default: nil, off: nil, allow_empty: false)
+      @name        = name.to_s
+      @states      = Set.new(states.map(&:to_s))
+      @allow_empty = allow_empty
+      unless allow_empty
+        @states.empty? and raise ArgumentError, 'states cannot be empty'
+      end
+      if default
+        @default  = default.to_s
+        unless allow_empty?
+          @states.member?(@default) or raise ArgumentError,
+            "default has to be one of #{@states.to_a * ', '}."
+        end
+        @selected = @default
+      else
+        @selected = @states.first
+      end
+      @off = Array(off)
+    end
+
+    # The selected reader returns the currently selected state of the switch.
+    #
+    # @return [Object] the currently selected state value
+    attr_reader :selected
+
+    # The selected= method sets the selected state of the switch.
+    #
+    # @param value [Object] the value to be converted to a string and set as
+    #   the selected state
+    #
+    # @raise [ArgumentError] if the provided value is not one of the valid states
+    def selected=(value)
+      value = value.to_s
+      unless allow_empty?
+        @states.member?(value) or raise ArgumentError,
+          "value has to be one of #{@states.to_a * ', '}."
+      end
+      @selected = value
+    end
+
+    # The allow_empty? method checks if the switch is allowed to be empty.
+    #
+    # @return [ TrueClass, FalseClass ] true if the switch is allowed to be
+    #   empty, false otherwise
+    def allow_empty?
+      !!@allow_empty
+    end
+
+    # The off? method checks if the current state is in the off set.
+    #
+    # @return [ TrueClass, FalseClass ] true if the selected state is in the
+    #   off set, false otherwise
+    def off?
+      @off.member?(@selected)
+    end
+
+    # The on? method checks if the switch is in the on state, returning true if
+    # it is enabled and false if it is disabled.
+    #
+    # @return [ TrueClass, FalseClass ] true if the switch is on, false if it
+    #   is off
+    def on?
+      !off?
+    end
+
+    # The choose method presents a menu to select from available states.
+    #
+    # This method displays the available states to the user and allows them to
+    # select one. It handles the user's choice by updating the selected state
+    # or exiting the chooser if the user selects '[EXIT]' or cancels the selection.
+    #
+    # @return [ nil ] This method does not return a value; it updates the instance
+    # variable @selected based on user input.
+    def choose
+      states = @states + [ '[EXIT]' ]
+      case chosen = OllamaChat::Utils::Chooser.choose(states)
+      when '[EXIT]', nil
+        STDOUT.puts "Exiting chooser."
+      when
+        @selected = chosen
+      end
+    end
+
+    # The show method outputs the current value of the state selector.
+    #
+    # This method displays the name of the state selector along with its
+    # currently selected state in a formatted message to standard output.
+    def show
+      STDOUT.puts "#{@name} is #{bold(to_s)}."
+    end
+
+    # The to_s method returns the string representation of the selected state.
+    #
+    # @return [ String ] the string representation of the currently selected
+    #   state
+    def to_s
+      @selected.to_s
+    end
+  end
+
+  # Sets up state selectors for document policy and think mode based on the
+  # provided configuration.
+  #
+  # @param config [ComplexConfig::Settings] the configuration object containing
+  #   settings for document policy and think mode
+  def setup_state_selectors(config)
+    @document_policy = StateSelector.new(
+      name: 'Document policy',
+      default: config.document_policy,
+      states: %w[ embedding ignoring importing summarizing ],
+      off: %w[ ignoring ],
+    )
+    @think_mode = StateSelector.new(
+      name: 'Think mode',
+      default: config.think.mode,
+      states: %w[ enabled disabled low medium high ],
+      off: %w[ disabled ],
+    )
+    @voices = StateSelector.new(
+      name: 'Voice',
+      default: config.voice.default,
+      states: config.voice.list,
+      allow_empty: true
+    )
+  end
+end

data/lib/ollama_chat/switches.rb

@@ -192,7 +192,7 @@ module OllamaChat::Switches
     )
 
     @think_loud = Switch.new(
-      value: config.think_loud,
+      value: config.think.loud,
       msg: {
         true  => "Thinking out loud, show thinking annotations.",
         false => "Thinking silently, don't show thinking annotations.",

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.56'
+  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.56 ruby lib
+# stub: ollama_chat 0.0.57 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "ollama_chat".freeze
-  s.version = "0.0.56".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,15 +12,15 @@ 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.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
 

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
@@ -134,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
 
@@ -235,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
@@ -252,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
@@ -274,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,7 +2,7 @@ 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

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
 
@@ -233,14 +233,14 @@ describe OllamaChat::Parsing do
 
     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)
@@ -253,7 +253,7 @@ describe OllamaChat::Parsing do
       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),
@@ -264,7 +264,7 @@ describe OllamaChat::Parsing do
       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

data/spec/ollama_chat/think_control_spec.rb

@@ -2,153 +2,93 @@ require 'spec_helper'
 
 describe OllamaChat::ThinkControl 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
 
   describe '#think' do
-    it 'returns the current think mode state' do
+    it 'returns false when the think mode selector is off' do
+      chat.think_mode.selected = 'disabled'
       expect(chat.think).to be false
-      chat.instance_variable_set(:@think, true)
+    end
+
+    it 'returns true when the think mode selector is enabled' do
+      chat.think_mode.selected = 'enabled'
       expect(chat.think).to be true
-      chat.instance_variable_set(:@think, false)
-      expect(chat.think).to be false
-      chat.instance_variable_set(:@think, 'low')
+    end
+
+    it 'returns the selected value when it is not “enabled”' do
+      chat.think_mode.selected = 'low'
       expect(chat.think).to eq 'low'
     end
   end
 
   describe '#think?' do
-    it 'returns true when think mode is enabled (boolean true)' do
-      chat.instance_variable_set(:@think, true)
+    it 'returns true when the think mode selector is on' do
+      chat.think_mode.selected = 'enabled'
       expect(chat.think?).to be true
     end
 
-    it 'returns true when think mode is enabled (string value)' do
-      chat.instance_variable_set(:@think, 'high')
+    it 'returns true when the think mode selector is on but the value is a string' do
+      chat.think_mode.selected = 'high'
       expect(chat.think?).to be true
     end
 
-    it 'returns false when think mode is disabled (boolean false)' do
-      chat.instance_variable_set(:@think, false)
-      expect(chat.think?).to be false
-    end
-
-    it 'returns false when think mode is nil' do
-      chat.instance_variable_set(:@think, nil)
+    it 'returns false when the think mode selector is off' do
+      chat.think_mode.selected = 'disabled'
       expect(chat.think?).to be false
     end
   end
 
   describe '#think_mode' do
-    it 'returns "enabled" when think is true' do
-      chat.instance_variable_set(:@think, true)
-      expect(chat.think_mode).to eq 'enabled'
+    it 'exposes a StateSelector instance' do
+      expect(chat.think_mode).to be_a OllamaChat::StateSelectors::StateSelector
     end
 
-    it 'returns the think value when it is a string' do
-      chat.instance_variable_set(:@think, 'medium')
-      expect(chat.think_mode).to eq 'medium'
-    end
-
-    it 'returns "disabled" when think is false' do
-      chat.instance_variable_set(:@think, false)
-      expect(chat.think_mode).to eq 'disabled'
-    end
-
-    it 'returns "disabled" when think is nil' do
-      chat.instance_variable_set(:@think, nil)
-      expect(chat.think_mode).to eq 'disabled'
-    end
-  end
-
-  describe '#think_show' do
-    it 'displays the current think mode status' do
-      chat.instance_variable_set(:@think, true)
-      expect(STDOUT).to receive(:puts).with(/Think mode is \e\[1menabled\e\[0m\./)
-      chat.think_show
-    end
-
-    it 'displays the think mode level when set to string' do
-      chat.instance_variable_set(:@think, 'high')
-      expect(STDOUT).to receive(:puts).with(/Think mode is \e\[1mhigh\e\[0m\./)
-      chat.think_show
-    end
-
-    it 'displays "disabled" when think is false' do
-      chat.instance_variable_set(:@think, false)
-      expect(STDOUT).to receive(:puts).with(/Think mode is \e\[1mdisabled\e\[0m\./)
-      chat.think_show
-    end
-
-    it 'displays "disabled" when think is nil' do
-      chat.instance_variable_set(:@think, nil)
-      expect(STDOUT).to receive(:puts).with(/Think mode is \e\[1mdisabled\e\[0m\./)
-      chat.think_show
+    it 'has the correct default selection (disabled)' do
+      expect(chat.think_mode.selected).to eq 'disabled'
     end
   end
 
   describe '#think_loud?' do
-    it 'returns false when think is disabled' do
-      chat.instance_variable_set(:@think, false)
+    it 'returns false when the think mode selector is off' do
+      chat.think_mode.selected = 'disabled'
       expect(chat.think_loud?).to be false
     end
 
-    it 'returns false when think_loud is off' do
-      chat.instance_variable_set(:@think, true)
+    it 'returns false when the think_loud switch is off' do
+      chat.think_mode.selected = 'enabled'
       allow(chat).to receive(:think_loud).and_return(double(on?: false))
       expect(chat.think_loud?).to be false
     end
 
-    it 'returns true when both think and think_loud are enabled' do
-      chat.instance_variable_set(:@think, true)
+    it 'returns true when both the think mode selector and think_loud switch are on' do
+      chat.think_mode.selected = 'enabled'
       allow(chat).to receive(:think_loud).and_return(double(on?: true))
       expect(chat.think_loud?).to be true
     end
   end
 
-  describe '#choose_think_mode' do
-    it 'can select "off" mode' do
-      expect(OllamaChat::Utils::Chooser).to receive(:choose).and_return('off')
-      chat.choose_think_mode
-      expect(chat.think).to be false
+  describe '#think_mode.show' do
+    it 'prints the current think mode in bold' do
+      chat.think_mode.selected = 'high'
+      expect(STDOUT).to receive(:puts).with(/Think mode is \e\[1mhigh\e\[0m/)
+      chat.think_mode.show
     end
 
-    it 'can select "on" mode' do
-      expect(OllamaChat::Utils::Chooser).to receive(:choose).and_return('on')
-      chat.choose_think_mode
-      expect(chat.think).to be true
+    it 'prints “disabled” when the selector is off' do
+      chat.think_mode.selected = 'disabled'
+      expect(STDOUT).to receive(:puts).with(/Think mode is \e\[1mdisabled\e\[0m/)
+      chat.think_mode.show
     end
+  end
 
-    it 'can select "low" mode' do
+  describe '#think_mode.choose' do
+    it 'updates the selector based on the user choice' do
       expect(OllamaChat::Utils::Chooser).to receive(:choose).and_return('low')
-      chat.choose_think_mode
-      expect(chat.think).to eq 'low'
-    end
-
-    it 'can select "medium" mode' do
-      expect(OllamaChat::Utils::Chooser).to receive(:choose).and_return('medium')
-      chat.choose_think_mode
-      expect(chat.think).to eq 'medium'
-    end
-
-    it 'can select "high" mode' do
-      expect(OllamaChat::Utils::Chooser).to receive(:choose).and_return('high')
-      chat.choose_think_mode
-      expect(chat.think).to eq 'high'
-    end
-
-    it 'can exit selection' do
-      expect(OllamaChat::Utils::Chooser).to receive(:choose).and_return('[EXIT]')
-      expect { chat.choose_think_mode }.not_to change { chat.think }
-    end
-
-    it 'can handle nil selection' do
-      expect(OllamaChat::Utils::Chooser).to receive(:choose).and_return(nil)
-      expect { chat.choose_think_mode }.not_to change { chat.think }
+      chat.think_mode.choose
+      expect(chat.think_mode.selected).to eq 'low'
     end
   end
 end

data/spec/ollama_chat/web_searching_spec.rb

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

data/spec/spec_helper.rb

@@ -86,6 +86,10 @@ module AssetHelpers
   end
 end
 
+def chat_default_config(a = [])
+  a + %w[ -f lib/ollama_chat/ollama_chat_config/default_config.yml ]
+end
+
 # A module that provides functionality for stubbing Ollama server responses.
 #
 # The StubOllamaServer module enables developers to simulate Ollama API

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ollama_chat
 version: !ruby/object:Gem::Version
-  version: 0.0.56
+  version: 0.0.57
 platform: ruby
 authors:
 - Florian Frank
@@ -430,6 +430,7 @@ extra_rdoc_files:
 - lib/ollama_chat/redis_cache.rb
 - lib/ollama_chat/server_socket.rb
 - lib/ollama_chat/source_fetching.rb
+- lib/ollama_chat/state_selectors.rb
 - lib/ollama_chat/switches.rb
 - lib/ollama_chat/think_control.rb
 - lib/ollama_chat/utils.rb
@@ -473,6 +474,7 @@ files:
 - lib/ollama_chat/redis_cache.rb
 - lib/ollama_chat/server_socket.rb
 - lib/ollama_chat/source_fetching.rb
+- lib/ollama_chat/state_selectors.rb
 - lib/ollama_chat/switches.rb
 - lib/ollama_chat/think_control.rb
 - lib/ollama_chat/utils.rb
@@ -516,6 +518,7 @@ files:
 - spec/ollama_chat/redis_cache_spec.rb
 - spec/ollama_chat/server_socket_spec.rb
 - spec/ollama_chat/source_fetching_spec.rb
+- spec/ollama_chat/state_selectors_spec.rb
 - spec/ollama_chat/switches_spec.rb
 - spec/ollama_chat/think_control_spec.rb
 - spec/ollama_chat/utils/cache_fetcher_spec.rb
@@ -566,6 +569,7 @@ test_files:
 - spec/ollama_chat/redis_cache_spec.rb
 - spec/ollama_chat/server_socket_spec.rb
 - spec/ollama_chat/source_fetching_spec.rb
+- spec/ollama_chat/state_selectors_spec.rb
 - spec/ollama_chat/switches_spec.rb
 - spec/ollama_chat/think_control_spec.rb
 - spec/ollama_chat/utils/cache_fetcher_spec.rb