Class: ActiveHarness::Providers::Mistral

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

Overview

Mistral AI — OpenAI-compatible API. docs.mistral.ai/api

Constant Summary

Constants inherited from Base

Base::HTTP, Base::STREAMING_HTTP

Instance Method Summary collapse

Instance Method Details

#call(model:, messages:, temperature: 0.7) ⇒ Object 



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
 
# File 'lib/active_harness/providers/mistral.rb', line 8

def call(model:, messages:, temperature: 0.7)
  raw  = post_json(URI(config.mistral_api_url),
    headers: {
      "Content-Type"  => "application/json",
      "Authorization" => "Bearer #{api_key}"
    },
    body: { model: model, messages: messages, temperature: temperature }
  )
  data = parse!(raw)
  handle_error!(data)

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