@assemblyline-agents/postgres 0.1.0

package/dist/adapter-delivery.js

@@ -0,0 +1,157 @@
+import { randomUUID } from "node:crypto";
+import { mapDelivery } from "./row-mappers.js";
+import { applyDeliveryPatchForPostgres, positiveLimit, postgresSandboxSyncRetryDelayMs, timestamp } from "./sql-helpers.js";
+export async function createDelivery(client, input) {
+    const now = timestamp();
+    const delivery = {
+        ...input,
+        attemptCount: input.attemptCount ?? 0,
+        maxAttempts: input.maxAttempts ?? 5,
+        nextAttemptAt: input.nextAttemptAt ?? now,
+        id: randomUUID(),
+        createdAt: now,
+        updatedAt: now
+    };
+    await client.query(`insert into openeve_delivery_obligations
+       (id, run_id, channel, payload, idempotency_key, status, attempt_count, max_attempts, next_attempt_at,
+        lease_token, lease_expires_at, last_error, sent_at, failed_at, created_at, updated_at)
+       values ($1,$2,$3,$4::jsonb,$5,$6,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16)`, [
+        delivery.id,
+        delivery.runId,
+        delivery.channel,
+        JSON.stringify(delivery.payload),
+        delivery.idempotencyKey,
+        delivery.status,
+        delivery.attemptCount,
+        delivery.maxAttempts,
+        delivery.nextAttemptAt,
+        delivery.leaseToken ?? null,
+        delivery.leaseExpiresAt ?? null,
+        delivery.lastError ?? null,
+        delivery.sentAt ?? null,
+        delivery.failedAt ?? null,
+        delivery.createdAt,
+        delivery.updatedAt
+    ]);
+    return delivery;
+}
+export async function updateDelivery(client, id, patch) {
+    const current = (await client.query("select * from openeve_delivery_obligations where id=$1", [id])).rows[0];
+    if (!current)
+        throw new Error(`Delivery not found: ${id}`);
+    const updated = applyDeliveryPatchForPostgres(mapDelivery(current), patch);
+    await persistDelivery(client, updated);
+    return updated;
+}
+async function persistDelivery(client, updated) {
+    await client.query(`update openeve_delivery_obligations
+       set status=$2, payload=$3::jsonb, attempt_count=$4, max_attempts=$5, next_attempt_at=$6,
+           lease_token=$7, lease_expires_at=$8, last_error=$9, sent_at=$10, failed_at=$11, updated_at=$12
+       where id=$1`, [
+        updated.id,
+        updated.status,
+        JSON.stringify(updated.payload),
+        updated.attemptCount,
+        updated.maxAttempts,
+        updated.nextAttemptAt,
+        updated.leaseToken ?? null,
+        updated.leaseExpiresAt ?? null,
+        updated.lastError ?? null,
+        updated.sentAt ?? null,
+        updated.failedAt ?? null,
+        updated.updatedAt
+    ]);
+}
+export async function listDeliveries(client, runId) {
+    const result = await client.query("select * from openeve_delivery_obligations where run_id=$1 order by created_at asc", [runId]);
+    return result.rows.map(mapDelivery);
+}
+export async function leaseDueDeliveries(client, options = {}) {
+    const now = options.now ?? new Date();
+    const leaseToken = randomUUID();
+    const leaseForMs = positiveLimit(options.leaseForMs, 60_000);
+    const leaseExpiresAt = new Date(now.getTime() + leaseForMs).toISOString();
+    const result = await client.query(`update openeve_delivery_obligations
+       set status='sending',
+           lease_token=$1,
+           lease_expires_at=$2,
+           attempt_count=attempt_count + 1,
+           updated_at=$3
+       where id in (
+         select id from openeve_delivery_obligations
+         where status='pending'
+           and next_attempt_at <= $3
+           and attempt_count < max_attempts
+         order by next_attempt_at asc, created_at asc
+         limit $4
+         for update skip locked
+       )
+       returning *`, [leaseToken, leaseExpiresAt, now.toISOString(), positiveLimit(options.limit, 10)]);
+    return result.rows.map(mapDelivery);
+}
+export async function completeDeliverySend(client, id, leaseToken, patch = {}) {
+    const current = (await client.query("select * from openeve_delivery_obligations where id=$1", [id])).rows[0];
+    if (!current)
+        throw new Error(`Delivery not found: ${id}`);
+    const delivery = mapDelivery(current);
+    if (delivery.status === "sent")
+        return delivery;
+    if (delivery.leaseToken !== leaseToken)
+        throw new Error(`Delivery lease mismatch: ${id}`);
+    const sentAt = timestamp();
+    // Token-enforced in the WHERE clause so the settle is atomic: if the lease
+    // expired mid-send and a worker re-leased the row between the SELECT above
+    // and this UPDATE, 0 rows match and we throw rather than clobbering the new
+    // holder's lease (which would re-open the double-send window).
+    const result = await client.query(`update openeve_delivery_obligations
+       set status='sent', payload=$2::jsonb, lease_token=null, lease_expires_at=null,
+           last_error=null, sent_at=$3, updated_at=$3
+       where id=$1 and lease_token=$4
+       returning *`, [id, JSON.stringify(patch.payload ?? delivery.payload), sentAt, leaseToken]);
+    const row = result.rows[0];
+    if (!row)
+        throw new Error(`Delivery lease mismatch: ${id}`);
+    return mapDelivery(row);
+}
+export async function failDeliverySend(client, id, leaseToken, failure) {
+    const current = (await client.query("select * from openeve_delivery_obligations where id=$1", [id])).rows[0];
+    if (!current)
+        throw new Error(`Delivery not found: ${id}`);
+    const delivery = mapDelivery(current);
+    if (delivery.leaseToken !== leaseToken)
+        throw new Error(`Delivery lease mismatch: ${id}`);
+    const now = failure.now ?? new Date();
+    const permanent = failure.permanent === true || delivery.attemptCount >= delivery.maxAttempts;
+    const nextAttemptAt = permanent
+        ? delivery.nextAttemptAt
+        : new Date(now.getTime() + positiveLimit(failure.retryDelayMs, postgresSandboxSyncRetryDelayMs(delivery.attemptCount))).toISOString();
+    const result = await client.query(`update openeve_delivery_obligations
+       set status=$2, next_attempt_at=$3, lease_token=null, lease_expires_at=null,
+           failed_at=$4, last_error=$5, updated_at=$6
+       where id=$1 and lease_token=$7
+       returning *`, [
+        id,
+        permanent ? "failed" : "pending",
+        nextAttemptAt,
+        permanent ? now.toISOString() : delivery.failedAt ?? null,
+        failure.error,
+        timestamp(),
+        leaseToken
+    ]);
+    const row = result.rows[0];
+    // 0 rows: the lease was reclaimed by another worker between SELECT and
+    // UPDATE; throw rather than clobber its state.
+    if (!row)
+        throw new Error(`Delivery lease mismatch: ${id}`);
+    return mapDelivery(row);
+}
+export async function recoverExpiredDeliveries(client, now = new Date()) {
+    const result = await client.query(`update openeve_delivery_obligations
+       set status='pending', next_attempt_at=$1, lease_token=null, lease_expires_at=null, updated_at=$1
+       where status='sending'
+         and lease_expires_at is not null
+         and lease_expires_at <= $1
+       returning *`, [now.toISOString()]);
+    return result.rows.map(mapDelivery);
+}
+//# sourceMappingURL=adapter-delivery.js.map

