fix: fix the if statement for checking tllm_disagg_params.opaque_state is none (#1679)

richardhuo-nv · atchernych · commit 579458fc1700 · 2025-07-09T09:58:27.000-07:00
diff --git a/examples/tensorrt_llm/common/protocol.py b/examples/tensorrt_llm/common/protocol.py
@@ -57,7 +57,7 @@ def to_oai_disaggregated_params(
         else:
             encoded_opaque_state = (
                 base64.b64encode(tllm_disagg_params.opaque_state).decode("utf-8")
-                if tllm_disagg_params is not None
+                if tllm_disagg_params.opaque_state is not None
                 else None
             )
             return DisaggregatedParams(

Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,7 @@ def to_oai_disaggregated_params(`
`57`	`57`	`else:`
`58`	`58`	`encoded_opaque_state = (`
`59`	`59`	`base64.b64encode(tllm_disagg_params.opaque_state).decode("utf-8")`
`60`		`- if tllm_disagg_params is not None`
	`60`	`+ if tllm_disagg_params.opaque_state is not None`
`61`	`61`	`else None`
`62`	`62`	`)`
`63`	`63`	`return DisaggregatedParams(`