chore: fixes

chatwoot · sojan-official · Jul 21, 2023 · May 22, 2023 · May 29, 2023 · Mar 20, 2023
commit dfa9b27bec1226fd1b1d95b41bd064820df272fd
diff --git a/app/views/api/v1/accounts/response_sources/remove_document.json.jbuilder b/app/views/api/v1/accounts/response_sources/remove_document.json.jbuilder
@@ -1 +1 @@
-json.partial! 'api/v1/models/response_source', formats: [:json], resource: @response_source
+json.partial! 'api/v1/models/response_source', formats: [:json], resource: @response_source
diff --git a/config/initializers/monkey_patches/schema_dumper.rb b/config/initializers/monkey_patches/schema_dumper.rb
@@ -0,0 +1,35 @@
+# When working with experimental extensions, which doesn't have support on all providers
+# This monkey patch will help us to ignore the extensions when dumping the schema
+# Additionally we will also ignore the tables associated with those features and exentions
+
+# Once the feature stabilizes, we can remove the tables/extension from the ignore list
+# Ensure you write appropriate migrations when you do that.
+
+module ActiveRecord
+  module ConnectionAdapters
+    module PostgreSQL
+      class SchemaDumper < ConnectionAdapters::SchemaDumper
+        cattr_accessor :ignore_extentions, default: []
+
+        private
+
+        def extensions(stream)
+          extensions = @connection.extensions
+          return unless extensions.any?
+
+          stream.puts '  # These are extensions that must be enabled in order to support this database'
+          extensions.sort.each do |extension|
+            stream.puts "  enable_extension #{extension.inspect}" unless ignore_extentions.include?(extension)
+          end
+          stream.puts
+        end
+      end
+    end
+  end
+end
+
+## Extentions / Tables to be ignored
+ActiveRecord::ConnectionAdapters::PostgreSQL::SchemaDumper.ignore_extentions << 'vector'
+ActiveRecord::SchemaDumper.ignore_tables << 'responses'
+ActiveRecord::SchemaDumper.ignore_tables << 'response_sources'
+ActiveRecord::SchemaDumper.ignore_tables << 'response_documents'
diff --git a/db/migrate/20230509122602_enable_pgvector_extension.rb b/db/migrate/20230509122602_enable_pgvector_extension.rb
diff --git a/db/migrate/20230709113830_create_responses.rb b/db/migrate/20230709113830_create_responses.rb
diff --git a/db/schema.rb b/db/schema.rb
@@ -16,7 +16,6 @@
   enable_extension "pg_trgm"
   enable_extension "pgcrypto"
   enable_extension "plpgsql"
-  enable_extension "vector"
 
   create_table "access_tokens", force: :cascade do |t|
     t.string "owner_type"
@@ -814,44 +813,6 @@
     t.index ["user_id"], name: "index_reporting_events_on_user_id"
   end
 
-  create_table "response_documents", force: :cascade do |t|
-    t.bigint "response_source_id", null: false
-    t.string "document_link"
-    t.string "document_type"
-    t.bigint "document_id"
-    t.text "content"
-    t.bigint "account_id", null: false
-    t.datetime "created_at", null: false
-    t.datetime "updated_at", null: false
-    t.index ["document_type", "document_id"], name: "index_response_documents_on_document"
-    t.index ["response_source_id"], name: "index_response_documents_on_response_source_id"
-  end
-
-  create_table "response_sources", force: :cascade do |t|
-    t.integer "source_type", default: 0, null: false
-    t.string "name", null: false
-    t.string "source_link"
-    t.string "source_model_type"
-    t.bigint "source_model_id"
-    t.bigint "account_id", null: false
-    t.bigint "inbox_id", null: false
-    t.datetime "created_at", null: false
-    t.datetime "updated_at", null: false
-    t.index ["source_model_type", "source_model_id"], name: "index_response_sources_on_source_model"
-  end
-
-  create_table "responses", force: :cascade do |t|
-    t.bigint "response_document_id"
-    t.string "question", null: false
-    t.text "answer", null: false
-    t.bigint "account_id", null: false
-    t.vector "embedding", limit: 1536
-    t.datetime "created_at", null: false
-    t.datetime "updated_at", null: false
-    t.index ["embedding"], name: "index_responses_on_embedding", using: :ivfflat
-    t.index ["response_document_id"], name: "index_responses_on_response_document_id"
-  end
-
   create_table "sla_policies", force: :cascade do |t|
     t.string "name", null: false
     t.float "frt_threshold"

diff --git a/enterprise/app/services/features/response_bot_service.rb b/enterprise/app/services/features/response_bot_service.rb
@@ -0,0 +1,76 @@
+class Features::ResponseBotService
+  MIGRATION_VERSION = ActiveRecord::Migration[7.0]
+
+  def enable_in_installation
+    enable_vector_extension
+    create_tables
+  end
+
+  def enable_vector_extension
+    MIGRATION_VERSION.enable_extension 'vector'
+  end
+
+  def disable_vector_extension
+    MIGRATION_VERSION.disable_extension 'vector'
+  end
+
+  def vector_extension_enabled?
+    raise 'Vector extension not available' unless ActiveRecord::Base.connection.extension_enabled?('vector')
+
+    true
+  end
+
+  def create_tables
+    vector_extension_enabled?
+    %i[response_sources response_documents responses].each do |table|
+      send("create_#{table}_table")
+    end
+  end
+
+  def drop_tables
+    %i[responses response_documents response_sources].each do |table|
+      MIGRATION_VERSION.drop_table table
+    end
+  end
+
+  private
+
+  def create_response_sources_table
+    MIGRATION_VERSION.create_table :response_sources do |t|
+      t.integer :source_type, null: false, default: 0
+      t.string :name, null: false
+      t.string :source_link
+      t.references :source_model, polymorphic: true
+      t.bigint :account_id, null: false
+      t.bigint :inbox_id, null: false
+      t.timestamps
+    end
+  end
+
+  def create_response_documents_table
+    MIGRATION_VERSION.create_table :response_documents do |t|
+      t.bigint :response_source_id, null: false
+      t.string :document_link
+      t.references :document, polymorphic: true
+      t.text :content
+      t.bigint :account_id, null: false
+      t.timestamps
+    end
+
+    MIGRATION_VERSION.add_index :response_documents, :response_source_id
+  end
+
+  def create_responses_table
+    MIGRATION_VERSION.create_table :responses do |t|
+      t.bigint :response_document_id
+      t.string :question, null: false
+      t.text :answer, null: false
+      t.bigint :account_id, null: false
+      t.vector :embedding, limit: 1536
+      t.timestamps
+    end
+
+    MIGRATION_VERSION.add_index :responses, :response_document_id
+    MIGRATION_VERSION.add_index :responses, :embedding, using: :ivfflat, opclass: :vector_l2_ops
+  end
+end
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		json.partial! 'api/v1/models/response_source', formats: [:json], resource: @response_source
		json.partial! 'api/v1/models/response_source', formats: [:json], resource: @response_source