diff --git a/app/sage.py b/app/sage.py index 04945c1..76bfa92 100644 --- a/app/sage.py +++ b/app/sage.py @@ -3,10 +3,12 @@ import argparse from appPublic.log import MyLogger, info, debug, warning from appPublic.folderUtils import ProgramPath from appPublic.jsonConfig import getConfig +from appPublic.zmqapi import run_proxy from appPublic.registerfunction import RegisterFunction from bricks_for_python.init import load_pybricks from ahserver.webapp import webapp from ahserver.serverenv import ServerEnv +from ahserver.configuredServer import add_cleanupctx from rbac.init import load_rbac from pricing.init import load_pricing from appbase.init import load_appbase @@ -28,6 +30,12 @@ from ext import * from rf import * __version__ = '0.0.1' +async def run_zmq_proxy(app): + task = asyncio.create_task(run_proxy()) + app['zmq_proxy_task'] = task + yield + task.cancel() + def init(): rf = RegisterFunction() set_globalvariable() @@ -47,6 +55,7 @@ def init(): load_checklang() load_skillagent() load_dagflow() + add_cleanupctx(run_zmq_proxy) if __name__ == '__main__': webapp(init)