23 lines
673 B
Ruby
23 lines
673 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
class AddProviderToCompletionPrompts < ActiveRecord::Migration[7.0]
|
||
|
|
def up
|
||
|
|
remove_index :completion_prompts, name: "index_completion_prompts_on_name"
|
||
|
|
add_column :completion_prompts, :provider, :text
|
||
|
|
add_index :completion_prompts, %i[name], unique: false
|
||
|
|
|
||
|
|
# set provider for existing prompts
|
||
|
|
DB.exec <<~SQL
|
||
|
|
UPDATE completion_prompts
|
||
|
|
SET provider = 'openai'
|
||
|
|
WHERE provider IS NULL;
|
||
|
|
SQL
|
||
|
|
end
|
||
|
|
|
||
|
|
def down
|
||
|
|
remove_column :completion_prompts, :provider
|
||
|
|
remove_index :completion_prompts, name: "index_completion_prompts_on_name"
|
||
|
|
add_index :completion_prompts, %i[name], unique: true
|
||
|
|
end
|
||
|
|
end
|