npm - @dudousxd/nestjs-agent-store-drizzle - Versions diffs - 0.3.4 → 0.5.0 - Mend

@dudousxd/nestjs-agent-store-drizzle 0.3.4 → 0.5.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (7) hide show

package/dist/index.d.cts CHANGED Viewed

@@ -1,11 +1,11 @@
 import * as drizzle_orm_sqlite_core from 'drizzle-orm/sqlite-core';
 import { BaseSQLiteDatabase } from 'drizzle-orm/sqlite-core';
-import { MessageRole, ToolCallRequest, ToolResult, MessageUsage, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, CreateThreadInput, ThreadSummary, ThreadDetail, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, RecordUsageInput, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, ModelPriceInput, CurrentModelPrice } from '@dudousxd/nestjs-agent-core';
+import { MessageRole, ToolCallRequest, ToolResult, MessageUsage, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, CreateThreadInput, ThreadSummary, ThreadDetail, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, RecordUsageInput, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, ModelPriceInput, CurrentModelPrice } from '@dudousxd/nestjs-agent-core';
 import { Type, DynamicModule } from '@nestjs/common';
 /**
  * Drizzle schema mirroring the MikroORM agent entities (same logical columns/relations):
- * threads, messages, tool calls, token usage and model pricing. Targets the SQLite core so the
+ * threads, messages, tool calls, token usage, model pricing and run outcomes. Targets the SQLite core so the
  * db-test runs on an in-memory better-sqlite3; timestamps are stored as integer epoch-ms columns
  * (`{ mode: 'timestamp_ms' }`) which Drizzle round-trips to/from JS `Date`, and the model-facing
  * message extras (tool calls/results, follow-ups, usage) live in JSON `text` columns.
@@ -987,7 +987,245 @@ declare const agentModelPricing: drizzle_orm_sqlite_core.SQLiteTableWithColumns<
     };
     dialect: "sqlite";
 }>;
-/** The five agent tables as one schema object, ready for `drizzle(client, { schema })`. */
+/** A run's lifecycle status. Not part of the core SPI (that surfaces `string`) — internal only. */
+type AgentRunStatus = 'running' | 'completed' | 'failed';
+/**
+ * One recorded run (turn) outcome — the durable half of what `aviary:agent:*` diagnostics report
+ * ephemerally. Written by {@link import('./drizzle-agent-store.js').DrizzleAgentStore.recordRunStart}/
+ * `recordRunEnd`/`bumpRunRetries`, read by
+ * {@link import('./drizzle-governance-queries.js').DrizzleGovernanceQueries} for the reliability
+ * surfaces (success rate, error breakdown, duration percentiles, trend, recent runs). `threadId`
+ * cascades on delete, like `agent_message`/`agent_token_usage`.
+ */
+declare const agentRun: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
+    name: "agent_run";
+    schema: undefined;
+    columns: {
+        id: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "id";
+            tableName: "agent_run";
+            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;
+        }>;
+        threadId: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "thread_id";
+            tableName: "agent_run";
+            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;
+        }>;
+        actorRef: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "actor_ref";
+            tableName: "agent_run";
+            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;
+        }>;
+        agentName: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "agent_name";
+            tableName: "agent_run";
+            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;
+        }>;
+        status: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "status";
+            tableName: "agent_run";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: AgentRunStatus;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+            $type: AgentRunStatus;
+        }>;
+        durationMs: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "duration_ms";
+            tableName: "agent_run";
+            dataType: "number";
+            columnType: "SQLiteInteger";
+            data: number;
+            driverParam: number;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        errorCode: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "error_code";
+            tableName: "agent_run";
+            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;
+        }>;
+        errorMessage: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "error_message";
+            tableName: "agent_run";
+            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;
+        }>;
+        retries: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "retries";
+            tableName: "agent_run";
+            dataType: "number";
+            columnType: "SQLiteInteger";
+            data: number;
+            driverParam: number;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        startedAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "started_at";
+            tableName: "agent_run";
+            dataType: "date";
+            columnType: "SQLiteTimestamp";
+            data: Date;
+            driverParam: number;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        settledAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "settled_at";
+            tableName: "agent_run";
+            dataType: "date";
+            columnType: "SQLiteTimestamp";
+            data: Date;
+            driverParam: number;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        promptHash: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "prompt_hash";
+            tableName: "agent_run";
+            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;
+        }>;
+    };
+    dialect: "sqlite";
+}>;
+/** The six agent tables as one schema object, ready for `drizzle(client, { schema })`. */
 declare const agentSchema: {
     agentThread: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         name: "agent_thread";
@@ -1953,6 +2191,234 @@ declare const agentSchema: {
         };
         dialect: "sqlite";
     }>;
