@agentuity/cli 0.0.64 → 0.0.66

package/dist/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,mBAAmB,IAAI,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAEjG,OAAO,EAAE,CAAC,IAAI,GAAG,EAAE,MAAM,KAAK,CAAC;AAG/B,OAAO,EAAE,UAAU,EAAE,CAAC;AAItB,MAAM,CAAC,MAAM,YAAY,GAAG,GAAG,CAAC,MAAM,CAAC;IACtC,IAAI,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;IAC3C,IAAI,EAAE,GAAG;SACP,MAAM,CAAC;QACP,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;QACnE,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;QACpD,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qCAAqC,CAAC;KAChF,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,+DAA+D,CAAC;IAC3E,OAAO,EAAE,GAAG;SACV,MAAM,CAAC;QACP,QAAQ,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;KACvE,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,gCAAgC,CAAC;IAC5C,SAAS,EAAE,GAAG;SACZ,MAAM,CAAC;QACP,OAAO,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QACzF,OAAO,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QACzF,aAAa,EAAE,GAAG;aAChB,GAAG,EAAE;aACL,OAAO,CAAC,kCAAkC,CAAC;aAC3C,QAAQ,CAAC,wBAAwB,CAAC;QACpC,UAAU,EAAE,GAAG;aACb,GAAG,EAAE;aACL,OAAO,CAAC,gCAAgC,CAAC;aACzC,QAAQ,CAAC,qBAAqB,CAAC;QACjC,MAAM,EAAE,GAAG;aACT,GAAG,EAAE;aACL,OAAO,CAAC,kCAAkC,CAAC;aAC3C,QAAQ,CAAC,uBAAuB,CAAC;QACnC,UAAU,EAAE,GAAG;aACb,GAAG,EAAE;aACL,OAAO,CAAC,kCAAkC,CAAC;aAC3C,QAAQ,CAAC,2BAA2B,CAAC;QACvC,UAAU,EAAE,GAAG;aACb,GAAG,EAAE;aACL,OAAO,CAAC,kCAAkC,CAAC;aAC3C,QAAQ,CAAC,2BAA2B,CAAC;QACvC,YAAY,EAAE,GAAG;aACf,GAAG,EAAE;aACL,OAAO,CAAC,kCAAkC,CAAC;aAC3C,QAAQ,CAAC,uBAAuB,CAAC;QACnC,OAAO,EAAE,GAAG;aACV,GAAG,EAAE;aACL,QAAQ,EAAE;aACV,OAAO,CAAC,6BAA6B,CAAC;aACtC,QAAQ,CAAC,kBAAkB,CAAC;QAC9B,WAAW,EAAE,GAAG;aACd,GAAG,EAAE;aACL,OAAO,CAAC,8BAA8B,CAAC;aACvC,QAAQ,CAAC,sBAAsB,CAAC;QAClC,kBAAkB,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;KAC1F,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,4BAA4B,CAAC;IACxC,WAAW,EAAE,GAAG;SACd,MAAM,CAAC;QACP,iBAAiB,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;QAClF,mBAAmB,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;QAC1F,mBAAmB,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC;QACxF,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;QAClE,WAAW,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;KAC5E,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,kBAAkB,CAAC;IAC9B,OAAO,EAAE,GAAG;SACV,MAAM,CAAC;QACP,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;QACxE,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;KACzE,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,gCAAgC,CAAC;CAC5C,CAAC,CAAC;AA4KH,MAAM,UAAU,gBAAgB,CAM9B,UA+BD;IACA,OAAO,UAA6C,CAAC;AACtD,CAAC;AAED,MAAM,UAAU,aAAa,CAM3B,UAgCD;IACA,OAAO,UAA0C,CAAC;AACnD,CAAC;AA+FD,MAAM,CAAC,MAAM,aAAa,GAAG,GAAG,CAAC,MAAM,CAAC;IACvC,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC;IAClD,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;IACnD,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yDAAyD,CAAC;IACxF,UAAU,EAAE,UAAU,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;CAC1E,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,mBAAmB,GAAG,yBAAyB,CAAC"}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,mBAAmB,IAAI,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAEjG,OAAO,EAAE,CAAC,IAAI,GAAG,EAAE,MAAM,KAAK,CAAC;AAG/B,OAAO,EAAE,UAAU,EAAE,CAAC;AAItB,MAAM,CAAC,MAAM,YAAY,GAAG,GAAG,CAAC,MAAM,CAAC;IACtC,IAAI,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;IAC3C,IAAI,EAAE,GAAG;SACP,MAAM,CAAC;QACP,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;QACnE,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;QACpD,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qCAAqC,CAAC;KAChF,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,+DAA+D,CAAC;IAC3E,OAAO,EAAE,GAAG;SACV,MAAM,CAAC;QACP,QAAQ,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;KACvE,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,gCAAgC,CAAC;IAC5C,SAAS,EAAE,GAAG;SACZ,MAAM,CAAC;QACP,OAAO,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QAC/D,OAAO,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QAC/D,aAAa,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;QACtE,UAAU,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;QAChE,MAAM,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QAC9D,UAAU,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;QACtE,UAAU,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;QACtE,YAAY,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QACpE,OAAO,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QAC1D,WAAW,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sBAAsB,CAAC;QAClE,kBAAkB,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;KAC1F,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,EAAE;SACV,QAAQ,CAAC,4BAA4B,CAAC;IACxC,WAAW,EAAE,GAAG;SACd,MAAM,CAAC;QACP,iBAAiB,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;QAClF,mBAAmB,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;QAC1F,mBAAmB,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC;QACxF,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;QAClE,WAAW,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;KAC5E,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,kBAAkB,CAAC;IAC9B,OAAO,EAAE,GAAG;SACV,MAAM,CAAC;QACP,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;QACxE,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;KACzE,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,gCAAgC,CAAC;CAC5C,CAAC,CAAC;AA4KH,MAAM,UAAU,gBAAgB,CAM9B,UA+BD;IACA,OAAO,UAA6C,CAAC;AACtD,CAAC;AAED,MAAM,UAAU,aAAa,CAM3B,UAgCD;IACA,OAAO,UAA0C,CAAC;AACnD,CAAC;AA+FD,MAAM,CAAC,MAAM,aAAa,GAAG,GAAG,CAAC,MAAM,CAAC;IACvC,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC;IAClD,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;IACnD,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yDAAyD,CAAC;IACxF,UAAU,EAAE,UAAU,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;CAC1E,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,mBAAmB,GAAG,yBAAyB,CAAC"}

