32 lines
720 B
Ruby
32 lines
720 B
Ruby
# frozen_string_literal: true
|
|
|
|
# Base service for LLM operations using RubyLLM.
|
|
# New features should inherit from this class.
|
|
class Llm::BaseAiService
|
|
DEFAULT_MODEL = Llm::Config::DEFAULT_MODEL
|
|
DEFAULT_TEMPERATURE = 1.0
|
|
|
|
attr_reader :model, :temperature
|
|
|
|
def initialize
|
|
Llm::Config.initialize!
|
|
setup_model
|
|
setup_temperature
|
|
end
|
|
|
|
def chat(model: @model, temperature: @temperature)
|
|
RubyLLM.chat(model: model).with_temperature(temperature)
|
|
end
|
|
|
|
private
|
|
|
|
def setup_model
|
|
config_value = InstallationConfig.find_by(name: 'CAPTAIN_OPEN_AI_MODEL')&.value
|
|
@model = (config_value.presence || DEFAULT_MODEL)
|
|
end
|
|
|
|
def setup_temperature
|
|
@temperature = DEFAULT_TEMPERATURE
|
|
end
|
|
end
|