Class: SmartPrompt::Engine

Inherits:
Object
  • Object
show all
Defined in:
lib/smart_prompt/engine.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(config_file) ⇒ Engine

Returns a new instance of Engine.



4
5
6
7
8
9
10
11
 
# File 'lib/smart_prompt/engine.rb', line 4

def initialize(config_file)
  @config_file = config_file
  @adapters={}
  @llms={}
  @templates={}
  load_config(config_file)
  SmartPrompt.logger.info "Started create the SmartPrompt engine."
end

Instance Attribute Details

#adapters ⇒ Object (readonly)

Returns the value of attribute adapters.



3
4
5
 
# File 'lib/smart_prompt/engine.rb', line 3

def adapters
  @adapters
end

#config ⇒ Object (readonly)

Returns the value of attribute config.



3
4
5
 
# File 'lib/smart_prompt/engine.rb', line 3

def config
  @config
end

#config_file ⇒ Object (readonly)

Returns the value of attribute config_file.



3
4
5
 
# File 'lib/smart_prompt/engine.rb', line 3

def config_file
  @config_file
end

#current_adapter ⇒ Object (readonly)

Returns the value of attribute current_adapter.



3
4
5
 
# File 'lib/smart_prompt/engine.rb', line 3

def current_adapter
  @current_adapter
end

#llms ⇒ Object (readonly)

Returns the value of attribute llms.



3
4
5
 
# File 'lib/smart_prompt/engine.rb', line 3

def llms
  @llms
end

#templates ⇒ Object (readonly)

Returns the value of attribute templates.



3
4
5
 
# File 'lib/smart_prompt/engine.rb', line 3

def templates
  @templates
end

Instance Method Details

#call_worker(worker_name, params = {}) ⇒ Object 



64
65
66
67
68
69
70
71
72
73
74
75
76
77
 
# File 'lib/smart_prompt/engine.rb', line 64

def call_worker(worker_name, params = {})
  SmartPrompt.logger.info "Calling worker: #{worker_name} with params: #{params}"
  worker = get_worker(worker_name)
  
  begin
    result = worker.execute(params)
    SmartPrompt.logger.info "Worker #{worker_name} executed successfully"
    result
  rescue => e
    SmartPrompt.logger.error "Error executing worker #{worker_name}: #{e.message}"
    SmartPrompt.logger.debug e.backtrace.join("\n")
    raise
  end
end

#check_worker(worker_name) ⇒ Object 



55
56
57
58
59
60
61
62
 
# File 'lib/smart_prompt/engine.rb', line 55

def check_worker(worker_name)
  if SmartPrompt::Worker.workers[worker_name]
    return true
  else
    SmartPrompt.logger.warn "Invalid worker: #{worker_name}"
    return false
  end
end

#load_config(config_file) ⇒ Object 



13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
 
# File 'lib/smart_prompt/engine.rb', line 13

def load_config(config_file)
  begin
    @config_file = config_file
    @config = YAML.load_file(config_file)
    if @config['logger_file']
      SmartPrompt.logger = Logger.new(@config['logger_file'])
    end
    SmartPrompt.logger.info "Loading configuration from file: #{config_file}"
    @config['adapters'].each do |adapter_name, adapter_class|
      adapter_class = SmartPrompt.const_get(adapter_class)
      @adapters[adapter_name] = adapter_class
    end
    @config['llms'].each do |llm_name,llm_config|
      adapter_class = @adapters[llm_config['adapter']]
      @llms[llm_name]=adapter_class.new(llm_config)
    end
    @current_llm = @config['default_llm'] if @config['default_llm']
    Dir.glob(File.join(@config['template_path'], '*.erb')).each do |file|
      template_name = file.gsub(@config['template_path']+"/","").gsub("\.erb","")
      @templates[template_name] = PromptTemplate.new(file)
    end
    load_workers          
  rescue Psych::SyntaxError => ex
    SmartPrompt.logger.error "YAML syntax error in config file: #{ex.message}"
    raise ConfigurationError, "Invalid YAML syntax in config file: #{ex.message}"
  rescue Errno::ENOENT => ex
    SmartPrompt.logger.error "Config file not found: #{ex.message}"
    raise ConfigurationError, "Config file not found: #{ex.message}"
  rescue StandardError => ex
    SmartPrompt.logger.error "Error loading configuration: #{ex.message}"
    raise ConfigurationError, "Error loading configuration: #{ex.message}"
  ensure
    SmartPrompt.logger.info "Configuration loaded successfully"          
  end        
end

#load_workers ⇒ Object 



49
50
51
52
53
 
# File 'lib/smart_prompt/engine.rb', line 49

def load_workers
  Dir.glob(File.join(@config['worker_path'], '*.rb')).each do |file|
    require(file)
  end
end