@pantheon.ai/agents 0.3.2 → 0.3.3

package/dist/db/migrations/db9/20260305_0002_add_agent_config_envs.sql

@@ -0,0 +1,5 @@
+-- introduced_in: 0.3.2
+-- Add optional env overrides for per-project agent config (db9/PostgreSQL).
+
+ALTER TABLE agent_project_config
+  ADD COLUMN IF NOT EXISTS envs JSONB;

package/dist/db/migrations/tidb/20260305_0002_add_agent_config_envs.sql

@@ -0,0 +1,5 @@
+-- introduced_in: 0.3.2
+-- Add optional env overrides for per-project agent config (TiDB/MySQL).
+
+ALTER TABLE agent_project_config
+  ADD COLUMN IF NOT EXISTS envs JSON NULL;

package/dist/index.js

@@ -399,7 +399,7 @@ var require_cli_options = /* @__PURE__ */ __commonJSMin(((exports, module) => {
 
 //#endregion
 //#region package.json
-var version = "0.3.1";
+var version = "0.3.3";
 
 //#endregion
 //#region src/db/db9.ts
@@ -639,6 +639,7 @@ var TaskListDb9Provider = class extends TaskListProvider {
 			"agent_project_config.retry_backoff_seconds",
 			"agent_project_config.role",
 			"agent_project_config.skills",
+			"agent_project_config.envs",
 			"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();
@@ -657,26 +658,30 @@ var TaskListDb9Provider = class extends TaskListProvider {
 			"agent_project_config.retry_backoff_seconds",
 			"agent_project_config.role",
 			"agent_project_config.skills",
+			"agent_project_config.envs",
 			"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 }) {
+	async setAgentConfig({ skills, envs, ...config }) {
 		const maxRetryAttempts = config.max_retry_attempts ?? 3;
 		const retryBackoffSeconds = config.retry_backoff_seconds ?? 30;
 		await this.db.insertInto("agent_project_config").values({
 			agent: this.agentName,
 			skills: JSON.stringify(skills),
+			envs: JSON.stringify(envs ?? {}),
 			...config,
 			max_retry_attempts: maxRetryAttempts,
 			retry_backoff_seconds: retryBackoffSeconds
 		}).execute();
 	}
-	async updateAgentConfig({ skills, ...config }) {
+	async updateAgentConfig({ skills, envs, ...config }) {
 		const maxRetryAttempts = config.max_retry_attempts;
 		const retryBackoffSeconds = config.retry_backoff_seconds;
+		const resolvedEnvs = envs == null ? void 0 : JSON.stringify(envs);
 		const result = await this.db.updateTable("agent_project_config").set({
 			skills: JSON.stringify(skills),
+			...resolvedEnvs == null ? {} : { envs: resolvedEnvs },
 			...config,
 			...maxRetryAttempts == null ? {} : { max_retry_attempts: maxRetryAttempts },
 			...retryBackoffSeconds == null ? {} : { retry_backoff_seconds: retryBackoffSeconds }
@@ -783,22 +788,25 @@ var TaskListTidbProvider = class extends TaskListProvider {
 	async getAgentConfigs() {
 		return await this.db.selectFrom("agent_project_config").innerJoin("task", "task.id", "agent_project_config.config_task_id").selectAll("agent_project_config").select((eb) => eb.ref("task.status").$castTo().as("config_task_status")).where("agent_project_config.agent", "=", this.agentName).execute();
 	}
-	async setAgentConfig({ skills, ...config }) {
+	async setAgentConfig({ skills, envs, ...config }) {
 		const maxRetryAttempts = config.max_retry_attempts ?? 3;
 		const retryBackoffSeconds = config.retry_backoff_seconds ?? 30;
 		await this.db.insertInto("agent_project_config").values({
 			agent: this.agentName,
 			skills: JSON.stringify(skills),
+			envs: JSON.stringify(envs ?? {}),
 			...config,
 			max_retry_attempts: maxRetryAttempts,
 			retry_backoff_seconds: retryBackoffSeconds
 		}).execute();
 	}
-	async updateAgentConfig({ skills, ...config }) {
+	async updateAgentConfig({ skills, envs, ...config }) {
 		const maxRetryAttempts = config.max_retry_attempts;
 		const retryBackoffSeconds = config.retry_backoff_seconds;
+		const resolvedEnvs = envs == null ? void 0 : JSON.stringify(envs);
 		const result = await this.db.updateTable("agent_project_config").set({
 			skills: JSON.stringify(skills),
+			...resolvedEnvs == null ? {} : { envs: resolvedEnvs },
 			...config,
 			...maxRetryAttempts == null ? {} : { max_retry_attempts: maxRetryAttempts },
 			...retryBackoffSeconds == null ? {} : { retry_backoff_seconds: retryBackoffSeconds }
@@ -902,6 +910,20 @@ function normalizeSkills(value) {
 	}
 	return [];
 }
+function normalizeEnvs(value) {
+	const parseRecord = (input) => {
+		if (!input || typeof input !== "object" || Array.isArray(input)) return {};
+		const result = {};
+		for (const [key, item] of Object.entries(input)) if (typeof item === "string") result[key] = item;
+		return result;
+	};
+	if (typeof value === "string") try {
+		return parseRecord(JSON.parse(value));
+	} catch {
+		return {};
+	}
+	return parseRecord(value);
+}
 function buildConfigSetupStep(options) {
 	const prototypeRepoUrl = normalizeRepoUrl(options.prototypeRepoUrl);
 	const skillsArg = JSON.stringify(options.skills.join(","));
@@ -1787,6 +1809,7 @@ const listProjectBackgroundTasks = defineApi(get`/api/v1/projects/${"projectId"}
 const createProjectExploration = defineApi(post`/api/v1/projects/${"projectId"}/explorations`, typeOf(), createExplorationResultSchema);
 const listProjectExplorations = defineApi(get`/api/v1/projects/${"projectId"}/explorations`, null, explorationSchema.array());
 const getProjectExplorationResult = defineApi(get`/api/v1/projects/${"projectId"}/explorations/${"explorationId"}/result`, null, explorationResultSchema);
+const killProjectBranch = defineApi(post`/api/v1/projects/${"projectId"}/branches/${"branchId"}/kill`, typeOf(), snapSchema.nullable());
 const getProjectBranchFs = defineApi(get`/api/v1/projects/${"projectId"}/branches/${"branchId"}/fs/${"path*"}`, null, FSResponseValidator);
 const getProjectBranchFsPreview = defineApi(get`/api/v1/projects/${"projectId"}/branches/${"branchId"}/preview/${"path*"}`, null, FSResponseValidator);
 
@@ -2027,7 +2050,6 @@ const executor = createApiExecutor({
 	headers: { Authorization: `Bearer ${process.env.PANTHEON_API_KEY}` },
 	validateResponse
 });
-const killProjectBranchExecutionApi = defineApi(post`/api/v1/projects/${"projectId"}/branches/${"branchId"}/kill`, typeOf(), "raw");
 async function getPantheonProjectInfo({ projectId }) {
 	return await executor.execute(getProject, { projectId }, null);
 }
@@ -2076,13 +2098,14 @@ async function getPantheonBranchInfo({ projectId, branchId }) {
 		branchId
 	}, null);
 }
-async function executeOnPantheon({ projectId, branchId, prompt, agent }) {
+async function executeOnPantheon({ projectId, branchId, prompt, agent, envs }) {
 	const promptSequence = Array.isArray(prompt) ? prompt : [prompt];
 	return (await executor.execute(createProjectExploration, { projectId }, {
 		shared_prompt_sequence: promptSequence,
 		num_branches: 1,
 		agent,
-		parent_branch_id: branchId
+		parent_branch_id: branchId,
+		envs
 	})).branches[0];
 }
 function isNotFoundError(error) {
@@ -2092,7 +2115,7 @@ function isNotFoundError(error) {
 }
 async function killPantheonBranchExecution({ projectId, branchId }) {
 	try {
-		await executor.execute(killProjectBranchExecutionApi, {
+		await executor.execute(killProjectBranch, {
 			projectId,
 			branchId
 		}, {});
@@ -2246,6 +2269,7 @@ async function pollRunningTaskState(provider, state, logger) {
 			execute_agent: config.execute_agent,
 			role: config.role,
 			skills: normalizeSkills(config.skills),
+			envs: normalizeEnvs(config.envs),
 			prototype_url: config.prototype_url
 		});
 	} else if (newStatus.state === "failed") {
@@ -2350,7 +2374,8 @@ async function startPendingTask(provider, task, config, logger) {
 			projectId: taskToStart.project_id,
 			branchId: taskToStart.base_branch_id,
 			prompt: promptSequence,
-			agent: config.execute_agent
+			agent: config.execute_agent,
+			envs: normalizeEnvs(config.envs)
 		});
 	});
 	logger.info(`Task ${taskToStart.id} started successfully on branch [id = ${taskBranch.branch_id}, snap_id = ${taskBranch.latest_snap_id}]`);
@@ -2877,6 +2902,7 @@ async function configAgent(name, options) {
 			return;
 		}
 		const resolvedSkills = options.skills ?? normalizeSkills(previousConfig?.skills);
+		const resolvedEnvs = options.envs ?? normalizeEnvs(previousConfig?.envs);
 		const resolvedExecuteAgent = options.executeAgent.trim() || previousConfig?.execute_agent || "codex";
 		const resolvedPrototypeUrl = options.prototypeUrl.trim() || previousConfig?.prototype_url || "https://github.com/pingcap-inc/pantheon-agents";
 		const baseBranchId = previousConfig ? previousConfig.base_branch_id : await resolveInitialBaseBranchId(options.projectId, options.rootBranchId);
@@ -2911,6 +2937,7 @@ async function configAgent(name, options) {
 				execute_agent: resolvedExecuteAgent,
 				role: resolvedRole ?? previousConfig.role,
 				skills: resolvedSkills,
+				envs: resolvedEnvs,
 				prototype_url: resolvedPrototypeUrl
 			});
 			if (!options.prompt) await provider.updateTask({
@@ -2980,6 +3007,7 @@ async function configAgent(name, options) {
 				execute_agent: resolvedExecuteAgent,
 				role: resolvedRole,
 				skills: resolvedSkills,
+				envs: resolvedEnvs,
 				prototype_url: resolvedPrototypeUrl
 			});
 			else await provider.updateAgentConfig({
@@ -2993,6 +3021,7 @@ async function configAgent(name, options) {
 				execute_agent: resolvedExecuteAgent,
 				role: resolvedRole ?? previousConfig.role,
 				skills: resolvedSkills,
+				envs: resolvedEnvs,
 				prototype_url: resolvedPrototypeUrl
 			});
 			for (const pendingTask of pendingTasks) {
@@ -3549,8 +3578,25 @@ function parsePositiveInteger(value, fieldName) {
 	if (!Number.isInteger(parsed) || parsed <= 0) throw new InvalidArgumentError(`${fieldName} must be a positive integer.`);
 	return parsed;
 }
+function parseEnvs(value) {
+	const trimmed = value.trim();
+	if (!trimmed) throw new InvalidArgumentError("envs must be a JSON object of string values.");
+	let parsed;
+	try {
+		parsed = JSON.parse(trimmed);
+	} catch {
+		throw new InvalidArgumentError("envs must be valid JSON.");
+	}
+	if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) throw new InvalidArgumentError("envs must be a JSON object.");
+	const out = {};
+	for (const [key, item] of Object.entries(parsed)) {
+		if (typeof item !== "string") throw new InvalidArgumentError(`envs.${key} must be a string value.`);
+		out[key] = item;
+	}
+	return out;
+}
 function createConfigAgentCommand(version, deps = {}) {
-	return createCommand("config").version(version).description("Queue a configuration task for an agent/project").argument("<name>", "The name of the agent.").argument("[prompt]", "The configuration task prompt.").option("--project-id <project-id>", "The project id of the agent. Defaults to DEFAULT_PANTHEON_PROJECT_ID.").option("--task-id <task-id>", "Optional parent task id.").option("--role <role>", "Role metadata. Required for first-time project config; optional override later.").option("--skills <skills>", "The skills of the agent. Multiple values are separated by comma.", parseUniqueCommaList).option("--execute-agent <agent>", "The execute agent of the agent.", "codex").option("--concurrency <number>", "Max number of parallel running tasks for this project.", parseConcurrency).option("--max-retry-attempts <number>", "Max automatic retry attempts per task after failures.", (value) => parseNonNegativeInteger(value, "max-retry-attempts")).option("--retry-backoff-seconds <seconds>", "Base delay (seconds) for exponential retry backoff.", (value) => parsePositiveInteger(value, "retry-backoff-seconds")).option("--root-branch-id <branchId>", "The root branch id of the agent. Defaults to DEFAULT_PANTHEON_ROOT_BRANCH_ID, then project root branch id.").option("--prototype-url <url>", "Role and skill definitions repo.", "https://github.com/pingcap-inc/pantheon-agents").action(async function() {
+	return createCommand("config").version(version).description("Queue a configuration task for an agent/project").argument("<name>", "The name of the agent.").argument("[prompt]", "The configuration task prompt.").option("--project-id <project-id>", "The project id of the agent. Defaults to DEFAULT_PANTHEON_PROJECT_ID.").option("--task-id <task-id>", "Optional parent task id.").option("--role <role>", "Role metadata. Required for first-time project config; optional override later.").option("--skills <skills>", "The skills of the agent. Multiple values are separated by comma.", parseUniqueCommaList).option("--execute-agent <agent>", "The execute agent of the agent.", "codex").option("--concurrency <number>", "Max number of parallel running tasks for this project.", parseConcurrency).option("--max-retry-attempts <number>", "Max automatic retry attempts per task after failures.", (value) => parseNonNegativeInteger(value, "max-retry-attempts")).option("--retry-backoff-seconds <seconds>", "Base delay (seconds) for exponential retry backoff.", (value) => parsePositiveInteger(value, "retry-backoff-seconds")).option("--root-branch-id <branchId>", "The root branch id of the agent. Defaults to DEFAULT_PANTHEON_ROOT_BRANCH_ID, then project root branch id.").option("--prototype-url <url>", "Role and skill definitions repo.", "https://github.com/pingcap-inc/pantheon-agents").option("--envs <json>", "Pantheon branch envs in JSON object format, e.g. '{\"ANTHROPIC_MODEL\":\"claude-opus-4-5\"}'.", parseEnvs).action(async function() {
 		const [name, prompt] = this.args;
 		const options = this.opts();
 		const resolvedProjectId = resolvePantheonProjectId(options.projectId);
@@ -3571,6 +3617,7 @@ function createConfigAgentCommand(version, deps = {}) {
 			maxRetryAttempts: options.maxRetryAttempts,
 			retryBackoffSeconds: options.retryBackoffSeconds,
 			skills: options.skills,
+			envs: options.envs,
 			prototypeUrl: options.prototypeUrl,
 			rootBranchId: resolvedRootBranchId
 		});
@@ -3774,7 +3821,8 @@ function createShowConfigCommand(version) {
 				role: config.role,
 				execute_agent: config.execute_agent,
 				prototype_url: config.prototype_url,
-				skills: Array.isArray(config.skills) ? config.skills.join(", ") : String(config.skills)
+				skills: Array.isArray(config.skills) ? config.skills.join(", ") : String(config.skills),
+				envs: config.envs && typeof config.envs === "object" ? JSON.stringify(config.envs) : String(config.envs ?? "{}")
 			}]);
 			return;
 		}
@@ -3800,7 +3848,8 @@ function createShowConfigCommand(version) {
 			role: config.role,
 			execute_agent: config.execute_agent,
 			prototype_url: config.prototype_url,
-			skills: Array.isArray(config.skills) ? config.skills.join(", ") : String(config.skills)
+			skills: Array.isArray(config.skills) ? config.skills.join(", ") : String(config.skills),
+			envs: config.envs && typeof config.envs === "object" ? JSON.stringify(config.envs) : String(config.envs ?? "{}")
 		})));
 	});
 }

package/package.json

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