@assemblyline-agents/postgres 0.1.0

package/dist/presets.d.ts

@@ -0,0 +1,19 @@
+import { type AdapterDefinition, type JsonObject } from "@assemblyline-agents/core";
+export interface PostgresAdapter extends AdapterDefinition {
+    kind: "postgres";
+}
+export type PostgresProvider = "neon" | "railway" | "supabase" | "local" | "custom";
+export interface RailwayPostgresOptions extends JsonObject {
+    /** Railway database service referenced by the deployed OpenEve service. */
+    databaseService?: string;
+    /** Create the Railway Postgres service when it does not exist. Defaults to true. */
+    provision?: boolean;
+    /** Railway's generated database certificate is self-signed. Defaults to false. */
+    sslRejectUnauthorized?: boolean;
+}
+export declare function postgresAdapter(options?: JsonObject): PostgresAdapter;
+export declare function neonPostgres(options?: JsonObject): PostgresAdapter;
+export declare function railwayPostgres(options?: RailwayPostgresOptions): PostgresAdapter;
+export declare function supabasePostgres(options?: JsonObject): PostgresAdapter;
+export declare function localPostgres(options?: JsonObject): PostgresAdapter;
+//# sourceMappingURL=presets.d.ts.map

package/dist/presets.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"presets.d.ts","sourceRoot":"","sources":["../src/presets.ts"],"names":[],"mappings":"AAAA,OAAO,EAAW,KAAK,iBAAiB,EAAE,KAAK,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAG7F,MAAM,WAAW,eAAgB,SAAQ,iBAAiB;IACxD,IAAI,EAAE,UAAU,CAAC;CAClB;AAED,MAAM,MAAM,gBAAgB,GAAG,MAAM,GAAG,SAAS,GAAG,UAAU,GAAG,OAAO,GAAG,QAAQ,CAAC;AAEpF,MAAM,WAAW,sBAAuB,SAAQ,UAAU;IACxD,2EAA2E;IAC3E,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,oFAAoF;IACpF,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,kFAAkF;IAClF,qBAAqB,CAAC,EAAE,OAAO,CAAC;CACjC;AAED,wBAAgB,eAAe,CAAC,OAAO,GAAE,UAAe,GAAG,eAAe,CAMzE;AAED,wBAAgB,YAAY,CAAC,OAAO,GAAE,UAAe,GAAG,eAAe,CAEtE;AAED,wBAAgB,eAAe,CAAC,OAAO,GAAE,sBAA2B,GAAG,eAAe,CASrF;AAED,wBAAgB,gBAAgB,CAAC,OAAO,GAAE,UAAe,GAAG,eAAe,CAE1E;AAED,wBAAgB,aAAa,CAAC,OAAO,GAAE,UAAe,GAAG,eAAe,CAEvE"}

package/dist/presets.js

@@ -0,0 +1,31 @@
+import { adapter } from "@assemblyline-agents/core";
+import { POSTGRES_STATE_PROVIDER_METADATA as postgresStateProviderMetadata } from "./provider.js";
+export function postgresAdapter(options = {}) {
+    const metadata = postgresStateProviderMetadata;
+    const definition = { ...adapter("postgres", options), kind: "postgres" };
+    if (metadata.requiredEnv)
+        definition.requiredEnv = metadata.requiredEnv;
+    if (metadata.capabilities)
+        definition.capabilities = metadata.capabilities;
+    return definition;
+}
+export function neonPostgres(options = {}) {
+    return postgresAdapter({ provider: "neon", connectionEnv: "DATABASE_URL", ...options });
+}
+export function railwayPostgres(options = {}) {
+    return postgresAdapter({
+        provider: "railway",
+        connectionEnv: "DATABASE_URL",
+        databaseService: "Postgres",
+        provision: true,
+        sslRejectUnauthorized: false,
+        ...options
+    });
+}
+export function supabasePostgres(options = {}) {
+    return postgresAdapter({ provider: "supabase", connectionEnv: "DATABASE_URL", ...options });
+}
+export function localPostgres(options = {}) {
+    return postgresAdapter({ provider: "local", connectionEnv: "DATABASE_URL", ssl: false, ...options });
+}
+//# sourceMappingURL=presets.js.map

