@pantheon.ai/agents 0.2.2 → 0.3.0

package/README.md

@@ -13,6 +13,7 @@
 Create/update `.env`:
 
 ```bash
+# Use mysql://... for TiDB, or postgresql://... for db9.
 DATABASE_URL=mysql://root@127.0.0.1:4000/pantheon_agents
 PANTHEON_API_KEY=<your_pantheon_api_key>
 DEFAULT_PANTHEON_PROJECT_ID=<optional_default_project_id>

package/dist/index.js

@@ -5,9 +5,10 @@ import process$1 from "node:process";
 import * as fs from "node:fs";
 import path from "node:path";
 import { multistream, pino, transport } from "pino";
-import { Kysely, MysqlDialect } from "kysely";
-import { createPool } from "mysql2";
+import { Kysely, MysqlDialect, PostgresDialect } from "kysely";
+import { Pool } from "pg";
 import z$1, { z } from "zod";
+import { createPool } from "mysql2";
 import readline from "node:readline/promises";
 import expandTilde from "expand-tilde";
 import blessed from "reblessed";
@@ -399,18 +400,16 @@ var require_cli_options = /* @__PURE__ */ __commonJSMin(((exports, module) => {
 var version = "0.2.2";
 
 //#endregion
-//#region src/db/tidb.ts
-function createTidbPoolOptions(databaseUrl) {
+//#region src/db/db9.ts
+function createDb9PoolOptions(databaseUrl) {
 	return {
-		uri: databaseUrl,
-		supportBigNumbers: true,
-		bigNumberStrings: true,
-		timezone: "Z"
+		connectionString: databaseUrl,
+		ssl: process.env.DB9_INSECURE === "true" ? { rejectUnauthorized: false } : true
 	};
 }
-function createTidbDb(databaseUrl = process.env.DATABASE_URL) {
+function createDb9Db(databaseUrl = process.env.DATABASE_URL) {
 	if (!databaseUrl) throw new Error("DATABASE_URL environment variable is not set.");
-	return new Kysely({ dialect: new MysqlDialect({ pool: createPool(createTidbPoolOptions(databaseUrl)) }) });
+	return new Kysely({ dialect: new PostgresDialect({ pool: new Pool(createDb9PoolOptions(databaseUrl)) }) });
 }
 
 //#endregion
@@ -602,6 +601,140 @@ var TaskListProvider = class {
 	}
 };
 
+//#endregion
+//#region src/providers/task-list-db9-provider.ts
+var TaskListDb9Provider = class extends TaskListProvider {
+	db;
+	ownsDb;
+	constructor(agentName, logger, options = {}) {
+		super(agentName, logger);
+		if (options.db) {
+			this.db = options.db;
+			this.ownsDb = false;
+		} else {
+			this.db = createDb9Db();
+			this.ownsDb = true;
+		}
+	}
+	async close() {
+		if (this.ownsDb) await this.db.destroy();
+	}
+	selectTask() {
+		return this.db.selectFrom("task").selectAll().where("agent", "=", this.agentName);
+	}
+	async getAgentConfig(projectId) {
+		const config = await this.db.selectFrom("agent_project_config").innerJoin("task", "task.id", "agent_project_config.config_task_id").select([
+			"agent_project_config.agent",
+			"agent_project_config.project_id",
+			"agent_project_config.base_branch_id",
+			"agent_project_config.config_version",
+			"agent_project_config.config_task_id",
+			"agent_project_config.concurrency",
+			"agent_project_config.role",
+			"agent_project_config.skills",
+			"agent_project_config.prototype_url",
+			"agent_project_config.execute_agent"
+		]).select((eb) => eb.ref("task.status").$castTo().as("config_task_status")).where("agent_project_config.project_id", "=", projectId).where("agent_project_config.agent", "=", this.agentName).executeTakeFirst();
+		if (config == null) return null;
+		return config;
+	}
+	async getAgentConfigs() {
+		return await this.db.selectFrom("agent_project_config").innerJoin("task", "task.id", "agent_project_config.config_task_id").select([
+			"agent_project_config.agent",
+			"agent_project_config.project_id",
+			"agent_project_config.base_branch_id",
+			"agent_project_config.config_version",
+			"agent_project_config.config_task_id",
+			"agent_project_config.concurrency",
+			"agent_project_config.role",
+			"agent_project_config.skills",
+			"agent_project_config.prototype_url",
+			"agent_project_config.execute_agent"
+		]).select((eb) => eb.ref("task.status").$castTo().as("config_task_status")).where("agent_project_config.agent", "=", this.agentName).execute();
+	}
+	async setAgentConfig({ skills, ...config }) {
+		await this.db.insertInto("agent_project_config").values({
+			agent: this.agentName,
+			skills: JSON.stringify(skills),
+			...config
+		}).execute();
+	}
+	async updateAgentConfig({ skills, ...config }) {
+		const result = await this.db.updateTable("agent_project_config").set({
+			skills: JSON.stringify(skills),
+			...config
+		}).where("agent", "=", this.agentName).where("project_id", "=", config.project_id).executeTakeFirst();
+		if (Number(result.numUpdatedRows ?? 0) === 0) throw new Error(`No config found to update for agent ${this.agentName} and project ${config.project_id}.`);
+	}
+	async getTask(taskId) {
+		const taskItem = await this.selectTask().where("id", "=", taskId).executeTakeFirst();
+		if (taskItem == null) return null;
+		return taskItemSchema.parse(taskItem);
+	}
+	async getTasks({ status, order_by, order_direction = "asc", from, limit, type, min_config_version, max_config_version, parent_task_id, project_id } = {}) {
+		let builder = this.selectTask();
+		if (status && status.length > 0) builder = builder.where("status", "in", status);
+		if (from) builder = builder.where("queued_at", ">=", from);
+		if (type) builder = builder.where("type", "=", type);
+		if (min_config_version != null) builder = builder.where("config_version", ">=", min_config_version);
+		if (max_config_version != null) builder = builder.where("config_version", "<=", max_config_version);
+		if (parent_task_id === null) builder = builder.where("parent_task_id", "is", null);
+		else if (parent_task_id) builder = builder.where("parent_task_id", "=", parent_task_id);
+		if (project_id != null) builder = builder.where("project_id", "=", project_id);
+		if (order_by) builder = builder.orderBy(order_by, order_direction);
+		if (limit != null) builder = builder.limit(limit);
+		return (await builder.execute()).map((item) => taskItemSchema.parse(item));
+	}
+	async listAgentNames() {
+		const [taskAgents, configAgents] = await Promise.all([this.db.selectFrom("task").select("agent").distinct().execute(), this.db.selectFrom("agent_project_config").select("agent").distinct().execute()]);
+		const agents = /* @__PURE__ */ new Set();
+		for (const item of taskAgents) agents.add(item.agent);
+		for (const item of configAgents) agents.add(item.agent);
+		return Array.from(agents).sort();
+	}
+	async updateTask(taskItem) {
+		const { id, started_at, ended_at, queued_at, cancelled_at, ...rest } = taskItem;
+		await this.db.updateTable("task").set({
+			started_at,
+			ended_at,
+			queued_at,
+			cancelled_at,
+			...rest
+		}).where("id", "=", id).where("agent", "=", this.agentName).execute();
+	}
+	async insertTask(taskItem) {
+		const { started_at, ended_at, queued_at, cancelled_at, ...rest } = taskItem;
+		const inserted = await this.db.insertInto("task").values({
+			agent: this.agentName,
+			started_at,
+			ended_at,
+			queued_at,
+			cancelled_at,
+			...rest
+		}).returningAll().executeTakeFirstOrThrow();
+		return taskItemSchema.parse(inserted);
+	}
+	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
+//#region src/db/tidb.ts
+function createTidbPoolOptions(databaseUrl) {
+	return {
+		uri: databaseUrl,
+		supportBigNumbers: true,
+		bigNumberStrings: true,
+		timezone: "Z"
+	};
+}
+function createTidbDb(databaseUrl = process.env.DATABASE_URL) {
+	if (!databaseUrl) throw new Error("DATABASE_URL environment variable is not set.");
+	return new Kysely({ dialect: new MysqlDialect({ pool: createPool(createTidbPoolOptions(databaseUrl)) }) });
+}
+
 //#endregion
 //#region src/providers/task-list-tidb-provider.ts
 var TaskListTidbProvider = class extends TaskListProvider {
@@ -702,6 +835,28 @@ var TaskListTidbProvider = class extends TaskListProvider {
 	}
 };
 
+//#endregion
+//#region src/providers/task-list-provider-factory.ts
+function getDatabaseProtocol(databaseUrl) {
+	try {
+		return new URL(databaseUrl).protocol.toLowerCase();
+	} catch {
+		throw new Error("Invalid DATABASE_URL. Expected a URL starting with mysql:// or postgresql://.");
+	}
+}
+function resolveTaskListProviderKind(databaseUrl) {
+	const protocol = getDatabaseProtocol(databaseUrl);
+	if (protocol === "mysql:") return "tidb";
+	if (protocol === "postgresql:") return "db9";
+	throw new Error(`Unsupported DATABASE_URL protocol '${protocol}'. Use mysql:// for TiDB or postgresql:// for db9.`);
+}
+function createTaskListProvider(agentName, logger) {
+	const databaseUrl = process.env.DATABASE_URL?.trim();
+	if (!databaseUrl) throw new Error("DATABASE_URL environment variable is not set.");
+	if (resolveTaskListProviderKind(databaseUrl) === "db9") return new TaskListDb9Provider(agentName, logger);
+	return new TaskListTidbProvider(agentName, logger);
+}
+
 //#endregion
 //#region src/core/agent-config-utils.ts
 function normalizeRepoUrl(value) {
@@ -1929,7 +2084,7 @@ async function startTaskListLoop(agentName, { loopInterval = 5 }, logger) {
 		abortController.abort(signal);
 		process.exit(1);
 	});
