Merge pull request #599 from bmgalego/node-plugin-lazy

feat: make node-plugin lazy-loaded for faster boot times

Author: lalalune

agent/src/index.ts

@@ -26,7 +26,7 @@ import { bootstrapPlugin } from "@ai16z/plugin-bootstrap";
 import { confluxPlugin } from "@ai16z/plugin-conflux";
 import { solanaPlugin } from "@ai16z/plugin-solana";
 import { zgPlugin } from "@ai16z/plugin-0g";
-import { nodePlugin } from "@ai16z/plugin-node";
+import { type NodePlugin, createNodePlugin } from "@ai16z/plugin-node";
 import {
     coinbaseCommercePlugin,
     coinbaseMassPaymentsPlugin,
@@ -238,6 +238,8 @@ function getSecret(character: Character, secret: string) {
     return character.settings.secrets?.[secret] || process.env[secret];
 }
 
+let nodePlugin: NodePlugin | undefined;
+
 export function createAgent(
     character: Character,
     db: IDatabaseAdapter,
@@ -249,6 +251,9 @@ export function createAgent(
         "Creating runtime for character",
         character.name
     );
+
+    nodePlugin ??= createNodePlugin();
+
     return new AgentRuntime({
         databaseAdapter: db,
         token,

packages/plugin-node/src/index.ts

@@ -12,18 +12,21 @@ import {
     VideoService,
 } from "./services/index.ts";
 
-export const nodePlugin: Plugin = {
-    name: "default",
-    description: "Default plugin, with basic actions and evaluators",
-    services: [
-        new BrowserService(),
-        new ImageDescriptionService(),
-        new LlamaService(),
-        new PdfService(),
-        new SpeechService(),
-        new TranscriptionService(),
-        new VideoService(),
-    ],
-};
 
-export default nodePlugin;
+export type NodePlugin = ReturnType<typeof createNodePlugin>
+
+export function createNodePlugin() {
+    return {
+        name: "default",
+        description: "Default plugin, with basic actions and evaluators",
+        services: [
+            new BrowserService(),
+            new ImageDescriptionService(),
+            new LlamaService(),
+            new PdfService(),
+            new SpeechService(),
+            new TranscriptionService(),
+            new VideoService(),
+        ],
+    } as const satisfies Plugin
+};

packages/plugin-node/src/services/browser.ts

@@ -87,7 +87,9 @@ export class BrowserService extends Service implements IBrowserService {
         );
     }
 
-    async initialize() {
+    async initialize() { }
+
+    async initializeBrowser() {
         if (!this.browser) {
             this.browser = await chromium.launch({
                 args: ["--no-sandbox", "--disable-setuid-sandbox"],
@@ -118,7 +120,7 @@ export class BrowserService extends Service implements IBrowserService {
         url: string,
         runtime: IAgentRuntime
     ): Promise<PageContent> {
-        await this.initialize();
+        await this.initializeBrowser();
         this.queue.push(url);
         this.processQueue(runtime);
 
@@ -181,7 +183,7 @@ export class BrowserService extends Service implements IBrowserService {
         try {
             if (!this.context) {
                 console.log(
-                    "Browser context not initialized. Call initialize() first."
+                    "Browser context not initialized. Call initializeBrowser() first."
                 );
             }