Skip to content

Commit 9cd62c9

Browse files
committed
fix(BA-3028): Make sure kernel registry is passed in as constructor
1 parent 948963d commit 9cd62c9

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/ai/backend/agent/runtime.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@ async def _create_agent(
6565
agent_config: AgentUnifiedConfig,
6666
) -> AbstractAgent:
6767
agent_kwargs = {
68+
"kernel_registry": self.kernel_registry,
6869
"stats_monitor": self.stats_monitor,
6970
"error_monitor": self.error_monitor,
7071
"agent_public_key": self.agent_public_key,
@@ -74,4 +75,4 @@ async def _create_agent(
7475
agent_mod = importlib.import_module(f"ai.backend.agent.{backend.value}")
7576
agent_cls = agent_mod.get_agent_cls()
7677

77-
return agent_cls.new(etcd_view, agent_config, **agent_kwargs)
78+
return await agent_cls.new(etcd_view, agent_config, **agent_kwargs)

0 commit comments

Comments
 (0)