@dudousxd/nestjs-agent-store-drizzle 0.11.0 → 0.12.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, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, ThreadTurnReader, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, 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, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, ThreadTurnReader, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, ToolCallApprovalState, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, 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';
 export { ThreadTurnPage, ThreadTurnQuery } from '@dudousxd/nestjs-agent-core';
 import { TablesRelationalConfig, SQL } from 'drizzle-orm';
 import { OnModuleInit, OnModuleDestroy, Type, DynamicModule } from '@nestjs/common';
@@ -751,6 +751,78 @@ declare const agentToolCall: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             length: number | undefined;
         }>;
+        approver: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "approver";
+            tableName: "agent_tool_call";
+            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;
+        }>;
+        expiresAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "expires_at";
+            tableName: "agent_tool_call";
+            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;
+        }, {}, {}>;
+        remember: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "remember";
+            tableName: "agent_tool_call";
+            dataType: "boolean";
+            columnType: "SQLiteBoolean";
+            data: boolean;
+            driverParam: number;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        decidedVia: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "decided_via";
+            tableName: "agent_tool_call";
+            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";
 }>;
@@ -2604,6 +2676,78 @@ declare const agentSchema: {
             }, {}, {
                 length: number | undefined;
             }>;
+            approver: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "approver";
+                tableName: "agent_tool_call";
+                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;
+            }>;
+            expiresAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "expires_at";
+                tableName: "agent_tool_call";
+                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;
+            }, {}, {}>;
+            remember: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "remember";
+                tableName: "agent_tool_call";
+                dataType: "boolean";
+                columnType: "SQLiteBoolean";
+                data: boolean;
+                driverParam: number;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {}>;
+            decidedVia: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "decided_via";
+                tableName: "agent_tool_call";
+                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";
     }>;
@@ -3743,6 +3887,15 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader {
     ownerOfThread(threadId: string): Promise<string | null>;
     ownerOfToolCall(toolCallId: string): Promise<string | null>;
     runForToolCall(toolCallId: string): Promise<string | null>;
+    /**
+     * The approval record of every call on `messageIds` that a policy put to a person, grouped by
+     * message. One query over the calls that carry an approver — a thread whose calls never asked
+     * anyone reads nothing back.
+     */
+    private approvalsFor;
+    /** Tools whose approval someone asked to remember in this thread. */
+    rememberedApprovals(threadId: string): Promise<string[]>;
+    toolCallApproval(toolCallId: string): Promise<ToolCallApprovalState | null>;
     ownerOfActiveStream(runId: string): Promise<string | null>;
     setTitle(threadId: string, title: string): Promise<void>;
     promoteThread(threadId: string): Promise<void>;

package/dist/index.d.ts

@@ -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, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, ThreadTurnReader, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, 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, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, ThreadTurnReader, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, ToolCallApprovalState, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, 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';
 export { ThreadTurnPage, ThreadTurnQuery } from '@dudousxd/nestjs-agent-core';
 import { TablesRelationalConfig, SQL } from 'drizzle-orm';
 import { OnModuleInit, OnModuleDestroy, Type, DynamicModule } from '@nestjs/common';
@@ -751,6 +751,78 @@ declare const agentToolCall: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             length: number | undefined;
         }>;
+        approver: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "approver";
+            tableName: "agent_tool_call";
+            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;
+        }>;
+        expiresAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "expires_at";
+            tableName: "agent_tool_call";
+            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;
+        }, {}, {}>;
+        remember: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "remember";
+            tableName: "agent_tool_call";
+            dataType: "boolean";
+            columnType: "SQLiteBoolean";
+            data: boolean;
+            driverParam: number;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        decidedVia: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "decided_via";
+            tableName: "agent_tool_call";
+            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";
 }>;
@@ -2604,6 +2676,78 @@ declare const agentSchema: {
             }, {}, {
                 length: number | undefined;
             }>;
+            approver: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "approver";
+                tableName: "agent_tool_call";
+                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;
+            }>;
+            expiresAt: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "expires_at";
+                tableName: "agent_tool_call";
+                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;
+            }, {}, {}>;
+            remember: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "remember";
+                tableName: "agent_tool_call";
+                dataType: "boolean";
+                columnType: "SQLiteBoolean";
+                data: boolean;
+                driverParam: number;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {}>;
+            decidedVia: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "decided_via";
+                tableName: "agent_tool_call";
+                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";
     }>;
