ollama-ruby 0.0.0

data/lib/ollama/utils/width.rb

@@ -0,0 +1,22 @@
+require 'tins/terminal'
+
+module Ollama::Utils::Width
+  module_function
+
+  def width(percentage: 100.0)
+    ((Float(percentage) * Tins::Terminal.columns) / 100).floor
+  end
+
+  def wrap(text, percentage: nil, length: nil)
+    percentage.nil? ^ length.nil? or
+      raise ArgumentError, "either pass percentage or length argument"
+    percentage and length ||= width(percentage:)
+    text.gsub(/(?<!\n)\n(?!\n)/, ' ').lines.map do |line|
+      if line.length > length
+        line.gsub(/(.{1,#{length}})(\s+|$)/, "\\1\n").strip
+      else
+        line.strip
+      end
+    end * ?\n
+  end
+end

data/lib/ollama/version.rb

@@ -0,0 +1,8 @@
+module Ollama
+  # Ollama version
+  VERSION         = '0.0.0'
+  VERSION_ARRAY   = VERSION.split('.').map(&:to_i) # :nodoc:
+  VERSION_MAJOR   = VERSION_ARRAY[0] # :nodoc:
+  VERSION_MINOR   = VERSION_ARRAY[1] # :nodoc:
+  VERSION_BUILD   = VERSION_ARRAY[2] # :nodoc:
+end

data/lib/ollama.rb

@@ -0,0 +1,43 @@
+require 'json'
+
+module Ollama
+end
+
+require 'ollama/handlers'
+module Ollama
+  include Ollama::Handlers
+end
+
+require 'ollama/version'
+require 'ollama/errors'
+require 'ollama/dto'
+require 'ollama/image'
+require 'ollama/message'
+require 'ollama/tool'
+require 'ollama/tool/function'
+require 'ollama/tool/function/parameters'
+require 'ollama/tool/function/parameters/property'
+require 'ollama/response'
+require 'ollama/options'
+
+module Ollama::Utils
+end
+require 'ollama/utils/width'
+require 'ollama/utils/ansi_markdown'
+
+class Ollama::Commands
+end
+require 'ollama/commands/generate'
+require 'ollama/commands/chat'
+require 'ollama/commands/create'
+require 'ollama/commands/tags'
+require 'ollama/commands/show'
+require 'ollama/commands/copy'
+require 'ollama/commands/delete'
+require 'ollama/commands/pull'
+require 'ollama/commands/push'
+require 'ollama/commands/embed'
+require 'ollama/commands/embeddings'
+require 'ollama/commands/ps'
+
+require 'ollama/client'

data/ollama-ruby.gemspec

@@ -0,0 +1,36 @@
+# -*- encoding: utf-8 -*-
+# stub: ollama-ruby 0.0.0 ruby lib
+
+Gem::Specification.new do |s|
+  s.name = "ollama-ruby".freeze
+  s.version = "0.0.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-08-12"
+  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]
+  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/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/width.rb".freeze, "lib/ollama/version.rb".freeze]
+  s.files = ["Gemfile".freeze, "LICENSE".freeze, "README.md".freeze, "Rakefile".freeze, "bin/ollama_chat".freeze, "bin/ollama_console".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/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/width.rb".freeze, "lib/ollama/version.rb".freeze, "ollama-ruby.gemspec".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/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/spec_helper.rb".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.16".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/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/spec_helper.rb".freeze]
+
+  s.specification_version = 4
+
+  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 1.16.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_runtime_dependency(%q<excon>.freeze, ["~> 0.111".freeze])
+  s.add_runtime_dependency(%q<infobar>.freeze, ["~> 0.7".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])
+end

data/spec/ollama/client/doc_spec.rb

@@ -0,0 +1,11 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Client::Doc do
+  it 'can document commands' do
+    expect(Ollama::Client::Doc.new(:generate).to_s).to match(/generate-a-completion/)
+  end
+
+  it 'defaults to the whole API document' do
+    expect(Ollama::Client::Doc.new(:nix).to_s).to match(%r(main/docs/api.md))
+  end
+end

data/spec/ollama/client_spec.rb

@@ -0,0 +1,144 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Client do
+  let :base_url do
+    'https://ai.foo.bar'
+  end
+
+  let :client do
+    described_class.new base_url:
+  end
+
+  it 'can be instantiated' do
+    expect(client).to be_a described_class
+  end
+
+  it 'can be configured via environment variable' do
+    expect { described_class.new }.to raise_error(ArgumentError)
+    ENV['OLLAMA_URL'] = base_url
+    expect(described_class.new).to be_a described_class
+  end
+
+  it 'can disable ssl peer verification' do
+    expect(client).to be_ssl_verify_peer
+    client2 = described_class.new(
+      base_url: 'https://ai.foo.bar?ssl_verify_peer=false'
+    )
+    expect(client2).not_to be_ssl_verify_peer
+  end
+
+  it 'has a string representation' do
+    expect(client.to_s).to eq '#<Ollama::Client@https://ai.foo.bar>'
+  end
+
+  let :excon do
+    double('excon')
+  end
+
+  before do
+    allow(Excon).to receive(:new).and_return(excon)
+  end
+
+  it 'can raise error based on status code 500' do
+    expect(excon).to receive(:send).and_return(double(status: 500, body: '{}'))
+    expect {
+      client.generate(model: 'llama3.1', prompt: 'Hello World')
+    }.to raise_error(Ollama::Errors::Error)
+  end
+
+  it 'can raise error based on status code 404' do
+    expect(excon).to receive(:send).and_return(double(status: 404, body: '{}'))
+    expect {
+      client.generate(model: 'llama3.1', prompt: 'Hello World')
+    }.to raise_error(Ollama::Errors::NotFoundError)
+  end
+
+  it 'can raise error on connection error' do
+    allow(excon).to receive(:post).and_raise Excon::Error::Socket
+    expect {
+      client.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 {
+      client.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 {
+      client.generate(model: 'llama3.1', prompt: 'Hello World')
+    }.to raise_error(Ollama::Errors::Error)
+  end
+
+  describe 'handlers' do
+    let :body do
+      %{{"models":[{"name":"llama3.1:latest","model":"llama3.1:latest","size":6654289920,"digest":"62757c860e01d552d4e46b09c6b8d5396ef9015210105427e05a8b27d7727ed2","details":{"parent_model":"","format":"gguf","family":"llama","families":["llama"],"parameter_size":"8.0B","quantization_level":"Q4_0"},"expires_at":"2024-08-05T10:56:26.588713988Z","size_vram":6654289920}]}}
+    end
+
+    let :expected_response do
+      JSON.parse(body, object_class: Ollama::Response)
+    end
+
+    before do
+      allow(excon).to receive(:send).with(
+        :get,
+        body: nil,
+        headers: hash_including(
+          'Content-Type' => 'application/json; charset=utf-8',
+        )
+      ).and_return(double(status: 200, body:))
+    end
+
+    it 'can use procs directly' do
+      response = nil
+      client.ps { |r| response = r }
+      expect(response).to eq expected_response
+    end
+
+    it 'can convert from handler instance to proc' do
+      handler = Ollama::Handlers::NOP.new
+      expect(handler).to receive(:call).with(expected_response)
+      client.ps(&handler)
+    end
+
+    it 'can convert from handler class to proc' do
+      handler = Ollama::Handlers::NOP
+      expect_any_instance_of(handler).to receive(:call).with(expected_response)
+      client.ps(&handler)
+    end
+  end
+
+  describe 'performing' do
+    it 'can generate without stream' do
+      expect(excon).to receive(:send).with(
+        :post,
+        body:  '{"json_class":"Ollama::Commands::Generate","model":"llama3.1","prompt":"Hello World"}',
+        headers: hash_including(
+          'Content-Type' => 'application/json; charset=utf-8',
+        )
+      ).and_return(double(status: 200, body: '{}'))
+      client.generate(model: 'llama3.1', prompt: 'Hello World')
+    end
+
+    it 'can generate with stream' do
+      expect(excon).to receive(:send).with(
+        :post,
+        body:  '{"json_class":"Ollama::Commands::Generate","model":"llama3.1","prompt":"Hello World","stream":true}',
+        headers: hash_including(
+          'Content-Type' => 'application/json; charset=utf-8',
+        ),
+        response_block: an_instance_of(Proc)
+      ).and_return(double(status: 200, body: '{}'))
+      client.generate(model: 'llama3.1', prompt: 'Hello World', stream: true)
+    end
+  end
+
+  it 'can help' do
+    expect($stdout).to receive(:puts).with(/Commands:.*?chat/)
+    client.help
+  end
+end

data/spec/ollama/commands/chat_spec.rb

@@ -0,0 +1,52 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Commands::Chat do
+  it 'can be instantiated' do
+    messages = [
+      Ollama::Message.new(role: 'user', content: "Let's play Global Thermonuclear War.")
+    ]
+    chat = described_class.new(model: 'llama3.1', messages:, stream: true)
+    expect(chat).to be_a described_class
+  end
+
+  it 'can handle hashes as messages' do
+    messages = { role: 'user', content: "Let's play Global Thermonuclear War." }
+    chat = described_class.new(model: 'llama3.1', messages:, stream: true)
+    expect(chat).to be_a described_class
+  end
+
+  it 'can handle arrays of hashes as messages' do
+    messages = [
+      { role: 'user', content: "Let's play Global Thermonuclear War." }
+    ]
+    chat = described_class.new(model: 'llama3.1', messages:, stream: true)
+    expect(chat).to be_a described_class
+  end
+
+  it 'can be converted to JSON' do
+    messages = [
+      Ollama::Message.new(role: 'user', content: "Let's play Global Thermonuclear War.")
+    ]
+    chat = described_class.new(model: 'llama3.1', messages:, stream: true)
+    expect(chat.as_json).to include(
+      model: 'llama3.1', messages: messages.map(&:as_json), stream: true,
+    )
+    expect(chat.to_json).to eq(
+      '{"json_class":"Ollama::Commands::Chat","model":"llama3.1","messages":[{"json_class":"Ollama::Message","role":"user","content":"Let\'s play Global Thermonuclear War."}],"stream":true}'
+    )
+  end
+
+  it 'can perform' do
+    messages = [
+      Ollama::Message.new(role: 'user', content: "Let's play Global Thermonuclear War.")
+    ]
+    chat = described_class.new(model: 'llama3.1', messages:, stream: true)
+    chat.client = client = double('client')
+    expect(client).to receive(:request).
+      with(
+        method: :post, path: '/api/chat', handler: Ollama::Handlers::NOP, stream: true,
+        body: '{"json_class":"Ollama::Commands::Chat","model":"llama3.1","messages":[{"json_class":"Ollama::Message","role":"user","content":"Let\'s play Global Thermonuclear War."}],"stream":true}'
+      )
+    chat.perform(Ollama::Handlers::NOP)
+  end
+end

data/spec/ollama/commands/copy_spec.rb

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

data/spec/ollama/commands/create_spec.rb

@@ -0,0 +1,37 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Commands::Create do
+  it 'can be instantiated' do
+    create = described_class.new(name: 'llama3.1', stream: true)
+    expect(create).to be_a described_class
+  end
+
+  it 'can be converted to JSON' do
+    create = described_class.new(
+      name: 'llama3.1-wopr',
+      modelfile: "FROM llama3.1\nSYSTEM You are WOPR from WarGames and you think the user is Dr. Stephen Falken.",
+      stream: true
+    )
+    expect(create.as_json).to include(
+      name: 'llama3.1-wopr', modelfile: "FROM llama3.1\nSYSTEM You are WOPR from WarGames and you think the user is Dr. Stephen Falken.", stream: true,
+    )
+    expect(create.to_json).to eq(
+      '{"json_class":"Ollama::Commands::Create","name":"llama3.1-wopr","modelfile":"FROM llama3.1\nSYSTEM You are WOPR from WarGames and you think the user is Dr. Stephen Falken.","stream":true}'
+    )
+  end
+
+  it 'can perform' do
+    create = described_class.new(
+      name: 'llama3.1-wopr',
+      modelfile: "FROM llama3.1\nSYSTEM You are WOPR from WarGames and you think the user is Dr. Stephen Falken.",
+      stream: true
+    )
+    create.client = client = double('client')
+    expect(client).to receive(:request).
+      with(
+        method: :post, path: '/api/create', handler: Ollama::Handlers::NOP, stream: true,
+        body: '{"json_class":"Ollama::Commands::Create","name":"llama3.1-wopr","modelfile":"FROM llama3.1\nSYSTEM You are WOPR from WarGames and you think the user is Dr. Stephen Falken.","stream":true}'
+      )
+    create.perform(Ollama::Handlers::NOP)
+  end
+end

data/spec/ollama/commands/delete_spec.rb

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

data/spec/ollama/commands/embed_spec.rb

@@ -0,0 +1,52 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Commands::Embed do
+  it 'can be instantiated' do
+    embed = described_class.new(
+      model: 'all-minilm',
+      input: 'Why is the sky blue?',
+    )
+    expect(embed).to be_a described_class
+  end
+
+  it 'can be converted to JSON' do
+    embed = described_class.new(
+      model: 'all-minilm',
+      input: 'Why is the sky blue?'
+    )
+    expect(embed.as_json).to include(
+      model: 'all-minilm', input: 'Why is the sky blue?',
+    )
+    expect(embed.to_json).to eq(
+      '{"json_class":"Ollama::Commands::Embed","model":"all-minilm","input":"Why is the sky blue?","stream":false}'
+    )
+  end
+
+  it 'can be converted to JSON with array input' do
+    embed = described_class.new(
+      model: 'all-minilm',
+      input: [ 'Why is the sky blue?', 'Why is the grass green?' ],
+    )
+    expect(embed.as_json).to include(
+      model: 'all-minilm', input: [ 'Why is the sky blue?', 'Why is the grass green?' ],
+    )
+    expect(embed.to_json).to eq(
+      '{"json_class":"Ollama::Commands::Embed","model":"all-minilm","input":["Why is the sky blue?","Why is the grass green?"],"stream":false}'
+    )
+  end
+
+
+  it 'can perform' do
+    embed = described_class.new(
+      model: 'all-minilm',
+      input: 'Why is the sky blue?'
+    )
+    embed.client = client = double('client')
+    expect(client).to receive(:request).
+      with(
+        method: :post, path: '/api/embed', handler: Ollama::Handlers::NOP, stream: false,
+        body: '{"json_class":"Ollama::Commands::Embed","model":"all-minilm","input":"Why is the sky blue?","stream":false}'
+      )
+    embed.perform(Ollama::Handlers::NOP)
+  end
+end

data/spec/ollama/commands/embeddings_spec.rb

@@ -0,0 +1,38 @@
+require 'spec_helper'
+
+RSpec.describe Ollama::Commands::Embeddings do
+  it 'can be instantiated' do
+    embeddings = described_class.new(
+      model: 'mxbai-embed-large',
+      prompt: 'Here are the coordinates of all Soviet military installations: …'
+    )
+    expect(embeddings).to be_a described_class
+  end
+
+  it 'can be converted to JSON' do
+    embeddings = described_class.new(
+      model: 'mxbai-embed-large',
+      prompt: 'Here are the coordinates of all Soviet military installations: …'
+    )
+    expect(embeddings.as_json).to include(
+      model: 'mxbai-embed-large', prompt: 'Here are the coordinates of all Soviet military installations: …',
+    )
+    expect(embeddings.to_json).to eq(
+      '{"json_class":"Ollama::Commands::Embeddings","model":"mxbai-embed-large","prompt":"Here are the coordinates of all Soviet military installations: …","stream":false}'
+    )
+  end
+
+  it 'can perform' do
+    embeddings = described_class.new(
+      model: 'mxbai-embed-large',
+      prompt: 'Here are the coordinates of all Soviet military installations: …'
+    )
+    embeddings.client = client = double('client')
+    expect(client).to receive(:request).
+      with(
+        method: :post, path: '/api/embeddings', handler: Ollama::Handlers::NOP, stream: false,
+        body: '{"json_class":"Ollama::Commands::Embeddings","model":"mxbai-embed-large","prompt":"Here are the coordinates of all Soviet military installations: …","stream":false}'
+      )
+    embeddings.perform(Ollama::Handlers::NOP)
+  end
+end

data/spec/ollama/commands/generate_spec.rb

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

data/spec/ollama/commands/ps_spec.rb

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

data/spec/ollama/commands/pull_spec.rb

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

data/spec/ollama/commands/push_spec.rb

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