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 



71
72
73
74
75
76
77
78
79
80
81
82
83
 
# File 'lib/smart_prompt/engine.rb', line 71

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

#call_worker_by_stream(worker_name, params = {}, &proc) ⇒ Object 



85
86
87
88
89
90
91
92
93
94
95
96
 
# File 'lib/smart_prompt/engine.rb', line 85

def call_worker_by_stream(worker_name, params = {}, &proc)
  SmartPrompt.logger.info "Calling worker: #{worker_name} with params: #{params}"
  worker = get_worker(worker_name)        
  begin
    worker.execute_by_stream(params, &proc)
    SmartPrompt.logger.info "Worker #{worker_name} executed(stream) successfully"          
  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 



62
63
64
65
66
67
68
69
 
# File 'lib/smart_prompt/engine.rb', line 62

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

#create_dir(filename) ⇒ Object 



13
14
15
16
17
 
# File 'lib/smart_prompt/engine.rb', line 13

def create_dir(filename)
  path = File::path(filename).to_s
  parent_dir = File::dirname(path) 
  Dir.mkdir(parent_dir, 0755) unless File.directory?(parent_dir)
end

#load_config(config_file) ⇒ Object 



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
48
49
50
51
52
53
54
 
# File 'lib/smart_prompt/engine.rb', line 19

def load_config(config_file)
  begin
    @config_file = config_file
    @config = YAML.load_file(config_file)
    if @config['logger_file']
      create_dir(@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 



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

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