@hilbras/omninode 0.3.0 → 0.5.0

package/dist/index.js

@@ -123,6 +123,8 @@ var agentConfigSchema = z.object({
   args: z.array(z.string()).optional(),
   cwd: z.string().optional(),
   env: z.record(z.string()).optional(),
+  input_mode: z.enum(["stdin", "arg", "protocol"]).default("stdin"),
+  timeout_ms: z.number().int().positive().optional(),
   metadata: z.record(z.unknown()).optional()
 }).strict();
 var roleConfigSchema = z.object({
@@ -234,6 +236,8 @@ function toAgentConfig(raw) {
     args: raw.args,
     cwd: raw.cwd,
     env: raw.env,
+    inputMode: raw.input_mode,
+    timeoutMs: raw.timeout_ms,
     metadata: raw.metadata
   };
 }
@@ -591,12 +595,520 @@ var ModelRegistry = class {
   }
 };
 
+// src/agents/process/index.ts
+import { spawn } from "child_process";
+var DEFAULT_TIMEOUT_MS2 = 6e5;
+var KILL_GRACE_MS = 5e3;
+function composeTaskText(input) {
+  const sections = [`# Objective
+${input.objective}`];
+  if (input.role) {
+    const lines = ["# Role", `${input.role.name} (${input.role.id})`];
+    if (input.role.responsibilities.length > 0) {
+      lines.push("", "Responsibilities:", ...input.role.responsibilities.map((r) => `- ${r}`));
+    }
+    if (input.role.expectedOutputs.length > 0) {
+      lines.push("", "Expected outputs:", ...input.role.expectedOutputs.map((o) => `- ${o}`));
+    }
+    sections.push(lines.join("\n"));
+  }
+  if (input.context) sections.push(`# Context
+${input.context}`);
+  if (input.instruction) sections.push(`# Instruction
+${input.instruction}`);
+  return sections.join("\n\n");
+}
+function buildProtocolMessages(input, agentName) {
+  const meta = {
+    taskId: input.taskId,
+    agent: agentName,
+    timestamp: (/* @__PURE__ */ new Date()).toISOString()
+  };
+  const messages = [
+    { type: "TASK", payload: { taskId: input.taskId, objective: input.objective } }
+  ];
+  if (input.role) messages.push({ type: "ROLE", payload: input.role });
+  if (input.context) messages.push({ type: "CONTEXT", payload: { body: input.context } });
+  if (input.instruction) messages.push({ type: "INSTRUCTION", payload: { text: input.instruction } });
+  return messages.map((message) => JSON.stringify({ meta, message }));
+}
+var ProcessAgent = class {
+  config;
+  info;
+  log;
+  running = /* @__PURE__ */ new Map();
+  constructor(config, log = logger) {
+    if (!config.command) {
+      throw new AgentError("AGENT_NOT_FOUND", `Agent "${config.name}" has no command configured.`);
+    }
+    this.config = config;
+    this.log = log.child({ agent: config.name });
+    this.info = {
+      name: config.name,
+      integration: "process",
+      command: config.command,
+      cwd: config.cwd,
+      status: "ready",
+      capabilities: [config.inputMode ?? "stdin"]
+    };
+  }
+  async run(input) {
+    const command = this.config.command;
+    if (!command) {
+      throw new AgentError("AGENT_NOT_FOUND", `Agent "${this.config.name}" has no command configured.`);
+    }
+    const mode = this.config.inputMode ?? "stdin";
+    const timeoutMs = this.config.timeoutMs ?? DEFAULT_TIMEOUT_MS2;
+    return new Promise((resolve, reject) => {
+      const child = spawn(
+        command,
+        [...this.config.args ?? [], ...mode === "arg" ? [composeTaskText(input)] : []],
+        {
+          cwd: this.config.cwd,
+          env: { ...process.env, ...this.config.env },
+          stdio: ["pipe", "pipe", "pipe"]
+        }
+      );
+      this.running.set(input.taskId, child);
+      let stdout = "";
+      let stderr = "";
+      let timedOut = false;
+      let settled = false;
+      const timer = setTimeout(() => {
+        timedOut = true;
+        this.log.warn(`Task ${input.taskId} exceeded ${timeoutMs}ms \u2014 killing process.`);
+        child.kill("SIGTERM");
+        setTimeout(() => child.kill("SIGKILL"), KILL_GRACE_MS).unref();
+      }, timeoutMs);
+      const settle = (finish) => {
+        if (settled) return;
+        settled = true;
+        clearTimeout(timer);
+        this.running.delete(input.taskId);
+        finish();
+      };
+      child.stdout?.on("data", (chunk) => {
+        stdout += chunk.toString();
+      });
+      child.stderr?.on("data", (chunk) => {
+        stderr += chunk.toString();
+      });
+      child.stdin?.on("error", () => {
+      });
+      if (mode !== "arg") {
+        const payload = mode === "protocol" ? buildProtocolMessages(input, this.config.name).join("\n") + "\n" : composeTaskText(input) + "\n";
+        child.stdin?.write(payload);
+        child.stdin?.end();
+      }
+      child.on("error", (error) => {
+        settle(() => {
+          if (error.code === "ENOENT") {
+            reject(
+              new AgentError(
+                "AGENT_NOT_FOUND",
+                `Agent "${this.config.name}": command "${command}" was not found. Is it installed and on PATH?`,
+                { cause: error }
+              )
+            );
+            return;
+          }
+          reject(
+            new AgentError("AGENT_FAILED", `Agent "${this.config.name}" failed to start: ${error.message}.`, {
+              cause: error
+            })
+          );
+        });
+      });
+      child.on("close", (code, signal) => {
+        settle(() => {
+          if (timedOut) {
+            reject(
+              new AgentError(
+                "AGENT_TIMEOUT",
+                `Agent "${this.config.name}" timed out after ${timeoutMs}ms and was killed.`,
+                { details: { taskId: input.taskId, stdout: stdout.slice(0, 4e3), stderr: stderr.slice(0, 2e3) } }
+              )
+            );
+            return;
+          }
+          const logs = stderr.trim().length > 0 ? stderr.split("\n") : void 0;
+          if (mode === "protocol") {
+            try {
+              resolve(
+                interpretProtocolRun({
+                  input,
+                  agentName: this.config.name,
+                  stdout,
+                  exitCode: code,
+                  logs
+                })
+              );
+            } catch (error) {
+              reject(error);
+            }
+            return;
+          }
+          const trimmed = stdout.trim();
+          resolve({
+            taskId: input.taskId,
+            status: code === 0 ? "completed" : "failed",
+            summary: trimmed.length > 0 ? firstLine(trimmed) : void 0,
+            rawOutput: trimmed.length > 0 ? trimmed : void 0,
+            exitCode: code ?? void 0,
+            logs,
+            error: code === 0 ? void 0 : `Process exited with code ${code ?? "unknown"}${signal ? ` (signal: ${signal})` : ""}`
+          });
+        });
+      });
+    });
+  }
+  /** Ask the agent to stop current work (SIGTERM). */
+  async cancel(taskId) {
+    const child = this.running.get(taskId);
+    if (child) child.kill("SIGTERM");
+  }
+  runningTaskIds() {
+    return [...this.running.keys()];
+  }
+};
+function interpretProtocolRun({ input, agentName, stdout, exitCode, logs }) {
+  const reports = [];
+  let summary;
+  let error;
+  let sawProtocolMessage = false;
+  for (const line of stdout.split("\n")) {
+    const trimmed2 = line.trim();
+    if (trimmed2.length === 0) continue;
+    const message = parseAgentMessage(trimmed2);
+    if (!message) continue;
+    sawProtocolMessage = true;
+    if (message.type === "REPORT") {
+      const report = normalizeReport(message.payload, input.taskId, agentName, reports.length);
+      if (report) reports.push(report);
+    } else if (message.type === "COMPLETION") {
+      const payload = message.payload;
+      if (typeof payload?.summary === "string") summary = payload.summary;
+    } else if (message.type === "ERROR") {
+      const payload = message.payload;
+      const detail = typeof payload?.message === "string" ? payload.message : "unknown error";
+      const code = typeof payload?.code === "string" ? ` [${payload.code}]` : "";
+      error = `Agent error${code}: ${detail}`;
+    }
+  }
+  if (!sawProtocolMessage && exitCode === 0) {
+    throw new ProtocolError(
+      `Agent "${agentName}" produced no protocol messages on stdout (input mode "protocol").`,
+      { details: { taskId: input.taskId, stdout: stdout.slice(0, 2e3) } }
+    );
+  }
+  const trimmed = stdout.trim();
+  return {
+    taskId: input.taskId,
+    status: error !== void 0 || exitCode !== 0 ? "failed" : "completed",
+    summary: summary ?? reports[0]?.summary,
+    rawOutput: trimmed.length > 0 ? trimmed : void 0,
+    exitCode: exitCode ?? void 0,
+    logs,
+    reports: reports.length > 0 ? reports : void 0,
+    error: error ?? (exitCode !== 0 ? `Process exited with code ${exitCode ?? "unknown"}` : void 0)
+  };
+}
+function parseAgentMessage(line) {
+  try {
+    const parsed = JSON.parse(line);
+    if (typeof parsed.type === "string") {
+      return { type: parsed.type, payload: parsed.payload };
+    }
+    const inner = parsed.message;
+    if (inner && typeof inner.type === "string") {
+      return { type: inner.type, payload: inner.payload };
+    }
+  } catch {
+  }
+  return void 0;
+}
+function normalizeReport(payload, taskId, agentName, index) {
+  if (payload === null || typeof payload !== "object") return void 0;
+  const raw = payload;
+  if (typeof raw.summary !== "string") return void 0;
+  return {
+    id: typeof raw.id === "string" ? raw.id : `${taskId}-report-${index}`,
+    taskId: typeof raw.taskId === "string" ? raw.taskId : taskId,
+    agent: typeof raw.agent === "string" ? raw.agent : agentName,
+    summary: raw.summary,
+    findings: Array.isArray(raw.findings) ? raw.findings : [],
+    recommendations: Array.isArray(raw.recommendations) ? raw.recommendations : [],
+    createdAt: typeof raw.createdAt === "string" ? raw.createdAt : (/* @__PURE__ */ new Date()).toISOString()
+  };
+}
+function firstLine(text, maxLength = 200) {
+  const line = text.split("\n", 1)[0] ?? text;
+  return line.length > maxLength ? `${line.slice(0, maxLength)}\u2026` : line;
+}
+
+// src/agents/registry.ts
+var AgentRegistry = class {
+  agents = /* @__PURE__ */ new Map();
+  register(agent) {
+    this.agents.set(agent.info.name, agent);
+  }
+  get(name) {
+    return this.agents.get(name);
+  }
+  list() {
+    return [...this.agents.values()].sort((a, b) => a.info.name.localeCompare(b.info.name));
+  }
+};
+
+// src/agents/factory.ts
+function createAgent(config) {
+  if (config.integration === "process") {
+    return new ProcessAgent(config);
+  }
+  if (config.integration === "native") {
+    throw new OmniNodeError(
+      "NOT_IMPLEMENTED",
+      `Native agent adapters are planned for a later phase. Use input over stdin/stdout (type "cli") for "${config.name}" for now.`
+    );
+  }
+  throw new OmniNodeError("NOT_IMPLEMENTED", `Agent integration "${config.integration}" has no adapter yet.`);
+}
+function buildAgentRegistry(config, log = logger) {
+  const registry = new AgentRegistry();
+  for (const agentConfig of config.project.agents) {
+    try {
+      registry.register(createAgent(agentConfig));
+    } catch (error) {
+      log.warn(
+        `Skipping agent "${agentConfig.name}": ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
+  return registry;
+}
+
+// src/roles/index.ts
+var RoleRegistry = class {
+  roles = /* @__PURE__ */ new Map();
+  register(role) {
+    this.roles.set(role.id, role);
+  }
+  registerAll(roles) {
+    for (const role of roles) this.register(role);
+  }
+  get(id) {
+    return this.roles.get(id);
+  }
+  list() {
+    return [...this.roles.values()].sort((a, b) => a.id.localeCompare(b.id));
+  }
+};
+function buildRoleRegistry(config) {
+  const registry = new RoleRegistry();
+  registry.registerAll(config.project.roles);
+  return registry;
+}
+
+// src/tasks/store.ts
+import { mkdir, readFile, rename, writeFile } from "fs/promises";
+import { dirname } from "path";
+var FileTaskStore = class {
+  filePath;
+  constructor(directory = `${process.cwd()}/.omninode`) {
+    this.filePath = `${directory}/tasks.json`;
+  }
+  async save(task) {
+    const tasks = await this.load();
+    const index = tasks.findIndex((t) => t.id === task.id);
+    if (index >= 0) tasks[index] = task;
+    else tasks.push(task);
+    await this.write(tasks);
+  }
+  async get(id) {
+    const tasks = await this.load();
+    return tasks.find((t) => t.id === id);
+  }
+  async list(filter = {}) {
+    const tasks = await this.load();
+    return tasks.filter((task) => filter.status ? task.status === filter.status : true).filter((task) => filter.project ? task.project === filter.project : true).sort((a, b) => a.createdAt.localeCompare(b.createdAt));
+  }
+  async load() {
+    try {
+      const raw = await readFile(this.filePath, "utf8");
+      const parsed = JSON.parse(raw);
+      return Array.isArray(parsed) ? parsed : [];
+    } catch {
+      return [];
+    }
+  }
+  async write(tasks) {
+    const tmp = `${this.filePath}.tmp`;
+    await mkdir(dirname(this.filePath), { recursive: true });
+    await writeFile(tmp, `${JSON.stringify(tasks, null, 2)}
+`, "utf8");
+    await rename(tmp, this.filePath);
+  }
+};
+
+// src/tasks/engine.ts
+import { randomBytes } from "crypto";
+var ALLOWED_TRANSITIONS = {
+  created: ["queued", "running", "cancelled"],
+  queued: ["running", "cancelled"],
+  running: ["completed", "failed", "cancelled"],
+  waiting: ["running", "cancelled"],
+  completed: [],
+  failed: [],
+  cancelled: []
+};
+var TaskEngine = class {
+  constructor(options) {
+    this.options = options;
+    this.log = (options.log ?? logger).child({ component: "tasks" });
+  }
+  options;
+  log;
+  get store() {
+    return this.options.store;
+  }
+  async create(input) {
+    if (!input.objective || input.objective.trim().length === 0) {
+      throw new TaskError("TASK_INVALID", "Task objective must not be empty.");
+    }
+    if (input.role && !this.options.roles.get(input.role)) {
+      throw new TaskError(
+        "TASK_INVALID",
+        `Role "${input.role}" is not defined. Add it under project.roles in omninode.yaml.`
+      );
+    }
+    if (input.agent && !this.options.agents.get(input.agent)) {
+      throw new TaskError(
+        "TASK_INVALID",
+        `Agent "${input.agent}" is not registered. Add it with \`omninode agent add\`.`
+      );
+    }
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    const task = {
+      id: generateTaskId(),
+      objective: input.objective.trim(),
+      status: "created",
+      createdAt: now,
+      updatedAt: now,
+      ...input.project !== void 0 ? { project: input.project } : {},
+      ...input.role !== void 0 ? { role: input.role } : {},
+      ...input.agent !== void 0 ? { agent: input.agent } : {},
+      ...input.context !== void 0 ? { context: input.context } : {}
+    };
+    await this.options.store.save(task);
+    this.log.info(`Created task ${task.id}.`);
+    return task;
+  }
+  async queue(id) {
+    const task = await this.mustGet(id);
+    return this.transition(task, "queued");
+  }
+  /** Runs the task through its assigned agent and records the result. */
+  async run(id) {
+    const task = await this.mustGet(id);
+    if (task.status !== "created" && task.status !== "queued") {
+      throw new TaskError("TASK_INVALID", `Task ${id} is "${task.status}" and cannot be run.`);
+    }
+    if (!task.agent) {
+      throw new TaskError("TASK_INVALID", `Task ${id} has no agent assigned. Re-create it with --agent.`);
+    }
+    const agent = this.options.agents.get(task.agent);
+    if (!agent) {
+      throw new TaskError("TASK_INVALID", `Agent "${task.agent}" is not registered anymore.`);
+    }
+    const role = task.role ? this.options.roles.get(task.role) : void 0;
+    const running = await this.transition(task, "running");
+    this.log.info(`Running task ${task.id} on agent "${task.agent}".`);
+    let output;
+    try {
+      output = await agent.run({
+        taskId: running.id,
+        objective: running.objective,
+        ...role ? { role } : {},
+        ...running.context?.background ? { context: running.context.background } : {}
+      });
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      this.log.error(`Task ${task.id} failed: ${message}`);
+      return this.transition(running, "failed", {
+        result: { error: message, finishedAt: (/* @__PURE__ */ new Date()).toISOString() }
+      });
+    }
+    const result = {
+      ...output.summary !== void 0 ? { summary: output.summary } : {},
+      ...output.reports !== void 0 && output.reports.length > 0 ? { reports: output.reports } : {},
+      ...output.error !== void 0 ? { error: output.error } : {},
+      finishedAt: (/* @__PURE__ */ new Date()).toISOString()
+    };
+    const finalStatus = output.status === "completed" ? "completed" : "failed";
+    this.log.info(`Task ${task.id} finished: ${finalStatus}.`);
+    return this.transition(running, finalStatus, { result });
+  }
+  async cancel(id) {
+    const task = await this.mustGet(id);
+    if (!ALLOWED_TRANSITIONS[task.status].includes("cancelled")) {
+      throw new TaskError("TASK_INVALID", `Task ${id} is "${task.status}" and cannot be cancelled.`);
+    }
+    if (task.agent) {
+      const agent = this.options.agents.get(task.agent);
+      await agent?.cancel?.(task.id);
+    }
+    return this.transition(task, "cancelled");
+  }
+  async get(id) {
+    return this.options.store.get(id);
+  }
+  async list(filter) {
+    return this.options.store.list(filter);
+  }
+  async mustGet(id) {
+    const task = await this.options.store.get(id);
+    if (!task) {
+      throw new TaskError("TASK_NOT_FOUND", `Task "${id}" does not exist in this project's task store.`);
+    }
+    return task;
+  }
+  async transition(task, to, extra = {}) {
+    const allowed = ALLOWED_TRANSITIONS[task.status] ?? [];
+    if (!allowed.includes(to)) {
+      throw new TaskError("TASK_INVALID", `Task ${task.id} cannot move from "${task.status}" to "${to}".`);
+    }
+    const updated = {
+      ...task,
+      status: to,
+      updatedAt: (/* @__PURE__ */ new Date()).toISOString(),
+      ...extra.result !== void 0 ? { result: { ...task.result, ...extra.result } } : {}
+    };
+    await this.options.store.save(updated);
+    return updated;
+  }
+};
+function generateTaskId() {
+  return `task-${Date.now().toString(36)}-${randomBytes(3).toString("hex")}`;
+}
+
+// src/tasks/engine-factory.ts
+function createTaskEngine(config, log = logger, directory) {
+  return new TaskEngine({
+    agents: buildAgentRegistry(config, log),
+    roles: buildRoleRegistry(config),
+    store: new FileTaskStore(directory),
+    log
+  });
+}
+
 // src/version.ts
-var OMNINODE_VERSION = "0.3.0";
+var OMNINODE_VERSION = "0.4.0";
 export {
   AgentError,
+  AgentRegistry,
   ConfigError,
   ConsoleLogSink,
+  FileTaskStore,
   Logger,
   MemoryError,
   ModelRegistry,
@@ -606,13 +1118,21 @@ export {
   OmniNodeError,
   OpenAICompatibleProvider,
   PipelineError,
+  ProcessAgent,
   ProtocolError,
   ProviderError,
+  RoleRegistry,
+  TaskEngine,
   TaskError,
   agentConfigSchema,
   appConfigSchema,
   authHeaders,
+  buildAgentRegistry,
+  buildRoleRegistry,
+  composeTaskText,
+  createAgent,
   createProvider,
+  createTaskEngine,
   defaultProjectConfigYaml,
   expandEnvRefs,
   findConfigFile,