dank-ai 1.0.7 → 1.0.9

package/lib/project.js

@@ -1,14 +1,11 @@
 /**
- * DankProject - Project Management Class
+ * Dank Project Management
  * 
- * This class manages a collection of agents and provides
- * project-level configuration and operations.
+ * Handles project initialization, configuration generation, and scaffolding
  */
 
 const fs = require('fs-extra');
 const path = require('path');
-const yaml = require('js-yaml');
-const { DankAgent } = require('./agent');
 
 class DankProject {
   constructor(name, options = {}) {
@@ -16,99 +13,15 @@ class DankProject {
     this.options = {
       configFile: 'dank.config.js',
       agentsDir: 'agents',
-      outputDir: '.dank',
+      outputDir: 'output',
+      template: 'basic',
       ...options
     };
-    
-    this.agents = new Map();
-    this.projectPath = process.cwd();
-    this.configPath = path.join(this.projectPath, this.options.configFile);
-    this.createdAt = new Date().toISOString();
-  }
-
-  /**
-   * Add an agent to the project
-   */
-  addAgent(agent) {
-    if (!(agent instanceof DankAgent)) {
-      throw new Error('Agent must be an instance of DankAgent');
-    }
-
-    if (this.agents.has(agent.name)) {
-      throw new Error(`Agent with name '${agent.name}' already exists`);
-    }
-
-    this.agents.set(agent.name, agent);
-    return this;
-  }
-
-  /**
-   * Remove an agent from the project
-   */
-  removeAgent(name) {
-    if (!this.agents.has(name)) {
-      throw new Error(`Agent '${name}' not found`);
-    }
-
-    this.agents.delete(name);
-    return this;
-  }
-
-  /**
-   * Get an agent by name
-   */
-  getAgent(name) {
-    return this.agents.get(name);
+    this.projectPath = path.resolve(process.cwd(), name);
   }
 
   /**
-   * Get all agents
-   */
-  getAllAgents() {
-    return Array.from(this.agents.values());
-  }
-
-  /**
-   * Save project configuration to file
-   */
-  async save() {
-    const config = this.toConfig();
-    
-    // Ensure output directory exists
-    await fs.ensureDir(path.join(this.projectPath, this.options.outputDir));
-    
-    // Save as YAML for readability
-    const yamlConfig = yaml.dump(config, { 
-      indent: 2,
-      lineWidth: 120,
-      noCompatMode: true
-    });
-    
-    const configFile = path.join(this.projectPath, this.options.outputDir, 'project.yaml');
-    await fs.writeFile(configFile, yamlConfig, 'utf8');
-    
-    console.log(`Project configuration saved to: ${configFile}`);
-    return configFile;
-  }
-
-  /**
-   * Load project configuration from file
-   */
-  async load() {
-    const configFile = path.join(this.projectPath, this.options.outputDir, 'project.yaml');
-    
-    if (!(await fs.pathExists(configFile))) {
-      throw new Error(`Project configuration not found: ${configFile}`);
-    }
-
-    const yamlContent = await fs.readFile(configFile, 'utf8');
-    const config = yaml.load(yamlContent);
-    
-    return this.fromConfig(config);
-  }
-
-  /**
-   * Initialize a new project structure
+   * Initialize project structure and create example files
    */
   async init() {
     const projectDir = this.projectPath;
@@ -126,68 +39,23 @@ class DankProject {
       console.log(`Created example configuration: ${configPath}`);
     }
 
-    // Create example agent
-    const exampleAgent = this._generateExampleAgent();
-    const agentPath = path.join(projectDir, this.options.agentsDir, 'example-agent.js');
-    
-    if (!(await fs.pathExists(agentPath))) {
-      await fs.writeFile(agentPath, exampleAgent, 'utf8');
-      console.log(`Created example agent: ${agentPath}`);
-    }
-
     console.log(`\\nDank project '${this.name}' initialized!`);
     console.log(`\\nNext steps:`);
     console.log(`1. Edit ${this.options.configFile} to configure your agents`);
     console.log(`2. Run 'dank run' to start your agents`);
     
-    return this;
-  }
-
-  /**
-   * Convert project to configuration object
-   */
-  toConfig() {
     return {
-      name: this.name,
-      version: '1.0.0',
-      createdAt: this.createdAt,
-      options: this.options,
-      agents: Object.fromEntries(
-        Array.from(this.agents.entries()).map(([name, agent]) => [
-          name,
-          agent.toConfig()
-        ])
-      )
+      projectPath: projectDir,
+      configFile: configPath
     };
   }
 
-  /**
-   * Create project from configuration object
-   */
-  fromConfig(config) {
-    this.name = config.name;
-    this.createdAt = config.createdAt;
-    this.options = { ...this.options, ...config.options };
-    
-    // Restore agents
-    this.agents.clear();
-    if (config.agents) {
-      Object.entries(config.agents).forEach(([name, agentConfig]) => {
-        const agent = DankAgent.fromConfig(agentConfig);
-        this.agents.set(name, agent);
-      });
-    }
-    
-    return this;
-  }
-
   /**
    * Generate example configuration file
    */
   _generateExampleConfig() {
-    // Detect if we're in development mode (inside the dank repo)
-    const isDevelopment = this.projectPath.includes('/dank') && 
-                         fs.existsSync(path.join(this.projectPath, '../lib/index.js'));
+    // Check if we're in development mode (local lib directory exists)
+    const isDevelopment = fs.existsSync(path.join(this.projectPath, '../lib/index.js'));
     
     const requirePath = isDevelopment ? '../lib/index.js' : 'dank-ai';
     
@@ -205,6 +73,7 @@ module.exports = {
   name: '${this.name}',
   
   // Define your agents
+  // Each agent can have custom Docker image configuration for production builds
   agents: [
     // Example 1: Direct Prompting Agent with Event Handlers
     createAgent('prompt-agent')
@@ -358,6 +227,12 @@ module.exports = {
         memory: '2g',
         cpu: 2
       })
+      // Agent image configuration for Docker builds
+      .setAgentImageConfig({
+        registry: 'ghcr.io',
+        namespace: 'mycompany',
+        tag: 'latest'
+      })
       // HTTP API routes
       .get('/creative', (req, res) => {
         res.json({ 
@@ -401,119 +276,8 @@ module.exports = {
       })
   ]
 };
-`;
-  }
-
-  /**
-   * Generate example agent file
-   */
-  _generateExampleAgent() {
-    return `/**
- * Example Dank Agent
- * 
- * This is an example of how to define a Dank agent with modern event handling.
- * You can create multiple agent files and import them in your config.
- */
-
-const { createAgent } = require('dank-ai');
-
-const exampleAgent = createAgent('example-agent')
-  .setLLM('openai', {
-    apiKey: process.env.OPENAI_API_KEY,
-    model: 'gpt-3.5-turbo',
-    temperature: 0.7
-  })
-  .setPrompt(\`
-    You are a helpful AI assistant with the following capabilities:
-    - Answer questions clearly and concisely
-    - Provide code examples when appropriate
-    - Be friendly and professional
-    - Help with problem-solving and creative tasks
-  \`)
-  .setBaseImage('nodejs-20')
-  .setPromptingServer({
-    protocol: 'http',
-    port: 3000
-  })
-  .setResources({
-    memory: '512m',
-    cpu: 1,
-    timeout: 30000
-  })
-  // HTTP API routes
-  .get('/info', (req, res) => {
-    res.json({ 
-      agent: 'example-agent',
-      status: 'running',
-      capabilities: ['direct-prompting', 'http-api'],
-      timestamp: new Date().toISOString()
-    });
-  })
-  .post('/chat', (req, res) => {
-    res.json({ 
-      message: 'Chat endpoint ready',
-      data: req.body,
-      timestamp: new Date().toISOString()
-    });
-  })
-  // Event handlers for prompt processing
-  .addHandler('request_output:start', (data) => {
-    console.log(\`[\${new Date().toISOString()}] Processing prompt:\`, data.conversationId);
-    console.log('Original prompt:', data.prompt);
-    
-    // Enhance the prompt
-    const enhancedPrompt = \`[Enhanced] \${data.prompt}\\n\\nPlease provide a helpful and detailed response.\`;
-    
-    return {
-      prompt: enhancedPrompt
-    };
-  })
-  .addHandler('request_output', (data) => {
-    console.log(\`[\${new Date().toISOString()}] LLM Response:\`, {
-      conversationId: data.conversationId,
-      promptModified: data.promptModified,
-      processingTime: data.processingTime,
-      model: data.model
-    });
-  })
-  .addHandler('request_output:end', (data) => {
-    console.log(\`[\${new Date().toISOString()}] Response completed in:\`, data.processingTime + 'ms');
-    
-    // Enhance the response
-    const enhancedResponse = \`\${data.response}\\n\\n---\\n🤖 Powered by Dank Framework\`;
-    
-    return {
-      response: enhancedResponse
-    };
-  })
-  .addHandler('request_output:error', (data) => {
-    console.error(\`[\${new Date().toISOString()}] Error processing prompt:\`, data.error);
-  })
-  // HTTP tool event handlers
-  .addHandler('tool:http-server:call', (data) => {
-    console.log(\`[\${new Date().toISOString()}] HTTP Request:\`, {
-      method: data.method,
-      path: data.path,
-      body: data.body
-    });
-  })
-  .addHandler('tool:http-server:response', (data) => {
-    console.log(\`[\${new Date().toISOString()}] HTTP Response:\`, {
-      statusCode: data.statusCode,
-      processingTime: data.processingTime
-    });
-  })
-  // System event handlers
-  .addHandler('output', (data) => {
-    console.log(\`[\${new Date().toISOString()}] Agent output:\`, data);
-  })
-  .addHandler('error', (error) => {
-    console.error(\`[\${new Date().toISOString()}] Agent error:\`, error);
-  });
-
-module.exports = exampleAgent;
 `;
   }
 }
 
-module.exports = { DankProject };
+module.exports = { DankProject };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dank-ai",
-  "version": "1.0.7",
+  "version": "1.0.9",
   "description": "Dank Agent Service - Docker-based AI agent orchestration platform",
   "main": "lib/index.js",
   "exports": {