ollama-ruby 0.3.0 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 510c31683b2251118a7c3b469620400016b29fd1ea6f17cc86e4f99f62ecea2f
-  data.tar.gz: a00c4275e42002a01a99f874a855386c83e502551360c1f4146eee0b74f2fd08
+  metadata.gz: 9304634c783477787792071f039d6941222bcca0577cd59cdbc5103f820ea299
+  data.tar.gz: 9edc7eaf98c0601e08bb5cde23fc2d46b745982172caea349982d8867790166a
 SHA512:
-  metadata.gz: 17e50d40e4c24b56b5c2923f0b73f3e4294587b61a302fbaf4cd8f891e879eb435541e1f7e195a4f88352fd99aa80ff44c8577b8c195d7ef31af1c33229018b7
-  data.tar.gz: a18bcef82e9481b75fee4a4c88598a7a374c48f841e9870b39174c0d1f5f235f1173bdbff939182481d431e4e9c8c85501be4f5ef917ca02865692cfc39fde5a
+  metadata.gz: 0bda727610f0793d6c360a63c76b2c42e032109f34011ed7cbd54df5663102c3738498be836adcb00645e4ce407cab1f422737970b83a3fa757ea0b4d90d8453
+  data.tar.gz: b9df134561b2a44accfdc1ec6da68eb0b3b9b5ce64c2d64c744834a0b74cbfde8018e270977440224d0433c681e7a203d272453e391b85a3d32f34e3b0c790ca

data/CHANGES.md

@@ -1,5 +1,64 @@
 # Changes
 
+## 2024-09-15 v0.3.2
+
+* Add color support to chooser module:
+	+ Include `Term::ANSIColor` in `Ollama::Utils::Chooser` module
+    + Use `blue`, `on_blue` ANSI color for selected item in query method
+* Refactor summarize method to also import sources:
+	+ Added `content` variable to store result of `parse_source`
+	+ Replaced `or return` with explicit assignment and return
+	+ Added calls to `source_io.rewind` and `import_document`
+* Add new test for `file_argument_spec.rb`
+* Refactor tag list initialization and merging:
+	+ Use array literals for initializing tags lists
+	+ Use array literals for passing to merge method
+* Update dependencies and dates in Rakefile and gemspec:
+	+ Removed '.utilsrc' from ignored files in Rakefile
+	+ Updated date in `ollama-ruby.gemspec` to "2024-09-13"
+	+ Removed 'utils' development dependency from `ollama-ruby.gemspec`
+* Refactor `search_web` method to allow n parameter to be optional and default
+  to 1.
+
+## 2024-09-12 v0.3.1
+
+* Update dependencies and date in gemspec files:
+  - Updated `complex_config` dependency to '~> 0.22'
+* Refactor FollowChat#eval_stats to add bold eval rates
+  * Improve formatting in eval_stats using bold and color for better
+    readability.
+  * Update import_document and add_image methods to handle nil values
+    correctly.
+  * Update width method in utils/width.rb to use uncolor when checking line
+    length.
+* Refactor eval stats output in FollowChat class
+  - Add indentation to eval stats output for better readability
+* FollowChat evaluation stats refactored
+  - Removed hardcoded eval_stats hash and replaced with method call
+    `eval_stats(response)`
+  - Added new method `eval_stats(response)` to calculate evaluation statistics
+    - Calculates eval duration, prompt eval duration, total duration, and load
+      duration
+    - Adds eval count, prompt eval count, eval rate, and prompt eval rate
+* Use default to_s tree representation of config.
+  * Update complex_config dependency to ~> 0.21, >= 0.21.1 in Rakefile
+  * Update complex_config dependency to ~> 0.21, >= 0.21.1 in
+    ollama-ruby.gemspec
+* Update dependencies and configuration display
+  * Update 'complex_config' dependency to '~> 0.21'
+  * Change OllamaChatConfig to display configuration as a tree instead of yaml
+* Improve /web search command
+  * Update infobar dependency to ~> 0.8
+  * Update /web command to summarize web sources as well as importing them
+    directly
+  * Add /clobber command to clear conversation messages and collection
+* Refactor Ollama chat configuration and summary generation.
+  * Update `OllamaChatConfig` to use `prompts.system` instead of `system`.
+  * Introduce `prompts.summarize` config as template for generating abstract
+    summaries.
+  * Replace hardcoded summary generation with call to `prompts.summarize`.
+  * Display /help for all unknown chat commands starting wit `/`
+
 ## 2024-09-05 v0.3.0
 
 * **New Features**

data/README.md

@@ -156,6 +156,7 @@ The following commands can be given inside the chat, if prefixed by a `/`:
 /markdown                                toggle markdown output
 /list                                    list the messages of the conversation
 /clear                                   clear the conversation messages
+/clobber                                 clear conversation messages and collection
 /pop [n]                                 pop the last n exchanges, defaults to 1
 /model                                   change the model
 /regenerate                              the last answer message

data/Rakefile

@@ -15,7 +15,7 @@ GemHadar do
   ignore      '.*.sw[pon]', 'pkg', 'Gemfile.lock', '.AppleDouble', '.bundle',
     '.yardoc', 'tags', 'errors.lst', 'cscope.out', 'coverage', 'tmp', 'corpus'
   package_ignore '.all_images.yml', '.tool-versions', '.gitignore', 'VERSION',
