ollama-ruby 0.5.0 → 0.7.0

data/spec/ollama/utils/fetcher_spec.rb

@@ -6,7 +6,7 @@ RSpec.describe Ollama::Utils::Fetcher do
   end
 
   let :fetcher do
-    described_class.new
+    described_class.new.expose
   end
 
   it 'can be instantiated' do
@@ -36,7 +36,7 @@ RSpec.describe Ollama::Utils::Fetcher do
   it 'can #get without ssl peer verification' do
     fetcher = described_class.new(
       http_options: { ssl_verify_peer: false }
-    )
+    ).expose
     stub_request(:get, 'https://www.example.com/hello').
       with(headers: fetcher.headers).
       to_return(

data/spec/ollama/utils/tags_spec.rb

@@ -14,11 +14,35 @@ 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 %w[ baz baz2 ]
-    expect(tags.to_a).to eq %w[ bar baz baz2 foo ]
+  end
+
+  it 'can increase in size' do
+    tags = described_class.new
+    expect { tags.add 'foo' }.to change { tags.size }.from(0).to(1)
+    expect { tags.add 'bar' }.to change { tags.size }.from(1).to(2)
+  end
+
+  it 'can be cleared' do
+    tags = described_class.new([ 'foo', 'bar' ])
+    expect { tags.clear }.to change { tags.size }.from(2).to(0)
+  end
+
+  it 'tags can be empt' do
+    tags = described_class.new([ 'foo' ])
+    expect { tags.clear }.to change { tags.empty? }.from(false).to(true)
   end
 
   it 'can be output nicely' do
     expect(described_class.new(%w[ foo bar ]).to_s).to eq '#bar #foo'
   end
+
+  it 'can be output nicely with links to source' do
+    tags = described_class.new([ 'foo' ], source: 'https://foo.example.com')
+    tags.add 'bar', source: '/path/to/bar.html'
+    expect(tags.to_a).to eq %w[ bar foo ]
+    tags.all? { expect(_1).to be_a(Ollama::Utils::Tags::Tag) }
+    expect(tags.to_s).to eq(
+      "\e]8;;file:///path/to/bar.html\e\\#bar\e]8;;\e\\ \e]8;;https://foo.example.com\e\\#foo\e]8;;\e\\"
+    )
+  end
 end

data/spec/ollama/utils/width_spec.rb

@@ -0,0 +1,82 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Utils::Width do
+  before do
+    allow(Tins::Terminal).to receive(:columns).and_return 80
+  end
+
+  describe '.width' do
+    it 'defaults to 100%' do
+      expect(described_class.width).to eq 80
+    end
+
+    it 'can be to 80%' do
+      expect(described_class.width(percentage: 80)).to eq 64
+    end
+  end
+
+  describe '.wrap' do
+    it 'can wrap with percentage' do
+      wrapped = described_class.wrap([ ?A * 10 ] * 10 * ' ', percentage: 80)
+      expect(wrapped).to eq(
+        "AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA\n"\
+        "AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA"
+      )
+      expect(wrapped.size).to eq 109
+    end
+
+    it 'can wrap with length' do
+      wrapped = described_class.wrap([ ?A * 10 ] * 10 * ' ', length: 64)
+      expect(wrapped).to eq(
+        "AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA\n"\
+        "AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA"
+      )
+      expect(wrapped.size).to eq 109
+    end
+
+    it "doesn't wrap with length 0" do
+      wrapped = described_class.wrap([ ?A * 10 ] * 10 * ' ', length: 0)
+      expect(wrapped).to eq(
+        "AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA "\
+        "AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA"
+      )
+    end
+  end
+
+  describe '.truncate' do
+    it 'can truncate with percentage' do
+      truncated = described_class.truncate([ ?A * 10 ] * 10 * ' ', percentage: 80)
+      expect(truncated).to eq(
+        "AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAA…"
+      )
+      expect(truncated.size).to eq 64
+    end
+
+    it 'can truncate with length' do
+      truncated = described_class.truncate([ ?A * 10 ] * 10 * ' ', length: 64)
+      expect(truncated).to eq(
+        "AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAA…"
+      )
+      expect(truncated.size).to eq 64
+    end
+
+    it 'cannot truncate if not necessary' do
+      text = [ ?A * 10 ] * 5 * ' '
+      truncated = described_class.truncate(text, length: 54)
+      expect(truncated).to eq text
+    end
+
+    it 'can truncate with length 0' do
+      truncated = described_class.truncate([ ?A * 10 ] * 10 * ' ', length: 0)
+      expect(truncated).to be_empty
+    end
+
+    it 'can truncate with ...' do
+      truncated = described_class.truncate([ ?A * 10 ] * 10 * ' ', length: 64, ellipsis: '...')
+      expect(truncated).to eq(
+        "AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAAAAAA AAAAAA..."
+      )
+      expect(truncated.size).to eq 64
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -5,6 +5,7 @@ if ENV['START_SIMPLECOV'].to_i == 1
   end
 end
 require 'rspec'
+require 'tins/xt/expose'
 begin
   require 'debug'
 rescue LoadError

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ollama-ruby
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.7.0
 platform: ruby
 authors:
 - Florian Frank
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-09-26 00:00:00.000000000 Z
+date: 2024-10-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gem_hadar
@@ -206,20 +206,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.1'
-- !ruby/object:Gem::Dependency
-  name: sorted_set
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '1.0'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '1.0'
 - !ruby/object:Gem::Dependency
   name: mime-types
   requirement: !ruby/object:Gem::Requirement
@@ -332,6 +318,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: xdg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+- !ruby/object:Gem::Dependency
+  name: tins
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.34'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.34'
 description: Library that allows interacting with the Ollama API
 email: flori@ping.de
 executables:
@@ -387,6 +401,7 @@ extra_rdoc_files:
 - lib/ollama/tool/function/parameters.rb
 - lib/ollama/tool/function/parameters/property.rb
 - lib/ollama/utils/ansi_markdown.rb
+- lib/ollama/utils/cache_fetcher.rb
 - lib/ollama/utils/chooser.rb
 - lib/ollama/utils/colorize_texts.rb
 - lib/ollama/utils/fetcher.rb
@@ -453,6 +468,7 @@ files:
 - lib/ollama/tool/function/parameters.rb
 - lib/ollama/tool/function/parameters/property.rb
 - lib/ollama/utils/ansi_markdown.rb
+- lib/ollama/utils/cache_fetcher.rb
 - lib/ollama/utils/chooser.rb
 - lib/ollama/utils/colorize_texts.rb
 - lib/ollama/utils/fetcher.rb
@@ -499,9 +515,11 @@ files:
 - spec/ollama/options_spec.rb
 - spec/ollama/tool_spec.rb
 - spec/ollama/utils/ansi_markdown_spec.rb
+- spec/ollama/utils/cache_fetcher_spec.rb
 - spec/ollama/utils/fetcher_spec.rb
 - spec/ollama/utils/file_argument_spec.rb
 - spec/ollama/utils/tags_spec.rb
+- spec/ollama/utils/width_spec.rb
 - spec/spec_helper.rb
 - tmp/.keep
 homepage: https://github.com/flori/ollama-ruby
@@ -566,7 +584,9 @@ test_files:
 - spec/ollama/options_spec.rb
 - spec/ollama/tool_spec.rb
 - spec/ollama/utils/ansi_markdown_spec.rb
+- spec/ollama/utils/cache_fetcher_spec.rb
 - spec/ollama/utils/fetcher_spec.rb
 - spec/ollama/utils/file_argument_spec.rb
 - spec/ollama/utils/tags_spec.rb
+- spec/ollama/utils/width_spec.rb
 - spec/spec_helper.rb