diff --git a/ee/lib/gitlab/llm/completions/chat.rb b/ee/lib/gitlab/llm/completions/chat.rb index 8c8c52b8d2acbbe989d5d30884bf45b125d18b48..fc76c7ff600f455c98db7eb3b2e84de6609d9d4c 100644 --- a/ee/lib/gitlab/llm/completions/chat.rb +++ b/ee/lib/gitlab/llm/completions/chat.rb @@ -94,7 +94,7 @@ def agent_or_tool_response(response_handler) options: { input: options[:content] }, stream_response_handler: stream_response_handler, command: command - ).perform + ).execute end Gitlab::Llm::Chain::Agents::ZeroShot::Executor.new( diff --git a/ee/spec/lib/gitlab/llm/completions/chat_spec.rb b/ee/spec/lib/gitlab/llm/completions/chat_spec.rb index 4f9b61f94f0191fd638270f93a157cdf24df4d69..53682707004cf9ae8476828c87a2dd31cd44c785 100644 --- a/ee/spec/lib/gitlab/llm/completions/chat_spec.rb +++ b/ee/spec/lib/gitlab/llm/completions/chat_spec.rb @@ -264,7 +264,7 @@ expect(::Gitlab::Llm::Chain::Agents::ZeroShot::Executor).not_to receive(:new) expect(expected_tool) .to receive(:new).with(expected_params).and_return(executor) - expect(executor).to receive(:perform).and_return(answer) + expect(executor).to receive(:execute).and_return(answer) subject end