@assemblyline-agents/postgres 0.1.0

package/dist/index.d.ts

@@ -0,0 +1,208 @@
+import { type JsonObject, type JsonValue, type MemoryListInput, type MemorySearchInput } from "@assemblyline-agents/core";
+import type { CheckpointRecord, CheckpointRetentionOptions, CheckpointRetentionResult, ConnectionAuthorizationSessionRecord, ConnectionAuthorizationSessionStore, ConnectionDefinitionStore, ConnectionGrantStore, ConversationRecord, ConversationTurnRecord, ConversationTurnStatus, DeliveryFailureInput, DeliveryLeaseOptions, DeliveryObligation, DeliveryObligationInput, DeliveryObligationPatch, FileIndexRecord, MemoryDocument, MemoryEmbeddingRecord, MemorySearchResult, MemoryScope, MessageRecord, RunEvent, RunEventType, RunRecord, RunControlIntent, RunStatus, RuntimeControlEvent, RuntimeSettingRecord, ResolvedStateAdapter, ScheduleRecord, ScheduleRunRecord, ScheduleStore, SandboxSnapshotRecord, SandboxSessionInput, SandboxSessionPatch, SandboxSessionRecord, SandboxSyncFailureInput, SandboxSyncJobInput, SandboxSyncJobPatch, SandboxSyncJobRecord, SandboxSyncLeaseOptions, SkillStore, StoredConnectionGrant, ToolCallRecord, UsageAggregate, UsageQuery, UsageRecord, UsageRecordInput, UsageSummaryQuery } from "@assemblyline-agents/runtime";
+import type { QueryClient } from "./client-types.js";
+import { type PostgresMigrationPlan } from "./migrations.js";
+import * as conversationsDb from "./adapter-conversations.js";
+import * as conversationTurnsDb from "./adapter-conversation-turns.js";
+import { type PostgresQueueRecord, type PostgresScheduleRunRecord } from "./adapter-misc.js";
+export { createPostgresPoolFromEnv, openeveProvider, POSTGRES_STATE_PROVIDER_METADATA } from "./provider.js";
+export { localPostgres, neonPostgres, postgresAdapter, railwayPostgres, supabasePostgres, type PostgresAdapter, type PostgresProvider, type RailwayPostgresOptions } from "./presets.js";
+export type { QueryClient } from "./client-types.js";
+export { postgresMigrations } from "./migrations.js";
+export type { PostgresMigration, PostgresMigrationPlan, PostgresMigrationPlanItem } from "./migrations.js";
+export { PostgresConnectionDefinitionStore, PostgresScheduleStore, PostgresSkillStore } from "./stores.js";
+export type { PostgresQueueRecord, PostgresScheduleRunRecord } from "./adapter-misc.js";
+export interface PostgresStateAdapterOptions {
+    optionalMigrations?: boolean | string[];
+    packageVersion?: string;
+    /**
+     * Set when the adapter created the pool itself (provider registration path).
+     * `close()` only ends pools the adapter owns; host-supplied clients are the
+     * host's responsibility.
+     */
+    ownsClient?: boolean;
+}
+export declare class PostgresStateAdapter implements ResolvedStateAdapter, ConnectionGrantStore, ConnectionAuthorizationSessionStore {
+    private readonly client;
+    private readonly options;
+    readonly skillStore: SkillStore;
+    readonly scheduleStore: ScheduleStore;
+    readonly connectionDefinitionStore: ConnectionDefinitionStore;
+    constructor(client: QueryClient, options?: PostgresStateAdapterOptions);
+    supportsMemoryEmbeddings(): boolean;
+    /** Graceful-shutdown hook: ends the pg pool, but only when this adapter created it. */
+    close(): Promise<void>;
+    migrate(): Promise<void>;
+    planMigrations(): Promise<PostgresMigrationPlan>;
+    private shouldRunOptionalMigration;
+    private packageVersion;
+    private ensureSchemaMigrationsTable;
+    private appliedMigration;
+    private appliedMigrations;
+    private applyMigration;
+    createRun(input: Omit<RunRecord, "id" | "createdAt" | "updatedAt">): Promise<RunRecord>;
+    updateRun(id: string, patch: Partial<Omit<RunRecord, "id" | "createdAt">>): Promise<RunRecord>;
+    touchRun(id: string): Promise<void>;
+    requestRunControl(id: string, intent: RunControlIntent): Promise<RunRecord>;
+    settleRunControl(id: string): Promise<RunRecord>;
+    getRun(id: string): Promise<RunRecord | undefined>;
+    listRuns(): Promise<RunRecord[]>;
+    listRunsByStatus(statuses: RunStatus[], options?: {
+        updatedBefore?: string;
+        limit?: number;
+    }): Promise<RunRecord[]>;
+    appendEvent(runId: string, type: RunEventType, data?: JsonObject): Promise<RunEvent>;
+    listEvents(runId: string): Promise<RunEvent[]>;
+    createToolCall(input: Omit<ToolCallRecord, "id" | "createdAt" | "updatedAt">): Promise<ToolCallRecord>;
+    updateToolCall(id: string, patch: Partial<Omit<ToolCallRecord, "id" | "createdAt">>): Promise<ToolCallRecord>;
+    listToolCalls(runId: string): Promise<ToolCallRecord[]>;
+    createCheckpoint(input: Omit<CheckpointRecord, "id" | "createdAt">): Promise<CheckpointRecord>;
+    listCheckpoints(runId: string): Promise<CheckpointRecord[]>;
+    compactCheckpoints(options?: CheckpointRetentionOptions): Promise<CheckpointRetentionResult>;
+    createDelivery(input: DeliveryObligationInput): Promise<DeliveryObligation>;
+    updateDelivery(id: string, patch: DeliveryObligationPatch): Promise<DeliveryObligation>;
+    listDeliveries(runId: string): Promise<DeliveryObligation[]>;
+    leaseDueDeliveries(options?: DeliveryLeaseOptions): Promise<DeliveryObligation[]>;
+    completeDeliverySend(id: string, leaseToken: string, patch?: {
+        payload?: JsonObject;
+    }): Promise<DeliveryObligation>;
+    failDeliverySend(id: string, leaseToken: string, failure: DeliveryFailureInput): Promise<DeliveryObligation>;
+    recoverExpiredDeliveries(now?: Date): Promise<DeliveryObligation[]>;
+    enqueueRun(runId: string, options?: {
+        queue?: string;
+        availableAt?: string;
+        idempotencyKey?: string;
+    }): Promise<PostgresQueueRecord>;
+    enqueueScheduleRun(input: {
+        scheduleId: string;
+        runId?: string;
+        runAt: string;
+        idempotencyKey: string;
+        status?: string;
+    }): Promise<PostgresScheduleRunRecord>;
+    reserveIdempotencyKey(scope: string, key: string, ownerId?: string): Promise<boolean>;
+    getSetting(scope: string, key: string): Promise<RuntimeSettingRecord | undefined>;
+    setSetting(scope: string, key: string, value: JsonValue): Promise<RuntimeSettingRecord>;
+    appendRuntimeControlEvent(scope: string, type: string, data?: JsonObject): Promise<RuntimeControlEvent>;
+    listRuntimeControlEvents(scope: string): Promise<RuntimeControlEvent[]>;
+    upsertConversation(input: {
+        id: string;
+        agentScope?: string;
+        agentRevision: string;
+        channel: string;
+        subject?: string;
+        title?: string;
+        metadata?: JsonObject;
+    }): Promise<ConversationRecord>;
+    getConversation(id: string): Promise<ConversationRecord | undefined>;
+    listConversations(input: Parameters<typeof conversationsDb.listConversations>[1]): ReturnType<typeof conversationsDb.listConversations>;
+    updateConversation(id: string, patch: Parameters<typeof conversationsDb.updateConversation>[2]): ReturnType<typeof conversationsDb.updateConversation>;
+    listRecentConversationsBySubject(input: {
+        channel: string;
+        subject: string;
+        since?: string;
+        limit?: number;
+        excludeId?: string;
+    }): Promise<ConversationRecord[]>;
+    appendMessage(input: {
+        conversationId: string;
+        runId?: string;
+        role: MessageRecord["role"];
+        content: JsonObject;
+    }): Promise<MessageRecord>;
+    listRecentMessages(conversationId: string, limit?: number): Promise<MessageRecord[]>;
+    listMessages(input: Parameters<typeof conversationsDb.listMessages>[1]): ReturnType<typeof conversationsDb.listMessages>;
+    enqueueConversationTurn(input: Parameters<typeof conversationTurnsDb.enqueueConversationTurn>[1]): ReturnType<typeof conversationTurnsDb.enqueueConversationTurn>;
+    leaseNextConversationTurn(options: Parameters<typeof conversationTurnsDb.leaseNextConversationTurn>[1]): ReturnType<typeof conversationTurnsDb.leaseNextConversationTurn>;
+    attachConversationTurnRun(id: string, leaseToken: string, runId: string): Promise<ConversationTurnRecord>;
+    settleConversationTurn(id: string, leaseToken: string, status: "blocked" | "completed" | "failed" | "cancelled", options?: {
+        runId?: string;
+        error?: string;
+    }): Promise<ConversationTurnRecord>;
+    settleConversationTurnByRunId(runId: string, status: "blocked" | "completed" | "failed" | "cancelled", options?: {
+        error?: string;
+    }): Promise<ConversationTurnRecord | undefined>;
+    retryConversationTurn(id: string, leaseToken: string, error: string): Promise<ConversationTurnRecord>;
+    recoverExpiredConversationTurns(now?: Date): Promise<ConversationTurnRecord[]>;
+    listConversationTurns(options?: {
+        agentScope?: string;
+        conversationId?: string;
+        status?: ConversationTurnStatus;
+    }): Promise<ConversationTurnRecord[]>;
+    registerSchedule(input: {
+        id: string;
+        name: string;
+        cron: string;
+        timezone?: string;
+        metadata?: JsonObject;
+    }): Promise<ScheduleRecord>;
+    recordScheduleRun(input: {
+        scheduleId: string;
+        runId?: string;
+        runAt: string;
+        idempotencyKey: string;
+        status?: string;
+    }): Promise<ScheduleRunRecord>;
+    recordFileIndex(input: {
+        runId?: string;
+        path: string;
+        kind: FileIndexRecord["kind"];
+        blobKey: string;
+        sha256: string;
+        sizeBytes: number;
+        metadata?: JsonObject;
+    }): Promise<FileIndexRecord>;
+    listFileIndexes(runId?: string): Promise<FileIndexRecord[]>;
+    recordUsage(input: UsageRecordInput): Promise<UsageRecord>;
+    listUsage(runId?: string): Promise<UsageRecord[]>;
+    queryUsage(query: UsageQuery): Promise<UsageRecord[]>;
+    summarizeUsage(query: UsageSummaryQuery): Promise<UsageAggregate[]>;
+    recordSandboxSnapshot(input: {
+        runId: string;
+        sandboxId: string;
+        snapshotId: string;
+        metadata?: JsonObject;
+    }): Promise<SandboxSnapshotRecord>;
+    upsertSandboxSession(input: SandboxSessionInput): Promise<SandboxSessionRecord>;
+    getSandboxSession(sessionKey: string): Promise<SandboxSessionRecord | undefined>;
+    listSandboxSessions(options?: {
+        dirty?: boolean;
+    }): Promise<SandboxSessionRecord[]>;
+    updateSandboxSession(sessionKey: string, patch: SandboxSessionPatch): Promise<SandboxSessionRecord>;
+    createSandboxSyncJob(input: SandboxSyncJobInput): Promise<SandboxSyncJobRecord>;
+    enqueueSandboxSyncJob(input: SandboxSyncJobInput): Promise<SandboxSyncJobRecord>;
+    updateSandboxSyncJob(id: string, patch: SandboxSyncJobPatch): Promise<SandboxSyncJobRecord>;
+    listSandboxSyncJobs(runId?: string): Promise<SandboxSyncJobRecord[]>;
+    leaseSandboxSyncJobs(options?: SandboxSyncLeaseOptions): Promise<SandboxSyncJobRecord[]>;
+    completeSandboxSyncJob(id: string, leaseToken: string, patch?: {
+        manifest?: JsonObject;
+        metadata?: JsonObject;
+    }): Promise<SandboxSyncJobRecord>;
+    failSandboxSyncJob(id: string, leaseToken: string, failure: SandboxSyncFailureInput): Promise<SandboxSyncJobRecord>;
+    recoverExpiredSandboxSyncJobs(now?: Date): Promise<SandboxSyncJobRecord[]>;
+    listMemoryDocuments(scope: MemoryScope, input?: MemoryListInput): Promise<MemoryDocument[]>;
+    searchMemoryDocuments(scope: MemoryScope, input: MemorySearchInput): Promise<MemorySearchResult[]>;
+    getMemoryDocument(scope: MemoryScope, path: string): Promise<MemoryDocument | undefined>;
+    upsertMemoryDocument(document: MemoryDocument): Promise<MemoryDocument>;
+    deleteMemoryDocument(scope: MemoryScope, path: string): Promise<void>;
+    listMemoryDocumentsNeedingEmbeddings(scope: MemoryScope, options: {
+        model: string;
+        limit?: number;
+    }): Promise<MemoryDocument[]>;
+    upsertMemoryEmbedding(input: Omit<MemoryEmbeddingRecord, "id" | "createdAt" | "updatedAt">): Promise<MemoryEmbeddingRecord>;
+    searchMemoryEmbeddings(scope: MemoryScope, input: MemorySearchInput, embedding: number[], options: {
+        model: string;
+        limit?: number;
+    }): Promise<MemorySearchResult[]>;
+    markMemoryEmbeddingsStale(scope: MemoryScope, path: string, currentSha256: string): Promise<void>;
+    private requireMemoryEmbeddings;
+    getGrant(connectionName: string, principalKey: string): Promise<StoredConnectionGrant | undefined>;
+    saveGrant(grant: StoredConnectionGrant): Promise<void>;
+    deleteGrant(connectionName: string, principalKey: string): Promise<void>;
+    createSession(session: ConnectionAuthorizationSessionRecord): Promise<void>;
+    updateSession(id: string, patch: Partial<Omit<ConnectionAuthorizationSessionRecord, "id" | "createdAt">>): Promise<ConnectionAuthorizationSessionRecord>;
+    getSession(id: string): Promise<ConnectionAuthorizationSessionRecord | undefined>;
+    getSessionByState(state: string): Promise<ConnectionAuthorizationSessionRecord | undefined>;
+    findPendingSession(connectionName: string, principalKey: string): Promise<ConnectionAuthorizationSessionRecord | undefined>;
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,UAAU,EACf,KAAK,SAAS,EACd,KAAK,eAAe,EACpB,KAAK,iBAAiB,EACvB,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EACV,gBAAgB,EAChB,0BAA0B,EAC1B,yBAAyB,EACzB,oCAAoC,EACpC,mCAAmC,EACnC,yBAAyB,EACzB,oBAAoB,EACpB,kBAAkB,EAClB,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,oBAAoB,EACpB,kBAAkB,EAClB,uBAAuB,EACvB,uBAAuB,EACvB,eAAe,EACf,cAAc,EACd,qBAAqB,EACrB,kBAAkB,EAClB,WAAW,EACX,aAAa,EACb,QAAQ,EACR,YAAY,EACZ,SAAS,EACT,gBAAgB,EAChB,SAAS,EACT,mBAAmB,EACnB,oBAAoB,EACpB,oBAAoB,EACpB,cAAc,EACd,iBAAiB,EACjB,aAAa,EACb,qBAAqB,EACrB,mBAAmB,EACnB,mBAAmB,EACnB,oBAAoB,EACpB,uBAAuB,EACvB,mBAAmB,EACnB,mBAAmB,EACnB,oBAAoB,EACpB,uBAAuB,EACvB,UAAU,EACV,qBAAqB,EACrB,cAAc,EACd,cAAc,EACd,UAAU,EACV,WAAW,EACX,gBAAgB,EAChB,iBAAiB,EAClB,MAAM,8BAA8B,CAAC;AACtC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAIL,KAAK,qBAAqB,EAE3B,MAAM,iBAAiB,CAAC;AAczB,OAAO,KAAK,eAAe,MAAM,4BAA4B,CAAC;AAC9D,OAAO,KAAK,mBAAmB,MAAM,iCAAiC,CAAC;AAIvE,OAAO,EAAE,KAAK,mBAAmB,EAAE,KAAK,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAK7F,OAAO,EAAE,yBAAyB,EAAE,eAAe,EAAE,gCAAgC,EAAE,MAAM,eAAe,CAAC;AAC7G,OAAO,EACL,aAAa,EACb,YAAY,EACZ,eAAe,EACf,eAAe,EACf,gBAAgB,EAChB,KAAK,eAAe,EACpB,KAAK,gBAAgB,EACrB,KAAK,sBAAsB,EAC5B,MAAM,cAAc,CAAC;AACtB,YAAY,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrD,YAAY,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,yBAAyB,EAAE,MAAM,iBAAiB,CAAC;AAC3G,OAAO,EAAE,iCAAiC,EAAE,qBAAqB,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAC3G,YAAY,EAAE,mBAAmB,EAAE,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAExF,MAAM,WAAW,2BAA2B;IAC1C,kBAAkB,CAAC,EAAE,OAAO,GAAG,MAAM,EAAE,CAAC;IACxC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;;OAIG;IACH,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAED,qBAAa,oBAAqB,YAAW,oBAAoB,EAAE,oBAAoB,EAAE,mCAAmC;IAK9G,OAAO,CAAC,QAAQ,CAAC,MAAM;IAAe,OAAO,CAAC,QAAQ,CAAC,OAAO;IAJ1E,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC;IAChC,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAC;IACtC,QAAQ,CAAC,yBAAyB,EAAE,yBAAyB,CAAC;gBAEjC,MAAM,EAAE,WAAW,EAAmB,OAAO,GAAE,2BAAgC;IAM5G,wBAAwB,IAAI,OAAO;IAInC,uFAAuF;IACjF,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAKtB,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAgBxB,cAAc,IAAI,OAAO,CAAC,qBAAqB,CAAC;IAyCtD,OAAO,CAAC,0BAA0B;IAKlC,OAAO,CAAC,cAAc;YAIR,2BAA2B;YAI3B,gBAAgB;YAahB,iBAAiB;YAYjB,cAAc;IA+BtB,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC;IAIvF,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC;IAI9F,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAInC,iBAAiB,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAAC,SAAS,CAAC;IAI3E,gBAAgB,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;IAIhD,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC;IAIlD,QAAQ,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;IAIhC,gBAAgB,CAAC,QAAQ,EAAE,SAAS,EAAE,EAAE,OAAO,GAAE;QAAE,aAAa,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAO,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAIvH,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,GAAE,UAAe,GAAG,OAAO,CAAC,QAAQ,CAAC;IAIxF,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;IAI9C,cAAc,CAAC,KAAK,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC;IAItG,cAAc,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC;IAI7G,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAIvD,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE,IAAI,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAI9F,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAI3D,kBAAkB,CAAC,OAAO,GAAE,0BAA+B,GAAG,OAAO,CAAC,yBAAyB,CAAC;IAIhG,cAAc,CAAC,KAAK,EAAE,uBAAuB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAI3E,cAAc,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,uBAAuB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAIvF,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAI5D,kBAAkB,CAAC,OAAO,GAAE,oBAAyB,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAIrF,oBAAoB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,GAAE;QAAE,OAAO,CAAC,EAAE,UAAU,CAAA;KAAO,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAIvH,gBAAgB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAI5G,wBAAwB,CAAC,GAAG,GAAE,IAAiB,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAI/E,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAC;QAAC,cAAc,CAAC,EAAE,MAAM,CAAA;KAAO,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAIxI,kBAAkB,CAAC,KAAK,EAAE;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,cAAc,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,yBAAyB,CAAC;IAI7J,qBAAqB,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAIrF,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,GAAG,SAAS,CAAC;IAIjF,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAIvF,yBAAyB,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,GAAE,UAAe,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAI3G,wBAAwB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAIvE,kBAAkB,CAAC,KAAK,EAAE;QAC9B,EAAE,EAAE,MAAM,CAAC;QACX,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,aAAa,EAAE,MAAM,CAAC;QACtB,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,QAAQ,CAAC,EAAE,UAAU,CAAC;KACvB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAIzB,eAAe,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,GAAG,SAAS,CAAC;IAIpE,iBAAiB,CAAC,KAAK,EAAE,UAAU,CAAC,OAAO,eAAe,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,OAAO,eAAe,CAAC,iBAAiB,CAAC;IAIvI,kBAAkB,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,CAAC,OAAO,eAAe,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,OAAO,eAAe,CAAC,kBAAkB,CAAC;IAItJ,gCAAgC,CAAC,KAAK,EAAE;QAC5C,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,CAAC;QAChB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAI3B,aAAa,CAAC,KAAK,EAAE;QACzB,cAAc,EAAE,MAAM,CAAC;QACvB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;QAC5B,OAAO,EAAE,UAAU,CAAC;KACrB,GAAG,OAAO,CAAC,aAAa,CAAC;IAIpB,kBAAkB,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,SAAK,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IAIhF,YAAY,CAAC,KAAK,EAAE,UAAU,CAAC,OAAO,eAAe,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,OAAO,eAAe,CAAC,YAAY,CAAC;IAIxH,uBAAuB,CAC3B,KAAK,EAAE,UAAU,CAAC,OAAO,mBAAmB,CAAC,uBAAuB,CAAC,CAAC,CAAC,CAAC,GACvE,UAAU,CAAC,OAAO,mBAAmB,CAAC,uBAAuB,CAAC;IAI3D,yBAAyB,CAC7B,OAAO,EAAE,UAAU,CAAC,OAAO,mBAAmB,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,GAC3E,UAAU,CAAC,OAAO,mBAAmB,CAAC,yBAAyB,CAAC;IAI7D,yBAAyB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAIzG,sBAAsB,CAC1B,EAAE,EAAE,MAAM,EACV,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,WAAW,EACxD,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAC3C,OAAO,CAAC,sBAAsB,CAAC;IAI5B,6BAA6B,CACjC,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,WAAW,EACxD,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAC3B,OAAO,CAAC,sBAAsB,GAAG,SAAS,CAAC;IAIxC,qBAAqB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAIrG,+BAA+B,CAAC,GAAG,CAAC,EAAE,IAAI,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC;IAI9E,qBAAqB,CAAC,OAAO,CAAC,EAAE;QACpC,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,MAAM,CAAC,EAAE,sBAAsB,CAAC;KACjC,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC;IAI/B,gBAAgB,CAAC,KAAK,EAAE;QAC5B,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,UAAU,CAAC;KACvB,GAAG,OAAO,CAAC,cAAc,CAAC;IAIrB,iBAAiB,CAAC,KAAK,EAAE;QAC7B,UAAU,EAAE,MAAM,CAAC;QACnB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,EAAE,MAAM,CAAC;QACvB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAIxB,eAAe,CAAC,KAAK,EAAE;QAC3B,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;QAC9B,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,UAAU,CAAC;KACvB,GAAG,OAAO,CAAC,eAAe,CAAC;IAItB,eAAe,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAI3D,WAAW,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;IAI1D,SAAS,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAIjD,UAAU,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAIrD,cAAc,CAAC,KAAK,EAAE,iBAAiB,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAInE,qBAAqB,CAAC,KAAK,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,UAAU,CAAA;KAAE,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAI7I,oBAAoB,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAI/E,iBAAiB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,GAAG,SAAS,CAAC;IAIhF,mBAAmB,CAAC,OAAO,GAAE;QAAE,KAAK,CAAC,EAAE,OAAO,CAAA;KAAO,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAIvF,oBAAoB,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAInG,oBAAoB,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAI/E,qBAAqB,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAIhF,oBAAoB,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAI3F,mBAAmB,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAIpE,oBAAoB,CAAC,OAAO,GAAE,uBAA4B,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAI5F,sBAAsB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,GAAE;QAAE,QAAQ,CAAC,EAAE,UAAU,CAAC;QAAC,QAAQ,CAAC,EAAE,UAAU,CAAA;KAAO,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAInJ,kBAAkB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,uBAAuB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAInH,6BAA6B,CAAC,GAAG,GAAE,IAAiB,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAItF,mBAAmB,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,GAAE,eAAoB,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAI/F,qBAAqB,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAIlG,iBAAiB,CAAC,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC;IAIxF,oBAAoB,CAAC,QAAQ,EAAE,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;IAIvE,oBAAoB,CAAC,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMrE,oCAAoC,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAK/H,qBAAqB,CAAC,KAAK,EAAE,IAAI,CAAC,qBAAqB,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAK3H,sBAAsB,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,iBAAiB,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAKpK,yBAAyB,CAAC,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAKvG,OAAO,CAAC,uBAAuB;IAOzB,QAAQ,CAAC,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,GAAG,SAAS,CAAC;IAIlG,SAAS,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;IAItD,WAAW,CAAC,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIxE,aAAa,CAAC,OAAO,EAAE,oCAAoC,GAAG,OAAO,CAAC,IAAI,CAAC;IAI3E,aAAa,CACjB,EAAE,EAAE,MAAM,EACV,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,oCAAoC,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GAC7E,OAAO,CAAC,oCAAoC,CAAC;IAI1C,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,oCAAoC,GAAG,SAAS,CAAC;IAIjF,iBAAiB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,oCAAoC,GAAG,SAAS,CAAC;IAI3F,kBAAkB,CAAC,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,oCAAoC,GAAG,SAAS,CAAC;CAGlI"}

package/dist/index.js

@@ -0,0 +1,424 @@
+import { POSTGRES_SCHEMA_MIGRATIONS_SQL, postgresMigrations } from "./migrations.js";
+import { postgresMigrationChecksum, timestamp, toIso } from "./sql-helpers.js";
+import { PostgresConnectionDefinitionStore, PostgresScheduleStore, PostgresSkillStore } from "./stores.js";
+import * as deliveryDb from "./adapter-delivery.js";
+import * as memoryDb from "./adapter-memory.js";
+import * as runsDb from "./adapter-runs.js";
+import * as sandboxDb from "./adapter-sandbox.js";
+import * as settingsDb from "./adapter-settings.js";
+import * as checkpointsDb from "./adapter-checkpoints.js";
+import * as conversationsDb from "./adapter-conversations.js";
+import * as conversationTurnsDb from "./adapter-conversation-turns.js";
+import * as grantsDb from "./adapter-grants.js";
+import * as miscDb from "./adapter-misc.js";
+import * as usageDb from "./adapter-usage.js";
+const OPENEVE_POSTGRES_PACKAGE_VERSION = "0.0.0";
+const MEMORY_EMBEDDINGS_MIGRATION_ID = "003_openeve_memory_embeddings_pgvector";
+export { createPostgresPoolFromEnv, openeveProvider, POSTGRES_STATE_PROVIDER_METADATA } from "./provider.js";
+export { localPostgres, neonPostgres, postgresAdapter, railwayPostgres, supabasePostgres } from "./presets.js";
+export { postgresMigrations } from "./migrations.js";
+export { PostgresConnectionDefinitionStore, PostgresScheduleStore, PostgresSkillStore } from "./stores.js";
+export class PostgresStateAdapter {
+    client;
+    options;
+    skillStore;
+    scheduleStore;
+    connectionDefinitionStore;
+    constructor(client, options = {}) {
+        this.client = client;
+        this.options = options;
+        this.skillStore = new PostgresSkillStore(client);
+        this.scheduleStore = new PostgresScheduleStore(client);
+        this.connectionDefinitionStore = new PostgresConnectionDefinitionStore(client);
+    }
+    supportsMemoryEmbeddings() {
+        return this.shouldRunOptionalMigration(MEMORY_EMBEDDINGS_MIGRATION_ID);
+    }
+    /** Graceful-shutdown hook: ends the pg pool, but only when this adapter created it. */
+    async close() {
+        if (!this.options.ownsClient)
+            return;
+        await this.client.end?.();
+    }
+    async migrate() {
+        await this.ensureSchemaMigrationsTable();
+        for (const migration of postgresMigrations) {
+            if (migration.optional && !this.shouldRunOptionalMigration(migration.id))
+                continue;
+            const checksum = postgresMigrationChecksum(migration);
+            const existing = await this.appliedMigration(migration.id);
+            if (existing) {
+                if (existing.checksum !== checksum) {
+                    throw new Error(`Postgres migration checksum mismatch for ${migration.id}. Expected ${existing.checksum}, current ${checksum}.`);
+                }
+                continue;
+            }
+            await this.applyMigration(migration, checksum);
+        }
+    }
+    async planMigrations() {
+        let applied = new Map();
+        try {
+            applied = await this.appliedMigrations();
+        }
+        catch {
+            applied = new Map();
+        }
+        const items = postgresMigrations.map((migration) => {
+            const checksum = postgresMigrationChecksum(migration);
+            if (migration.optional && !this.shouldRunOptionalMigration(migration.id)) {
+                return {
+                    id: migration.id,
+                    description: migration.description,
+                    checksum,
+                    optional: true,
+                    status: "skipped_optional"
+                };
+            }
+            const current = applied.get(migration.id);
+            if (!current) {
+                return {
+                    id: migration.id,
+                    description: migration.description,
+                    checksum,
+                    optional: migration.optional === true,
+                    status: "pending"
+                };
+            }
+            return {
+                id: migration.id,
+                description: migration.description,
+                checksum,
+                optional: migration.optional === true,
+                status: current.checksum === checksum ? "applied" : "checksum_mismatch",
+                appliedAt: current.appliedAt,
+                packageVersion: current.packageVersion
+            };
+        });
+        return { packageVersion: this.packageVersion(), items };
+    }
+    shouldRunOptionalMigration(id) {
+        if (this.options.optionalMigrations === true)
+            return true;
+        return Array.isArray(this.options.optionalMigrations) && this.options.optionalMigrations.includes(id);
+    }
+    packageVersion() {
+        return this.options.packageVersion ?? OPENEVE_POSTGRES_PACKAGE_VERSION;
+    }
+    async ensureSchemaMigrationsTable() {
+        await this.client.query(POSTGRES_SCHEMA_MIGRATIONS_SQL);
+    }
+    async appliedMigration(id) {
+        const result = await this.client.query("select checksum, applied_at, package_version from openeve_schema_migrations where id=$1", [id]);
+        const row = result.rows[0];
+        return row ? {
+            checksum: String(row.checksum),
+            appliedAt: toIso(row.applied_at),
+            packageVersion: String(row.package_version)
+        } : undefined;
+    }
+    async appliedMigrations() {
+        const result = await this.client.query("select id, checksum, applied_at, package_version from openeve_schema_migrations");
+        return new Map(result.rows.map((row) => [
+            String(row.id),
+            {
+                checksum: String(row.checksum),
+                appliedAt: toIso(row.applied_at),
+                packageVersion: String(row.package_version)
+            }
+        ]));
+    }
+    async applyMigration(migration, checksum) {
+        // Transactions must be pinned to a single connection: pg pools may route
+        // each pool.query() to a different client, silently splitting
+        // begin/commit across connections. Pin one client when the injected
+        // QueryClient is a pool; plain single-connection clients run as-is.
+        const poolCandidate = this.client;
+        const pinned = typeof poolCandidate.connect === "function" ? await poolCandidate.connect() : undefined;
+        const client = pinned ?? this.client;
+        try {
+            await client.query("begin");
+            await client.query(migration.sql);
+            await client.query(`insert into openeve_schema_migrations (id, checksum, description, package_version, applied_at)
+         values ($1,$2,$3,$4,$5)`, [migration.id, checksum, migration.description, this.packageVersion(), timestamp()]);
+            await client.query("commit");
+        }
+        catch (error) {
+            try {
+                await client.query("rollback");
+            }
+            catch {
+                // Ignore rollback failures so the original migration error is preserved.
+            }
+            throw error;
+        }
+        finally {
+            pinned?.release();
+        }
+    }
+    async createRun(input) {
+        return runsDb.createRun(this.client, input);
+    }
+    async updateRun(id, patch) {
+        return runsDb.updateRun(this.client, id, patch);
+    }
+    async touchRun(id) {
+        return runsDb.touchRun(this.client, id);
+    }
+    async requestRunControl(id, intent) {
+        return runsDb.requestRunControl(this.client, id, intent);
+    }
+    async settleRunControl(id) {
+        return runsDb.settleRunControl(this.client, id);
+    }
+    async getRun(id) {
+        return runsDb.getRun(this.client, id);
+    }
+    async listRuns() {
+        return runsDb.listRuns(this.client);
+    }
+    async listRunsByStatus(statuses, options = {}) {
+        return runsDb.listRunsByStatus(this.client, statuses, options);
+    }
+    async appendEvent(runId, type, data = {}) {
+        return runsDb.appendEvent(this.client, runId, type, data);
+    }
+    async listEvents(runId) {
+        return runsDb.listEvents(this.client, runId);
+    }
+    async createToolCall(input) {
+        return runsDb.createToolCall(this.client, input);
+    }
+    async updateToolCall(id, patch) {
+        return runsDb.updateToolCall(this.client, id, patch);
+    }
+    async listToolCalls(runId) {
+        return runsDb.listToolCalls(this.client, runId);
+    }
+    async createCheckpoint(input) {
+        return checkpointsDb.createCheckpoint(this.client, input);
+    }
+    async listCheckpoints(runId) {
+        return checkpointsDb.listCheckpoints(this.client, runId);
+    }
+    async compactCheckpoints(options = {}) {
+        return checkpointsDb.compactCheckpoints(this.client, options);
+    }
+    async createDelivery(input) {
+        return deliveryDb.createDelivery(this.client, input);
+    }
+    async updateDelivery(id, patch) {
+        return deliveryDb.updateDelivery(this.client, id, patch);
+    }
+    async listDeliveries(runId) {
+        return deliveryDb.listDeliveries(this.client, runId);
+    }
+    async leaseDueDeliveries(options = {}) {
+        return deliveryDb.leaseDueDeliveries(this.client, options);
+    }
+    async completeDeliverySend(id, leaseToken, patch = {}) {
+        return deliveryDb.completeDeliverySend(this.client, id, leaseToken, patch);
+    }
+    async failDeliverySend(id, leaseToken, failure) {
+        return deliveryDb.failDeliverySend(this.client, id, leaseToken, failure);
+    }
+    async recoverExpiredDeliveries(now = new Date()) {
+        return deliveryDb.recoverExpiredDeliveries(this.client, now);
+    }
+    async enqueueRun(runId, options = {}) {
+        return miscDb.enqueueRun(this.client, runId, options);
+    }
+    async enqueueScheduleRun(input) {
+        return miscDb.enqueueScheduleRun(this.client, input);
+    }
+    async reserveIdempotencyKey(scope, key, ownerId) {
+        return miscDb.reserveIdempotencyKey(this.client, scope, key, ownerId);
+    }
+    async getSetting(scope, key) {
+        return settingsDb.getSetting(this.client, scope, key);
+    }
+    async setSetting(scope, key, value) {
+        return settingsDb.setSetting(this.client, scope, key, value);
+    }
+    async appendRuntimeControlEvent(scope, type, data = {}) {
+        return settingsDb.appendRuntimeControlEvent(this.client, scope, type, data);
+    }
+    async listRuntimeControlEvents(scope) {
+        return settingsDb.listRuntimeControlEvents(this.client, scope);
+    }
+    async upsertConversation(input) {
+        return conversationsDb.upsertConversation(this.client, input);
+    }
+    async getConversation(id) {
+        return conversationsDb.getConversation(this.client, id);
+    }
+    async listConversations(input) {
+        return conversationsDb.listConversations(this.client, input);
+    }
+    async updateConversation(id, patch) {
+        return conversationsDb.updateConversation(this.client, id, patch);
+    }
+    async listRecentConversationsBySubject(input) {
+        return conversationsDb.listRecentConversationsBySubject(this.client, input);
+    }
+    async appendMessage(input) {
+        return conversationsDb.appendMessage(this.client, input);
+    }
+    async listRecentMessages(conversationId, limit = 20) {
+        return conversationsDb.listRecentMessages(this.client, conversationId, limit);
+    }
+    async listMessages(input) {
+        return conversationsDb.listMessages(this.client, input);
+    }
+    async enqueueConversationTurn(input) {
+        return conversationTurnsDb.enqueueConversationTurn(this.client, input);
+    }
+    async leaseNextConversationTurn(options) {
+        return conversationTurnsDb.leaseNextConversationTurn(this.client, options);
+    }
+    async attachConversationTurnRun(id, leaseToken, runId) {
+        return conversationTurnsDb.attachConversationTurnRun(this.client, id, leaseToken, runId);
+    }
+    async settleConversationTurn(id, leaseToken, status, options) {
+        return conversationTurnsDb.settleConversationTurn(this.client, id, leaseToken, status, options);
+    }
+    async settleConversationTurnByRunId(runId, status, options) {
+        return conversationTurnsDb.settleConversationTurnByRunId(this.client, runId, status, options);
+    }
+    async retryConversationTurn(id, leaseToken, error) {
+        return conversationTurnsDb.retryConversationTurn(this.client, id, leaseToken, error);
+    }
+    async recoverExpiredConversationTurns(now) {
+        return conversationTurnsDb.recoverExpiredConversationTurns(this.client, now);
+    }
+    async listConversationTurns(options) {
+        return conversationTurnsDb.listConversationTurns(this.client, options);
+    }
+    async registerSchedule(input) {
+        return miscDb.registerSchedule(this.client, input);
+    }
+    async recordScheduleRun(input) {
+        return miscDb.recordScheduleRun(this.client, input);
+    }
+    async recordFileIndex(input) {
+        return miscDb.recordFileIndex(this.client, input);
+    }
+    async listFileIndexes(runId) {
+        return miscDb.listFileIndexes(this.client, runId);
+    }
+    async recordUsage(input) {
+        return usageDb.recordUsage(this.client, input);
+    }
+    async listUsage(runId) {
+        return usageDb.listUsage(this.client, runId);
+    }
+    async queryUsage(query) {
+        return usageDb.queryUsage(this.client, query);
+    }
+    async summarizeUsage(query) {
+        return usageDb.summarizeUsage(this.client, query);
+    }
+    async recordSandboxSnapshot(input) {
+        return sandboxDb.recordSandboxSnapshot(this.client, input);
+    }
+    async upsertSandboxSession(input) {
+        return sandboxDb.upsertSandboxSession(this.client, input);
+    }
+    async getSandboxSession(sessionKey) {
+        return sandboxDb.getSandboxSession(this.client, sessionKey);
+    }
+    async listSandboxSessions(options = {}) {
+        return sandboxDb.listSandboxSessions(this.client, options);
+    }
+    async updateSandboxSession(sessionKey, patch) {
+        return sandboxDb.updateSandboxSession(this.client, sessionKey, patch);
+    }
+    async createSandboxSyncJob(input) {
+        return sandboxDb.createSandboxSyncJob(this.client, input);
+    }
+    async enqueueSandboxSyncJob(input) {
+        return sandboxDb.enqueueSandboxSyncJob(this.client, input);
+    }
+    async updateSandboxSyncJob(id, patch) {
+        return sandboxDb.updateSandboxSyncJob(this.client, id, patch);
+    }
+    async listSandboxSyncJobs(runId) {
+        return sandboxDb.listSandboxSyncJobs(this.client, runId);
+    }
+    async leaseSandboxSyncJobs(options = {}) {
+        return sandboxDb.leaseSandboxSyncJobs(this.client, options);
+    }
+    async completeSandboxSyncJob(id, leaseToken, patch = {}) {
+        return sandboxDb.completeSandboxSyncJob(this.client, id, leaseToken, patch);
+    }
+    async failSandboxSyncJob(id, leaseToken, failure) {
+        return sandboxDb.failSandboxSyncJob(this.client, id, leaseToken, failure);
+    }
+    async recoverExpiredSandboxSyncJobs(now = new Date()) {
+        return sandboxDb.recoverExpiredSandboxSyncJobs(this.client, now);
+    }
+    async listMemoryDocuments(scope, input = {}) {
+        return memoryDb.listMemoryDocuments(this.client, scope, input);
+    }
+    async searchMemoryDocuments(scope, input) {
+        return memoryDb.searchMemoryDocuments(this.client, scope, input);
+    }
+    async getMemoryDocument(scope, path) {
+        return memoryDb.getMemoryDocument(this.client, scope, path);
+    }
+    async upsertMemoryDocument(document) {
+        return memoryDb.upsertMemoryDocument(this.client, document);
+    }
+    async deleteMemoryDocument(scope, path) {
+        return memoryDb.deleteMemoryDocument(this.client, scope, path, {
+            embeddingsAvailable: this.supportsMemoryEmbeddings()
+        });
+    }
+    async listMemoryDocumentsNeedingEmbeddings(scope, options) {
+        this.requireMemoryEmbeddings();
+        return memoryDb.listMemoryDocumentsNeedingEmbeddings(this.client, scope, options);
+    }
+    async upsertMemoryEmbedding(input) {
+        this.requireMemoryEmbeddings();
+        return memoryDb.upsertMemoryEmbedding(this.client, input);
+    }
+    async searchMemoryEmbeddings(scope, input, embedding, options) {
+        this.requireMemoryEmbeddings();
+        return memoryDb.searchMemoryEmbeddings(this.client, scope, input, embedding, options);
+    }
+    async markMemoryEmbeddingsStale(scope, path, currentSha256) {
+        if (!this.supportsMemoryEmbeddings())
+            return;
+        return memoryDb.markMemoryEmbeddingsStale(this.client, scope, path, currentSha256);
+    }
+    requireMemoryEmbeddings() {
+        if (this.supportsMemoryEmbeddings())
+            return;
+        throw new Error(`Postgres memory embeddings require optional migration ${MEMORY_EMBEDDINGS_MIGRATION_ID}.`);
+    }
+    async getGrant(connectionName, principalKey) {
+        return grantsDb.getGrant(this.client, connectionName, principalKey);
+    }
+    async saveGrant(grant) {
+        return grantsDb.saveGrant(this.client, grant);
+    }
+    async deleteGrant(connectionName, principalKey) {
+        return grantsDb.deleteGrant(this.client, connectionName, principalKey);
+    }
+    async createSession(session) {
+        return grantsDb.createSession(this.client, session);
+    }
+    async updateSession(id, patch) {
+        return grantsDb.updateSession(this.client, id, patch);
+    }
+    async getSession(id) {
+        return grantsDb.getSession(this.client, id);
+    }
+    async getSessionByState(state) {
+        return grantsDb.getSessionByState(this.client, state);
+    }
+    async findPendingSession(connectionName, principalKey) {
+        return grantsDb.findPendingSession(this.client, connectionName, principalKey);
+    }
+}
+//# sourceMappingURL=index.js.map