3a4249da11
This PR implements the following features - FAQs from conversations will be generated in account language - Contact notes will be generated in account language - Copilot chat will respond in user language, unless the agent asks the question in a different language ## Changes ### Copilot Chat - Update the prompt to include an instruction for the language, the bot will reply in asked language, but will default to account language - Update the `ChatService` class to include pass the language to `SystemPromptsService` ### FAQ and Contact note generation - Update contact note generator and conversation generator to include account locale - Pass the account locale to `SystemPromptsService` <details><summary>Screenshots</summary> #### FAQs being generated in system langauge  #### Copilot responding in system language  </details> --------- Co-authored-by: Muhsin Keloth <muhsinkeramam@gmail.com> Co-authored-by: Pranav <pranav@chatwoot.com>
89 lines
3.9 KiB
Ruby
89 lines
3.9 KiB
Ruby
require 'rails_helper'
|
|
|
|
RSpec.describe Captain::Copilot::ChatService do
|
|
let(:account) { create(:account, custom_attributes: { plan_name: 'startups' }) }
|
|
let(:captain_inbox_association) { create(:captain_inbox, captain_assistant: assistant, inbox: inbox) }
|
|
let(:mock_captain_agent) { instance_double(Captain::Agent) }
|
|
let(:mock_captain_tool) { instance_double(Captain::Tool) }
|
|
let(:mock_openai_client) { instance_double(OpenAI::Client) }
|
|
let(:inbox) { create(:inbox, account: account) }
|
|
let(:assistant) { create(:captain_assistant, account: account) }
|
|
|
|
before do
|
|
create(:installation_config, name: 'CAPTAIN_OPEN_AI_API_KEY', value: 'test-key')
|
|
end
|
|
|
|
describe '#initialize' do
|
|
it 'sets default language to english when not specified' do
|
|
service = described_class.new(assistant, { previous_messages: [], conversation_history: '' })
|
|
expect(service.instance_variable_get(:@language)).to eq('english')
|
|
end
|
|
|
|
it 'uses the specified language when provided' do
|
|
service = described_class.new(assistant, {
|
|
previous_messages: [],
|
|
conversation_history: '',
|
|
language: 'spanish'
|
|
})
|
|
expect(service.instance_variable_get(:@language)).to eq('spanish')
|
|
end
|
|
end
|
|
|
|
describe '#generate_response' do
|
|
before do
|
|
allow(OpenAI::Client).to receive(:new).and_return(mock_openai_client)
|
|
allow(mock_openai_client).to receive(:chat).and_return({ choices: [{ message: { content: '{ "result": "Hey" }' } }] }.with_indifferent_access)
|
|
|
|
allow(Captain::Agent).to receive(:new).and_return(mock_captain_agent)
|
|
allow(mock_captain_agent).to receive(:execute).and_return(true)
|
|
allow(mock_captain_agent).to receive(:register_tool).and_return(true)
|
|
|
|
allow(Captain::Tool).to receive(:new).and_return(mock_captain_tool)
|
|
allow(mock_captain_tool).to receive(:register_method).and_return(true)
|
|
|
|
allow(account).to receive(:increment_response_usage).and_return(true)
|
|
end
|
|
|
|
it 'increments usage' do
|
|
described_class.new(assistant, { previous_messages: ['Hello'], conversation_history: 'Hi' }).generate_response('Hey')
|
|
expect(account).to have_received(:increment_response_usage).once
|
|
end
|
|
|
|
it 'includes language in system message' do
|
|
service = described_class.new(assistant, {
|
|
previous_messages: [],
|
|
conversation_history: '',
|
|
language: 'spanish'
|
|
})
|
|
|
|
allow(Captain::Llm::SystemPromptsService).to receive(:copilot_response_generator)
|
|
.with(assistant.config['product_name'], 'spanish')
|
|
.and_return('Spanish system prompt')
|
|
|
|
system_message = service.send(:system_message)
|
|
expect(system_message[:content]).to eq('Spanish system prompt')
|
|
end
|
|
end
|
|
|
|
describe '#execute' do
|
|
before do
|
|
allow(OpenAI::Client).to receive(:new).and_return(mock_openai_client)
|
|
allow(mock_openai_client).to receive(:chat).and_return({ choices: [{ message: { content: '{ "result": "Hey" }' } }] }.with_indifferent_access)
|
|
|
|
allow(Captain::Agent).to receive(:new).and_return(mock_captain_agent)
|
|
allow(mock_captain_agent).to receive(:execute).and_return(true)
|
|
allow(mock_captain_agent).to receive(:register_tool).and_return(true)
|
|
|
|
allow(Captain::Tool).to receive(:new).and_return(mock_captain_tool)
|
|
allow(mock_captain_tool).to receive(:register_method).and_return(true)
|
|
|
|
allow(account).to receive(:increment_response_usage).and_return(true)
|
|
end
|
|
|
|
it 'increments usage' do
|
|
described_class.new(assistant, { previous_messages: ['Hello'], conversation_history: 'Hi' }).generate_response('Hey')
|
|
expect(account).to have_received(:increment_response_usage).once
|
|
end
|
|
end
|
|
end
|