@dudousxd/nestjs-agent-store-drizzle 0.10.8 → 0.11.1

package/README.md

@@ -116,6 +116,9 @@ ALTER TABLE agent_thread ADD COLUMN default_agent TEXT;
 ALTER TABLE agent_message ADD COLUMN run_id TEXT;
 ALTER TABLE agent_message ADD COLUMN attachments TEXT;
 ALTER TABLE agent_run ADD COLUMN parent_run_id TEXT;
+ALTER TABLE agent_message ADD COLUMN reasoning TEXT;
+ALTER TABLE agent_message ADD COLUMN reasoning_ms INTEGER;
+ALTER TABLE agent_message ADD COLUMN ui TEXT;
 CREATE INDEX agent_tool_call_message_idx ON agent_tool_call (message_id);
 ```
 

package/dist/index.cjs

@@ -90,6 +90,13 @@ var agentMessage = (0, import_sqlite_core.sqliteTable)("agent_message", {
   }).$type(),
   agentName: (0, import_sqlite_core.text)("agent_name"),
   runId: (0, import_sqlite_core.text)("run_id"),
+  /** The step's streamed thinking; `null` when the model produced none. */
+  reasoning: (0, import_sqlite_core.text)("reasoning"),
+  reasoningMs: (0, import_sqlite_core.integer)("reasoning_ms"),
+  /** Components pushed during the step (`ui` stream frames), replayed on reload. */
+  ui: (0, import_sqlite_core.text)("ui", {
+    mode: "json"
+  }).$type(),
   createdAt: (0, import_sqlite_core.integer)("created_at", {
     mode: "timestamp_ms"
   }).notNull()
@@ -286,6 +293,9 @@ var statements = [
     usage TEXT,
     agent_name TEXT,
     run_id TEXT,
+    reasoning TEXT,
+    reasoning_ms INTEGER,
+    ui TEXT,
     created_at INTEGER NOT NULL
   )`,
   `CREATE INDEX IF NOT EXISTS agent_message_thread_created_idx
@@ -405,6 +415,21 @@ var additiveColumns = [
     table: "agent_run",
     column: "parent_run_id",
     ddl: "ALTER TABLE agent_run ADD COLUMN parent_run_id TEXT"
+  },
+  {
+    table: "agent_message",
+    column: "reasoning",
+    ddl: "ALTER TABLE agent_message ADD COLUMN reasoning TEXT"
+  },
+  {
+    table: "agent_message",
+    column: "reasoning_ms",
+    ddl: "ALTER TABLE agent_message ADD COLUMN reasoning_ms INTEGER"
+  },
+  {
+    table: "agent_message",
+    column: "ui",
+    ddl: "ALTER TABLE agent_message ADD COLUMN ui TEXT"
   }
 ];
 async function ensureAgentSchema(db) {
@@ -568,6 +593,9 @@ var DrizzleAgentStore = class {
         // turn produced this text gets the truthful answer; the run's own thread is still the
         // original, so a run-scoped read never picks the fork's rows up.
         runId: message.runId,
+        reasoning: message.reasoning,
+        reasoningMs: message.reasoningMs,
+        ui: message.ui,
         createdAt: message.createdAt
       })));
     }
@@ -711,6 +739,9 @@ var DrizzleAgentStore = class {
       usage: input.usage ?? null,
       agentName: input.agentName ?? null,
       runId: input.runId ?? null,
+      reasoning: input.reasoning ?? null,
+      reasoningMs: input.reasoningMs ?? null,
+      ui: input.ui ?? null,
       createdAt: now
     };
     await this.db.insert(agentMessage).values(message);
@@ -871,6 +902,15 @@ var DrizzleAgentStore = class {
       } : {},
       ...message.runId != null ? {
         runId: message.runId
+      } : {},
+      ...message.reasoning != null ? {
+        reasoning: message.reasoning
+      } : {},
+      ...message.reasoningMs != null ? {
+        reasoningMs: message.reasoningMs
+      } : {},
+      ...message.ui != null ? {
+        ui: message.ui
       } : {}
     };
   }