# 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) | |