@dudousxd/nestjs-agent-store-drizzle 0.12.0 → 0.14.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>;
@@ -3932,6 +3933,7 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader {
     bumpRunRetries(runId: string): Promise<void>;
     appendMessage(input: AppendMessageInput): Promise<StoredMessage>;
     setMessageToolResults(messageId: string, results: ToolResult[]): Promise<void>;
+    setMessageUi(messageId: string, ui: AgentUiComponent[]): Promise<void>;
     truncateFrom(threadId: string, messageId: string): Promise<void>;
     recordToolCall(input: RecordToolCallInput): Promise<void>;
     updateToolCall(input: UpdateToolCallInput): 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>;
@@ -3932,6 +3933,7 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader {
     bumpRunRetries(runId: string): Promise<void>;
     appendMessage(input: AppendMessageInput): Promise<StoredMessage>;
     setMessageToolResults(messageId: string, results: ToolResult[]): Promise<void>;
+    setMessageUi(messageId: string, ui: AgentUiComponent[]): Promise<void>;
     truncateFrom(threadId: string, messageId: string): Promise<void>;
     recordToolCall(input: RecordToolCallInput): Promise<void>;
     updateToolCall(input: UpdateToolCallInput): 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,
@@ -817,6 +823,11 @@ var DrizzleAgentStore = class {
       toolResults: results
     }).where(eq(agentMessage.id, messageId));
   }
+  async setMessageUi(messageId, ui) {
+    await this.db.update(agentMessage).set({
+      ui: ui.length > 0 ? ui : null
+    }).where(eq(agentMessage.id, messageId));
+  }
   async truncateFrom(threadId, messageId) {
     const messages = await this.db.select().from(agentMessage).where(eq(agentMessage.threadId, threadId)).orderBy(asc(agentMessage.createdAt), asc(agentMessage.id));
     const cutoff = messages.findIndex((message) => message.id === messageId);