@johpaz/hive 1.1.0

package/packages/core/src/tools/coordinator-tools.ts

@@ -0,0 +1,179 @@
+/**
+ * Coordinator-exclusive tools — native Tool interface, no LangChain.
+ *
+ * These tools are only available to agents with role='coordinator'.
+ * Workers receive only domain-specific tools + save_note + report_result.
+ */
+
+import type { Tool } from "../agent/native-tools"
+import { getDb } from "../storage/sqlite"
+import { emitCanvas } from "../canvas/emitter"
+import { saveScratchpadNote } from "../agent/conversation-store"
+import { logger } from "../utils/logger"
+
+const log = logger.child("coordinator-tools")
+
+// ─── create_agent ─────────────────────────────────────────────────────────────
+
+export const createAgentNative: Tool = {
+  name: "create_agent",
+  description:
+    "Crea un nuevo agente especializado (worker) en la base de datos. " +
+    "Úsalo cuando ningún agente existente sea adecuado para la tarea. " +
+    "Antes de crear, busca en la BD si ya existe uno con habilidades similares.",
+  parameters: {
+    type: "object",
+    properties: {
+      name: { type: "string", description: "Nombre corto del agente, ej: 'researcher'" },
+      description: {
+        type: "string",
+        description: "Qué sabe hacer este agente (para búsqueda futura)"
+      },
+      system_prompt: {
+        type: "string",
+        description: "Prompt de sistema completo para este agente"
+      },
+      tools_json: {
+        type: "string",
+        description: "JSON array de IDs de tools asignadas, ej: [\"web_search\",\"read\"]. null = todas"
+      },
+      skills_json: {
+        type: "string",
+        description: "JSON array de IDs de skills asignadas. null = todas"
+      },
+      model_id: { type: "string", description: "Modelo LLM a usar (hereda del coordinador si omite)" },
+      provider_id: { type: "string", description: "Provider ID (hereda del coordinador si omite)" },
+    },
+    required: ["name", "description", "system_prompt"],
+  },
+  execute: async (params: Record<string, unknown>, config?: any) => {
+    const db = getDb()
+    const userId = config?.configurable?.user_id || process.env.HIVE_USER_ID
+    const coordinator = db.query<any, []>(
+      "SELECT * FROM agents WHERE is_coordinator = 1 LIMIT 1"
+    ).get()
+
+    const result = db.query(`
+      INSERT INTO agents (user_id, name, description, system_prompt, role, status,
+        is_coordinator, enabled, provider_id, model_id, tools_json, skills_json, parent_id)
+      VALUES (?, ?, ?, ?, 'worker', 'idle', 0, 1, ?, ?, ?, ?, ?)
+      RETURNING id
+    `).get(
+      userId,
+      params.name,
+      params.description,
+      params.system_prompt,
+      params.provider_id || coordinator?.provider_id || "openai",
+      params.model_id || coordinator?.model_id || "gpt-4o-mini",
+      params.tools_json ?? null,
+      params.skills_json ?? null,
+      coordinator?.id ?? null,
+    ) as { id: string } | undefined
+
+    const agentId = result?.id ?? crypto.randomUUID()
+
+    emitCanvas("canvas:node_add", {
+      node: {
+        id: agentId,
+        name: params.name,
+        description: params.description,
+        role: "worker",
+        status: "idle",
+        type: "agent",
+      },
+    })
+
+    log.info(`[create_agent] Created worker '${params.name}' id=${agentId}`)
+    return JSON.stringify({
+      ok: true,
+      agentId,
+      name: params.name,
+      message: `Worker '${params.name}' creado exitosamente`,
+    })
+  },
+}
+
+// ─── save_note ────────────────────────────────────────────────────────────────
+
+export const saveNoteNative: Tool = {
+  name: "save_note",
+  description:
+    "Guarda una nota persistente en el scratchpad de la conversación. " +
+    "Útil para recordar datos del usuario, preferencias, contexto de tareas. " +
+    "Las notas sobreviven a la compresión del historial.",
+  parameters: {
+    type: "object",
+    properties: {
+      key: {
+        type: "string",
+        description: "Etiqueta de la nota, ej: 'user_name', 'task_context', 'preference'"
+      },
+      value: { type: "string", description: "Contenido de la nota" },
+    },
+    required: ["key", "value"],
+  },
+  execute: async (params: Record<string, unknown>, config?: any) => {
+    const threadId = config?.configurable?.thread_id || "default"
+    const source = config?.configurable?.agent_id || "agent"
+    saveScratchpadNote(threadId, params.key as string, params.value as string, source)
+    return JSON.stringify({ ok: true, key: params.key })
+  },
+}
+
+// ─── find_agent ───────────────────────────────────────────────────────────────
+
+export const findAgentNative: Tool = {
+  name: "find_agent",
+  description:
+    "Busca en la base de datos si ya existe un agente worker con habilidades similares " +
+    "a las requeridas. Úsalo ANTES de crear un agente nuevo para evitar duplicados.",
+  parameters: {
+    type: "object",
+    properties: {
+      query: {
+        type: "string",
+        description: "Descripción de las habilidades que necesitas"
+      },
+    },
+    required: ["query"],
+  },
+  execute: async (params: Record<string, unknown>) => {
+    const db = getDb()
+    const agents = db.query<any, []>(
+      "SELECT id, name, description, tools_json, skills_json FROM agents WHERE role = 'worker' AND status != 'archived' AND enabled = 1"
+    ).all()
+
+    if (agents.length === 0) return JSON.stringify({ agents: [] })
+
+    // Simple text search — in production the Context Compiler does FTS5
+    const q = (params.query as string).toLowerCase()
+    const matches = agents.filter((a: any) => {
+      const text = `${a.name} ${a.description}`.toLowerCase()
+      return q.split(" ").some((w) => text.includes(w))
+    })
+
+    return JSON.stringify({ agents: matches.slice(0, 5) })
+  },
+}
+
+// ─── archive_agent ────────────────────────────────────────────────────────────
+
+export const archiveAgentNative: Tool = {
+  name: "archive_agent",
+  description: "Archiva un agente worker que ya no es necesario.",
+  parameters: {
+    type: "object",
+    properties: {
+      agent_id: { type: "string", description: "ID del agente a archivar" },
+      reason: { type: "string", description: "Motivo del archivado" },
+    },
+    required: ["agent_id"],
+  },
+  execute: async (params: Record<string, unknown>) => {
+    const db = getDb()
+    db.query(
+      "UPDATE agents SET status = 'archived', updated_at = unixepoch() WHERE id = ? AND is_coordinator = 0"
+    ).run(params.agent_id as string)
+    return JSON.stringify({ ok: true, agent_id: params.agent_id })
+  },
+}