20 lines
643 B
Ruby
20 lines
643 B
Ruby
# frozen_string_literal: true
|
|
class MigratePersonaToLlmModelId < ActiveRecord::Migration[7.2]
|
|
def up
|
|
add_column :ai_personas, :default_llm_id, :bigint
|
|
add_column :ai_personas, :question_consolidator_llm_id, :bigint
|
|
# personas are seeded, we do not mark stuff as readonline
|
|
|
|
execute <<~SQL
|
|
UPDATE ai_personas
|
|
set
|
|
default_llm_id = (select id from llm_models where ('custom:' || id) = default_llm),
|
|
question_consolidator_llm_id = (select id from llm_models where ('custom:' || id) = question_consolidator_llm)
|
|
SQL
|
|
end
|
|
|
|
def down
|
|
raise ActiveRecord::IrreversibleMigration
|
|
end
|
|
end
|