package/dist/adapter-delivery.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"adapter-delivery.js","sourceRoot":"","sources":["../src/adapter-delivery.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAUzC,OAAO,EAAE,WAAW,EAAY,MAAM,kBAAkB,CAAC;AACzD,OAAO,EAAE,6BAA6B,EAAE,aAAa,EAAE,+BAA+B,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAE5H,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,MAAmB,EAAE,KAA8B;IACtF,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;IACxB,MAAM,QAAQ,GAAuB;QACnC,GAAG,KAAK;QACR,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC;QACrC,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,CAAC;QACnC,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,GAAG;QACzC,EAAE,EAAE,UAAU,EAAE;QAChB,SAAS,EAAE,GAAG;QACd,SAAS,EAAE,GAAG;KACf,CAAC;IACF,MAAM,MAAM,CAAC,KAAK,CAChB;;;8EAG0E,EAC1E;QACE,QAAQ,CAAC,EAAE;QACX,QAAQ,CAAC,KAAK;QACd,QAAQ,CAAC,OAAO;QAChB,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC;QAChC,QAAQ,CAAC,cAAc;QACvB,QAAQ,CAAC,MAAM;QACf,QAAQ,CAAC,YAAY;QACrB,QAAQ,CAAC,WAAW;QACpB,QAAQ,CAAC,aAAa;QACtB,QAAQ,CAAC,UAAU,IAAI,IAAI;QAC3B,QAAQ,CAAC,cAAc,IAAI,IAAI;QAC/B,QAAQ,CAAC,SAAS,IAAI,IAAI;QAC1B,QAAQ,CAAC,MAAM,IAAI,IAAI;QACvB,QAAQ,CAAC,QAAQ,IAAI,IAAI;QACzB,QAAQ,CAAC,SAAS;QAClB,QAAQ,CAAC,SAAS;KACnB,CACF,CAAC;IACF,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,MAAmB,EAAE,EAAU,EAAE,KAA8B;IAClG,MAAM,OAAO,GAAG,CAAC,MAAM,MAAM,CAAC,KAAK,CAAM,wDAAwD,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClH,IAAI,CAAC,OAAO;QAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAC;IAC3D,MAAM,OAAO,GAAG,6BAA6B,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,KAAK,CAAC,CAAC;IAC3E,MAAM,eAAe,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACvC,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,KAAK,UAAU,eAAe,CAAC,MAAmB,EAAE,OAA2B;IAC7E,MAAM,MAAM,CAAC,KAAK,CAChB;;;mBAGe,EACf;QACE,OAAO,CAAC,EAAE;QACV,OAAO,CAAC,MAAM;QACd,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;QAC/B,OAAO,CAAC,YAAY;QACpB,OAAO,CAAC,WAAW;QACnB,OAAO,CAAC,aAAa;QACrB,OAAO,CAAC,UAAU,IAAI,IAAI;QAC1B,OAAO,CAAC,cAAc,IAAI,IAAI;QAC9B,OAAO,CAAC,SAAS,IAAI,IAAI;QACzB,OAAO,CAAC,MAAM,IAAI,IAAI;QACtB,OAAO,CAAC,QAAQ,IAAI,IAAI;QACxB,OAAO,CAAC,SAAS;KAClB,CACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,MAAmB,EAAE,KAAa;IACrE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAM,oFAAoF,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;IACtI,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;AACtC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,MAAmB,EAAE,UAAgC,EAAE;IAC9F,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC;IACtC,MAAM,UAAU,GAAG,UAAU,EAAE,CAAC;IAChC,MAAM,UAAU,GAAG,aAAa,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAC7D,MAAM,cAAc,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;IAC1E,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B;;;;;;;;;;;;;;;mBAee,EACf,CAAC,UAAU,EAAE,cAAc,EAAE,GAAG,CAAC,WAAW,EAAE,EAAE,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAClF,CAAC;IACF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;AACtC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,MAAmB,EAAE,EAAU,EAAE,UAAkB,EAAE,QAAkC,EAAE;IAClI,MAAM,OAAO,GAAG,CAAC,MAAM,MAAM,CAAC,KAAK,CAAM,wDAAwD,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClH,IAAI,CAAC,OAAO;QAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAC;IAC3D,MAAM,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACtC,IAAI,QAAQ,CAAC,MAAM,KAAK,MAAM;QAAE,OAAO,QAAQ,CAAC;IAChD,IAAI,QAAQ,CAAC,UAAU,KAAK,UAAU;QAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,EAAE,EAAE,CAAC,CAAC;IAC1F,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,2EAA2E;IAC3E,2EAA2E;IAC3E,4EAA4E;IAC5E,+DAA+D;IAC/D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B;;;;mBAIe,EACf,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,CAC5E,CAAC;IACF,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC3B,IAAI,CAAC,GAAG;QAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,EAAE,EAAE,CAAC,CAAC;IAC5D,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;AAC1B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,MAAmB,EAAE,EAAU,EAAE,UAAkB,EAAE,OAA6B;IACvH,MAAM,OAAO,GAAG,CAAC,MAAM,MAAM,CAAC,KAAK,CAAM,wDAAwD,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClH,IAAI,CAAC,OAAO;QAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAC;IAC3D,MAAM,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACtC,IAAI,QAAQ,CAAC,UAAU,KAAK,UAAU;QAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,EAAE,EAAE,CAAC,CAAC;IAC1F,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC;IACtC,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,KAAK,IAAI,IAAI,QAAQ,CAAC,YAAY,IAAI,QAAQ,CAAC,WAAW,CAAC;IAC9F,MAAM,aAAa,GAAG,SAAS;QAC7B,CAAC,CAAC,QAAQ,CAAC,aAAa;QACxB,CAAC,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,aAAa,CAAC,OAAO,CAAC,YAAY,EAAE,+BAA+B,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;IACxI,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B;;;;mBAIe,EACf;QACE,EAAE;QACF,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;QAChC,aAAa;QACb,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,IAAI,IAAI;QACzD,OAAO,CAAC,KAAK;QACb,SAAS,EAAE;QACX,UAAU;KACX,CACF,CAAC;IACF,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC3B,uEAAuE;IACvE,+CAA+C;IAC/C,IAAI,CAAC,GAAG;QAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,EAAE,EAAE,CAAC,CAAC;IAC5D,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;AAC1B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,MAAmB,EAAE,MAAY,IAAI,IAAI,EAAE;IACxF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B;;;;;mBAKe,EACf,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CACpB,CAAC;IACF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;AACtC,CAAC"}

