ollama_chat 0.0.3 → 0.0.5

data/docker-compose.yml

@@ -1,10 +1,31 @@
 services:
   redis:
+    container_name: redis
     image: valkey/valkey:7.2.8-alpine
     restart: unless-stopped
     ports: [ "127.0.0.1:9736:6379" ]
     volumes:
       - "redis-data:/data:delegated"
       - "./redis/redis.conf:/etc/redis.conf"
+  searxng:
+    container_name: searxng
+    image: searxng/searxng:latest
+    ports:
+      - "127.0.0.1:8088:8080"
+    restart: unless-stopped
+    cap_drop:
+      - ALL
+    cap_add:
+      - CHOWN
+      - SETGID
+      - SETUID
+      - DAC_OVERRIDE
+    logging:
+      driver: "json-file"
+      options:
+        max-size: "1m"
+        max-file: "1"
+    volumes:
+      - "./config/searxng:/etc/searxng"
 volumes:
   redis-data:

data/lib/ollama_chat/chat.rb

@@ -19,6 +19,7 @@ class OllamaChat::Chat
   include OllamaChat::ModelHandling
   include OllamaChat::Parsing
   include OllamaChat::SourceFetching
+  include OllamaChat::WebSearching
   include OllamaChat::Dialog
   include OllamaChat::Information
   include OllamaChat::Clipboard
@@ -45,14 +46,14 @@ class OllamaChat::Chat
     embedding_enabled.set(config.embedding.enabled && !@opts[?E])
     @messages        = OllamaChat::MessageList.new(self)
     if @opts[?c]
-      @messages.load_conversation(@opts[?c])
+      messages.load_conversation(@opts[?c])
     else
       default = config.system_prompts.default? || model_system
       if @opts[?s] =~ /\A\?/
         change_system_prompt(default, system: @opts[?s])
       else
         system = OllamaChat::Utils::FileArgument.get_file_argument(@opts[?s], default:)
-        system.present? and @messages.set_system_prompt(system)
+        system.present? and messages.set_system_prompt(system)
       end
     end
     @documents     = setup_documents
@@ -65,6 +66,8 @@ class OllamaChat::Chat
 
   attr_reader :documents
 
+  attr_reader :messages
+
   def links
     @links ||= Set.new
   end
@@ -83,8 +86,17 @@ class OllamaChat::Chat
 
   def start
     info
+    if messages.size > 1
+      messages.list_conversation(2)
+    end
     STDOUT.puts "\nType /help to display the chat help."
 
+    interact_with_user
+  end
+
+  private
+
+  def interact_with_user
     loop do
       parse_content = true
       input_prompt = bold { color(172) { message_type(@images) + " user" } } + bold { "> " }
@@ -114,15 +126,15 @@ class OllamaChat::Chat
         next
       when %r(^/list(?:\s+(\d*))?$)
         last = 2 * $1.to_i if $1
-        @messages.list_conversation(last)
+        messages.list_conversation(last)
         next
       when %r(^/clear$)
-        @messages.clear
+        messages.clear
         STDOUT.puts "Cleared messages."
         next
       when %r(^/clobber$)
         if ask?(prompt: 'Are you sure to clear messages and collection? (y/n) ') =~ /\Ay/i
-          @messages.clear
+          messages.clear
           @documents.clear
           links.clear
           STDOUT.puts "Cleared messages and collection #{bold{@documents.collection}}."
@@ -131,8 +143,8 @@ class OllamaChat::Chat
         end
         next
       when %r(^/drop(?:\s+(\d*))?$)
-        @messages.drop($1)
-        @messages.list_conversation(2)
+        messages.drop($1)
+        messages.list_conversation(2)
         next
       when %r(^/model$)
         @model = choose_model('', @model)
@@ -142,9 +154,9 @@ class OllamaChat::Chat
         info
         next
       when %r(^/regenerate$)
-        if content = @messages.second_last&.content
+        if content = messages.second_last&.content
           content.gsub!(/\nConsider these chunks for your answer.*\z/, '')
-          @messages.drop(2)
+          messages.drop(2)
         else
           STDOUT.puts "Not enough messages in this conversation."
           redo
