Add ai request to troubleshoot job

ee/lib/gitlab/llm/chain/tools/troubleshoot_job/executor.rb

@@ -84,6 +84,11 @@ def unit_primitive
               'troubleshoot_job'
             end
 
+            def ai_request
+              ::Gitlab::Llm::Chain::Requests::AiGateway.new(context.current_user, service_name: :troubleshoot_job,
+                tracking_context: tracking_context)
+            end
+
             def tracking_context
               {
                 request_id: context.request_id,

ee/spec/lib/gitlab/llm/chain/tools/troubleshoot_job/executor_spec.rb

@@ -82,6 +82,13 @@
       end
 
       it 'sets the correct unit primitive' do
+        allow(Gitlab::Llm::Chain::Requests::AiGateway).to receive(:new).with(user, {
+          service_name: :troubleshoot_job,
+          tracking_context: {
+            request_id: nil,
+            action: 'troubleshoot_job'
+          }
+        }).and_return(ai_request_double)
         expect(ai_request_double).to receive(:request).with(tool.prompt, unit_primitive: 'troubleshoot_job')
 
         tool.execute