OmAgent / compile_container.py
韩宇
init
1b7e88c
raw
history blame
633 Bytes
# Import core modules and components
# Import workflow related modules
from pathlib import Path
from omagent_core.utils.container import container
from omagent_core.utils.registry import registry
# Set up path and import modules
CURRENT_PATH = root_path = Path(__file__).parents[0]
registry.import_module(project_path=CURRENT_PATH.joinpath("agent"))
# Register required components
container.register_callback(callback="DefaultCallback")
container.register_input(input="AppInput")
container.register_stm("SharedMemSTM")
container.register_ltm(ltm="VideoMilvusLTM")
# Compile container config
container.compile_config(CURRENT_PATH)