ollama-ruby 0.0.0

data/spec/ollama/commands/show_spec.rb

@@ -0,0 +1,28 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Commands::Show do
+  it 'can be instantiated' do
+    show = described_class.new(name: 'llama3.1')
+    expect(show).to be_a described_class
+  end
+
+  it 'can be converted to JSON' do
+    show = described_class.new(name: 'llama3.1')
+    expect(show.as_json).to include(
+      name: 'llama3.1', stream: false
+    )
+    expect(show.to_json).to eq(
+      '{"json_class":"Ollama::Commands::Show","name":"llama3.1","stream":false}'
+    )
+  end
+
+  it 'can perform' do
+    show = described_class.new(name: 'llama3.1')
+    show.client = client = double('client')
+    expect(client).to receive(:request).with(
+      method: :post, path: '/api/show', handler: Ollama::Handlers::NOP ,stream: false,
+      body: '{"json_class":"Ollama::Commands::Show","name":"llama3.1","stream":false}'
+    )
+    show.perform(Ollama::Handlers::NOP)
+  end
+end

data/spec/ollama/commands/tags_spec.rb

@@ -0,0 +1,22 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Commands::Tags do
+  let :tags do
+    described_class.new
+  end
+
+  it 'can be instantiated' do
+    expect(tags).to be_a described_class
+  end
+
+  it 'cannot be converted to JSON' do
+    expect(tags).not_to respond_to(:as_json)
+  end
+
+  it 'can perform' do
+    tags.client = client = double('client')
+    expect(client).to receive(:request).
+      with(method: :get, path: '/api/tags', stream: false, handler: Ollama::Handlers::NOP)
+    tags.perform(Ollama::Handlers::NOP)
+  end
+end

data/spec/ollama/handlers/collector_spec.rb

@@ -0,0 +1,15 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Handlers::Collector do
+  it 'has .call' do
+    expect_any_instance_of(described_class).to receive(:call).with(:foo)
+    described_class.call(:foo)
+  end
+
+  it 'can collect responses in an array' do
+    collector = described_class.new(output:)
+    response = Ollama::Response[foo: 'testing']
+    collector.call(response)
+    expect(collector.result).to eq [ response ]
+  end
+end

data/spec/ollama/handlers/dump_json_spec.rb

@@ -0,0 +1,16 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Handlers::DumpJSON do
+  it 'has .call' do
+    expect_any_instance_of(described_class).to receive(:call).with(:foo)
+    described_class.call(:foo)
+  end
+
+  it 'can print pretty JSON' do
+    output = double('output')
+    expect(output).to receive(:puts).with(%Q'{\n  "foo": "testing"\n}')
+    print = described_class.new(output:)
+    response = Ollama::Response[foo: 'testing']
+    print.call(response)
+  end
+end

data/spec/ollama/handlers/dump_yaml_spec.rb

@@ -0,0 +1,18 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Handlers::DumpYAML do
+  it 'has .call' do
+    expect_any_instance_of(described_class).to receive(:call).with(:foo)
+    described_class.call(:foo)
+  end
+
+  it 'can print YAML' do
+    output = double('output')
+    expect(output).to receive(:puts).with(
+      "--- !ruby/object:Ollama::Response\nfoo: testing\n"
+    )
+    print = described_class.new(output:)
+    response = Ollama::Response[foo: 'testing']
+    print.call(response)
+  end
+end

data/spec/ollama/handlers/markdown_spec.rb