package/dist/presets.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"presets.js","sourceRoot":"","sources":["../src/presets.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAA2C,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,gCAAgC,IAAI,6BAA6B,EAAE,MAAM,eAAe,CAAC;AAiBlG,MAAM,UAAU,eAAe,CAAC,UAAsB,EAAE;IACtD,MAAM,QAAQ,GAAG,6BAA6B,CAAC;IAC/C,MAAM,UAAU,GAAoB,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;IAC1F,IAAI,QAAQ,CAAC,WAAW;QAAE,UAAU,CAAC,WAAW,GAAG,QAAQ,CAAC,WAAW,CAAC;IACxE,IAAI,QAAQ,CAAC,YAAY;QAAE,UAAU,CAAC,YAAY,GAAG,QAAQ,CAAC,YAAY,CAAC;IAC3E,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,UAAsB,EAAE;IACnD,OAAO,eAAe,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,cAAc,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;AAC1F,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,UAAkC,EAAE;IAClE,OAAO,eAAe,CAAC;QACrB,QAAQ,EAAE,SAAS;QACnB,aAAa,EAAE,cAAc;QAC7B,eAAe,EAAE,UAAU;QAC3B,SAAS,EAAE,IAAI;QACf,qBAAqB,EAAE,KAAK;QAC5B,GAAG,OAAO;KACX,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,UAAsB,EAAE;IACvD,OAAO,eAAe,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,aAAa,EAAE,cAAc,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;AAC9F,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,UAAsB,EAAE;IACpD,OAAO,eAAe,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;AACvG,CAAC"}

package/dist/provider.d.ts

@@ -0,0 +1,12 @@
+import type { AdapterProviderMetadata, JsonObject, ProviderModule } from "@assemblyline-agents/core";
+import { type QueryClient } from "./index.js";
+/** Metadata owned by this package; the core registry mirrors it as an offline default. */
+export declare const POSTGRES_STATE_PROVIDER_METADATA: AdapterProviderMetadata;
+/**
+ * Creates a `pg` Pool from environment variables. `OPENEVE_POSTGRES_CONNECTION_ENV`
+ * overrides which variable holds the connection string (default `DATABASE_URL`).
+ */
+export declare function createPostgresPoolFromEnv(env: Record<string, string | undefined>, options?: JsonObject): Promise<QueryClient>;
+export declare const openeveProvider: ProviderModule;
+export declare function optionalMigrationsFromOptions(options: JsonObject): boolean | string[] | undefined;
+//# sourceMappingURL=provider.d.ts.map

package/dist/provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.d.ts","sourceRoot":"","sources":["../src/provider.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,UAAU,EAA0B,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC7H,OAAO,EAAwB,KAAK,WAAW,EAAE,MAAM,YAAY,CAAC;AAEpE,0FAA0F;AAC1F,eAAO,MAAM,gCAAgC,EAAE,uBAc9C,CAAC;AAIF;;;GAGG;AACH,wBAAsB,yBAAyB,CAC7C,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,EACvC,OAAO,GAAE,UAAe,GACvB,OAAO,CAAC,WAAW,CAAC,CA2BtB;AAED,eAAO,MAAM,eAAe,EAAE,cAgB7B,CAAC;AAEF,wBAAgB,6BAA6B,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,GAAG,MAAM,EAAE,GAAG,SAAS,CAWjG"}

package/dist/provider.js

@@ -0,0 +1,86 @@
+import { PostgresStateAdapter } from "./index.js";
+/** Metadata owned by this package; the core registry mirrors it as an offline default. */
+export const POSTGRES_STATE_PROVIDER_METADATA = {
+    kind: "postgres",
+    role: "state",
+    packageName: "@assemblyline-agents/postgres",
+    requiredEnv: ["DATABASE_URL"],
+    capabilities: ["durable-state", "migrations"],
+    setup: [
+        {
+            kind: "migration",
+            name: "postgres-schema",
+            required: true,
+            message: "Apply OpenEve state migrations to Postgres before production runs."
+        }
+    ]
+};
+/**
+ * Creates a `pg` Pool from environment variables. `OPENEVE_POSTGRES_CONNECTION_ENV`
+ * overrides which variable holds the connection string (default `DATABASE_URL`).
+ */
+export async function createPostgresPoolFromEnv(env, options = {}) {
+    const configuredConnectionEnv = typeof options.connectionEnv === "string" ? options.connectionEnv : undefined;
+    const connectionEnv = env.OPENEVE_POSTGRES_CONNECTION_ENV ?? configuredConnectionEnv ?? "DATABASE_URL";
+    const connectionString = env[connectionEnv];
+    if (!connectionString)
+        throw new Error(`Postgres state requires ${connectionEnv}.`);
+    const pg = await dynamicImport("pg");
+    const Pool = pg.Pool ?? pg.default?.Pool;
+    if (!Pool)
+        throw new Error("The pg package did not expose a Pool constructor.");
+    const sslDisabled = options.ssl === false || /(?:[?&]sslmode=disable)(?:&|$)/.test(connectionString);
+    const configuredRejectUnauthorized = typeof options.sslRejectUnauthorized === "boolean"
+        ? options.sslRejectUnauthorized
+        : undefined;
+    const rejectUnauthorized = env.OPENEVE_POSTGRES_SSL_REJECT_UNAUTHORIZED !== undefined
+        ? env.OPENEVE_POSTGRES_SSL_REJECT_UNAUTHORIZED !== "false"
+        : configuredRejectUnauthorized ?? true;
+    return new Pool({
+        connectionString,
+        ssl: sslDisabled
+            ? false
+            : {
+                // Secure by default: verify the server certificate so a MITM cannot
+                // read runtime state (including stored OAuth/connection tokens).
+                // Provider presets may opt out for generated/self-signed certificates;
+                // OPENEVE_POSTGRES_SSL_REJECT_UNAUTHORIZED remains the final override.
+                rejectUnauthorized
+            }
+    });
+}
+export const openeveProvider = {
+    providers: [
+        {
+            metadata: POSTGRES_STATE_PROVIDER_METADATA,
+            async create(ctx) {
+                // The provider constructs the pool, so the adapter owns it and closes it on shutdown.
+                const optionalMigrations = optionalMigrationsFromOptions(ctx.options);
+                const state = new PostgresStateAdapter(await createPostgresPoolFromEnv(ctx.env, ctx.options), {
+                    ownsClient: true,
+                    ...(optionalMigrations !== undefined ? { optionalMigrations } : {})
+                });
+                if (ctx.env.OPENEVE_AUTO_MIGRATE !== "false")
+                    await state.migrate();
+                return state;
+            }
+        }
+    ]
+};
+export function optionalMigrationsFromOptions(options) {
+    const configured = options.optionalMigrations;
+    if (configured === undefined || configured === false)
+        return undefined;
+    if (configured === true)
+        return true;
+    if (Array.isArray(configured)
+        && configured.every((migration) => typeof migration === "string" && migration.trim().length > 0)) {
+        return [...new Set(configured.map((migration) => migration.trim()))];
+    }
+    throw new Error("Postgres optionalMigrations must be true or an array of migration IDs.");
+}
+async function dynamicImport(specifier) {
+    const load = new Function("specifier", "return import(specifier)");
+    return load(specifier);
+}
+//# sourceMappingURL=provider.js.map

