Class: LlmCostTracker::Generators::InstallGenerator

Inherits:
Rails::Generators::Base
  • Object
show all
Includes:
ActiveRecord::Generators::Migration
Defined in:
lib/llm_cost_tracker/generators/llm_cost_tracker/install_generator.rb

Instance Method Summary collapse

Instance Method Details

#create_initializerObject



24
25
26
27
28
29
# File 'lib/llm_cost_tracker/generators/llm_cost_tracker/install_generator.rb', line 24

def create_initializer
  template(
    "initializer.rb.erb",
    "config/initializers/llm_cost_tracker.rb"
  )
end

#create_migration_fileObject



17
18
19
20
21
22
# File 'lib/llm_cost_tracker/generators/llm_cost_tracker/install_generator.rb', line 17

def create_migration_file
  migration_template(
    "create_llm_api_calls.rb.erb",
    "db/migrate/create_llm_api_calls.rb"
  )
end

#create_prices_fileObject



31
32
33
34
35
# File 'lib/llm_cost_tracker/generators/llm_cost_tracker/install_generator.rb', line 31

def create_prices_file
  return unless options[:prices]

  invoke "llm_cost_tracker:prices"
end

#mount_engineObject



37
38
39
40
41
42
# File 'lib/llm_cost_tracker/generators/llm_cost_tracker/install_generator.rb', line 37

def mount_engine
  return unless options[:dashboard]

  add_engine_require
  route %(mount LlmCostTracker::Engine => "/llm-costs")
end