@@ -201,7 +213,7 @@ class OllamaChat::Chat
         content       = embed($1) or next
       when %r(^/web\s+(?:(\d+)\s+)?(.+))
         parse_content   = false
-        urls            = search_web($2, $1.to_i)
+        urls            = search_web($2, $1.to_i) or next
         urls.each do |url|
           fetch_source(url) { |url_io| embed_source(url_io, url) }
         end
@@ -211,7 +223,7 @@ class OllamaChat::Chat
           map { |u, s| "%s as \n:%s" % [ u, s ] } * "\n\n"
         content = config.prompts.web % { query:, results: }
       when %r(^/save\s+(.+)$)
-        @messages.save_conversation($1)
+        messages.save_conversation($1)
         STDOUT.puts "Saved conversation to #$1."
         next
       when %r(^/links(?:\s+(clear))?$)
@@ -251,7 +263,10 @@ class OllamaChat::Chat
         end
         next
       when %r(^/load\s+(.+)$)
-        @messages.load_conversation($1)
+        messages.load_conversation($1)
+        if messages.size > 1
+          messages.list_conversation(2)
+        end
         STDOUT.puts "Loaded conversation from #$1."
         next
       when %r(^/config$)
@@ -302,16 +317,16 @@ class OllamaChat::Chat
         end
       end
 
-      @messages << Ollama::Message.new(role: 'user', content:, images: @images.dup)
+      messages << Ollama::Message.new(role: 'user', content:, images: @images.dup)
       @images.clear
       handler = OllamaChat::FollowChat.new(
         chat:     self,
-        messages: @messages,
+        messages:,
         voice:    (@current_voice if voice.on?)
       )
       ollama.chat(
         model: @model,
-        messages: @messages,
+        messages:,
         options: @model_options,
         stream: stream.on?,
         &handler
@@ -325,16 +340,16 @@ class OllamaChat::Chat
                  end
           [ link, record.tags.first ]
         }.uniq.map { |l, t| hyperlink(l, t) }.join(' ')
-        config.debug and jj @messages.to_ary
+        config.debug and jj messages.to_ary
       end
+    rescue Ollama::Errors::TimeoutError
+      STDOUT.puts "#{bold('Error')}: Currently lost connection to ollama server and cannot send command."
     rescue Interrupt
       STDOUT.puts "Type /quit to quit."
     end
     0
   end
 
-  private
-
   def setup_documents
     if embedding.on?
       @embedding_model         = config.embedding.model.name
@@ -345,6 +360,7 @@ class OllamaChat::Chat
         ollama:,
         model:             @embedding_model,
         model_options:     config.embedding.model.options,
+        embedding_length:  config.embedding.model.embedding_length,
         database_filename: config.embedding.database_filename || @ollama_chat_config.database_path,
         collection:        ,
         cache:             configure_cache,
@@ -392,7 +408,7 @@ class OllamaChat::Chat
       Documentrix::Documents::RedisCache.new(
         prefix: 'Expiring-',
         url:,
-        ex:     config.redis.expiring.ex,
+        ex:     config.redis.expiring.ex?.to_i,
       )
     end
   end

data/lib/ollama_chat/follow_chat.rb

@@ -52,7 +52,7 @@ class OllamaChat::FollowChat
       prompt_eval_rate:     bold { "%.2f c/s" % (response.prompt_eval_count.to_i / prompt_eval_duration) } + color(111),
       total_duration:       Tins::Duration.new(response.total_duration / 1e9),
       load_duration:        Tins::Duration.new(response.load_duration / 1e9),
-    }.map { _1 * '=' } * ' '
+    }.map { _1 * ?= } * ' '
     '📊 ' + color(111) {
       Kramdown::ANSI::Width.wrap(stats_text, percentage: 90).gsub(/(?<!\A)^/, '   ')
     }

data/lib/ollama_chat/information.rb

@@ -28,6 +28,7 @@ module OllamaChat::Information
   end
 
   def info
+    STDOUT.puts "Running ollama_chat version: #{bold(OllamaChat::VERSION)}"
     STDOUT.puts "Connected to ollama server version: #{bold(server_version)}"
     STDOUT.puts "Current model is #{bold{@model}}."
     if @model_options.present?
