dank-ai 1.0.41 → 1.0.45

package/docker/entrypoint.js

@@ -494,10 +494,11 @@ class AgentRuntime {
   }
 
   /**
-   * Emit an event to all matching handlers
+   * Emit an event to all matching handlers (fire-and-forget)
+   * Supports both sync and async handlers
    * Supports pattern matching for tool events
    */
-  emitEvent(eventName, data = null) {
+  async emitEvent(eventName, data = null) {
     // Find all matching handlers (exact match and pattern match)
     const matchingHandlers = [];
 
@@ -507,19 +508,27 @@ class AgentRuntime {
       }
     }
 
-    // Execute all matching handlers
-    matchingHandlers.forEach((handler) => {
+    // Execute all matching handlers (in parallel for fire-and-forget)
+    const promises = matchingHandlers.map(async (handler) => {
       try {
+        let result;
         if (typeof handler === "function") {
-          handler(data);
+          result = handler(data);
         } else if (handler.handler && typeof handler.handler === "function") {
-          handler.handler(data);
+          result = handler.handler(data);
+        }
+        // Await if handler returns a promise
+        if (result && typeof result.then === "function") {
+          await result;
         }
       } catch (error) {
         logger.error(`Error in event handler for '${eventName}':`, error);
       }
     });
 
+    // Wait for all handlers to complete
+    await Promise.all(promises);
+
     if (matchingHandlers.length > 0) {
       logger.debug(
         `Emitted event '${eventName}' to ${matchingHandlers.length} handlers`
@@ -529,8 +538,10 @@ class AgentRuntime {
 
   /**
    * Emit an event and collect responses from handlers that return modified data
+   * Supports both sync and async handlers
+   * Handlers are executed sequentially to allow proper chaining of modifications
    */
-  emitEventWithResponse(eventName, data) {
+  async emitEventWithResponse(eventName, data) {
     // Find all matching handlers (exact match and pattern match)
     const matchingHandlers = [];
 
@@ -542,8 +553,8 @@ class AgentRuntime {
 
     let modifiedData = { ...data };
 
-    // Execute all matching handlers and collect responses
-    matchingHandlers.forEach((handler) => {
+    // Execute handlers sequentially to allow chaining of modifications
+    for (const handler of matchingHandlers) {
       try {
         let result;
         if (typeof handler === "function") {
@@ -552,6 +563,11 @@ class AgentRuntime {
           result = handler.handler(modifiedData);
         }
 
+        // Await if handler returns a promise
+        if (result && typeof result.then === "function") {
+          result = await result;
+        }
+
         // If handler returns an object, merge it with the current data
         if (result && typeof result === "object" && !Array.isArray(result)) {
           modifiedData = { ...modifiedData, ...result };
@@ -559,7 +575,7 @@ class AgentRuntime {
       } catch (error) {
         logger.error(`Handler error for event '${eventName}':`, error);
       }
-    });
+    }
 
     if (matchingHandlers.length > 0) {
       logger.debug(
@@ -1193,7 +1209,7 @@ class AgentRuntime {
     this.mainApp.post("/prompt", async (req, res) => {
       logger.info(`📥 Received POST /prompt request from ${req.ip || 'unknown'}`);
       try {
-        const { prompt, conversationId, metadata } = req.body;
+        const { prompt, ...requestBodyFields } = req.body;
 
         if (!prompt) {
           return res.status(400).json({
@@ -1202,16 +1218,18 @@ class AgentRuntime {
           });
         }
 
-        const response = await this.processDirectPrompt(prompt, {
-          conversationId,
-          metadata,
+        // Build metadata object with all user-provided fields from request body (except prompt)
+        const metadata = {
+          ...requestBodyFields,
+        };
+
+        const response = await this.processDirectPrompt(prompt, metadata, {
           protocol: "http",
           clientIp: req.ip,
         });
 
         res.json({
           response: response.content,
-          conversationId: conversationId || response.conversationId,
           metadata: response.metadata,
           timestamp: new Date().toISOString(),
         });
@@ -1233,24 +1251,23 @@ class AgentRuntime {
   /**
    * Process a direct prompt and emit events
    */
-  async processDirectPrompt(prompt, context = {}) {
+  async processDirectPrompt(prompt, metadata = {}, systemFields = {}) {
     const startTime = Date.now();
-    const conversationId = context.conversationId || require("uuid").v4();
+    let finalPrompt = prompt; // Declare outside try block so it's available in catch
 
     try {
       // Emit request start event and allow handlers to modify the prompt
       const startEventData = {
         prompt,
-        conversationId,
-        context,
+        metadata,
+        ...systemFields, // protocol, clientIp, etc.
         timestamp: new Date().toISOString(),
       };
       
-      const modifiedData = this.emitEventWithResponse("request_output:start", startEventData);
+      const modifiedData = await this.emitEventWithResponse("request_output:start", startEventData);
       
       // Use modified prompt if handlers returned one, otherwise use original
-      const finalPrompt = modifiedData?.prompt || prompt;
-
+      finalPrompt = modifiedData?.prompt || prompt;
       // Process with LLM
       let response;
       if (this.llmProvider === "openai") {
@@ -1281,12 +1298,12 @@ class AgentRuntime {
       const processingTime = Date.now() - startTime;
 
       // Emit request output event
-      this.emitEvent("request_output", {
+      await this.emitEvent("request_output", {
         prompt,
         finalPrompt, // Include the final prompt that was sent to LLM
         response: response.content,
-        conversationId,
-        context,
+        metadata,
+        ...systemFields, // protocol, clientIp, etc.
         usage: response.usage,
         model: response.model,
         processingTime,
@@ -1296,11 +1313,11 @@ class AgentRuntime {
 
       // Emit end event and allow handlers to modify the response before returning
       const endEventData = {
-        conversationId,
         prompt,
         finalPrompt,
         response: response.content,
-        context,
+        metadata,
+        ...systemFields, // protocol, clientIp, etc.
         usage: response.usage,
         model: response.model,
         processingTime,
@@ -1309,14 +1326,13 @@ class AgentRuntime {
         timestamp: new Date().toISOString(),
       };
 
-      const modifiedEndData = this.emitEventWithResponse("request_output:end", endEventData);
+      const modifiedEndData = await this.emitEventWithResponse("request_output:end", endEventData);
       
       // Use the final response (potentially modified by handlers)
       const finalResponse = modifiedEndData.response || response.content;
 
       return {
         content: finalResponse,
-        conversationId,
         metadata: {
           usage: response.usage,
           model: response.model,
@@ -1328,11 +1344,11 @@ class AgentRuntime {
       const processingTime = Date.now() - startTime;
 
       // Emit error event
-      this.emitEvent("request_output:error", {
+      await this.emitEvent("request_output:error", {
         prompt,
         finalPrompt,
-        conversationId,
-        context,
+        metadata,
+        ...systemFields, // protocol, clientIp, etc.
         error: error.message,
         processingTime,
         promptModified: finalPrompt !== prompt,

package/lib/agent.js

@@ -10,6 +10,7 @@ const { validate: validateUUID, v4: uuidv4 } = require('uuid');
 const { DEFAULT_CONFIG, SUPPORTED_LLMS, DOCKER_CONFIG, INSTANCE_TYPES } = require('./constants');
 const { ToolRegistry, ToolExecutor } = require('./tools');
 const builtinTools = require('./tools/builtin');
+const { PluginManager } = require('./plugins');
 
 class DankAgent {
   constructor(name, config = {}) {
@@ -26,6 +27,10 @@ class DankAgent {
     this.toolRegistry = new ToolRegistry();
     this.toolExecutor = new ToolExecutor(this.toolRegistry);
     
+    // Initialize plugin system
+    this.pluginManager = new PluginManager(this);
+    this.pendingPlugins = new Map(); // Store plugins to be loaded during finalization
+    
     // Register built-in tools if enabled
     if (config.enableBuiltinTools !== false) {
       this.registerBuiltinTools();
@@ -420,6 +425,68 @@ class DankAgent {
     return this.toolRegistry.toOpenAISchema();
   }
 
+  /**
+   * Add a plugin to this agent
+   * Plugin will be loaded and initialized during finalization
+   * @param {string|PluginBase} name - Plugin name, npm package, local path, or PluginBase instance
+   * @param {object} config - Plugin configuration
+   */
+  addPlugin(name, config = {}) {
+    this.pendingPlugins.set(name, config);
+    return this;
+  }
+
+  /**
+   * Add multiple plugins at once
+   * Plugins will be loaded and initialized during finalization
+   * @param {object} plugins - Object mapping plugin names to configs
+   */
+  addPlugins(plugins) {
+    Object.entries(plugins).forEach(([name, config]) => {
+      this.pendingPlugins.set(name, config);
+    });
+    return this;
+  }
+
+  /**
+   * Get a plugin by name
+   * @param {string} name - Plugin name
+   */
+  getPlugin(name) {
+    return this.pluginManager.getPlugin(name);
+  }
+
+  /**
+   * Get all plugins
+   */
+  getPlugins() {
+    return this.pluginManager.getAllPlugins();
+  }
+
+  /**
+   * Check if a plugin is loaded
+   * @param {string} name - Plugin name
+   */
+  hasPlugin(name) {
+    return this.pluginManager.hasPlugin(name);
+  }
+
+  /**
+   * Remove a plugin
+   * @param {string} name - Plugin name
+   */
+  async removePlugin(name) {
+    await this.pluginManager.removePlugin(name);
+    return this;
+  }
+
+  /**
+   * Get plugin manager metadata
+   */
+  getPluginMetadata() {
+    return this.pluginManager.getMetadata();
+  }
+
   /**
    * Register built-in tools
    */
@@ -590,6 +657,7 @@ class DankAgent {
   /**
    * Finalize agent configuration by auto-detecting features
    * This should be called before the agent is deployed
+   * Plugins will be loaded when the agent is started
    */
   finalize() {
     // Validate that ID is set before finalization (REQUIRED)
@@ -599,6 +667,18 @@ class DankAgent {
     return this;
   }
 
+  /**
+   * Initialize plugins (called internally when agent starts)
+   * @private
+   */
+  async _initializePlugins() {
+    if (this.pendingPlugins.size > 0) {
+      const plugins = Object.fromEntries(this.pendingPlugins);
+      await this.pluginManager.addPlugins(plugins);
+      this.pendingPlugins.clear();
+    }
+  }
+
   /**
    * Set custom configuration
    */

package/lib/cli/init.js

@@ -126,7 +126,9 @@ async function createPackageJson(npmProjectName, projectPath) {
       clean: 'dank clean'
     },
     dependencies: {
-      'dank-ai': '^1.0.0'
+      'dank-ai': '^1.0.0',
+      'axios': '^1.6.0',
+      'date-fns': '^3.0.0'
     },
     keywords: ['dank', 'ai', 'agents', 'automation', 'llm'],
     author: '',
@@ -270,6 +272,7 @@ A Dank AI agent project with modern event handling and Docker orchestration.
 
 - 🤖 **AI Agents**: Powered by multiple LLM providers (OpenAI, Anthropic, Google AI)
 - 🐳 **Docker Integration**: Containerized agents with automatic management
+- 📦 **NPM Packages**: Use any npm package in your handlers with top-level imports
 - 📡 **Event System**: Real-time event handling for prompts, responses, and tools
 - 🔧 **Auto-Detection**: Automatically enables features based on usage
 - 📊 **Monitoring**: Built-in logging and status monitoring
@@ -349,6 +352,27 @@ Each agent in your project has a unique UUIDv4 identifier that is automatically
 - Once agents register with Dank Cloud services, these IDs become locked to your account
 - You can generate new UUIDs if needed: \`require('uuid').v4()\`
 
+## Using NPM Packages
+
+You can use any npm package in your handlers by importing them at the top of \`dank.config.js\`:
+
+\`\`\`javascript
+// Import packages at the top
+const axios = require('axios');
+const { format } = require('date-fns');
+
+// Use them in your handlers
+.addHandler('request_output', async (data) => {
+  const timestamp = format(new Date(), 'yyyy-MM-dd HH:mm');
+  await axios.post('https://api.example.com/log', {
+    response: data.response,
+    timestamp
+  });
+})
+\`\`\`
+
+Just add any packages you need to your \`package.json\` and run \`npm install\`.
+
 ## Configuration
 
 Edit \`dank.config.js\` to:

package/lib/cli/production-build.js

@@ -173,7 +173,9 @@ async function productionBuildCommand(options) {
           force: options.force || false,
           push: options.push || false,
           baseImageOverride: options.baseImageOverride || null,
-          projectDir: projectDir  // Pass project directory so external files can be copied
+          projectDir: projectDir,  // Pass project directory so external files can be copied
+          configPath: configPath,  // Pass config path for extracting top-level requires
+          projectRoot: process.cwd()  // Pass project root for package.json location
         };
         
         const result = await dockerManager.buildProductionImage(agent, buildOptions);

package/lib/cli/run.js

@@ -71,7 +71,9 @@ async function runCommand(options) {
           
           const container = await dockerManager.startAgent(agent, {
             rebuild: !options.noBuild,  // Rebuild by default unless --no-build is specified
-            projectDir: projectDir  // Pass project directory so external files can be copied
+            projectDir: projectDir,  // Pass project directory so external files can be copied
+            configPath: configPath,  // Pass config path for extracting top-level requires
+            projectRoot: process.cwd()  // Pass project root for package.json location
           });
           
           console.log(chalk.green(`  ✅ ${agent.name} started (${container.id.substring(0, 12)})`));