FIX: Explicit check for empty string in compat migration (#463)
This commit is contained in:
parent
9168c75eb6
commit
0ff5c0c2c4
|
@ -0,0 +1,47 @@
|
||||||
|
# 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
|
Loading…
Reference in New Issue