chore(CE): embedding model changes (#600)

Co-authored-by: afthab vp <[email protected]>

Author: github-actions[bot]

server/app/models/embedding_model.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+class EmbeddingModel < ApplicationRecord
+  enum status: { inactive: 0, active: 1 }
+
+  validates :mode, presence: true
+  validates :models, presence: true
+end

server/db/data/20250207090028_create_embedding_model_openai.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+class CreateEmbeddingModelOpenai < ActiveRecord::Migration[7.1]
+  def up
+    EmbeddingModel.create!(
+      mode: "openai",
+      models: [
+        "text-embedding-3-small",
+        "text-embedding-3-large",
+        "text-embedding-ada-002"
+      ]
+    )
+  end
+
+  def down
+    EmbeddingModel.where(mode: "openai").destroy_all
+  end
+end

server/db/data_schema.rb

@@ -1,3 +1,3 @@
 # frozen_string_literal: true
 
-DataMigrate::Data.define(version: 20_250_206_101_336)
+DataMigrate::Data.define(version: 20_250_207_090_028)

server/db/migrate/20250207084006_create_embedding_models.rb

@@ -0,0 +1,12 @@
+class CreateEmbeddingModels < ActiveRecord::Migration[7.1]
+  def change
+    create_table :embedding_models do |t|
+      t.string :mode, null: false
+      t.integer :status, default: 1
+      t.string :models, null: false, array: true, default: []
+
+      t.timestamps
+    end
+
+  end
+end

server/db/schema.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :embedding_model do
+    mode { "openai" }
+    status { 1 }
+    models { ["text-embedding-3-small", "text-embedding-3-large"] }
+
+    trait :inactive do
+      status { 0 }
+    end
+
+    trait :with_custom_models do
+      models { ["text-embedding-ada-002"] }
+    end
+  end
+end