@dudousxd/nestjs-agent-store-drizzle 0.20.2 → 0.22.0

package/dist/index.d.cts

@@ -1,6 +1,6 @@
 import * as drizzle_orm_sqlite_core from 'drizzle-orm/sqlite-core';
 import { BaseSQLiteDatabase } from 'drizzle-orm/sqlite-core';
-import { MessageRole, ToolCallRequest, ToolResult, MessageAttachment, MessageUsage, AgentUiComponent, MessageFeedback, Actor, PageContext, QueuePause, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, ThreadTurnReader, ChatQueueStore, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, ToolCallApprovalState, EnqueueMessageInput, QueuedMessage, QueuedMessagePatch, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, ToolCallOutcome, RecordUsageInput, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, GovernancePageQuery, ToolCallWhere, GovernancePage, ThreadWhere, RunWhere, ApprovalWhere, GovernanceRunDetail, GovernanceThreadDetailQuery, GovernanceThreadDetail, MemoryProvider, ListMemoriesInput, MemoryRecord, ForgetMemoryInput, StoreMemoryInput, ModelPriceInput, CurrentModelPrice } from '@dudousxd/nestjs-agent-core';
+import { MessageRole, ToolCallRequest, ToolResult, MessageAttachment, MessageUsage, AgentUiComponent, MessageFeedback, Actor, PageContext, QueuePause, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, ThreadTurnReader, ChatQueueStore, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, ToolCallApprovalState, EnqueueMessageInput, QueuedMessage, QueuedMessagePatch, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, ToolCallOutcome, RecordUsageInput, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, GovernancePageQuery, ToolCallWhere, GovernancePage, ThreadWhere, RunWhere, ApprovalWhere, GovernanceRunDetail, GovernanceThreadDetailQuery, GovernanceThreadDetail, MemoryProvider, ListMemoriesInput, MemoryRecord, ForgetMemoryInput, StoreMemoryInput, ModelPriceInput, CurrentModelPrice, StreamFrameTable, StreamFrameRow, SqlTokenStreamSink, SqlTokenStreamSinkOptions, ConfirmTokenStore, ConfirmTokenClaim } from '@dudousxd/nestjs-agent-core';
 export { ThreadTurnPage, ThreadTurnQuery } from '@dudousxd/nestjs-agent-core';
 import { TablesRelationalConfig, SQL } from 'drizzle-orm';
 import { OnModuleInit, OnModuleDestroy, Type, DynamicModule } from '@nestjs/common';
@@ -2243,6 +2243,213 @@ declare const ragIngestionLog: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
     };
     dialect: "sqlite";
 }>;
+/**
+ * Spent confirm tokens (`defineConfirmedTool`), one row per token, keyed by its SHA-256 — the
+ * primary key is the lock that makes a confirmation single use across replicas. Only the hash, the
+ * actor and the tool are kept, never an argument. `expires_at` is epoch-ms; past it the row is dead
+ * weight for `DrizzleConfirmTokenStore.purgeExpired`.
+ */
+declare const agentConfirmToken: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
+    name: "agent_confirm_token";
+    schema: undefined;
+    columns: {
+        hash: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "hash";
+            tableName: "agent_confirm_token";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: true;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        actorRef: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "actor_ref";
+            tableName: "agent_confirm_token";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        tool: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "tool";
+            tableName: "agent_confirm_token";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        expiresAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "expires_at";
+            tableName: "agent_confirm_token";
+            dataType: "number";
+            columnType: "SQLiteInteger";
+            data: number;
+            driverParam: number;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        createdAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "created_at";
+            tableName: "agent_confirm_token";
+            dataType: "number";
+            columnType: "SQLiteInteger";
+            data: number;
+            driverParam: number;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+    };
+    dialect: "sqlite";
+}>;
+/**
+ * One row per frame of a run's live stream (`DrizzleTokenStreamSink`), numbered per run from 1 with
+ * no gaps. `frame` is one NDJSON line; `null` marks the end, with `error` set when the run failed.
+ * `created_at` (epoch-ms) is what the TTL counts from — a run's LAST row.
+ */
+declare const agentStreamFrame: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
+    name: "agent_stream_frame";
+    schema: undefined;
+    columns: {
+        runId: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "run_id";
+            tableName: "agent_stream_frame";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        seq: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "seq";
+            tableName: "agent_stream_frame";
+            dataType: "number";
+            columnType: "SQLiteInteger";
+            data: number;
+            driverParam: number;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        frame: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "frame";
+            tableName: "agent_stream_frame";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        error: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "error";
+            tableName: "agent_stream_frame";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        createdAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "created_at";
+            tableName: "agent_stream_frame";
+            dataType: "number";
+            columnType: "SQLiteInteger";
+            data: number;
+            driverParam: number;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+    };
+    dialect: "sqlite";
+}>;
 /** The tables this store owns, as one schema object, ready for `drizzle(client, { schema })`. */
 declare const agentSchema: {
     agentThread: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
@@ -4401,6 +4608,104 @@ declare const agentSchema: {
         };
         dialect: "sqlite";
     }>;
