@agentuity/server 1.0.23 → 1.0.25

package/src/api/sandbox/disk-checkpoint.ts

@@ -0,0 +1,179 @@
+import { z } from 'zod';
+import type { z as zType } from 'zod';
+import { type APIClient, APIResponseSchema, APIResponseSchemaNoData } from '../api';
+import { API_VERSION, throwSandboxError } from './util';
+
+// --- Schemas ---
+
+export const DiskCheckpointInfoSchema = z.object({
+	id: z.string().describe('Globally unique checkpoint ID (ckpt_xxx)'),
+	name: z.string().describe('User-provided checkpoint name'),
+	createdAt: z.string().describe('ISO timestamp of creation'),
+	parent: z.string().describe('Parent checkpoint name (empty for base)'),
+});
+
+export type DiskCheckpointInfo = z.infer<typeof DiskCheckpointInfoSchema>;
+
+const CreateDiskCheckpointDataSchema = DiskCheckpointInfoSchema;
+export const CreateDiskCheckpointResponseSchema = APIResponseSchema(CreateDiskCheckpointDataSchema);
+
+const ListDiskCheckpointsDataSchema = z.object({
+	checkpoints: z.array(DiskCheckpointInfoSchema),
+});
+export const ListDiskCheckpointsResponseSchema = APIResponseSchema(ListDiskCheckpointsDataSchema);
+
+export const RestoreDiskCheckpointResponseSchema = APIResponseSchemaNoData();
+export const DeleteDiskCheckpointResponseSchema = APIResponseSchemaNoData();
+
+// --- Params ---
+
+export interface DiskCheckpointCreateParams {
+	sandboxId: string;
+	name: string;
+	orgId?: string;
+}
+
+export interface DiskCheckpointListParams {
+	sandboxId: string;
+	orgId?: string;
+}
+
+export interface DiskCheckpointRestoreParams {
+	sandboxId: string;
+	checkpointId: string;
+	orgId?: string;
+}
+
+export interface DiskCheckpointDeleteParams {
+	sandboxId: string;
+	checkpointId: string;
+	orgId?: string;
+}
+
+// --- API Functions ---
+
+/**
+ * Creates a new disk checkpoint for a sandbox.
+ *
+ * @param client - The API client to use for the request
+ * @param params - Parameters including the sandbox ID and checkpoint name
+ * @throws {SandboxResponseError} If the sandbox is not found or checkpoint creation fails
+ */
+export async function diskCheckpointCreate(
+	client: APIClient,
+	params: DiskCheckpointCreateParams
+): Promise<DiskCheckpointInfo> {
+	const { sandboxId, name, orgId } = params;
+	const queryParams = new URLSearchParams();
+	if (orgId) {
+		queryParams.set('orgId', orgId);
+	}
+	const queryString = queryParams.toString();
+	const url = `/sandbox/${API_VERSION}/${encodeURIComponent(sandboxId)}/checkpoint${queryString ? `?${queryString}` : ''}`;
+
+	const resp = await client.post<zType.infer<typeof CreateDiskCheckpointResponseSchema>>(
+		url,
+		{ name },
+		CreateDiskCheckpointResponseSchema
+	);
+
+	if (resp.success) {
+		return resp.data;
+	}
+
+	throwSandboxError(resp, { sandboxId });
+}
+
+/**
+ * Lists all disk checkpoints for a sandbox.
+ *
+ * @param client - The API client to use for the request
+ * @param params - Parameters including the sandbox ID
+ * @throws {SandboxResponseError} If the sandbox is not found or listing fails
+ */
+export async function diskCheckpointList(
+	client: APIClient,
+	params: DiskCheckpointListParams
+): Promise<DiskCheckpointInfo[]> {
+	const { sandboxId, orgId } = params;
+	const queryParams = new URLSearchParams();
+	if (orgId) {
+		queryParams.set('orgId', orgId);
+	}
+	const queryString = queryParams.toString();
+	const url = `/sandbox/${API_VERSION}/checkpoints/${encodeURIComponent(sandboxId)}${queryString ? `?${queryString}` : ''}`;
+
+	const resp = await client.get<zType.infer<typeof ListDiskCheckpointsResponseSchema>>(
+		url,
+		ListDiskCheckpointsResponseSchema
+	);
+
+	if (resp.success) {
+		return resp.data.checkpoints;
+	}
+
+	throwSandboxError(resp, { sandboxId });
+}
+
+/**
+ * Restores a sandbox to a specific disk checkpoint.
+ *
+ * @param client - The API client to use for the request
+ * @param params - Parameters including the sandbox ID and checkpoint ID
+ * @throws {SandboxResponseError} If the sandbox or checkpoint is not found, or restore fails
+ */
+export async function diskCheckpointRestore(
+	client: APIClient,
+	params: DiskCheckpointRestoreParams
+): Promise<void> {
+	const { sandboxId, checkpointId, orgId } = params;
+	const queryParams = new URLSearchParams();
+	if (orgId) {
+		queryParams.set('orgId', orgId);
+	}
+	const queryString = queryParams.toString();
+	const url = `/sandbox/${API_VERSION}/${encodeURIComponent(sandboxId)}/checkpoint/${encodeURIComponent(checkpointId)}/restore${queryString ? `?${queryString}` : ''}`;
+
+	const resp = await client.post<zType.infer<typeof RestoreDiskCheckpointResponseSchema>>(
+		url,
+		undefined,
+		RestoreDiskCheckpointResponseSchema
+	);
+
+	if (resp.success) {
+		return;
+	}
+
+	throwSandboxError(resp, { sandboxId });
+}
+
+/**
+ * Deletes a disk checkpoint from a sandbox.
+ *
+ * @param client - The API client to use for the request
+ * @param params - Parameters including the sandbox ID and checkpoint ID
+ * @throws {SandboxResponseError} If the sandbox or checkpoint is not found, or deletion fails
+ */
+export async function diskCheckpointDelete(
+	client: APIClient,
+	params: DiskCheckpointDeleteParams
+): Promise<void> {
+	const { sandboxId, checkpointId, orgId } = params;
+	const queryParams = new URLSearchParams();
+	if (orgId) {
+		queryParams.set('orgId', orgId);
+	}
+	const queryString = queryParams.toString();
+	const url = `/sandbox/${API_VERSION}/${encodeURIComponent(sandboxId)}/checkpoint/${encodeURIComponent(checkpointId)}${queryString ? `?${queryString}` : ''}`;
+
+	const resp = await client.delete<zType.infer<typeof DeleteDiskCheckpointResponseSchema>>(
+		url,
+		DeleteDiskCheckpointResponseSchema
+	);
+
+	if (resp.success) {
+		return;
+	}
+
+	throwSandboxError(resp, { sandboxId });
+}

