Signed-off-by: Shaun Walsh <shaunwalsh24@gmail.com> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
15 lines
423 B
Python
15 lines
423 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
|
"""Request models for Omni server control routes."""
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class OmniSleepRequest(BaseModel):
|
|
stage_ids: list[int] = Field(..., min_length=1)
|
|
level: int = Field(default=2, ge=0)
|
|
|
|
|
|
class OmniWakeupRequest(BaseModel):
|
|
stage_ids: list[int] = Field(..., min_length=1)
|