@dudousxd/nestjs-agent-store-drizzle 0.1.0

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/schema.ts","../src/ensure-schema.ts","../src/drizzle-agent-store.ts","../src/drizzle-governance-queries.ts","../src/drizzle-pricing-store.ts","../src/drizzle-agent-store.module.ts"],"sourcesContent":["import type {\n  MessageRole,\n  MessageUsage,\n  ToolCallRequest,\n  ToolCallStatus,\n  ToolKind,\n  ToolResult,\n  UsagePurpose,\n} from '@dudousxd/nestjs-agent-core';\nimport {\n  type BaseSQLiteDatabase,\n  index,\n  integer,\n  real,\n  sqliteTable,\n  text,\n} from 'drizzle-orm/sqlite-core';\n\n/**\n * Drizzle schema mirroring the MikroORM agent entities (same logical columns/relations):\n * threads, messages, tool calls, token usage and model pricing. Targets the SQLite core so the\n * db-test runs on an in-memory better-sqlite3; timestamps are stored as integer epoch-ms columns\n * (`{ mode: 'timestamp_ms' }`) which Drizzle round-trips to/from JS `Date`, and the model-facing\n * message extras (tool calls/results, follow-ups, usage) live in JSON `text` columns.\n */\n\n/**\n * A conversation thread. `deletedAt` drives soft delete: a deleted thread is hidden from\n * {@link import('./drizzle-agent-store.js').DrizzleAgentStore.getThread}/`listThreads` but its\n * row survives.\n */\nexport const agentThread = sqliteTable(\n  'agent_thread',\n  {\n    id: text('id').primaryKey(),\n    actorRef: text('actor_ref').notNull(),\n    tenantRef: text('tenant_ref'),\n    title: text('title').notNull(),\n    persona: text('persona').notNull().default('default'),\n    transient: integer('transient', { mode: 'boolean' }).notNull().default(false),\n    activeStreamId: text('active_stream_id'),\n    createdAt: integer('created_at', { mode: 'timestamp_ms' }).notNull(),\n    updatedAt: integer('updated_at', { mode: 'timestamp_ms' }).notNull(),\n    deletedAt: integer('deleted_at', { mode: 'timestamp_ms' }),\n  },\n  (table) => [index('agent_thread_actor_updated_idx').on(table.actorRef, table.updatedAt)],\n);\n\n/**\n * One stored message in a thread. The model-facing extras (`toolCalls`/`toolResults`/\n * `followUps`/`usage`) are folded into nullable JSON columns so a single row round-trips to a\n * {@link import('@dudousxd/nestjs-agent-core').StoredMessage}. `threadId` cascades on delete.\n */\nexport const agentMessage = sqliteTable(\n  'agent_message',\n  {\n    id: text('id').primaryKey(),\n    threadId: text('thread_id')\n      .notNull()\n      .references(() => agentThread.id, { onDelete: 'cascade' }),\n    role: text('role').$type<MessageRole>().notNull(),\n    content: text('content').notNull(),\n    toolCalls: text('tool_calls', { mode: 'json' }).$type<ToolCallRequest[]>(),\n    toolResults: text('tool_results', { mode: 'json' }).$type<ToolResult[]>(),\n    followUps: text('follow_ups', { mode: 'json' }).$type<string[]>(),\n    usage: text('usage', { mode: 'json' }).$type<MessageUsage>(),\n    persona: text('persona'),\n    createdAt: integer('created_at', { mode: 'timestamp_ms' }).notNull(),\n  },\n  (table) => [index('agent_message_thread_created_idx').on(table.threadId, table.createdAt)],\n);\n\n/** A tool call requested during an assistant turn. The pk is the model-supplied `toolCallId`. */\nexport const agentToolCall = sqliteTable('agent_tool_call', {\n  id: text('id').primaryKey(),\n  messageId: text('message_id')\n    .notNull()\n    .references(() => agentMessage.id, { onDelete: 'cascade' }),\n  toolName: text('tool_name').notNull(),\n  toolType: text('tool_type').$type<ToolKind>().notNull(),\n  input: text('input', { mode: 'json' }),\n  output: text('output', { mode: 'json' }),\n  status: text('status').$type<ToolCallStatus>().notNull(),\n  executedByRef: text('executed_by_ref'),\n  executionMs: integer('execution_ms'),\n  error: text('error'),\n  createdAt: integer('created_at', { mode: 'timestamp_ms' }).notNull(),\n  executedAt: integer('executed_at', { mode: 'timestamp_ms' }),\n});\n\n/** A token-usage ledger row, summed per actor per day by `quotaToday`. */\nexport const agentTokenUsage = sqliteTable(\n  'agent_token_usage',\n  {\n    id: text('id').primaryKey(),\n    threadId: text('thread_id')\n      .notNull()\n      .references(() => agentThread.id, { onDelete: 'cascade' }),\n    actorRef: text('actor_ref').notNull(),\n    messageId: text('message_id'),\n    modelId: text('model_id').notNull(),\n    purpose: text('purpose').$type<UsagePurpose>().notNull(),\n    inputTokens: integer('input_tokens').notNull(),\n    outputTokens: integer('output_tokens').notNull(),\n    /** Subset of `inputTokens` written to the prompt cache this turn; null when not reported. */\n    cacheWriteTokens: integer('cache_write_tokens'),\n    /** Subset of `inputTokens` served from the prompt cache this turn; null when not reported. */\n    cacheReadTokens: integer('cache_read_tokens'),\n    /** Provider-reported actual USD cost for the turn; null when only tokens were reported. */\n    costUsd: real('cost_usd'),\n    createdAt: integer('created_at', { mode: 'timestamp_ms' }).notNull(),\n  },\n  (table) => [index('agent_token_usage_actor_created_idx').on(table.actorRef, table.createdAt)],\n);\n\n/** Per-model token pricing, versioned by `effectiveFrom`; `isCurrent` flags the live row. */\nexport const agentModelPricing = sqliteTable('agent_model_pricing', {\n  id: text('id').primaryKey(),\n  modelId: text('model_id').notNull(),\n  inputPricePer1m: real('input_price_per_1m').notNull(),\n  outputPricePer1m: real('output_price_per_1m').notNull(),\n  /** Per-1M price for cache-write input tokens; null → price them at the input rate. */\n  cacheWritePricePer1m: real('cache_write_price_per_1m'),\n  /** Per-1M price for cache-read input tokens; null → price them at the input rate. */\n  cacheReadPricePer1m: real('cache_read_price_per_1m'),\n  effectiveFrom: integer('effective_from', { mode: 'timestamp_ms' }).notNull(),\n  isCurrent: integer('is_current', { mode: 'boolean' }).notNull(),\n});\n\n/** The five agent tables as one schema object, ready for `drizzle(client, { schema })`. */\nexport const agentSchema = {\n  agentThread,\n  agentMessage,\n  agentToolCall,\n  agentTokenUsage,\n  agentModelPricing,\n};\n\n/**\n * The dialect-portable Drizzle SQLite database handle the store operates on. Accepts any SQLite\n * driver (better-sqlite3, libsql, D1, …) in either sync or async result mode; the host app owns\n * the connection and passes the `drizzle(...)` instance in.\n */\nexport type AgentDrizzleDb = BaseSQLiteDatabase<'sync' | 'async', unknown>;\n\n/** A persisted thread row as Drizzle selects it. */\nexport type AgentThreadRow = typeof agentThread.$inferSelect;\n/** A persisted message row as Drizzle selects it. */\nexport type AgentMessageRow = typeof agentMessage.$inferSelect;\n","import { sql } from 'drizzle-orm';\nimport type { AgentDrizzleDb } from './schema.js';\n\n/**\n * Idempotent `CREATE TABLE IF NOT EXISTS` DDL for the five agent tables, mirroring\n * {@link import('./schema.js').agentSchema}. SQLite dialect (the db-test driver); kept here rather\n * than relying on drizzle-kit so the package can stand up its schema with no migration files. Safe\n * to run on boot against a shared database — it never drops or alters existing columns.\n */\nconst statements: string[] = [\n  `CREATE TABLE IF NOT EXISTS agent_thread (\n    id TEXT PRIMARY KEY NOT NULL,\n    actor_ref TEXT NOT NULL,\n    tenant_ref TEXT,\n    title TEXT NOT NULL,\n    persona TEXT NOT NULL DEFAULT 'default',\n    transient INTEGER NOT NULL DEFAULT 0,\n    active_stream_id TEXT,\n    created_at INTEGER NOT NULL,\n    updated_at INTEGER NOT NULL,\n    deleted_at INTEGER\n  )`,\n  `CREATE INDEX IF NOT EXISTS agent_thread_actor_updated_idx\n    ON agent_thread (actor_ref, updated_at)`,\n  `CREATE TABLE IF NOT EXISTS agent_message (\n    id TEXT PRIMARY KEY NOT NULL,\n    thread_id TEXT NOT NULL REFERENCES agent_thread(id) ON DELETE CASCADE,\n    role TEXT NOT NULL,\n    content TEXT NOT NULL,\n    tool_calls TEXT,\n    tool_results TEXT,\n    follow_ups TEXT,\n    usage TEXT,\n    persona TEXT,\n    created_at INTEGER NOT NULL\n  )`,\n  `CREATE INDEX IF NOT EXISTS agent_message_thread_created_idx\n    ON agent_message (thread_id, created_at)`,\n  `CREATE TABLE IF NOT EXISTS agent_tool_call (\n    id TEXT PRIMARY KEY NOT NULL,\n    message_id TEXT NOT NULL REFERENCES agent_message(id) ON DELETE CASCADE,\n    tool_name TEXT NOT NULL,\n    tool_type TEXT NOT NULL,\n    input TEXT,\n    output TEXT,\n    status TEXT NOT NULL,\n    executed_by_ref TEXT,\n    execution_ms INTEGER,\n    error TEXT,\n    created_at INTEGER NOT NULL,\n    executed_at INTEGER\n  )`,\n  `CREATE TABLE IF NOT EXISTS agent_token_usage (\n    id TEXT PRIMARY KEY NOT NULL,\n    thread_id TEXT NOT NULL REFERENCES agent_thread(id) ON DELETE CASCADE,\n    actor_ref TEXT NOT NULL,\n    message_id TEXT,\n    model_id TEXT NOT NULL,\n    purpose TEXT NOT NULL,\n    input_tokens INTEGER NOT NULL,\n    output_tokens INTEGER NOT NULL,\n    cache_write_tokens INTEGER,\n    cache_read_tokens INTEGER,\n    cost_usd REAL,\n    created_at INTEGER NOT NULL\n  )`,\n  `CREATE INDEX IF NOT EXISTS agent_token_usage_actor_created_idx\n    ON agent_token_usage (actor_ref, created_at)`,\n  `CREATE TABLE IF NOT EXISTS agent_model_pricing (\n    id TEXT PRIMARY KEY NOT NULL,\n    model_id TEXT NOT NULL,\n    input_price_per_1m REAL NOT NULL,\n    output_price_per_1m REAL NOT NULL,\n    cache_write_price_per_1m REAL,\n    cache_read_price_per_1m REAL,\n    effective_from INTEGER NOT NULL,\n    is_current INTEGER NOT NULL\n  )`,\n];\n\n/** Runs the `CREATE TABLE IF NOT EXISTS` DDL above against the supplied Drizzle SQLite db. */\nexport async function ensureAgentSchema(db: AgentDrizzleDb): Promise<void> {\n  for (const statement of statements) {\n    await db.run(sql.raw(statement));\n  }\n}\n","import type {\n  AgentStore,\n  AppendMessageInput,\n  CreateThreadInput,\n  RecordToolCallInput,\n  RecordUsageInput,\n  StoredMessage,\n  ThreadDetail,\n  ThreadSummary,\n  UpdateToolCallInput,\n} from '@dudousxd/nestjs-agent-core';\nimport { and, asc, desc, eq, gte, inArray, isNull, lte } from 'drizzle-orm';\nimport {\n  type AgentDrizzleDb,\n  type AgentMessageRow,\n  type AgentThreadRow,\n  agentMessage,\n  agentThread,\n  agentTokenUsage,\n  agentToolCall,\n} from './schema.js';\n\n/**\n * {@link AgentStore} backed by Drizzle ORM — a second adapter alongside the MikroORM one, proving\n * the store SPI is ORM-portable. A POJO receiving a Drizzle SQLite database handle (the host app\n * owns the connection). Behaviour mirrors {@link import('@dudousxd/nestjs-agent-store-mikro-orm')}\n * exactly (fork/truncate/quota/active-stream/soft-delete semantics) so the two are interchangeable.\n */\nexport class DrizzleAgentStore implements AgentStore {\n  constructor(private readonly db: AgentDrizzleDb) {}\n\n  async createThread(input: CreateThreadInput): Promise<ThreadSummary> {\n    const now = new Date();\n    const thread: AgentThreadRow = {\n      id: crypto.randomUUID(),\n      actorRef: input.actor.id,\n      tenantRef: input.actor.tenantRef ?? null,\n      title: input.title ?? 'New chat',\n      persona: input.persona,\n      transient: input.transient ?? false,\n      activeStreamId: null,\n      createdAt: now,\n      updatedAt: now,\n      deletedAt: null,\n    };\n    await this.db.insert(agentThread).values(thread);\n    return this.toSummary(thread);\n  }\n\n  async getThread(threadId: string): Promise<ThreadDetail | null> {\n    const [thread] = await this.db\n      .select()\n      .from(agentThread)\n      .where(and(eq(agentThread.id, threadId), isNull(agentThread.deletedAt)));\n    if (thread === undefined) {\n      return null;\n    }\n    const messages = await this.db\n      .select()\n      .from(agentMessage)\n      .where(eq(agentMessage.threadId, threadId))\n      .orderBy(asc(agentMessage.createdAt), asc(agentMessage.id));\n    const last = messages[messages.length - 1];\n    return {\n      ...this.toSummary(thread, last?.content),\n      messages: messages.map((message) => this.toStoredMessage(message)),\n      ...(thread.activeStreamId != null ? { activeStreamId: thread.activeStreamId } : {}),\n    };\n  }\n\n  async listThreads(actorRef: string, limit = 50): Promise<ThreadSummary[]> {\n    const threads = await this.db\n      .select()\n      .from(agentThread)\n      .where(\n        and(\n          eq(agentThread.actorRef, actorRef),\n          eq(agentThread.transient, false),\n          isNull(agentThread.deletedAt),\n        ),\n      )\n      .orderBy(desc(agentThread.updatedAt))\n      .limit(limit);\n    const summaries: ThreadSummary[] = [];\n    for (const thread of threads) {\n      const [last] = await this.db\n        .select()\n        .from(agentMessage)\n        .where(eq(agentMessage.threadId, thread.id))\n        .orderBy(desc(agentMessage.createdAt), desc(agentMessage.id))\n        .limit(1);\n      summaries.push(this.toSummary(thread, last?.content));\n    }\n    return summaries;\n  }\n\n  async softDeleteThread(threadId: string): Promise<void> {\n    await this.db\n      .update(agentThread)\n      .set({ deletedAt: new Date() })\n      .where(eq(agentThread.id, threadId));\n  }\n\n  async forkThread(threadId: string, fromMessageId: string): Promise<ThreadSummary> {\n    const [source] = await this.db.select().from(agentThread).where(eq(agentThread.id, threadId));\n    if (source === undefined) {\n      throw new Error(`thread ${threadId} not found`);\n    }\n    const messages = await this.db\n      .select()\n      .from(agentMessage)\n      .where(eq(agentMessage.threadId, threadId))\n      .orderBy(asc(agentMessage.createdAt), asc(agentMessage.id));\n    const cutoff = messages.findIndex((message) => message.id === fromMessageId);\n    const kept = cutoff >= 0 ? messages.slice(0, cutoff + 1) : messages;\n    const now = new Date();\n    const fork: AgentThreadRow = {\n      id: crypto.randomUUID(),\n      actorRef: source.actorRef,\n      tenantRef: source.tenantRef,\n      title: source.title,\n      persona: source.persona,\n      transient: false,\n      activeStreamId: null,\n      createdAt: now,\n      updatedAt: now,\n      deletedAt: null,\n    };\n    await this.db.insert(agentThread).values(fork);\n    if (kept.length > 0) {\n      await this.db.insert(agentMessage).values(\n        kept.map((message) => ({\n          id: crypto.randomUUID(),\n          threadId: fork.id,\n          role: message.role,\n          content: message.content,\n          toolCalls: message.toolCalls,\n          toolResults: message.toolResults,\n          followUps: message.followUps,\n          usage: message.usage,\n          persona: message.persona,\n          createdAt: message.createdAt,\n        })),\n      );\n    }\n    return this.toSummary(fork);\n  }\n\n  async ownerOfThread(threadId: string): Promise<string | null> {\n    const [thread] = await this.db\n      .select({ actorRef: agentThread.actorRef })\n      .from(agentThread)\n      .where(and(eq(agentThread.id, threadId), isNull(agentThread.deletedAt)));\n    return thread?.actorRef ?? null;\n  }\n\n  async ownerOfToolCall(toolCallId: string): Promise<string | null> {\n    const [row] = await this.db\n      .select({ actorRef: agentThread.actorRef })\n      .from(agentToolCall)\n      .innerJoin(agentMessage, eq(agentToolCall.messageId, agentMessage.id))\n      .innerJoin(agentThread, eq(agentMessage.threadId, agentThread.id))\n      .where(eq(agentToolCall.id, toolCallId));\n    return row?.actorRef ?? null;\n  }\n\n  async runForToolCall(toolCallId: string): Promise<string | null> {\n    const [row] = await this.db\n      .select({ activeStreamId: agentThread.activeStreamId })\n      .from(agentToolCall)\n      .innerJoin(agentMessage, eq(agentToolCall.messageId, agentMessage.id))\n      .innerJoin(agentThread, eq(agentMessage.threadId, agentThread.id))\n      .where(eq(agentToolCall.id, toolCallId));\n    return row?.activeStreamId ?? null;\n  }\n\n  async ownerOfActiveStream(runId: string): Promise<string | null> {\n    const [thread] = await this.db\n      .select({ actorRef: agentThread.actorRef })\n      .from(agentThread)\n      .where(and(eq(agentThread.activeStreamId, runId), isNull(agentThread.deletedAt)));\n    return thread?.actorRef ?? null;\n  }\n\n  async setTitle(threadId: string, title: string): Promise<void> {\n    await this.db\n      .update(agentThread)\n      .set({ title, updatedAt: new Date() })\n      .where(eq(agentThread.id, threadId));\n  }\n\n  async setActiveStream(threadId: string, runId: string | null): Promise<void> {\n    await this.db\n      .update(agentThread)\n      .set({ activeStreamId: runId })\n      .where(eq(agentThread.id, threadId));\n  }\n\n  async appendMessage(input: AppendMessageInput): Promise<StoredMessage> {\n    const [thread] = await this.db\n      .select()\n      .from(agentThread)\n      .where(eq(agentThread.id, input.threadId));\n    if (thread === undefined) {\n      throw new Error(`thread ${input.threadId} not found`);\n    }\n    const now = new Date();\n    const message: AgentMessageRow = {\n      id: crypto.randomUUID(),\n      threadId: input.threadId,\n      role: input.role,\n      content: input.content,\n      toolCalls: input.toolCalls ?? null,\n      toolResults: input.toolResults ?? null,\n      followUps: input.followUps ?? null,\n      usage: input.usage ?? null,\n      persona: input.persona ?? null,\n      createdAt: now,\n    };\n    await this.db.insert(agentMessage).values(message);\n    await this.db\n      .update(agentThread)\n      .set({ updatedAt: now })\n      .where(eq(agentThread.id, input.threadId));\n    return this.toStoredMessage(message);\n  }\n\n  async truncateFrom(threadId: string, messageId: string): Promise<void> {\n    const messages = await this.db\n      .select()\n      .from(agentMessage)\n      .where(eq(agentMessage.threadId, threadId))\n      .orderBy(asc(agentMessage.createdAt), asc(agentMessage.id));\n    const cutoff = messages.findIndex((message) => message.id === messageId);\n    if (cutoff < 0) {\n      return;\n    }\n    const doomedIds = messages.slice(cutoff).map((message) => message.id);\n    await this.db.delete(agentToolCall).where(inArray(agentToolCall.messageId, doomedIds));\n    await this.db.delete(agentMessage).where(inArray(agentMessage.id, doomedIds));\n  }\n\n  async recordToolCall(input: RecordToolCallInput): Promise<void> {\n    await this.db.insert(agentToolCall).values({\n      id: input.toolCallId,\n      messageId: input.messageId,\n      toolName: input.toolName,\n      toolType: input.toolType,\n      input: input.input,\n      output: null,\n      status: input.status,\n      executedByRef: null,\n      executionMs: null,\n      error: null,\n      createdAt: new Date(),\n      executedAt: null,\n    });\n  }\n\n  async updateToolCall(input: UpdateToolCallInput): Promise<void> {\n    const updates: Partial<typeof agentToolCall.$inferInsert> = { status: input.status };\n    if (input.output !== undefined) {\n      updates.output = input.output;\n    }\n    if (input.error !== undefined) {\n      updates.error = input.error;\n    }\n    if (input.executionMs !== undefined) {\n      updates.executionMs = input.executionMs;\n    }\n    if (input.executedByRef !== undefined) {\n      updates.executedByRef = input.executedByRef;\n    }\n    if (input.status === 'executed' || input.status === 'auto_executed') {\n      updates.executedAt = new Date();\n    }\n    await this.db.update(agentToolCall).set(updates).where(eq(agentToolCall.id, input.toolCallId));\n  }\n\n  async recordUsage(input: RecordUsageInput): Promise<void> {\n    await this.db.insert(agentTokenUsage).values({\n      id: crypto.randomUUID(),\n      threadId: input.threadId,\n      actorRef: input.actorRef,\n      messageId: input.messageId ?? null,\n      modelId: input.modelId,\n      purpose: input.purpose,\n      inputTokens: input.usage.inputTokens,\n      outputTokens: input.usage.outputTokens,\n      cacheWriteTokens: input.usage.cacheWriteTokens ?? null,\n      cacheReadTokens: input.usage.cacheReadTokens ?? null,\n      costUsd: input.costUsd ?? null,\n      createdAt: new Date(),\n    });\n  }\n\n  async quotaToday(actorRef: string, day: string): Promise<{ usedTokens: number }> {\n    const start = new Date(`${day}T00:00:00.000Z`);\n    const end = new Date(`${day}T23:59:59.999Z`);\n    const rows = await this.db\n      .select()\n      .from(agentTokenUsage)\n      .where(\n        and(\n          eq(agentTokenUsage.actorRef, actorRef),\n          gte(agentTokenUsage.createdAt, start),\n          lte(agentTokenUsage.createdAt, end),\n        ),\n      );\n    const usedTokens = rows.reduce((sum, row) => sum + row.inputTokens + row.outputTokens, 0);\n    return { usedTokens };\n  }\n\n  private toSummary(thread: AgentThreadRow, lastContent?: string): ThreadSummary {\n    return {\n      id: thread.id,\n      title: thread.title,\n      persona: thread.persona,\n      transient: thread.transient,\n      createdAt: thread.createdAt.toISOString(),\n      updatedAt: thread.updatedAt.toISOString(),\n      ...(lastContent !== undefined ? { lastMessagePreview: lastContent.slice(0, 120) } : {}),\n    };\n  }\n\n  private toStoredMessage(message: AgentMessageRow): StoredMessage {\n    return {\n      id: message.id,\n      role: message.role,\n      content: message.content,\n      createdAt: message.createdAt.toISOString(),\n      ...(message.toolCalls != null ? { toolCalls: message.toolCalls } : {}),\n      ...(message.toolResults != null ? { toolResults: message.toolResults } : {}),\n      ...(message.followUps != null ? { followUps: message.followUps } : {}),\n      ...(message.usage != null ? { usage: message.usage } : {}),\n    };\n  }\n}\n","import type {\n  ActorSpendRow,\n  AgentGovernanceQueries,\n  GovernanceRange,\n  ModelSpendRow,\n  ThreadActivityRow,\n  ToolCallActivityRow,\n  UsageTrendPoint,\n} from '@dudousxd/nestjs-agent-core';\nimport { and, count, desc, eq, gte, isNull, lte } from 'drizzle-orm';\nimport {\n  type AgentDrizzleDb,\n  agentMessage,\n  agentModelPricing,\n  agentThread,\n  agentTokenUsage,\n  agentToolCall,\n} from './schema.js';\n\n/** The current per-1M token prices for one model; cache rates fall back to the input rate. */\ninterface ModelPrice {\n  inputPricePer1m: number;\n  outputPricePer1m: number;\n  cacheWritePricePer1m?: number | null;\n  cacheReadPricePer1m?: number | null;\n}\n\n/**\n * Token-ledger estimate for one usage row against the current pricing row: the uncached input at the\n * input rate, cache-write/cache-read tokens at their own rates (falling back to the input rate when\n * unpriced), plus output at the output rate. An unpriced model contributes 0 (tokens still count).\n * Cache token counts are subsets of `inputTokens`, so the uncached remainder is the difference.\n */\nfunction estimateFromTokens(\n  pricing: Map<string, ModelPrice>,\n  row: typeof agentTokenUsage.$inferSelect,\n): number {\n  const price = pricing.get(row.modelId);\n  if (price === undefined) {\n    return 0;\n  }\n  const cacheWriteTokens = row.cacheWriteTokens ?? 0;\n  const cacheReadTokens = row.cacheReadTokens ?? 0;\n  const uncachedInputTokens = row.inputTokens - cacheWriteTokens - cacheReadTokens;\n  return (\n    (uncachedInputTokens / 1_000_000) * price.inputPricePer1m +\n    (cacheWriteTokens / 1_000_000) * (price.cacheWritePricePer1m ?? price.inputPricePer1m) +\n    (cacheReadTokens / 1_000_000) * (price.cacheReadPricePer1m ?? price.inputPricePer1m) +\n    (row.outputTokens / 1_000_000) * price.outputPricePer1m\n  );\n}\n\n/**\n * The cost of one usage row: the provider-reported `costUsd` when present (gateways report real\n * spend), otherwise the cache-aware token estimate against the current pricing row.\n */\nfunction rowCost(\n  pricing: Map<string, ModelPrice>,\n  row: typeof agentTokenUsage.$inferSelect,\n): number {\n  return row.costUsd ?? estimateFromTokens(pricing, row);\n}\n\n/** Turns an inclusive `YYYY-MM-DD` day range into the UTC datetime bounds used by `quotaToday`. */\nfunction dayBounds(range: GovernanceRange): { start: Date; end: Date } {\n  return {\n    start: new Date(`${range.fromDay}T00:00:00.000Z`),\n    end: new Date(`${range.toDay}T23:59:59.999Z`),\n  };\n}\n\n/**\n * {@link AgentGovernanceQueries} backed by Drizzle ORM — the read/analytics half of the store SPI,\n * mirroring {@link import('@dudousxd/nestjs-agent-store-mikro-orm')} exactly. Cost is the token\n * ledger joined to the current pricing row per model (`agentModelPricing.isCurrent`); an unpriced\n * model contributes 0 cost. Aggregation is in-process (like `quotaToday`) so day-bucketing stays\n * dialect-portable.\n */\nexport class DrizzleGovernanceQueries implements AgentGovernanceQueries {\n  constructor(private readonly db: AgentDrizzleDb) {}\n\n  private async loadPricing(): Promise<Map<string, ModelPrice>> {\n    const rows = await this.db\n      .select()\n      .from(agentModelPricing)\n      .where(eq(agentModelPricing.isCurrent, true));\n    const pricing = new Map<string, ModelPrice>();\n    for (const row of rows) {\n      pricing.set(row.modelId, {\n        inputPricePer1m: row.inputPricePer1m,\n        outputPricePer1m: row.outputPricePer1m,\n        cacheWritePricePer1m: row.cacheWritePricePer1m,\n        cacheReadPricePer1m: row.cacheReadPricePer1m,\n      });\n    }\n    return pricing;\n  }\n\n  private async usageInRange(\n    range: GovernanceRange,\n  ): Promise<(typeof agentTokenUsage.$inferSelect)[]> {\n    const { start, end } = dayBounds(range);\n    const rows = await this.db\n      .select()\n      .from(agentTokenUsage)\n      .where(and(gte(agentTokenUsage.createdAt, start), lte(agentTokenUsage.createdAt, end)));\n    return rows;\n  }\n\n  async spendByModel(range: GovernanceRange): Promise<ModelSpendRow[]> {\n    const pricing = await this.loadPricing();\n    const rows = await this.usageInRange(range);\n    const byModel = new Map<\n      string,\n      { requests: number; inputTokens: number; outputTokens: number; costUsd: number }\n    >();\n    for (const row of rows) {\n      const bucket = byModel.get(row.modelId) ?? {\n        requests: 0,\n        inputTokens: 0,\n        outputTokens: 0,\n        costUsd: 0,\n      };\n      bucket.requests += 1;\n      bucket.inputTokens += row.inputTokens;\n      bucket.outputTokens += row.outputTokens;\n      bucket.costUsd += rowCost(pricing, row);\n      byModel.set(row.modelId, bucket);\n    }\n    const result: ModelSpendRow[] = [];\n    for (const [modelId, bucket] of byModel) {\n      result.push({\n        modelId,\n        requests: bucket.requests,\n        inputTokens: bucket.inputTokens,\n        outputTokens: bucket.outputTokens,\n        costUsd: bucket.costUsd,\n      });\n    }\n    result.sort(\n      (left, right) => right.costUsd - left.costUsd || left.modelId.localeCompare(right.modelId),\n    );\n    return result;\n  }\n\n  async spendByActor(range: GovernanceRange): Promise<ActorSpendRow[]> {\n    const pricing = await this.loadPricing();\n    const rows = await this.usageInRange(range);\n    const byActor = new Map<string, { requests: number; totalTokens: number; costUsd: number }>();\n    for (const row of rows) {\n      const bucket = byActor.get(row.actorRef) ?? { requests: 0, totalTokens: 0, costUsd: 0 };\n      bucket.requests += 1;\n      bucket.totalTokens += row.inputTokens + row.outputTokens;\n      bucket.costUsd += rowCost(pricing, row);\n      byActor.set(row.actorRef, bucket);\n    }\n    const result: ActorSpendRow[] = [];\n    for (const [actorRef, bucket] of byActor) {\n      result.push({\n        actorRef,\n        requests: bucket.requests,\n        totalTokens: bucket.totalTokens,\n        costUsd: bucket.costUsd,\n      });\n    }\n    result.sort(\n      (left, right) => right.costUsd - left.costUsd || left.actorRef.localeCompare(right.actorRef),\n    );\n    return result;\n  }\n\n  async usageTrend(range: GovernanceRange): Promise<UsageTrendPoint[]> {\n    const pricing = await this.loadPricing();\n    const rows = await this.usageInRange(range);\n    const byDay = new Map<string, { totalTokens: number; costUsd: number }>();\n    for (const row of rows) {\n      const day = row.createdAt.toISOString().slice(0, 10);\n      const bucket = byDay.get(day) ?? { totalTokens: 0, costUsd: 0 };\n      bucket.totalTokens += row.inputTokens + row.outputTokens;\n      bucket.costUsd += rowCost(pricing, row);\n      byDay.set(day, bucket);\n    }\n    const result: UsageTrendPoint[] = [];\n    for (const [day, bucket] of byDay) {\n      result.push({ day, totalTokens: bucket.totalTokens, costUsd: bucket.costUsd });\n    }\n    result.sort((left, right) => left.day.localeCompare(right.day));\n    return result;\n  }\n\n  async recentToolCalls(limit: number): Promise<ToolCallActivityRow[]> {\n    const rows = await this.db\n      .select({\n        toolCallId: agentToolCall.id,\n        toolName: agentToolCall.toolName,\n        toolType: agentToolCall.toolType,\n        status: agentToolCall.status,\n        threadId: agentMessage.threadId,\n        createdAt: agentToolCall.createdAt,\n      })\n      .from(agentToolCall)\n      .innerJoin(agentMessage, eq(agentToolCall.messageId, agentMessage.id))\n      .orderBy(desc(agentToolCall.createdAt), desc(agentToolCall.id))\n      .limit(limit);\n    return rows.map((row) => ({\n      toolCallId: row.toolCallId,\n      toolName: row.toolName,\n      toolType: row.toolType,\n      status: row.status,\n      threadId: row.threadId,\n      createdAt: row.createdAt.toISOString(),\n    }));\n  }\n\n  async recentThreads(limit: number): Promise<ThreadActivityRow[]> {\n    const threads = await this.db\n      .select()\n      .from(agentThread)\n      .where(isNull(agentThread.deletedAt))\n      .orderBy(desc(agentThread.updatedAt), desc(agentThread.id))\n      .limit(limit);\n    const result: ThreadActivityRow[] = [];\n    for (const thread of threads) {\n      const [messageCountRow] = await this.db\n        .select({ value: count() })\n        .from(agentMessage)\n        .where(eq(agentMessage.threadId, thread.id));\n      const usageRows = await this.db\n        .select()\n        .from(agentTokenUsage)\n        .where(eq(agentTokenUsage.threadId, thread.id));\n      const totalTokens = usageRows.reduce(\n        (sum, row) => sum + row.inputTokens + row.outputTokens,\n        0,\n      );\n      result.push({\n        threadId: thread.id,\n        title: thread.title,\n        actorRef: thread.actorRef,\n        messageCount: messageCountRow?.value ?? 0,\n        totalTokens,\n        lastActivityAt: thread.updatedAt.toISOString(),\n      });\n    }\n    return result;\n  }\n}\n","import type {\n  AgentPricingStore,\n  CurrentModelPrice,\n  ModelPriceInput,\n} from '@dudousxd/nestjs-agent-core';\nimport { and, eq } from 'drizzle-orm';\nimport { type AgentDrizzleDb, agentModelPricing } from './schema.js';\n\n/**\n * {@link AgentPricingStore} backed by Drizzle ORM — the write side of the pricing table\n * {@link import('./drizzle-governance-queries.js').DrizzleGovernanceQueries} joins against. A POJO\n * receiving a Drizzle SQLite database handle (the host app owns the connection), mirroring\n * {@link import('@dudousxd/nestjs-agent-store-mikro-orm')} exactly (atomic supersede semantics).\n */\nexport class DrizzlePricingStore implements AgentPricingStore {\n  constructor(private readonly db: AgentDrizzleDb) {}\n\n  async upsertModelPrice(input: ModelPriceInput): Promise<void> {\n    await this.db\n      .update(agentModelPricing)\n      .set({ isCurrent: false })\n      .where(\n        and(eq(agentModelPricing.modelId, input.modelId), eq(agentModelPricing.isCurrent, true)),\n      );\n    await this.db.insert(agentModelPricing).values({\n      id: crypto.randomUUID(),\n      modelId: input.modelId,\n      inputPricePer1m: input.inputPricePer1m,\n      outputPricePer1m: input.outputPricePer1m,\n      cacheWritePricePer1m: input.cacheWritePricePer1m ?? null,\n      cacheReadPricePer1m: input.cacheReadPricePer1m ?? null,\n      effectiveFrom: new Date(),\n      isCurrent: true,\n    });\n  }\n\n  async listCurrentPrices(): Promise<CurrentModelPrice[]> {\n    const rows = await this.db\n      .select()\n      .from(agentModelPricing)\n      .where(eq(agentModelPricing.isCurrent, true));\n    return rows.map((row) => ({\n      modelId: row.modelId,\n      inputPricePer1m: row.inputPricePer1m,\n      outputPricePer1m: row.outputPricePer1m,\n      effectiveFrom: row.effectiveFrom.toISOString(),\n      ...(row.cacheWritePricePer1m != null\n        ? { cacheWritePricePer1m: row.cacheWritePricePer1m }\n        : {}),\n      ...(row.cacheReadPricePer1m != null ? { cacheReadPricePer1m: row.cacheReadPricePer1m } : {}),\n    }));\n  }\n}\n","import {\n  AGENT_GOVERNANCE_QUERIES,\n  AGENT_PRICING_STORE,\n  AGENT_STORE,\n} from '@dudousxd/nestjs-agent-core';\nimport { type DynamicModule, Module } from '@nestjs/common';\nimport { DrizzleAgentStore } from './drizzle-agent-store.js';\nimport { DrizzleGovernanceQueries } from './drizzle-governance-queries.js';\nimport { DrizzlePricingStore } from './drizzle-pricing-store.js';\nimport type { AgentDrizzleDb } from './schema.js';\n\n/** Options for {@link DrizzleAgentStoreModule.forRoot}. The app owns and supplies the db handle. */\nexport interface DrizzleAgentStoreModuleOptions {\n  /** A Drizzle SQLite database instance (`drizzle(client, { schema: agentSchema })`). */\n  db: AgentDrizzleDb;\n}\n\n/**\n * Binds {@link DrizzleAgentStore} to the {@link AGENT_STORE} token consumed by\n * `@dudousxd/nestjs-agent`, plus {@link DrizzleGovernanceQueries} to\n * {@link AGENT_GOVERNANCE_QUERIES} (the read-model the dashboard/telescope surfaces consume) and\n * {@link DrizzlePricingStore} to {@link AGENT_PRICING_STORE} (the write side of that pricing table).\n * The host app supplies an already-opened Drizzle db — this module never opens a connection itself.\n *\n * The returned module is global, so {@link AGENT_STORE}, {@link AGENT_GOVERNANCE_QUERIES} and\n * {@link AGENT_PRICING_STORE} are visible app-wide — `AgentModule` resolves the store without an\n * explicit `store` option, and the dashboard/telescope resolve {@link AGENT_GOVERNANCE_QUERIES}\n * without the host re-binding it.\n *\n * ```ts\n * @Module({ imports: [DrizzleAgentStoreModule.forRoot({ db })] })\n * export class AppModule {}\n * ```\n */\n@Module({})\nexport class DrizzleAgentStoreModule {\n  static forRoot(options: DrizzleAgentStoreModuleOptions): DynamicModule {\n    return {\n      module: DrizzleAgentStoreModule,\n      global: true,\n      providers: [\n        { provide: DrizzleAgentStore, useFactory: () => new DrizzleAgentStore(options.db) },\n        { provide: AGENT_STORE, useExisting: DrizzleAgentStore },\n        {\n          provide: DrizzleGovernanceQueries,\n          useFactory: () => new DrizzleGovernanceQueries(options.db),\n        },\n        { provide: AGENT_GOVERNANCE_QUERIES, useExisting: DrizzleGovernanceQueries },\n        {\n          provide: DrizzlePricingStore,\n          useFactory: () => new DrizzlePricingStore(options.db),\n        },\n        { provide: AGENT_PRICING_STORE, useExisting: DrizzlePricingStore },\n      ],\n      exports: [\n        DrizzleAgentStore,\n        AGENT_STORE,\n        DrizzleGovernanceQueries,\n        AGENT_GOVERNANCE_QUERIES,\n        DrizzlePricingStore,\n        AGENT_PRICING_STORE,\n      ],\n    };\n  }\n}\n"],"mappings":";;;;AASA,SAEEA,OACAC,SACAC,MACAC,aACAC,YACK;AAeA,IAAMC,cAAcF,YACzB,gBACA;EACEG,IAAIF,KAAK,IAAA,EAAMG,WAAU;EACzBC,UAAUJ,KAAK,WAAA,EAAaK,QAAO;EACnCC,WAAWN,KAAK,YAAA;EAChBO,OAAOP,KAAK,OAAA,EAASK,QAAO;EAC5BG,SAASR,KAAK,SAAA,EAAWK,QAAO,EAAGI,QAAQ,SAAA;EAC3CC,WAAWb,QAAQ,aAAa;IAAEc,MAAM;EAAU,CAAA,EAAGN,QAAO,EAAGI,QAAQ,KAAA;EACvEG,gBAAgBZ,KAAK,kBAAA;EACrBa,WAAWhB,QAAQ,cAAc;IAAEc,MAAM;EAAe,CAAA,EAAGN,QAAO;EAClES,WAAWjB,QAAQ,cAAc;IAAEc,MAAM;EAAe,CAAA,EAAGN,QAAO;EAClEU,WAAWlB,QAAQ,cAAc;IAAEc,MAAM;EAAe,CAAA;AAC1D,GACA,CAACK,UAAU;EAACpB,MAAM,gCAAA,EAAkCqB,GAAGD,MAAMZ,UAAUY,MAAMF,SAAS;CAAE;AAQnF,IAAMI,eAAenB,YAC1B,iBACA;EACEG,IAAIF,KAAK,IAAA,EAAMG,WAAU;EACzBgB,UAAUnB,KAAK,WAAA,EACZK,QAAO,EACPe,WAAW,MAAMnB,YAAYC,IAAI;IAAEmB,UAAU;EAAU,CAAA;EAC1DC,MAAMtB,KAAK,MAAA,EAAQuB,MAAK,EAAgBlB,QAAO;EAC/CmB,SAASxB,KAAK,SAAA,EAAWK,QAAO;EAChCoB,WAAWzB,KAAK,cAAc;IAAEW,MAAM;EAAO,CAAA,EAAGY,MAAK;EACrDG,aAAa1B,KAAK,gBAAgB;IAAEW,MAAM;EAAO,CAAA,EAAGY,MAAK;EACzDI,WAAW3B,KAAK,cAAc;IAAEW,MAAM;EAAO,CAAA,EAAGY,MAAK;EACrDK,OAAO5B,KAAK,SAAS;IAAEW,MAAM;EAAO,CAAA,EAAGY,MAAK;EAC5Cf,SAASR,KAAK,SAAA;EACda,WAAWhB,QAAQ,cAAc;IAAEc,MAAM;EAAe,CAAA,EAAGN,QAAO;AACpE,GACA,CAACW,UAAU;EAACpB,MAAM,kCAAA,EAAoCqB,GAAGD,MAAMG,UAAUH,MAAMH,SAAS;CAAE;AAIrF,IAAMgB,gBAAgB9B,YAAY,mBAAmB;EAC1DG,IAAIF,KAAK,IAAA,EAAMG,WAAU;EACzB2B,WAAW9B,KAAK,YAAA,EACbK,QAAO,EACPe,WAAW,MAAMF,aAAahB,IAAI;IAAEmB,UAAU;EAAU,CAAA;EAC3DU,UAAU/B,KAAK,WAAA,EAAaK,QAAO;EACnC2B,UAAUhC,KAAK,WAAA,EAAauB,MAAK,EAAalB,QAAO;EACrD4B,OAAOjC,KAAK,SAAS;IAAEW,MAAM;EAAO,CAAA;EACpCuB,QAAQlC,KAAK,UAAU;IAAEW,MAAM;EAAO,CAAA;EACtCwB,QAAQnC,KAAK,QAAA,EAAUuB,MAAK,EAAmBlB,QAAO;EACtD+B,eAAepC,KAAK,iBAAA;EACpBqC,aAAaxC,QAAQ,cAAA;EACrByC,OAAOtC,KAAK,OAAA;EACZa,WAAWhB,QAAQ,cAAc;IAAEc,MAAM;EAAe,CAAA,EAAGN,QAAO;EAClEkC,YAAY1C,QAAQ,eAAe;IAAEc,MAAM;EAAe,CAAA;AAC5D,CAAA;AAGO,IAAM6B,kBAAkBzC,YAC7B,qBACA;EACEG,IAAIF,KAAK,IAAA,EAAMG,WAAU;EACzBgB,UAAUnB,KAAK,WAAA,EACZK,QAAO,EACPe,WAAW,MAAMnB,YAAYC,IAAI;IAAEmB,UAAU;EAAU,CAAA;EAC1DjB,UAAUJ,KAAK,WAAA,EAAaK,QAAO;EACnCyB,WAAW9B,KAAK,YAAA;EAChByC,SAASzC,KAAK,UAAA,EAAYK,QAAO;EACjCqC,SAAS1C,KAAK,SAAA,EAAWuB,MAAK,EAAiBlB,QAAO;EACtDsC,aAAa9C,QAAQ,cAAA,EAAgBQ,QAAO;EAC5CuC,cAAc/C,QAAQ,eAAA,EAAiBQ,QAAO;;EAE9CwC,kBAAkBhD,QAAQ,oBAAA;;EAE1BiD,iBAAiBjD,QAAQ,mBAAA;;EAEzBkD,SAASjD,KAAK,UAAA;EACde,WAAWhB,QAAQ,cAAc;IAAEc,MAAM;EAAe,CAAA,EAAGN,QAAO;AACpE,GACA,CAACW,UAAU;EAACpB,MAAM,qCAAA,EAAuCqB,GAAGD,MAAMZ,UAAUY,MAAMH,SAAS;CAAE;AAIxF,IAAMmC,oBAAoBjD,YAAY,uBAAuB;EAClEG,IAAIF,KAAK,IAAA,EAAMG,WAAU;EACzBsC,SAASzC,KAAK,UAAA,EAAYK,QAAO;EACjC4C,iBAAiBnD,KAAK,oBAAA,EAAsBO,QAAO;EACnD6C,kBAAkBpD,KAAK,qBAAA,EAAuBO,QAAO;;EAErD8C,sBAAsBrD,KAAK,0BAAA;;EAE3BsD,qBAAqBtD,KAAK,yBAAA;EAC1BuD,eAAexD,QAAQ,kBAAkB;IAAEc,MAAM;EAAe,CAAA,EAAGN,QAAO;EAC1EiD,WAAWzD,QAAQ,cAAc;IAAEc,MAAM;EAAU,CAAA,EAAGN,QAAO;AAC/D,CAAA;AAGO,IAAMkD,cAAc;EACzBtD;EACAiB;EACAW;EACAW;EACAQ;AACF;;;ACxIA,SAASQ,WAAW;AASpB,IAAMC,aAAuB;EAC3B;;;;;;;;;;;;EAYA;;EAEA;;;;;;;;;;;;EAYA;;EAEA;;;;;;;;;;;;;;EAcA;;;;;;;;;;;;;;EAcA;;EAEA;;;;;;;;;;;AAaF,eAAsBC,kBAAkBC,IAAkB;AACxD,aAAWC,aAAaH,YAAY;AAClC,UAAME,GAAGE,IAAIC,IAAIC,IAAIH,SAAAA,CAAAA;EACvB;AACF;AAJsBF;;;ACtEtB,SAASM,KAAKC,KAAKC,MAAMC,IAAIC,KAAKC,SAASC,QAAQC,WAAW;AAiBvD,IAAMC,oBAAN,MAAMA;EAjBb,OAiBaA;;;;EACX,YAA6BC,IAAoB;SAApBA,KAAAA;EAAqB;EAElD,MAAMC,aAAaC,OAAkD;AACnE,UAAMC,MAAM,oBAAIC,KAAAA;AAChB,UAAMC,SAAyB;MAC7BC,IAAIC,OAAOC,WAAU;MACrBC,UAAUP,MAAMQ,MAAMJ;MACtBK,WAAWT,MAAMQ,MAAMC,aAAa;MACpCC,OAAOV,MAAMU,SAAS;MACtBC,SAASX,MAAMW;MACfC,WAAWZ,MAAMY,aAAa;MAC9BC,gBAAgB;MAChBC,WAAWb;MACXc,WAAWd;MACXe,WAAW;IACb;AACA,UAAM,KAAKlB,GAAGmB,OAAOC,WAAAA,EAAaC,OAAOhB,MAAAA;AACzC,WAAO,KAAKiB,UAAUjB,MAAAA;EACxB;EAEA,MAAMkB,UAAUC,UAAgD;AAC9D,UAAM,CAACnB,MAAAA,IAAU,MAAM,KAAKL,GACzByB,OAAM,EACNC,KAAKN,WAAAA,EACLO,MAAMC,IAAIC,GAAGT,YAAYd,IAAIkB,QAAAA,GAAWM,OAAOV,YAAYF,SAAS,CAAA,CAAA;AACvE,QAAIb,WAAW0B,QAAW;AACxB,aAAO;IACT;AACA,UAAMC,WAAW,MAAM,KAAKhC,GACzByB,OAAM,EACNC,KAAKO,YAAAA,EACLN,MAAME,GAAGI,aAAaT,UAAUA,QAAAA,CAAAA,EAChCU,QAAQC,IAAIF,aAAajB,SAAS,GAAGmB,IAAIF,aAAa3B,EAAE,CAAA;AAC3D,UAAM8B,OAAOJ,SAASA,SAASK,SAAS,CAAA;AACxC,WAAO;MACL,GAAG,KAAKf,UAAUjB,QAAQ+B,MAAME,OAAAA;MAChCN,UAAUA,SAASO,IAAI,CAACC,YAAY,KAAKC,gBAAgBD,OAAAA,CAAAA;MACzD,GAAInC,OAAOU,kBAAkB,OAAO;QAAEA,gBAAgBV,OAAOU;MAAe,IAAI,CAAC;IACnF;EACF;EAEA,MAAM2B,YAAYjC,UAAkBkC,QAAQ,IAA8B;AACxE,UAAMC,UAAU,MAAM,KAAK5C,GACxByB,OAAM,EACNC,KAAKN,WAAAA,EACLO,MACCC,IACEC,GAAGT,YAAYX,UAAUA,QAAAA,GACzBoB,GAAGT,YAAYN,WAAW,KAAA,GAC1BgB,OAAOV,YAAYF,SAAS,CAAA,CAAA,EAG/BgB,QAAQW,KAAKzB,YAAYH,SAAS,CAAA,EAClC0B,MAAMA,KAAAA;AACT,UAAMG,YAA6B,CAAA;AACnC,eAAWzC,UAAUuC,SAAS;AAC5B,YAAM,CAACR,IAAAA,IAAQ,MAAM,KAAKpC,GACvByB,OAAM,EACNC,KAAKO,YAAAA,EACLN,MAAME,GAAGI,aAAaT,UAAUnB,OAAOC,EAAE,CAAA,EACzC4B,QAAQW,KAAKZ,aAAajB,SAAS,GAAG6B,KAAKZ,aAAa3B,EAAE,CAAA,EAC1DqC,MAAM,CAAA;AACTG,gBAAUC,KAAK,KAAKzB,UAAUjB,QAAQ+B,MAAME,OAAAA,CAAAA;IAC9C;AACA,WAAOQ;EACT;EAEA,MAAME,iBAAiBxB,UAAiC;AACtD,UAAM,KAAKxB,GACRiD,OAAO7B,WAAAA,EACP8B,IAAI;MAAEhC,WAAW,oBAAId,KAAAA;IAAO,CAAA,EAC5BuB,MAAME,GAAGT,YAAYd,IAAIkB,QAAAA,CAAAA;EAC9B;EAEA,MAAM2B,WAAW3B,UAAkB4B,eAA+C;AAChF,UAAM,CAACC,MAAAA,IAAU,MAAM,KAAKrD,GAAGyB,OAAM,EAAGC,KAAKN,WAAAA,EAAaO,MAAME,GAAGT,YAAYd,IAAIkB,QAAAA,CAAAA;AACnF,QAAI6B,WAAWtB,QAAW;AACxB,YAAM,IAAIuB,MAAM,UAAU9B,QAAAA,YAAoB;IAChD;AACA,UAAMQ,WAAW,MAAM,KAAKhC,GACzByB,OAAM,EACNC,KAAKO,YAAAA,EACLN,MAAME,GAAGI,aAAaT,UAAUA,QAAAA,CAAAA,EAChCU,QAAQC,IAAIF,aAAajB,SAAS,GAAGmB,IAAIF,aAAa3B,EAAE,CAAA;AAC3D,UAAMiD,SAASvB,SAASwB,UAAU,CAAChB,YAAYA,QAAQlC,OAAO8C,aAAAA;AAC9D,UAAMK,OAAOF,UAAU,IAAIvB,SAAS0B,MAAM,GAAGH,SAAS,CAAA,IAAKvB;AAC3D,UAAM7B,MAAM,oBAAIC,KAAAA;AAChB,UAAMuD,OAAuB;MAC3BrD,IAAIC,OAAOC,WAAU;MACrBC,UAAU4C,OAAO5C;MACjBE,WAAW0C,OAAO1C;MAClBC,OAAOyC,OAAOzC;MACdC,SAASwC,OAAOxC;MAChBC,WAAW;MACXC,gBAAgB;MAChBC,WAAWb;MACXc,WAAWd;MACXe,WAAW;IACb;AACA,UAAM,KAAKlB,GAAGmB,OAAOC,WAAAA,EAAaC,OAAOsC,IAAAA;AACzC,QAAIF,KAAKpB,SAAS,GAAG;AACnB,YAAM,KAAKrC,GAAGmB,OAAOc,YAAAA,EAAcZ,OACjCoC,KAAKlB,IAAI,CAACC,aAAa;QACrBlC,IAAIC,OAAOC,WAAU;QACrBgB,UAAUmC,KAAKrD;QACfsD,MAAMpB,QAAQoB;QACdtB,SAASE,QAAQF;QACjBuB,WAAWrB,QAAQqB;QACnBC,aAAatB,QAAQsB;QACrBC,WAAWvB,QAAQuB;QACnBC,OAAOxB,QAAQwB;QACfnD,SAAS2B,QAAQ3B;QACjBG,WAAWwB,QAAQxB;MACrB,EAAA,CAAA;IAEJ;AACA,WAAO,KAAKM,UAAUqC,IAAAA;EACxB;EAEA,MAAMM,cAAczC,UAA0C;AAC5D,UAAM,CAACnB,MAAAA,IAAU,MAAM,KAAKL,GACzByB,OAAO;MAAEhB,UAAUW,YAAYX;IAAS,CAAA,EACxCiB,KAAKN,WAAAA,EACLO,MAAMC,IAAIC,GAAGT,YAAYd,IAAIkB,QAAAA,GAAWM,OAAOV,YAAYF,SAAS,CAAA,CAAA;AACvE,WAAOb,QAAQI,YAAY;EAC7B;EAEA,MAAMyD,gBAAgBC,YAA4C;AAChE,UAAM,CAACC,GAAAA,IAAO,MAAM,KAAKpE,GACtByB,OAAO;MAAEhB,UAAUW,YAAYX;IAAS,CAAA,EACxCiB,KAAK2C,aAAAA,EACLC,UAAUrC,cAAcJ,GAAGwC,cAAcE,WAAWtC,aAAa3B,EAAE,CAAA,EACnEgE,UAAUlD,aAAaS,GAAGI,aAAaT,UAAUJ,YAAYd,EAAE,CAAA,EAC/DqB,MAAME,GAAGwC,cAAc/D,IAAI6D,UAAAA,CAAAA;AAC9B,WAAOC,KAAK3D,YAAY;EAC1B;EAEA,MAAM+D,eAAeL,YAA4C;AAC/D,UAAM,CAACC,GAAAA,IAAO,MAAM,KAAKpE,GACtByB,OAAO;MAAEV,gBAAgBK,YAAYL;IAAe,CAAA,EACpDW,KAAK2C,aAAAA,EACLC,UAAUrC,cAAcJ,GAAGwC,cAAcE,WAAWtC,aAAa3B,EAAE,CAAA,EACnEgE,UAAUlD,aAAaS,GAAGI,aAAaT,UAAUJ,YAAYd,EAAE,CAAA,EAC/DqB,MAAME,GAAGwC,cAAc/D,IAAI6D,UAAAA,CAAAA;AAC9B,WAAOC,KAAKrD,kBAAkB;EAChC;EAEA,MAAM0D,oBAAoBC,OAAuC;AAC/D,UAAM,CAACrE,MAAAA,IAAU,MAAM,KAAKL,GACzByB,OAAO;MAAEhB,UAAUW,YAAYX;IAAS,CAAA,EACxCiB,KAAKN,WAAAA,EACLO,MAAMC,IAAIC,GAAGT,YAAYL,gBAAgB2D,KAAAA,GAAQ5C,OAAOV,YAAYF,SAAS,CAAA,CAAA;AAChF,WAAOb,QAAQI,YAAY;EAC7B;EAEA,MAAMkE,SAASnD,UAAkBZ,OAA8B;AAC7D,UAAM,KAAKZ,GACRiD,OAAO7B,WAAAA,EACP8B,IAAI;MAAEtC;MAAOK,WAAW,oBAAIb,KAAAA;IAAO,CAAA,EACnCuB,MAAME,GAAGT,YAAYd,IAAIkB,QAAAA,CAAAA;EAC9B;EAEA,MAAMoD,gBAAgBpD,UAAkBkD,OAAqC;AAC3E,UAAM,KAAK1E,GACRiD,OAAO7B,WAAAA,EACP8B,IAAI;MAAEnC,gBAAgB2D;IAAM,CAAA,EAC5B/C,MAAME,GAAGT,YAAYd,IAAIkB,QAAAA,CAAAA;EAC9B;EAEA,MAAMqD,cAAc3E,OAAmD;AACrE,UAAM,CAACG,MAAAA,IAAU,MAAM,KAAKL,GACzByB,OAAM,EACNC,KAAKN,WAAAA,EACLO,MAAME,GAAGT,YAAYd,IAAIJ,MAAMsB,QAAQ,CAAA;AAC1C,QAAInB,WAAW0B,QAAW;AACxB,YAAM,IAAIuB,MAAM,UAAUpD,MAAMsB,QAAQ,YAAY;IACtD;AACA,UAAMrB,MAAM,oBAAIC,KAAAA;AAChB,UAAMoC,UAA2B;MAC/BlC,IAAIC,OAAOC,WAAU;MACrBgB,UAAUtB,MAAMsB;MAChBoC,MAAM1D,MAAM0D;MACZtB,SAASpC,MAAMoC;MACfuB,WAAW3D,MAAM2D,aAAa;MAC9BC,aAAa5D,MAAM4D,eAAe;MAClCC,WAAW7D,MAAM6D,aAAa;MAC9BC,OAAO9D,MAAM8D,SAAS;MACtBnD,SAASX,MAAMW,WAAW;MAC1BG,WAAWb;IACb;AACA,UAAM,KAAKH,GAAGmB,OAAOc,YAAAA,EAAcZ,OAAOmB,OAAAA;AAC1C,UAAM,KAAKxC,GACRiD,OAAO7B,WAAAA,EACP8B,IAAI;MAAEjC,WAAWd;IAAI,CAAA,EACrBwB,MAAME,GAAGT,YAAYd,IAAIJ,MAAMsB,QAAQ,CAAA;AAC1C,WAAO,KAAKiB,gBAAgBD,OAAAA;EAC9B;EAEA,MAAMsC,aAAatD,UAAkB+C,WAAkC;AACrE,UAAMvC,WAAW,MAAM,KAAKhC,GACzByB,OAAM,EACNC,KAAKO,YAAAA,EACLN,MAAME,GAAGI,aAAaT,UAAUA,QAAAA,CAAAA,EAChCU,QAAQC,IAAIF,aAAajB,SAAS,GAAGmB,IAAIF,aAAa3B,EAAE,CAAA;AAC3D,UAAMiD,SAASvB,SAASwB,UAAU,CAAChB,YAAYA,QAAQlC,OAAOiE,SAAAA;AAC9D,QAAIhB,SAAS,GAAG;AACd;IACF;AACA,UAAMwB,YAAY/C,SAAS0B,MAAMH,MAAAA,EAAQhB,IAAI,CAACC,YAAYA,QAAQlC,EAAE;AACpE,UAAM,KAAKN,GAAGgF,OAAOX,aAAAA,EAAe1C,MAAMsD,QAAQZ,cAAcE,WAAWQ,SAAAA,CAAAA;AAC3E,UAAM,KAAK/E,GAAGgF,OAAO/C,YAAAA,EAAcN,MAAMsD,QAAQhD,aAAa3B,IAAIyE,SAAAA,CAAAA;EACpE;EAEA,MAAMG,eAAehF,OAA2C;AAC9D,UAAM,KAAKF,GAAGmB,OAAOkD,aAAAA,EAAehD,OAAO;MACzCf,IAAIJ,MAAMiE;MACVI,WAAWrE,MAAMqE;MACjBY,UAAUjF,MAAMiF;MAChBC,UAAUlF,MAAMkF;MAChBlF,OAAOA,MAAMA;MACbmF,QAAQ;MACRC,QAAQpF,MAAMoF;MACdC,eAAe;MACfC,aAAa;MACbC,OAAO;MACPzE,WAAW,oBAAIZ,KAAAA;MACfsF,YAAY;IACd,CAAA;EACF;EAEA,MAAMC,eAAezF,OAA2C;AAC9D,UAAM0F,UAAsD;MAAEN,QAAQpF,MAAMoF;IAAO;AACnF,QAAIpF,MAAMmF,WAAWtD,QAAW;AAC9B6D,cAAQP,SAASnF,MAAMmF;IACzB;AACA,QAAInF,MAAMuF,UAAU1D,QAAW;AAC7B6D,cAAQH,QAAQvF,MAAMuF;IACxB;AACA,QAAIvF,MAAMsF,gBAAgBzD,QAAW;AACnC6D,cAAQJ,cAActF,MAAMsF;IAC9B;AACA,QAAItF,MAAMqF,kBAAkBxD,QAAW;AACrC6D,cAAQL,gBAAgBrF,MAAMqF;IAChC;AACA,QAAIrF,MAAMoF,WAAW,cAAcpF,MAAMoF,WAAW,iBAAiB;AACnEM,cAAQF,aAAa,oBAAItF,KAAAA;IAC3B;AACA,UAAM,KAAKJ,GAAGiD,OAAOoB,aAAAA,EAAenB,IAAI0C,OAAAA,EAASjE,MAAME,GAAGwC,cAAc/D,IAAIJ,MAAMiE,UAAU,CAAA;EAC9F;EAEA,MAAM0B,YAAY3F,OAAwC;AACxD,UAAM,KAAKF,GAAGmB,OAAO2E,eAAAA,EAAiBzE,OAAO;MAC3Cf,IAAIC,OAAOC,WAAU;MACrBgB,UAAUtB,MAAMsB;MAChBf,UAAUP,MAAMO;MAChB8D,WAAWrE,MAAMqE,aAAa;MAC9BwB,SAAS7F,MAAM6F;MACfC,SAAS9F,MAAM8F;MACfC,aAAa/F,MAAM8D,MAAMiC;MACzBC,cAAchG,MAAM8D,MAAMkC;MAC1BC,kBAAkBjG,MAAM8D,MAAMmC,oBAAoB;MAClDC,iBAAiBlG,MAAM8D,MAAMoC,mBAAmB;MAChDC,SAASnG,MAAMmG,WAAW;MAC1BrF,WAAW,oBAAIZ,KAAAA;IACjB,CAAA;EACF;EAEA,MAAMkG,WAAW7F,UAAkB8F,KAA8C;AAC/E,UAAMC,QAAQ,oBAAIpG,KAAK,GAAGmG,GAAAA,gBAAmB;AAC7C,UAAME,MAAM,oBAAIrG,KAAK,GAAGmG,GAAAA,gBAAmB;AAC3C,UAAMG,OAAO,MAAM,KAAK1G,GACrByB,OAAM,EACNC,KAAKoE,eAAAA,EACLnE,MACCC,IACEC,GAAGiE,gBAAgBrF,UAAUA,QAAAA,GAC7BkG,IAAIb,gBAAgB9E,WAAWwF,KAAAA,GAC/BI,IAAId,gBAAgB9E,WAAWyF,GAAAA,CAAAA,CAAAA;AAGrC,UAAMI,aAAaH,KAAKI,OAAO,CAACC,KAAK3C,QAAQ2C,MAAM3C,IAAI6B,cAAc7B,IAAI8B,cAAc,CAAA;AACvF,WAAO;MAAEW;IAAW;EACtB;EAEQvF,UAAUjB,QAAwB2G,aAAqC;AAC7E,WAAO;MACL1G,IAAID,OAAOC;MACXM,OAAOP,OAAOO;MACdC,SAASR,OAAOQ;MAChBC,WAAWT,OAAOS;MAClBE,WAAWX,OAAOW,UAAUiG,YAAW;MACvChG,WAAWZ,OAAOY,UAAUgG,YAAW;MACvC,GAAID,gBAAgBjF,SAAY;QAAEmF,oBAAoBF,YAAYtD,MAAM,GAAG,GAAA;MAAK,IAAI,CAAC;IACvF;EACF;EAEQjB,gBAAgBD,SAAyC;AAC/D,WAAO;MACLlC,IAAIkC,QAAQlC;MACZsD,MAAMpB,QAAQoB;MACdtB,SAASE,QAAQF;MACjBtB,WAAWwB,QAAQxB,UAAUiG,YAAW;MACxC,GAAIzE,QAAQqB,aAAa,OAAO;QAAEA,WAAWrB,QAAQqB;MAAU,IAAI,CAAC;MACpE,GAAIrB,QAAQsB,eAAe,OAAO;QAAEA,aAAatB,QAAQsB;MAAY,IAAI,CAAC;MAC1E,GAAItB,QAAQuB,aAAa,OAAO;QAAEA,WAAWvB,QAAQuB;MAAU,IAAI,CAAC;MACpE,GAAIvB,QAAQwB,SAAS,OAAO;QAAEA,OAAOxB,QAAQwB;MAAM,IAAI,CAAC;IAC1D;EACF;AACF;;;ACxUA,SAASmD,OAAAA,MAAKC,OAAOC,QAAAA,OAAMC,MAAAA,KAAIC,OAAAA,MAAKC,UAAAA,SAAQC,OAAAA,YAAW;AAwBvD,SAASC,mBACPC,SACAC,KAAwC;AAExC,QAAMC,QAAQF,QAAQG,IAAIF,IAAIG,OAAO;AACrC,MAAIF,UAAUG,QAAW;AACvB,WAAO;EACT;AACA,QAAMC,mBAAmBL,IAAIK,oBAAoB;AACjD,QAAMC,kBAAkBN,IAAIM,mBAAmB;AAC/C,QAAMC,sBAAsBP,IAAIQ,cAAcH,mBAAmBC;AACjE,SACGC,sBAAsB,MAAaN,MAAMQ,kBACzCJ,mBAAmB,OAAcJ,MAAMS,wBAAwBT,MAAMQ,mBACrEH,kBAAkB,OAAcL,MAAMU,uBAAuBV,MAAMQ,mBACnET,IAAIY,eAAe,MAAaX,MAAMY;AAE3C;AAjBSf;AAuBT,SAASgB,QACPf,SACAC,KAAwC;AAExC,SAAOA,IAAIe,WAAWjB,mBAAmBC,SAASC,GAAAA;AACpD;AALSc;AAQT,SAASE,UAAUC,OAAsB;AACvC,SAAO;IACLC,OAAO,oBAAIC,KAAK,GAAGF,MAAMG,OAAO,gBAAgB;IAChDC,KAAK,oBAAIF,KAAK,GAAGF,MAAMK,KAAK,gBAAgB;EAC9C;AACF;AALSN;AAcF,IAAMO,2BAAN,MAAMA;EArEb,OAqEaA;;;;EACX,YAA6BC,IAAoB;SAApBA,KAAAA;EAAqB;EAElD,MAAcC,cAAgD;AAC5D,UAAMC,OAAO,MAAM,KAAKF,GACrBG,OAAM,EACNC,KAAKC,iBAAAA,EACLC,MAAMC,IAAGF,kBAAkBG,WAAW,IAAA,CAAA;AACzC,UAAMjC,UAAU,oBAAIkC,IAAAA;AACpB,eAAWjC,OAAO0B,MAAM;AACtB3B,cAAQmC,IAAIlC,IAAIG,SAAS;QACvBM,iBAAiBT,IAAIS;QACrBI,kBAAkBb,IAAIa;QACtBH,sBAAsBV,IAAIU;QAC1BC,qBAAqBX,IAAIW;MAC3B,CAAA;IACF;AACA,WAAOZ;EACT;EAEA,MAAcoC,aACZlB,OACkD;AAClD,UAAM,EAAEC,OAAOG,IAAG,IAAKL,UAAUC,KAAAA;AACjC,UAAMS,OAAO,MAAM,KAAKF,GACrBG,OAAM,EACNC,KAAKQ,eAAAA,EACLN,MAAMO,KAAIC,KAAIF,gBAAgBG,WAAWrB,KAAAA,GAAQsB,KAAIJ,gBAAgBG,WAAWlB,GAAAA,CAAAA,CAAAA;AACnF,WAAOK;EACT;EAEA,MAAMe,aAAaxB,OAAkD;AACnE,UAAMlB,UAAU,MAAM,KAAK0B,YAAW;AACtC,UAAMC,OAAO,MAAM,KAAKS,aAAalB,KAAAA;AACrC,UAAMyB,UAAU,oBAAIT,IAAAA;AAIpB,eAAWjC,OAAO0B,MAAM;AACtB,YAAMiB,SAASD,QAAQxC,IAAIF,IAAIG,OAAO,KAAK;QACzCyC,UAAU;QACVpC,aAAa;QACbI,cAAc;QACdG,SAAS;MACX;AACA4B,aAAOC,YAAY;AACnBD,aAAOnC,eAAeR,IAAIQ;AAC1BmC,aAAO/B,gBAAgBZ,IAAIY;AAC3B+B,aAAO5B,WAAWD,QAAQf,SAASC,GAAAA;AACnC0C,cAAQR,IAAIlC,IAAIG,SAASwC,MAAAA;IAC3B;AACA,UAAME,SAA0B,CAAA;AAChC,eAAW,CAAC1C,SAASwC,MAAAA,KAAWD,SAAS;AACvCG,aAAOC,KAAK;QACV3C;QACAyC,UAAUD,OAAOC;QACjBpC,aAAamC,OAAOnC;QACpBI,cAAc+B,OAAO/B;QACrBG,SAAS4B,OAAO5B;MAClB,CAAA;IACF;AACA8B,WAAOE,KACL,CAACC,MAAMC,UAAUA,MAAMlC,UAAUiC,KAAKjC,WAAWiC,KAAK7C,QAAQ+C,cAAcD,MAAM9C,OAAO,CAAA;AAE3F,WAAO0C;EACT;EAEA,MAAMM,aAAalC,OAAkD;AACnE,UAAMlB,UAAU,MAAM,KAAK0B,YAAW;AACtC,UAAMC,OAAO,MAAM,KAAKS,aAAalB,KAAAA;AACrC,UAAMmC,UAAU,oBAAInB,IAAAA;AACpB,eAAWjC,OAAO0B,MAAM;AACtB,YAAMiB,SAASS,QAAQlD,IAAIF,IAAIqD,QAAQ,KAAK;QAAET,UAAU;QAAGU,aAAa;QAAGvC,SAAS;MAAE;AACtF4B,aAAOC,YAAY;AACnBD,aAAOW,eAAetD,IAAIQ,cAAcR,IAAIY;AAC5C+B,aAAO5B,WAAWD,QAAQf,SAASC,GAAAA;AACnCoD,cAAQlB,IAAIlC,IAAIqD,UAAUV,MAAAA;IAC5B;AACA,UAAME,SAA0B,CAAA;AAChC,eAAW,CAACQ,UAAUV,MAAAA,KAAWS,SAAS;AACxCP,aAAOC,KAAK;QACVO;QACAT,UAAUD,OAAOC;QACjBU,aAAaX,OAAOW;QACpBvC,SAAS4B,OAAO5B;MAClB,CAAA;IACF;AACA8B,WAAOE,KACL,CAACC,MAAMC,UAAUA,MAAMlC,UAAUiC,KAAKjC,WAAWiC,KAAKK,SAASH,cAAcD,MAAMI,QAAQ,CAAA;AAE7F,WAAOR;EACT;EAEA,MAAMU,WAAWtC,OAAoD;AACnE,UAAMlB,UAAU,MAAM,KAAK0B,YAAW;AACtC,UAAMC,OAAO,MAAM,KAAKS,aAAalB,KAAAA;AACrC,UAAMuC,QAAQ,oBAAIvB,IAAAA;AAClB,eAAWjC,OAAO0B,MAAM;AACtB,YAAM+B,MAAMzD,IAAIuC,UAAUmB,YAAW,EAAGC,MAAM,GAAG,EAAA;AACjD,YAAMhB,SAASa,MAAMtD,IAAIuD,GAAAA,KAAQ;QAAEH,aAAa;QAAGvC,SAAS;MAAE;AAC9D4B,aAAOW,eAAetD,IAAIQ,cAAcR,IAAIY;AAC5C+B,aAAO5B,WAAWD,QAAQf,SAASC,GAAAA;AACnCwD,YAAMtB,IAAIuB,KAAKd,MAAAA;IACjB;AACA,UAAME,SAA4B,CAAA;AAClC,eAAW,CAACY,KAAKd,MAAAA,KAAWa,OAAO;AACjCX,aAAOC,KAAK;QAAEW;QAAKH,aAAaX,OAAOW;QAAavC,SAAS4B,OAAO5B;MAAQ,CAAA;IAC9E;AACA8B,WAAOE,KAAK,CAACC,MAAMC,UAAUD,KAAKS,IAAIP,cAAcD,MAAMQ,GAAG,CAAA;AAC7D,WAAOZ;EACT;EAEA,MAAMe,gBAAgBC,OAA+C;AACnE,UAAMnC,OAAO,MAAM,KAAKF,GACrBG,OAAO;MACNmC,YAAYC,cAAcC;MAC1BC,UAAUF,cAAcE;MACxBC,UAAUH,cAAcG;MACxBC,QAAQJ,cAAcI;MACtBC,UAAUC,aAAaD;MACvB7B,WAAWwB,cAAcxB;IAC3B,CAAA,EACCX,KAAKmC,aAAAA,EACLO,UAAUD,cAActC,IAAGgC,cAAcQ,WAAWF,aAAaL,EAAE,CAAA,EACnEQ,QAAQC,MAAKV,cAAcxB,SAAS,GAAGkC,MAAKV,cAAcC,EAAE,CAAA,EAC5DH,MAAMA,KAAAA;AACT,WAAOnC,KAAKgD,IAAI,CAAC1E,SAAS;MACxB8D,YAAY9D,IAAI8D;MAChBG,UAAUjE,IAAIiE;MACdC,UAAUlE,IAAIkE;MACdC,QAAQnE,IAAImE;MACZC,UAAUpE,IAAIoE;MACd7B,WAAWvC,IAAIuC,UAAUmB,YAAW;IACtC,EAAA;EACF;EAEA,MAAMiB,cAAcd,OAA6C;AAC/D,UAAMe,UAAU,MAAM,KAAKpD,GACxBG,OAAM,EACNC,KAAKiD,WAAAA,EACL/C,MAAMgD,QAAOD,YAAYE,SAAS,CAAA,EAClCP,QAAQC,MAAKI,YAAYG,SAAS,GAAGP,MAAKI,YAAYb,EAAE,CAAA,EACxDH,MAAMA,KAAAA;AACT,UAAMhB,SAA8B,CAAA;AACpC,eAAWoC,UAAUL,SAAS;AAC5B,YAAM,CAACM,eAAAA,IAAmB,MAAM,KAAK1D,GAClCG,OAAO;QAAEwD,OAAOC,MAAAA;MAAQ,CAAA,EACxBxD,KAAKyC,YAAAA,EACLvC,MAAMC,IAAGsC,aAAaD,UAAUa,OAAOjB,EAAE,CAAA;AAC5C,YAAMqB,YAAY,MAAM,KAAK7D,GAC1BG,OAAM,EACNC,KAAKQ,eAAAA,EACLN,MAAMC,IAAGK,gBAAgBgC,UAAUa,OAAOjB,EAAE,CAAA;AAC/C,YAAMV,cAAc+B,UAAUC,OAC5B,CAACC,KAAKvF,QAAQuF,MAAMvF,IAAIQ,cAAcR,IAAIY,cAC1C,CAAA;AAEFiC,aAAOC,KAAK;QACVsB,UAAUa,OAAOjB;QACjBwB,OAAOP,OAAOO;QACdnC,UAAU4B,OAAO5B;QACjBoC,cAAcP,iBAAiBC,SAAS;QACxC7B;QACAoC,gBAAgBT,OAAOD,UAAUtB,YAAW;MAC9C,CAAA;IACF;AACA,WAAOb;EACT;AACF;;;ACjPA,SAAS8C,OAAAA,MAAKC,MAAAA,WAAU;AASjB,IAAMC,sBAAN,MAAMA;EATb,OASaA;;;;EACX,YAA6BC,IAAoB;SAApBA,KAAAA;EAAqB;EAElD,MAAMC,iBAAiBC,OAAuC;AAC5D,UAAM,KAAKF,GACRG,OAAOC,iBAAAA,EACPC,IAAI;MAAEC,WAAW;IAAM,CAAA,EACvBC,MACCC,KAAIC,IAAGL,kBAAkBM,SAASR,MAAMQ,OAAO,GAAGD,IAAGL,kBAAkBE,WAAW,IAAA,CAAA,CAAA;AAEtF,UAAM,KAAKN,GAAGW,OAAOP,iBAAAA,EAAmBQ,OAAO;MAC7CC,IAAIC,OAAOC,WAAU;MACrBL,SAASR,MAAMQ;MACfM,iBAAiBd,MAAMc;MACvBC,kBAAkBf,MAAMe;MACxBC,sBAAsBhB,MAAMgB,wBAAwB;MACpDC,qBAAqBjB,MAAMiB,uBAAuB;MAClDC,eAAe,oBAAIC,KAAAA;MACnBf,WAAW;IACb,CAAA;EACF;EAEA,MAAMgB,oBAAkD;AACtD,UAAMC,OAAO,MAAM,KAAKvB,GACrBwB,OAAM,EACNC,KAAKrB,iBAAAA,EACLG,MAAME,IAAGL,kBAAkBE,WAAW,IAAA,CAAA;AACzC,WAAOiB,KAAKG,IAAI,CAACC,SAAS;MACxBjB,SAASiB,IAAIjB;MACbM,iBAAiBW,IAAIX;MACrBC,kBAAkBU,IAAIV;MACtBG,eAAeO,IAAIP,cAAcQ,YAAW;MAC5C,GAAID,IAAIT,wBAAwB,OAC5B;QAAEA,sBAAsBS,IAAIT;MAAqB,IACjD,CAAC;MACL,GAAIS,IAAIR,uBAAuB,OAAO;QAAEA,qBAAqBQ,IAAIR;MAAoB,IAAI,CAAC;IAC5F,EAAA;EACF;AACF;;;ACpDA,SACEU,0BACAC,qBACAC,mBACK;AACP,SAA6BC,cAAc;;;;;;;;AA8BpC,IAAMC,0BAAN,MAAMA,yBAAAA;SAAAA;;;EACX,OAAOC,QAAQC,SAAwD;AACrE,WAAO;MACLC,QAAQH;MACRI,QAAQ;MACRC,WAAW;QACT;UAAEC,SAASC;UAAmBC,YAAY,6BAAM,IAAID,kBAAkBL,QAAQO,EAAE,GAAtC;QAAwC;QAClF;UAAEH,SAASI;UAAaC,aAAaJ;QAAkB;QACvD;UACED,SAASM;UACTJ,YAAY,6BAAM,IAAII,yBAAyBV,QAAQO,EAAE,GAA7C;QACd;QACA;UAAEH,SAASO;UAA0BF,aAAaC;QAAyB;QAC3E;UACEN,SAASQ;UACTN,YAAY,6BAAM,IAAIM,oBAAoBZ,QAAQO,EAAE,GAAxC;QACd;QACA;UAAEH,SAASS;UAAqBJ,aAAaG;QAAoB;;MAEnEE,SAAS;QACPT;QACAG;QACAE;QACAC;QACAC;QACAC;;IAEJ;EACF;AACF;;;;","names":["index","integer","real","sqliteTable","text","agentThread","id","primaryKey","actorRef","notNull","tenantRef","title","persona","default","transient","mode","activeStreamId","createdAt","updatedAt","deletedAt","table","on","agentMessage","threadId","references","onDelete","role","$type","content","toolCalls","toolResults","followUps","usage","agentToolCall","messageId","toolName","toolType","input","output","status","executedByRef","executionMs","error","executedAt","agentTokenUsage","modelId","purpose","inputTokens","outputTokens","cacheWriteTokens","cacheReadTokens","costUsd","agentModelPricing","inputPricePer1m","outputPricePer1m","cacheWritePricePer1m","cacheReadPricePer1m","effectiveFrom","isCurrent","agentSchema","sql","statements","ensureAgentSchema","db","statement","run","sql","raw","and","asc","desc","eq","gte","inArray","isNull","lte","DrizzleAgentStore","db","createThread","input","now","Date","thread","id","crypto","randomUUID","actorRef","actor","tenantRef","title","persona","transient","activeStreamId","createdAt","updatedAt","deletedAt","insert","agentThread","values","toSummary","getThread","threadId","select","from","where","and","eq","isNull","undefined","messages","agentMessage","orderBy","asc","last","length","content","map","message","toStoredMessage","listThreads","limit","threads","desc","summaries","push","softDeleteThread","update","set","forkThread","fromMessageId","source","Error","cutoff","findIndex","kept","slice","fork","role","toolCalls","toolResults","followUps","usage","ownerOfThread","ownerOfToolCall","toolCallId","row","agentToolCall","innerJoin","messageId","runForToolCall","ownerOfActiveStream","runId","setTitle","setActiveStream","appendMessage","truncateFrom","doomedIds","delete","inArray","recordToolCall","toolName","toolType","output","status","executedByRef","executionMs","error","executedAt","updateToolCall","updates","recordUsage","agentTokenUsage","modelId","purpose","inputTokens","outputTokens","cacheWriteTokens","cacheReadTokens","costUsd","quotaToday","day","start","end","rows","gte","lte","usedTokens","reduce","sum","lastContent","toISOString","lastMessagePreview","and","count","desc","eq","gte","isNull","lte","estimateFromTokens","pricing","row","price","get","modelId","undefined","cacheWriteTokens","cacheReadTokens","uncachedInputTokens","inputTokens","inputPricePer1m","cacheWritePricePer1m","cacheReadPricePer1m","outputTokens","outputPricePer1m","rowCost","costUsd","dayBounds","range","start","Date","fromDay","end","toDay","DrizzleGovernanceQueries","db","loadPricing","rows","select","from","agentModelPricing","where","eq","isCurrent","Map","set","usageInRange","agentTokenUsage","and","gte","createdAt","lte","spendByModel","byModel","bucket","requests","result","push","sort","left","right","localeCompare","spendByActor","byActor","actorRef","totalTokens","usageTrend","byDay","day","toISOString","slice","recentToolCalls","limit","toolCallId","agentToolCall","id","toolName","toolType","status","threadId","agentMessage","innerJoin","messageId","orderBy","desc","map","recentThreads","threads","agentThread","isNull","deletedAt","updatedAt","thread","messageCountRow","value","count","usageRows","reduce","sum","title","messageCount","lastActivityAt","and","eq","DrizzlePricingStore","db","upsertModelPrice","input","update","agentModelPricing","set","isCurrent","where","and","eq","modelId","insert","values","id","crypto","randomUUID","inputPricePer1m","outputPricePer1m","cacheWritePricePer1m","cacheReadPricePer1m","effectiveFrom","Date","listCurrentPrices","rows","select","from","map","row","toISOString","AGENT_GOVERNANCE_QUERIES","AGENT_PRICING_STORE","AGENT_STORE","Module","DrizzleAgentStoreModule","forRoot","options","module","global","providers","provide","DrizzleAgentStore","useFactory","db","AGENT_STORE","useExisting","DrizzleGovernanceQueries","AGENT_GOVERNANCE_QUERIES","DrizzlePricingStore","AGENT_PRICING_STORE","exports"]}