package/dist/utils/workbench-notify.d.ts

@@ -0,0 +1,29 @@
+/**
+ * Utility to send notifications directly to workbench clients via WebSocket
+ */
+export interface WorkbenchNotifyOptions {
+    port?: number;
+    message: 'restarting' | 'alive';
+}
+/**
+ * Send a notification directly to workbench clients via WebSocket
+ *
+ * @param options - Configuration for the notification
+ * @returns Promise that resolves when notification is sent
+ *
+ * @example
+ * ```typescript
+ * // Notify clients that server is restarting
+ * await notifyWorkbenchClients({
+ *   baseUrl: 'ws://localhost:3500',
+ *   message: 'restarting'
+ * });
+ *
+ * // Notify clients that server is alive
+ * await notifyWorkbenchClients({
+ *   message: 'alive'
+ * });
+ * ```
+ */
+export declare function notifyWorkbenchClients(options: WorkbenchNotifyOptions): Promise<void>;
+//# sourceMappingURL=workbench-notify.d.ts.map

package/dist/utils/workbench-notify.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workbench-notify.d.ts","sourceRoot":"","sources":["../../src/utils/workbench-notify.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,MAAM,WAAW,sBAAsB;IACtC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC;CAChC;AAED;;;;;;;;;;;;;;;;;;;GAmBG;AACH,wBAAsB,sBAAsB,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,IAAI,CAAC,CAqC3F"}

package/dist/utils/workbench-notify.js