package/dist/adapter-grants.d.ts

@@ -0,0 +1,11 @@
+import type { ConnectionAuthorizationSessionRecord, StoredConnectionGrant } from "@assemblyline-agents/runtime";
+import type { QueryClient } from "./client-types.js";
+export declare function getGrant(client: QueryClient, connectionName: string, principalKey: string): Promise<StoredConnectionGrant | undefined>;
+export declare function saveGrant(client: QueryClient, grant: StoredConnectionGrant): Promise<void>;
+export declare function deleteGrant(client: QueryClient, connectionName: string, principalKey: string): Promise<void>;
+export declare function createSession(client: QueryClient, session: ConnectionAuthorizationSessionRecord): Promise<void>;
+export declare function updateSession(client: QueryClient, id: string, patch: Partial<Omit<ConnectionAuthorizationSessionRecord, "id" | "createdAt">>): Promise<ConnectionAuthorizationSessionRecord>;
+export declare function getSession(client: QueryClient, id: string): Promise<ConnectionAuthorizationSessionRecord | undefined>;
+export declare function getSessionByState(client: QueryClient, state: string): Promise<ConnectionAuthorizationSessionRecord | undefined>;
+export declare function findPendingSession(client: QueryClient, connectionName: string, principalKey: string): Promise<ConnectionAuthorizationSessionRecord | undefined>;
+//# sourceMappingURL=adapter-grants.d.ts.map

package/dist/adapter-grants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"adapter-grants.d.ts","sourceRoot":"","sources":["../src/adapter-grants.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,oCAAoC,EACpC,qBAAqB,EACtB,MAAM,8BAA8B,CAAC;AACtC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAIrD,wBAAsB,QAAQ,CAAC,MAAM,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,GAAG,SAAS,CAAC,CAO5I;AAED,wBAAsB,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC,CA4BhG;AAED,wBAAsB,WAAW,CAAC,MAAM,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAKlH;AAED,wBAAsB,aAAa,CAAC,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,oCAAoC,GAAG,OAAO,CAAC,IAAI,CAAC,CA0BrH;AAED,wBAAsB,aAAa,CACjC,MAAM,EAAE,WAAW,EACnB,EAAE,EAAE,MAAM,EACV,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,oCAAoC,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GAC7E,OAAO,CAAC,oCAAoC,CAAC,CA4C/C;AAED,wBAAsB,UAAU,CAAC,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,oCAAoC,GAAG,SAAS,CAAC,CAI3H;AAED,wBAAsB,iBAAiB,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,oCAAoC,GAAG,SAAS,CAAC,CAIrI;AAED,wBAAsB,kBAAkB,CAAC,MAAM,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,oCAAoC,GAAG,SAAS,CAAC,CAUrK"}

package/dist/adapter-grants.js

