Junhong LiuandHongsheng Liu
9af6fb9a1f
Signed-off-by: Junhong Liu <98734602+LJH-LBJ@users.noreply.github.com> Signed-off-by: Junhong Liu <ljh_lbj@163.com> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
14 lines
454 B
Python
14 lines
454 B
Python
import asyncio
|
|
|
|
from vllm_omni.metrics import OrchestratorAggregator
|
|
|
|
|
|
class ClientRequestState:
|
|
"""Tracks the state of an individual request in the orchestrator."""
|
|
|
|
def __init__(self, request_id: str, queue: asyncio.Queue | None = None):
|
|
self.request_id = request_id
|
|
self.stage_id: int | None = None
|
|
self.queue = queue if queue is not None else asyncio.Queue()
|
|
self.metrics: OrchestratorAggregator | None = None
|