ollama-ruby 0.7.0 → 0.9.0

data/docker-compose.yml

@@ -1,6 +1,6 @@
 services:
   redis:
-    image: redis:7.2.5-alpine
+    image: valkey/valkey:7.2.7-alpine
     restart: unless-stopped
     ports: [ "127.0.0.1:9736:6379" ]
     volumes:

data/lib/ollama/documents.rb

@@ -55,8 +55,9 @@ class Ollama::Documents
     @cache.prefix = prefix
   end
 
-  def add(inputs, batch_size: 10, source: nil, tags: [])
+  def add(inputs, batch_size: nil, source: nil, tags: [])
     inputs = Array(inputs)
+    batch_size ||= 10
     tags = Ollama::Utils::Tags.new(tags, source:)
     if source
       tags.add(File.basename(source).gsub(/\?.*/, ''), source:)

data/lib/ollama/handlers/markdown.rb

@@ -7,7 +7,7 @@ class Ollama::Handlers::Markdown
   def initialize(output: $stdout)
     super
     @output.sync = true
-    @content = ''
+    @content     = ''
   end
 
   def call(response)
@@ -16,7 +16,6 @@ class Ollama::Handlers::Markdown
       markdown_content = Ollama::Utils::ANSIMarkdown.parse(@content)
       @output.print clear_screen, move_home, markdown_content
     end
-    response.done and @output.puts
     self
   end
 end

data/lib/ollama/utils/fetcher.rb

@@ -45,6 +45,8 @@ class Ollama::Utils::Fetcher
         file.content_type = MIME::Types.type_for(filename).first
         block.(file)
       end
+    else
+      STDERR.puts "File #{filename.to_s.inspect} doesn't exist."
     end
   end
 

data/lib/ollama/version.rb

@@ -1,6 +1,6 @@
 module Ollama
   # Ollama version
-  VERSION         = '0.7.0'
+  VERSION         = '0.9.0'
   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,14 +1,14 @@
 # -*- encoding: utf-8 -*-
-# stub: ollama-ruby 0.7.0 ruby lib
+# stub: ollama-ruby 0.9.0 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "ollama-ruby".freeze
-  s.version = "0.7.0".freeze
+  s.version = "0.9.0".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-10-02"
+  s.date = "2024-10-18"
   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]
@@ -24,7 +24,7 @@ Gem::Specification.new do |s|
 
   s.specification_version = 4
 
-  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 1.18.0".freeze])
+  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 1.19".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<webmock>.freeze, [">= 0".freeze])

data/spec/ollama/client_spec.rb

@@ -54,21 +54,21 @@ RSpec.describe Ollama::Client do
   end
 
   it 'can raise error on connection error' do
-    allow(excon).to receive(:post).and_raise Excon::Error::Socket
+    expect(excon).to receive(:post).and_raise Excon::Error::Socket
     expect {
       ollama.generate(model: 'llama3.1', prompt: 'Hello World')
     }.to raise_error(Ollama::Errors::SocketError)
   end
 
   it 'can raise error on timeout' do
-    allow(excon).to receive(:post).and_raise Excon::Errors::Timeout
+    expect(excon).to receive(:post).and_raise Excon::Errors::Timeout
     expect {
       ollama.generate(model: 'llama3.1', prompt: 'Hello World')
     }.to raise_error(Ollama::Errors::TimeoutError)
   end
 
   it 'can raise a generic error' do
-    allow(excon).to receive(:post).and_raise Excon::Errors::Error
+    expect(excon).to receive(:post).and_raise Excon::Errors::Error
     expect {
       ollama.generate(model: 'llama3.1', prompt: 'Hello World')
     }.to raise_error(Ollama::Errors::Error)

data/spec/ollama/documents/redis_backed_memory_cache_spec.rb

@@ -78,7 +78,7 @@ RSpec.describe Ollama::Documents::RedisBackedMemoryCache do
     end
 
     it 'returns size' do
-      allow(cache).to receive(:count).and_return 3
+      expect(cache).to receive(:count).and_return 3
       expect(cache.size).to eq 3
     end
 

data/spec/ollama/documents/redis_cache_spec.rb

