mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-07-10 08:03:28 +00:00
19 lines
598 B
Ruby
19 lines
598 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
class ConfigurableEmbeddingsPrefixes < ActiveRecord::Migration[7.2]
|
||
|
|
def up
|
||
|
|
add_column :embedding_definitions, :embed_prompt, :string, null: false, default: ""
|
||
|
|
add_column :embedding_definitions, :search_prompt, :string, null: false, default: ""
|
||
|
|
|
||
|
|
# 4 is bge-large-en. Default model and the only one using this so far.
|
||
|
|
execute <<~SQL
|
||
|
|
UPDATE embedding_definitions
|
||
|
|
SET search_prompt='Represent this sentence for searching relevant passages:'
|
||
|
|
WHERE id = 4
|
||
|
|
SQL
|
||
|
|
end
|
||
|
|
|
||
|
|
def down
|
||
|
|
raise ActiveRecord::IrreversibleMigration
|
||
|
|
end
|
||
|
|
end
|