Sha256: f7e4478e90e3047d2c8c7c4c7f2f0be9a62384fb4c772eaaf5cc410b1f81fb20
Contents?: true
Size: 984 Bytes
Versions: 21
Compression:
Stored size: 984 Bytes
Contents
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(
'{"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 = ollama = double('Ollama::Client')
expect(ollama).to receive(:request).
with(
method: :post, path: '/api/generate', handler: Ollama::Handlers::NOP, stream: true,
body: '{"model":"llama3.1","prompt":"Hello World","stream":true}'
)
generate.perform(Ollama::Handlers::NOP)
end
end
Version data entries
21 entries across 21 versions & 1 rubygems