hanuscode 1.0.0__py3-none-any.whl

hanus/subagent/types.py

@@ -0,0 +1,266 @@
+# hanus/subagent/types.py
+"""
+Tipos de datos para el sistema de subagentes.
+
+Incluye todos los tipos de agentes del sistema ECC:
+- Architecture agents (5)
+- Reviewer agents (22)
+- Build resolvers (11)
+- Specialized agents (26)
+"""
+from __future__ import annotations
+from dataclasses import dataclass, field
+from typing import List, Dict, Any, Optional
+from enum import Enum
+
+
+class SubagentType(Enum):
+    """
+    Tipos de subagentes disponibles.
+
+    Categorías:
+    - EXPLORE: Solo lectura, exploración de código
+    - PLAN: Planificación y diseño
+    - GENERAL: Propósito general con todas las herramientas
+    - REVIEW: Revisión de código
+    - TEST: Escritura y ejecución de tests
+
+    Agentes especializados de ECC:
+    - ARCHITECT: Arquitectura de software
+    - CODE_ARCHITECT: Arquitectura de código
+    - NETWORK_ARCHITECT: Arquitectura de red
+    - SECURITY_REVIEWER: Revisión de seguridad
+    - PYTHON_REVIEWER: Revisión de Python
+    - TYPESCRIPT_REVIEWER: Revisión de TypeScript
+    - ... y más
+    """
+    # Tipos básicos
+    EXPLORE = "explore"
+    PLAN = "plan"
+    GENERAL = "general"
+    REVIEW = "review"
+    TEST = "test"
+
+    # Architecture Agents
+    ARCHITECT = "architect"
+    CODE_ARCHITECT = "code-architect"
+    NETWORK_ARCHITECT = "network-architect"
+    A11Y_ARCHITECT = "a11y-architect"
+    HOMELAB_ARCHITECT = "homelab-architect"
+
+    # Reviewer Agents
+    CODE_REVIEWER = "code-reviewer"
+    PYTHON_REVIEWER = "python-reviewer"
+    TYPESCRIPT_REVIEWER = "typescript-reviewer"
+    SECURITY_REVIEWER = "security-reviewer"
+    GO_REVIEWER = "go-reviewer"
+    RUST_REVIEWER = "rust-reviewer"
+    REACT_REVIEWER = "react-reviewer"
+    JAVA_REVIEWER = "java-reviewer"
+    FASTAPI_REVIEWER = "fastapi-reviewer"
+    DJANGO_REVIEWER = "django-reviewer"
+    DATABASE_REVIEWER = "database-reviewer"
+    SWIFT_REVIEWER = "swift-reviewer"
+    KOTLIN_REVIEWER = "kotlin-reviewer"
+    PHP_REVIEWER = "php-reviewer"
+    CPP_REVIEWER = "cpp-reviewer"
+    CSHARP_REVIEWER = "csharp-reviewer"
+    MLE_REVIEWER = "mle-reviewer"
+    FLUTTER_REVIEWER = "flutter-reviewer"
+
+    # Build Resolver Agents
+    BUILD_ERROR_RESOLVER = "build-error-resolver"
+    PYTHON_BUILD_RESOLVER = "python-build-resolver"
+    REACT_BUILD_RESOLVER = "react-build-resolver"
+    GO_BUILD_RESOLVER = "go-build-resolver"
+    RUST_BUILD_RESOLVER = "rust-build-resolver"
+    JAVA_BUILD_RESOLVER = "java-build-resolver"
+    SWIFT_BUILD_RESOLVER = "swift-build-resolver"
+    DART_BUILD_RESOLVER = "dart-build-resolver"
+
+    # Specialized Agents
+    PLANNER = "planner"
+    TDD_GUIDE = "tdd-guide"
+    PERFORMANCE_OPTIMIZER = "performance-optimizer"
+    CODE_EXPLORER = "code-explorer"
+    DOC_UPDATER = "doc-updater"
+    REFACTOR_CLEANER = "refactor-cleaner"
+    CHIEF_OF_STAFF = "chief-of-staff"
+    CODE_SIMPLIFIER = "code-simplifier"
+    SILENT_FAILURE_HUNTER = "silent-failure-hunter"
+    E2E_RUNNER = "e2e-runner"
+    OPENSOURCE_SANITIZER = "opensource-sanitizer"
+    OPENSOURCE_PACKAGER = "opensource-packager"
+    LOOP_OPERATOR = "loop-operator"
+    HARNESS_OPTIMIZER = "harness-optimizer"
+
+
+# Mapeo de herramientas permitidas por tipo de subagente
+SUBAGENT_TOOLS: Dict[SubagentType, Optional[List[str]]] = {
+    # Tipos básicos
+    SubagentType.EXPLORE: [
+        "read_file", "grep_search", "glob_search", "list_files",
+        "git_status", "git_diff", "git_log",
+    ],
+    SubagentType.PLAN: [
+        "read_file", "grep_search", "glob_search", "list_files",
+        "write_file",
+    ],
+    SubagentType.GENERAL: None,  # Todas las herramientas
+    SubagentType.REVIEW: [
+        "read_file", "grep_search", "glob_search", "list_files",
+        "structured_output", "exec_cmd",
+    ],
+    SubagentType.TEST: [
+        "read_file", "write_file", "exec_cmd",
+        "grep_search", "glob_search",
+    ],
+
+    # Architecture Agents - Solo lectura y análisis
+    SubagentType.ARCHITECT: ["read_file", "grep_search", "glob_search", "list_files"],
+    SubagentType.CODE_ARCHITECT: ["read_file", "grep_search", "glob_search", "list_files", "write_file"],
+    SubagentType.NETWORK_ARCHITECT: ["read_file", "grep_search", "glob_search", "list_files"],
+    SubagentType.A11Y_ARCHITECT: ["read_file", "grep_search", "glob_search", "list_files"],
+    SubagentType.HOMELAB_ARCHITECT: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+
+    # Reviewer Agents - Lectura + comandos de análisis
+    SubagentType.CODE_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.PYTHON_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.TYPESCRIPT_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.SECURITY_REVIEWER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.GO_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.RUST_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.REACT_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.JAVA_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.FASTAPI_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.DJANGO_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.DATABASE_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.SWIFT_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.KOTLIN_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.PHP_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.CPP_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.CSHARP_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.MLE_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+    SubagentType.FLUTTER_REVIEWER: ["read_file", "grep_search", "glob_search", "list_files", "exec_cmd"],
+
+    # Build Resolver Agents - Escritura limitada para fixes
+    SubagentType.BUILD_ERROR_RESOLVER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.PYTHON_BUILD_RESOLVER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.REACT_BUILD_RESOLVER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.GO_BUILD_RESOLVER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.RUST_BUILD_RESOLVER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.JAVA_BUILD_RESOLVER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.SWIFT_BUILD_RESOLVER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.DART_BUILD_RESOLVER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+
+    # Specialized Agents
+    SubagentType.PLANNER: ["read_file", "grep_search", "glob_search", "list_files"],
+    SubagentType.TDD_GUIDE: ["read_file", "write_file", "edit_file", "grep_search", "exec_cmd"],
+    SubagentType.PERFORMANCE_OPTIMIZER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.CODE_EXPLORER: ["read_file", "grep_search", "glob_search", "list_files"],
+    SubagentType.DOC_UPDATER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search"],
+    SubagentType.REFACTOR_CLEANER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.CHIEF_OF_STAFF: ["read_file", "grep_search", "glob_search", "list_files"],
+    SubagentType.CODE_SIMPLIFIER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search"],
+    SubagentType.SILENT_FAILURE_HUNTER: ["read_file", "grep_search", "glob_search", "list_files"],
+    SubagentType.E2E_RUNNER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.OPENSOURCE_SANITIZER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.OPENSOURCE_PACKAGER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search"],
+    SubagentType.LOOP_OPERATOR: ["read_file", "grep_search", "glob_search", "exec_cmd"],
+    SubagentType.HARNESS_OPTIMIZER: ["read_file", "write_file", "edit_file", "grep_search", "glob_search"],
+}
+
+
+# Prompts de sistema por tipo de subagente
+SUBAGENT_SYSTEM_PROMPTS: Dict[SubagentType, str] = {
+    # Tipos básicos
+    SubagentType.EXPLORE: """Eres un agente de exploración de código. Tu trabajo es:
+1. Leer y analizar archivos de código
+2. Buscar patrones y estructuras
+3. Resumir lo encontrado de forma clara y concisa
+
+NO puedes modificar archivos. Solo leer y reportar.
+Usa grep_search y read_file para explorar eficientemente.
+Proporciona resúmenes estructurados con rutas de archivo y números de línea.""",
+
+    SubagentType.PLAN: """Eres un agente de planificación. Tu trabajo es:
+1. Analizar el problema o requerimiento
+2. Diseñar una solución paso a paso
+3. Documentar el plan de implementación
+
+Escribe planes claros y detallados.
+Considera edge cases y posibles problemas.
+Prioriza soluciones simples y mantenibles.""",
+
+    SubagentType.GENERAL: """Eres un agente de propósito general.
+Tienes acceso a todas las herramientas para completar tu tarea.
+Trabaja de forma autónoma y completa las tareas de principio a fin.
+Verifica tu trabajo antes de reportar completado.""",
+
+    SubagentType.REVIEW: """Eres un agente de revisión de código. Tu trabajo es:
+1. Analizar código en busca de bugs
+2. Identificar problemas de seguridad
+3. Sugerir mejoras de calidad
+4. Verificar adherencia a best practices
+
+Proporciona feedback específico con rutas de archivo y números de línea.
+Prioriza problemas críticos sobre mejoras cosméticas.""",
+
+    SubagentType.TEST: """Eres un agente de testing. Tu trabajo es:
+1. Escribir tests para el código proporcionado
+2. Ejecutar los tests y verificar que pasan
+3. Reportar cobertura y problemas encontrados
+
+Usa pytest para Python, jest para JavaScript, etc.
+Los tests deben ser claros, mantenibles y comprehensivos.""",
+
+    # Los prompts especializados se cargan desde los archivos .md
+}
+
+
+@dataclass
+class SubagentConfig:
+    """Configuración para crear un subagente."""
+    agent_type: SubagentType
+    task_description: str
+    context: str = ""
+    max_iterations: int = 15
+    max_tokens: int = 4096
+    isolation_mode: str = "shared"  # "shared" | "worktree"
+    model_override: Optional[str] = None
+
+
+@dataclass
+class SubagentResult:
+    """Resultado de la ejecución de un subagente."""
+    success: bool
+    output: str
+    files_modified: List[str] = field(default_factory=list)
+    tokens_used: int = 0
+    cost_usd: float = 0.0
+    error: Optional[str] = None
+    agent_id: str = ""
+
+
+# Utilidades para mapeo de tipos
+
+def get_subagent_type(name: str) -> Optional[SubagentType]:
+    """Obtiene el tipo de subagente desde un nombre string."""
+    name_lower = name.lower().replace("-", "_").replace(" ", "_")
+    for st in SubagentType:
+        if st.value == name or st.name.lower() == name_lower:
+            return st
+    return None
+
+
+def get_tools_for_type(agent_type: SubagentType) -> Optional[List[str]]:
+    """Obtiene las herramientas permitidas para un tipo de agente."""
+    return SUBAGENT_TOOLS.get(agent_type)
+
+
+def get_prompt_for_type(agent_type: SubagentType) -> str:
+    """Obtiene el prompt de sistema para un tipo de agente."""
+    return SUBAGENT_SYSTEM_PROMPTS.get(
+        agent_type,
+        SUBAGENT_SYSTEM_PROMPTS[SubagentType.GENERAL]
+    )

hanus/suggestions/__init__.py

@@ -0,0 +1,5 @@
+# hanus/suggestions/__init__.py
+"""Proactive suggestions system for HanusCode."""
+from .proactive import ProactiveSuggester, Suggestion, SuggestionType
+
+__all__ = ["ProactiveSuggester", "Suggestion", "SuggestionType"]