@agentuity/cli 2.0.13 → 2.0.15

package/src/cmd/coder/workspace/common.ts

@@ -2,16 +2,32 @@ import {
 	type CoderCreateWorkspaceRequest,
 	type CoderUpdateWorkspaceRequest,
 	type CoderWorkspaceDetail,
+	type CoderWorkspaceSystemPromptMode,
 } from '@agentuity/core/coder';
 import { StructuredError } from '@agentuity/core';
 import * as tui from '../../../tui';
 
 export const EMPTY_WORKSPACE_ERROR =
-	'A workspace needs at least one repo, dependency, setup script, saved skill, skill bucket, or agent';
+	'A workspace needs at least one repo, dependency, setup script, system prompt, saved skill, skill bucket, or agent';
 export const SetupScriptValidationError = StructuredError('SetupScriptValidationError')<{
 	message: string;
 	path?: string;
 }>();
+export const SystemPromptValidationError = StructuredError('SystemPromptValidationError')<{
+	message: string;
+	path?: string;
+}>();
+
+export function normalizeSystemPromptMode(
+	value?: string
+): CoderWorkspaceSystemPromptMode | undefined {
+	if (value === undefined) return undefined;
+	const normalized = value.trim().toLowerCase();
+	if (normalized === 'append' || normalized === 'overwrite') return normalized;
+	throw new SystemPromptValidationError({
+		message: 'Use --system-prompt-mode append or --system-prompt-mode overwrite.',
+	});
+}
 
 export function parseCommaList(value?: string): string[] {
 	return value
@@ -46,11 +62,36 @@ export async function readSetupScript(input: {
 	}
 }
 
+export async function readSystemPrompt(input: {
+	systemPrompt?: string;
+	systemPromptFile?: string;
+}): Promise<string | undefined> {
+	if (input.systemPrompt !== undefined && input.systemPromptFile) {
+		throw new SystemPromptValidationError({
+			message: 'Use either --system-prompt or --system-prompt-file, not both.',
+		});
+	}
+	if (input.systemPrompt !== undefined) return input.systemPrompt;
+	if (!input.systemPromptFile) return undefined;
+	try {
+		return await Bun.file(input.systemPromptFile).text();
+	} catch (error) {
+		throw new SystemPromptValidationError({
+			message: `Failed to read system prompt file "${input.systemPromptFile}": ${
+				error instanceof Error ? error.message : String(error)
+			}`,
+			path: input.systemPromptFile,
+			cause: error,
+		});
+	}
+}
+
 export function hasWorkspaceSelections(input: CoderCreateWorkspaceRequest): boolean {
 	return (
 		(input.repos?.length ?? 0) > 0 ||
 		(input.dependencies?.length ?? 0) > 0 ||
 		Boolean(input.setupScript?.trim()) ||
+		Boolean(input.systemPrompt?.trim()) ||
 		(input.savedSkillIds?.length ?? 0) > 0 ||
 		(input.skillBucketIds?.length ?? 0) > 0 ||
 		(input.enabledAgents?.length ?? 0) > 0
@@ -67,7 +108,7 @@ export function formatWorkspaceValidationMessage(issues: Array<{ message: string
 		return 'Invalid workspace configuration';
 	}
 	if (messages.includes(EMPTY_WORKSPACE_ERROR)) {
-		return `${EMPTY_WORKSPACE_ERROR}. Use --repo, --dependency, --setup-script, or --enabled-agents.`;
+		return `${EMPTY_WORKSPACE_ERROR}. Use --repo, --dependency, --setup-script, --system-prompt, or --enabled-agents.`;
 	}
 	return messages.join('; ');
 }
@@ -94,6 +135,9 @@ export function printWorkspaceSummary(workspace: CoderWorkspaceDetail): void {
 	if (workspace.setupScript) {
 		tui.output('  Setup:       configured');
 	}
+	if (workspace.systemPrompt) {
+		tui.output(`  Prompt:      configured (${workspace.systemPromptMode})`);
+	}
 	if (workspace.snapshot?.status) {
 		tui.output(`  Snapshot:    ${workspace.snapshot.status}`);
 	}

package/src/cmd/coder/workspace/create.ts

@@ -14,8 +14,10 @@ import {
 	EMPTY_WORKSPACE_ERROR,
 	formatWorkspaceValidationMessage,
 	hasWorkspaceSelections,
+	normalizeSystemPromptMode,
 	parseCommaList,
 	printWorkspaceSummary,
+	readSystemPrompt,
 	readSetupScript,
 } from './common';
 
@@ -38,6 +40,12 @@ export const createWorkspaceSubcommand = createSubcommand({
 			),
 			description: 'Create an org-scoped workspace with dependencies and a setup script',
 		},
+		{
+			command: getCommand(
+				'coder workspace create "My Workspace" --system-prompt-file ./WORKSPACE_PROMPT.md --system-prompt-mode overwrite'
+			),
+			description: 'Create a workspace with Lead system prompt instructions',
+		},
 		{
 			command: getCommand('coder workspace create "My Workspace" --enabled-agents code-review'),
 			description: 'Create a workspace with an agent roster',
@@ -71,6 +79,18 @@ export const createWorkspaceSubcommand = createSubcommand({
 				.string()
 				.optional()
 				.describe('Path to a shell script to run while preparing workspace snapshots'),
+			systemPrompt: z
+				.string()
+				.optional()
+				.describe('Inline Lead system prompt to apply to sessions created from this workspace'),
+			systemPromptFile: z
+				.string()
+				.optional()
+				.describe('Path to a file containing the workspace Lead system prompt'),
+			systemPromptMode: z
+				.string()
+				.optional()
+				.describe('How to apply the system prompt: append or overwrite'),
 			enabledAgents: z
 				.string()
 				.optional()
@@ -116,12 +136,25 @@ export const createWorkspaceSubcommand = createSubcommand({
 			tui.fatal(`Failed to read setup script: ${msg}`, ErrorCode.VALIDATION_FAILED);
 			return;
 		}
+		try {
+			const systemPrompt = await readSystemPrompt({
+				systemPrompt: opts?.systemPrompt,
+				systemPromptFile: opts?.systemPromptFile,
+			});
+			if (systemPrompt !== undefined) body.systemPrompt = systemPrompt;
+			const systemPromptMode = normalizeSystemPromptMode(opts?.systemPromptMode);
+			if (systemPromptMode !== undefined) body.systemPromptMode = systemPromptMode;
+		} catch (err) {
+			const msg = err instanceof Error ? err.message : String(err);
+			tui.fatal(`Failed to read system prompt: ${msg}`, ErrorCode.VALIDATION_FAILED);
+			return;
+		}
 		if (opts?.enabledAgents) {
 			body.enabledAgents = parseCommaList(opts.enabledAgents);
 		}
 		if (!hasWorkspaceSelections(body)) {
 			tui.fatal(
-				`Failed to create workspace: ${EMPTY_WORKSPACE_ERROR}. Use --repo, --dependency, --setup-script, or --enabled-agents.`,
+				`Failed to create workspace: ${EMPTY_WORKSPACE_ERROR}. Use --repo, --dependency, --setup-script, --system-prompt, or --enabled-agents.`,
 				ErrorCode.VALIDATION_FAILED
 			);
 		}

package/src/cmd/coder/workspace/update.ts

@@ -13,8 +13,10 @@ import { resolveGitHubRepo } from '../resolve-repo';
 import {
 	formatWorkspaceValidationMessage,
 	hasWorkspaceUpdate,
+	normalizeSystemPromptMode,
 	parseCommaList,
 	printWorkspaceSummary,
+	readSystemPrompt,
 	readSetupScript,
 } from './common';
 
@@ -33,6 +35,12 @@ export const updateWorkspaceSubcommand = createSubcommand({
 			command: getCommand('coder workspace update ws_abc123 --setup-script-file ./setup.sh'),
 			description: 'Update the workspace setup script',
 		},
+		{
+			command: getCommand(
+				'coder workspace update ws_abc123 --system-prompt-file ./WORKSPACE_PROMPT.md --system-prompt-mode append'
+			),
+			description: 'Update the workspace Lead system prompt',
+		},
 	],
 	schema: {
 		args: z.object({
@@ -57,6 +65,18 @@ export const updateWorkspaceSubcommand = createSubcommand({
 				.string()
 				.optional()
 				.describe('Path to a shell script to run while preparing workspace snapshots'),
+			systemPrompt: z
+				.string()
+				.optional()
+				.describe('Inline Lead system prompt to apply to sessions created from this workspace'),
+			systemPromptFile: z
+				.string()
+				.optional()
+				.describe('Path to a file containing the workspace Lead system prompt'),
+			systemPromptMode: z
+				.string()
+				.optional()
+				.describe('How to apply the system prompt: append or overwrite'),
 			enabledAgents: z
 				.string()
 				.optional()
@@ -102,6 +122,19 @@ export const updateWorkspaceSubcommand = createSubcommand({
 			tui.fatal(`Failed to read setup script: ${msg}`, ErrorCode.VALIDATION_FAILED);
 			return;
 		}
+		try {
+			const systemPrompt = await readSystemPrompt({
+				systemPrompt: opts?.systemPrompt,
+				systemPromptFile: opts?.systemPromptFile,
+			});
+			if (systemPrompt !== undefined) body.systemPrompt = systemPrompt;
+			const systemPromptMode = normalizeSystemPromptMode(opts?.systemPromptMode);
+			if (systemPromptMode !== undefined) body.systemPromptMode = systemPromptMode;
+		} catch (err) {
+			const msg = err instanceof Error ? err.message : String(err);
+			tui.fatal(`Failed to read system prompt: ${msg}`, ErrorCode.VALIDATION_FAILED);
+			return;
+		}
 		if (opts?.enabledAgents) {
 			body.enabledAgents = parseCommaList(opts.enabledAgents);
 		}

package/src/cmd/dev/download.ts

@@ -1,5 +1,5 @@
 import { randomUUID } from 'node:crypto';
-import { mkdirSync, rmSync, writeFileSync } from 'node:fs';
+import { existsSync, mkdirSync, readdirSync, rmSync, writeFileSync } from 'node:fs';
 import { tmpdir, platform } from 'node:os';
 import { join, dirname } from 'node:path';
 import * as tar from 'tar';
@@ -11,6 +11,37 @@ interface GravityClient {
 	version: string;
 }
 
+/**
+ * Remove previously downloaded gravity version directories after a newer
+ * version has started successfully.
+ *
+ * Safety guard: only removes sibling directories that contain a gravity
+ * binary, leaving any unrelated files/folders untouched.
+ */
+export function sweepOldGravityVersions(gravityDir: string, currentVersion: string): string[] {
+	if (!existsSync(gravityDir)) {
+		return [];
+	}
+
+	const removed: string[] = [];
+	for (const entry of readdirSync(gravityDir, { withFileTypes: true })) {
+		if (!entry.isDirectory() || entry.name === currentVersion) {
+			continue;
+		}
+
+		const candidateDir = join(gravityDir, entry.name);
+		const candidateBinary = join(candidateDir, 'gravity');
+		if (!existsSync(candidateBinary)) {
+			continue;
+		}
+
+		rmSync(candidateDir, { recursive: true, force: true });
+		removed.push(candidateDir);
+	}
+
+	return removed;
+}
+
 const GravityVersionError = StructuredError('GravityVersionError')<{
 	status: number;
 	statusText: string;

package/src/cmd/dev/index.ts

@@ -9,7 +9,7 @@ import * as tui from '../../tui';
 import { getCommand } from '../../command-prefix';
 import { generateEndpoint, type DevmodeResponse } from './api';
 import { APIClient, getAPIBaseURL, getAppBaseURL, getGravityDevModeURL } from '../../api';
-import { download } from './download';
+import { download, sweepOldGravityVersions } from './download';
 import { createDevmodeSyncService } from './sync';
 import { getDevmodeDeploymentId } from '../build/ids';
 import { getDefaultConfigDir, saveConfig, loadProjectSDKKey, getAuth } from '../../config';
@@ -411,6 +411,7 @@ export const command = createCommand({
 			let devmode: DevmodeResponse | undefined;
 			let gravityBin: string | undefined;
 			let gravityURL: string | undefined;
+			let gravitySweepTarget: { gravityDir: string; version: string } | null = null;
 			let appURL: string | undefined;
 			let savedPrivateKey: string | undefined = config?.devmode?.privateKey
 				? Buffer.from(config.devmode.privateKey, 'base64').toString('utf-8')
@@ -465,8 +466,12 @@ export const command = createCommand({
 				}
 
 				if (mustCheck) {
+					const previousGravityVersion = config?.gravity?.version;
 					const res = await download(gravityDir);
 					gravityBin = res.filename;
+					if (previousGravityVersion !== res.version) {
+						gravitySweepTarget = { gravityDir, version: res.version };
+					}
 					const _config = { ...config } as Config;
 					_config.gravity = {
 						checked: Date.now(),
@@ -494,10 +499,6 @@ export const command = createCommand({
 
 			// Calculate URLs for banner
 			const padding = 12;
-			const workbenchUrl =
-				auth && project?.projectId
-					? `${getAppBaseURL(config)}/w/${project.projectId}`
-					: `http://127.0.0.1:${opts.port}${workbench.config?.route ?? '/workbench'}`;
 
 			const devmodebody =
 				tui.muted(tui.padRight('Local:', padding)) +
@@ -506,9 +507,6 @@ export const command = createCommand({
 				tui.muted(tui.padRight('Public:', padding)) +
 				(devmode?.hostname ? tui.link(`https://${devmode.hostname}`) : tui.warn('Disabled')) +
 				'\n' +
-				tui.muted(tui.padRight('Workbench:', padding)) +
-				(workbench.hasWorkbench ? tui.link(workbenchUrl) : tui.warn('Disabled')) +
-				'\n' +
 				tui.muted(tui.padRight('Dashboard:', padding)) +
 				(appURL ? tui.link(appURL) : tui.warn('Disabled')) +
 				'\n' +
@@ -1133,6 +1131,24 @@ export const command = createCommand({
 											sendHeartbeat();
 											gravityHeartbeatInterval = setInterval(sendHeartbeat, 5000);
 										}
+
+										if (gravitySweepTarget) {
+											try {
+												const removed = sweepOldGravityVersions(
+													gravitySweepTarget.gravityDir,
+													gravitySweepTarget.version
+												);
+												logger.debug(
+													'Swept %d old gravity version director%s after successful startup',
+													removed.length,
+													removed.length === 1 ? 'y' : 'ies'
+												);
+											} catch (error) {
+												logger.warn('Failed to sweep old gravity versions: %s', error);
+											} finally {
+												gravitySweepTarget = null;
+											}
+										}
 									} else if (trimmed) {
 										logger.debug('[gravity] %s', trimmed);
 									}

package/src/config.ts

@@ -226,6 +226,9 @@ export async function loadConfig(
 			if (process.env.AGENTUITY_VECTOR_URL) {
 				overrides.vector_url = process.env.AGENTUITY_VECTOR_URL;
 			}
+			if (process.env.AGENTUITY_AIGATEWAY_URL) {
+				overrides.aigateway_url = process.env.AGENTUITY_AIGATEWAY_URL;
+			}
 			if (process.env.AGENTUITY_STREAM_URL) {
 				overrides.stream_url = process.env.AGENTUITY_STREAM_URL;
 			}

package/src/types.ts

@@ -41,6 +41,7 @@ export const ConfigSchema = zod.object({
 			kv_url: zod.url().optional().describe('Override keyvalue URL'),
 			sandbox_url: zod.url().optional().describe('Override sandbox URL'),
 			vector_url: zod.url().optional().describe('Override vector store URL'),
+			aigateway_url: zod.url().optional().describe('Override AI Gateway URL'),
 			catalyst_url: zod.url().optional().describe('Override catalyst URL'),
 			ion_url: zod.url().optional().describe('Override ion URL'),
 			gravity_url: zod.url().optional().describe('Override gravity URL'),