package/dist/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../src/provider.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAoB,MAAM,YAAY,CAAC;AAEpE,0FAA0F;AAC1F,MAAM,CAAC,MAAM,gCAAgC,GAA4B;IACvE,IAAI,EAAE,UAAU;IAChB,IAAI,EAAE,OAAO;IACb,WAAW,EAAE,+BAA+B;IAC5C,WAAW,EAAE,CAAC,cAAc,CAAC;IAC7B,YAAY,EAAE,CAAC,eAAe,EAAE,YAAY,CAAC;IAC7C,KAAK,EAAE;QACL;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,iBAAiB;YACvB,QAAQ,EAAE,IAAI;YACd,OAAO,EAAE,oEAAoE;SAC9E;KACF;CACF,CAAC;AAIF;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,GAAuC,EACvC,UAAsB,EAAE;IAExB,MAAM,uBAAuB,GAAG,OAAO,OAAO,CAAC,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9G,MAAM,aAAa,GAAG,GAAG,CAAC,+BAA+B,IAAI,uBAAuB,IAAI,cAAc,CAAC;IACvG,MAAM,gBAAgB,GAAG,GAAG,CAAC,aAAa,CAAC,CAAC;IAC5C,IAAI,CAAC,gBAAgB;QAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,aAAa,GAAG,CAAC,CAAC;IACpF,MAAM,EAAE,GAAG,MAAM,aAAa,CAAmE,IAAI,CAAC,CAAC;IACvG,MAAM,IAAI,GAAG,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC;IACzC,IAAI,CAAC,IAAI;QAAE,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;IAChF,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,KAAK,KAAK,IAAI,gCAAgC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACrG,MAAM,4BAA4B,GAAG,OAAO,OAAO,CAAC,qBAAqB,KAAK,SAAS;QACrF,CAAC,CAAC,OAAO,CAAC,qBAAqB;QAC/B,CAAC,CAAC,SAAS,CAAC;IACd,MAAM,kBAAkB,GAAG,GAAG,CAAC,wCAAwC,KAAK,SAAS;QACnF,CAAC,CAAC,GAAG,CAAC,wCAAwC,KAAK,OAAO;QAC1D,CAAC,CAAC,4BAA4B,IAAI,IAAI,CAAC;IACzC,OAAO,IAAI,IAAI,CAAC;QACd,gBAAgB;QAChB,GAAG,EAAE,WAAW;YACd,CAAC,CAAC,KAAK;YACP,CAAC,CAAC;gBACE,oEAAoE;gBACpE,iEAAiE;gBACjE,uEAAuE;gBACvE,uEAAuE;gBACvE,kBAAkB;aACnB;KACN,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,MAAM,eAAe,GAAmB;IAC7C,SAAS,EAAE;QACT;YACE,QAAQ,EAAE,gCAAgC;YAC1C,KAAK,CAAC,MAAM,CAAC,GAA2B;gBACtC,sFAAsF;gBACtF,MAAM,kBAAkB,GAAG,6BAA6B,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACtE,MAAM,KAAK,GAAG,IAAI,oBAAoB,CAAC,MAAM,yBAAyB,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,EAAE;oBAC5F,UAAU,EAAE,IAAI;oBAChB,GAAG,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACpE,CAAC,CAAC;gBACH,IAAI,GAAG,CAAC,GAAG,CAAC,oBAAoB,KAAK,OAAO;oBAAE,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;gBACpE,OAAO,KAAK,CAAC;YACf,CAAC;SACF;KACF;CACF,CAAC;AAEF,MAAM,UAAU,6BAA6B,CAAC,OAAmB;IAC/D,MAAM,UAAU,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAC9C,IAAI,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,KAAK;QAAE,OAAO,SAAS,CAAC;IACvE,IAAI,UAAU,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IACrC,IACE,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC;WACtB,UAAU,CAAC,KAAK,CAAC,CAAC,SAAS,EAAuB,EAAE,CAAC,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,EACrH,CAAC;QACD,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACvE,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,wEAAwE,CAAC,CAAC;AAC5F,CAAC;AAED,KAAK,UAAU,aAAa,CAAI,SAAiB;IAC/C,MAAM,IAAI,GAAG,IAAI,QAAQ,CAAC,WAAW,EAAE,0BAA0B,CAAsC,CAAC;IACxG,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC;AACzB,CAAC"}

