Unverified Commit 1e18a341 authored by Tony Lu's avatar Tony Lu Committed by GitHub
Browse files

[Bugfix] fix pd chat completion protocol for batching support (#10016)


Signed-off-by: default avatarTony Lu <tonylu@linux.alibaba.com>
parent 2c562fd2
...@@ -542,9 +542,9 @@ class ChatCompletionRequest(BaseModel): ...@@ -542,9 +542,9 @@ class ChatCompletionRequest(BaseModel):
rid: Optional[Union[List[str], str]] = None rid: Optional[Union[List[str], str]] = 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
class ChatMessage(BaseModel): class ChatMessage(BaseModel):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment