shipdeck/lib/ultimate/agents/infrastructure-maintainer.js

Ship MVPs in 48 hours. Fix bugs in 30 seconds. The command deck for developers who ship.

/**
 * InfrastructureMaintainer Agent
 * Monitors system health and manages infrastructure
 */

const BaseAgent = require('./base-agent');
const { AnthropicClient } = require('../anthropic/client');

/**
 * InfrastructureMaintainer - Monitors system health and manages infrastructure
 */
class InfrastructureMaintainerAgent extends BaseAgent {
  constructor(config = {}) {
    super('InfrastructureMaintainerAgent', {
      ...config,
      description: 'Monitors system health and manages infrastructure',
      version: '1.0.0'
    });

    this.capabilities = ["monitoring","scaling","reliability","performance","health"];
    this.anthropicClient = config.anthropicClient || new AnthropicClient();
  }

  /**
   * Get agent capabilities
   * @returns {Array} List of capabilities
   */
  getCapabilities() {
    return this.capabilities;
  }

  /**
   * Get system prompt for the agent
   * @returns {string} System prompt
   */
  getSystemPrompt() {
    return `You are InfrastructureMaintainerAgent, an AI agent specialized in monitoring, scaling, reliability, performance, health.

Your core responsibilities:
- monitoring
- scaling
- reliability
- performance
- health

Always provide practical, production-ready solutions.`;
  }

  /**
   * Execute agent task
   * @param {Object} task - Task to execute
   * @param {Object} context - Execution context
   * @returns {Promise<Object>} Execution result
   */
  async execute(task, context = {}) {
    try {
      this.validateTask(task);
      
      const systemPrompt = this.getSystemPrompt();
      const userPrompt = this.createPrompt(task, context);

      const response = await this.anthropicClient.createMessage(
        `${systemPrompt}

${userPrompt}`,
        {
          maxTokens: 4096,
          temperature: 0.7
        }
      );

      return this.parseResponse(response, task.type);
    } catch (error) {
      this.emit('error', error);
      throw new Error(`InfrastructureMaintainerAgent execution failed: ${error.message}`);
    }
  }

  /**
   * Create prompt for the task
   * @param {Object} task - Task object
   * @param {Object} context - Context object
   * @returns {string} Generated prompt
   */
  createPrompt(task, context) {
    const { type, description, requirements = [] } = task;
    
    return `Task Type: ${type}
Description: ${description}
Requirements: ${requirements.join(', ')}

Please provide a comprehensive solution.`;
  }

  /**
   * Parse agent response
   * @param {Object} response - API response
   * @param {string} type - Task type
   * @returns {Object} Parsed result
   */
  parseResponse(response, type) {
    const content = response.content?.[0]?.text || '';
    
    return {
      success: true,
      type: type,
      content: content,
      timestamp: new Date().toISOString()
    };
  }


  /**
   * monitorHealth implementation
   * @param {Object} params - Method parameters
   * @returns {Promise<Object>} Method result
   */
  async monitorHealth(params = {}) {
    const task = {
      type: 'monitorHealth',
      ...params
    };
    
    return this.execute(task);
  }

  /**
   * optimizePerformance implementation
   * @param {Object} params - Method parameters
   * @returns {Promise<Object>} Method result
   */
  async optimizePerformance(params = {}) {
    const task = {
      type: 'optimizePerformance',
      ...params
    };
    
    return this.execute(task);
  }

  /**
   * ensureReliability implementation
   * @param {Object} params - Method parameters
   * @returns {Promise<Object>} Method result
   */
  async ensureReliability(params = {}) {
    const task = {
      type: 'ensureReliability',
      ...params
    };
    
    return this.execute(task);
  }
}

module.exports = InfrastructureMaintainerAgent;