From 9f2076664ea33d574f8e01dde696e027b7ae7a19 Mon Sep 17 00:00:00 2001 From: aakashb95 Date: Mon, 4 May 2026 17:14:57 +0530 Subject: [PATCH] test(captain): simplify classifier model spec --- .../llm/assistant_action_classifier_service_spec.rb | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/spec/enterprise/services/captain/llm/assistant_action_classifier_service_spec.rb b/spec/enterprise/services/captain/llm/assistant_action_classifier_service_spec.rb index 3a899c9b7..1fa874ab8 100644 --- a/spec/enterprise/services/captain/llm/assistant_action_classifier_service_spec.rb +++ b/spec/enterprise/services/captain/llm/assistant_action_classifier_service_spec.rb @@ -67,15 +67,15 @@ RSpec.describe Captain::Llm::AssistantActionClassifierService do ) end - it 'uses the configured Captain model from the base LLM service' do - create(:installation_config, name: 'CAPTAIN_OPEN_AI_MODEL', value: 'provider/custom-model') + it 'uses the configured Captain model' do + create(:installation_config, name: 'CAPTAIN_OPEN_AI_MODEL', value: 'gpt-4.1-nano') - expect(RubyLLM).to receive(:chat).with(model: 'provider/custom-model').and_return(mock_chat) + expect(RubyLLM).to receive(:chat).with(model: 'gpt-4.1-nano').and_return(mock_chat) allow(mock_chat).to receive(:ask).and_return(mock_response) result = service.classify(message_history: message_history, assistant_response: 'Would you like to talk to support?') - expect(result).to include('model' => 'provider/custom-model') + expect(result).to include('model' => 'gpt-4.1-nano') end context 'when the assistant has no custom instructions' do