+    agentConfirmToken: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
+        name: "agent_confirm_token";
+        schema: undefined;
+        columns: {
+            hash: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "hash";
+                tableName: "agent_confirm_token";
+                dataType: "string";
+                columnType: "SQLiteText";
+                data: string;
+                driverParam: string;
+                notNull: true;
+                hasDefault: false;
+                isPrimaryKey: true;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: [string, ...string[]];
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                length: number | undefined;
+            }>;
+            actorRef: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "actor_ref";
+                tableName: "agent_confirm_token";
+                dataType: "string";
+                columnType: "SQLiteText";
+                data: string;
+                driverParam: string;
+                notNull: true;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: [string, ...string[]];
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                length: number | undefined;
+            }>;
+            tool: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "tool";
+                tableName: "agent_confirm_token";
+                dataType: "string";
+                columnType: "SQLiteText";
+                data: string;
+                driverParam: string;
+                notNull: true;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: [string, ...string[]];
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                length: number | undefined;
+            }>;
+            expiresAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "expires_at";
+                tableName: "agent_confirm_token";
+                dataType: "number";
+                columnType: "SQLiteInteger";
+                data: number;
+                driverParam: number;
+                notNull: true;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {}>;
+            createdAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "created_at";
+                tableName: "agent_confirm_token";
+                dataType: "number";
+                columnType: "SQLiteInteger";
+                data: number;
+                driverParam: number;
+                notNull: true;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {}>;
+        };
+        dialect: "sqlite";
+    }>;
 };
 /**
  * The dialect-portable Drizzle SQLite database handle the store operates on. Accepts any SQLite
@@ -4745,6 +5050,53 @@ declare class DrizzlePricingStore implements AgentPricingStore {
     listCurrentPrices(): Promise<CurrentModelPrice[]>;
 }
 
+/** {@link StreamFrameTable} over `agent_stream_frame`, through a Drizzle SQLite handle. */
+declare class DrizzleStreamFrameTable implements StreamFrameTable {
+    private readonly db;
+    constructor(db: AgentDrizzleDb);
+    append(runId: string, row: {
+        frame: string | null;
+        error: string | null;
+        createdAt: number;
+    }): Promise<void>;
+    read(runId: string, after: number, limit: number): Promise<StreamFrameRow[]>;
+    has(runId: string): Promise<boolean>;
+    remove(runIds: readonly string[]): Promise<void>;
+    lapsedRuns(cutoff: number): Promise<string[]>;
+    isUniqueViolation(error: unknown): boolean;
+}
+/**
+ * A multi-replica `TokenStreamSink` over the app's Drizzle database — several replicas, no Redis.
+ * Polled, with streamed text coalesced into one row per `flushMs`; see `SqlTokenStreamSink` for the
+ * trade-offs. The `agent_stream_frame` table comes from `ensureAgentSchema` (or your migration).
+ *
+ * ```ts
+ * AgentModule.forRoot({ model, sink: new DrizzleTokenStreamSink(db) });
+ * ```
+ */
+declare class DrizzleTokenStreamSink extends SqlTokenStreamSink {
+    constructor(db: AgentDrizzleDb, options?: SqlTokenStreamSinkOptions);
+}
+
+/**
+ * A production `ConfirmTokenStore` (`defineConfirmedTool`) over the `agent_confirm_token` table —
+ * what makes a confirmed write single use across replicas.
+ *
+ * `claim` is one `INSERT … ON CONFLICT DO NOTHING RETURNING`, so the primary key is the lock: of two
+ * concurrent confirmations one inserts and the other touches no row. A statement that does not FAIL
+ * on the duplicate, rather than an insert whose violation is caught, so it never aborts a
+ * surrounding transaction. Only the token's SHA-256, the actor and the tool name are stored; call
+ * {@link purgeExpired} from a scheduled job to drop the dead rows. The table comes from
+ * `ensureAgentSchema` (or your own migration).
+ */
+declare class DrizzleConfirmTokenStore implements ConfirmTokenStore {
+    private readonly db;
+    constructor(db: AgentDrizzleDb);
+    claim(input: ConfirmTokenClaim): Promise<boolean>;
+    release(hash: string): Promise<void>;
+    purgeExpired(now?: number): Promise<number>;
+}
+
 /**
  * The paging order: newest first, tiebroken on the primary key.
  *
@@ -4956,7 +5308,8 @@ interface DrizzleAgentStoreModuleOptions {
  * Binds {@link DrizzleAgentStore} to the {@link AGENT_STORE} token consumed by
  * `@dudousxd/nestjs-agent`, plus {@link DrizzleGovernanceQueries} to
  * {@link AGENT_GOVERNANCE_QUERIES} (the read-model the dashboard/telescope surfaces consume) and
- * {@link DrizzlePricingStore} to {@link AGENT_PRICING_STORE} (the write side of that pricing table).
+ * {@link DrizzlePricingStore} to {@link AGENT_PRICING_STORE} (the write side of that pricing table),
+ * and {@link DrizzleConfirmTokenStore} to {@link AGENT_CONFIRM_TOKEN_STORE} (confirmed writes).
  * The host app supplies an already-opened Drizzle db — this module never opens a connection itself.
  *
  * The returned module is global, so {@link AGENT_STORE}, {@link AGENT_GOVERNANCE_QUERIES} and
@@ -4973,4 +5326,4 @@ declare class DrizzleAgentStoreModule {
     static forRoot(options: DrizzleAgentStoreModuleOptions): DynamicModule;
 }
 
-export { type AgentDrizzleDb, type AgentMemoryRow, type AgentMessageRow, type AgentQueuedMessageRow, type AgentRunRow, type AgentRunStatus, type AgentThreadRow, DrizzleAgentStore, DrizzleAgentStoreModule, type DrizzleAgentStoreModuleOptions, DrizzleGovernanceQueries, DrizzleMemoryProvider, DrizzlePricingStore, DrizzleRagIngestionLog, RAG_INGESTION_LOG_PAGE_ORDER, type RagIngestionLogCursor, type RagIngestionLogIterateOptions, type RagIngestionLogQuery, type RagIngestionLogRow, type RagIngestionLogWhere, type RagIngestionStatus, agentMemory, agentMessage, agentModelPricing, agentQueuedMessage, agentRun, agentSchema, agentThread, agentTokenUsage, agentToolCall, ensureAgentSchema, ragIngestionLog };
+export { type AgentDrizzleDb, type AgentMemoryRow, type AgentMessageRow, type AgentQueuedMessageRow, type AgentRunRow, type AgentRunStatus, type AgentThreadRow, DrizzleAgentStore, DrizzleAgentStoreModule, type DrizzleAgentStoreModuleOptions, DrizzleConfirmTokenStore, DrizzleGovernanceQueries, DrizzleMemoryProvider, DrizzlePricingStore, DrizzleRagIngestionLog, DrizzleStreamFrameTable, DrizzleTokenStreamSink, RAG_INGESTION_LOG_PAGE_ORDER, type RagIngestionLogCursor, type RagIngestionLogIterateOptions, type RagIngestionLogQuery, type RagIngestionLogRow, type RagIngestionLogWhere, type RagIngestionStatus, agentConfirmToken, agentMemory, agentMessage, agentModelPricing, agentQueuedMessage, agentRun, agentSchema, agentStreamFrame, agentThread, agentTokenUsage, agentToolCall, ensureAgentSchema, ragIngestionLog };