package/dist/row-mappers.d.ts

@@ -0,0 +1,24 @@
+import type { MemorySearchInput } from "@assemblyline-agents/core";
+import type { CheckpointRecord, ConnectionAuthorizationSessionRecord, ConversationRecord, DeliveryObligation, DynamicScheduleRecord, FileIndexRecord, MemoryDocument, MemoryEmbeddingRecord, MemorySearchResult, MessageRecord, RunEvent, RunRecord, SandboxSessionRecord, SandboxSyncJobRecord, StoredConnectionDefinition, StoredConnectionGrant, StoredSkill, ToolCallRecord, UsageRecord } from "@assemblyline-agents/runtime";
+export type Row = Record<string, unknown>;
+export declare function mapRun(row: Row): RunRecord;
+export declare function mapEvent(row: Row): RunEvent;
+export declare function mapToolCall(row: Row): ToolCallRecord;
+export declare function mapCheckpoint(row: Row): CheckpointRecord;
+export declare function mapDelivery(row: Row): DeliveryObligation;
+export declare function mapUsage(row: Row): UsageRecord;
+export declare function mapSandboxSession(row: Row): SandboxSessionRecord;
+export declare function mapSandboxSyncJob(row: Row): SandboxSyncJobRecord;
+export declare function mapMemoryDocument(row: Row): MemoryDocument;
+export declare function memorySearchResultFromPostgres(row: Row, query: string, mode: MemorySearchInput["mode"] | undefined): MemorySearchResult;
+export declare function memoryEmbeddingSearchResultFromPostgres(row: Row, query: string): MemorySearchResult;
+export declare function mapMemoryEmbedding(row: Row): MemoryEmbeddingRecord;
+export declare function mapFileIndexRecord(row: Row): FileIndexRecord;
+export declare function mapConversation(row: Row): ConversationRecord;
+export declare function mapMessage(row: Row): MessageRecord;
+export declare function mapConnectionGrant(row: Row): StoredConnectionGrant;
+export declare function mapConnectionAuthorizationSession(row: Row): ConnectionAuthorizationSessionRecord;
+export declare function mapSkill(row: Row): StoredSkill;
+export declare function mapDynamicSchedule(row: Row): DynamicScheduleRecord;
+export declare function mapConnectionDefinition(row: Row): StoredConnectionDefinition;
+//# sourceMappingURL=row-mappers.d.ts.map

