[fix] Remove duplicated KVCache transmission check (#6022)

Signed-off-by: Iman Tabrizian <[email protected]>

Author: Tabrizian

tensorrt_llm/_torch/pyexecutor/py_executor.py

@@ -966,19 +966,14 @@ def _executor_loop(self):
                         self._prepare_disagg_gen_transmission_complete(
                             scheduled_batch)
 
+                        # Return the first token to the client
+                        self._handle_first_token_response(scheduled_batch)
+
                     self.resource_manager.prepare_resources(scheduled_batch)
                     if self.drafter is not None:
                         self.drafter.prepare_draft_tokens(
                             scheduled_batch, self.resource_manager)
 
-                    if self.kv_cache_transceiver:
-                        # For generation requests which have completed KV cache transfer
-                        self._prepare_disagg_gen_transmission_complete(
-                            scheduled_batch)
-
-                        # Return the first token to the client
-                        self._handle_first_token_response(scheduled_batch)
-
                     batch_outputs = self._forward_step(scheduled_batch)
 
                     if self.guided_decoder is not None: