27 lines
1.0 KiB
Ruby
Executable File
27 lines
1.0 KiB
Ruby
Executable File
# frozen_string_literal: true
|
|
|
|
require 'agents'
|
|
|
|
Rails.application.config.after_initialize do
|
|
api_key = InstallationConfig.find_by(name: 'CAPTAIN_OPEN_AI_API_KEY')&.value.presence || ENV.fetch('OPENAI_API_KEY', nil)
|
|
model = InstallationConfig.find_by(name: 'CAPTAIN_OPEN_AI_MODEL')&.value.presence || LlmConstants::DEFAULT_MODEL
|
|
api_endpoint = InstallationConfig.find_by(name: 'CAPTAIN_OPEN_AI_ENDPOINT')&.value || LlmConstants::OPENAI_API_ENDPOINT
|
|
|
|
if api_key.present?
|
|
# Sanitize the key: remove common accidental image suffixes and whitespace
|
|
sanitized_key = api_key.to_s.gsub(/\.(png|jpg|jpeg|gif|webp|svg|@2x|@3x).*$/i, '').strip
|
|
|
|
Agents.configure do |config|
|
|
config.openai_api_key = sanitized_key
|
|
if api_endpoint.present?
|
|
api_base = "#{api_endpoint.chomp('/')}/v1"
|
|
config.openai_api_base = api_base
|
|
end
|
|
config.default_model = model
|
|
config.debug = false
|
|
end
|
|
end
|
|
rescue StandardError => e
|
|
Rails.logger.error "Failed to configure AI Agents SDK: #{e.message}"
|
|
end
|