Class: Raif::Generators::AgentGenerator

Inherits:
BaseGenerator
  • Object
show all
Defined in:
lib/generators/raif/agent/agent_generator.rb

Instance Method Summary collapse

Instance Method Details

#create_agent ⇒ Object 



25
26
27
 
# File 'lib/generators/raif/agent/agent_generator.rb', line 25

def create_agent
  template "agent.rb.tt", File.join("app/models/raif/agents", class_path, "#{file_name}.rb")
end

#create_application_agent ⇒ Object 



21
22
23
 
# File 'lib/generators/raif/agent/agent_generator.rb', line 21

def create_application_agent
  template "application_agent.rb.tt", "app/models/raif/application_agent.rb" unless File.exist?("app/models/raif/application_agent.rb")
end

#create_directory ⇒ Object 



35
36
37
 
# File 'lib/generators/raif/agent/agent_generator.rb', line 35

def create_directory
  empty_directory "app/models/raif/agents" unless File.directory?("app/models/raif/agents")
end

#create_eval_set ⇒ Object 



39
40
41
42
43
 
# File 'lib/generators/raif/agent/agent_generator.rb', line 39

def create_eval_set
  return if options[:skip_eval_set]

  template "agent_eval_set.rb.tt", eval_set_file_path
end

#create_system_prompt_template ⇒ Object 



29
30
31
32
33
 
# File 'lib/generators/raif/agent/agent_generator.rb', line 29

def create_system_prompt_template
  return if options[:skip_prompt_template]

  template "system_prompt.erb.tt", system_prompt_template_file_path
end

#show_instructions ⇒ Object 



45
46
47
48
49
50
51
 
# File 'lib/generators/raif/agent/agent_generator.rb', line 45

def show_instructions
  say "\nAgent created!"
  unless options[:skip_prompt_template]
    say "  System prompt template: #{system_prompt_template_file_path}"
  end
  say ""
end