@@ -3743,6 +3887,15 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader {
     ownerOfThread(threadId: string): Promise<string | null>;
     ownerOfToolCall(toolCallId: string): Promise<string | null>;
     runForToolCall(toolCallId: string): Promise<string | null>;
+    /**
+     * The approval record of every call on `messageIds` that a policy put to a person, grouped by
+     * message. One query over the calls that carry an approver — a thread whose calls never asked
+     * anyone reads nothing back.
+     */
+    private approvalsFor;
+    /** Tools whose approval someone asked to remember in this thread. */
+    rememberedApprovals(threadId: string): Promise<string[]>;
+    toolCallApproval(toolCallId: string): Promise<ToolCallApprovalState | null>;
     ownerOfActiveStream(runId: string): Promise<string | null>;
     setTitle(threadId: string, title: string): Promise<void>;
     promoteThread(threadId: string): Promise<void>;

package/dist/index.js

@@ -87,7 +87,19 @@ var agentToolCall = sqliteTable("agent_tool_call", {
     mode: "timestamp_ms"
   }),
   /** The run (turn) this call belongs to, for a trace deep-link; null for a pre-rollout row. */
-  runId: text("run_id")
+  runId: text("run_id"),
+  /** Who had to approve it (`requester` or a role); null for a call no policy put to anyone. */
+  approver: text("approver"),
+  /** When the approval request lapses; null → never. */
+  expiresAt: integer("expires_at", {
+    mode: "timestamp_ms"
+  }),
+  /** The approval covers later calls of this tool in this thread. */
+  remember: integer("remember", {
+    mode: "boolean"
+  }),
+  /** The surface the decision came through (`web`, `slack`, `remembered`, …). */
+  decidedVia: text("decided_via")
 }, (table) => [
   index("agent_tool_call_message_idx").on(table.messageId)
 ]);
