[bug] fix pd completion protocol for batching support (#8317)
This commit is contained in:
@@ -192,9 +192,9 @@ class CompletionRequest(BaseModel):
|
|||||||
session_params: Optional[Dict] = None
|
session_params: Optional[Dict] = None
|
||||||
|
|
||||||
# For PD disaggregation
|
# For PD disaggregation
|
||||||
bootstrap_host: Optional[str] = None
|
bootstrap_host: Optional[Union[List[str], str]] = None
|
||||||
bootstrap_port: Optional[int] = None
|
bootstrap_port: Optional[Union[List[Optional[int]], int]] = None
|
||||||
bootstrap_room: Optional[int] = None
|
bootstrap_room: Optional[Union[List[int], int]] = None
|
||||||
|
|
||||||
# For request id
|
# For request id
|
||||||
rid: Optional[Union[List[str], str]] = None
|
rid: Optional[Union[List[str], str]] = None
|
||||||
|
|||||||
Reference in New Issue
Block a user