@@ -47,6 +48,7 @@ module OllamaChat::Information
     stream.show
     location.show
     STDOUT.puts "Document policy for references in user text: #{bold{@document_policy}}"
+    STDOUT.puts "Currently selected search engine is #{bold(search_engine)}."
     if @voice.on?
       STDOUT.puts "Using voice #{bold{@current_voice}} to speak."
     end

data/lib/ollama_chat/ollama_chat_config/default_config.yml

@@ -21,7 +21,7 @@ prompts:
     Answer the the query %{query} using these sources and summaries:
 
     %{results}
-  location: You are at %{location_name} (%{location_decimal_degrees}), on %{localtime}, preferring %{units}
+  location: You are at %{location_name}, %{location_decimal_degrees}, on %{localtime}, preferring %{units}
 system_prompts:
   default: <%= ENV.fetch('OLLAMA_CHAT_SYSTEM', 'null') %>
   assistant: You are a helpful assistant.
@@ -60,3 +60,10 @@ request_headers:
   Accept: 'text/*,application/*,image/*'
 ssl_no_verify: []
 copy: pbcopy
+web_search:
+  use: duckduckgo
+  engines:
+    duckduckgo:
+      url: 'https://www.duckduckgo.com/html/?q=%{query}'
+    searxng:
+      url: <%= ENV.fetch('OLLAMA_SEARXNG_URL', 'http://localhost:8088/search?q=%{query}&language=en&format=json') %>

data/lib/ollama_chat/source_fetching.rb

@@ -141,37 +141,4 @@ module OllamaChat::SourceFetching
       summarize(source)
     end
   end
