state ¶

def clear_state() -> None:
    """Clear server state on shutdown."""
    global _generator, _serving_engine, _fastvideo_args, _served_model_name, _default_request
    _generator = None
    _serving_engine = None
    _fastvideo_args = None
    _served_model_name = None
    _default_request = None

def get_default_request() -> GenerationRequest | None:
    """Return the ServeConfig.default_request set at startup, if any."""
    return _default_request

def get_generator() -> ServingGenerator:
    """Return the global VideoGenerator instance (set during startup)."""
    assert _generator is not None, "Server not initialized — generator is None"
    return _generator

def get_output_dir() -> str:
    """Return the configured output directory."""
    return _output_dir

def get_served_model_name() -> str:
    """Return the public model id advertised by the OpenAI server."""
    args = get_server_args()
    if args.lora_path:
        return args.lora_nickname
    return _served_model_name or args.model_path

def get_server_args() -> FastVideoArgs:
    """Return the global FastVideoArgs (set during startup)."""
    assert _fastvideo_args is not None, "Server not initialized — args is None"
    return _fastvideo_args

def get_serving_engine() -> OpenAIServingEngine:
    """Return the shared model-agnostic OpenAI serving engine."""
    assert _serving_engine is not None, "Server not initialized — serving engine is None"
    return _serving_engine

def set_state(
    generator: ServingGenerator,
    serving_engine: OpenAIServingEngine,
    fastvideo_args: FastVideoArgs,
    output_dir: str,
    default_request: GenerationRequest | None = None,
    served_model_name: str | None = None,
) -> None:
    """Set all server state at once (called from lifespan)."""
    global _generator, _serving_engine, _fastvideo_args, _output_dir, _served_model_name, _default_request
    _generator = generator
    _serving_engine = serving_engine
    _fastvideo_args = fastvideo_args
    _output_dir = output_dir
    _served_model_name = served_model_name
    _default_request = default_request