Currently, it’s unclear whether an FAQ item is generated from a document, derived from a conversation, or added manually. This PR resolves the issue by providing visibility into the source of each FAQ. Users can now see whether an FAQ was generated or manually added and, if applicable, by whom. - Move the document_id to a polymorphic relation (documentable). - Updated the APIs to accommodate the change. - Update the service to add corresponding references. - Updated the specs. <img width="1007" alt="Screenshot 2025-01-15 at 11 27 56 PM" src="https://github.com/user-attachments/assets/7d58f798-19c0-4407-b3e2-748a919d14af" /> --------- Co-authored-by: Sivin Varghese <64252451+iamsivin@users.noreply.github.com>
48 lines
1.6 KiB
Ruby
48 lines
1.6 KiB
Ruby
require 'rails_helper'
|
|
|
|
RSpec.describe Captain::Documents::ResponseBuilderJob, type: :job do
|
|
let(:assistant) { create(:captain_assistant) }
|
|
let(:document) { create(:captain_document, assistant: assistant) }
|
|
let(:faq_generator) { instance_double(Captain::Llm::FaqGeneratorService) }
|
|
let(:faqs) do
|
|
[
|
|
{ 'question' => 'What is Ruby?', 'answer' => 'A programming language' },
|
|
{ 'question' => 'What is Rails?', 'answer' => 'A web framework' }
|
|
]
|
|
end
|
|
|
|
before do
|
|
allow(Captain::Llm::FaqGeneratorService).to receive(:new)
|
|
.with(document.content)
|
|
.and_return(faq_generator)
|
|
allow(faq_generator).to receive(:generate).and_return(faqs)
|
|
end
|
|
|
|
describe '#perform' do
|
|
context 'when processing a document' do
|
|
it 'deletes previous responses' do
|
|
existing_response = create(:captain_assistant_response, documentable: document)
|
|
|
|
described_class.new.perform(document)
|
|
|
|
expect { existing_response.reload }.to raise_error(ActiveRecord::RecordNotFound)
|
|
end
|
|
|
|
it 'creates new responses for each FAQ' do
|
|
expect do
|
|
described_class.new.perform(document)
|
|
end.to change(Captain::AssistantResponse, :count).by(2)
|
|
|
|
responses = document.responses.reload
|
|
expect(responses.count).to eq(2)
|
|
|
|
first_response = responses.first
|
|
expect(first_response.question).to eq('What is Ruby?')
|
|
expect(first_response.answer).to eq('A programming language')
|
|
expect(first_response.assistant).to eq(assistant)
|
|
expect(first_response.documentable).to eq(document)
|
|
end
|
|
end
|
|
end
|
|
end
|