@@ -0,0 +1,125 @@
+import { mapConnectionAuthorizationSession, mapConnectionGrant } from "./row-mappers.js";
+import { timestamp } from "./sql-helpers.js";
+export async function getGrant(client, connectionName, principalKey) {
+    const result = await client.query("select * from openeve_connection_grants where connection_name=$1 and principal_key=$2", [connectionName, principalKey]);
+    const row = result.rows[0];
+    return row ? mapConnectionGrant(row) : undefined;
+}
+export async function saveGrant(client, grant) {
+    await client.query(`insert into openeve_connection_grants
+     (connection_name, principal_key, principal, token, refresh_token, token_type, scopes, expires_at, credential, created_at, updated_at)
+     values ($1,$2,$3::jsonb,$4,$5,$6,$7::jsonb,$8,$9::jsonb,$10,$11)
+     on conflict (connection_name, principal_key) do update set
+       principal=$3::jsonb,
+       token=$4,
+       refresh_token=$5,
+       token_type=$6,
+       scopes=$7::jsonb,
+       expires_at=$8,
+       credential=$9::jsonb,
+       updated_at=$11`, [
+        grant.connectionName,
+        grant.principalKey,
+        JSON.stringify(grant.principal),
+        grant.token,
+        grant.refreshToken ?? null,
+        grant.tokenType ?? null,
+        grant.scopes ? JSON.stringify(grant.scopes) : null,
+        grant.expiresAt ?? null,
+        grant.credential ? JSON.stringify(grant.credential) : null,
+        grant.createdAt,
+        grant.updatedAt
+    ]);
+}
+export async function deleteGrant(client, connectionName, principalKey) {
+    await client.query("delete from openeve_connection_grants where connection_name=$1 and principal_key=$2", [
+        connectionName,
+        principalKey
+    ]);
+}
+export async function createSession(client, session) {
+    await client.query(`insert into openeve_connection_authorization_sessions
+     (id, state, connection_name, principal_key, principal, session, callback_url, code_verifier, resume, run_id, tool_call_id, tool_name, challenge, continuation, status, expires_at, created_at, updated_at)
+     values ($1,$2,$3,$4,$5::jsonb,$6::jsonb,$7,$8,$9::jsonb,$10,$11,$12,$13::jsonb,$14::jsonb,$15,$16,$17,$18)`, [
+        session.id,
+        session.state,
+        session.connectionName,
+        session.principalKey,
+        JSON.stringify(session.principal),
+        JSON.stringify(session.session),
+        session.callbackUrl,
+        session.codeVerifier ?? null,
+        session.resume === undefined ? null : JSON.stringify(session.resume),
+        session.runId ?? null,
+        session.toolCallId ?? null,
+        session.toolName ?? null,
+        JSON.stringify(session.challenge),
+        session.continuation === undefined ? null : JSON.stringify(session.continuation),
+        session.status,
+        session.expiresAt,
+        session.createdAt,
+        session.updatedAt
+    ]);
+}
+export async function updateSession(client, id, patch) {
+    const current = await getSession(client, id);
+    if (!current)
+        throw new Error(`Connection authorization session not found: ${id}`);
+    const updated = { ...current, ...patch, updatedAt: timestamp() };
+    await client.query(`update openeve_connection_authorization_sessions
+     set state=$2,
+         connection_name=$3,
+         principal_key=$4,
+         principal=$5::jsonb,
+         session=$6::jsonb,
+         callback_url=$7,
+         code_verifier=$8,
+         resume=$9::jsonb,
+         run_id=$10,
+         tool_call_id=$11,
+         tool_name=$12,
+         challenge=$13::jsonb,
+         continuation=$14::jsonb,
+         status=$15,
+         expires_at=$16,
+         updated_at=$17
+     where id=$1`, [
+        updated.id,
+        updated.state,
+        updated.connectionName,
+        updated.principalKey,
+        JSON.stringify(updated.principal),
+        JSON.stringify(updated.session),
+        updated.callbackUrl,
+        updated.codeVerifier ?? null,
+        updated.resume === undefined ? null : JSON.stringify(updated.resume),
+        updated.runId ?? null,
+        updated.toolCallId ?? null,
+        updated.toolName ?? null,
+        JSON.stringify(updated.challenge),
+        updated.continuation === undefined ? null : JSON.stringify(updated.continuation),
+        updated.status,
+        updated.expiresAt,
+        updated.updatedAt
+    ]);
+    return updated;
+}
+export async function getSession(client, id) {
+    const result = await client.query("select * from openeve_connection_authorization_sessions where id=$1", [id]);
+    const row = result.rows[0];
+    return row ? mapConnectionAuthorizationSession(row) : undefined;
+}
+export async function getSessionByState(client, state) {
+    const result = await client.query("select * from openeve_connection_authorization_sessions where state=$1", [state]);
+    const row = result.rows[0];
+    return row ? mapConnectionAuthorizationSession(row) : undefined;
+}
+export async function findPendingSession(client, connectionName, principalKey) {
+    const result = await client.query(`select * from openeve_connection_authorization_sessions
+     where connection_name=$1 and principal_key=$2 and status='pending' and expires_at>$3
+     order by created_at desc
+     limit 1`, [connectionName, principalKey, Date.now()]);
+    const row = result.rows[0];
+    return row ? mapConnectionAuthorizationSession(row) : undefined;
+}
+//# sourceMappingURL=adapter-grants.js.map

package/dist/adapter-grants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"adapter-grants.js","sourceRoot":"","sources":["../src/adapter-grants.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,iCAAiC,EAAE,kBAAkB,EAAY,MAAM,kBAAkB,CAAC;AACnG,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAE7C,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,MAAmB,EAAE,cAAsB,EAAE,YAAoB;IAC9F,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B,uFAAuF,EACvF,CAAC,cAAc,EAAE,YAAY,CAAC,CAC/B,CAAC;IACF,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC3B,OAAO,GAAG,CAAC,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AACnD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,MAAmB,EAAE,KAA4B;IAC/E,MAAM,MAAM,CAAC,KAAK,CAChB;;;;;;;;;;;sBAWkB,EAClB;QACE,KAAK,CAAC,cAAc;QACpB,KAAK,CAAC,YAAY;QAClB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC;QAC/B,KAAK,CAAC,KAAK;QACX,KAAK,CAAC,YAAY,IAAI,IAAI;QAC1B,KAAK,CAAC,SAAS,IAAI,IAAI;QACvB,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI;QAClD,KAAK,CAAC,SAAS,IAAI,IAAI;QACvB,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI;QAC1D,KAAK,CAAC,SAAS;QACf,KAAK,CAAC,SAAS;KAChB,CACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,MAAmB,EAAE,cAAsB,EAAE,YAAoB;IACjG,MAAM,MAAM,CAAC,KAAK,CAAC,qFAAqF,EAAE;QACxG,cAAc;QACd,YAAY;KACb,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,MAAmB,EAAE,OAA6C;IACpG,MAAM,MAAM,CAAC,KAAK,CAChB;;gHAE4G,EAC5G;QACE,OAAO,CAAC,EAAE;QACV,OAAO,CAAC,KAAK;QACb,OAAO,CAAC,cAAc;QACtB,OAAO,CAAC,YAAY;QACpB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC;QACjC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;QAC/B,OAAO,CAAC,WAAW;QACnB,OAAO,CAAC,YAAY,IAAI,IAAI;QAC5B,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC;QACpE,OAAO,CAAC,KAAK,IAAI,IAAI;QACrB,OAAO,CAAC,UAAU,IAAI,IAAI;QAC1B,OAAO,CAAC,QAAQ,IAAI,IAAI;QACxB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC;QACjC,OAAO,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,CAAC;QAChF,OAAO,CAAC,MAAM;QACd,OAAO,CAAC,SAAS;QACjB,OAAO,CAAC,SAAS;QACjB,OAAO,CAAC,SAAS;KAClB,CACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,MAAmB,EACnB,EAAU,EACV,KAA8E;IAE9E,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAC7C,IAAI,CAAC,OAAO;QAAE,MAAM,IAAI,KAAK,CAAC,+CAA+C,EAAE,EAAE,CAAC,CAAC;IACnF,MAAM,OAAO,GAAyC,EAAE,GAAG,OAAO,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,CAAC;IACvG,MAAM,MAAM,CAAC,KAAK,CAChB;;;;;;;;;;;;;;;;;iBAiBa,EACb;QACE,OAAO,CAAC,EAAE;QACV,OAAO,CAAC,KAAK;QACb,OAAO,CAAC,cAAc;QACtB,OAAO,CAAC,YAAY;QACpB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC;QACjC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;QAC/B,OAAO,CAAC,WAAW;QACnB,OAAO,CAAC,YAAY,IAAI,IAAI;QAC5B,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC;QACpE,OAAO,CAAC,KAAK,IAAI,IAAI;QACrB,OAAO,CAAC,UAAU,IAAI,IAAI;QAC1B,OAAO,CAAC,QAAQ,IAAI,IAAI;QACxB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC;QACjC,OAAO,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,CAAC;QAChF,OAAO,CAAC,MAAM;QACd,OAAO,CAAC,SAAS;QACjB,OAAO,CAAC,SAAS;KAClB,CACF,CAAC;IACF,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,MAAmB,EAAE,EAAU;IAC9D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAM,qEAAqE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACpH,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC3B,OAAO,GAAG,CAAC,CAAC,CAAC,iCAAiC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAClE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,MAAmB,EAAE,KAAa;IACxE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAM,wEAAwE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;IAC1H,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC3B,OAAO,GAAG,CAAC,CAAC,CAAC,iCAAiC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAClE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,MAAmB,EAAE,cAAsB,EAAE,YAAoB;IACxG,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B;;;aAGS,EACT,CAAC,cAAc,EAAE,YAAY,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAC3C,CAAC;IACF,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC3B,OAAO,GAAG,CAAC,CAAC,CAAC,iCAAiC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAClE,CAAC"}

