add Agent Container

src/container/compat.ts

@@ -12,6 +12,7 @@ import { Context } from "./context";
 import { Container } from "./impl";
 
 const UI_KEY = "NEOS_UI";
+const AGENT_KEY = "NEOS_AGENT";
 
 export function initUIContainer(conn: WebSocketStream) {
   const context = new Context({
@@ -35,3 +36,20 @@ export function getUIContainer(): Container {
     throw Error("UI Container not initialized !!");
   }
 }
+
+export function initAgentContainer(conn: WebSocketStream) {
+  const context = new Context({});
+  const container = new Container(context, conn);
+
+  CONTAINERS.set(AGENT_KEY, container);
+}
+
+export function getAgentContainer(): Container {
+  const container = CONTAINERS.get(AGENT_KEY);
+
+  if (container) {
+    return container;
+  } else {
+    throw Error("Agent Container not initialized !!");
+  }
+}