@@ -0,0 +1,56 @@
+/**
+ * Utility to send notifications directly to workbench clients via WebSocket
+ */
+/**
+ * Send a notification directly to workbench clients via WebSocket
+ *
+ * @param options - Configuration for the notification
+ * @returns Promise that resolves when notification is sent
+ *
+ * @example
+ * ```typescript
+ * // Notify clients that server is restarting
+ * await notifyWorkbenchClients({
+ *   baseUrl: 'ws://localhost:3500',
+ *   message: 'restarting'
+ * });
+ *
+ * // Notify clients that server is alive
+ * await notifyWorkbenchClients({
+ *   message: 'alive'
+ * });
+ * ```
+ */
+export async function notifyWorkbenchClients(options) {
+    const { port = 3500, message } = options;
+    const wsUrl = new URL(`ws://localhost:${port}`);
+    return new Promise((resolve) => {
+        try {
+            wsUrl.pathname = '/_agentuity/workbench/ws';
+            const ws = new WebSocket(wsUrl.toString());
+            // Set a timeout to avoid hanging
+            const timeout = setTimeout(() => {
+                ws.close();
+                resolve();
+            }, 2000);
+            ws.onopen = () => {
+                ws.send(message);
+                ws.close();
+            };
+            ws.onclose = () => {
+                clearTimeout(timeout);
+                resolve();
+            };
+            ws.onerror = () => {
+                clearTimeout(timeout);
+                ws.close();
+                resolve();
+            };
+        }
+        catch (_error) {
+            // Silently fail - this ensures the CLI doesn't fail if the app server isn't running
+            resolve();
+        }
+    });
+}
+//# sourceMappingURL=workbench-notify.js.map

package/dist/utils/workbench-notify.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workbench-notify.js","sourceRoot":"","sources":["../../src/utils/workbench-notify.ts"],"names":[],"mappings":"AAAA;;GAEG;AAOH;;;;;;;;;;;;;;;;;;;GAmBG;AACH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,OAA+B;IAC3E,MAAM,EAAE,IAAI,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IAEzC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,kBAAkB,IAAI,EAAE,CAAC,CAAC;IAEhD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC9B,IAAI,CAAC;YACJ,KAAK,CAAC,QAAQ,GAAG,0BAA0B,CAAC;YAE5C,MAAM,EAAE,GAAG,IAAI,SAAS,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;YAE3C,iCAAiC;YACjC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC/B,EAAE,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO,EAAE,CAAC;YACX,CAAC,EAAE,IAAI,CAAC,CAAC;YAET,EAAE,CAAC,MAAM,GAAG,GAAG,EAAE;gBAChB,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACjB,EAAE,CAAC,KAAK,EAAE,CAAC;YACZ,CAAC,CAAC;YAEF,EAAE,CAAC,OAAO,GAAG,GAAG,EAAE;gBACjB,YAAY,CAAC,OAAO,CAAC,CAAC;gBACtB,OAAO,EAAE,CAAC;YACX,CAAC,CAAC;YAEF,EAAE,CAAC,OAAO,GAAG,GAAG,EAAE;gBACjB,YAAY,CAAC,OAAO,CAAC,CAAC;gBACtB,EAAE,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO,EAAE,CAAC;YACX,CAAC,CAAC;QACH,CAAC;QAAC,OAAO,MAAM,EAAE,CAAC;YACjB,oFAAoF;YACpF,OAAO,EAAE,CAAC;QACX,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@agentuity/cli",
-	"version": "0.0.64",
+	"version": "0.0.66",
 	"license": "Apache-2.0",
 	"author": "Agentuity employees and contributors",
 	"type": "module",
