Tiny support whitelisted headers in request logging (#16342)

2026-01-03 20:10:58 +08:00
parent 6bc5a52fd2
commit c7c0d97fc6
3 changed files with 61 additions and 10 deletions
@@ -509,7 +509,7 @@ class TokenizerManager(TokenizerCommunicatorMixin, TokenizerManagerMultiItemMixi
            self._attach_multi_http_worker_info(obj)

        # Log the request
-        self.request_logger.log_received_request(obj, self.tokenizer)
+        self.request_logger.log_received_request(obj, self.tokenizer, request)

        async with self.is_pause_cond:
            await self.is_pause_cond.wait_for(lambda: not self.is_pause)
@@ -1214,7 +1214,10 @@ class TokenizerManager(TokenizerCommunicatorMixin, TokenizerManagerMultiItemMixi
                        "response_sent_to_client_ts"
                    ] = state.response_sent_to_client_ts
                self.request_logger.log_finished_request(
-                    obj, out, is_multimodal_gen=self.model_config.is_multimodal_gen
+                    obj,
+                    out,
+                    is_multimodal_gen=self.model_config.is_multimodal_gen,
+                    request=request,
                )

                if self.request_metrics_exporter_manager.exporter_enabled():