826d9ec5a7
This PR refactors the schema we introduced in #7518 based on the feedback from production tests. Here is the change log - Decouple Inbox association to a new table inbox_response_sources -> this lets us share the same response source between multiple inboxes - Add a status field to responses. This ensures that, by default, responses are created in pending status. You can do quality assurance before making them active. In future, this status can be leveraged by the bot to auto-generate response questions from conversations which require a handoff - Add response_source association to responses and remove hard dependency from response_documents. This lets users write free-form question answers based on conversations, which doesn't necessarily need a response source.
97 lines
2.9 KiB
Ruby
97 lines
2.9 KiB
Ruby
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'
|
|
rescue ActiveRecord::StatementInvalid
|
|
print 'Vector extension not available'
|
|
end
|
|
|
|
def disable_vector_extension
|
|
MIGRATION_VERSION.disable_extension 'vector'
|
|
end
|
|
|
|
def vector_extension_enabled?
|
|
ActiveRecord::Base.connection.extension_enabled?('vector')
|
|
end
|
|
|
|
def create_tables
|
|
return unless vector_extension_enabled?
|
|
|
|
%i[response_sources response_documents responses inbox_response_sources].each do |table|
|
|
send("create_#{table}_table")
|
|
end
|
|
end
|
|
|
|
def drop_tables
|
|
%i[responses response_documents response_sources inbox_response_sources].each do |table|
|
|
MIGRATION_VERSION.drop_table table if MIGRATION_VERSION.table_exists?(table)
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def create_inbox_response_sources_table
|
|
return if MIGRATION_VERSION.table_exists?(:inbox_response_sources)
|
|
|
|
MIGRATION_VERSION.create_table :inbox_response_sources do |t|
|
|
t.references :inbox, null: false
|
|
t.references :response_source, null: false
|
|
t.index [:inbox_id, :response_source_id], name: 'index_inbox_response_sources_on_inbox_id_and_response_source_id', unique: true
|
|
t.index [:response_source_id, :inbox_id], name: 'index_inbox_response_sources_on_response_source_id_and_inbox_id', unique: true
|
|
t.timestamps
|
|
end
|
|
end
|
|
|
|
def create_response_sources_table
|
|
return if MIGRATION_VERSION.table_exists?(:response_sources)
|
|
|
|
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.timestamps
|
|
end
|
|
end
|
|
|
|
def create_response_documents_table
|
|
return if MIGRATION_VERSION.table_exists?(:response_documents)
|
|
|
|
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
|
|
return if MIGRATION_VERSION.table_exists?(:responses)
|
|
|
|
MIGRATION_VERSION.create_table :responses do |t|
|
|
t.bigint :response_source_id, null: false
|
|
t.bigint :response_document_id
|
|
t.string :question, null: false
|
|
t.text :answer, null: false
|
|
t.integer :status, default: 0
|
|
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
|