@@ -35,8 +35,8 @@
 		"prepublishOnly": "bun run clean && bun run build"
 	},
 	"dependencies": {
-		"@agentuity/core": "0.0.64",
-		"@agentuity/server": "0.0.64",
+		"@agentuity/core": "0.0.66",
+		"@agentuity/server": "0.0.66",
 		"@datasert/cronjs-parser": "^1.4.0",
 		"@terascope/fetch-github-release": "^2.2.1",
 		"acorn-loose": "^8.5.2",

package/src/api.ts

@@ -85,10 +85,21 @@ export class APIClient extends BaseAPIClient {
 
 export function getAPIBaseURL(config?: Config | null): string {
 	const overrides = config?.overrides as { api_url?: string } | undefined;
-	return baseGetAPIBaseURL(overrides);
+	return baseGetAPIBaseURL(config?.name, overrides);
 }
 
 export function getAppBaseURL(config?: Config | null): string {
 	const overrides = config?.overrides as { app_url?: string } | undefined;
-	return baseGetAppBaseURL(overrides);
+	return baseGetAppBaseURL(config?.name, overrides);
+}
+
+export function getGravityDevModeURL(region: string, config?: Config | null): string {
+	const overrides = config?.overrides as { gravity_url?: string } | undefined;
+	if (overrides?.gravity_url) {
+		return overrides.gravity_url;
+	}
+	if (config?.name === 'local') {
+		return 'grpc://gravity.agentuity.io:8443';
+	}
+	return `grpc://gravity-${region}.agentuity.cloud`;
 }

package/src/cli.ts

@@ -398,6 +398,7 @@ interface ResolveRegionOptions {
 	apiClient: APIClientType;
 	logger: Logger;
 	required: boolean;
+	region?: string;
 }
 
 async function resolveRegion(opts: ResolveRegionOptions): Promise<string | undefined> {
@@ -425,7 +426,7 @@ async function resolveRegion(opts: ResolveRegionOptions): Promise<string | undef
 	}
 
 	// Check if region was provided via flag
-	let region = options.region as string | undefined;
+	let region = opts.region ?? (options.region as string | undefined);
 
 	// Validate --region flag if provided
 	if (region) {
@@ -746,6 +747,7 @@ async function registerSubcommand(
 									apiClient,
 									logger: baseCtx.logger,
 									required: !!normalized.requiresRegion,
+									region: project?.region,
 								});
 							},
 						});
