@electric-ax/agents-server 0.4.16 → 0.4.17

package/dist/entrypoint.js

@@ -4,7 +4,7 @@ import { DurableStreamTestServer } from "@durable-streams/server";
 import { createServer } from "node:http";
 import { createServerAdapter } from "@whatwg-node/server";
 import { Agent } from "undici";
-import { appendPathToUrl, assertTags, buildEventSourceManifestEntry, buildTagsIndex, createEntityRegistry, createRuntimeHandler, entityStateSchema, eventSourceSubscriptionManifestKey, getCronStreamPath, getCronStreamPathFromSpec, getEntitiesStreamPath, getNextCronFireAt, getSharedStateStreamPath, getWebhookStreamPath, hashString, manifestChildKey, manifestSharedStateKey, manifestSourceKey, normalizeTags, parseCronStreamPath, resolveCronScheduleSpec, resolveEventSourceSubscription, sourceRefForTags, verifyWebhookSignature } from "@electric-ax/agents-runtime";
+import { COMPOSER_INPUT_MESSAGE_TYPE, appendPathToUrl, assertTags, buildEventSourceManifestEntry, buildTagsIndex, createEntityRegistry, createRuntimeHandler, entityStateSchema, eventSourceSubscriptionManifestKey, getCronStreamPath, getCronStreamPathFromSpec, getEntitiesStreamPath, getNextCronFireAt, getSharedStateStreamPath, getWebhookStreamPath, hashString, manifestChildKey, manifestSharedStateKey, manifestSourceKey, normalizeTags, parseCronStreamPath, resolveCronScheduleSpec, resolveEventSourceSubscription, sourceRefForTags, validateComposerInputPayload, validateSlashCommandDefinitions, verifyWebhookSignature } from "@electric-ax/agents-runtime";
 import fs, { existsSync } from "node:fs";
 import path, { dirname, resolve } from "node:path";
 import { drizzle } from "drizzle-orm/postgres-js";
