discourse-ai/db/migrate/20250122131007_matryoshka_dimensions_support.rb

# frozen_string_literal: true
class MatryoshkaDimensionsSupport < ActiveRecord::Migration[7.2]
  def change
    add_column :embedding_definitions, :matryoshka_dimensions, :boolean, null: false, default: false

    execute <<~SQL
      UPDATE embedding_definitions
      SET matryoshka_dimensions = TRUE
      WHERE 
        provider = 'open_ai' AND 
        provider_params IS NOT NULL AND
        (
          (provider_params->>'model_name') = 'text-embedding-3-large' OR
          (provider_params->>'model_name') = 'text-embedding-3-small'
        )
    SQL
  end

  def down
    raise ActiveRecord::IrreversibleMigration
  end
end