@agentuity/core 3.0.0-alpha.6 → 3.0.0-beta.0

package/src/services/sandbox/get.ts

@@ -156,7 +156,7 @@ export const SandboxInfoDataSchema = z
 		user: SandboxUserInfoSchema.optional().describe('User who created the sandbox'),
 		agent: SandboxAgentInfoSchema.optional().describe('Agent associated with the sandbox'),
 		project: SandboxProjectInfoSchema.optional().describe('Project associated with the sandbox'),
-		org: SandboxOrgInfoSchema.describe('Organization associated with the sandbox'),
+		org: SandboxOrgInfoSchema.nullish().describe('Organization associated with the sandbox'),
 		timeout: z
 			.object({
 				idle: z.string().optional().describe('Idle timeout duration (e.g., "5m", "1h").'),
@@ -164,6 +164,10 @@ export const SandboxInfoDataSchema = z
 					.string()
 					.optional()
 					.describe('Execution timeout duration (e.g., "30m", "2h").'),
+				paused: z
+					.string()
+					.optional()
+					.describe('Paused timeout duration (e.g., "24h", "0s" for infinite).'),
 			})
 			.optional()
 			.describe('Timeout configuration for the sandbox.'),
@@ -189,6 +193,16 @@ export const SandboxGetParamsSchema = z.object({
 		.boolean()
 		.optional()
 		.describe('Whether deleted sandboxes should be included in lookup'),
+	waitForStatus: z
+		.union([z.string(), z.array(z.string())])
+		.optional()
+		.describe('Optional desired status or statuses to wait for before responding'),
+	waitMs: z
+		.number()
+		.int()
+		.nonnegative()
+		.optional()
+		.describe('Maximum time in milliseconds to wait for the desired status'),
 });
 
 export type SandboxGetParams = z.infer<typeof SandboxGetParamsSchema>;