package/package.json

@@ -0,0 +1,47 @@
+{
+  "name": "@dudousxd/nestjs-agent-store-drizzle",
+  "version": "0.1.0",
+  "description": "nestjs-agent Drizzle ORM agent store (threads, messages, tool calls, usage)",
+  "license": "MIT",
+  "author": "Davide Carvalho",
+  "type": "module",
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      },
+      "require": {
+        "types": "./dist/index.d.cts",
+        "default": "./dist/index.cjs"
+      }
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "dependencies": {
+    "@dudousxd/nestjs-agent-core": "0.1.0"
+  },
+  "peerDependencies": {
+    "@nestjs/common": "^10.0.0 || ^11.0.0",
+    "drizzle-orm": ">=0.40.0 <1.0.0"
+  },
+  "devDependencies": {
+    "@nestjs/common": "11.1.6",
+    "@types/better-sqlite3": "7.6.13",
+    "better-sqlite3": "12.11.1",
+    "drizzle-orm": "0.45.2",
+    "tsup": "8.3.5",
+    "typescript": "5.9.3",
+    "vitest": "2.1.9",
+    "@dudousxd/nestjs-agent-core": "0.1.0"
+  },
+  "scripts": {
+    "build": "tsup",
+    "typecheck": "tsc -p tsconfig.json --noEmit"
+  }
+}