taskmanager-ai/utils/ai/anthropic.js

A CLI tool to manage your projects tasks with AI

/**
 * Módulo para integração com Anthropic (Claude)
 *
 * Este módulo implementa a geração de tarefas usando a API da Anthropic.
 */

import chalk from 'chalk';
import { simulateTasks } from './simulation.js';

/**
 * Gera tarefas usando a API da Anthropic (Claude)
 * @param {String} projectDescription - Descrição detalhada do projeto
 * @param {String} projectType - Tipo de projeto ('new' ou 'existing')
 * @param {Number} taskCount - Número de tarefas a serem geradas
 * @param {Object} aiConfig - Configuração da IA
 * @returns {Array} Array de tarefas geradas
 */
export async function generateTasksWithAnthropic(projectDescription, projectType, taskCount, aiConfig) {
  // Verificar se a chave de API está configurada
  if (!aiConfig.apiKey) {
    throw new Error('Chave de API da Anthropic não configurada');
  }

  try {
    // Aqui seria implementada a integração real com a API da Anthropic
    // Para implementação futura

    console.log(chalk.yellow('Funcionalidade de integração com Anthropic ainda não implementada.'));
    return simulateTasks(projectDescription, projectType, taskCount);

    /*
    // Exemplo de como a implementação seria:

    import Anthropic from '@anthropic-ai/sdk';

    const anthropic = new Anthropic({ apiKey: aiConfig.apiKey });

    const prompt = createPromptForTaskGeneration(projectDescription, projectType, taskCount);

    const response = await anthropic.messages.create({
      model: aiConfig.model || 'claude-3-5-sonnet-20240409',
      max_tokens: 4000,
      temperature: 0.7,
      messages: [
        { role: 'user', content: prompt }
      ]
    });

    const tasksJson = parseTasksFromResponse(response.content[0].text);
    return tasksJson;
    */

  } catch (error) {
    console.error(chalk.red(`Erro na API da Anthropic: ${error.message}`));
    // Em caso de erro, retorna tarefas simuladas
    return simulateTasks(projectDescription, projectType, taskCount);
  }
}

/**
 * Cria um prompt para geração de tarefas com Anthropic
 * @param {String} projectDescription - Descrição do projeto
 * @param {String} projectType - Tipo de projeto
 * @param {Number} taskCount - Número de tarefas
 * @returns {String} Prompt formatado
 */
function createPromptForTaskGeneration(projectDescription, projectType, taskCount) {
  return `
<instructions>
Você é um especialista em gerenciamento de projetos e desenvolvimento de software.
Gere exatamente ${taskCount} tarefas para um ${projectType === 'new' ? 'novo' : 'existente'} projeto com a seguinte descrição:

${projectDescription}

Retorne as tarefas no formato JSON, seguindo esta estrutura:
[
  {
    "id": 1,
    "title": "Título da tarefa",
    "description": "Descrição detalhada da tarefa",
    "status": "pending",
    "priority": "high|medium|low",
    "dependencies": [array de IDs das tarefas dependentes],
    "details": "Detalhes de implementação",
    "testStrategy": "Estratégia de teste",
    "category": "setup|frontend|backend|database|etc",
    "subtasks": [
      {
        "id": 1,
        "title": "Título da subtarefa",
        "description": "Descrição da subtarefa",
        "status": "pending"
      }
    ]
  }
]

Gere tarefas lógicas e bem estruturadas que sigam boas práticas de desenvolvimento.
Se o projeto for novo, comece com tarefas de setup e configuração inicial.
Se for um projeto existente, concentre-se em tarefas de desenvolvimento, melhoria ou correção.

Seu resultado deve conter apenas e exatamente o array JSON com as tarefas, sem explicações adicionais.
</instructions>
`;
}

/**
 * Faz o parse da resposta da API em formato JSON para um array de tarefas
 * @param {String} responseText - Texto da resposta da API
 * @returns {Array} Array de tarefas
 */
function parseTasksFromResponse(responseText) {
  try {
    // Encontra o objeto JSON na resposta
    const jsonMatch = responseText.match(/\[\s*{.*}\s*\]/s);

    if (jsonMatch) {
      // Adiciona timestamps a todas as tarefas
      const tasks = JSON.parse(jsonMatch[0]);
      const now = new Date().toISOString();

      for (const task of tasks) {
        task.createdAt = now;
        task.updatedAt = now;

        if (task.subtasks) {
          for (const subtask of task.subtasks) {
            subtask.createdAt = now;
            subtask.updatedAt = now;
          }
        }
      }

      return tasks;
    } else {
      throw new Error('Não foi possível encontrar JSON válido na resposta');
    }
  } catch (error) {
    console.error(chalk.red(`Erro ao fazer parse da resposta: ${error.message}`));
    throw error;
  }
}