@dudousxd/nestjs-agent-store-drizzle 0.12.0 → 0.13.0

package/dist/index.d.cts

@@ -3895,6 +3895,7 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader {
     private approvalsFor;
     /** Tools whose approval someone asked to remember in this thread. */
     rememberedApprovals(threadId: string): Promise<string[]>;
+    toolCallInput(toolCallId: string): Promise<unknown>;
     toolCallApproval(toolCallId: string): Promise<ToolCallApprovalState | null>;
     ownerOfActiveStream(runId: string): Promise<string | null>;
     setTitle(threadId: string, title: string): Promise<void>;

package/dist/index.d.ts

@@ -3895,6 +3895,7 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader {
     private approvalsFor;
     /** Tools whose approval someone asked to remember in this thread. */
     rememberedApprovals(threadId: string): Promise<string[]>;
+    toolCallInput(toolCallId: string): Promise<unknown>;
     toolCallApproval(toolCallId: string): Promise<ToolCallApprovalState | null>;
     ownerOfActiveStream(runId: string): Promise<string | null>;
     setTitle(threadId: string, title: string): Promise<void>;

package/dist/index.js

@@ -667,6 +667,12 @@ var DrizzleAgentStore = class {
     }).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 toolCallInput(toolCallId) {
+    const [row] = await this.db.select({
+      input: agentToolCall.input
+    }).from(agentToolCall).where(eq(agentToolCall.id, toolCallId));
+    return row?.input ?? null;
+  }
   async toolCallApproval(toolCallId) {
     const [row] = await this.db.select({
       status: agentToolCall.status,