package/dist/row-mappers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"row-mappers.d.ts","sourceRoot":"","sources":["../src/row-mappers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AACnE,OAAO,KAAK,EACV,gBAAgB,EAChB,oCAAoC,EACpC,kBAAkB,EAClB,kBAAkB,EAClB,qBAAqB,EACrB,eAAe,EACf,cAAc,EACd,qBAAqB,EAErB,kBAAkB,EAClB,aAAa,EACb,QAAQ,EACR,SAAS,EACT,oBAAoB,EACpB,oBAAoB,EACpB,0BAA0B,EAC1B,qBAAqB,EACrB,WAAW,EACX,cAAc,EACd,WAAW,EACZ,MAAM,8BAA8B,CAAC;AAatC,MAAM,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAE1C,wBAAgB,MAAM,CAAC,GAAG,EAAE,GAAG,GAAG,SAAS,CAiB1C;AAED,wBAAgB,QAAQ,CAAC,GAAG,EAAE,GAAG,GAAG,QAAQ,CAS3C;AAED,wBAAgB,WAAW,CAAC,GAAG,EAAE,GAAG,GAAG,cAAc,CAepD;AAED,wBAAgB,aAAa,CAAC,GAAG,EAAE,GAAG,GAAG,gBAAgB,CAQxD;AAED,wBAAgB,WAAW,CAAC,GAAG,EAAE,GAAG,GAAG,kBAAkB,CAqBxD;AAED,wBAAgB,QAAQ,CAAC,GAAG,EAAE,GAAG,GAAG,WAAW,CAwC9C;AAkBD,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,GAAG,GAAG,oBAAoB,CAkBhE;AAED,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,GAAG,GAAG,oBAAoB,CAuBhE;AAED,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,GAAG,GAAG,cAAc,CAmB1D;AAED,wBAAgB,8BAA8B,CAAC,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,CAAC,MAAM,CAAC,GAAG,SAAS,GAAG,kBAAkB,CAevI;AAED,wBAAgB,uCAAuC,CAAC,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,kBAAkB,CAcnG;AAED,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,GAAG,GAAG,qBAAqB,CAclE;AAED,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,GAAG,GAAG,eAAe,CAa5D;AAED,wBAAgB,eAAe,CAAC,GAAG,EAAE,GAAG,GAAG,kBAAkB,CAc5D;AAED,wBAAgB,UAAU,CAAC,GAAG,EAAE,GAAG,GAAG,aAAa,CAUlD;AAED,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,GAAG,GAAG,qBAAqB,CAelE;AAED,wBAAgB,iCAAiC,CAAC,GAAG,EAAE,GAAG,GAAG,oCAAoC,CAsBhG;AAED,wBAAgB,QAAQ,CAAC,GAAG,EAAE,GAAG,GAAG,WAAW,CAW9C;AAED,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,GAAG,GAAG,qBAAqB,CA8BlE;AAED,wBAAgB,uBAAuB,CAAC,GAAG,EAAE,GAAG,GAAG,0BAA0B,CAa5E"}

package/dist/row-mappers.js

