feature: Pass Root context of manager to plugins (#1829)

Co-authored-by: Joongi Kim <[email protected]>
lablup · Jan 18, 2024 · 4461215 · 4461215
1 parent b052e29
commit 4461215
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/changes/1829.feature.md b/changes/1829.feature.md
@@ -0,0 +1 @@
+Pass the root context to the manager plugins so that they can access database connection pools and other globals
diff --git a/src/ai/backend/manager/server.py b/src/ai/backend/manager/server.py
@@ -301,6 +301,7 @@ async def webapp_plugin_ctx(root_app: web.Application) -> AsyncIterator[None]:
     root_ctx: RootContext = root_app["_root.context"]
     plugin_ctx = WebappPluginContext(root_ctx.shared_config.etcd, root_ctx.local_config)
     await plugin_ctx.init(
+        context=root_ctx,
         allowlist=root_ctx.local_config["manager"]["allowed-plugins"],
         blocklist=root_ctx.local_config["manager"]["disabled-plugins"],
     )
@@ -439,7 +440,7 @@ async def hook_plugin_ctx(root_ctx: RootContext) -> AsyncIterator[None]:
     ctx = HookPluginContext(root_ctx.shared_config.etcd, root_ctx.local_config)
     root_ctx.hook_plugin_ctx = ctx
     await ctx.init(
-        context=ctx,
+        context=root_ctx,
         allowlist=root_ctx.local_config["manager"]["allowed-plugins"],
         blocklist=root_ctx.local_config["manager"]["disabled-plugins"],
     )