@@ -273,7 +285,11 @@ var statements = [
     error TEXT,
     created_at INTEGER NOT NULL,
     executed_at INTEGER,
-    run_id TEXT
+    run_id TEXT,
+    approver TEXT,
+    expires_at INTEGER,
+    remember INTEGER,
+    decided_via TEXT
   )`,
   `CREATE INDEX IF NOT EXISTS agent_tool_call_message_idx
     ON agent_tool_call (message_id)`,
@@ -390,6 +406,26 @@ var additiveColumns = [
     table: "agent_message",
     column: "ui",
     ddl: "ALTER TABLE agent_message ADD COLUMN ui TEXT"
+  },
+  {
+    table: "agent_tool_call",
+    column: "approver",
+    ddl: "ALTER TABLE agent_tool_call ADD COLUMN approver TEXT"
+  },
+  {
+    table: "agent_tool_call",
+    column: "expires_at",
+    ddl: "ALTER TABLE agent_tool_call ADD COLUMN expires_at INTEGER"
+  },
+  {
+    table: "agent_tool_call",
+    column: "remember",
+    ddl: "ALTER TABLE agent_tool_call ADD COLUMN remember INTEGER"
+  },
+  {
+    table: "agent_tool_call",
+    column: "decided_via",
+    ddl: "ALTER TABLE agent_tool_call ADD COLUMN decided_via TEXT"
   }
 ];
 async function ensureAgentSchema(db) {
@@ -406,6 +442,7 @@ async function ensureAgentSchema(db) {
 __name(ensureAgentSchema, "ensureAgentSchema");
 
 // src/drizzle-agent-store.ts
+import { toolCallApprovalFromRow } from "@dudousxd/nestjs-agent-core";
 import { and, asc, desc, eq, gte, inArray, isNotNull, isNull, lte, sql as sql2 } from "drizzle-orm";
 var TURN_MESSAGE_COLUMNS = {
   id: agentMessage.id,
@@ -449,9 +486,17 @@ var DrizzleAgentStore = class {
     }
     const messages = await this.db.select().from(agentMessage).where(eq(agentMessage.threadId, threadId)).orderBy(asc(agentMessage.createdAt), asc(agentMessage.id));
     const last = messages[messages.length - 1];
+    const approvals = await this.approvalsFor(messages.map((message) => message.id));
     return {
       ...this.toSummary(thread, last?.content),
-      messages: messages.map((message) => this.toStoredMessage(message)),
+      messages: messages.map((message) => {
+        const stored = this.toStoredMessage(message);
+        const onMessage = approvals.get(message.id);
+        return onMessage !== void 0 ? {
+          ...stored,
+          approvals: onMessage
+        } : stored;
+      }),
       ...thread.activeStreamId != null ? {
         activeStreamId: thread.activeStreamId
       } : {}
@@ -580,6 +625,63 @@ var DrizzleAgentStore = class {
     }).from(agentToolCall).innerJoin(agentMessage, eq(agentToolCall.messageId, agentMessage.id)).innerJoin(agentThread, eq(agentMessage.threadId, agentThread.id)).where(eq(agentToolCall.id, toolCallId));
     return row?.runId ?? row?.activeStreamId ?? null;
   }
+  /**
+  * The approval record of every call on `messageIds` that a policy put to a person, grouped by
+  * message. One query over the calls that carry an approver — a thread whose calls never asked
+  * anyone reads nothing back.
+  */
+  async approvalsFor(messageIds) {
+    const byMessage = /* @__PURE__ */ new Map();
+    if (messageIds.length === 0) {
+      return byMessage;
+    }
+    const rows = await this.db.select({
+      id: agentToolCall.id,
+      messageId: agentToolCall.messageId,
+      status: agentToolCall.status,
+      approver: agentToolCall.approver,
+      expiresAt: agentToolCall.expiresAt,
+      remember: agentToolCall.remember,
+      executedByRef: agentToolCall.executedByRef,
+      decidedVia: agentToolCall.decidedVia,
+      error: agentToolCall.error
+    }).from(agentToolCall).where(and(inArray(agentToolCall.messageId, messageIds), isNotNull(agentToolCall.approver))).orderBy(asc(agentToolCall.createdAt), asc(agentToolCall.id));
+    for (const row of rows) {
+      const approval = toolCallApprovalFromRow({
+        ...row,
+        toolCallId: row.id
+      });
+      if (approval === null) {
+        continue;
+      }
+      const list = byMessage.get(row.messageId) ?? [];
+      list.push(approval);
+      byMessage.set(row.messageId, list);
+    }
+    return byMessage;
+  }
+  /** Tools whose approval someone asked to remember in this thread. */
+  async rememberedApprovals(threadId) {
+    const rows = await this.db.selectDistinct({
+      toolName: agentToolCall.toolName
+    }).from(agentToolCall).innerJoin(agentMessage, eq(agentToolCall.messageId, agentMessage.id)).where(and(eq(agentMessage.threadId, threadId), eq(agentToolCall.remember, true)));
+    return rows.map((row) => row.toolName);
+  }
+  async toolCallApproval(toolCallId) {
+    const [row] = await this.db.select({
+      status: agentToolCall.status,
+      approver: agentToolCall.approver,
+      expiresAt: agentToolCall.expiresAt
+    }).from(agentToolCall).where(eq(agentToolCall.id, toolCallId));
+    if (row === void 0) {
+      return null;
+    }
+    return {
+      status: row.status,
+      approver: row.approver,
+      expiresAt: row.expiresAt?.toISOString() ?? null
+    };
+  }
   async ownerOfActiveStream(runId) {
     const [thread] = await this.db.select({
       actorRef: agentThread.actorRef
@@ -739,7 +841,11 @@ var DrizzleAgentStore = class {
       error: null,
       createdAt: /* @__PURE__ */ new Date(),
       executedAt: null,
-      runId: input.runId ?? null
+      runId: input.runId ?? null,
+      approver: input.approver ?? null,
+      expiresAt: input.expiresAt !== void 0 ? new Date(input.expiresAt) : null,
+      remember: null,
+      decidedVia: null
     });
   }
   async updateToolCall(input) {
@@ -758,6 +864,12 @@ var DrizzleAgentStore = class {
     if (input.executedByRef !== void 0) {
       updates.executedByRef = input.executedByRef;
     }
+    if (input.remember !== void 0) {
+      updates.remember = input.remember;
+    }
+    if (input.decidedVia !== void 0) {
+      updates.decidedVia = input.decidedVia;
+    }
     if (input.status === "executed" || input.status === "auto_executed") {
       updates.executedAt = /* @__PURE__ */ new Date();
     }
@@ -886,7 +998,8 @@ var TOOL_CALL_STATUSES = [
   "pending_approval",
   "executed",
   "rejected",
-  "failed"
+  "failed",
+  "expired"
 ];
 function isToolCallStatus(value) {
   return TOOL_CALL_STATUSES.includes(value);