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



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

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

#create_migration_fileObject



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

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

#create_prices_fileObject



33
34
35
36
37
# File 'lib/llm_cost_tracker/generators/llm_cost_tracker/install_generator.rb', line 33

def create_prices_file
  return unless options[:prices]

  invoke "llm_cost_tracker:prices"
end

#mount_engineObject



39
40
41
42
43
44
45
# File 'lib/llm_cost_tracker/generators/llm_cost_tracker/install_generator.rb', line 39

def mount_engine
  return unless options[:dashboard]

  add_engine_require
  route %(mount LlmCostTracker::Engine => "/llm-costs")
  say "Mount /llm-costs behind your app's admin auth before deploying.", :yellow
end