Class: Rails::Nl2sql::Providers::OpenaiProvider

Inherits:

Base

Object
Base
Rails::Nl2sql::Providers::OpenaiProvider

show all

Defined in:: lib/rails/nl2sql/providers/openai_provider.rb

Instance Method Summary collapse

#complete(prompt:, **params) ⇒ Object
#initialize(api_key:, model: 'gpt-3.5-turbo-instruct') ⇒ OpenaiProvider constructor

A new instance of OpenaiProvider.

Constructor Details

#initialize(api_key:, model: 'gpt-3.5-turbo-instruct') ⇒ `OpenaiProvider`

Returns a new instance of OpenaiProvider.

# File 'lib/rails/nl2sql/providers/openai_provider.rb', line 7

def initialize(api_key:, model: 'gpt-3.5-turbo-instruct')
  @client = ::OpenAI::Client.new(access_token: api_key)
  @model = model
end

Instance Method Details

#complete(prompt:, **params) ⇒ `Object`



12
13
14

# File 'lib/rails/nl2sql/providers/openai_provider.rb', line 12

def complete(prompt:, **params)
  @client.completions(parameters: {model: @model, prompt: prompt}.merge(params))
end