mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-29 12:06:51 +08:00
79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
"""宿主启动阶段构建的类型化运行时上下文。"""
|
|
|
|
from collections.abc import AsyncGenerator, Generator
|
|
from dataclasses import dataclass
|
|
from typing import Protocol
|
|
|
|
from app.application.messaging.chat import (
|
|
AsyncAgentChatRepository,
|
|
AsyncUnitOfWork,
|
|
)
|
|
|
|
|
|
class AgentChatRepositoryFactory(Protocol):
|
|
"""由请求会话构造 Agent 会话仓储的工厂端口。"""
|
|
|
|
def __call__(self, session: object) -> AsyncAgentChatRepository:
|
|
"""绑定请求会话并返回 Agent 会话仓储。"""
|
|
...
|
|
|
|
|
|
class AsyncUnitOfWorkFactory(Protocol):
|
|
"""由请求会话构造异步事务端口的工厂。"""
|
|
|
|
def __call__(self, session: object) -> AsyncUnitOfWork:
|
|
"""绑定请求会话并返回异步事务端口。"""
|
|
...
|
|
|
|
|
|
class AsyncSessionProvider(Protocol):
|
|
"""FastAPI 请求级异步会话提供器。"""
|
|
|
|
def __call__(self) -> AsyncGenerator[object, None]:
|
|
"""生成一个请求独占的异步数据库会话。"""
|
|
...
|
|
|
|
|
|
class SyncSessionProvider(Protocol):
|
|
"""兼容 API Facade 使用的同步会话提供器。"""
|
|
|
|
def __call__(self) -> Generator[object, None, None]:
|
|
"""生成一个请求独占的同步数据库会话。"""
|
|
...
|
|
|
|
|
|
class CompatibilityApiData(Protocol):
|
|
"""未迁移 API 领域继续使用的结构化兼容 Facade。"""
|
|
|
|
sync_session: SyncSessionProvider
|
|
async_session: AsyncSessionProvider
|
|
|
|
def repository(self, name: str, session: object) -> object:
|
|
"""按旧能力名构造请求级仓储。"""
|
|
...
|
|
|
|
def standalone_repository(self, name: str) -> object:
|
|
"""按旧能力名构造独立仓储。"""
|
|
...
|
|
|
|
def transaction(self, name: str, session: object) -> object:
|
|
"""按旧能力名构造事务端口。"""
|
|
...
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class AgentChatRuntime:
|
|
"""Agent 会话 API 可见的最小数据运行时。"""
|
|
|
|
async_session: AsyncSessionProvider
|
|
repository: AgentChatRepositoryFactory
|
|
transaction: AsyncUnitOfWorkFactory
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class HostRuntime:
|
|
"""宿主组合根构建且在一个 FastAPI lifespan 内共享的运行时对象。"""
|
|
|
|
agent_chat: AgentChatRuntime
|
|
compatibility_api_data: CompatibilityApiData
|