package/dist/adapter-memory.d.ts

@@ -0,0 +1,21 @@
+import type { MemoryListInput, MemorySearchInput } from "@assemblyline-agents/core";
+import type { MemoryDocument, MemoryEmbeddingRecord, MemoryScope, MemorySearchResult } from "@assemblyline-agents/runtime";
+import type { QueryClient } from "./client-types.js";
+export declare function listMemoryDocuments(client: QueryClient, scope: MemoryScope, input?: MemoryListInput): Promise<MemoryDocument[]>;
+export declare function searchMemoryDocuments(client: QueryClient, scope: MemoryScope, input: MemorySearchInput): Promise<MemorySearchResult[]>;
+export declare function getMemoryDocument(client: QueryClient, scope: MemoryScope, path: string): Promise<MemoryDocument | undefined>;
+export declare function upsertMemoryDocument(client: QueryClient, document: MemoryDocument): Promise<MemoryDocument>;
+export declare function deleteMemoryDocument(client: QueryClient, scope: MemoryScope, path: string, options?: {
+    embeddingsAvailable?: boolean;
+}): Promise<void>;
+export declare function listMemoryDocumentsNeedingEmbeddings(client: QueryClient, scope: MemoryScope, options: {
+    model: string;
+    limit?: number;
+}): Promise<MemoryDocument[]>;
+export declare function upsertMemoryEmbedding(client: QueryClient, input: Omit<MemoryEmbeddingRecord, "id" | "createdAt" | "updatedAt">): Promise<MemoryEmbeddingRecord>;
+export declare function searchMemoryEmbeddings(client: QueryClient, scope: MemoryScope, input: MemorySearchInput, embedding: number[], options: {
+    model: string;
+    limit?: number;
+}): Promise<MemorySearchResult[]>;
+export declare function markMemoryEmbeddingsStale(client: QueryClient, scope: MemoryScope, path: string, currentSha256: string): Promise<void>;
+//# sourceMappingURL=adapter-memory.d.ts.map

package/dist/adapter-memory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"adapter-memory.d.ts","sourceRoot":"","sources":["../src/adapter-memory.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AACpF,OAAO,KAAK,EAAE,cAAc,EAAE,qBAAqB,EAAE,WAAW,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAC3H,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAkBrD,wBAAsB,mBAAmB,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,GAAE,eAAoB,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAYzI;AAED,wBAAsB,qBAAqB,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAuC5I;AAED,wBAAsB,iBAAiB,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC,CAQlI;AAED,wBAAsB,oBAAoB,CAAC,MAAM,EAAE,WAAW,EAAE,QAAQ,EAAE,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC,CA0CjH;AAED,wBAAsB,oBAAoB,CACxC,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,WAAW,EAClB,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE;IAAE,mBAAmB,CAAC,EAAE,OAAO,CAAA;CAAO,GAC9C,OAAO,CAAC,IAAI,CAAC,CAOf;AAED,wBAAsB,oCAAoC,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAezK;AAED,wBAAsB,qBAAqB,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,IAAI,CAAC,qBAAqB,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAqCrK;AAED,wBAAsB,sBAAsB,CAC1C,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,WAAW,EAClB,KAAK,EAAE,iBAAiB,EACxB,SAAS,EAAE,MAAM,EAAE,EACnB,OAAO,EAAE;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,GACzC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAyB/B;AAED,wBAAsB,yBAAyB,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAO3I"}

package/dist/adapter-memory.js

