Contents

require 'webmock/rspec'
require_relative '../lib/rubyai/client.rb'

RSpec.describe RubyAI::Client do
  let(:api_key) { 'your_api_key' }
  let(:messages) { 'Hello, how are you?' }
  let(:temperature) { 0.7 }
  let(:model) { 'gpt-3.5-turbo' }
  let(:client) { described_class.new(api_key: api_key, messages: messages, temperature: temperature, model: model) }

  describe '#call' do
    let(:response_body) { { 'completion' => 'This is a response from the model.' } }
    let(:status) { 200 }

    before do
      stub_request(:post, RubyAI::Configuration::BASE_URL)
        .to_return(status: status, body: response_body.to_json, headers: { 'Content-Type' => 'application/json' })
    end

    it 'returns parsed JSON response when passing through client directly' do
      expect(client.call).to eq(response_body)
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
rubyai-0.5	spec/client_spec.rb