Class: Rails::Nl2sql::Providers::AnthropicProvider

Inherits:
Base
  • Object
show all
Defined in:
lib/rails/nl2sql/providers/anthropic_provider.rb

Instance Method Summary collapse

Constructor Details

#initialize(api_key:, model: 'claude-3-opus-20240229') ⇒ AnthropicProvider

Returns a new instance of AnthropicProvider.



11
12
13
14
15
 
# File 'lib/rails/nl2sql/providers/anthropic_provider.rb', line 11

def initialize(api_key:, model: 'claude-3-opus-20240229')
  raise 'anthropic gem missing' unless defined?(::Anthropic::Client)
  @client = ::Anthropic::Client.new(api_key: api_key)
  @model = model
end

Instance Method Details

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



17
18
19
 
# File 'lib/rails/nl2sql/providers/anthropic_provider.rb', line 17

def complete(prompt:, **params)
  @client.completions(model: @model, prompt: prompt, **params)
end