@@ -0,0 +1,181 @@
+import { randomUUID } from "node:crypto";
+import { mapMemoryDocument, mapMemoryEmbedding, memoryEmbeddingSearchResultFromPostgres, memorySearchResultFromPostgres } from "./row-mappers.js";
+import { memoryScopeMatchesForPostgres, normalizeMemoryPathForPostgres, positiveLimit, postgresMemoryKey, postgresMemoryScopeWhere, postgresVectorLiteral, timestamp } from "./sql-helpers.js";
+export async function listMemoryDocuments(client, scope, input = {}) {
+    const result = await client.query("select * from openeve_memory_documents where scope->>'agentScope'=$1 order by updated_at desc", [scope.agentScope]);
+    const prefix = normalizeMemoryPathForPostgres(input.pathPrefix ?? "");
+    const limit = input.limit === undefined ? Number.MAX_SAFE_INTEGER : positiveLimit(input.limit, 50);
+    return result.rows
+        .map(mapMemoryDocument)
+        .filter((document) => memoryScopeMatchesForPostgres(scope, document.scope))
+        .filter((document) => !prefix || document.path.startsWith(prefix))
+        .slice(0, limit);
+}
+export async function searchMemoryDocuments(client, scope, input) {
+    const query = String(input.query ?? "").trim();
+    const pathPrefix = normalizeMemoryPathForPostgres(input.pathPrefix ?? "");
+    const limit = positiveLimit(input.limit, 8);
+    const mode = input.mode ?? "hybrid";
+    const scopeFilter = postgresMemoryScopeWhere(scope, 5);
+    const params = [scope.agentScope, query, pathPrefix, limit, ...scopeFilter.params];
+    const exact = mode === "exact";
+    const result = await client.query(exact
+        ? `select *, 1::float as rank
+           from openeve_memory_documents
+           where scope->>'agentScope'=$1
+             and ($3 = '' or normalized_path like ($3 || '%'))
+             and (normalized_path = $2 or path = $2)
+             ${scopeFilter.sql}
+           order by updated_at desc
+           limit $4`
+        : `select *,
+              ts_rank(search_vector, plainto_tsquery('simple', $2)) as rank,
+              ts_headline('simple', body::text, plainto_tsquery('simple', $2), 'MaxWords=32, MinWords=8, ShortWord=3') as snippet
+           from openeve_memory_documents
+           where scope->>'agentScope'=$1
+             and ($3 = '' or normalized_path like ($3 || '%'))
+             and (
+               $2 = ''
+               or normalized_path = $2
+               or normalized_path like ('%' || $2 || '%')
+               or search_vector @@ plainto_tsquery('simple', $2)
+             )
+             ${scopeFilter.sql}
+           order by
+             case when normalized_path = $2 then 1000 else 0 end desc,
+             rank desc,
+             updated_at desc
+           limit $4`, params);
+    return result.rows.map((row) => memorySearchResultFromPostgres(row, query, mode));
+}
+export async function getMemoryDocument(client, scope, path) {
+    const normalizedPath = normalizeMemoryPathForPostgres(path);
+    const result = await client.query("select * from openeve_memory_documents where scope_key=$1", [postgresMemoryKey(scope, normalizedPath)]);
+    const row = result.rows[0];
+    return row ? mapMemoryDocument(row) : undefined;
+}
+export async function upsertMemoryDocument(client, document) {
+    const result = await client.query(`insert into openeve_memory_documents
+       (id, scope_key, scope, path, normalized_path, body, content_type, title, summary, tags, confidence, source_refs, token_estimate, sha256, revision, created_at, updated_at)
+       values ($1,$2,$3::jsonb,$4,$5,$6::jsonb,$7,$8,$9,$10::jsonb,$11,$12::jsonb,$13,$14,$15,$16,$17)
+       on conflict (scope_key) do update set
+         scope=$3::jsonb,
+         path=$4,
+         normalized_path=$5,
+         body=$6::jsonb,
+         content_type=$7,
+         title=$8,
+         summary=$9,
+         tags=$10::jsonb,
+         confidence=$11,
+         source_refs=$12::jsonb,
+         token_estimate=$13,
+         sha256=$14,
+         revision=$15,
+         updated_at=$17
+       returning *`, [
+        randomUUID(),
+        postgresMemoryKey(document.scope, document.path),
+        JSON.stringify(document.scope),
+        document.path,
+        normalizeMemoryPathForPostgres(document.path),
+        JSON.stringify(document.body),
+        document.contentType,
+        document.title ?? null,
+        document.summary ?? null,
+        JSON.stringify(document.tags),
+        document.confidence ?? null,
+        JSON.stringify(document.sourceRefs ?? []),
+        document.tokenEstimate,
+        document.sha256,
+        document.revision,
+        document.createdAt,
+        document.updatedAt
+    ]);
+    return result.rows[0] ? mapMemoryDocument(result.rows[0]) : document;
+}
+export async function deleteMemoryDocument(client, scope, path, options = {}) {
+    const normalizedPath = normalizeMemoryPathForPostgres(path);
+    const key = postgresMemoryKey(scope, normalizedPath);
+    if (options.embeddingsAvailable !== false) {
+        await client.query("delete from openeve_memory_embeddings where scope_key=$1", [key]);
+    }
+    await client.query("delete from openeve_memory_documents where scope_key=$1", [key]);
+}
+export async function listMemoryDocumentsNeedingEmbeddings(client, scope, options) {
+    const scopeFilter = postgresMemoryScopeWhere(scope, 4);
+    const result = await client.query(`select d.*
+       from openeve_memory_documents d
+       left join openeve_memory_embeddings e
+         on e.scope_key=d.scope_key and e.model=$2
+       where d.scope->>'agentScope'=$1
+         and (e.id is null or e.stale=true or e.content_sha256 <> d.sha256)
+         ${scopeFilter.sql.replace(/scope/g, "d.scope")}
+       order by d.updated_at desc
+       limit $3`, [scope.agentScope, options.model, positiveLimit(options.limit, 25), ...scopeFilter.params]);
+    return result.rows.map(mapMemoryDocument);
+}
+export async function upsertMemoryEmbedding(client, input) {
+    const now = timestamp();
+    const result = await client.query(`insert into openeve_memory_embeddings
+       (id, scope_key, scope, path, model, dimensions, content_sha256, embedding, stale, metadata, created_at, updated_at)
+       values ($1,$2,$3::jsonb,$4,$5,$6,$7,$8::vector,$9,$10::jsonb,$11,$12)
+       on conflict (scope_key, model) do update set
+         scope=$3::jsonb,
+         path=$4,
+         dimensions=$6,
+         content_sha256=$7,
+         embedding=$8::vector,
+         stale=$9,
+         metadata=$10::jsonb,
+         updated_at=$12
+       returning *`, [
+        randomUUID(),
+        postgresMemoryKey(input.scope, input.path),
+        JSON.stringify(input.scope),
+        normalizeMemoryPathForPostgres(input.path),
+        input.model,
+        input.dimensions,
+        input.contentSha256,
+        postgresVectorLiteral(input.embedding),
+        input.stale,
+        JSON.stringify(input.metadata),
+        now,
+        now
+    ]);
+    return result.rows[0] ? mapMemoryEmbedding(result.rows[0]) : {
+        ...input,
+        id: randomUUID(),
+        createdAt: now,
+        updatedAt: now
+    };
+}
+export async function searchMemoryEmbeddings(client, scope, input, embedding, options) {
+    const pathPrefix = normalizeMemoryPathForPostgres(input.pathPrefix ?? "");
+    const scopeFilter = postgresMemoryScopeWhere(scope, 6);
+    const result = await client.query(`select d.*, (e.embedding <=> $2::vector) as distance
+       from openeve_memory_embeddings e
+       join openeve_memory_documents d
+         on d.scope_key=e.scope_key
+       where e.model=$3
+         and e.stale=false
+         and d.scope->>'agentScope'=$1
+         and ($4 = '' or d.normalized_path like ($4 || '%'))
+         ${scopeFilter.sql.replace(/scope/g, "d.scope")}
+       order by distance asc, d.updated_at desc
+       limit $5`, [
+        scope.agentScope,
+        postgresVectorLiteral(embedding),
+        options.model,
+        pathPrefix,
+        positiveLimit(options.limit ?? input.limit, 8),
+        ...scopeFilter.params
+    ]);
+    return result.rows.map((row) => memoryEmbeddingSearchResultFromPostgres(row, input.query));
+}
+export async function markMemoryEmbeddingsStale(client, scope, path, currentSha256) {
+    await client.query(`update openeve_memory_embeddings
+       set stale=true, updated_at=$4
+       where scope_key=$1 and path=$2 and content_sha256 <> $3`, [postgresMemoryKey(scope, path), normalizeMemoryPathForPostgres(path), currentSha256, timestamp()]);
+}
+//# sourceMappingURL=adapter-memory.js.map