package/src/api/sandbox/index.ts

@@ -16,6 +16,24 @@ export {
 } from './create.ts';
 export type { SandboxDestroyParams } from './destroy.ts';
 export { DestroyResponseSchema, sandboxDestroy } from './destroy.ts';
+export type {
+	DiskCheckpointInfo,
+	DiskCheckpointCreateParams,
+	DiskCheckpointListParams,
+	DiskCheckpointRestoreParams,
+	DiskCheckpointDeleteParams,
+} from './disk-checkpoint.ts';
+export {
+	DiskCheckpointInfoSchema,
+	CreateDiskCheckpointResponseSchema,
+	ListDiskCheckpointsResponseSchema,
+	RestoreDiskCheckpointResponseSchema,
+	DeleteDiskCheckpointResponseSchema,
+	diskCheckpointCreate,
+	diskCheckpointList,
+	diskCheckpointRestore,
+	diskCheckpointDelete,
+} from './disk-checkpoint.ts';
 export type { SandboxPauseParams } from './pause.ts';
 export { PauseResponseSchema, sandboxPause } from './pause.ts';
 export type { SandboxResumeParams } from './resume.ts';

package/src/api/services/index.ts

@@ -0,0 +1 @@
+export * from './stats.ts';

package/src/api/services/stats.ts