-
-  def search_web(query, n = nil)
-    if l = @messages.at_location.full?
-      query += " #{l}"
-    end
-    n = n.to_i.clamp(1..)
-    query = URI.encode_uri_component(query)
-    url = "https://www.duckduckgo.com/html/?q=#{query}"
-    OllamaChat::Utils::Fetcher.get(
-      url,
-      headers: config.request_headers?.to_h,
-      debug:   config.debug
-    ) do |tmp|
-      result = []
-      doc = Nokogiri::HTML(tmp)
-      doc.css('.results_links').each do |link|
-        if n > 0
-          url = link.css('.result__a').first&.[]('href')
-          url.sub!(%r(\A(//duckduckgo\.com)?/l/\?uddg=), '')
-          url.sub!(%r(&rut=.*), '')
-          url = URI.decode_uri_component(url)
-          url = URI.parse(url)
-          url.host =~ /duckduckgo\.com/ and next
-          links.add(url.to_s)
-          result << url
-          n -= 1
-        else
-          break
-        end
-      end
-      result
-    end
-  end
 end

data/lib/ollama_chat/version.rb

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

data/lib/ollama_chat/web_searching.rb

@@ -0,0 +1,60 @@
+module OllamaChat::WebSearching
+  def search_web(query, n = nil)
+    l     = @messages.at_location.full? and query += " #{l}"
+    n     = n.to_i.clamp(1..)
+    query = URI.encode_uri_component(query)
+    search_command = :"search_web_with_#{search_engine}"
+    if respond_to?(search_command, true)
+      send(search_command, query, n)
+    else
+      STDOUT.puts "Search engine #{bold{search_engine}} not implemented!"
+      nil
+    end
+  end
+
+  private
+
+  def search_engine
+    config.web_search.use
+  end
+
+  def search_web_with_searxng(query, n)
+    url = config.web_search.engines.searxng.url % { query: }
+    OllamaChat::Utils::Fetcher.get(
+      url,
+      headers: config.request_headers?.to_h,
+      debug:   config.debug
+    ) do |tmp|
+      data = JSON.parse(tmp.read, object_class: JSON::GenericObject)
+      data.results.first(n).map(&:url)
+    end
+  end
+
+  def search_web_with_duckduckgo(query, n)
+    url = config.web_search.engines.duckduckgo.url % { query: }
+    OllamaChat::Utils::Fetcher.get(
+      url,
+      headers: config.request_headers?.to_h,
+      debug:   config.debug
+    ) do |tmp|
+      result = []
+      doc = Nokogiri::HTML(tmp)
+      doc.css('.results_links').each do |link|
+        if n > 0
+          url = link.css('.result__a').first&.[]('href')
+          url.sub!(%r(\A(//duckduckgo\.com)?/l/\?uddg=), '')
+          url.sub!(%r(&rut=.*), '')
+          url = URI.decode_uri_component(url)
+          url = URI.parse(url)
+          url.host =~ /duckduckgo\.com/ and next
+          links.add(url.to_s)
+          result << url
+          n -= 1
+        else
+          break
+        end
+      end
+      result
+    end
+  end
+end

data/lib/ollama_chat.rb

@@ -13,6 +13,7 @@ require 'ollama_chat/message_list'
 require 'ollama_chat/model_handling'
 require 'ollama_chat/parsing'
 require 'ollama_chat/source_fetching'
+require 'ollama_chat/web_searching'
 require 'ollama_chat/dialog'
 require 'ollama_chat/information'
 require 'ollama_chat/clipboard'

data/ollama_chat.gemspec

@@ -1,30 +1,30 @@
 # -*- encoding: utf-8 -*-
-# stub: ollama_chat 0.0.3 ruby lib
+# stub: ollama_chat 0.0.5 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "ollama_chat".freeze
-  s.version = "0.0.3".freeze
+  s.version = "0.0.5".freeze
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0".freeze) if s.respond_to? :required_rubygems_version=
   s.require_paths = ["lib".freeze]
   s.authors = ["Florian Frank".freeze]
-  s.date = "2025-02-17"
+  s.date = "2025-03-22"
   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]
-  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/information.rb".freeze, "lib/ollama_chat/message_list.rb".freeze, "lib/ollama_chat/message_type.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/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]
-  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, "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/information.rb".freeze, "lib/ollama_chat/message_list.rb".freeze, "lib/ollama_chat/message_type.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/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, "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/ollama_chat/chat_spec.rb".freeze, "spec/ollama_chat/clipboard_spec.rb".freeze, "spec/ollama_chat/follow_chat_spec.rb".freeze, "spec/ollama_chat/information_spec.rb".freeze, "spec/ollama_chat/message_list_spec.rb".freeze, "spec/ollama_chat/model_handling_spec.rb".freeze, "spec/ollama_chat/parsing_spec.rb".freeze, "spec/ollama_chat/source_fetching_spec.rb".freeze, "spec/ollama_chat/switches_spec.rb".freeze, "spec/ollama_chat/utils/cache_fetcher_spec.rb".freeze, "spec/ollama_chat/utils/fetcher_spec.rb".freeze, "spec/ollama_chat/utils/file_argument_spec.rb".freeze, "spec/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/dialog.rb".freeze, "lib/ollama_chat/document_cache.rb".freeze, "lib/ollama_chat/follow_chat.rb".freeze, "lib/ollama_chat/information.rb".freeze, "lib/ollama_chat/message_list.rb".freeze, "lib/ollama_chat/message_type.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/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/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, "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/information.rb".freeze, "lib/ollama_chat/message_list.rb".freeze, "lib/ollama_chat/message_type.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/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/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/model_handling_spec.rb".freeze, "spec/ollama_chat/parsing_spec.rb".freeze, "spec/ollama_chat/source_fetching_spec.rb".freeze, "spec/ollama_chat/switches_spec.rb".freeze, "spec/ollama_chat/utils/cache_fetcher_spec.rb".freeze, "spec/ollama_chat/utils/fetcher_spec.rb".freeze, "spec/ollama_chat/utils/file_argument_spec.rb".freeze, "spec/ollama_chat/web_searching_spec.rb".freeze, "spec/spec_helper.rb".freeze, "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.2".freeze
   s.summary = "A command-line interface (CLI) for interacting with an Ollama AI model.".freeze
-  s.test_files = ["spec/assets/example.rb".freeze, "spec/ollama_chat/chat_spec.rb".freeze, "spec/ollama_chat/clipboard_spec.rb".freeze, "spec/ollama_chat/follow_chat_spec.rb".freeze, "spec/ollama_chat/information_spec.rb".freeze, "spec/ollama_chat/message_list_spec.rb".freeze, "spec/ollama_chat/model_handling_spec.rb".freeze, "spec/ollama_chat/parsing_spec.rb".freeze, "spec/ollama_chat/source_fetching_spec.rb".freeze, "spec/ollama_chat/switches_spec.rb".freeze, "spec/ollama_chat/utils/cache_fetcher_spec.rb".freeze, "spec/ollama_chat/utils/fetcher_spec.rb".freeze, "spec/ollama_chat/utils/file_argument_spec.rb".freeze, "spec/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/message_list_spec.rb".freeze, "spec/ollama_chat/model_handling_spec.rb".freeze, "spec/ollama_chat/parsing_spec.rb".freeze, "spec/ollama_chat/source_fetching_spec.rb".freeze, "spec/ollama_chat/switches_spec.rb".freeze, "spec/ollama_chat/utils/cache_fetcher_spec.rb".freeze, "spec/ollama_chat/utils/fetcher_spec.rb".freeze, "spec/ollama_chat/utils/file_argument_spec.rb".freeze, "spec/ollama_chat/web_searching_spec.rb".freeze, "spec/spec_helper.rb".freeze]
 
   s.specification_version = 4
 
-  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 1.19".freeze])
+  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 1.20".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])

data/spec/assets/searxng.json

@@ -0,0 +1,111 @@
+{
+  "query": "foo",
+  "number_of_results": 0,
+  "results": [
+    {
+      "url": "https://en.wikipedia.org/wiki/Foo_Fighters",
+      "title": "Foo Fighters - Wikipedia",
+      "content": "The Foo Fighters are an American rock band formed in Seattle in 1994. Initially founded as a one-man project by former Nirvana drummer Dave Grohl, the band comprises vocalist/guitarist Grohl, bassist Nate Mendel, guitarist Pat Smear, guitarist Chris Shiflett, keyboardist Rami Jaffee and drummer Josh Freese.Guitarist Franz Stahl and drummers William Goldsmith and Taylor Hawkins are former ...",
+      "publishedDate": null,
+      "thumbnail": null,
+      "engine": "brave",
+      "template": "default.html",
+      "parsed_url": [
+        "https",
+        "en.wikipedia.org",
+        "/wiki/Foo_Fighters",
+        "",
+        "",
+        ""
+      ],
+      "engines": [
+        "startpage",
+        "brave",
+        "google",
+        "duckduckgo"
+      ],
+      "positions": [
+        2,
+        1,
+        2,
+        1
+      ],
+      "score": 12.0,
+      "category": "general"
+    },
+    {
+      "url": "https://www.foofighters.com",
+      "title": "Home - Foo Fighters",
+      "content": "Official website of Foo Fighters",
+      "publishedDate": "2023-04-19T00:00:00",
+      "thumbnail": null,
+      "engine": "brave",
+      "template": "default.html",
+      "parsed_url": [
+        "https",
+        "www.foofighters.com",
+        "",
+        "",
+        "",
+        ""
+      ],
+      "engines": [
+        "startpage",
+        "brave",
+        "google",
+        "duckduckgo"
+      ],
+      "positions": [
+        1,
+        2,
+        7,
+        2
+      ],
+      "score": 8.571428571428571,
+      "category": "general"
+    },
+    {
+      "url": "https://www.dictionary.com/e/tech-science/foo/",
+      "title": "foo | Meaning & Origin | Dictionary.com",
+      "content": "\"Foo was here\" was a popular piece of graffiti drawn by Australian soldiers in WWII, and possibly even WWI, that depicts a little man poking his head and large nose over the wall à la Kilroy. While the origins of this foo are unclear, it appears to be unrelated to Holman's foo.",
+      "publishedDate": "2022-07-04T00:00:00",
+      "thumbnail": "",
+      "engine": "brave",
+      "template": "default.html",
+      "parsed_url": [
+        "https",
+        "www.dictionary.com",
+        "/e/tech-science/foo/",
+        "",
+        "",
+        ""
+      ],
+      "engines": [
+        "brave",
+        "duckduckgo"
+      ],
+      "positions": [
+        13,
+        1
+      ],
+      "score": 2.1538461538461537,
+      "category": "general"
+    }
+  ],
+  "suggestions": [
+    "foo fighters - everlong",
+    "Foo meaning",
+    "Foo Fighters WW2",
+    "Foo Fighters - Learn to Fly",
+    "foo fighters - the colour and the shape",
+    "Foo Fighters tour 2024",
+    "Foo programming",
+    "Foo word"
+  ],
+  "unresponsive_engines": [
+    [
+      "qwant",
+      "Suspended: CAPTCHA"
+    ]
+  ]
+}

data/spec/ollama_chat/chat_spec.rb

@@ -15,6 +15,19 @@ RSpec.describe OllamaChat::Chat do
     expect(chat).to be_a described_class
   end
 
+  context 'loading conversations' do
+    let :argv do
+      %w[ -C test -c ] << asset('conversation.json')
+    end
+
+    it 'dispays the last exchange of the converstation' do
+      expect(chat).to receive(:interact_with_user).and_return 0
+      expect(STDOUT).to receive(:puts).at_least(1)
+      expect(chat.messages).to receive(:list_conversation)
+      chat.start
+    end
+  end
+
   describe OllamaChat::DocumentCache do
     context 'with MemoryCache' do
       let :argv do
@@ -71,6 +84,7 @@ RSpec.describe OllamaChat::Chat do
       expect(STDOUT).to receive(:puts).
         with(
           /
+            Running\ ollama_chat\ version|
             Connected\ to\ ollama\ server|
             Current\ model|
             Options|
@@ -80,7 +94,8 @@ RSpec.describe OllamaChat::Chat do
             output\ content|
             Streaming|
             Location|
-            Document\ policy
+            Document\ policy|
+            Currently\ selected\ search\ engine
           /x
         ).at_least(1)
       expect(chat.info).to be_nil

data/spec/ollama_chat/source_fetching_spec.rb

@@ -36,13 +36,4 @@ RSpec.describe OllamaChat::SourceFetching do
       'This source was now embedded: ./spec/assets/example.html'
     )
   end
-
-  it 'can search web' do
-    stub_request(:get, "https://www.duckduckgo.com/html/?q=foo").
-      with(headers: { 'Host'=>'www.duckduckgo.com' }).
-      to_return(status: 200, body: asset_content('duckduckgo.html'), headers: {})
-    expect(chat.search_web('foo').first.to_s).to eq(
-      'https://en.wikipedia.org/wiki/Foo_Fighters'
-    )
-  end
 end

data/spec/ollama_chat/web_searching_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+
+RSpec.describe OllamaChat::WebSearching do
+  let :chat do
+    OllamaChat::Chat.new
+  end
+
+  connect_to_ollama_server
+
+  it 'can search web with duckduckgo' do
+    url = 'https://www.duckduckgo.com/html/?q=foo'
+    chat.config.web_search.engines.duckduckgo.attributes_update(url:)
+    expect(chat).to receive(:search_engine).and_return 'duckduckgo'
+    stub_request(:get, url).
+      with(headers: { 'Host'=> 'www.duckduckgo.com' }).
+      to_return(status: 200, body: asset_content('duckduckgo.html'), headers: {})
+    expect(chat.search_web('foo').first.to_s).to eq(
+      'https://en.wikipedia.org/wiki/Foo_Fighters'
+    )
+  end
+
+  it 'can search web with searxng' do
+    url = 'http://localhost:8088/search?format=json&language=en&q=foo'
+    chat.config.web_search.engines.searxng.attributes_update(url:)
+    expect(chat).to receive(:search_engine).and_return 'searxng'
+    stub_request(:get, url).
+      with(headers: { 'Host'=>'localhost:8088' }).
+      to_return(status: 200, body: asset_content('searxng.json'), headers: {})
+    expect(chat.search_web('foo').first.to_s).to eq(
+      'https://en.wikipedia.org/wiki/Foo_Fighters'
+    )
+  end
+end

data/spec/spec_helper.rb

@@ -14,6 +14,8 @@ require 'webmock/rspec'
 WebMock.disable_net_connect!
 require 'ollama_chat'
 
+ComplexConfig::Provider.deep_freeze = false
+
 def asset(name)
   File.join(__dir__, 'assets', name)
 end