Fix MaxListenersExceededWarning memory leak

Author: charlesvien

apps/twig/src/main/lib/typed-event-emitter.ts

@@ -1,6 +1,11 @@
 import { EventEmitter, on } from "node:events";
 
 export class TypedEventEmitter<TEvents> extends EventEmitter {
+  constructor() {
+    super();
+    this.setMaxListeners(0);
+  }
+
   emit<K extends keyof TEvents & string>(
     event: K,
     payload: TEvents[K],

packages/agent/src/agent.ts

@@ -100,6 +100,9 @@ export class Agent {
   }
 
   async cleanup(): Promise<void> {
+    if (this.sessionLogWriter && this.taskRunId) {
+      await this.sessionLogWriter.flush(this.taskRunId);
+    }
     await this.acpConnection?.cleanup();
   }
 }

packages/agent/src/session-log-writer.ts

@@ -18,25 +18,14 @@ export class SessionLogWriter {
     this.posthogAPI = posthogAPI;
     this.logger =
       logger ?? new Logger({ debug: false, prefix: "[SessionLogWriter]" });
+  }
 
-    const flushAllAndExit = async () => {
-      const flushPromises: Promise<void>[] = [];
-      for (const sessionId of this.configs.keys()) {
-        flushPromises.push(this.flush(sessionId));
-      }
-      await Promise.all(flushPromises);
-      process.exit(0);
-    };
-
-    process.on("beforeExit", () => {
-      flushAllAndExit().catch((e) => this.logger.error("Flush failed:", e));
-    });
-    process.on("SIGINT", () => {
-      flushAllAndExit().catch((e) => this.logger.error("Flush failed:", e));
-    });
-    process.on("SIGTERM", () => {
-      flushAllAndExit().catch((e) => this.logger.error("Flush failed:", e));
-    });
+  async flushAll(): Promise<void> {
+    const flushPromises: Promise<void>[] = [];
+    for (const sessionId of this.configs.keys()) {
+      flushPromises.push(this.flush(sessionId));
+    }
+    await Promise.all(flushPromises);
   }
 
   register(sessionId: string, config: SessionLogConfig): void {