@@ -57,7 +57,7 @@ RSpec.describe Ollama::Documents::RedisCache do
       key, value = 'foo', { test: true }
       expect(redis).to receive(:set).with('test-' + key, JSON(value), ex: 3_600)
       cache[key] = value
-      allow(redis).to receive(:ttl).with('test-' + key).and_return 3_600
+      expect(redis).to receive(:ttl).with('test-' + key).and_return 3_600
       expect(cache.ttl(key)).to eq 3_600
     end
 

data/spec/ollama/documents_spec.rb

@@ -42,7 +42,7 @@ RSpec.describe Ollama::Documents do
   end
 
   it 'can find strings' do
-    allow(ollama).to receive(:embed).
+    expect(ollama).to receive(:embed).
       with(model:, input: [ 'foo' ], options: nil).
       and_return(double(embeddings: [ [ 0.1 ] ]))
     expect(documents << 'foo').to eq documents
@@ -57,7 +57,7 @@ RSpec.describe Ollama::Documents do
   end
 
   it 'can find only tagged strings' do
-    allow(ollama).to receive(:embed).
+    expect(ollama).to receive(:embed).
       with(model:, input: [ 'foo' ], options: nil).
       and_return(double(embeddings: [ [ 0.1 ] ]))
     expect(documents.add('foo', tags: %i[ test ])).to eq documents
@@ -77,10 +77,10 @@ RSpec.describe Ollama::Documents do
   end
 
   it 'can find strings conditionally' do
-    allow(ollama).to receive(:embed).
+    expect(ollama).to receive(:embed).
       with(model:, input: [ 'foobar' ], options: nil).
       and_return(double(embeddings: [ [ 0.01 ] ]))
-    allow(ollama).to receive(:embed).
+    expect(ollama).to receive(:embed).
       with(model:, input: [ 'foo' ], options: nil).
       and_return(double(embeddings: [ [ 0.1 ] ]))
     expect(documents << 'foobar').to eq documents
@@ -132,7 +132,7 @@ RSpec.describe Ollama::Documents do
     end
 
     it 'can clear texts with tags' do
-      allow(ollama).to receive(:embed).
+      expect(ollama).to receive(:embed).
         with(model:, input: %w[ bar ], options: nil).
         and_return(double(embeddings: [ [ 0.1 ] ]))
       expect(documents.add('foo', tags: %i[ test ])).to eq documents

data/spec/ollama/handlers/markdown_spec.rb

@@ -25,7 +25,6 @@ RSpec.describe Ollama::Handlers::Markdown do
   it 'can markdown response as markdown' do
     output = double('output', :sync= => true)
     expect(output).to receive(:print).with("\e[2J", "\e[1;1H", ansi)
-    expect(output).to receive(:puts)
     markdown = described_class.new(output:)
     response = double('response', response: md, done: false)
     markdown.call(response)
@@ -36,7 +35,6 @@ RSpec.describe Ollama::Handlers::Markdown do
   it 'can markdown message content as markdown' do
     output = double('output', :sync= => true)
     expect(output).to receive(:print).with("\e[2J", "\e[1;1H", ansi)
-    expect(output).to receive(:puts)
     markdown = described_class.new(output:)
     response = double('response', response: nil, message: double(content: md), done: false)
     markdown.call(response)

data/spec/ollama/utils/fetcher_spec.rb

@@ -105,7 +105,7 @@ RSpec.describe Ollama::Utils::Fetcher do
   end
 
   it 'can .execute and fail' do
-    allow(IO).to receive(:popen).and_raise StandardError
+    expect(IO).to receive(:popen).and_raise StandardError
     described_class.execute('foobar') do |file|
       expect(file).to be_a StringIO
       expect(file.read).to be_empty

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ollama-ruby
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.9.0
 platform: ruby
 authors:
 - Florian Frank
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-02 00:00:00.000000000 Z
+date: 2024-10-18 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.18.0
+        version: '1.19'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.18.0
+        version: '1.19'
 - !ruby/object:Gem::Dependency
   name: all_images
   requirement: !ruby/object:Gem::Requirement