48 lines
1.5 KiB
Ruby
48 lines
1.5 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
# Some sites defaults weren't migrated correctly due to the previous migration
|
||
|
|
# using the value as the if condition instead of checking with String#empty?
|
||
|
|
class FixLlmBackedSettingDefaults < ActiveRecord::Migration[7.0]
|
||
|
|
def up
|
||
|
|
backfill_settings("composer_ai_helper_enabled", "ai_helper_model")
|
||
|
|
backfill_settings(
|
||
|
|
"ai_embeddings_semantic_search_enabled",
|
||
|
|
"ai_embeddings_semantic_search_hyde_model",
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
def down
|
||
|
|
raise ActiveRecord::IrreversibleMigration
|
||
|
|
end
|
||
|
|
|
||
|
|
def backfill_settings(feature_setting_name, llm_setting_name)
|
||
|
|
feature_enabled =
|
||
|
|
DB.query_single(
|
||
|
|
"SELECT value FROM site_settings WHERE name = :setting_name",
|
||
|
|
setting_name: feature_setting_name,
|
||
|
|
).first == "t"
|
||
|
|
|
||
|
|
setting_value =
|
||
|
|
DB
|
||
|
|
.query_single(
|
||
|
|
"SELECT value FROM site_settings WHERE name = :llm_setting",
|
||
|
|
llm_setting: llm_setting_name,
|
||
|
|
)
|
||
|
|
.first
|
||
|
|
.to_s
|
||
|
|
using_old_default = setting_value.empty?
|
||
|
|
|
||
|
|
providers = %w[aws_bedrock anthropic open_ai hugging_face vllm google]
|
||
|
|
# Sanity check to make sure we won't add provider twice.
|
||
|
|
return if providers.include?(setting_value.split(":").first)
|
||
|
|
|
||
|
|
if using_old_default && feature_enabled
|
||
|
|
# Enabled and using old default (gpt-3.5-turbo)
|
||
|
|
DB.exec(<<~SQL, llm_setting: llm_setting_name, default: "open_ai:gpt-3.5-turbo")
|
||
|
|
INSERT INTO site_settings(name, data_type, value, created_at, updated_at)
|
||
|
|
VALUES (:llm_setting, 1, :default, NOW(), NOW())
|
||
|
|
SQL
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|