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

package/dist/index.cjs

@@ -707,6 +707,12 @@ var DrizzleAgentStore = class {
     }).from(agentToolCall).innerJoin(agentMessage, (0, import_drizzle_orm2.eq)(agentToolCall.messageId, agentMessage.id)).where((0, import_drizzle_orm2.and)((0, import_drizzle_orm2.eq)(agentMessage.threadId, threadId), (0, import_drizzle_orm2.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((0, import_drizzle_orm2.eq)(agentToolCall.id, toolCallId));
+    return row?.input ?? null;
+  }
   async toolCallApproval(toolCallId) {
     const [row] = await this.db.select({
       status: agentToolCall.status,
@@ -857,6 +863,11 @@ var DrizzleAgentStore = class {
       toolResults: results
     }).where((0, import_drizzle_orm2.eq)(agentMessage.id, messageId));
   }
+  async setMessageUi(messageId, ui) {
+    await this.db.update(agentMessage).set({
+      ui: ui.length > 0 ? ui : null
+    }).where((0, import_drizzle_orm2.eq)(agentMessage.id, messageId));
+  }
   async truncateFrom(threadId, messageId) {
     const messages = await this.db.select().from(agentMessage).where((0, import_drizzle_orm2.eq)(agentMessage.threadId, threadId)).orderBy((0, import_drizzle_orm2.asc)(agentMessage.createdAt), (0, import_drizzle_orm2.asc)(agentMessage.id));
     const cutoff = messages.findIndex((message) => message.id === messageId);