Class: SmartBrain::ModelProvider::OpenAI

Inherits:
Base
  • Object
show all
Includes:
Rerankable
Defined in:
lib/smart_brain/model_provider/openai.rb

Overview

通用 OpenAI 兼容 provider（覆盖 silicon_flow 等任意 /v1/chat/completions 端点）。

Instance Method Summary collapse

Methods included from Rerankable

#rerank

Methods inherited from Base

#rerank

Constructor Details

#initialize(model:, base_url:, api_key:, rerank_model: nil, temperature: 0.2, timeout_seconds: 30, **_) ⇒ OpenAI

Returns a new instance of OpenAI.



13
14
15
16
17
18
19
20
 
# File 'lib/smart_brain/model_provider/openai.rb', line 13

def initialize(model:, base_url:, api_key:, rerank_model: nil, temperature: 0.2, timeout_seconds: 30, **_)
  @model = model
  @rerank_model = rerank_model || model
  @temperature = temperature
  @timeout_seconds = timeout_seconds
  @api_key = api_key
  @base_url = base_url.to_s.chomp('/')
end

Instance Method Details

#complete(prompt:, system: nil, temperature: nil, max_tokens: nil) ⇒ Object 



30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
 
# File 'lib/smart_brain/model_provider/openai.rb', line 30

def complete(prompt:, system: nil, temperature: nil, max_tokens: nil)
  messages = []
  messages << { 'role' => 'system', 'content' => system } if system && !system.empty?
  messages << { 'role' => 'user', 'content' => prompt }

  body = { 'model' => @model, 'messages' => messages, 'temperature' => temperature || @temperature }
  body['max_tokens'] = max_tokens if max_tokens

  response = connection.post('/v1/chat/completions') do |req|
    req.headers['Authorization'] = "Bearer #{@api_key}" unless @api_key.to_s.empty?
    req.body = JSON.generate(body)
  end
  raise "openai HTTP #{response.status}: #{response.body[0, 200]}" unless response.status.between?(200, 299)

  { text: strip_think(JSON.parse(response.body).dig('choices', 0, 'message', 'content').to_s) }
rescue StandardError => e
  { text: '', error: "#{e.class}: #{e.message}" }
end

#enabled? ⇒ Boolean

Returns:

  • (Boolean) 


26
27
28
 
# File 'lib/smart_brain/model_provider/openai.rb', line 26

def enabled?
  !@api_key.to_s.empty?
end

#llm? ⇒ Boolean

Returns:

  • (Boolean) 


22
23
24
 
# File 'lib/smart_brain/model_provider/openai.rb', line 22

def llm?
  true
end