+    agentRun: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
+        name: "agent_run";
+        schema: undefined;
+        columns: {
+            id: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "id";
+                tableName: "agent_run";
+                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;
+            }>;
+            threadId: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "thread_id";
+                tableName: "agent_run";
+                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;
+            }>;
+            actorRef: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "actor_ref";
+                tableName: "agent_run";
+                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;
+            }>;
+            agentName: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "agent_name";
+                tableName: "agent_run";
+                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;
+            }>;
+            status: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "status";
+                tableName: "agent_run";
+                dataType: "string";
+                columnType: "SQLiteText";
+                data: AgentRunStatus;
+                driverParam: string;
+                notNull: true;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: [string, ...string[]];
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                length: number | undefined;
+                $type: AgentRunStatus;
+            }>;
+            durationMs: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "duration_ms";
+                tableName: "agent_run";
+                dataType: "number";
+                columnType: "SQLiteInteger";
+                data: number;
+                driverParam: number;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {}>;
+            errorCode: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "error_code";
+                tableName: "agent_run";
+                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;
+            }>;
+            errorMessage: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "error_message";
+                tableName: "agent_run";
+                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;
+            }>;
+            retries: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "retries";
+                tableName: "agent_run";
+                dataType: "number";
+                columnType: "SQLiteInteger";
+                data: number;
+                driverParam: number;
+                notNull: true;
+                hasDefault: true;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {}>;
+            startedAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "started_at";
+                tableName: "agent_run";
+                dataType: "date";
+                columnType: "SQLiteTimestamp";
+                data: Date;
+                driverParam: number;
+                notNull: true;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {}>;
+            settledAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "settled_at";
+                tableName: "agent_run";
+                dataType: "date";
+                columnType: "SQLiteTimestamp";
+                data: Date;
+                driverParam: number;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {}>;
+            promptHash: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "prompt_hash";
+                tableName: "agent_run";
+                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;
+            }>;
+        };
+        dialect: "sqlite";
+    }>;
 };
 /**
  * The dialect-portable Drizzle SQLite database handle the store operates on. Accepts any SQLite
@@ -1964,6 +2430,8 @@ type AgentDrizzleDb = BaseSQLiteDatabase<'sync' | 'async', unknown>;
 type AgentThreadRow = typeof agentThread.$inferSelect;
 /** A persisted message row as Drizzle selects it. */
 type AgentMessageRow = typeof agentMessage.$inferSelect;
+/** A persisted run row as Drizzle selects it. */
+type AgentRunRow = typeof agentRun.$inferSelect;
 /** Runs the `CREATE TABLE IF NOT EXISTS` DDL above against the supplied Drizzle SQLite db. */
 declare function ensureAgentSchema(db: AgentDrizzleDb): Promise<void>;
@@ -1989,6 +2457,24 @@ declare class DrizzleAgentStore implements AgentStore {
     setTitle(threadId: string, title: string): Promise<void>;
     promoteThread(threadId: string): Promise<void>;
     setActiveStream(threadId: string, runId: string | null): Promise<void>;
+    /** Persist the start of a run (turn). Replay-safe: called under a durable localStep. */
+    recordRunStart(run: {
+        runId: string;
+        threadId: string;
+        actorRef: string;
+        agentName?: string;
+        promptHash?: string;
+    }): Promise<void>;
+    /** Settle a run's outcome. A no-op when the run is unknown (mirrors `setTitle`/`updateThread`). */
+    recordRunEnd(end: {
+        runId: string;
+        status: 'completed' | 'failed';
+        durationMs?: number;
+        errorCode?: string;
+        errorMessage?: string;
+    }): Promise<void>;
+    /** Bump the run's llm-step retry counter. Atomic `retries = retries + 1`, no read-modify-write. */
+    bumpRunRetries(runId: string): Promise<void>;
     appendMessage(input: AppendMessageInput): Promise<StoredMessage>;
     truncateFrom(threadId: string, messageId: string): Promise<void>;
     recordToolCall(input: RecordToolCallInput): Promise<void>;
@@ -2027,6 +2513,24 @@ declare class DrizzleGovernanceQueries implements AgentGovernanceQueries {
     usageTrend(range: GovernanceRange): Promise<UsageTrendPoint[]>;
     recentToolCalls(limit: number): Promise<ToolCallActivityRow[]>;
     recentThreads(limit: number): Promise<ThreadActivityRow[]>;
+    private runsInRange;
+    runMetrics(range: GovernanceRange): Promise<RunMetrics>;
+    runsByAgent(range: GovernanceRange): Promise<RunAgentBreakdownRow[]>;
+    runErrors(range: GovernanceRange): Promise<RunErrorBreakdownRow[]>;
+    runTrend(range: GovernanceRange): Promise<RunTrendPoint[]>;
+    recentRuns(limit: number): Promise<RecentRunRow[]>;
+    /**
+     * Tool calls sitting `pending_approval`, oldest first (an inbox drains from the back), joined
+     * through their message to the owning thread for title/actorRef and to the message itself for
+     * `agentName` (null when the message carries none).
+     */
+    pendingApprovals(limit: number): Promise<PendingApprovalRow[]>;
+    /**
+     * Per-tool call/failure/rejection/latency rollup over the range, highest call count first.
+     * `p95ExecutionMs` is computed over calls that recorded a non-null `executionMs` (regardless of
+     * their final status); `null` when none did.
+     */
+    toolStats(range: GovernanceRange): Promise<ToolStatRow[]>;
 }
 /**
@@ -2074,4 +2578,4 @@ declare class DrizzleAgentStoreModule {
     static forRoot(options: DrizzleAgentStoreModuleOptions): DynamicModule;
 }
-export { type AgentDrizzleDb, type AgentMessageRow, type AgentThreadRow, DrizzleAgentStore, DrizzleAgentStoreModule, type DrizzleAgentStoreModuleOptions, DrizzleGovernanceQueries, DrizzlePricingStore, agentMessage, agentModelPricing, agentSchema, agentThread, agentTokenUsage, agentToolCall, ensureAgentSchema };
+export { type AgentDrizzleDb, type AgentMessageRow, type AgentRunRow, type AgentRunStatus, type AgentThreadRow, DrizzleAgentStore, DrizzleAgentStoreModule, type DrizzleAgentStoreModuleOptions, DrizzleGovernanceQueries, DrizzlePricingStore, agentMessage, agentModelPricing, agentRun, agentSchema, agentThread, agentTokenUsage, agentToolCall, ensureAgentSchema };