@@ -810,6 +812,7 @@ async function registerSubcommand(
 							apiClient,
 							logger: baseCtx.logger,
 							required: !!normalized.requiresRegion,
+							region: project?.region,
 						});
 					});
 					if (region) {
@@ -921,6 +924,7 @@ async function registerSubcommand(
 							apiClient,
 							logger: baseCtx.logger,
 							required: !!normalized.requiresRegion,
+							region: project?.region,
 						});
 						if (region) {
 							ctx.region = region;
@@ -1092,6 +1096,7 @@ async function registerSubcommand(
 						apiClient,
 						logger: baseCtx.logger,
 						required: !!normalized.requiresRegion,
+						region: project?.region,
 					});
 					if (region) {
 						ctx.region = region;

package/src/cmd/auth/signup.ts

@@ -20,7 +20,7 @@ export const signupCommand = createSubcommand({
 
 	async handler(ctx) {
 		const { logger, config, apiClient } = ctx;
-		const appUrl = getAppBaseURL(config?.overrides);
+		const appUrl = getAppBaseURL(config?.name, config?.overrides);
 
 		try {
 			const otp = generateSignupOTP();

package/src/cmd/build/ast.test.ts

@@ -1,5 +1,5 @@
 import { describe, test, expect } from 'bun:test';
-import { parseRoute } from './ast';
+import { parseRoute, parseAgentMetadata } from './ast';
 import { writeFileSync, mkdirSync, rmSync } from 'node:fs';
 import { join } from 'node:path';
 
@@ -171,3 +171,248 @@ export default router;
 		cleanup();
 	});
 });
+
+describe('parseAgentMetadata - Schema Code Extraction', () => {
+	const setup = () => {
+		rmSync(TEST_DIR, { recursive: true, force: true });
+		mkdirSync(TEST_DIR, { recursive: true });
+	};
+
+	const cleanup = () => {
+		rmSync(TEST_DIR, { recursive: true, force: true });
+	};
+
+	test('should extract input and output schema code', async () => {
+		setup();
+		const agentFile = join(TEST_DIR, 'agent.ts');
+		const code = `
+import { createAgent } from '@agentuity/runtime';
+import { z } from 'zod';
+
+const agent = createAgent({
+	metadata: {
+		name: 'test-agent',
+		description: 'Test agent',
+	},
+	schema: {
+		input: z.object({
+			name: z.string(),
+			age: z.number(),
+		}),
+		output: z.object({
+			result: z.string(),
+		}),
+	},
+	handler: async (ctx, input) => {
+		return { result: 'success' };
+	},
+});
+
+export default agent;
+		`;
+		writeFileSync(agentFile, code);
+
+		const transpiler = new Bun.Transpiler({ loader: 'ts', target: 'bun' });
+		const contents = transpiler.transformSync(code);
+		const [, metadata] = await parseAgentMetadata(
+			TEST_DIR,
+			agentFile,
+			contents,
+			'proj_1',
+			'dep_1'
+		);
+
+		expect(metadata.has('inputSchemaCode')).toBe(true);
+		expect(metadata.has('outputSchemaCode')).toBe(true);
+
+		const inputSchemaCode = metadata.get('inputSchemaCode');
+		const outputSchemaCode = metadata.get('outputSchemaCode');
+
+		expect(inputSchemaCode).toContain('z.object');
+		expect(inputSchemaCode).toContain('name');
+		expect(inputSchemaCode).toContain('age');
+
+		expect(outputSchemaCode).toContain('z.object');
+		expect(outputSchemaCode).toContain('result');
+
+		cleanup();
+	});
+
+	test('should extract only input schema code when output is missing', async () => {
+		setup();
+		const agentFile = join(TEST_DIR, 'agent.ts');
+		const code = `
+import { createAgent } from '@agentuity/runtime';
+import { z } from 'zod';
+
+const agent = createAgent({
+	metadata: {
+		name: 'test-agent',
+	},
+	schema: {
+		input: z.string(),
+	},
+	handler: async (ctx, input) => {
+		return 'success';
+	},
+});
+
+export default agent;
+		`;
+		writeFileSync(agentFile, code);
+
+		const transpiler = new Bun.Transpiler({ loader: 'ts', target: 'bun' });
+		const contents = transpiler.transformSync(code);
+		const [, metadata] = await parseAgentMetadata(
+			TEST_DIR,
+			agentFile,
+			contents,
+			'proj_1',
+			'dep_1'
+		);
+
+		expect(metadata.has('inputSchemaCode')).toBe(true);
+		expect(metadata.has('outputSchemaCode')).toBe(false);
+
+		const inputSchemaCode = metadata.get('inputSchemaCode');
+		expect(inputSchemaCode).toContain('z.string');
+
+		cleanup();
+	});
+
+	test('should extract only output schema code when input is missing', async () => {
+		setup();
+		const agentFile = join(TEST_DIR, 'agent.ts');
+		const code = `
+import { createAgent } from '@agentuity/runtime';
+import { z } from 'zod';
+
+const agent = createAgent({
+	metadata: {
+		name: 'test-agent',
+	},
+	schema: {
+		output: z.array(z.string()),
+	},
+	handler: async (ctx) => {
+		return ['item1', 'item2'];
+	},
+});
+
+export default agent;
+		`;
+		writeFileSync(agentFile, code);
+
+		const transpiler = new Bun.Transpiler({ loader: 'ts', target: 'bun' });
+		const contents = transpiler.transformSync(code);
+		const [, metadata] = await parseAgentMetadata(
+			TEST_DIR,
+			agentFile,
+			contents,
+			'proj_1',
+			'dep_1'
+		);
+
+		expect(metadata.has('inputSchemaCode')).toBe(false);
+		expect(metadata.has('outputSchemaCode')).toBe(true);
+
+		const outputSchemaCode = metadata.get('outputSchemaCode');
+		expect(outputSchemaCode).toContain('z.array');
+		expect(outputSchemaCode).toContain('z.string');
+
+		cleanup();
+	});
+
+	test('should handle complex nested schemas', async () => {
+		setup();
+		const agentFile = join(TEST_DIR, 'agent.ts');
+		const code = `
+import { createAgent } from '@agentuity/runtime';
+import { z } from 'zod';
+
+const agent = createAgent({
+	metadata: {
+		name: 'test-agent',
+	},
+	schema: {
+		input: z.object({
+			user: z.object({
+				name: z.string(),
+				email: z.string().email(),
+			}),
+			tags: z.array(z.string()),
+		}),
+		output: z.union([
+			z.object({ success: z.boolean() }),
+			z.object({ error: z.string() }),
+		]),
+	},
+	handler: async (ctx, input) => {
+		return { success: true };
+	},
+});
+
+export default agent;
+		`;
+		writeFileSync(agentFile, code);
+
+		const transpiler = new Bun.Transpiler({ loader: 'ts', target: 'bun' });
+		const contents = transpiler.transformSync(code);
+		const [, metadata] = await parseAgentMetadata(
+			TEST_DIR,
+			agentFile,
+			contents,
+			'proj_1',
+			'dep_1'
+		);
+
+		expect(metadata.has('inputSchemaCode')).toBe(true);
+		expect(metadata.has('outputSchemaCode')).toBe(true);
+
+		const inputSchemaCode = metadata.get('inputSchemaCode');
+		const outputSchemaCode = metadata.get('outputSchemaCode');
+
+		expect(inputSchemaCode).toContain('z.object');
+		expect(inputSchemaCode).toContain('user');
+		expect(inputSchemaCode).toContain('tags');
+
+		expect(outputSchemaCode).toContain('z.union');
+
+		cleanup();
+	});
+
+	test('should handle agent without schema property', async () => {
+		setup();
+		const agentFile = join(TEST_DIR, 'agent.ts');
+		const code = `
+import { createAgent } from '@agentuity/runtime';
+
+const agent = createAgent({
+	metadata: {
+		name: 'test-agent',
+	},
+	handler: async (ctx) => {
+		return 'success';
+	},
+});
+
+export default agent;
+		`;
+		writeFileSync(agentFile, code);
+
+		const transpiler = new Bun.Transpiler({ loader: 'ts', target: 'bun' });
+		const contents = transpiler.transformSync(code);
+		const [, metadata] = await parseAgentMetadata(
+			TEST_DIR,
+			agentFile,
+			contents,
+			'proj_1',
+			'dep_1'
+		);
+
+		expect(metadata.has('inputSchemaCode')).toBe(false);
+		expect(metadata.has('outputSchemaCode')).toBe(false);
+
+		cleanup();
+	});
+});

package/src/cmd/build/ast.ts

@@ -174,6 +174,49 @@ function generateStableEvalId(projectId: string, agentId: string, name: string):
 	return `evalid_${hashSHA1(projectId, agentId, name)}`.substring(0, 64);
 }
 
