Class: ActiveHarness::Providers::OpenAI

Inherits:
Base
  • Object
show all
Defined in:
lib/active_harness/providers/openai.rb

Constant Summary

Constants inherited from Base

Base::HTTP, Base::STREAMING_HTTP

Instance Method Summary collapse

Instance Method Details

#call(model:, messages:, temperature: 0.7, stream: nil) ⇒ Hash

Returns { content:, provider:, model: }.

Parameters:

  • model (String)
  • messages (Array<Hash>) —
    content:, …
  • temperature (Float) (defaults to: 0.7)

Returns:

  • (Hash) —

    { content:, provider:, model: }



10
11
12
13
14
15
16
17
18
19
20
21
 
# File 'lib/active_harness/providers/openai.rb', line 10

def call(model:, messages:, temperature: 0.7, stream: nil)
  headers = { "Content-Type" => "application/json", "Authorization" => "Bearer #{api_key}" }
  body    = { model: model, messages: messages, temperature: temperature }

  return call_streaming(url: config.openai_api_url, headers: headers, body: body, stream: stream, provider: :openai, model: model) if stream

  raw  = post_json(URI(config.openai_api_url), headers: headers, body: body)
  data = parse!(raw)
  handle_error!(data)

  { content: data.dig("choices", 0, "message", "content").to_s.strip, provider: :openai, model: data["model"] || model, usage: extract_usage_openai(data) }
end