@@ -0,0 +1,213 @@
+import { z } from 'zod';
+import { type APIClient, APIResponseSchema } from '../api.ts';
+import { StructuredError } from '@agentuity/core';
+
+// --- Error ---
+
+export const ServiceStatsError = StructuredError('ServiceStatsError')<{
+	message: string;
+}>();
+
+// --- Per-Service Stat Schemas ---
+
+export const KeyValueStatSchema = z.object({
+	namespaceCount: z.number(),
+	keyCount: z.number(),
+	totalSizeBytes: z.number(),
+});
+
+export const VectorStatSchema = z.object({
+	namespaceCount: z.number(),
+	documentCount: z.number(),
+	totalSizeBytes: z.number(),
+});
+
+export const QueueStatSchema = z.object({
+	queueCount: z.number(),
+	totalMessages: z.number(),
+	totalDlq: z.number(),
+});
+
+export const StreamStatSchema = z.object({
+	streamCount: z.number(),
+	totalSizeBytes: z.number(),
+});
+
+export const SandboxStatSchema = z.object({
+	totalActive: z.number(),
+	running: z.number(),
+	idle: z.number(),
+	creating: z.number(),
+	totalExecutions: z.number(),
+	totalCpuTimeMs: z.number(),
+	totalMemoryByteSec: z.number(),
+	totalNetworkEgressBytes: z.number(),
+});
+
+export const EmailStatSchema = z.object({
+	addressCount: z.number(),
+	inboundCount: z.number(),
+	outboundCount: z.number(),
+	outboundSuccess: z.number(),
+	outboundFailed: z.number(),
+});
+
+export const TaskStatSchema = z.object({
+	total: z.number().default(0),
+	open: z.number().default(0),
+	inProgress: z.number().default(0),
+	done: z.number().default(0),
+	closed: z.number().default(0),
+	cancelled: z.number().default(0),
+});
+
+export const ScheduleStatSchema = z.object({
+	scheduleCount: z.number(),
+	totalDeliveries: z.number(),
+	successDeliveries: z.number(),
+	failedDeliveries: z.number(),
+});
+
+export const DatabaseStatSchema = z.object({
+	databaseCount: z.number(),
+	totalTableCount: z.number(),
+	totalRecordCount: z.number(),
+	totalSizeBytes: z.number(),
+});
+
+// --- Aggregate Schema ---
+
+export const ServiceStatsDataSchema = z.object({
+	services: z.object({
+		database: DatabaseStatSchema.optional(),
+		keyvalue: KeyValueStatSchema.optional(),
+		vector: VectorStatSchema.optional(),
+		queue: QueueStatSchema.optional(),
+		stream: StreamStatSchema.optional(),
+		sandbox: SandboxStatSchema.optional(),
+		email: EmailStatSchema.optional(),
+		task: TaskStatSchema.optional(),
+		schedule: ScheduleStatSchema.optional(),
+	}),
+});
+
+export const ServiceStatsResponseSchema = APIResponseSchema(ServiceStatsDataSchema);
+
+// --- Types ---
+
+export type KeyValueStat = z.infer<typeof KeyValueStatSchema>;
+export type VectorStat = z.infer<typeof VectorStatSchema>;
+export type QueueStat = z.infer<typeof QueueStatSchema>;
+export type StreamStat = z.infer<typeof StreamStatSchema>;
+export type SandboxStat = z.infer<typeof SandboxStatSchema>;
+export type EmailStat = z.infer<typeof EmailStatSchema>;
+export type TaskStat = z.infer<typeof TaskStatSchema>;
+export type ScheduleStat = z.infer<typeof ScheduleStatSchema>;
+export type DatabaseStat = z.infer<typeof DatabaseStatSchema>;
+export type ServiceStatsData = z.infer<typeof ServiceStatsDataSchema>;
+export type ServiceStatsResponse = z.infer<typeof ServiceStatsResponseSchema>;
+
+// --- Valid Services ---
+
+/**
+ * Valid service names that can be used to filter stats.
+ */
+export const VALID_SERVICES = [
+	'database',
+	'keyvalue',
+	'email',
+	'vector',
+	'schedule',
+	'task',
+	'stream',
+	'sandbox',
+	'queue',
+] as const;
+
+export type ServiceName = (typeof VALID_SERVICES)[number];
+
+// --- Options ---
+
+export interface ServiceStatsOptions {
+	/**
+	 * Filter to a specific service. If omitted, returns stats for all services.
+	 */
+	service?: ServiceName;
+	/**
+	 * Start time filter (ISO 8601 timestamp).
+	 */
+	start?: string;
+	/**
+	 * End time filter (ISO 8601 timestamp).
+	 */
+	end?: string;
+	/**
+	 * For CLI auth: sets x-agentuity-orgid header.
+	 * Required when using CLI token auth (bearer tokens without embedded org).
+	 */
+	orgIdHeader?: string;
+}
+
+// --- API Function ---
+
+/**
+ * Get aggregated stats for services used by an organization.
+ *
+ * Returns per-service stats with service-specific fields (counts, sizes, etc.).
+ * Services that error on the backend are omitted from the response.
+ * Services with no provisioned tenant DB return zero values.
+ *
+ * @param client - The API client instance
+ * @param orgId - The organization ID
+ * @param options - Optional filtering (service, time range)
+ * @returns Service stats data with per-service breakdown
+ * @throws {ServiceStatsError} If the API request fails
+ *
+ * @example
+ * ```typescript
+ * // Get stats for all services
+ * const stats = await getServiceStats(client, 'org_123');
+ * console.log(`KV keys: ${stats.services.keyvalue?.keyCount}`);
+ * ```
+ *
+ * @example
+ * ```typescript
+ * // Get stats for a specific service
+ * const stats = await getServiceStats(client, 'org_123', {
+ *   service: 'keyvalue',
+ * });
+ * ```
+ */
+export async function getServiceStats(
+	client: APIClient,
+	orgId: string,
+	options?: ServiceStatsOptions
+): Promise<ServiceStatsData> {
+	const params = new URLSearchParams();
+	if (options?.service) params.set('service', options.service);
+	if (options?.start) params.set('start', options.start);
+	if (options?.end) params.set('end', options.end);
+
+	const queryString = params.toString();
+	const url = `/services/stats/2026-02-26/${encodeURIComponent(orgId)}${queryString ? `?${queryString}` : ''}`;
+
+	const headers: Record<string, string> = {};
+	if (options?.orgIdHeader) {
+		headers['x-agentuity-orgid'] = options.orgIdHeader;
+	}
+
+	const resp = await client.get(
+		url,
+		ServiceStatsResponseSchema,
+		undefined,
+		Object.keys(headers).length > 0 ? headers : undefined
+	);
+
+	if (resp.success) {
+		return resp.data;
+	}
+
+	throw new ServiceStatsError({
+		message: resp.message || 'Failed to get service stats',
+	});
+}

package/src/server.ts

@@ -269,6 +269,10 @@ class ServerFetchAdapter implements FetchAdapter {
 		) {
 			headers['Content-Type'] = 'application/octet-stream';
 		}
+		// Ensure we request JSON responses for proper error handling
+		if (!headers['Accept'] && !headers['accept']) {
+			headers['Accept'] = 'application/json';
+		}
 		const method: HttpMethod = options.method ?? 'POST';
 		this.#logger.trace('sending %s to %s with headers: %s', method, url, redactHeaders(headers));
 		const res = await fetch(url, {