@dudousxd/nestjs-agent-store-drizzle 0.19.0 → 0.20.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, 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 } 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';
@@ -4537,6 +4537,8 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader, ChatQue
     truncateFrom(threadId: string, messageId: string): Promise<void>;
     recordToolCall(input: RecordToolCallInput): Promise<void>;
     updateToolCall(input: UpdateToolCallInput): Promise<void>;
+    toolCallOutcomes(toolCallIds: readonly string[]): Promise<ToolCallOutcome[]>;
+    failUnsettledToolCalls(runId: string, error: string): Promise<number>;
     /**
      * Of `mediaIds`, the ones a surviving message in one of this actor's threads still carries.
      *

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, 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, 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 } 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';
@@ -4537,6 +4537,8 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader, ChatQue
     truncateFrom(threadId: string, messageId: string): Promise<void>;
     recordToolCall(input: RecordToolCallInput): Promise<void>;
     updateToolCall(input: UpdateToolCallInput): Promise<void>;
+    toolCallOutcomes(toolCallIds: readonly string[]): Promise<ToolCallOutcome[]>;
+    failUnsettledToolCalls(runId: string, error: string): Promise<number>;
     /**
      * Of `mediaIds`, the ones a surviving message in one of this actor's threads still carries.
      *

package/dist/index.js

@@ -863,6 +863,9 @@ var DrizzleAgentStore = class {
     if (patch.attachments !== void 0) {
       updates.attachments = patch.attachments === null || patch.attachments.length === 0 ? null : patch.attachments;
     }
+    if (patch.interrupt !== void 0) {
+      updates.interrupt = patch.interrupt;
+    }
     await this.db.update(agentQueuedMessage).set(updates).where(eq(agentQueuedMessage.id, id));
     return this.getQueuedMessage(id);
   }
@@ -1107,6 +1110,43 @@ var DrizzleAgentStore = class {
     }
     await this.db.update(agentToolCall).set(updates).where(eq(agentToolCall.id, input.toolCallId));
   }
+  async toolCallOutcomes(toolCallIds) {
+    if (toolCallIds.length === 0) {
+      return [];
+    }
+    const rows = await this.db.select({
+      id: agentToolCall.id,
+      status: agentToolCall.status,
+      output: agentToolCall.output,
+      error: agentToolCall.error
+    }).from(agentToolCall).where(inArray(agentToolCall.id, [
+      ...toolCallIds
+    ]));
+    return rows.map((row) => ({
+      id: row.id,
+      status: row.status,
+      ...row.output !== void 0 && row.output !== null ? {
+        output: row.output
+      } : {},
+      ...typeof row.error === "string" ? {
+        error: row.error
+      } : {}
+    }));
+  }
+  async failUnsettledToolCalls(runId, error) {
+    const pending = and(eq(agentToolCall.runId, runId), eq(agentToolCall.status, "pending_approval"));
+    const rows = await this.db.select({
+      id: agentToolCall.id
+    }).from(agentToolCall).where(pending);
+    if (rows.length === 0) {
+      return 0;
+    }
+    await this.db.update(agentToolCall).set({
+      status: "failed",
+      error
+    }).where(pending);
+    return rows.length;
+  }
   /**
   * Of `mediaIds`, the ones a surviving message in one of this actor's threads still carries.
   *