@pantheon.ai/agents 0.0.4 → 0.0.5

package/dist/index.js

@@ -4,6 +4,7 @@ import { createCommand } from "commander";
 import expandTilde from "expand-tilde";
 import path from "node:path";
 import * as process$1 from "node:process";
+import readline from "node:readline/promises";
 import { multistream, pino, transport } from "pino";
 import * as fs$1 from "node:fs";
 import { Kysely, MysqlDialect } from "kysely";
@@ -15,7 +16,7 @@ var __commonJSMin = (cb, mod) => () => (mod || cb((mod = { exports: {} }).export
 var __require = /* @__PURE__ */ createRequire(import.meta.url);
 
 //#endregion
-//#region schemas/task-list.ts
+//#region src/schemas/task-list.ts
 const taskItemSchema = z.discriminatedUnion("status", [
 	z.object({
 		status: z.literal("pending"),
@@ -70,13 +71,13 @@ const taskItemSchema = z.discriminatedUnion("status", [
 ]);
 
 //#endregion
-//#region utils/get-error-message.ts
+//#region src/utils/get-error-message.ts
 function getErrorMessage(error) {
 	return String(error?.message ?? error);
 }
 
 //#endregion
-//#region task-list/task-list-provider.ts
+//#region src/providers/task-list-provider.ts
 var TaskListProvider = class {
 	logger;
 	agentName;
@@ -156,7 +157,7 @@ var TaskListProvider = class {
 };
 
 //#endregion
-//#region task-list/task-list-tidb-provider.ts
+//#region src/providers/task-list-tidb-provider.ts
 var TaskListTidbProvider = class extends TaskListProvider {
 	db;
 	constructor(agentName, logger) {
@@ -178,6 +179,9 @@ var TaskListTidbProvider = class extends TaskListProvider {
 		if (config == null) return null;
 		return config;
 	}
+	async getAgentConfigs() {
+		return await this.db.selectFrom("agent_project_config").selectAll().where("agent", "=", this.agentName).execute();
+	}
 	async setAgentConfig({ skills, ...config }) {
 		await this.db.insertInto("agent_project_config").values({
 			agent: this.agentName,
@@ -196,6 +200,9 @@ var TaskListTidbProvider = class extends TaskListProvider {
 		if (order_by) builder = builder.orderBy(order_by, order_direction);
 		return (await builder.execute()).map((item) => taskItemSchema.parse(item));
 	}
+	async listAgentNames() {
+		return (await this.db.selectFrom("task").select("agent").distinct().execute()).map((item) => item.agent);
+	}
 	async updateTask(taskItem) {
 		const { id, started_at, ended_at, queued_at, ...rest } = taskItem;
 		await this.db.updateTable("task").set({
@@ -219,6 +226,10 @@ var TaskListTidbProvider = class extends TaskListProvider {
 			id: String(insertId)
 		};
 	}
+	async deleteTask(taskId) {
+		const result = await this.db.deleteFrom("task").where("id", "=", taskId).where("agent", "=", this.agentName).executeTakeFirst();
+		return Number(result.numDeletedRows ?? 0) > 0;
+	}
 };
 
 //#endregion
@@ -1326,7 +1337,7 @@ function getServerErrorMessage(error) {
 }
 
 //#endregion
-//#region core/pantheon.ts
+//#region src/core/pantheon.ts
 const executor = createApiExecutor({
 	baseUrl: "https://pantheon-ai.tidb.ai",
 	headers: { Authorization: `Bearer ${process.env.PANTHEON_API_KEY}` },
@@ -1376,7 +1387,7 @@ async function executeOnPantheon({ projectId, branchId, prompt, agent }) {
 }
 
 //#endregion
-//#region core/task-list.ts
+//#region src/core/task-list.ts
 async function startTaskListLoop(agentName, { loopInterval = 5 }, logger) {
 	logger.info("Starting task list loop...");
 	let i = 0;
@@ -1458,7 +1469,7 @@ async function startPendingTask(provider, task, logger) {
 }
 
 //#endregion
-//#region core/index.ts
+//#region src/core/index.ts
 async function runAgent(name, options, logger) {
 	const agentDir = path.join(options.dataDir, "agents", name);
 	const pidFile = path.join(agentDir, "pid");
@@ -1577,6 +1588,145 @@ async function addTask(name, options) {
 	console.log(`Task ${task.task} queued successfully.`);
 	await provider.close();
 }
+async function deleteTask(agentName, taskId) {
+	const provider = new TaskListTidbProvider(agentName, pino());
+	try {
+		const task = await provider.getTask(taskId);
+		if (!task) return null;
+		if (!await provider.deleteTask(taskId)) return null;
+		return task;
+	} finally {
+		await provider.close();
+	}
+}
+async function showAgentConfig(agentName, projectId) {
+	const provider = new TaskListTidbProvider(agentName, pino());
+	try {
+		return await provider.getAgentConfig(projectId);
+	} finally {
+		await provider.close();
+	}
+}
+async function showAgentConfigs(agentName) {
+	const provider = new TaskListTidbProvider(agentName, pino());
+	try {
+		return await provider.getAgentConfigs();
+	} finally {
+		await provider.close();
+	}
+}
+async function showTasks(name, options) {
+	const provider = new TaskListTidbProvider(name, pino());
+	try {
+		return await provider.getTasks({
+			status: options.status,
+			order_by: options.orderBy,
+			order_direction: options.orderDirection
+		});
+	} finally {
+		await provider.close();
+	}
+}
+async function listAgentNames() {
+	const provider = new TaskListTidbProvider("list-agents", pino());
+	try {
+		return await provider.listAgentNames();
+	} finally {
+		await provider.close();
+	}
+}
+async function showTasksForAgents(options) {
+	const selectedAgents = options.allAgents ? await listAgentNames() : options.agents;
+	const tasksWithAgent = await Promise.all(selectedAgents.map(async (agent) => {
+		return (await showTasks(agent, {
+			status: options.status,
+			orderBy: options.orderBy,
+			orderDirection: options.orderDirection
+		})).map((task) => ({
+			agent,
+			task
+		}));
+	})).then((groups) => groups.flat());
+	const orderKey = options.orderBy ?? "queued_at";
+	const orderDirection = options.orderDirection ?? "desc";
+	tasksWithAgent.sort((a, b) => {
+		const aValue = a.task[orderKey];
+		const bValue = b.task[orderKey];
+		const aTime = aValue instanceof Date ? aValue.getTime() : 0;
+		const bTime = bValue instanceof Date ? bValue.getTime() : 0;
+		return orderDirection === "asc" ? aTime - bTime : bTime - aTime;
+	});
+	if (options.limit != null) return tasksWithAgent.slice(0, options.limit);
+	return tasksWithAgent;
+}
+function formatRelativeTime(value) {
+	if (!value) return "";
+	const diffMs = Date.now() - value.getTime();
+	const absMs = Math.abs(diffMs);
+	const tense = diffMs >= 0 ? "ago" : "from now";
+	const seconds = Math.floor(absMs / 1e3);
+	const minutes = Math.floor(seconds / 60);
+	const hours = Math.floor(minutes / 60);
+	const days = Math.floor(hours / 24);
+	if (days > 0) return `${days}d ${tense}`;
+	if (hours > 0) return `${hours}h ${tense}`;
+	if (minutes > 0) return `${minutes}m ${tense}`;
+	return `${seconds}s ${tense}`;
+}
+function formatDate(value) {
+	return value ? value.toISOString() : "";
+}
+function truncateText(value, maxLength) {
+	if (value.length <= maxLength) return value;
+	return `${value.slice(0, Math.max(0, maxLength - 1))}…`;
+}
+function colorText(value, colorCode) {
+	return `\u001b[${colorCode}m${value}\u001b[0m`;
+}
+function formatStatus(status, useColor) {
+	if (!useColor) return status;
+	switch (status) {
+		case "pending": return colorText(status, 33);
+		case "running": return colorText(status, 36);
+		case "completed": return colorText(status, 32);
+		case "failed": return colorText(status, 31);
+		case "cancelled": return colorText(status, 90);
+	}
+}
+function formatConciseTaskLine(options) {
+	const { agent, task, maxTaskLength, useColor } = options;
+	let timestamp = task.queued_at;
+	if (task.status === "running") timestamp = task.started_at;
+	if (task.status === "completed" || task.status === "failed") timestamp = task.ended_at;
+	if (task.status === "cancelled") timestamp = task.cancelled_at ?? task.queued_at;
+	const statusText = formatStatus(task.status, useColor);
+	const timeText = formatRelativeTime(timestamp);
+	const taskText = truncateText(task.task, maxTaskLength);
+	return [
+		agent,
+		statusText,
+		task.id,
+		timeText,
+		taskText
+	].filter((part) => part !== "").join("  ");
+}
+function formatTaskTableRow(options) {
+	const { agent, task, maxTaskLength } = options;
+	const branchId = "branch_id" in task ? task.branch_id ?? "" : "";
+	const startedAt = "started_at" in task ? task.started_at : void 0;
+	const endedAt = "ended_at" in task ? task.ended_at : void 0;
+	return {
+		agent,
+		id: task.id,
+		status: task.status,
+		project_id: task.project_id,
+		branch_id: branchId,
+		queued_at: formatDate(task.queued_at),
+		started_at: formatDate(startedAt),
+		ended_at: formatDate(endedAt),
+		task: truncateText(task.task, maxTaskLength)
+	};
+}
 async function assertsSingleton(logger, pidFile) {
 	try {
 		const pid = await fs$1.promises.readFile(pidFile, "utf-8");
@@ -1969,10 +2119,26 @@ var require_cli_options = /* @__PURE__ */ __commonJSMin(((exports, module) => {
 
 //#endregion
 //#region package.json
-var version = "0.0.4";
+var version = "0.0.5";
 
 //#endregion
-//#region index.ts
+//#region src/cli/index.ts
+const taskStatuses = [
+	"pending",
+	"running",
+	"completed",
+	"failed",
+	"cancelled"
+];
+const orderByFields = [
+	"queued_at",
+	"started_at",
+	"ended_at"
+];
+const orderDirections = ["asc", "desc"];
+function parseCommaList(value) {
+	return value.split(",").map((item) => item.trim()).filter((item) => item !== "");
+}
 if (!process$1.env.PANTHEON_API_KEY) console.error("PANTHEON_API_KEY environment variable is not set.");
 if (!process$1.env.DATABASE_URL) console.error("DATABASE_URL environment variable is not set.");
 const configAgentCommand = createCommand("pantheon-agents config").version(version).description("Configure agent for pantheon project").argument("<name>", "The name of the agent.").argument("<role>", "The role of the agent.").argument("<project-id>", "The project id of the agent.").option("--skills <skills>", "The skills of the agent. Multiple values are separated by comma.", (val) => val.split(",").map((s) => s.trim()).filter((s) => s !== ""), []).option("--execute-agent <agent>", "The execute agent of the agent.", "codex").option("--root-branch-id <branchId>", "The root branch id of the agent. Default to project root branch id.").option("--no-bootstrap", "Prevent bootstrap base branch for agent. Use the root branch as base branch.").action(async function() {
@@ -1990,6 +2156,74 @@ const addTaskCommand = createCommand("pantheon-agents add-task").version(version
 		prompt: taskPrompt
 	});
 });
+const deleteTaskCommand = createCommand("pantheon-agents delete-task").version(version).description("Delete a task for an agent").argument("<name>", "The name of the agent.").argument("<task-id>", "The id of the task.").action(async function() {
+	const [name, taskId] = this.args;
+	const rl = readline.createInterface({
+		input: process$1.stdin,
+		output: process$1.stdout
+	});
+	try {
+		if ((await rl.question(`Type the task id (${taskId}) to confirm deletion: `)).trim() !== taskId) {
+			console.error("Confirmation failed. Task id did not match.");
+			process$1.exit(1);
+		}
+		if ((await rl.question("Type DELETE to permanently remove this task: ")).trim() !== "DELETE") {
+			console.error("Confirmation failed. Aborting deletion.");
+			process$1.exit(1);
+		}
+		const deletedTask = await deleteTask(name, taskId);
+		if (!deletedTask) {
+			console.error(`Task ${taskId} not found for agent ${name}.`);
+			process$1.exit(1);
+		}
+		console.log(`Deleted task ${taskId} for agent ${name}. Status was ${deletedTask.status}.`);
+	} finally {
+		rl.close();
+	}
+});
+const showConfigCommand = createCommand("pantheon-agents show-config").version(version).description("Show agent config for a project").argument("<name>", "The name of the agent.").argument("[project-id]", "The project id.").option("--json", "Output config as JSON.").action(async function() {
+	const [name, projectId] = this.args;
+	const options = this.opts();
+	if (projectId) {
+		const config = await showAgentConfig(name, projectId);
+		if (!config) {
+			console.error(`No config found for agent ${name} and project ${projectId}.`);
+			process$1.exit(1);
+		}
+		if (options.json) {
+			console.log(JSON.stringify(config, null, 2));
+			return;
+		}
+		console.table([{
+			agent: config.agent,
+			project_id: config.project_id,
+			base_branch_id: config.base_branch_id,
+			role: config.role,
+			execute_agent: config.execute_agent,
+			prototype_url: config.prototype_url,
+			skills: Array.isArray(config.skills) ? config.skills.join(", ") : String(config.skills)
+		}]);
+		return;
+	}
+	const configs = await showAgentConfigs(name);
+	if (!configs.length) {
+		console.error(`No configs found for agent ${name}.`);
+		process$1.exit(1);
+	}
+	if (options.json) {
+		console.log(JSON.stringify(configs, null, 2));
+		return;
+	}
+	console.table(configs.map((config) => ({
+		agent: config.agent,
+		project_id: config.project_id,
+		base_branch_id: config.base_branch_id,
+		role: config.role,
+		execute_agent: config.execute_agent,
+		prototype_url: config.prototype_url,
+		skills: Array.isArray(config.skills) ? config.skills.join(", ") : String(config.skills)
+	})));
+});
 const runAgentCommand = createCommand("pantheon-agents run").version(version).description("Start a pantheon agents").argument("<name>", "The name of the agent.").option("--data-dir [dir]", "Data directory.", expandTilde("~/.pantheon-agents")).option("--mcp-port", "The port of the MCP server. Defaults to a random port.").option("--loop-interval <seconds>", "The interval of the loop in seconds. Defaults to 5.", (val) => parseInt(val, 10), 5).action(async function() {
 	const [name] = this.args;
 	const options = this.opts();
@@ -2015,6 +2249,79 @@ const runAgentCommand = createCommand("pantheon-agents run").version(version).de
 		} }
 	}, multistream([{ stream: logFileTransport }, { stream: prettyTransport }])));
 });
+const showTasksCommand = createCommand("pantheon-agents show-tasks").version(version).description("Show tasks for an agent").argument("[name]", "The name of the agent.").option("--agents <names>", "Comma-separated agent names to query in one call.", parseCommaList, []).option("--all", "Show tasks for all agents.").option("--status <status>", "Filter tasks by status. Multiple values are separated by comma.", parseCommaList, []).option("--order-by <field>", "Order by queued_at, started_at, or ended_at.", "queued_at").option("--order-direction <direction>", "Order direction: asc or desc.", "desc").option("--limit <number>", "Limit the number of tasks shown.", (val) => parseInt(val, 10)).option("--max-task-length <number>", "Maximum task length for table output.", (val) => parseInt(val, 10), 120).option("--full-task", "Do not truncate task text in output.").option("--concise", "Output concise one-line entries.", true).option("--no-color", "Disable colored output.").option("--json", "Output tasks as JSON.").action(async function() {
+	const [name] = this.args;
+	const options = this.opts();
+	const agentNames = /* @__PURE__ */ new Set();
+	if (name) agentNames.add(name);
+	options.agents.forEach((agent) => agentNames.add(agent));
+	if (options.all && agentNames.size > 0) {
+		console.error("Use either a specific agent name or --all, not both.");
+		process$1.exit(1);
+	}
+	if (!options.all && agentNames.size === 0) {
+		console.error("Provide an agent name, --agents, or --all.");
+		process$1.exit(1);
+	}
+	const invalidStatuses = options.status.filter((status) => !taskStatuses.includes(status));
+	if (invalidStatuses.length > 0) {
+		console.error(`Invalid status values: ${invalidStatuses.join(", ")}. Allowed: ${taskStatuses.join(", ")}.`);
+		process$1.exit(1);
+	}
+	if (!orderByFields.includes(options.orderBy)) {
+		console.error(`Invalid order-by value: ${options.orderBy}. Allowed: ${orderByFields.join(", ")}.`);
+		process$1.exit(1);
+	}
+	if (!orderDirections.includes(options.orderDirection)) {
+		console.error(`Invalid order-direction value: ${options.orderDirection}. Allowed: ${orderDirections.join(", ")}.`);
+		process$1.exit(1);
+	}
+	if (options.limit != null && Number.isNaN(options.limit)) {
+		console.error("Invalid limit value. Must be a number.");
+		process$1.exit(1);
+	}
+	if (Number.isNaN(options.maxTaskLength) || options.maxTaskLength <= 0) {
+		console.error("Invalid max-task-length value. Must be a positive number.");
+		process$1.exit(1);
+	}
+	if (options.fullTask && options.maxTaskLength) options.maxTaskLength = Number.POSITIVE_INFINITY;
+	if (options.json && options.concise) {
+		console.error("Use either --json or --concise, not both.");
+		process$1.exit(1);
+	}
+	const limitedTasks = await showTasksForAgents({
+		agents: Array.from(agentNames),
+		allAgents: options.all,
+		status: options.status,
+		orderBy: options.orderBy,
+		orderDirection: options.orderDirection,
+		limit: options.limit
+	});
+	if (options.json) {
+		const payload = limitedTasks.map(({ agent, task }) => ({
+			agent,
+			...task
+		}));
+		console.log(JSON.stringify(payload, null, 2));
+		return;
+	}
+	if (options.concise) {
+		for (const { agent, task } of limitedTasks) console.log(formatConciseTaskLine({
+			agent,
+			task,
+			maxTaskLength: options.maxTaskLength,
+			useColor: options.color
+		}));
+		return;
+	}
+	const rows = limitedTasks.map(({ agent, task }) => formatTaskTableRow({
+		agent,
+		task,
+		maxTaskLength: options.maxTaskLength
+	}));
+	console.table(rows);
+	console.log(`${limitedTasks.length} task(s) shown.`);
+});
 function printCommandHelpAndExit(command) {
 	console.error(`Invalid command: ${command}. Supported commands: ${Object.keys(commands).join(", ")}.`);
 	console.error(`  Run pantheon-agents help <command> for more information.`);
@@ -2023,7 +2330,10 @@ function printCommandHelpAndExit(command) {
 const commands = {
 	"add-task": addTaskCommand,
 	config: configAgentCommand,
-	run: runAgentCommand
+	"delete-task": deleteTaskCommand,
+	run: runAgentCommand,
+	"show-config": showConfigCommand,
+	"show-tasks": showTasksCommand
 };
 if (process$1.argv[2] === "help") {
 	const command = process$1.argv[3];

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@pantheon.ai/agents",
   "type": "module",
-  "version": "0.0.4",
+  "version": "0.0.5",
   "bin": {
     "pantheon-agents": "dist/index.js"
   },