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

package/dist/index.d.cts

@@ -3933,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

@@ -3933,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

@@ -823,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);