ai-kefu-admin/app/engine/routes/runtime.py

"""路由取当前租户运行时——Task 5b→7 的过渡兜底。

Task 7 的 Host 中间件上线前,没有人为请求挂 ``state.tenant``。本助手的
兜底规则 = state 里有租户用 state 的(中间件灰度接入即生效);没有则用
``app.state.default_runtime``(lifespan 预热的默认租户 jiancai),API 测试
与直连(无 Host 语义)在 5b→7 过渡期不断。

已知过渡限制(记录在案,Task 7 一并解决):默认租户经 registry_reload
热切换后 default_runtime 仍指向旧 runtime——过渡期不修,Task 7 中间件
按请求 Host 重新 resolve 后本模块整体删除。
"""
from typing import TYPE_CHECKING

from fastapi import Request, WebSocket

if TYPE_CHECKING:
    # 仅类型注解用: engine 包运行期零 import platform(依赖方向保持
    # platform → engine,反向只在类型层)
    from app.platform.registry import TenantRuntime


def get_runtime(request_or_ws: Request | WebSocket) -> "TenantRuntime":
    """state.tenant 优先,缺省落 app.state.default_runtime(Task 7 删)。"""
    runtime = getattr(request_or_ws.state, "tenant", None)
    if runtime is not None:
        return runtime
    runtime = getattr(request_or_ws.app.state, "default_runtime", None)
    if runtime is None or runtime.session_factory is None:
        raise RuntimeError(
            "默认租户运行时未就绪——lifespan 未运行、未预热 default_runtime,"
            "或 registry_reload 已 dispose 旧 runtime(过渡期限制,Task 7 解决)")
    return runtime