-     '.utilsrc', '.rspec', *Dir.glob('.github/**/*', File::FNM_DOTMATCH)
+     '.rspec', *Dir.glob('.github/**/*', File::FNM_DOTMATCH)
   readme      'README.md'
 
   executables << 'ollama_console' << 'ollama_chat' <<
@@ -24,7 +24,7 @@ GemHadar do
   required_ruby_version  '~> 3.1'
 
   dependency             'excon',                 '~> 0.111'
-  dependency             'infobar',               '~> 0.7'
+  dependency             'infobar',               '~> 0.8'
   dependency             'term-ansicolor',        '~> 1.11'
   dependency             'kramdown-parser-gfm',   '~> 1.1'
   dependency             'terminal-table',        '~> 3.0'
@@ -34,13 +34,12 @@ GemHadar do
   dependency             'sorted_set',            '~> 1.0'
   dependency             'mime-types',            '~> 3.0'
   dependency             'reverse_markdown',      '~> 2.0'
-  dependency             'complex_config',        '~> 0.20'
+  dependency             'complex_config',        '~> 0.22'
   dependency             'search_ui',             '~> 0.0'
   dependency             'amatch',                '~> 0.4.1'
   dependency             'pdf-reader',            '~> 2.0'
   development_dependency 'all_images',            '~> 0.4'
   development_dependency 'rspec',                 '~> 3.2'
-  development_dependency 'utils'
   development_dependency 'webmock'
 
   licenses << 'MIT'

data/bin/ollama_chat

@@ -26,7 +26,12 @@ class OllamaChatConfig
       name: <%= ENV.fetch('OLLAMA_CHAT_MODEL', 'llama3.1') %>
       options:
         num_ctx: 8192
-    system: <%= ENV.fetch('OLLAMA_CHAT_SYSTEM', 'null') %>
+    prompts:
+      system: <%= ENV.fetch('OLLAMA_CHAT_SYSTEM', 'null') %>
+      summarize: |
+        Generate an abstract summary of the content in this document:
+
+        %s
     voice: Samantha
     markdown: true
     embedding:
@@ -50,7 +55,7 @@ class OllamaChatConfig
 
   def initialize(filename = nil)
     @filename = filename || default_path
-    @config = Provider.config(@filename)
+    @config = Provider.config(@filename, '⚙️')
     retried = false
   rescue ConfigurationFileMissing
     if @filename == default_path && !retried
@@ -115,22 +120,33 @@ class FollowChat
       @say.call(response)
     end
     if response.done
-      @output.puts
-      eval_stats = {
-        eval_duration:        Tins::Duration.new(response.eval_duration / 1e9),
-        eval_count:           response.eval_count,
-        prompt_eval_duration: Tins::Duration.new(response.prompt_eval_duration / 1e9),
-        prompt_eval_count:    response.prompt_eval_count,
-        total_duration:       Tins::Duration.new(response.total_duration / 1e9),
-        load_duration:        Tins::Duration.new(response.load_duration / 1e9),
-      }.map { _1 * '=' } * ' '
-      @output.puts '📊 ' + color(111) { Utils::Width.wrap(eval_stats, percentage: 90) }
+      @output.puts "", eval_stats(response)
     end
     self
   end