package/dist/adapter-memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"adapter-memory.js","sourceRoot":"","sources":["../src/adapter-memory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAIzC,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,uCAAuC,EACvC,8BAA8B,EAE/B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,6BAA6B,EAC7B,8BAA8B,EAC9B,aAAa,EACb,iBAAiB,EACjB,wBAAwB,EACxB,qBAAqB,EACrB,SAAS,EACV,MAAM,kBAAkB,CAAC;AAE1B,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,MAAmB,EAAE,KAAkB,EAAE,QAAyB,EAAE;IAC5G,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B,+FAA+F,EAC/F,CAAC,KAAK,CAAC,UAAU,CAAC,CACnB,CAAC;IACF,MAAM,MAAM,GAAG,8BAA8B,CAAC,KAAK,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;IACtE,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACnG,OAAO,MAAM,CAAC,IAAI;SACf,GAAG,CAAC,iBAAiB,CAAC;SACtB,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,6BAA6B,CAAC,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;SAC1E,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,MAAM,IAAI,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;SACjE,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;AACrB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,MAAmB,EAAE,KAAkB,EAAE,KAAwB;IAC3G,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAC/C,MAAM,UAAU,GAAG,8BAA8B,CAAC,KAAK,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;IAC1E,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAC5C,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,IAAI,QAAQ,CAAC;IACpC,MAAM,WAAW,GAAG,wBAAwB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IACvD,MAAM,MAAM,GAAc,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;IAC9F,MAAM,KAAK,GAAG,IAAI,KAAK,OAAO,CAAC;IAC/B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B,KAAK;QACH,CAAC,CAAC;;;;;eAKO,WAAW,CAAC,GAAG;;oBAEV;QACd,CAAC,CAAC;;;;;;;;;;;;eAYO,WAAW,CAAC,GAAG;;;;;oBAKV,EAChB,MAAM,CACP,CAAC;IACF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,8BAA8B,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC;AACpF,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,MAAmB,EAAE,KAAkB,EAAE,IAAY;IAC3F,MAAM,cAAc,GAAG,8BAA8B,CAAC,IAAI,CAAC,CAAC;IAC5D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B,2DAA2D,EAC3D,CAAC,iBAAiB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC,CAC3C,CAAC;IACF,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC3B,OAAO,GAAG,CAAC,CAAC,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAClD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,MAAmB,EAAE,QAAwB;IACtF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B;;;;;;;;;;;;;;;;;;mBAkBe,EACf;QACE,UAAU,EAAE;QACZ,iBAAiB,CAAC,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC;QAChD,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC;QAC9B,QAAQ,CAAC,IAAI;QACb,8BAA8B,CAAC,QAAQ,CAAC,IAAI,CAAC;QAC7C,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC;QAC7B,QAAQ,CAAC,WAAW;QACpB,QAAQ,CAAC,KAAK,IAAI,IAAI;QACtB,QAAQ,CAAC,OAAO,IAAI,IAAI;QACxB,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC;QAC7B,QAAQ,CAAC,UAAU,IAAI,IAAI;QAC3B,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,UAAU,IAAI,EAAE,CAAC;QACzC,QAAQ,CAAC,aAAa;QACtB,QAAQ,CAAC,MAAM;QACf,QAAQ,CAAC,QAAQ;QACjB,QAAQ,CAAC,SAAS;QAClB,QAAQ,CAAC,SAAS;KACnB,CACF,CAAC;IACF,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;AACvE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,MAAmB,EACnB,KAAkB,EAClB,IAAY,EACZ,UAA6C,EAAE;IAE/C,MAAM,cAAc,GAAG,8BAA8B,CAAC,IAAI,CAAC,CAAC;IAC5D,MAAM,GAAG,GAAG,iBAAiB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;IACrD,IAAI,OAAO,CAAC,mBAAmB,KAAK,KAAK,EAAE,CAAC;QAC1C,MAAM,MAAM,CAAC,KAAK,CAAC,0DAA0D,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACxF,CAAC;IACD,MAAM,MAAM,CAAC,KAAK,CAAC,yDAAyD,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;AACvF,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oCAAoC,CAAC,MAAmB,EAAE,KAAkB,EAAE,OAA0C;IAC5I,MAAM,WAAW,GAAG,wBAAwB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IACvD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B;;;;;;WAMO,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,SAAS,CAAC;;gBAEvC,EACZ,CAAC,KAAK,CAAC,UAAU,EAAE,OAAO,CAAC,KAAK,EAAE,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,GAAG,WAAW,CAAC,MAAM,CAAC,CAC3F,CAAC;IACF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;AAC5C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,MAAmB,EAAE,KAAoE;IACnI,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;IACxB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B;;;;;;;;;;;;mBAYe,EACf;QACE,UAAU,EAAE;QACZ,iBAAiB,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC;QAC1C,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC;QAC3B,8BAA8B,CAAC,KAAK,CAAC,IAAI,CAAC;QAC1C,KAAK,CAAC,KAAK;QACX,KAAK,CAAC,UAAU;QAChB,KAAK,CAAC,aAAa;QACnB,qBAAqB,CAAC,KAAK,CAAC,SAAS,CAAC;QACtC,KAAK,CAAC,KAAK;QACX,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC9B,GAAG;QACH,GAAG;KACJ,CACF,CAAC;IACF,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3D,GAAG,KAAK;QACR,EAAE,EAAE,UAAU,EAAE;QAChB,SAAS,EAAE,GAAG;QACd,SAAS,EAAE,GAAG;KACf,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,MAAmB,EACnB,KAAkB,EAClB,KAAwB,EACxB,SAAmB,EACnB,OAA0C;IAE1C,MAAM,UAAU,GAAG,8BAA8B,CAAC,KAAK,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;IAC1E,MAAM,WAAW,GAAG,wBAAwB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IACvD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAC/B;;;;;;;;WAQO,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,SAAS,CAAC;;gBAEvC,EACZ;QACE,KAAK,CAAC,UAAU;QAChB,qBAAqB,CAAC,SAAS,CAAC;QAChC,OAAO,CAAC,KAAK;QACb,UAAU;QACV,aAAa,CAAC,OAAO,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;QAC9C,GAAG,WAAW,CAAC,MAAM;KACtB,CACF,CAAC;IACF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,uCAAuC,CAAC,GAAG,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;AAC7F,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAAC,MAAmB,EAAE,KAAkB,EAAE,IAAY,EAAE,aAAqB;IAC1H,MAAM,MAAM,CAAC,KAAK,CAChB;;+DAE2D,EAC3D,CAAC,iBAAiB,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,8BAA8B,CAAC,IAAI,CAAC,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC,CACnG,CAAC;AACJ,CAAC"}

