[bug] fix pd completion protocol for batching support (#8317)

This commit is contained in:
Simo Lin
2025-07-23 23:18:17 -07:00
committed by GitHub
parent f7e102d56a
commit 5dd0f870ab

View File

@@ -192,9 +192,9 @@ class CompletionRequest(BaseModel):
session_params: Optional[Dict] = None
# For PD disaggregation
bootstrap_host: Optional[str] = None
bootstrap_port: Optional[int] = None
bootstrap_room: Optional[int] = None
bootstrap_host: Optional[Union[List[str], str]] = None
bootstrap_port: Optional[Union[List[Optional[int]], int]] = None
bootstrap_room: Optional[Union[List[int], int]] = None
# For request id
rid: Optional[Union[List[str], str]] = None