@agentuity/server 1.0.16 → 1.0.18

package/src/api/project/malware.ts

@@ -1,5 +1,6 @@
 import { z } from 'zod';
-import { type APIClient, APIResponseSchema } from '../api';
+import { type APIClient, APIResponseSchema } from '../api.ts';
+import { MalwareCheckError } from './util.ts';
 
 export const PackageRefSchema = z.object({
 	name: z.string(),
@@ -57,7 +58,10 @@ export async function projectDeploymentMalwareCheck(
 	);
 
 	if (!resp.success) {
-		throw new Error(resp.message || 'Malware check request failed');
+		throw new MalwareCheckError({
+			message: resp.message || 'Malware check request failed',
+			deploymentId,
+		});
 	}
 
 	return resp.data;

package/src/api/project/update-region.ts

@@ -1,6 +1,6 @@
 import { z } from 'zod';
-import { type APIClient, APIResponseSchema } from '../api';
-import { ProjectResponseError } from './util';
+import { type APIClient, APIResponseSchema } from '../api.ts';
+import { ProjectResponseError } from './util.ts';
 
 export const UpdateRegionRequestSchema = z.object({
 	cloudRegion: z.string().describe('the cloud region to update the project to'),

package/src/api/project/util.ts

@@ -3,3 +3,6 @@ import { StructuredError } from '@agentuity/core';
 export const ProjectResponseError = StructuredError('ProjectResponseError');
 export const ProjectNotFoundError = StructuredError('ProjectNotFoundError')<{ id: string }>();
 export const AgentNotFoundError = StructuredError('AgentNotFoundError')<{ id: string }>();
+export const MalwareCheckError = StructuredError('MalwareCheckError')<{
+	deploymentId: string;
+}>();

package/src/api/queue/analytics.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { type APIClient, APIResponseSchema } from '../api';
+import { type APIClient, APIResponseSchema } from '../api.ts';
 import {
 	type AnalyticsOptions,
 	type OrgAnalytics,
@@ -11,9 +11,15 @@ import {
 	type StreamAnalyticsOptions,
 	type TimeSeriesData,
 	TimeSeriesDataSchema,
-} from './types';
-import { buildQueueHeaders, QueueError, QueueNotFoundError, queueApiPathWithQuery } from './util';
-import { validateQueueName } from './validation';
+} from './types.ts';
+import {
+	buildQueueHeaders,
+	QueueError,
+	QueueNotFoundError,
+	queueApiPathWithQuery,
+	withQueueErrorHandling,
+} from './util.ts';
+import { validateQueueName } from './validation.ts';
 
 export const OrgAnalyticsResponseSchema = APIResponseSchema(
 	z.object({ analytics: OrgAnalyticsSchema })
@@ -113,24 +119,21 @@ export async function getQueueAnalytics(
 	validateQueueName(name);
 	const queryString = buildAnalyticsQuery(options);
 	const url = queueApiPathWithQuery('analytics/queue', queryString, name);
-	const resp = await client.get(
-		url,
-		QueueAnalyticsResponseSchema,
-		undefined,
-		buildQueueHeaders(options?.orgId)
+	const resp = await withQueueErrorHandling(
+		() =>
+			client.get(
+				url,
+				QueueAnalyticsResponseSchema,
+				undefined,
+				buildQueueHeaders(options?.orgId)
+			),
+		{ queueName: name }
 	);
 
 	if (resp.success) {
 		return resp.data.analytics;
 	}
 
-	if (resp.message?.includes('not found')) {
-		throw new QueueNotFoundError({
-			queueName: name,
-			message: resp.message,
-		});
-	}
-
 	throw new QueueError({
 		queueName: name,
 		message: resp.message || 'Failed to get queue analytics',
@@ -168,24 +171,21 @@ export async function getQueueTimeSeries(
 	validateQueueName(name);
 	const queryString = buildAnalyticsQuery(options);
 	const url = queueApiPathWithQuery('analytics/timeseries', queryString, name);
-	const resp = await client.get(
-		url,
-		TimeSeriesResponseSchema,
-		undefined,
-		buildQueueHeaders(options?.orgId)
+	const resp = await withQueueErrorHandling(
+		() =>
+			client.get(
+				url,
+				TimeSeriesResponseSchema,
+				undefined,
+				buildQueueHeaders(options?.orgId)
+			),
+		{ queueName: name }
 	);
 
 	if (resp.success) {
 		return resp.data.timeseries;
 	}
 
-	if (resp.message?.includes('not found')) {
-		throw new QueueNotFoundError({
-			queueName: name,
-			message: resp.message,
-		});
-	}
-
 	throw new QueueError({
 		queueName: name,
 		message: resp.message || 'Failed to get queue time series',

package/src/api/queue/destinations.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { type APIClient, APIError, APIResponseSchema, APIResponseSchemaNoData } from '../api';
+import { type APIClient, APIResponseSchema, APIResponseSchemaNoData } from '../api.ts';
 import {
 	type CreateDestinationRequest,
 	CreateDestinationRequestSchema,
@@ -8,16 +8,14 @@ import {
 	type QueueApiOptions,
 	type UpdateDestinationRequest,
 	UpdateDestinationRequestSchema,
-} from './types';
+} from './types.ts';
 import {
 	buildQueueHeaders,
-	DestinationAlreadyExistsError,
-	DestinationNotFoundError,
 	QueueError,
-	QueueNotFoundError,
 	queueApiPath,
-} from './util';
-import { validateDestinationConfig, validateDestinationId, validateQueueName } from './validation';
+	withQueueErrorHandling,
+} from './util.ts';
+import { validateDestinationConfig, validateDestinationId, validateQueueName } from './validation.ts';
 
 export const DestinationResponseSchema = APIResponseSchema(
 	z.object({ destination: DestinationSchema })
@@ -68,63 +66,27 @@ export async function createDestination(
 	}
 
 	const url = queueApiPath('destinations/create', queueName);
+	const resp = await withQueueErrorHandling(
+		() =>
+			client.post(
+				url,
+				params,
+				DestinationResponseSchema,
+				CreateDestinationRequestSchema,
+				undefined,
+				buildQueueHeaders(options?.orgId)
+			),
+		{ queueName, destinationUrl: params.config?.url }
+	);
 
-	try {
-		const resp = await client.post(
-			url,
-			params,
-			DestinationResponseSchema,
-			CreateDestinationRequestSchema,
-			undefined,
-			buildQueueHeaders(options?.orgId)
-		);
-
-		if (resp.success) {
-			return resp.data.destination;
-		}
-
-		if (resp.message?.includes('queue') && resp.message?.includes('not found')) {
-			throw new QueueNotFoundError({
-				queueName,
-				message: resp.message,
-			});
-		}
-
-		if (resp.message?.includes('already exists')) {
-			throw new DestinationAlreadyExistsError({
-				queueName,
-				url: params.config?.url,
-				message: `A destination with URL "${params.config?.url}" already exists for queue "${queueName}"`,
-			});
-		}
-
-		throw new QueueError({
-			queueName,
-			message: resp.message || 'Failed to create destination',
-		});
-	} catch (error) {
-		if (error instanceof APIError) {
-			const message = error.message || '';
-			if (message.includes('already exists')) {
-				throw new DestinationAlreadyExistsError({
-					queueName,
-					url: params.config?.url,
-					message: `A destination with URL "${params.config?.url}" already exists for queue "${queueName}"`,
-				});
-			}
-			if (message.includes('queue') && message.includes('not found')) {
-				throw new QueueNotFoundError({
-					queueName,
-					message,
-				});
-			}
-			throw new QueueError({
-				queueName,
-				message: message || 'Failed to create destination',
-			});
-		}
-		throw error;
+	if (resp.success) {
+		return resp.data.destination;
 	}
+
+	throw new QueueError({
+		queueName,
+		message: resp.message || 'Failed to create destination',
+	});
 }
 
 /**
@@ -155,24 +117,21 @@ export async function listDestinations(
 ): Promise<Destination[]> {
 	validateQueueName(queueName);
 	const url = queueApiPath('destinations/list', queueName);
-	const resp = await client.get(
-		url,
-		DestinationsListResponseSchema,
-		undefined,
-		buildQueueHeaders(options?.orgId)
+	const resp = await withQueueErrorHandling(
+		() =>
+			client.get(
+				url,
+				DestinationsListResponseSchema,
+				undefined,
+				buildQueueHeaders(options?.orgId)
+			),
+		{ queueName }
 	);
 
 	if (resp.success) {
 		return resp.data.destinations;
 	}
 
-	if (resp.message?.includes('not found')) {
-		throw new QueueNotFoundError({
-			queueName,
-			message: resp.message,
-		});
-	}
-
 	throw new QueueError({
 		queueName,
 		message: resp.message || 'Failed to list destinations',
@@ -225,34 +184,23 @@ export async function updateDestination(
 	}
 
 	const url = queueApiPath('destinations/update', queueName, destinationId);
-	const resp = await client.patch(
-		url,
-		params,
-		DestinationResponseSchema,
-		UpdateDestinationRequestSchema,
-		undefined,
-		buildQueueHeaders(options?.orgId)
+	const resp = await withQueueErrorHandling(
+		() =>
+			client.patch(
+				url,
+				params,
+				DestinationResponseSchema,
+				UpdateDestinationRequestSchema,
+				undefined,
+				buildQueueHeaders(options?.orgId)
+			),
+		{ queueName, destinationId }
 	);
 
 	if (resp.success) {
 		return resp.data.destination;
 	}
 
-	if (resp.message?.includes('destination') && resp.message?.includes('not found')) {
-		throw new DestinationNotFoundError({
-			queueName,
-			destinationId,
-			message: resp.message,
-		});
-	}
-
-	if (resp.message?.includes('queue') && resp.message?.includes('not found')) {
-		throw new QueueNotFoundError({
-			queueName,
-			message: resp.message,
-		});
-	}
-
 	throw new QueueError({
 		queueName,
 		message: resp.message || 'Failed to update destination',
@@ -290,32 +238,21 @@ export async function deleteDestination(
 	validateDestinationId(destinationId);
 
 	const url = queueApiPath('destinations/delete', queueName, destinationId);
-	const resp = await client.delete(
-		url,
-		DeleteDestinationResponseSchema,
-		undefined,
-		buildQueueHeaders(options?.orgId)
+	const resp = await withQueueErrorHandling(
+		() =>
+			client.delete(
+				url,
+				DeleteDestinationResponseSchema,
+				undefined,
+				buildQueueHeaders(options?.orgId)
+			),
+		{ queueName, destinationId }
 	);
 
 	if (resp.success) {
 		return;
 	}
 
-	if (resp.message?.includes('destination') && resp.message?.includes('not found')) {
-		throw new DestinationNotFoundError({
-			queueName,
-			destinationId,
-			message: resp.message,
-		});
-	}
-
-	if (resp.message?.includes('queue') && resp.message?.includes('not found')) {
-		throw new QueueNotFoundError({
-			queueName,
-			message: resp.message,
-		});
-	}
-
 	throw new QueueError({
 		queueName,
 		message: resp.message || 'Failed to delete destination',

package/src/api/queue/dlq.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { type APIClient, APIResponseSchema, APIResponseSchemaNoData } from '../api';
+import { type APIClient, APIResponseSchema, APIResponseSchemaNoData } from '../api.ts';
 import {
 	type DeadLetterMessage,
 	DeadLetterMessageSchema,
@@ -7,16 +7,15 @@ import {
 	type Message,
 	MessageSchema,
 	type QueueApiOptions,
-} from './types';
+} from './types.ts';
 import {
 	buildQueueHeaders,
-	MessageNotFoundError,
 	QueueError,
-	QueueNotFoundError,
 	queueApiPath,
 	queueApiPathWithQuery,
-} from './util';
-import { validateLimit, validateMessageId, validateOffset, validateQueueName } from './validation';
+	withQueueErrorHandling,
+} from './util.ts';
+import { validateLimit, validateMessageId, validateOffset, validateQueueName } from './validation.ts';
 
 export const DlqListResponseSchema = APIResponseSchema(
 	z.object({
@@ -75,24 +74,15 @@ export async function listDeadLetterMessages(
 
 	const queryString = searchParams.toString();
 	const url = queueApiPathWithQuery('dlq/list', queryString || undefined, queueName);
-	const resp = await client.get(
-		url,
-		DlqListResponseSchema,
-		undefined,
-		buildQueueHeaders(options?.orgId)
+	const resp = await withQueueErrorHandling(
+		() => client.get(url, DlqListResponseSchema, undefined, buildQueueHeaders(options?.orgId)),
+		{ queueName }
 	);
 
 	if (resp.success) {
 		return { messages: resp.data.messages, total: resp.data.total };
 	}
 
-	if (resp.message?.includes('not found')) {
-		throw new QueueNotFoundError({
-			queueName,
-			message: resp.message,
-		});
-	}
-
 	throw new QueueError({
 		queueName,
 		message: resp.message || 'Failed to list dead letter messages',
@@ -132,34 +122,23 @@ export async function replayDeadLetterMessage(
 	validateMessageId(messageId);
 
 	const url = queueApiPath('dlq/replay', queueName, messageId);
-	const resp = await client.post(
-		url,
-		undefined,
-		ReplayDlqResponseSchema,
-		undefined,
-		undefined,
-		buildQueueHeaders(options?.orgId)
+	const resp = await withQueueErrorHandling(
+		() =>
+			client.post(
+				url,
+				undefined,
+				ReplayDlqResponseSchema,
+				undefined,
+				undefined,
+				buildQueueHeaders(options?.orgId)
+			),
+		{ queueName, messageId }
 	);
 
 	if (resp.success) {
 		return resp.data.message;
 	}
 
-	if (resp.message?.includes('message') && resp.message?.includes('not found')) {
-		throw new MessageNotFoundError({
-			queueName,
-			messageId,
-			message: resp.message,
-		});
-	}
-
-	if (resp.message?.includes('queue') && resp.message?.includes('not found')) {
-		throw new QueueNotFoundError({
-			queueName,
-			message: resp.message,
-		});
-	}
-
 	throw new QueueError({
 		queueName,
 		message: resp.message || 'Failed to replay dead letter message',
@@ -194,24 +173,16 @@ export async function purgeDeadLetter(
 ): Promise<void> {
 	validateQueueName(queueName);
 	const url = queueApiPath('dlq/purge', queueName);
-	const resp = await client.delete(
-		url,
-		DeleteDlqResponseSchema,
-		undefined,
-		buildQueueHeaders(options?.orgId)
+	const resp = await withQueueErrorHandling(
+		() =>
+			client.delete(url, DeleteDlqResponseSchema, undefined, buildQueueHeaders(options?.orgId)),
+		{ queueName }
 	);
 
 	if (resp.success) {
 		return;
 	}
 
-	if (resp.message?.includes('not found')) {
-		throw new QueueNotFoundError({
-			queueName,
-			message: resp.message,
-		});
-	}
-
 	throw new QueueError({
 		queueName,
 		message: resp.message || 'Failed to purge dead letter queue',
@@ -250,32 +221,21 @@ export async function deleteDeadLetterMessage(
 	validateMessageId(messageId);
 
 	const url = queueApiPath('dlq/delete', queueName, messageId);
-	const resp = await client.delete(
-		url,
-		DeleteDlqResponseSchema,
-		undefined,
-		buildQueueHeaders(options?.orgId)
+	const resp = await withQueueErrorHandling(
+		() =>
+			client.delete(
+				url,
+				DeleteDlqResponseSchema,
+				undefined,
+				buildQueueHeaders(options?.orgId)
+			),
+		{ queueName, messageId }
 	);
 
 	if (resp.success) {
 		return;
 	}
 
-	if (resp.message?.includes('queue') && resp.message?.includes('not found')) {
-		throw new QueueNotFoundError({
-			queueName,
-			message: resp.message,
-		});
-	}
-
-	if (resp.message?.includes('message') && resp.message?.includes('not found')) {
-		throw new MessageNotFoundError({
-			queueName,
-			messageId,
-			message: resp.message,
-		});
-	}
-
 	throw new QueueError({
 		queueName,
 		message: resp.message || 'Failed to delete dead letter message',

package/src/api/queue/index.ts

@@ -136,7 +136,7 @@ export {
 	WebSocketAuthResponseSchema,
 	type WebSocketMessage,
 	WebSocketMessageSchema,
-} from './types';
+} from './types.ts';
 
 // ============================================================================
 // Errors
@@ -151,7 +151,7 @@ export {
 	QueueNotFoundError,
 	SourceAlreadyExistsError,
 	SourceNotFoundError,
-} from './util';
+} from './util.ts';
 
 // ============================================================================
 // Queue Operations
@@ -168,7 +168,7 @@ export {
 	QueuesListResponseSchema,
 	resumeQueue,
 	updateQueue,
-} from './queues';
+} from './queues.ts';
 
 // ============================================================================
 // Message Operations
@@ -195,7 +195,7 @@ export {
 	ReceiveResponseSchema,
 	receiveMessage,
 	replayMessage,
-} from './messages';
+} from './messages.ts';
 
 // ============================================================================
 // Dead Letter Queue Operations
@@ -209,7 +209,7 @@ export {
 	purgeDeadLetter,
 	ReplayDlqResponseSchema,
 	replayDeadLetterMessage,
-} from './dlq';
+} from './dlq.ts';
 
 // ============================================================================
 // Destination Operations
@@ -223,7 +223,7 @@ export {
 	deleteDestination,
 	listDestinations,
 	updateDestination,
-} from './destinations';
+} from './destinations.ts';
 
 // ============================================================================
 // Source Operations
@@ -238,7 +238,7 @@ export {
 	SourceResponseSchema,
 	SourcesListResponseSchema,
 	updateSource,
-} from './sources';
+} from './sources.ts';
 
 // ============================================================================
 // Analytics Operations
@@ -253,7 +253,7 @@ export {
 	streamOrgAnalytics,
 	streamQueueAnalytics,
 	TimeSeriesResponseSchema,
-} from './analytics';
+} from './analytics.ts';
 
 // ============================================================================
 // WebSocket Operations
@@ -266,7 +266,7 @@ export {
 	type QueueWebSocketConnection,
 	type QueueWebSocketState,
 	type SubscribeToQueueOptions,
-} from './websocket';
+} from './websocket.ts';
 
 // ============================================================================
 // Validation Utilities
@@ -305,4 +305,4 @@ export {
 	validateTTL,
 	validateVisibilityTimeout,
 	validateWebhookUrl,
-} from './validation';
+} from './validation.ts';