@agentuity/cli 2.0.14 → 2.0.15

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'),