49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import TYPE_CHECKING
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from yuxi.channel.container import ChannelContainer
|
||
|
|
|
||
|
|
from yuxi.channel.container import ChannelContainerFactory, setup_channel
|
||
|
|
|
||
|
|
_container: ChannelContainer | None = None
|
||
|
|
|
||
|
|
|
||
|
|
async def init_channel(
|
||
|
|
redis_url: str = "",
|
||
|
|
agent_port=None,
|
||
|
|
*,
|
||
|
|
config_yaml_path: str = "channel_config.yaml",
|
||
|
|
mq_workers: int = 4,
|
||
|
|
mq_max_concurrent: int = 20,
|
||
|
|
default_agent_config_id: int = 1,
|
||
|
|
) -> ChannelContainer:
|
||
|
|
global _container
|
||
|
|
_container = await ChannelContainerFactory.create(
|
||
|
|
redis_url=redis_url,
|
||
|
|
agent_port=agent_port,
|
||
|
|
config_yaml_path=config_yaml_path,
|
||
|
|
mq_workers=mq_workers,
|
||
|
|
mq_max_concurrent=mq_max_concurrent,
|
||
|
|
default_agent_config_id=default_agent_config_id,
|
||
|
|
)
|
||
|
|
return _container
|
||
|
|
|
||
|
|
|
||
|
|
async def shutdown_channel(container: ChannelContainer | None = None) -> None:
|
||
|
|
global _container
|
||
|
|
target = container or _container
|
||
|
|
if target:
|
||
|
|
await target.shutdown()
|
||
|
|
_container = None
|
||
|
|
|
||
|
|
|
||
|
|
def get_container() -> ChannelContainer | None:
|
||
|
|
return _container
|
||
|
|
|
||
|
|
|
||
|
|
def register_channel_middleware(app) -> None:
|
||
|
|
if _container:
|
||
|
|
setup_channel(app, _container)
|