@@ -0,0 +1,450 @@
+import { microsToUsd } from "@assemblyline-agents/runtime";
+import { memoryBodyTextForPostgres, memorySnippetForPostgres, normalizeJsonArray, normalizeJsonObject, normalizeJsonValue, postgresMemoryRankReason, postgresVectorValue, toIso } from "./sql-helpers.js";
+export function mapRun(row) {
+    const run = {
+        id: String(row.id),
+        agentRevision: String(row.agent_revision),
+        status: String(row.status),
+        input: normalizeJsonObject(row.input),
+        createdAt: toIso(row.created_at),
+        updatedAt: toIso(row.updated_at)
+    };
+    if (row.context_bundle_id)
+        run.contextBundleId = String(row.context_bundle_id);
+    if (row.sandbox_id)
+        run.sandboxId = String(row.sandbox_id);
+    if (row.sandbox_snapshot_id)
+        run.sandboxSnapshotId = String(row.sandbox_snapshot_id);
+    if (row.memory_revision)
+        run.memoryRevision = String(row.memory_revision);
+    if (row.control_intent === "suspend" || row.control_intent === "cancel")
+        run.controlIntent = row.control_intent;
+    if (row.terminal_reason)
+        run.terminalReason = String(row.terminal_reason);
+    if (row.terminal_error)
+        run.terminalError = String(row.terminal_error);
+    return run;
+}
+export function mapEvent(row) {
+    return {
+        id: String(row.id),
+        runId: String(row.run_id),
+        sequence: Number(row.sequence),
+        type: String(row.type),
+        timestamp: toIso(row.created_at),
+        data: normalizeJsonObject(row.data)
+    };
+}
+export function mapToolCall(row) {
+    const record = {
+        id: String(row.id),
+        runId: String(row.run_id),
+        toolName: String(row.tool_name),
+        status: String(row.status),
+        input: row.input,
+        idempotencyKey: String(row.idempotency_key),
+        createdAt: toIso(row.created_at),
+        updatedAt: toIso(row.updated_at)
+    };
+    if (row.result !== null && row.result !== undefined)
+        record.result = row.result;
+    if (row.model_output !== null && row.model_output !== undefined)
+        record.modelOutput = row.model_output;
+    if (row.error)
+        record.error = String(row.error);
+    return record;
+}
+export function mapCheckpoint(row) {
+    return {
+        id: String(row.id),
+        runId: String(row.run_id),
+        name: String(row.name),
+        createdAt: toIso(row.created_at),
+        data: normalizeJsonObject(row.data)
+    };
+}
+export function mapDelivery(row) {
+    const createdAt = toIso(row.created_at);
+    const record = {
+        id: String(row.id),
+        runId: String(row.run_id),
+        channel: String(row.channel),
+        payload: normalizeJsonObject(row.payload),
+        idempotencyKey: String(row.idempotency_key),
+        status: String(row.status),
+        attemptCount: Number(row.attempt_count ?? 0),
+        maxAttempts: Number(row.max_attempts ?? 5),
+        nextAttemptAt: row.next_attempt_at ? toIso(row.next_attempt_at) : createdAt,
+        createdAt,
+        updatedAt: toIso(row.updated_at)
+    };
+    if (row.lease_token)
+        record.leaseToken = String(row.lease_token);
+    if (row.lease_expires_at)
+        record.leaseExpiresAt = toIso(row.lease_expires_at);
+    if (row.last_error)
+        record.lastError = String(row.last_error);
+    if (row.sent_at)
+        record.sentAt = toIso(row.sent_at);
+    if (row.failed_at)
+        record.failedAt = toIso(row.failed_at);
+    return record;
+}
+export function mapUsage(row) {
+    const tokenSource = actualTokenSource(row.token_source);
+    const costSource = actualCostSource(row.cost_source);
+    const costMicros = costSource === "unavailable" ? null : nullableString(row.cost_micros);
+    const upstreamCostMicros = costSource === "unavailable" ? null : nullableString(row.upstream_cost_micros);
+    const record = {
+        id: String(row.id),
+        idempotencyKey: String(row.idempotency_key ?? `legacy:${String(row.id)}`),
+        billingMode: String(row.billing_mode ?? "unknown"),
+        recordKind: String(row.record_kind ?? "transaction"),
+        inputTokens: tokenSource === "unavailable" ? 0 : Number(row.input_tokens ?? 0),
+        outputTokens: tokenSource === "unavailable" ? 0 : Number(row.output_tokens ?? 0),
+        cachedTokens: tokenSource === "unavailable" ? 0 : Number(row.cached_tokens ?? 0),
+        cacheWriteTokens: tokenSource === "unavailable" ? 0 : Number(row.cache_write_tokens ?? 0),
+        reasoningTokens: tokenSource === "unavailable" ? 0 : Number(row.reasoning_tokens ?? 0),
+        costMicros,
+        upstreamCostMicros,
+        costUsd: costMicros === null ? null : microsToUsd(costMicros),
+        upstreamCostUsd: upstreamCostMicros === null ? null : microsToUsd(upstreamCostMicros),
+        currency: String(row.currency ?? "USD"),
+        tokenSource,
+        costSource,
+        providerReceipt: normalizeJsonObject(row.provider_receipt),
+        metadata: normalizeJsonObject(row.metadata),
+        occurredAt: toIso(row.occurred_at ?? row.created_at),
+        createdAt: toIso(row.created_at)
+    };
+    if (row.run_id)
+        record.runId = String(row.run_id);
+    if (row.parent_run_id)
+        record.parentRunId = String(row.parent_run_id);
+    if (row.agent_id)
+        record.agentId = String(row.agent_id);
+    if (row.agent_revision)
+        record.agentRevision = String(row.agent_revision);
+    if (row.subagent)
+        record.subagent = String(row.subagent);
+    if (row.turn_id)
+        record.turnId = String(row.turn_id);
+    if (row.iteration !== null && row.iteration !== undefined)
+        record.iteration = Number(row.iteration);
+    if (row.provider)
+        record.provider = String(row.provider);
+    if (row.provider_request_id)
+        record.providerRequestId = String(row.provider_request_id);
+    if (row.model)
+        record.model = String(row.model);
+    if (row.response_model)
+        record.responseModel = String(row.response_model);
+    if (row.receipt_sha256)
+        record.receiptSha256 = String(row.receipt_sha256);
+    return record;
+}
+function nullableString(value) {
+    return value === null || value === undefined ? null : String(value);
+}
+function actualTokenSource(value) {
+    return value === "provider_reported" || value === "provider_reconciled"
+        ? value
+        : "unavailable";
+}
+function actualCostSource(value) {
+    return value === "provider_reported" || value === "provider_reconciled"
+        ? value
+        : "unavailable";
+}
+export function mapSandboxSession(row) {
+    const record = {
+        id: String(row.id),
+        sessionKey: String(row.session_key),
+        provider: String(row.provider),
+        sandboxId: String(row.sandbox_id),
+        state: String(row.state),
+        dirty: Boolean(row.dirty),
+        lastSeenAt: toIso(row.last_seen_at),
+        manifest: normalizeJsonObject(row.manifest),
+        createdAt: toIso(row.created_at),
+        updatedAt: toIso(row.updated_at)
+    };
+    if (row.run_id)
+        record.runId = String(row.run_id);
+    if (row.last_synced_at)
+        record.lastSyncedAt = toIso(row.last_synced_at);
+    if (row.snapshot_id)
+        record.snapshotId = String(row.snapshot_id);
+    if (row.sync_job_id)
+        record.syncJobId = String(row.sync_job_id);
+    return record;
+}
+export function mapSandboxSyncJob(row) {
+    const record = {
+        id: String(row.id),
+        runId: String(row.run_id),
+        sandboxId: String(row.sandbox_id),
+        status: String(row.status),
+        idempotencyKey: String(row.idempotency_key),
+        reason: String(row.reason),
+        manifest: normalizeJsonObject(row.manifest),
+        attemptCount: Number(row.attempt_count ?? 0),
+        maxAttempts: Number(row.max_attempts ?? 5),
+        nextAttemptAt: toIso(row.next_attempt_at ?? row.created_at),
+        metadata: normalizeJsonObject(row.metadata),
+        createdAt: toIso(row.created_at),
+        updatedAt: toIso(row.updated_at)
+    };
+    if (row.error)
+        record.error = String(row.error);
+    if (row.lease_token)
+        record.leaseToken = String(row.lease_token);
+    if (row.lease_expires_at)
+        record.leaseExpiresAt = toIso(row.lease_expires_at);
+    if (row.completed_at)
+        record.completedAt = toIso(row.completed_at);
+    if (row.failed_at)
+        record.failedAt = toIso(row.failed_at);
+    if (row.last_error)
+        record.lastError = String(row.last_error);
+    return record;
+}
+export function mapMemoryDocument(row) {
+    const sourceRefs = normalizeJsonArray(row.source_refs).map(String);
+    const document = {
+        scope: normalizeJsonObject(row.scope),
+        path: String(row.path),
+        body: row.body,
+        contentType: String(row.content_type),
+        tags: normalizeJsonArray(row.tags).map(String),
+        tokenEstimate: Number(row.token_estimate ?? 0),
+        sha256: String(row.sha256),
+        revision: String(row.revision),
+        createdAt: toIso(row.created_at),
+        updatedAt: toIso(row.updated_at)
+    };
+    if (row.title)
+        document.title = String(row.title);
+    if (row.summary)
+        document.summary = String(row.summary);
+    if (row.confidence)
+        document.confidence = String(row.confidence);
+    if (sourceRefs.length > 0)
+        document.sourceRefs = sourceRefs;
+    return document;
+}
+export function memorySearchResultFromPostgres(row, query, mode) {
+    const document = mapMemoryDocument(row);
+    const rank = Number(row.rank ?? 0);
+    const rowSnippet = typeof row.snippet === "string" ? row.snippet.replace(/<\/?b>/gu, "") : "";
+    const body = memoryBodyTextForPostgres(document);
+    return {
+        path: document.path,
+        ...(document.title ? { title: document.title } : {}),
+        ...(document.summary ? { summary: document.summary } : {}),
+        snippet: rowSnippet || memorySnippetForPostgres(body, query),
+        revision: document.revision,
+        tags: document.tags,
+        updatedAt: document.updatedAt,
+        rankReason: postgresMemoryRankReason(document, query, rank, mode)
+    };
+}
+export function memoryEmbeddingSearchResultFromPostgres(row, query) {
+    const document = mapMemoryDocument(row);
+    const distance = Number(row.distance ?? 1);
+    const similarity = Math.max(0, 1 - distance);
+    return {
+        path: document.path,
+        ...(document.title ? { title: document.title } : {}),
+        ...(document.summary ? { summary: document.summary } : {}),
+        snippet: memorySnippetForPostgres(memoryBodyTextForPostgres(document), query),
+        revision: document.revision,
+        tags: document.tags,
+        updatedAt: document.updatedAt,
+        rankReason: `semantic similarity ${similarity.toFixed(3)}`
+    };
+}
+export function mapMemoryEmbedding(row) {
+    return {
+        id: String(row.id),
+        scope: normalizeJsonObject(row.scope),
+        path: String(row.path),
+        model: String(row.model),
+        dimensions: Number(row.dimensions ?? 0),
+        contentSha256: String(row.content_sha256),
+        embedding: postgresVectorValue(row.embedding),
+        stale: row.stale === true || row.stale === "true",
+        metadata: normalizeJsonObject(row.metadata),
+        createdAt: toIso(row.created_at),
+        updatedAt: toIso(row.updated_at)
+    };
+}
+export function mapFileIndexRecord(row) {
+    const record = {
+        id: String(row.id),
+        path: String(row.path),
+        kind: String(row.kind),
+        blobKey: String(row.blob_key),
+        sha256: String(row.sha256),
+        sizeBytes: Number(row.size_bytes ?? 0),
+        metadata: normalizeJsonObject(row.metadata),
+        createdAt: toIso(row.created_at)
+    };
+    if (row.run_id)
+        record.runId = String(row.run_id);
+    return record;
+}
+export function mapConversation(row) {
+    const record = {
+        id: String(row.id),
+        agentRevision: String(row.agent_revision ?? row.agent_id ?? ""),
+        channel: String(row.channel),
+        metadata: normalizeJsonObject(row.metadata),
+        createdAt: toIso(row.created_at),
+        updatedAt: toIso(row.updated_at)
+    };
+    if (row.agent_scope)
+        record.agentScope = String(row.agent_scope);
+    if (row.subject)
+        record.subject = String(row.subject);
+    if (row.title)
+        record.title = String(row.title);
+    if (row.archived_at)
+        record.archivedAt = toIso(row.archived_at);
+    return record;
+}
+export function mapMessage(row) {
+    const record = {
+        id: String(row.id),
+        conversationId: String(row.conversation_id),
+        role: String(row.role),
+        content: normalizeJsonObject(row.content),
+        createdAt: toIso(row.created_at)
+    };
+    if (row.run_id)
+        record.runId = String(row.run_id);
+    return record;
+}
+export function mapConnectionGrant(row) {
+    const grant = {
+        connectionName: String(row.connection_name),
+        principalKey: String(row.principal_key),
+        principal: normalizeJsonObject(row.principal),
+        token: String(row.token),
+        createdAt: toIso(row.created_at),
+        updatedAt: toIso(row.updated_at)
+    };
+    if (row.refresh_token)
+        grant.refreshToken = String(row.refresh_token);
+    if (row.token_type)
+        grant.tokenType = String(row.token_type);
+    if (Array.isArray(row.scopes))
+        grant.scopes = row.scopes.map(String);
+    if (row.expires_at !== null && row.expires_at !== undefined)
+        grant.expiresAt = Number(row.expires_at);
+    if (row.credential && typeof row.credential === "object")
+        grant.credential = normalizeJsonObject(row.credential);
+    return grant;
+}
+export function mapConnectionAuthorizationSession(row) {
+    const record = {
+        id: String(row.id),
+        state: String(row.state),
+        connectionName: String(row.connection_name),
+        principalKey: String(row.principal_key),
+        principal: normalizeJsonObject(row.principal),
+        session: normalizeJsonObject(row.session),
+        callbackUrl: String(row.callback_url),
+        challenge: normalizeJsonObject(row.challenge),
+        status: String(row.status),
+        expiresAt: Number(row.expires_at),
+        createdAt: toIso(row.created_at),
+        updatedAt: toIso(row.updated_at)
+    };
+    if (row.code_verifier)
+        record.codeVerifier = String(row.code_verifier);
+    if (row.resume !== null && row.resume !== undefined)
+        record.resume = normalizeJsonValue(row.resume);
+    if (row.run_id)
+        record.runId = String(row.run_id);
+    if (row.tool_call_id)
+        record.toolCallId = String(row.tool_call_id);
+    if (row.tool_name)
+        record.toolName = String(row.tool_name);
+    if (row.continuation !== null && row.continuation !== undefined)
+        record.continuation = normalizeJsonObject(row.continuation);
+    return record;
+}
+export function mapSkill(row) {
+    const record = {
+        name: String(row.name),
+        body: String(row.body),
+        origin: String(row.origin),
+        bodySha256: String(row.body_sha256),
+        updatedAt: toIso(row.updated_at)
+    };
+    if (row.seed_sha256)
+        record.seedSha256 = String(row.seed_sha256);
+    if (row.pending)
+        record.pending = Boolean(row.pending);
+    return record;
+}
+export function mapDynamicSchedule(row) {
+    const record = {
+        id: String(row.id),
+        scope: String(row.scope),
+        prompt: String(row.prompt),
+        channel: String(row.channel),
+        nextRunAt: toIso(row.next_run_at),
+        enabled: Boolean(row.enabled),
+        createdAt: toIso(row.created_at),
+        updatedAt: toIso(row.updated_at)
+    };
+    if (row.message)
+        record.message = String(row.message);
+    if (row.user_id)
+        record.userId = String(row.user_id);
+    if (row.conversation_id)
+        record.conversationId = String(row.conversation_id);
+    if (row.project_id)
+        record.projectId = String(row.project_id);
+    if (row.target !== null && row.target !== undefined)
+        record.target = normalizeJsonObject(row.target);
+    if (row.trigger !== null && row.trigger !== undefined) {
+        record.trigger = normalizeJsonObject(row.trigger);
+    }
+    if (row.cron)
+        record.cron = String(row.cron);
+    if (row.every_minutes !== null && row.every_minutes !== undefined)
+        record.everyMinutes = Number(row.every_minutes);
+    if (row.timezone)
+        record.timezone = String(row.timezone);
+    if (row.lease_token)
+        record.leaseToken = String(row.lease_token);
+    if (row.lease_expires_at)
+        record.leaseExpiresAt = toIso(row.lease_expires_at);
+    if (row.last_run_at)
+        record.lastRunAt = toIso(row.last_run_at);
+    if (row.consecutive_failures !== null && row.consecutive_failures !== undefined) {
+        record.consecutiveFailures = Number(row.consecutive_failures);
+    }
+    if (row.metadata !== null && row.metadata !== undefined)
+        record.metadata = normalizeJsonObject(row.metadata);
+    return record;
+}
+export function mapConnectionDefinition(row) {
+    const record = {
+        name: String(row.name),
+        scope: String(row.scope),
+        protocol: String(row.protocol),
+        description: String(row.description),
+        definition: normalizeJsonObject(row.definition),
+        origin: String(row.origin),
+        createdAt: toIso(row.created_at),
+        updatedAt: toIso(row.updated_at)
+    };
+    if (row.metadata !== null && row.metadata !== undefined)
+        record.metadata = normalizeJsonObject(row.metadata);
+    return record;
+}
+//# sourceMappingURL=row-mappers.js.map