@@ -64,6 +64,7 @@ const entityTypes = pgTable(`entity_types`, {
 	creationSchema: jsonb(`creation_schema`),
 	inboxSchemas: jsonb(`inbox_schemas`),
 	stateSchemas: jsonb(`state_schemas`),
+	slashCommands: jsonb(`slash_commands`),
 	serveEndpoint: text(`serve_endpoint`),
 	defaultDispatchPolicy: jsonb(`default_dispatch_policy`),
 	revision: integer(`revision`).notNull().default(1),
@@ -1152,7 +1153,7 @@ function buildElectricProxyTarget(options) {
 			permissionBypass: options.permissionBypass
 		}));
 	} else if (table === `entity_types`) {
-		target.searchParams.set(`columns`, `"tenant_id","name","description","creation_schema","inbox_schemas","state_schemas","serve_endpoint","default_dispatch_policy","revision","created_at","updated_at"`);
+		target.searchParams.set(`columns`, `"tenant_id","name","description","creation_schema","inbox_schemas","state_schemas","slash_commands","serve_endpoint","default_dispatch_policy","revision","created_at","updated_at"`);
 		applyShapeWhere(target, buildSpawnableEntityTypesWhere({
 			tenantId: options.tenantId,
 			principalUrl: options.principalUrl ?? ``,
@@ -2725,6 +2726,7 @@ var PostgresRegistry = class {
 			creationSchema: et.creation_schema ?? null,
 			inboxSchemas: et.inbox_schemas ?? null,
 			stateSchemas: et.state_schemas ?? null,
+			slashCommands: et.slash_commands ?? null,
 			serveEndpoint: et.serve_endpoint ?? null,
 			defaultDispatchPolicy: et.default_dispatch_policy ?? null,
 			revision: et.revision,
@@ -2737,6 +2739,7 @@ var PostgresRegistry = class {
 				creationSchema: et.creation_schema ?? null,
 				inboxSchemas: et.inbox_schemas ?? null,
 				stateSchemas: et.state_schemas ?? null,
+				slashCommands: et.slash_commands ?? null,
 				serveEndpoint: et.serve_endpoint ?? null,
 				defaultDispatchPolicy: et.default_dispatch_policy ?? null,
 				revision: et.revision,
@@ -2754,6 +2757,7 @@ var PostgresRegistry = class {
 			creationSchema: et.creation_schema ?? null,
 			inboxSchemas: et.inbox_schemas ?? null,
 			stateSchemas: et.state_schemas ?? null,
+			slashCommands: et.slash_commands ?? null,
 			serveEndpoint: et.serve_endpoint ?? null,
 			defaultDispatchPolicy: et.default_dispatch_policy ?? null,
 			revision: et.revision,
@@ -2780,6 +2784,7 @@ var PostgresRegistry = class {
 			creationSchema: et.creation_schema ?? null,
 			inboxSchemas: et.inbox_schemas ?? null,
 			stateSchemas: et.state_schemas ?? null,
+			slashCommands: et.slash_commands ?? null,
 			serveEndpoint: et.serve_endpoint ?? null,
 			defaultDispatchPolicy: et.default_dispatch_policy ?? null,
 			revision: et.revision,
@@ -3370,6 +3375,7 @@ var PostgresRegistry = class {
 			creation_schema: row.creationSchema,
 			inbox_schemas: row.inboxSchemas,
 			state_schemas: row.stateSchemas,
+			slash_commands: row.slashCommands ?? void 0,
 			serve_endpoint: row.serveEndpoint ?? void 0,
 			default_dispatch_policy: row.defaultDispatchPolicy ?? void 0,
 			revision: row.revision,
@@ -3696,6 +3702,7 @@ var EntityManager = class {
 		this.validateSchema(req.creation_schema);
 		this.validateSchemaMap(req.inbox_schemas);
 		this.validateSchemaMap(req.state_schemas);
+		this.validateSlashCommands(req.slash_commands);
 		const defaultDispatchPolicy = req.default_dispatch_policy ? this.validateDispatchPolicy(req.default_dispatch_policy, { label: `default_dispatch_policy` }) : void 0;
 		const existing = await this.registry.getEntityType(req.name);
 		const now = new Date().toISOString();
@@ -3705,6 +3712,7 @@ var EntityManager = class {
 			creation_schema: req.creation_schema,
 			inbox_schemas: req.inbox_schemas,
 			state_schemas: req.state_schemas,
+			slash_commands: req.slash_commands,
 			serve_endpoint: req.serve_endpoint,
 			default_dispatch_policy: defaultDispatchPolicy,
 			revision: existing ? existing.revision + 1 : 1,
@@ -3866,6 +3874,18 @@ var EntityManager = class {
 			}
 		});
 		const initialEvents = [createdEvent];
+		const slashCommandTimestamp = new Date().toISOString();
+		for (const command of entityType.slash_commands ?? []) {
+			const slashCommandEvent = entityStateSchema.slashCommands.insert({
+				key: command.name,
+				value: {
+					...command,
+					source: `static`,
+					updated_at: slashCommandTimestamp
+				}
+			});
+			initialEvents.push(slashCommandEvent);
+		}
 		if (req.initialMessage !== void 0) {
 			const msgNow = new Date().toISOString();
 			const inboxEvent = entityStateSchema.inbox.insert({
@@ -3873,6 +3893,7 @@ var EntityManager = class {
 				value: {
 					from: req.created_by ?? req.parent ?? `spawn`,
 					payload: req.initialMessage,
+					message_type: req.initialMessageType,
 					timestamp: msgNow
 				}
 			});
@@ -5318,7 +5339,9 @@ var EntityManager = class {
 			creation_schema: existing.creation_schema,
 			inbox_schemas: mergedInbox,
 			state_schemas: mergedState,
+			slash_commands: existing.slash_commands,
 			serve_endpoint: existing.serve_endpoint,
+			default_dispatch_policy: existing.default_dispatch_policy,
 			revision: nextRevision,
 			created_at: existing.created_at,
 			updated_at: now
@@ -5372,11 +5395,19 @@ var EntityManager = class {
 			throw new ElectricAgentsError(ErrCodeInvalidRequest, error instanceof Error ? error.message : `Invalid tags`, 400);
 		}
 	}
+	validateSlashCommands(input) {
+		const validationError = validateSlashCommandDefinitions(input);
+		if (!validationError) return;
+		throw new ElectricAgentsError(ErrCodeSchemaValidationFailed, validationError.message, 422, validationError.details);
+	}
 	async validateSendRequest(entityUrl, req) {
 		const entity = await this.registry.getEntity(entityUrl);
 		if (!entity) throw new ElectricAgentsError(ErrCodeNotFound, `Entity not found`, 404);
 		if (rejectsNormalWrites(entity.status)) throw new ElectricAgentsError(ErrCodeNotRunning, `Entity is not accepting writes`, 409);
-		if (req.type && entity.type) {
+		if (req.type === COMPOSER_INPUT_MESSAGE_TYPE) {
+			const valErr = validateComposerInputPayload(req.payload);
+			if (valErr) throw new ElectricAgentsError(ErrCodeSchemaValidationFailed, valErr.message, 422, valErr.details);
+		} else if (req.type && entity.type) {
 			const { inboxSchemas } = await this.getEffectiveSchemas(entity);
 			if (inboxSchemas) {
 				const schema = inboxSchemas[req.type];
@@ -5657,6 +5688,7 @@ const spawnBodySchema = Type.Object({
 	sandbox: Type.Optional(sandboxChoiceSchema),
 	initialMessage: Type.Optional(Type.Unknown()),
 	grants: Type.Optional(Type.Array(entityPermissionGrantInputSchema)),
+	initialMessageType: Type.Optional(Type.String()),
 	wake: Type.Optional(Type.Object({
 		subscriberUrl: Type.String(),
 		condition: wakeConditionSchema,
@@ -6212,6 +6244,7 @@ async function spawnEntity(request, ctx) {
 		dispatch_policy: dispatchPolicy,
 		sandbox: parsed.sandbox,
 		initialMessage: void 0,
+		initialMessageType: void 0,
 		wake: parsed.wake,
 		created_by: principal.url
 	});
@@ -6230,7 +6263,8 @@ async function spawnEntity(request, ctx) {
 	if (linkBeforeInitialMessage) await linkEntityDispatchSubscription(ctx, entity);
 	if (parsed.initialMessage !== void 0) await ctx.entityManager.send(entity.url, {
 		from: principal.url,
-		payload: parsed.initialMessage
+		payload: parsed.initialMessage,
+		type: parsed.initialMessageType
 	});
 	if (!linkBeforeInitialMessage) await linkEntityDispatchSubscription(ctx, entity);
 	return json({
@@ -6277,6 +6311,21 @@ async function signalEntity(request, ctx) {
 //#region src/routing/entity-types-router.ts
 const jsonObjectSchema = Type.Record(Type.String(), Type.Unknown());
 const schemaMapSchema = Type.Record(Type.String(), jsonObjectSchema);
+const slashCommandArgumentSchema = Type.Object({
+	name: Type.String(),
+	type: Type.Union([
+		Type.Literal(`string`),
+		Type.Literal(`number`),
+		Type.Literal(`boolean`)
+	]),
+	required: Type.Optional(Type.Boolean()),
+	description: Type.Optional(Type.String())
+}, { additionalProperties: false });
+const slashCommandSchema = Type.Object({
+	name: Type.String(),
+	description: Type.Optional(Type.String()),
+	arguments: Type.Optional(Type.Array(slashCommandArgumentSchema))
+}, { additionalProperties: false });
 const typePermissionGrantInputSchema = Type.Object({
 	subject_kind: Type.Union([Type.Literal(`principal`), Type.Literal(`principal_kind`)]),
 	subject_value: Type.String(),
@@ -6289,6 +6338,7 @@ const registerEntityTypeBodySchema = Type.Object({
 	creation_schema: Type.Optional(jsonObjectSchema),
 	inbox_schemas: Type.Optional(schemaMapSchema),
 	state_schemas: Type.Optional(schemaMapSchema),
+	slash_commands: Type.Optional(Type.Array(slashCommandSchema)),
 	serve_endpoint: Type.Optional(Type.String()),
 	default_dispatch_policy: Type.Optional(dispatchPolicySchema),
 	permission_grants: Type.Optional(Type.Array(typePermissionGrantInputSchema))
@@ -6430,6 +6480,7 @@ function normalizeEntityTypeRequest(parsed) {
 		creation_schema: parsed.creation_schema,
 		inbox_schemas: parsed.inbox_schemas,
 		state_schemas: parsed.state_schemas,
+		slash_commands: parsed.slash_commands,
 		serve_endpoint: serveEndpoint,
 		default_dispatch_policy: parsed.default_dispatch_policy ?? (serveEndpoint ? { targets: [{
 			type: `webhook`,

package/dist/index.cjs

@@ -78,6 +78,7 @@ const entityTypes = (0, drizzle_orm_pg_core.pgTable)(`entity_types`, {
 	creationSchema: (0, drizzle_orm_pg_core.jsonb)(`creation_schema`),
 	inboxSchemas: (0, drizzle_orm_pg_core.jsonb)(`inbox_schemas`),
 	stateSchemas: (0, drizzle_orm_pg_core.jsonb)(`state_schemas`),
+	slashCommands: (0, drizzle_orm_pg_core.jsonb)(`slash_commands`),
 	serveEndpoint: (0, drizzle_orm_pg_core.text)(`serve_endpoint`),
 	defaultDispatchPolicy: (0, drizzle_orm_pg_core.jsonb)(`default_dispatch_policy`),
 	revision: (0, drizzle_orm_pg_core.integer)(`revision`).notNull().default(1),
@@ -849,6 +850,7 @@ var PostgresRegistry = class {
 			creationSchema: et.creation_schema ?? null,
 			inboxSchemas: et.inbox_schemas ?? null,
 			stateSchemas: et.state_schemas ?? null,
+			slashCommands: et.slash_commands ?? null,
 			serveEndpoint: et.serve_endpoint ?? null,
 			defaultDispatchPolicy: et.default_dispatch_policy ?? null,
 			revision: et.revision,
@@ -861,6 +863,7 @@ var PostgresRegistry = class {
 				creationSchema: et.creation_schema ?? null,
 				inboxSchemas: et.inbox_schemas ?? null,
 				stateSchemas: et.state_schemas ?? null,
+				slashCommands: et.slash_commands ?? null,
 				serveEndpoint: et.serve_endpoint ?? null,
 				defaultDispatchPolicy: et.default_dispatch_policy ?? null,
 				revision: et.revision,
@@ -878,6 +881,7 @@ var PostgresRegistry = class {
 			creationSchema: et.creation_schema ?? null,
 			inboxSchemas: et.inbox_schemas ?? null,
 			stateSchemas: et.state_schemas ?? null,
+			slashCommands: et.slash_commands ?? null,
 			serveEndpoint: et.serve_endpoint ?? null,
 			defaultDispatchPolicy: et.default_dispatch_policy ?? null,
 			revision: et.revision,
@@ -904,6 +908,7 @@ var PostgresRegistry = class {
 			creationSchema: et.creation_schema ?? null,
 			inboxSchemas: et.inbox_schemas ?? null,
 			stateSchemas: et.state_schemas ?? null,
+			slashCommands: et.slash_commands ?? null,
 			serveEndpoint: et.serve_endpoint ?? null,
 			defaultDispatchPolicy: et.default_dispatch_policy ?? null,
 			revision: et.revision,
@@ -1494,6 +1499,7 @@ var PostgresRegistry = class {
 			creation_schema: row.creationSchema,
 			inbox_schemas: row.inboxSchemas,
 			state_schemas: row.stateSchemas,
+			slash_commands: row.slashCommands ?? void 0,
 			serve_endpoint: row.serveEndpoint ?? void 0,
 			default_dispatch_policy: row.defaultDispatchPolicy ?? void 0,
 			revision: row.revision,
@@ -3387,6 +3393,7 @@ var EntityManager = class {
 		this.validateSchema(req.creation_schema);
 		this.validateSchemaMap(req.inbox_schemas);
 		this.validateSchemaMap(req.state_schemas);
+		this.validateSlashCommands(req.slash_commands);
 		const defaultDispatchPolicy = req.default_dispatch_policy ? this.validateDispatchPolicy(req.default_dispatch_policy, { label: `default_dispatch_policy` }) : void 0;
 		const existing = await this.registry.getEntityType(req.name);
 		const now = new Date().toISOString();
@@ -3396,6 +3403,7 @@ var EntityManager = class {
 			creation_schema: req.creation_schema,
 			inbox_schemas: req.inbox_schemas,
 			state_schemas: req.state_schemas,
+			slash_commands: req.slash_commands,
 			serve_endpoint: req.serve_endpoint,
 			default_dispatch_policy: defaultDispatchPolicy,
 			revision: existing ? existing.revision + 1 : 1,
@@ -3557,6 +3565,18 @@ var EntityManager = class {
 			}
 		});
 		const initialEvents = [createdEvent];
+		const slashCommandTimestamp = new Date().toISOString();
+		for (const command of entityType.slash_commands ?? []) {
+			const slashCommandEvent = __electric_ax_agents_runtime.entityStateSchema.slashCommands.insert({
+				key: command.name,
+				value: {
+					...command,
+					source: `static`,
+					updated_at: slashCommandTimestamp
+				}
+			});
+			initialEvents.push(slashCommandEvent);
+		}
 		if (req.initialMessage !== void 0) {
 			const msgNow = new Date().toISOString();
 			const inboxEvent = __electric_ax_agents_runtime.entityStateSchema.inbox.insert({
@@ -3564,6 +3584,7 @@ var EntityManager = class {
 				value: {
 					from: req.created_by ?? req.parent ?? `spawn`,
 					payload: req.initialMessage,
+					message_type: req.initialMessageType,
 					timestamp: msgNow
 				}
 			});
@@ -5009,7 +5030,9 @@ var EntityManager = class {
 			creation_schema: existing.creation_schema,
 			inbox_schemas: mergedInbox,
 			state_schemas: mergedState,
+			slash_commands: existing.slash_commands,
 			serve_endpoint: existing.serve_endpoint,
+			default_dispatch_policy: existing.default_dispatch_policy,
 			revision: nextRevision,
 			created_at: existing.created_at,
 			updated_at: now
@@ -5063,11 +5086,19 @@ var EntityManager = class {
 			throw new ElectricAgentsError(ErrCodeInvalidRequest, error instanceof Error ? error.message : `Invalid tags`, 400);
 		}
 	}
+	validateSlashCommands(input) {
+		const validationError = (0, __electric_ax_agents_runtime.validateSlashCommandDefinitions)(input);
+		if (!validationError) return;
+		throw new ElectricAgentsError(ErrCodeSchemaValidationFailed, validationError.message, 422, validationError.details);
+	}
 	async validateSendRequest(entityUrl, req) {
 		const entity = await this.registry.getEntity(entityUrl);
 		if (!entity) throw new ElectricAgentsError(ErrCodeNotFound, `Entity not found`, 404);
 		if (rejectsNormalWrites(entity.status)) throw new ElectricAgentsError(ErrCodeNotRunning, `Entity is not accepting writes`, 409);
-		if (req.type && entity.type) {
+		if (req.type === __electric_ax_agents_runtime.COMPOSER_INPUT_MESSAGE_TYPE) {
+			const valErr = (0, __electric_ax_agents_runtime.validateComposerInputPayload)(req.payload);
+			if (valErr) throw new ElectricAgentsError(ErrCodeSchemaValidationFailed, valErr.message, 422, valErr.details);
+		} else if (req.type && entity.type) {
 			const { inboxSchemas } = await this.getEffectiveSchemas(entity);
 			if (inboxSchemas) {
 				const schema = inboxSchemas[req.type];
@@ -7140,7 +7171,7 @@ function buildElectricProxyTarget(options) {
 			permissionBypass: options.permissionBypass
 		}));
 	} else if (table === `entity_types`) {
-		target.searchParams.set(`columns`, `"tenant_id","name","description","creation_schema","inbox_schemas","state_schemas","serve_endpoint","default_dispatch_policy","revision","created_at","updated_at"`);
+		target.searchParams.set(`columns`, `"tenant_id","name","description","creation_schema","inbox_schemas","state_schemas","slash_commands","serve_endpoint","default_dispatch_policy","revision","created_at","updated_at"`);
 		applyShapeWhere(target, buildSpawnableEntityTypesWhere({
 			tenantId: options.tenantId,
 			principalUrl: options.principalUrl ?? ``,
@@ -7948,6 +7979,7 @@ const spawnBodySchema = __sinclair_typebox.Type.Object({
 	sandbox: __sinclair_typebox.Type.Optional(sandboxChoiceSchema),
 	initialMessage: __sinclair_typebox.Type.Optional(__sinclair_typebox.Type.Unknown()),
 	grants: __sinclair_typebox.Type.Optional(__sinclair_typebox.Type.Array(entityPermissionGrantInputSchema)),
+	initialMessageType: __sinclair_typebox.Type.Optional(__sinclair_typebox.Type.String()),
 	wake: __sinclair_typebox.Type.Optional(__sinclair_typebox.Type.Object({
 		subscriberUrl: __sinclair_typebox.Type.String(),
 		condition: wakeConditionSchema,
@@ -8503,6 +8535,7 @@ async function spawnEntity(request, ctx) {
 		dispatch_policy: dispatchPolicy,
 		sandbox: parsed.sandbox,
 		initialMessage: void 0,
+		initialMessageType: void 0,
 		wake: parsed.wake,
 		created_by: principal.url
 	});
@@ -8521,7 +8554,8 @@ async function spawnEntity(request, ctx) {
 	if (linkBeforeInitialMessage) await linkEntityDispatchSubscription(ctx, entity);
 	if (parsed.initialMessage !== void 0) await ctx.entityManager.send(entity.url, {
 		from: principal.url,
-		payload: parsed.initialMessage
+		payload: parsed.initialMessage,
+		type: parsed.initialMessageType
 	});
 	if (!linkBeforeInitialMessage) await linkEntityDispatchSubscription(ctx, entity);
 	return (0, itty_router.json)({
@@ -8568,6 +8602,21 @@ async function signalEntity(request, ctx) {
 //#region src/routing/entity-types-router.ts
 const jsonObjectSchema = __sinclair_typebox.Type.Record(__sinclair_typebox.Type.String(), __sinclair_typebox.Type.Unknown());
 const schemaMapSchema = __sinclair_typebox.Type.Record(__sinclair_typebox.Type.String(), jsonObjectSchema);
+const slashCommandArgumentSchema = __sinclair_typebox.Type.Object({
+	name: __sinclair_typebox.Type.String(),
+	type: __sinclair_typebox.Type.Union([
+		__sinclair_typebox.Type.Literal(`string`),
+		__sinclair_typebox.Type.Literal(`number`),
+		__sinclair_typebox.Type.Literal(`boolean`)
+	]),
+	required: __sinclair_typebox.Type.Optional(__sinclair_typebox.Type.Boolean()),
+	description: __sinclair_typebox.Type.Optional(__sinclair_typebox.Type.String())
+}, { additionalProperties: false });
+const slashCommandSchema = __sinclair_typebox.Type.Object({
+	name: __sinclair_typebox.Type.String(),
+	description: __sinclair_typebox.Type.Optional(__sinclair_typebox.Type.String()),
+	arguments: __sinclair_typebox.Type.Optional(__sinclair_typebox.Type.Array(slashCommandArgumentSchema))
+}, { additionalProperties: false });
 const typePermissionGrantInputSchema = __sinclair_typebox.Type.Object({
 	subject_kind: __sinclair_typebox.Type.Union([__sinclair_typebox.Type.Literal(`principal`), __sinclair_typebox.Type.Literal(`principal_kind`)]),
 	subject_value: __sinclair_typebox.Type.String(),
@@ -8580,6 +8629,7 @@ const registerEntityTypeBodySchema = __sinclair_typebox.Type.Object({
 	creation_schema: __sinclair_typebox.Type.Optional(jsonObjectSchema),
 	inbox_schemas: __sinclair_typebox.Type.Optional(schemaMapSchema),
 	state_schemas: __sinclair_typebox.Type.Optional(schemaMapSchema),
+	slash_commands: __sinclair_typebox.Type.Optional(__sinclair_typebox.Type.Array(slashCommandSchema)),
 	serve_endpoint: __sinclair_typebox.Type.Optional(__sinclair_typebox.Type.String()),
 	default_dispatch_policy: __sinclair_typebox.Type.Optional(dispatchPolicySchema),
 	permission_grants: __sinclair_typebox.Type.Optional(__sinclair_typebox.Type.Array(typePermissionGrantInputSchema))
@@ -8721,6 +8771,7 @@ function normalizeEntityTypeRequest(parsed) {
 		creation_schema: parsed.creation_schema,
 		inbox_schemas: parsed.inbox_schemas,
 		state_schemas: parsed.state_schemas,
+		slash_commands: parsed.slash_commands,
 		serve_endpoint: serveEndpoint,
 		default_dispatch_policy: parsed.default_dispatch_policy ?? (serveEndpoint ? { targets: [{
 			type: `webhook`,