+
+  def eval_stats(response)
+    eval_duration = response.eval_duration / 1e9
+    prompt_eval_duration = response.prompt_eval_duration / 1e9
+    stats_text = {
+      eval_duration:        Tins::Duration.new(eval_duration),
+      eval_count:           response.eval_count,
+      eval_rate:            bold { "%.2f c/s" % (response.eval_count / eval_duration) } + color(111),
+      prompt_eval_duration: Tins::Duration.new(prompt_eval_duration),
+      prompt_eval_count:    response.prompt_eval_count,
+      prompt_eval_rate:     bold { "%.2f c/s" % (response.prompt_eval_count / 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 * '=' } * ' '
+    '📊 ' + color(111) {
+      Utils::Width.wrap(stats_text, percentage: 90).gsub(/(?<!\A)^/, '   ')
+    }
+  end
 end
 
-def search_web(query, n = 5)
+def search_web(query, n = nil)
+  n = n.to_i
+  n < 1 and n = 1
   query = URI.encode_uri_component(query)
   url = "https://www.duckduckgo.com/html/?q=#{query}"
   Ollama::Utils::Fetcher.new.get(url) do |tmp|
@@ -312,7 +328,7 @@ def import_document(source_io, source)
     STDOUT.puts "Embedding disabled, I won't import any documents, try: /summarize"
     return
   end
-  infobar.puts "Importing #{italic { source_io.content_type }} document #{source.to_s.inspect}."
+  puts "Importing #{italic { source_io&.content_type }} document #{source.to_s.inspect}."
   text = parse_source(source_io) or return
   text.downcase!
   splitter_config = $config.embedding.splitter
@@ -340,7 +356,7 @@ def import_document(source_io, source)
 end
 
 def add_image(images, source_io, source)
-  STDERR.puts "Adding #{source_io.content_type} image #{source.to_s.inspect}."
+  STDERR.puts "Adding #{source_io&.content_type} image #{source.to_s.inspect}."
   image = Image.for_io(source_io, path: source.to_s)
   (images << image).uniq!
 end
@@ -365,16 +381,15 @@ rescue => e
 end
 
 def summarize(source)
-  puts  "Now summarizing #{source.inspect}."
+  puts "Now summarizing #{source.to_s.inspect}."
   source_content =
     fetch_source(source) do |source_io|
-      parse_source(source_io) or return
+      content = parse_source(source_io) or return
+      source_io.rewind
+      import_document(source_io, source)
+      content
     end
-  <<~end
-    # Generate an abstract summary of the content in this document:
-
-    #{source_content}
-  end
+  $config.prompts.summarize % source_content
 end
 
 def parse_content(content, images)
@@ -459,6 +474,7 @@ def display_chat_help
     /markdown                                toggle markdown output
     /list                                    list the messages of the conversation
     /clear                                   clear the conversation messages
+    /clobber                                 clear conversation messages and collection
     /pop [n]                                 pop the last n exchanges, defaults to 1
     /model                                   change the model
     /regenerate                              the last answer message
@@ -501,8 +517,7 @@ $config = config.config
 
 opts[?h] and usage
 
-puts "Configuration read from #{config.filename.inspect} is:"
-y $config.to_h
+puts "Configuration read from #{config.filename.inspect} is:", $config
 
 base_url = opts[?u] || $config.url
 $ollama      = Client.new(base_url:, debug: $config.debug)
@@ -540,7 +555,7 @@ if $config.embedding.enabled
         File.expand_path(doc)
       end
     end
-    infobar.puts "Collection #{bold{collection}}: Adding #{document_list.size} documents…"
+    puts "Collection #{bold{collection}}: Adding #{document_list.size} documents…"
     document_list.each_slice(25) do |docs|
       docs.each do |doc|
         fetch_source(doc) do |doc_io|
@@ -563,7 +578,7 @@ if opts[?c]
   messages.concat load_conversation(opts[?c])
 else
   if system = Ollama::Utils::FileArgument.
-      get_file_argument(opts[?s], default: $config.system? || model_system)
+      get_file_argument(opts[?s], default: $config.prompts.system? || model_system)
     messages << Message.new(role: 'system', content: system)
     puts "Configured system prompt is:\n#{italic { system }}"
   end
@@ -584,7 +599,7 @@ loop do
   when %r(^/quit$)
     puts "Goodbye."
     exit 0
-  when %r(^/markdown)
+  when %r(^/markdown$)
     markdown = set_markdown(!markdown)
     next
   when %r(^/list$)
@@ -594,6 +609,11 @@ loop do
     messages.clear
     puts "Cleared messages."
     next
+  when %r(^/clobber$)
+    messages.clear
+    $documents.clear
+    puts "Cleared messages and collection."
+    next
   when %r(^/collection\s+(clear|stats|change|new)(?:\s+(.+))?$)
     command, arg = $1, $2
     case command
@@ -633,16 +653,24 @@ loop do
       puts "Not enough messages in this conversation."
       redo
     end
+    parse_content = false
+    content
   when %r(^/summarize\s+(.+))
     parse_content = false
     content       = summarize($1) or next
-  when %r(^/web\s+(?:(\d+)\s+)?(.+)$)
-    parse_content = true
-    urls = search_web($2, $1.to_i)
+  when %r(^/web\s+(?:(\d+)\s+)?(.+))
+    parse_content   = false
+    urls            = search_web($2, $1.to_i)
+    urls.each do |url|
+      fetch_source(url) do |url_io|
+        import_document(url_io, url)
+      end
+    end
+    urls_summarized = urls.map { summarize(_1) }
     content = <<~end
-      Answer the the query #{$2.inspect} using these sources:
+      Answer the the query #{$2.inspect} using these sources and summaries:
 
-      #{urls * ?\n}
+      #{urls.zip(urls_summarized).map { |u, s| "%s as \n:%s" % [ u, s ] } * "\n\n"}
     end
   when %r(^/save\s+(.+)$)
     save_conversation($1, messages)
@@ -652,12 +680,15 @@ loop do
     messages = load_conversation($1)
     puts "Loaded conversation from #$1."
     next
-  when %r(^/help$)
+  when %r(^/)
     display_chat_help
     next
-  when nil, ''
+  when ''
     puts "Type /quit to quit."
     next
+  when nil
+    puts "Goodbye."
+    exit 0
   end
 
   content, tags = if parse_content

data/lib/ollama/utils/chooser.rb

@@ -1,8 +1,10 @@
 require 'amatch'
 require 'search_ui'
+require 'term/ansicolor'
 
 module Ollama::Utils::Chooser
   include SearchUI
+  include Term::ANSIColor
 
   module_function
 
@@ -17,7 +19,7 @@ module Ollama::Utils::Chooser
       },
       query: -> _answer, matches, selector {
         matches.each_with_index.map { |m, i|
-          i == selector ? "#{Term::ANSIColor.blue{?⮕}} #{m.on_blue}" : "  #{m}"
+          i == selector ? "#{blue{?⮕}} #{on_blue{m}}" : "  #{m}"
         } * ?\n
       },
       found: -> _answer, matches, selector {

data/lib/ollama/utils/file_argument.rb

@@ -1,14 +1,14 @@
 module Ollama::Utils::FileArgument
   module_function
 
-  def get_file_argument(prompt, default: nil)
-    if prompt.present? && prompt.size < 2 ** 15 &&
-        File.basename(prompt).size < 2 ** 8 &&
-        File.exist?(prompt)
+  def get_file_argument(path_or_content, default: nil)
+    if path_or_content.present? && path_or_content.size < 2 ** 15 &&
+        File.basename(path_or_content).size < 2 ** 8 &&
+        File.exist?(path_or_content)
       then
-      File.read(prompt)
-    elsif prompt.present?
-      prompt
+      File.read(path_or_content)
+    elsif path_or_content.present?
+      path_or_content
     else
       default
     end

data/lib/ollama/utils/width.rb

@@ -1,6 +1,8 @@
 require 'tins/terminal'
 
 module Ollama::Utils::Width
+  include Term::ANSIColor
+
   module_function
 
   def width(percentage: 100.0)
@@ -12,7 +14,7 @@ module Ollama::Utils::Width
       raise ArgumentError, "either pass percentage or length argument"
     percentage and length ||= width(percentage:)
     text.gsub(/(?<!\n)\n(?!\n)/, ' ').lines.map do |line|
-      if length >= 1 && line.length > length
+      if length >= 1 && uncolor { line }.length > length
         line.gsub(/(.{1,#{length}})(\s+|$)/, "\\1\n").strip
       else
         line.strip

data/lib/ollama/version.rb

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

data/ollama-ruby.gemspec

@@ -1,36 +1,35 @@
 # -*- encoding: utf-8 -*-
-# stub: ollama-ruby 0.3.0 ruby lib
+# stub: ollama-ruby 0.3.2 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "ollama-ruby".freeze
-  s.version = "0.3.0".freeze
+  s.version = "0.3.2".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 = "2024-09-06"
+  s.date = "2024-09-15"
   s.description = "Library that allows interacting with the Ollama API".freeze
   s.email = "flori@ping.de".freeze
   s.executables = ["ollama_console".freeze, "ollama_chat".freeze, "ollama_update".freeze, "ollama_cli".freeze]
   s.extra_rdoc_files = ["README.md".freeze, "lib/ollama.rb".freeze, "lib/ollama/client.rb".freeze, "lib/ollama/client/command.rb".freeze, "lib/ollama/client/doc.rb".freeze, "lib/ollama/commands/chat.rb".freeze, "lib/ollama/commands/copy.rb".freeze, "lib/ollama/commands/create.rb".freeze, "lib/ollama/commands/delete.rb".freeze, "lib/ollama/commands/embed.rb".freeze, "lib/ollama/commands/embeddings.rb".freeze, "lib/ollama/commands/generate.rb".freeze, "lib/ollama/commands/ps.rb".freeze, "lib/ollama/commands/pull.rb".freeze, "lib/ollama/commands/push.rb".freeze, "lib/ollama/commands/show.rb".freeze, "lib/ollama/commands/tags.rb".freeze, "lib/ollama/documents.rb".freeze, "lib/ollama/documents/memory_cache.rb".freeze, "lib/ollama/documents/redis_cache.rb".freeze, "lib/ollama/documents/splitters/character.rb".freeze, "lib/ollama/documents/splitters/semantic.rb".freeze, "lib/ollama/dto.rb".freeze, "lib/ollama/errors.rb".freeze, "lib/ollama/handlers.rb".freeze, "lib/ollama/handlers/collector.rb".freeze, "lib/ollama/handlers/concern.rb".freeze, "lib/ollama/handlers/dump_json.rb".freeze, "lib/ollama/handlers/dump_yaml.rb".freeze, "lib/ollama/handlers/markdown.rb".freeze, "lib/ollama/handlers/nop.rb".freeze, "lib/ollama/handlers/print.rb".freeze, "lib/ollama/handlers/progress.rb".freeze, "lib/ollama/handlers/say.rb".freeze, "lib/ollama/handlers/single.rb".freeze, "lib/ollama/image.rb".freeze, "lib/ollama/message.rb".freeze, "lib/ollama/options.rb".freeze, "lib/ollama/response.rb".freeze, "lib/ollama/tool.rb".freeze, "lib/ollama/tool/function.rb".freeze, "lib/ollama/tool/function/parameters.rb".freeze, "lib/ollama/tool/function/parameters/property.rb".freeze, "lib/ollama/utils/ansi_markdown.rb".freeze, "lib/ollama/utils/chooser.rb".freeze, "lib/ollama/utils/colorize_texts.rb".freeze, "lib/ollama/utils/fetcher.rb".freeze, "lib/ollama/utils/file_argument.rb".freeze, "lib/ollama/utils/math.rb".freeze, "lib/ollama/utils/tags.rb".freeze, "lib/ollama/utils/width.rb".freeze, "lib/ollama/version.rb".freeze]
-  s.files = [".envrc".freeze, "CHANGES.md".freeze, "Gemfile".freeze, "LICENSE".freeze, "README.md".freeze, "Rakefile".freeze, "bin/ollama_chat".freeze, "bin/ollama_cli".freeze, "bin/ollama_console".freeze, "bin/ollama_update".freeze, "config/redis.conf".freeze, "docker-compose.yml".freeze, "lib/ollama.rb".freeze, "lib/ollama/client.rb".freeze, "lib/ollama/client/command.rb".freeze, "lib/ollama/client/doc.rb".freeze, "lib/ollama/commands/chat.rb".freeze, "lib/ollama/commands/copy.rb".freeze, "lib/ollama/commands/create.rb".freeze, "lib/ollama/commands/delete.rb".freeze, "lib/ollama/commands/embed.rb".freeze, "lib/ollama/commands/embeddings.rb".freeze, "lib/ollama/commands/generate.rb".freeze, "lib/ollama/commands/ps.rb".freeze, "lib/ollama/commands/pull.rb".freeze, "lib/ollama/commands/push.rb".freeze, "lib/ollama/commands/show.rb".freeze, "lib/ollama/commands/tags.rb".freeze, "lib/ollama/documents.rb".freeze, "lib/ollama/documents/memory_cache.rb".freeze, "lib/ollama/documents/redis_cache.rb".freeze, "lib/ollama/documents/splitters/character.rb".freeze, "lib/ollama/documents/splitters/semantic.rb".freeze, "lib/ollama/dto.rb".freeze, "lib/ollama/errors.rb".freeze, "lib/ollama/handlers.rb".freeze, "lib/ollama/handlers/collector.rb".freeze, "lib/ollama/handlers/concern.rb".freeze, "lib/ollama/handlers/dump_json.rb".freeze, "lib/ollama/handlers/dump_yaml.rb".freeze, "lib/ollama/handlers/markdown.rb".freeze, "lib/ollama/handlers/nop.rb".freeze, "lib/ollama/handlers/print.rb".freeze, "lib/ollama/handlers/progress.rb".freeze, "lib/ollama/handlers/say.rb".freeze, "lib/ollama/handlers/single.rb".freeze, "lib/ollama/image.rb".freeze, "lib/ollama/message.rb".freeze, "lib/ollama/options.rb".freeze, "lib/ollama/response.rb".freeze, "lib/ollama/tool.rb".freeze, "lib/ollama/tool/function.rb".freeze, "lib/ollama/tool/function/parameters.rb".freeze, "lib/ollama/tool/function/parameters/property.rb".freeze, "lib/ollama/utils/ansi_markdown.rb".freeze, "lib/ollama/utils/chooser.rb".freeze, "lib/ollama/utils/colorize_texts.rb".freeze, "lib/ollama/utils/fetcher.rb".freeze, "lib/ollama/utils/file_argument.rb".freeze, "lib/ollama/utils/math.rb".freeze, "lib/ollama/utils/tags.rb".freeze, "lib/ollama/utils/width.rb".freeze, "lib/ollama/version.rb".freeze, "ollama-ruby.gemspec".freeze, "spec/assets/embeddings.json".freeze, "spec/assets/kitten.jpg".freeze, "spec/ollama/client/doc_spec.rb".freeze, "spec/ollama/client_spec.rb".freeze, "spec/ollama/commands/chat_spec.rb".freeze, "spec/ollama/commands/copy_spec.rb".freeze, "spec/ollama/commands/create_spec.rb".freeze, "spec/ollama/commands/delete_spec.rb".freeze, "spec/ollama/commands/embed_spec.rb".freeze, "spec/ollama/commands/embeddings_spec.rb".freeze, "spec/ollama/commands/generate_spec.rb".freeze, "spec/ollama/commands/ps_spec.rb".freeze, "spec/ollama/commands/pull_spec.rb".freeze, "spec/ollama/commands/push_spec.rb".freeze, "spec/ollama/commands/show_spec.rb".freeze, "spec/ollama/commands/tags_spec.rb".freeze, "spec/ollama/documents/memory_cache_spec.rb".freeze, "spec/ollama/documents/redis_cache_spec.rb".freeze, "spec/ollama/documents/splitters/character_spec.rb".freeze, "spec/ollama/documents/splitters/semantic_spec.rb".freeze, "spec/ollama/documents_spec.rb".freeze, "spec/ollama/handlers/collector_spec.rb".freeze, "spec/ollama/handlers/dump_json_spec.rb".freeze, "spec/ollama/handlers/dump_yaml_spec.rb".freeze, "spec/ollama/handlers/markdown_spec.rb".freeze, "spec/ollama/handlers/nop_spec.rb".freeze, "spec/ollama/handlers/print_spec.rb".freeze, "spec/ollama/handlers/progress_spec.rb".freeze, "spec/ollama/handlers/say_spec.rb".freeze, "spec/ollama/handlers/single_spec.rb".freeze, "spec/ollama/image_spec.rb".freeze, "spec/ollama/message_spec.rb".freeze, "spec/ollama/options_spec.rb".freeze, "spec/ollama/tool_spec.rb".freeze, "spec/ollama/utils/ansi_markdown_spec.rb".freeze, "spec/ollama/utils/fetcher_spec.rb".freeze, "spec/ollama/utils/tags_spec.rb".freeze, "spec/spec_helper.rb".freeze, "tmp/.keep".freeze]
+  s.files = [".envrc".freeze, "CHANGES.md".freeze, "Gemfile".freeze, "LICENSE".freeze, "README.md".freeze, "Rakefile".freeze, "bin/ollama_chat".freeze, "bin/ollama_cli".freeze, "bin/ollama_console".freeze, "bin/ollama_update".freeze, "config/redis.conf".freeze, "docker-compose.yml".freeze, "lib/ollama.rb".freeze, "lib/ollama/client.rb".freeze, "lib/ollama/client/command.rb".freeze, "lib/ollama/client/doc.rb".freeze, "lib/ollama/commands/chat.rb".freeze, "lib/ollama/commands/copy.rb".freeze, "lib/ollama/commands/create.rb".freeze, "lib/ollama/commands/delete.rb".freeze, "lib/ollama/commands/embed.rb".freeze, "lib/ollama/commands/embeddings.rb".freeze, "lib/ollama/commands/generate.rb".freeze, "lib/ollama/commands/ps.rb".freeze, "lib/ollama/commands/pull.rb".freeze, "lib/ollama/commands/push.rb".freeze, "lib/ollama/commands/show.rb".freeze, "lib/ollama/commands/tags.rb".freeze, "lib/ollama/documents.rb".freeze, "lib/ollama/documents/memory_cache.rb".freeze, "lib/ollama/documents/redis_cache.rb".freeze, "lib/ollama/documents/splitters/character.rb".freeze, "lib/ollama/documents/splitters/semantic.rb".freeze, "lib/ollama/dto.rb".freeze, "lib/ollama/errors.rb".freeze, "lib/ollama/handlers.rb".freeze, "lib/ollama/handlers/collector.rb".freeze, "lib/ollama/handlers/concern.rb".freeze, "lib/ollama/handlers/dump_json.rb".freeze, "lib/ollama/handlers/dump_yaml.rb".freeze, "lib/ollama/handlers/markdown.rb".freeze, "lib/ollama/handlers/nop.rb".freeze, "lib/ollama/handlers/print.rb".freeze, "lib/ollama/handlers/progress.rb".freeze, "lib/ollama/handlers/say.rb".freeze, "lib/ollama/handlers/single.rb".freeze, "lib/ollama/image.rb".freeze, "lib/ollama/message.rb".freeze, "lib/ollama/options.rb".freeze, "lib/ollama/response.rb".freeze, "lib/ollama/tool.rb".freeze, "lib/ollama/tool/function.rb".freeze, "lib/ollama/tool/function/parameters.rb".freeze, "lib/ollama/tool/function/parameters/property.rb".freeze, "lib/ollama/utils/ansi_markdown.rb".freeze, "lib/ollama/utils/chooser.rb".freeze, "lib/ollama/utils/colorize_texts.rb".freeze, "lib/ollama/utils/fetcher.rb".freeze, "lib/ollama/utils/file_argument.rb".freeze, "lib/ollama/utils/math.rb".freeze, "lib/ollama/utils/tags.rb".freeze, "lib/ollama/utils/width.rb".freeze, "lib/ollama/version.rb".freeze, "ollama-ruby.gemspec".freeze, "spec/assets/embeddings.json".freeze, "spec/assets/kitten.jpg".freeze, "spec/assets/prompt.txt".freeze, "spec/ollama/client/doc_spec.rb".freeze, "spec/ollama/client_spec.rb".freeze, "spec/ollama/commands/chat_spec.rb".freeze, "spec/ollama/commands/copy_spec.rb".freeze, "spec/ollama/commands/create_spec.rb".freeze, "spec/ollama/commands/delete_spec.rb".freeze, "spec/ollama/commands/embed_spec.rb".freeze, "spec/ollama/commands/embeddings_spec.rb".freeze, "spec/ollama/commands/generate_spec.rb".freeze, "spec/ollama/commands/ps_spec.rb".freeze, "spec/ollama/commands/pull_spec.rb".freeze, "spec/ollama/commands/push_spec.rb".freeze, "spec/ollama/commands/show_spec.rb".freeze, "spec/ollama/commands/tags_spec.rb".freeze, "spec/ollama/documents/memory_cache_spec.rb".freeze, "spec/ollama/documents/redis_cache_spec.rb".freeze, "spec/ollama/documents/splitters/character_spec.rb".freeze, "spec/ollama/documents/splitters/semantic_spec.rb".freeze, "spec/ollama/documents_spec.rb".freeze, "spec/ollama/handlers/collector_spec.rb".freeze, "spec/ollama/handlers/dump_json_spec.rb".freeze, "spec/ollama/handlers/dump_yaml_spec.rb".freeze, "spec/ollama/handlers/markdown_spec.rb".freeze, "spec/ollama/handlers/nop_spec.rb".freeze, "spec/ollama/handlers/print_spec.rb".freeze, "spec/ollama/handlers/progress_spec.rb".freeze, "spec/ollama/handlers/say_spec.rb".freeze, "spec/ollama/handlers/single_spec.rb".freeze, "spec/ollama/image_spec.rb".freeze, "spec/ollama/message_spec.rb".freeze, "spec/ollama/options_spec.rb".freeze, "spec/ollama/tool_spec.rb".freeze, "spec/ollama/utils/ansi_markdown_spec.rb".freeze, "spec/ollama/utils/fetcher_spec.rb".freeze, "spec/ollama/utils/file_argument_spec.rb".freeze, "spec/ollama/utils/tags_spec.rb".freeze, "spec/spec_helper.rb".freeze, "tmp/.keep".freeze]
   s.homepage = "https://github.com/flori/ollama-ruby".freeze
   s.licenses = ["MIT".freeze]
   s.rdoc_options = ["--title".freeze, "Ollama-ruby - Interacting with the Ollama API".freeze, "--main".freeze, "README.md".freeze]
   s.required_ruby_version = Gem::Requirement.new("~> 3.1".freeze)
   s.rubygems_version = "3.5.18".freeze
   s.summary = "Interacting with the Ollama API".freeze
-  s.test_files = ["spec/ollama/client/doc_spec.rb".freeze, "spec/ollama/client_spec.rb".freeze, "spec/ollama/commands/chat_spec.rb".freeze, "spec/ollama/commands/copy_spec.rb".freeze, "spec/ollama/commands/create_spec.rb".freeze, "spec/ollama/commands/delete_spec.rb".freeze, "spec/ollama/commands/embed_spec.rb".freeze, "spec/ollama/commands/embeddings_spec.rb".freeze, "spec/ollama/commands/generate_spec.rb".freeze, "spec/ollama/commands/ps_spec.rb".freeze, "spec/ollama/commands/pull_spec.rb".freeze, "spec/ollama/commands/push_spec.rb".freeze, "spec/ollama/commands/show_spec.rb".freeze, "spec/ollama/commands/tags_spec.rb".freeze, "spec/ollama/documents/memory_cache_spec.rb".freeze, "spec/ollama/documents/redis_cache_spec.rb".freeze, "spec/ollama/documents/splitters/character_spec.rb".freeze, "spec/ollama/documents/splitters/semantic_spec.rb".freeze, "spec/ollama/documents_spec.rb".freeze, "spec/ollama/handlers/collector_spec.rb".freeze, "spec/ollama/handlers/dump_json_spec.rb".freeze, "spec/ollama/handlers/dump_yaml_spec.rb".freeze, "spec/ollama/handlers/markdown_spec.rb".freeze, "spec/ollama/handlers/nop_spec.rb".freeze, "spec/ollama/handlers/print_spec.rb".freeze, "spec/ollama/handlers/progress_spec.rb".freeze, "spec/ollama/handlers/say_spec.rb".freeze, "spec/ollama/handlers/single_spec.rb".freeze, "spec/ollama/image_spec.rb".freeze, "spec/ollama/message_spec.rb".freeze, "spec/ollama/options_spec.rb".freeze, "spec/ollama/tool_spec.rb".freeze, "spec/ollama/utils/ansi_markdown_spec.rb".freeze, "spec/ollama/utils/fetcher_spec.rb".freeze, "spec/ollama/utils/tags_spec.rb".freeze, "spec/spec_helper.rb".freeze]
+  s.test_files = ["spec/ollama/client/doc_spec.rb".freeze, "spec/ollama/client_spec.rb".freeze, "spec/ollama/commands/chat_spec.rb".freeze, "spec/ollama/commands/copy_spec.rb".freeze, "spec/ollama/commands/create_spec.rb".freeze, "spec/ollama/commands/delete_spec.rb".freeze, "spec/ollama/commands/embed_spec.rb".freeze, "spec/ollama/commands/embeddings_spec.rb".freeze, "spec/ollama/commands/generate_spec.rb".freeze, "spec/ollama/commands/ps_spec.rb".freeze, "spec/ollama/commands/pull_spec.rb".freeze, "spec/ollama/commands/push_spec.rb".freeze, "spec/ollama/commands/show_spec.rb".freeze, "spec/ollama/commands/tags_spec.rb".freeze, "spec/ollama/documents/memory_cache_spec.rb".freeze, "spec/ollama/documents/redis_cache_spec.rb".freeze, "spec/ollama/documents/splitters/character_spec.rb".freeze, "spec/ollama/documents/splitters/semantic_spec.rb".freeze, "spec/ollama/documents_spec.rb".freeze, "spec/ollama/handlers/collector_spec.rb".freeze, "spec/ollama/handlers/dump_json_spec.rb".freeze, "spec/ollama/handlers/dump_yaml_spec.rb".freeze, "spec/ollama/handlers/markdown_spec.rb".freeze, "spec/ollama/handlers/nop_spec.rb".freeze, "spec/ollama/handlers/print_spec.rb".freeze, "spec/ollama/handlers/progress_spec.rb".freeze, "spec/ollama/handlers/say_spec.rb".freeze, "spec/ollama/handlers/single_spec.rb".freeze, "spec/ollama/image_spec.rb".freeze, "spec/ollama/message_spec.rb".freeze, "spec/ollama/options_spec.rb".freeze, "spec/ollama/tool_spec.rb".freeze, "spec/ollama/utils/ansi_markdown_spec.rb".freeze, "spec/ollama/utils/fetcher_spec.rb".freeze, "spec/ollama/utils/file_argument_spec.rb".freeze, "spec/ollama/utils/tags_spec.rb".freeze, "spec/spec_helper.rb".freeze]
 
   s.specification_version = 4
 
-  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 1.17.0".freeze])
+  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 1.17.1".freeze])
   s.add_development_dependency(%q<all_images>.freeze, ["~> 0.4".freeze])
   s.add_development_dependency(%q<rspec>.freeze, ["~> 3.2".freeze])
-  s.add_development_dependency(%q<utils>.freeze, [">= 0".freeze])
   s.add_development_dependency(%q<webmock>.freeze, [">= 0".freeze])
   s.add_runtime_dependency(%q<excon>.freeze, ["~> 0.111".freeze])
-  s.add_runtime_dependency(%q<infobar>.freeze, ["~> 0.7".freeze])
+  s.add_runtime_dependency(%q<infobar>.freeze, ["~> 0.8".freeze])
   s.add_runtime_dependency(%q<term-ansicolor>.freeze, ["~> 1.11".freeze])
   s.add_runtime_dependency(%q<kramdown-parser-gfm>.freeze, ["~> 1.1".freeze])
   s.add_runtime_dependency(%q<terminal-table>.freeze, ["~> 3.0".freeze])
@@ -40,7 +39,7 @@ Gem::Specification.new do |s|
   s.add_runtime_dependency(%q<sorted_set>.freeze, ["~> 1.0".freeze])
   s.add_runtime_dependency(%q<mime-types>.freeze, ["~> 3.0".freeze])
   s.add_runtime_dependency(%q<reverse_markdown>.freeze, ["~> 2.0".freeze])
-  s.add_runtime_dependency(%q<complex_config>.freeze, ["~> 0.20".freeze])
+  s.add_runtime_dependency(%q<complex_config>.freeze, ["~> 0.22".freeze])
   s.add_runtime_dependency(%q<search_ui>.freeze, ["~> 0.0".freeze])
   s.add_runtime_dependency(%q<amatch>.freeze, ["~> 0.4.1".freeze])
   s.add_runtime_dependency(%q<pdf-reader>.freeze, ["~> 2.0".freeze])

data/spec/assets/prompt.txt

@@ -0,0 +1 @@
+You are a test prompt just used for testing.

data/spec/ollama/utils/file_argument_spec.rb

@@ -0,0 +1,17 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Utils::FileArgument do
+  it 'it can return content' do
+    expect(described_class.get_file_argument('foo')).to eq 'foo'
+  end
+
+  it 'it can return content at path' do
+    expect(described_class.get_file_argument(asset('prompt.txt'))).to include\
+      'test prompt'
+  end
+
+  it 'it can return default content' do
+    expect(described_class.get_file_argument('', default: 'foo')).to eq 'foo'
+    expect(described_class.get_file_argument(nil, default: 'foo')).to eq 'foo'
+  end
+end

data/spec/ollama/utils/tags_spec.rb

@@ -6,7 +6,7 @@ RSpec.describe Ollama::Utils::Tags do
   end
 
   it 'can contain unique tags and is sorted' do
-    tags = described_class.new([ 'bar', 'foo'])
+    tags = described_class.new(%w[ bar foo ])
     expect(tags.to_a).to eq %w[ bar foo ]
   end
 
@@ -14,11 +14,11 @@ RSpec.describe Ollama::Utils::Tags do
     tags = described_class.new([ 'foo' ])
     tags.add 'bar'
     expect(tags.to_a).to eq %w[ bar foo ]
-    tags.merge [ 'baz', 'baz2' ]
+    tags.merge %w[ baz baz2 ]
     expect(tags.to_a).to eq %w[ bar baz baz2 foo ]
   end
 
   it 'can be output nicely' do
-    expect(described_class.new(%w[foo bar]).to_s).to eq '#bar #foo'
+    expect(described_class.new(%w[ foo bar ]).to_s).to eq '#bar #foo'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ollama-ruby
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.2
 platform: ruby
 authors:
 - Florian Frank
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-09-06 00:00:00.000000000 Z
+date: 2024-09-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gem_hadar
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.17.0
+        version: 1.17.1
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.17.0
+        version: 1.17.1
 - !ruby/object:Gem::Dependency
   name: all_images
   requirement: !ruby/object:Gem::Requirement
@@ -52,20 +52,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.2'
-- !ruby/object:Gem::Dependency
-  name: utils
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
 - !ruby/object:Gem::Dependency
   name: webmock
   requirement: !ruby/object:Gem::Requirement
@@ -100,14 +86,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.7'
+        version: '0.8'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.7'
+        version: '0.8'
 - !ruby/object:Gem::Dependency
   name: term-ansicolor
   requirement: !ruby/object:Gem::Requirement
@@ -240,14 +226,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.20'
+        version: '0.22'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.20'
+        version: '0.22'
 - !ruby/object:Gem::Dependency
   name: search_ui
   requirement: !ruby/object:Gem::Requirement
@@ -418,6 +404,7 @@ files:
 - ollama-ruby.gemspec
 - spec/assets/embeddings.json
 - spec/assets/kitten.jpg
+- spec/assets/prompt.txt
 - spec/ollama/client/doc_spec.rb
 - spec/ollama/client_spec.rb
 - spec/ollama/commands/chat_spec.rb
@@ -452,6 +439,7 @@ files:
 - spec/ollama/tool_spec.rb
 - spec/ollama/utils/ansi_markdown_spec.rb
 - spec/ollama/utils/fetcher_spec.rb
+- spec/ollama/utils/file_argument_spec.rb
 - spec/ollama/utils/tags_spec.rb
 - spec/spec_helper.rb
 - tmp/.keep
@@ -517,5 +505,6 @@ test_files:
 - spec/ollama/tool_spec.rb
 - spec/ollama/utils/ansi_markdown_spec.rb
 - spec/ollama/utils/fetcher_spec.rb
+- spec/ollama/utils/file_argument_spec.rb
 - spec/ollama/utils/tags_spec.rb
 - spec/spec_helper.rb