+/**
+ * Extract schema code from createAgent call arguments
+ * Returns input and output schema code as strings
+ */
+function extractSchemaCode(callargexp: ASTObjectExpression): {
+	inputSchemaCode?: string;
+	outputSchemaCode?: string;
+} {
+	let schemaObj: ASTObjectExpression | undefined;
+
+	// Find the schema property
+	for (const prop of callargexp.properties) {
+		if (prop.key.type === 'Identifier' && prop.key.name === 'schema') {
+			if (prop.value.type === 'ObjectExpression') {
+				schemaObj = prop.value as ASTObjectExpression;
+				break;
+			}
+		}
+	}
+
+	if (!schemaObj) {
+		return {};
+	}
+
+	let inputSchemaCode: string | undefined;
+	let outputSchemaCode: string | undefined;
+
+	// Extract input and output schema code
+	for (const prop of schemaObj.properties) {
+		if (prop.key.type === 'Identifier') {
+			if (prop.key.name === 'input' && prop.value) {
+				// Generate source code from AST node
+				inputSchemaCode = generate(prop.value);
+			} else if (prop.key.name === 'output' && prop.value) {
+				// Generate source code from AST node
+				outputSchemaCode = generate(prop.value);
+			}
+		}
+	}
+
+	return { inputSchemaCode, outputSchemaCode };
+}
+
 type AcornParseResultType = ReturnType<typeof acornLoose.parse>;
 
 const MetadataError = StructuredError('MetatadataNameMissingError')<{
@@ -189,7 +232,9 @@ function augmentAgentMetadataNode(
 	version: string,
 	ast: AcornParseResultType,
 	propvalue: ASTObjectExpression,
-	filename: string
+	filename: string,
+	inputSchemaCode?: string,
+	outputSchemaCode?: string
 ): [string, Map<string, string>] {
 	const metadata = parseObjectExpressionToMap(propvalue);
 	if (!metadata.has('name')) {
@@ -218,6 +263,12 @@ function augmentAgentMetadataNode(
 	metadata.set('id', id);
 	metadata.set('agentId', agentId);
 	metadata.set('description', description);
+	if (inputSchemaCode) {
+		metadata.set('inputSchemaCode', inputSchemaCode);
+	}
+	if (outputSchemaCode) {
+		metadata.set('outputSchemaCode', outputSchemaCode);
+	}
 	propvalue.properties.push(
 		createObjectPropertyNode('id', id),
 		createObjectPropertyNode('agentId', agentId),
@@ -226,6 +277,12 @@ function augmentAgentMetadataNode(
 		createObjectPropertyNode('filename', rel),
 		createObjectPropertyNode('description', description)
 	);
+	if (inputSchemaCode) {
+		propvalue.properties.push(createObjectPropertyNode('inputSchemaCode', inputSchemaCode));
+	}
+	if (outputSchemaCode) {
+		propvalue.properties.push(createObjectPropertyNode('outputSchemaCode', outputSchemaCode));
+	}
 
 	const newsource = generate(ast);
 
@@ -626,6 +683,7 @@ export async function parseAgentMetadata(
 	const id = getAgentId(projectId, deploymentId, rel, version);
 
 	let result: [string, Map<string, string>] | undefined;
+	let schemaCodeExtracted = false;
 
 	for (const body of ast.body) {
 		if (body.type === 'ExportDefaultDeclaration') {
@@ -634,6 +692,17 @@ export async function parseAgentMetadata(
 				if (call.callee.name === 'createAgent') {
 					for (const callarg of call.arguments) {
 						const callargexp = callarg as ASTObjectExpression;
+
+						// Extract schema code before processing metadata
+						let inputSchemaCode: string | undefined;
+						let outputSchemaCode: string | undefined;
+						if (!schemaCodeExtracted) {
+							const schemaCode = extractSchemaCode(callargexp);
+							inputSchemaCode = schemaCode.inputSchemaCode;
+							outputSchemaCode = schemaCode.outputSchemaCode;
+							schemaCodeExtracted = true;
+						}
+
 						for (const prop of callargexp.properties) {
 							if (prop.key.type === 'Identifier' && prop.key.name === 'metadata') {
 								result = augmentAgentMetadataNode(
@@ -644,7 +713,9 @@ export async function parseAgentMetadata(
 									version,
 									ast,
 									prop.value as ASTObjectExpression,
-									filename
+									filename,
+									inputSchemaCode,
+									outputSchemaCode
 								);
 								break;
 							}
@@ -689,6 +760,17 @@ export async function parseAgentMetadata(
 								if (call.callee.name === 'createAgent') {
 									for (const callarg of call.arguments) {
 										const callargexp = callarg as ASTObjectExpression;
+
+										// Extract schema code before processing metadata
+										let inputSchemaCode: string | undefined;
+										let outputSchemaCode: string | undefined;
+										if (!schemaCodeExtracted) {
+											const schemaCode = extractSchemaCode(callargexp);
+											inputSchemaCode = schemaCode.inputSchemaCode;
+											outputSchemaCode = schemaCode.outputSchemaCode;
+											schemaCodeExtracted = true;
+										}
+
 										for (const prop of callargexp.properties) {
 											if (
 												prop.key.type === 'Identifier' &&
@@ -702,7 +784,9 @@ export async function parseAgentMetadata(
 													version,
 													ast,
 													prop.value as ASTObjectExpression,
-													filename
+													filename,
+													inputSchemaCode,
+													outputSchemaCode
 												);
 												break;
 											}
@@ -1572,7 +1656,7 @@ export function analyzeWorkbench(content: string): WorkbenchAnalysis {
 
 		// Set default config if workbench is used but no config was parsed
 		if (hasImport && hasUsage && !config) {
-			config = { route: '/workbench' };
+			config = { route: '/workbench', headers: {}, port: 3500 };
 		}
 
 		return {