Fix formatting

Signed-off-by: Zhongxuan Wang <[email protected]>

components/src/dynamo/vllm/handlers.py

@@ -259,10 +259,10 @@ async def generate_tokens(
                     out = {"token_ids": output.token_ids[num_output_tokens_so_far:]}
                     if output.finish_reason:
                         out["finish_reason"] = output.finish_reason
-                        out["completion_usage"] = (
-                            BaseWorkerHandler._build_completion_usage(
-                                request_output=res,
-                            )
+                        out[
+                            "completion_usage"
+                        ] = BaseWorkerHandler._build_completion_usage(
+                            request_output=res,
                         )
                     if output.stop_reason:
                         out["stop_reason"] = output.stop_reason
@@ -511,9 +511,9 @@ async def generate(self, request, context):
                     disaggregated_params: Optional[Dict[str, Any]] = {}
 
                     if res.kv_transfer_params:
-                        disaggregated_params["kv_transfer_params"] = (
-                            res.kv_transfer_params
-                        )
+                        disaggregated_params[
+                            "kv_transfer_params"
+                        ] = res.kv_transfer_params
 
                     if include_timing and timing_metrics:
                         timing_metrics["prefill_end_seconds"] = time.time()

components/src/dynamo/vllm/tests/test_vllm_extra_fields.py

@@ -22,7 +22,6 @@
     _request_contains_timing_metrics,
 )
 
-
 pytestmark = [
     pytest.mark.unit,
     pytest.mark.vllm,