-	const provider = new TaskListTidbProvider(agentName, logger);
+	const provider = createTaskListProvider(agentName, logger);
 	while (!abortController.signal.aborted) {
 		const loopIndex = i++;
 		await loopOnce(provider, logger.child({ name: `loop:${loopIndex}` }));
@@ -2575,7 +2730,7 @@ function resolveConfigRole(options) {
 	return { error: "--role is required for first-time config in a project." };
 }
 async function configAgent(name, options) {
-	const provider = new TaskListTidbProvider(name, pino());
+	const provider = createTaskListProvider(name, pino());
 	try {
 		const previousConfig = await provider.getAgentConfig(options.projectId);
 		const resolvedRoleResult = resolveConfigRole({
@@ -2735,7 +2890,7 @@ async function resolveInitialBaseBranchId(projectId, rootBranchId) {
 	return project.root_branch_id;
 }
 async function addTask(name, options) {
-	const provider = new TaskListTidbProvider(name, pino());
+	const provider = createTaskListProvider(name, pino());
 	try {
 		const config = await provider.getAgentConfig(options.projectId);
 		if (!config) throw new Error(`Agent ${name} not configured for project ${options.projectId}`);
@@ -2751,7 +2906,7 @@ async function addTask(name, options) {
 	}
 }
 async function deleteTask(agentName, taskId) {
-	const provider = new TaskListTidbProvider(agentName, pino());
+	const provider = createTaskListProvider(agentName, pino());
 	try {
 		const task = await provider.getTask(taskId);
 		if (!task) return null;
@@ -2762,7 +2917,7 @@ async function deleteTask(agentName, taskId) {
 	}
 }
 async function getTask(agentName, taskId) {
-	const provider = new TaskListTidbProvider(agentName, pino());
+	const provider = createTaskListProvider(agentName, pino());
 	try {
 		return await provider.getTask(taskId);
 	} finally {
@@ -2770,7 +2925,7 @@ async function getTask(agentName, taskId) {
 	}
 }
 async function cancelTask(agentName, taskId, reason) {
-	const provider = new TaskListTidbProvider(agentName, pino());
+	const provider = createTaskListProvider(agentName, pino());
 	try {
 		const task = await provider.getTask(taskId);
 		if (!task) return null;
@@ -2782,7 +2937,7 @@ async function cancelTask(agentName, taskId, reason) {
 	}
 }
 async function showAgentConfig(agentName, projectId) {
-	const provider = new TaskListTidbProvider(agentName, pino());
+	const provider = createTaskListProvider(agentName, pino());
 	try {
 		return await provider.getAgentConfig(projectId);
 	} finally {
@@ -2790,7 +2945,7 @@ async function showAgentConfig(agentName, projectId) {
 	}
 }
 async function showAgentConfigs(agentName) {
-	const provider = new TaskListTidbProvider(agentName, pino());
+	const provider = createTaskListProvider(agentName, pino());
 	try {
 		return await provider.getAgentConfigs();
 	} finally {
@@ -2798,7 +2953,7 @@ async function showAgentConfigs(agentName) {
 	}
 }
 async function showTasks(name, options) {
-	const provider = new TaskListTidbProvider(name, pino());
+	const provider = createTaskListProvider(name, pino());
 	try {
 		return await provider.getTasks({
 			status: options.status,
@@ -2811,7 +2966,7 @@ async function showTasks(name, options) {
 	}
 }
 async function listAgentNames() {
-	const provider = new TaskListTidbProvider("list-agents", pino());
+	const provider = createTaskListProvider("list-agents", pino());
 	try {
 		return await provider.listAgentNames();
 	} finally {
@@ -5068,6 +5223,11 @@ async function resolveWatchTargets(options) {
 		warnings
 	};
 }
+function createWatchDb() {
+	const databaseUrl = process.env.DATABASE_URL?.trim();
+	if (!databaseUrl) throw new Error("DATABASE_URL environment variable is not set.");
+	return resolveTaskListProviderKind(databaseUrl) === "db9" ? createDb9Db() : createTidbDb();
+}
 const ANSI = {
 	reset: "\x1B[0m",
 	bold: "\x1B[1m",
@@ -5153,7 +5313,7 @@ function createWatchCommand(version) {
 			console.error("Use either --tasks or --agents, not both.");
 			process.exit(1);
 		}
-		const db = createTidbDb();
+		const db = createWatchDb();
 		try {
 			const { targets, warnings } = await resolveWatchTargets({
 				db,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@pantheon.ai/agents",
   "type": "module",
-  "version": "0.2.2",
+  "version": "0.3.0",
   "bin": {
     "pantheon-agents": "dist/index.js"
   },
@@ -17,15 +17,17 @@
     "test": "bun test",
     "test:coverage": "bun test --coverage",
     "dev:db:start": "tiup playground --without-monitor --tag pantheon-agents",
-    "dev:db:gen": "kysely-codegen "
+    "dev:db:gen": "kysely-codegen --no-domains"
   },
   "dependencies": {
+    "@types/pg": "^8.18.0",
     "@types/shell-quote": "^1.7.5",
     "commander": "^14.0.3",
     "dotenv": "^17.2.4",
     "expand-tilde": "^2.0.2",
     "kysely": "^0.28.11",
     "mysql2": "^3.16.3",
+    "pg": "^8.19.0",
     "pino": "^10.3.0",
     "pino-pretty": "^13.1.3",
     "pino-roll": "^4.0.0",