@@ -205,7 +219,7 @@ export async function sandboxGet(
 	client: APIClient,
 	params: SandboxGetParams
 ): Promise<SandboxInfo> {
-	const { sandboxId, orgId, includeDeleted } = params;
+	const { sandboxId, orgId, includeDeleted, waitForStatus, waitMs } = params;
 	const queryParams = new URLSearchParams();
 	if (orgId) {
 		queryParams.set('orgId', orgId);
@@ -213,6 +227,15 @@ export async function sandboxGet(
 	if (includeDeleted) {
 		queryParams.set('includeDeleted', 'true');
 	}
+	if (waitForStatus) {
+		queryParams.set(
+			'waitForStatus',
+			Array.isArray(waitForStatus) ? waitForStatus.join(',') : waitForStatus
+		);
+	}
+	if (waitMs != null) {
+		queryParams.set('waitMs', String(waitMs));
+	}
 	const queryString = queryParams.toString();
 	const url = `/sandbox/${encodeURIComponent(sandboxId)}${queryString ? `?${queryString}` : ''}`;
 
@@ -222,6 +245,12 @@ export async function sandboxGet(
 	);
 
 	if (resp.success) {
+		// Newly created sandboxes can be served from the server's pending cache before
+		// the backing row and related org hydration are fully available. That response
+		// serializes `org: null`, so normalize it to a placeholder object instead of
+		// failing response validation for an otherwise valid sandbox status poll.
+		const org = resp.data.org ?? { id: orgId ?? '', name: '' };
+
 		return {
 			sandboxId: resp.data.sandboxId,
 			identifier: resp.data.identifier,
@@ -252,7 +281,7 @@ export async function sandboxGet(
 			user: resp.data.user,
 			agent: resp.data.agent,
 			project: resp.data.project,
-			org: resp.data.org,
+			org,
 			timeout: resp.data.timeout,
 			command: resp.data.command,
 		};

package/src/services/sandbox/getStatus.ts

@@ -13,6 +13,17 @@ const SandboxStatusResponseSchema = APIResponseSchema(SandboxStatusDataSchema);
 export const SandboxGetStatusParamsSchema = z.object({
 	sandboxId: z.string().describe('Sandbox ID to retrieve status for'),
 	orgId: z.string().optional().describe('Optional org id for CLI auth context'),
+	waitForStatus: z
+		.union([z.string(), z.array(z.string())])
+		.optional()
+		.describe('Optional desired status or statuses to wait for before responding'),
+	waitMs: z
+		.number()
+		.int()
+		.nonnegative()
+		.optional()
+		.describe('Maximum time in milliseconds to wait for the desired status'),
+	signal: z.custom<AbortSignal>().optional().describe('abort signal for cancellation'),
 });
 
 export type SandboxGetStatusParams = z.infer<typeof SandboxGetStatusParamsSchema>;
@@ -26,17 +37,27 @@ export async function sandboxGetStatus(
 	client: APIClient,
 	params: SandboxGetStatusParams
 ): Promise<SandboxStatusResult> {
-	const { sandboxId, orgId } = params;
+	const { sandboxId, orgId, waitForStatus, waitMs, signal } = params;
 	const queryParams = new URLSearchParams();
 	if (orgId) {
 		queryParams.set('orgId', orgId);
 	}
+	if (waitForStatus) {
+		queryParams.set(
+			'waitForStatus',
+			Array.isArray(waitForStatus) ? waitForStatus.join(',') : waitForStatus
+		);
+	}
+	if (waitMs != null) {
+		queryParams.set('waitMs', String(waitMs));
+	}
 	const queryString = queryParams.toString();
 	const url = `/sandbox/status/${sandboxId}${queryString ? `?${queryString}` : ''}`;
 
 	const resp = await client.get<z.infer<typeof SandboxStatusResponseSchema>>(
 		url,
-		SandboxStatusResponseSchema
+		SandboxStatusResponseSchema,
+		signal
 	);
 
 	if (resp.success) {

package/src/services/sandbox/index.ts

@@ -52,7 +52,7 @@ export {
 	diskCheckpointRestore,
 	diskCheckpointDelete,
 } from './disk-checkpoint.ts';
-export type { SandboxPauseParams } from './pause.ts';
+export type { SandboxPauseParams, SandboxPauseResult } from './pause.ts';
 export { PauseResponseSchema, SandboxPauseParamsSchema, sandboxPause } from './pause.ts';
 export type { SandboxResumeParams } from './resume.ts';
 export { ResumeResponseSchema, SandboxResumeParamsSchema, sandboxResume } from './resume.ts';

package/src/services/sandbox/list.ts

@@ -112,6 +112,10 @@ export const SandboxInfoSchema = z
 					.string()
 					.optional()
 					.describe('Execution timeout duration (e.g., "30m", "2h").'),
+				paused: z
+					.string()
+					.optional()
+					.describe('Paused timeout duration (e.g., "24h", "0s" for infinite).'),
 			})
 			.optional()
 			.describe('Timeout configuration for the sandbox.'),
@@ -225,6 +229,7 @@ export async function sandboxList(
 				networkPort: s.networkPort,
 				url: s.url,
 				org: s.org,
+				timeout: s.timeout,
 			})),
 			total: resp.data.total,
 		};

package/src/services/sandbox/pause.ts

@@ -1,8 +1,21 @@
 import { z } from 'zod';
-import { type APIClient, APIResponseSchemaNoData } from '../api.ts';
+import { type APIClient } from '../api.ts';
 import { throwSandboxError } from './util.ts';
 
-export const PauseResponseSchema = APIResponseSchemaNoData();
+export const PauseResponseSchema = z.discriminatedUnion('success', [
+	z.object({
+		success: z.literal<false>(false),
+		message: z.string(),
+		code: z.string().optional(),
+	}),
+	z.object({
+		success: z.literal<true>(true),
+		sandboxId: z.string(),
+		status: z.string(),
+		checkpointId: z.string().optional(),
+		terminatesAt: z.string().optional(),
+	}),
+]);
 
 export const SandboxPauseParamsSchema = z.object({
 	sandboxId: z.string().describe('Sandbox ID to pause'),
@@ -11,14 +24,30 @@ export const SandboxPauseParamsSchema = z.object({
 
 export type SandboxPauseParams = z.infer<typeof SandboxPauseParamsSchema>;
 
+/** Result returned from pausing a sandbox */
+export interface SandboxPauseResult {
+	/** The sandbox ID that was paused */
+	sandboxId: string;
+	/** New status (typically "suspended") */
+	status: string;
+	/** Checkpoint ID created during pause */
+	checkpointId?: string;
+	/** ISO 8601 timestamp when sandbox will auto-terminate if not resumed (omitted if no paused timeout) */
+	terminatesAt?: string;
+}
+
 /**
  * Pauses a running sandbox, creating a checkpoint of its current state.
  *
  * @param client - The API client to use for the request
  * @param params - Parameters including the sandbox ID to pause
+ * @returns Pause result including terminatesAt if a paused timeout is configured
  * @throws {SandboxResponseError} If the sandbox is not found or pause fails
  */
-export async function sandboxPause(client: APIClient, params: SandboxPauseParams): Promise<void> {
+export async function sandboxPause(
+	client: APIClient,
+	params: SandboxPauseParams
+): Promise<SandboxPauseResult> {
 	const { sandboxId, orgId } = params;
 	const queryParams = new URLSearchParams();
 	if (orgId) {
@@ -34,7 +63,12 @@ export async function sandboxPause(client: APIClient, params: SandboxPauseParams
 	);
 
 	if (resp.success) {
-		return;
+		return {
+			sandboxId: resp.sandboxId,
+			status: resp.status,
+			checkpointId: resp.checkpointId,
+			terminatesAt: resp.terminatesAt,
+		};
 	}
 
 	throwSandboxError(resp, { sandboxId });