@dudousxd/nestjs-agent-store-drizzle 0.25.0 → 0.27.0

package/dist/index.d.cts

@@ -1,5 +1,5 @@
 import * as _dudousxd_nestjs_agent_core from '@dudousxd/nestjs-agent-core';
-import { MessageRole, ToolCallRequest, ToolResult, MessageAttachment, MessageUsage, AgentUiComponent, MessageFeedback, Actor, PageContext, QueuePause, ActionProposal, ToolKind, ToolCallStatus, ToolConfirmation, UsagePurpose, AgentStore, ThreadTurnReader, ChatQueueStore, ActionProposalStore, ActionProposalWorkerStore, ActionProposalDiscoveryIndexStore, ActionProposalOutcomeStore, ActionProposalStoreOptions, ActionProposalSupersessionStore, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, ToolCallApprovalState, EnqueueMessageInput, QueuedMessage, QueuedMessagePatch, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, ToolCallOutcome, RecordUsageInput, CreateActionProposal, CreateActionProposalResult, ActionProposalScope, ActionProposalMutationResult, ListActionProposals, ActionProposalDecisionCommand, ClaimActionProposal, ExtendActionProposalLease, SettleActionProposal, ActionProposalOutcomeLease, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, GovernancePageQuery, ToolCallWhere, GovernancePage, ThreadWhere, RunWhere, ApprovalWhere, GovernanceRunDetail, GovernanceThreadDetailQuery, GovernanceThreadDetail, MemoryProvider, ListMemoriesInput, MemoryRecord, ForgetMemoryInput, StoreMemoryInput, ModelPriceInput, CurrentModelPrice, StreamFrameTable, StreamFrameRow, SqlTokenStreamSink, SqlTokenStreamSinkOptions, ConfirmTokenStore, ConfirmTokenClaim } from '@dudousxd/nestjs-agent-core';
+import { MessageRole, ToolCallRequest, ToolResult, MessageAttachment, MessageUsage, AgentUiComponent, MessageFeedback, AgentHostContext, Actor, PageContext, QueuePause, ActionProposal, ToolKind, ToolCallStatus, ToolConfirmation, UsagePurpose, AgentStore, ThreadTurnReader, ChatQueueStore, ActionProposalStore, ActionProposalWorkerStore, ActionProposalDiscoveryIndexStore, ActionProposalOutcomeStore, ActionProposalStoreOptions, ActionProposalSupersessionStore, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, ToolCallApprovalState, EnqueueMessageInput, QueuedMessage, QueuedMessagePatch, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, ToolCallOutcome, RecordUsageInput, CreateActionProposal, CreateActionProposalResult, ActionProposalScope, ActionProposalMutationResult, ListActionProposals, ActionProposalDecisionCommand, ClaimActionProposal, ExtendActionProposalLease, SettleActionProposal, ActionProposalOutcomeLease, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, GovernancePageQuery, ToolCallWhere, GovernancePage, ThreadWhere, RunWhere, ApprovalWhere, GovernanceRunDetail, GovernanceThreadDetailQuery, GovernanceThreadDetail, MemoryProvider, ListMemoriesInput, MemoryRecord, ForgetMemoryInput, StoreMemoryInput, ModelPriceInput, CurrentModelPrice, StreamFrameTable, StreamFrameRow, SqlTokenStreamSink, SqlTokenStreamSinkOptions, ConfirmTokenStore, ConfirmTokenClaim } from '@dudousxd/nestjs-agent-core';
 export { ThreadTurnPage, ThreadTurnQuery } from '@dudousxd/nestjs-agent-core';
 import * as drizzle_orm_sqlite_core from 'drizzle-orm/sqlite-core';
 import { BaseSQLiteDatabase } from 'drizzle-orm/sqlite-core';
@@ -1048,6 +1048,25 @@ declare const agentQueuedMessage: drizzle_orm_sqlite_core.SQLiteTableWithColumns
         }, {}, {
             $type: UiCapabilities;
         }>;