@@ -0,0 +1,46 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Handlers::Markdown do
+  it 'has .call' do
+    expect_any_instance_of(described_class).to receive(:call).with(:foo)
+    described_class.call(:foo)
+  end
+
+  let :md do
+    <<~end
+      - **strong**
+      - *emphasized*
+    end
+  end
+
+  let :ansi do
+    <<~end
+    · \e[1mstrong\e[0m
+
+    · \e[3memphasized\e[0m
+
+    end
+  end
+
+  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)
+    response = double('response', response: nil, message: nil, done: true)
+    markdown.call(response)
+  end
+
+  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)
+    response = double('response', response: nil, message: nil, done: true)
+    markdown.call(response)
+  end
+end

data/spec/ollama/handlers/nop_spec.rb

@@ -0,0 +1,15 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Handlers::NOP do
+  it 'has .to_proc' do
+    expect_any_instance_of(described_class).to receive(:call).with(:foo)
+    described_class.call(:foo)
+  end
+
+  it 'can do nothing at all' do
+    nop = described_class.new(output:)
+    response = Ollama::Response[foo: 'testing']
+    nop.call(response)
+    expect(nop.result).to be_nil
+  end
+end

data/spec/ollama/handlers/print_spec.rb

@@ -0,0 +1,30 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Handlers::Print do
+  it 'has .to_proc' do
+    expect_any_instance_of(described_class).to receive(:call).with(:foo)
+    described_class.call(:foo)
+  end
+
+  it 'can print response' do
+    output = double('output', :sync= => true)
+    expect(output).to receive(:print).with('testing')
+    expect(output).to receive(:puts)
+    print = described_class.new(output:)
+    response = double('response', response: 'testing', done: false)
+    print.call(response)
+    response = double('response', response: nil, message: nil, done: true)
+    print.call(response)
+  end
+
+  it 'can print message content' do
+    output = double('output', :sync= => true)
+    expect(output).to receive(:print).with('testing')
+    expect(output).to receive(:puts)
+    print = described_class.new(output:)
+    response = double('response', response: nil, message: double(content: 'testing'), done: false)
+    print.call(response)
+    response = double('response', response: nil, message: nil, done: true)
+    print.call(response)
+  end
+end

data/spec/ollama/handlers/progress_spec.rb

@@ -0,0 +1,22 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Handlers::Progress do
+  it 'has .to_proc' do
+    expect_any_instance_of(described_class).to receive(:call).with(:foo)
+    described_class.call(:foo)
+  end
+
+  it 'can display progress' do
+    response = double('response', status: 'testing', completed: 23, total: 666)
+    expect(infobar.counter).to receive(:progress).with(by: 23).and_call_original
+    expect(infobar.display).to receive(:update).and_call_original
+    described_class.new.call(response)
+  end
+
+  it 'can display errors in progress' do
+    response = double('response', error: 'foo', status: nil, completed: nil, total: nil)
+    progress = described_class.new
+    expect(progress.output).to receive(:puts).with(/Error: .*foo/)
+    progress.call(response)
+  end
+end

data/spec/ollama/handlers/say_spec.rb

@@ -0,0 +1,30 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Handlers::Say do
+  it 'has .to_proc' do
+    expect_any_instance_of(described_class).to receive(:call).with(:foo)
+    described_class.call(:foo)
+  end
+
+  it 'can print response' do
+    output = double('output', :sync= => true)
+    expect(output).to receive(:print).with('testing')
+    expect(output).to receive(:close)
+    print = described_class.new(output:)
+    response = double('response', response: 'testing', done: false)
+    print.call(response)
+    response = double('response', response: nil, message: nil, done: true)
+    print.call(response)
+  end
+
+  it 'can print message content' do
+    output = double('output', :sync= => true)
+    expect(output).to receive(:print).with('testing')
+    expect(output).to receive(:close)
+    print = described_class.new(output:)
+    response = double('response', response: nil, message: double(content: 'testing'), done: false)
+    print.call(response)
+    response = double('response', response: nil, message: nil, done: true)
+    print.call(response)
+  end
+end

data/spec/ollama/handlers/single_spec.rb

@@ -0,0 +1,24 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Handlers::Single do
+  it 'has .call' do
+    expect_any_instance_of(described_class).to receive(:call).with(:foo)
+    described_class.call(:foo)
+  end
+
+  it 'can collect responses in an array' do
+    single = described_class.new(output:)
+    response1 = Ollama::Response[foo: 'testing1']
+    response2 = Ollama::Response[foo: 'testing2']
+    single.call(response1)
+    single.call(response2)
+    expect(single.result).to eq [ response1, response2 ]
+  end
+
+  it 'can return only the single result' do
+    single = described_class.new(output:)
+    response = Ollama::Response[foo: 'testing']
+    single.call(response)
+    expect(single.result).to eq response
+  end
+end

data/spec/ollama/image_spec.rb

@@ -0,0 +1,23 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Image do
+  let :image do
+    described_class.for_filename(asset('kitten.jpg'))
+  end
+
+  it 'can be instantiated' do
+    expect(image).to be_a described_class
+  end
+
+  it 'cannot be created via .new' do
+    expect {
+      described_class.new('nix')
+    }.to raise_error NoMethodError
+  end
+
+  it 'can be converted to base64 string' do
+    expect(image.to_s.size).to eq 134400
+    expect(image.to_s.sum).to eq 42460
+    expect(image.to_s[0, 40]).to eq '/9j/4AAQSkZJRgABAQAASABIAAD/4QBYRXhpZgAA'
+  end
+end

data/spec/ollama/message_spec.rb

@@ -0,0 +1,37 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Message do
+  let :image do
+    Ollama::Image.for_string("test")
+  end
+
+  let :message do
+    described_class.new(
+      role: 'user',
+      content: 'hello world',
+      images: image
+    )
+  end
+
+  it 'can be instantiated' do
+    expect(message).to be_a described_class
+  end
+
+  it 'can be converted to JSON' do
+    expect(message.as_json).to eq(
+      json_class: described_class.name,
+      role: 'user',
+      content: 'hello world',
+      images: [ image ],
+    )
+    expect(message.to_json).to eq(
+      '{"json_class":"Ollama::Message","role":"user","content":"hello world","images":["dGVzdA==\n"]}'
+    )
+  end
+
+  it 'can be restored from JSON' do
+    expect(JSON(<<~'end', create_additions: true)).to be_a described_class
+      {"json_class":"Ollama::Message","role":"user","content":"hello world","images":["dGVzdA==\n"]}
+    end
+  end
+end

data/spec/ollama/options_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Options do
+  let :options do
+    described_class.new(
+      penalize_newline: true,
+      num_ctx: 8192,
+      temperature: 0.7,
+    )
+  end
+
+  it 'can be instantiated' do
+    expect(options).to be_a described_class
+  end
+
+  it 'throws error for invalid types' do
+    expect { described_class.new(temperature: Class.new) }.
+      to raise_error(TypeError)
+  end
+
+  it 'throws error for invalid boolean values' do
+    expect { described_class.new(penalize_newline: :tertium) }.
+      to raise_error(TypeError)
+  end
+end

data/spec/ollama/tool_spec.rb

@@ -0,0 +1,78 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Tool do
+  let :location do
+    Ollama::Tool::Function::Parameters::Property.new(
+      type: 'string',
+      description: 'The location to get the weather for, e.g. Berlin, Berlin',
+    )
+  end
+
+  let :format do
+    Ollama::Tool::Function::Parameters::Property.new(
+      type: 'string',
+      description: "The format to return the weather in, e.g. 'celsius' or 'fahrenheit'",
+      enum: %w[ celsius fahrenheit ]
+    )
+  end
+
+  let :parameters do
+    Ollama::Tool::Function::Parameters.new(
+      type: 'object',
+      properties: { location:, format: },
+      required: %w[ location format ],
+    )
+  end
+
+  let :function do
+    Ollama::Tool::Function.new(
+      name: 'get_current_weather',
+      description: 'Get the current weather for a location',
+      parameters:,
+    )
+  end
+
+  let :tool do
+    described_class.new(
+      type: 'function',
+      function:,
+    )
+  end
+
+  it 'can be instantiated' do
+    expect(tool).to be_a described_class
+  end
+
+  it 'cannot be converted to JSON' do
+    expect(tool.as_json).to eq(
+      json_class: described_class.name,
+      type: 'function',
+      function: {
+        json_class: "Ollama::Tool::Function",
+        name: 'get_current_weather',
+        description: "Get the current weather for a location",
+        parameters: {
+          json_class: "Ollama::Tool::Function::Parameters",
+          type: "object",
+          properties: {
+            location: {
+              json_class: "Ollama::Tool::Function::Parameters::Property",
+              type: "string",
+              description: "The location to get the weather for, e.g. Berlin, Berlin"
+            },
+            format: {
+              json_class: "Ollama::Tool::Function::Parameters::Property",
+              type: "string",
+              description: "The format to return the weather in, e.g. 'celsius' or 'fahrenheit'",
+              enum: ["celsius", "fahrenheit"]
+            }
+          },
+          required: ["location", "format"]
+        },
+      }
+    )
+    expect(tool.to_json).to eq(
+      %{{"json_class":"Ollama::Tool","type":"function","function":{"json_class":"Ollama::Tool::Function","name":"get_current_weather","description":"Get the current weather for a location","parameters":{"json_class":"Ollama::Tool::Function::Parameters","type":"object","properties":{"location":{"json_class":"Ollama::Tool::Function::Parameters::Property","type":"string","description":"The location to get the weather for, e.g. Berlin, Berlin"},"format":{"json_class":"Ollama::Tool::Function::Parameters::Property","type":"string","description":"The format to return the weather in, e.g. 'celsius' or 'fahrenheit'","enum":["celsius","fahrenheit"]}},"required":["location","format"]}}}}
+    )
+  end
+end

data/spec/ollama/utils/ansi_markdown_spec.rb

@@ -0,0 +1,15 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Utils::ANSIMarkdown do
+  let :source  do
+    File.read(Pathname.new(__dir__) + '..' + '..' + '..' + 'README.md')
+  end
+
+  it 'can parse' do
+    File.open('tmp/README.ansi', ?w) do |output|
+      ansi = described_class.parse(source)
+      expect(ansi).to match("This is the end.")
+      output.puts ansi
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,16 @@
+if ENV['START_SIMPLECOV'].to_i == 1
+  require 'simplecov'
+  SimpleCov.start do
+    add_filter "#{File.basename(File.dirname(__FILE__))}/"
+  end
+end
+require 'rspec'
+begin
+  require 'debug'
+rescue LoadError
+end
+require 'ollama'
+
+def asset(name)
+  File.join(__dir__, 'assets', name)
+end