Class: Inferkit::Client

Inherits:
Object
  • Object
show all
Defined in:
lib/inferkit/client.rb

Constant Summary collapse

HOST = 
'https://api.inferkit.com'
GENERATE_ENDPOINT = 
'/v1/models/standard/generate'

Instance Method Summary collapse

Instance Method Details

#generate(prompt, **params) ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/inferkit/client.rb', line 10

def generate(prompt, **params)
  request_params = default_params.merge(params)

  response =
    conn.post(GENERATE_ENDPOINT) do |req|
      req.headers['Content-Type'] = 'application/json'
      req.headers['Authorization'] =
        "Bearer #{Inferkit.configuration.api_key}"
      req.body = generate_body(prompt, request_params)
    end

  Oj.load(response.body, symbol_keys: true).dig(:data, :text)
end