package/dist/adapter-misc.d.ts

@@ -0,0 +1,61 @@
+import type { JsonObject } from "@assemblyline-agents/core";
+import type { FileIndexRecord, ScheduleRecord, ScheduleRunRecord } from "@assemblyline-agents/runtime";
+import type { QueryClient } from "./client-types.js";
+export interface PostgresQueueRecord {
+    id: string;
+    runId: string;
+    queue: string;
+    status: string;
+    availableAt: string;
+    idempotencyKey: string;
+    createdAt: string;
+    updatedAt: string;
+}
+export interface PostgresScheduleRunRecord {
+    id: string;
+    scheduleId: string;
+    runId?: string;
+    status: string;
+    runAt: string;
+    idempotencyKey: string;
+    createdAt: string;
+    updatedAt: string;
+}
+export declare function enqueueRun(client: QueryClient, runId: string, options?: {
+    queue?: string;
+    availableAt?: string;
+    idempotencyKey?: string;
+}): Promise<PostgresQueueRecord>;
+export declare function enqueueScheduleRun(client: QueryClient, input: {
+    scheduleId: string;
+    runId?: string;
+    runAt: string;
+    idempotencyKey: string;
+    status?: string;
+}): Promise<PostgresScheduleRunRecord>;
+export declare function reserveIdempotencyKey(client: QueryClient, scope: string, key: string, ownerId?: string): Promise<boolean>;
+export declare function registerSchedule(client: QueryClient, input: {
+    id: string;
+    name: string;
+    cron: string;
+    timezone?: string;
+    metadata?: JsonObject;
+}): Promise<ScheduleRecord>;
+export declare function recordScheduleRun(client: QueryClient, input: {
+    scheduleId: string;
+    runId?: string;
+    runAt: string;
+    idempotencyKey: string;
+    status?: string;
+}): Promise<ScheduleRunRecord>;
+export declare function recordFileIndex(client: QueryClient, input: {
+    runId?: string;
+    path: string;
+    kind: FileIndexRecord["kind"];
+    blobKey: string;
+    sha256: string;
+    sizeBytes: number;
+    metadata?: JsonObject;
+}): Promise<FileIndexRecord>;
+export declare function listFileIndexes(client: QueryClient, runId?: string): Promise<FileIndexRecord[]>;
+//# sourceMappingURL=adapter-misc.d.ts.map

package/dist/adapter-misc.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"adapter-misc.d.ts","sourceRoot":"","sources":["../src/adapter-misc.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,KAAK,EACV,eAAe,EACf,cAAc,EACd,iBAAiB,EAClB,MAAM,8BAA8B,CAAC;AACtC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAIrD,MAAM,WAAW,mBAAmB;IAClC,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,yBAAyB;IACxC,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,wBAAsB,UAAU,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE;IAAE,KAAK,CAAC,EAAE,MAAM,CAAC;IAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAAC,cAAc,CAAC,EAAE,MAAM,CAAA;CAAO,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAoBlL;AAED,wBAAsB,kBAAkB,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE;IAAE,UAAU,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,cAAc,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAyBvM;AAED,wBAAsB,qBAAqB,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAS/H;AAED,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE;IACjE,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,UAAU,CAAC;CACvB,GAAG,OAAO,CAAC,cAAc,CAAC,CAiB1B;AAED,wBAAsB,iBAAiB,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE;IAClE,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,cAAc,EAAE,MAAM,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAS7B;AAED,wBAAsB,eAAe,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE;IAChE,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,UAAU,CAAC;CACvB,GAAG,OAAO,CAAC,eAAe,CAAC,CAuC3B;AAED,wBAAsB,eAAe,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC,CAarG"}