+        hostContext: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "host_context";
+            tableName: "agent_queued_message";
+            dataType: "json";
+            columnType: "SQLiteTextJson";
+            data: AgentHostContext;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            $type: AgentHostContext;
+        }>;
         id: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "id";
             tableName: "agent_queued_message";
@@ -3861,6 +3880,25 @@ declare const agentSchema: {
             }, {}, {
                 $type: UiCapabilities;
             }>;
+            hostContext: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "host_context";
+                tableName: "agent_queued_message";
+                dataType: "json";
+                columnType: "SQLiteTextJson";
+                data: AgentHostContext;
+                driverParam: string;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                $type: AgentHostContext;
+            }>;
             id: drizzle_orm_sqlite_core.SQLiteColumn<{
                 name: "id";
                 tableName: "agent_queued_message";
@@ -5770,8 +5808,8 @@ declare function agentSchemaDdl(dialect: AgentDialect): string[];
 
 /**
  * {@link AgentStore} backed by Drizzle ORM — a second adapter alongside the MikroORM one, proving
- * the store SPI is ORM-portable. A POJO receiving a Drizzle SQLite database handle (the host app
- * owns the connection). Behaviour mirrors {@link import('@dudousxd/nestjs-agent-store-mikro-orm')}
+ * the store SPI is ORM-portable. A POJO receiving a Drizzle database handle — SQLite, Postgres or
+ * MySQL (the host app owns the connection). Behaviour mirrors {@link import('@dudousxd/nestjs-agent-store-mikro-orm')}
  * exactly (fork/truncate/quota/active-stream/soft-delete semantics) so the two are interchangeable.
  */
 declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader, ChatQueueStore, ActionProposalStore, ActionProposalWorkerStore, ActionProposalDiscoveryIndexStore, ActionProposalOutcomeStore {
@@ -7373,6 +7411,25 @@ declare const pgAgentSchema: {
             }, {}, {
                 $type: UiCapabilities;
             }>;
+            hostContext: drizzle_orm_pg_core.PgColumn<{
+                name: "host_context";
+                tableName: "agent_queued_message";
+                dataType: "json";
+                columnType: "PgJsonb";
+                data: AgentHostContext;
+                driverParam: unknown;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                $type: AgentHostContext;
+            }>;
             id: drizzle_orm_pg_core.PgColumn<{
                 name: "id";
                 tableName: "agent_queued_message";
@@ -10021,6 +10078,25 @@ declare const mysqlAgentSchema: {
             }, {}, {
                 $type: UiCapabilities;
             }>;
+            hostContext: drizzle_orm_mysql_core.MySqlColumn<{
+                name: "host_context";
+                tableName: "agent_queued_message";
+                dataType: "json";
+                columnType: "MySqlJson";
+                data: AgentHostContext;
+                driverParam: string;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                $type: AgentHostContext;
+            }>;
             id: drizzle_orm_mysql_core.MySqlColumn<{
                 name: string;
                 tableName: "agent_queued_message";

package/dist/index.d.ts

@@ -1,5 +1,5 @@
 import * as _dudousxd_nestjs_agent_core from '@dudousxd/nestjs-agent-core';
-import { MessageRole, ToolCallRequest, ToolResult, MessageAttachment, MessageUsage, AgentUiComponent, MessageFeedback, Actor, PageContext, QueuePause, ActionProposal, ToolKind, ToolCallStatus, ToolConfirmation, UsagePurpose, AgentStore, ThreadTurnReader, ChatQueueStore, ActionProposalStore, ActionProposalWorkerStore, ActionProposalDiscoveryIndexStore, ActionProposalOutcomeStore, ActionProposalStoreOptions, ActionProposalSupersessionStore, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, ToolCallApprovalState, EnqueueMessageInput, QueuedMessage, QueuedMessagePatch, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, ToolCallOutcome, RecordUsageInput, CreateActionProposal, CreateActionProposalResult, ActionProposalScope, ActionProposalMutationResult, ListActionProposals, ActionProposalDecisionCommand, ClaimActionProposal, ExtendActionProposalLease, SettleActionProposal, ActionProposalOutcomeLease, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, GovernancePageQuery, ToolCallWhere, GovernancePage, ThreadWhere, RunWhere, ApprovalWhere, GovernanceRunDetail, GovernanceThreadDetailQuery, GovernanceThreadDetail, MemoryProvider, ListMemoriesInput, MemoryRecord, ForgetMemoryInput, StoreMemoryInput, ModelPriceInput, CurrentModelPrice, StreamFrameTable, StreamFrameRow, SqlTokenStreamSink, SqlTokenStreamSinkOptions, ConfirmTokenStore, ConfirmTokenClaim } from '@dudousxd/nestjs-agent-core';
+import { MessageRole, ToolCallRequest, ToolResult, MessageAttachment, MessageUsage, AgentUiComponent, MessageFeedback, AgentHostContext, Actor, PageContext, QueuePause, ActionProposal, ToolKind, ToolCallStatus, ToolConfirmation, UsagePurpose, AgentStore, ThreadTurnReader, ChatQueueStore, ActionProposalStore, ActionProposalWorkerStore, ActionProposalDiscoveryIndexStore, ActionProposalOutcomeStore, ActionProposalStoreOptions, ActionProposalSupersessionStore, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, ToolCallApprovalState, EnqueueMessageInput, QueuedMessage, QueuedMessagePatch, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, ToolCallOutcome, RecordUsageInput, CreateActionProposal, CreateActionProposalResult, ActionProposalScope, ActionProposalMutationResult, ListActionProposals, ActionProposalDecisionCommand, ClaimActionProposal, ExtendActionProposalLease, SettleActionProposal, ActionProposalOutcomeLease, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, GovernancePageQuery, ToolCallWhere, GovernancePage, ThreadWhere, RunWhere, ApprovalWhere, GovernanceRunDetail, GovernanceThreadDetailQuery, GovernanceThreadDetail, MemoryProvider, ListMemoriesInput, MemoryRecord, ForgetMemoryInput, StoreMemoryInput, ModelPriceInput, CurrentModelPrice, StreamFrameTable, StreamFrameRow, SqlTokenStreamSink, SqlTokenStreamSinkOptions, ConfirmTokenStore, ConfirmTokenClaim } from '@dudousxd/nestjs-agent-core';
 export { ThreadTurnPage, ThreadTurnQuery } from '@dudousxd/nestjs-agent-core';
 import * as drizzle_orm_sqlite_core from 'drizzle-orm/sqlite-core';
 import { BaseSQLiteDatabase } from 'drizzle-orm/sqlite-core';
@@ -1048,6 +1048,25 @@ declare const agentQueuedMessage: drizzle_orm_sqlite_core.SQLiteTableWithColumns
         }, {}, {
             $type: UiCapabilities;
         }>;
+        hostContext: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "host_context";
+            tableName: "agent_queued_message";
+            dataType: "json";
+            columnType: "SQLiteTextJson";
+            data: AgentHostContext;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            $type: AgentHostContext;
+        }>;
         id: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "id";
             tableName: "agent_queued_message";
@@ -3861,6 +3880,25 @@ declare const agentSchema: {
             }, {}, {
                 $type: UiCapabilities;
             }>;
+            hostContext: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "host_context";
+                tableName: "agent_queued_message";
+                dataType: "json";
+                columnType: "SQLiteTextJson";
+                data: AgentHostContext;
+                driverParam: string;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                $type: AgentHostContext;
+            }>;
             id: drizzle_orm_sqlite_core.SQLiteColumn<{
                 name: "id";
                 tableName: "agent_queued_message";
@@ -5770,8 +5808,8 @@ declare function agentSchemaDdl(dialect: AgentDialect): string[];
 
 /**
  * {@link AgentStore} backed by Drizzle ORM — a second adapter alongside the MikroORM one, proving
- * the store SPI is ORM-portable. A POJO receiving a Drizzle SQLite database handle (the host app
- * owns the connection). Behaviour mirrors {@link import('@dudousxd/nestjs-agent-store-mikro-orm')}
+ * the store SPI is ORM-portable. A POJO receiving a Drizzle database handle — SQLite, Postgres or
+ * MySQL (the host app owns the connection). Behaviour mirrors {@link import('@dudousxd/nestjs-agent-store-mikro-orm')}
  * exactly (fork/truncate/quota/active-stream/soft-delete semantics) so the two are interchangeable.
  */
 declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader, ChatQueueStore, ActionProposalStore, ActionProposalWorkerStore, ActionProposalDiscoveryIndexStore, ActionProposalOutcomeStore {
@@ -7373,6 +7411,25 @@ declare const pgAgentSchema: {
             }, {}, {
                 $type: UiCapabilities;
             }>;
+            hostContext: drizzle_orm_pg_core.PgColumn<{
+                name: "host_context";
+                tableName: "agent_queued_message";
+                dataType: "json";
+                columnType: "PgJsonb";
+                data: AgentHostContext;
+                driverParam: unknown;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                $type: AgentHostContext;
+            }>;
             id: drizzle_orm_pg_core.PgColumn<{
                 name: "id";
                 tableName: "agent_queued_message";
@@ -10021,6 +10078,25 @@ declare const mysqlAgentSchema: {
             }, {}, {
                 $type: UiCapabilities;
             }>;
+            hostContext: drizzle_orm_mysql_core.MySqlColumn<{
+                name: "host_context";
+                tableName: "agent_queued_message";
+                dataType: "json";
+                columnType: "MySqlJson";
+                data: AgentHostContext;
+                driverParam: string;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                $type: AgentHostContext;
+            }>;
             id: drizzle_orm_mysql_core.MySqlColumn<{
                 name: string;
                 tableName: "agent_queued_message";

package/dist/index.js

@@ -133,6 +133,9 @@ var agentQueuedMessage = sqliteTable("agent_queued_message", {
   uiCapabilities: text("ui_capabilities", {
     mode: "json"
   }).$type(),
+  hostContext: text("host_context", {
+    mode: "json"
+  }).$type(),
   id: text("id").primaryKey(),
   threadId: text("thread_id").notNull().references(() => agentThread.id, {
     onDelete: "cascade"
@@ -444,6 +447,7 @@ var agentToolCall2 = mysqlTable("agent_tool_call", {
 ]);
 var agentQueuedMessage2 = mysqlTable("agent_queued_message", {
   uiCapabilities: json("ui_capabilities").$type(),
+  hostContext: json("host_context").$type(),
   id: key("id").primaryKey(),
   threadId: key("thread_id").notNull().references(() => agentThread2.id, {
     onDelete: "cascade"
@@ -711,6 +715,7 @@ var agentToolCall3 = pgTable("agent_tool_call", {
 ]);
 var agentQueuedMessage3 = pgTable("agent_queued_message", {
   uiCapabilities: jsonb("ui_capabilities").$type(),
+  hostContext: jsonb("host_context").$type(),
   id: text2("id").primaryKey(),
   threadId: text2("thread_id").notNull().references(() => agentThread3.id, {
     onDelete: "cascade"
@@ -1149,6 +1154,8 @@ function actionProposalTransactionMode(kind) {
   if (kind !== void 0 && [
     "NodePgDatabase",
     "NodePgTransaction",
+    "PostgresJsDatabase",
+    "PostgresJsTransaction",
     "MySql2Database",
     "MySql2Transaction",
     "LibSQLDatabase",
@@ -2067,6 +2074,7 @@ var DrizzleAgentStore = class {
       actor: input.actor,
       content: input.content,
       uiCapabilities: input.uiCapabilities ?? null,
+      hostContext: input.hostContext ?? null,
       attachments: input.attachments !== void 0 && input.attachments.length > 0 ? input.attachments : null,
       agentName: input.agentName ?? null,
       persona: input.persona ?? null,
@@ -2534,6 +2542,9 @@ function toQueuedMessage(row) {
     ...row.uiCapabilities !== null ? {
       uiCapabilities: row.uiCapabilities
     } : {},
+    ...row.hostContext != null ? {
+      hostContext: row.hostContext
+    } : {},
     id: row.id,
     threadId: row.threadId,
     actor: row.actor,