@adhd/agent-store-runtime 2.1.0

package/src/db/schema.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.d.ts","sourceRoot":"","sources":["../../../../../../packages/agent/agent-store-runtime/src/db/schema.ts"],"names":[],"mappings":"AAWA,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAaxB,CAAC;AAKH,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAcxB,CAAC;AAKH,eAAO,MAAM,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAoBrB,CAAC;AAKH,eAAO,MAAM,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAoB1B,CAAC;AAKH,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAgBhC,CAAC;AAKF,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAQrC,CAAC;AAKH,eAAO,MAAM,cAAc;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAoC1B,CAAC"}

package/src/db/schema.js

@@ -0,0 +1,146 @@
+import { index, integer, sqliteTable, text, uniqueIndex } from "drizzle-orm/sqlite-core";
+// ──────────────────────────────────────────────
+// sessions
+// ──────────────────────────────────────────────
+export const sessionsTable = sqliteTable("sessions", {
+    id: text("id").primaryKey(),
+    agentName: text("agent_name")
+        .notNull(),
+    agentVersion: integer("agent_version").notNull(),
+    agentData: text("agent_data").notNull(),
+    status: text("status", {
+        enum: ["active", "closed"]
+    }).notNull().default("active"),
+    createdAt: text("created_at").notNull(),
+    updatedAt: text("updated_at").notNull(),
+    closedAt: text("closed_at"),
+    composedPromptId: text("composed_prompt_id"),
+});
+// ──────────────────────────────────────────────
+// messages
+// ──────────────────────────────────────────────
+export const messagesTable = sqliteTable("messages", {
+    id: text("id").primaryKey(),
+    sessionId: text("session_id")
+        .notNull()
+        .references(() => sessionsTable.id, {
+        onDelete: "cascade"
+    }),
+    role: text("role", {
+        enum: ["system", "user", "assistant", "tool"]
+    }).notNull(),
+    content: text("content"),
+    toolCalls: text("tool_calls"),
+    toolResults: text("tool_results"),
+    createdAt: text("created_at").notNull(),
+});
+// ──────────────────────────────────────────────
+// tasks
+// ──────────────────────────────────────────────
+export const tasksTable = sqliteTable("tasks", {
+    id: text("id").primaryKey(),
+    sessionId: text("session_id"),
+    parentTaskId: text("parent_task_id"),
+    isEphemeral: integer("is_ephemeral").notNull().default(0),
+    recursionDepth: integer("recursion_depth").notNull().default(0),
+    status: text("status", {
+        enum: ["pending", "running", "completed", "failed", "cancelled", "waiting", "awaiting_input"]
+    }).notNull().default("pending"),
+    prompt: text("prompt").notNull(),
+    result: text("result"),
+    error: text("error"),
+    createdAt: text("created_at").notNull(),
+    updatedAt: text("updated_at").notNull(),
+    completedAt: text("completed_at"),
+    cancelledAt: text("cancelled_at"),
+    depends_on: text("depends_on"),
+    on_upstream_failure: text("on_upstream_failure"),
+    inputs: text("inputs"),
+    resume_token: text("resume_token"),
+});
+// ──────────────────────────────────────────────
+// task_events
+// ──────────────────────────────────────────────
+export const taskEventsTable = sqliteTable("task_events", {
+    id: text("id").primaryKey(),
+    taskId: text("task_id")
+        .notNull()
+        .references(() => tasksTable.id, {
+        onDelete: "cascade"
+    }),
+    type: text("type", {
+        enum: [
+            "MODEL_REQUEST",
+            "MODEL_RESPONSE",
+            "TOOL_CALL",
+            "TOOL_RESULT",
+            "TASK_COMPLETED",
+            "TASK_FAILED",
+            "TASK_CANCELLED"
+        ]
+    }).notNull(),
+    payload: text("payload"),
+    createdAt: text("created_at").notNull(),
+});
+// ──────────────────────────────────────────────
+// composed_prompts
+// ──────────────────────────────────────────────
+export const composedPromptsTable = sqliteTable("composed_prompts", {
+    id: text("id").primaryKey(),
+    agentSlug: text("agent_slug").notNull(),
+    contextHash: text("context_hash").notNull(),
+    content: text("content").notNull(),
+    componentVersions: text("component_versions").notNull(),
+    createdAt: text("created_at").notNull(),
+}, (table) => [
+    uniqueIndex("idx_composed_prompts_agent_ctx").on(table.agentSlug, table.contextHash),
+]);
+// ──────────────────────────────────────────────
+// experiment_assignments
+// ──────────────────────────────────────────────
+export const experimentAssignmentsTable = sqliteTable("experiment_assignments", {
+    id: text("id").primaryKey(),
+    sessionId: text("session_id")
+        .notNull()
+        .references(() => sessionsTable.id, { onDelete: "cascade" }),
+    experimentSlug: text("experiment_slug").notNull(),
+    variant: text("variant").notNull(),
+    createdAt: text("created_at").notNull(),
+});
+// ──────────────────────────────────────────────
+// task_usage
+// ──────────────────────────────────────────────
+export const taskUsageTable = sqliteTable("task_usage", {
+    taskId: text("task_id").primaryKey(),
+    rootTaskId: text("root_task_id"),
+    agentName: text("agent_name").notNull(),
+    providerType: text("provider_type").notNull(),
+    model: text("model").notNull(),
+    // CUMULATIVE across the task's model calls (SUM). Because chat APIs are stateless,
+    // the whole history is re-sent every call, so this is billed spend — NOT a context
+    // size. Do not read it as one. See peakContextTokens. (FINDING-ORCH-007)
+    inputTokens: integer("input_tokens").notNull().default(0),
+    outputTokens: integer("output_tokens").notNull().default(0),
+    toolCallCount: integer("tool_call_count").notNull().default(0),
+    modelCalls: integer("model_calls").notNull().default(0),
+    latencyMs: integer("latency_ms").notNull().default(0),
+    isComplete: integer("is_complete").notNull().default(0),
+    stopReason: text("stop_reason"),
+    maxTokens: integer("max_tokens"),
+    cacheReadTokens: integer("cache_read_input_tokens"),
+    cacheCreationTokens: integer("cache_creation_input_tokens"),
+    // Full-price (cache-miss) input. Cache-hit vs cache-miss is a 50x price difference
+    // on deepseek-v4-flash, so this — not inputTokens — is what actually drives cost.
+    uncachedInputTokens: integer("uncached_input_tokens"),
+    // Reasoning tokens (reasoning models); billed as output.
+    reasoningTokens: integer("reasoning_tokens"),
+    // PEAK single-call input (MAX, not SUM) — the real context high-water mark, and the
+    // only number that may be compared against a model's context window.
+    peakContextTokens: integer("peak_context_tokens"),
+    // Which model call (1-based) hit that peak.
+    peakContextAt: integer("peak_context_at"),
+    createdAt: text("created_at").notNull(),
+}, (table) => [
+    index("idx_task_usage_root_task_id").on(table.rootTaskId),
+]);
+//# sourceMappingURL=schema.js.map

package/src/db/schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../../../../../packages/agent/agent-store-runtime/src/db/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,KAAK,EACL,OAAO,EACP,WAAW,EACX,IAAI,EACJ,WAAW,EACd,MAAM,yBAAyB,CAAC;AAEjC,iDAAiD;AACjD,WAAW;AACX,iDAAiD;AACjD,MAAM,CAAC,MAAM,aAAa,GAAG,WAAW,CAAC,UAAU,EAAE;IACjD,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE;IAC3B,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC;SACxB,OAAO,EAAE;IACd,YAAY,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC,OAAO,EAAE;IAChD,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACvC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE;QACnB,IAAI,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC;KAC7B,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC;IAC9B,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACvC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACvC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC;IAC3B,gBAAgB,EAAE,IAAI,CAAC,oBAAoB,CAAC;CAC/C,CAAC,CAAC;AAEH,iDAAiD;AACjD,WAAW;AACX,iDAAiD;AACjD,MAAM,CAAC,MAAM,aAAa,GAAG,WAAW,CAAC,UAAU,EAAE;IACjD,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE;IAC3B,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC;SACxB,OAAO,EAAE;SACT,UAAU,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE;QAChC,QAAQ,EAAE,SAAS;KACtB,CAAC;IACN,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;QACf,IAAI,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,CAAC;KAChD,CAAC,CAAC,OAAO,EAAE;IACZ,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;IACxB,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC;IAC7B,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC;IACjC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;CAC1C,CAAC,CAAC;AAEH,iDAAiD;AACjD,QAAQ;AACR,iDAAiD;AACjD,MAAM,CAAC,MAAM,UAAU,GAAG,WAAW,CAAC,OAAO,EAAE;IAC3C,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE;IAC3B,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC;IAC7B,YAAY,EAAE,IAAI,CAAC,gBAAgB,CAAC;IACpC,WAAW,EAAE,OAAO,CAAC,cAAc,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACzD,cAAc,EAAE,OAAO,CAAC,iBAAiB,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAC/D,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE;QACnB,IAAI,EAAE,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,gBAAgB,CAAC;KAChG,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC;IAC/B,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE;IAChC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC;IACtB,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC;IACpB,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACvC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACvC,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC;IACjC,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC;IACjC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC;IAC9B,mBAAmB,EAAE,IAAI,CAAC,qBAAqB,CAAC;IAChD,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC;IACtB,YAAY,EAAE,IAAI,CAAC,cAAc,CAAC;CACrC,CAAC,CAAC;AAEH,iDAAiD;AACjD,cAAc;AACd,iDAAiD;AACjD,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC,aAAa,EAAE;IACtD,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE;IAC3B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC;SAClB,OAAO,EAAE;SACT,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,EAAE,EAAE;QAC7B,QAAQ,EAAE,SAAS;KACtB,CAAC;IACN,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;QACf,IAAI,EAAE;YACF,eAAe;YACf,gBAAgB;YAChB,WAAW;YACX,aAAa;YACb,gBAAgB;YAChB,aAAa;YACb,gBAAgB;SACnB;KACJ,CAAC,CAAC,OAAO,EAAE;IACZ,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;IACxB,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;CAC1C,CAAC,CAAC;AAEH,iDAAiD;AACjD,mBAAmB;AACnB,iDAAiD;AACjD,MAAM,CAAC,MAAM,oBAAoB,GAAG,WAAW,CAC3C,kBAAkB,EAClB;IACI,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE;IAC3B,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACvC,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,OAAO,EAAE;IAC3C,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;IAClC,iBAAiB,EAAE,IAAI,CAAC,oBAAoB,CAAC,CAAC,OAAO,EAAE;IACvD,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;CAC1C,EACD,CAAC,KAAK,EAAE,EAAE,CAAC;IACP,WAAW,CAAC,gCAAgC,CAAC,CAAC,EAAE,CAC5C,KAAK,CAAC,SAAS,EACf,KAAK,CAAC,WAAW,CACpB;CACJ,CACJ,CAAC;AAEF,iDAAiD;AACjD,yBAAyB;AACzB,iDAAiD;AACjD,MAAM,CAAC,MAAM,0BAA0B,GAAG,WAAW,CAAC,wBAAwB,EAAE;IAC5E,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE;IAC3B,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC;SACxB,OAAO,EAAE;SACT,UAAU,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;IAChE,cAAc,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC,OAAO,EAAE;IACjD,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;IAClC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;CAC1C,CAAC,CAAC;AAEH,iDAAiD;AACjD,aAAa;AACb,iDAAiD;AACjD,MAAM,CAAC,MAAM,cAAc,GAAG,WAAW,CACrC,YAAY,EACZ;IACI,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,UAAU,EAAE;IACpC,UAAU,EAAE,IAAI,CAAC,cAAc,CAAC;IAChC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACvC,YAAY,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC,OAAO,EAAE;IAC7C,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE;IAC9B,mFAAmF;IACnF,mFAAmF;IACnF,yEAAyE;IACzE,WAAW,EAAE,OAAO,CAAC,cAAc,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACzD,YAAY,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3D,aAAa,EAAE,OAAO,CAAC,iBAAiB,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAC9D,UAAU,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACvD,SAAS,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACrD,UAAU,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACvD,UAAU,EAAE,IAAI,CAAC,aAAa,CAAC;IAC/B,SAAS,EAAE,OAAO,CAAC,YAAY,CAAC;IAChC,eAAe,EAAE,OAAO,CAAC,yBAAyB,CAAC;IACnD,mBAAmB,EAAE,OAAO,CAAC,6BAA6B,CAAC;IAC3D,mFAAmF;IACnF,kFAAkF;IAClF,mBAAmB,EAAE,OAAO,CAAC,uBAAuB,CAAC;IACrD,yDAAyD;IACzD,eAAe,EAAE,OAAO,CAAC,kBAAkB,CAAC;IAC5C,oFAAoF;IACpF,qEAAqE;IACrE,iBAAiB,EAAE,OAAO,CAAC,qBAAqB,CAAC;IACjD,4CAA4C;IAC5C,aAAa,EAAE,OAAO,CAAC,iBAAiB,CAAC;IACzC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;CAC1C,EACD,CAAC,KAAK,EAAE,EAAE,CAAC;IACP,KAAK,CAAC,6BAA6B,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC;CAC5D,CACJ,CAAC"}

package/src/index.d.ts

@@ -0,0 +1,13 @@
+export { SessionStore, estimateTokens, windowMessages } from './store/session-store.js';
+export { TaskStore } from './store/task-store.js';
+export { UsageClient } from './runtime/usage-client.js';
+export type { Scope, UsageTotals } from './runtime/usage-client.js';
+export * from './db/schema.js';
+export { generateId } from './utils/ids.js';
+export { nowIso } from './utils/timestamps.js';
+export { ToolError } from './validation/errors.js';
+export type { ErrorCode } from './validation/errors.js';
+export { sessionSchema, agentDefinitionStoredSchema, taskSchema, } from './validation/schemas.js';
+export type { SessionListInput, TaskListInput, } from './validation/schemas.js';
+export { logger } from './logger.js';
+//# sourceMappingURL=index.d.ts.map

package/src/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../packages/agent/agent-store-runtime/src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AACxF,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,YAAY,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAEpE,cAAc,gBAAgB,CAAC;AAE/B,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AAE/C,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,YAAY,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EACL,aAAa,EACb,2BAA2B,EAC3B,UAAU,GACX,MAAM,yBAAyB,CAAC;AACjC,YAAY,EACV,gBAAgB,EAChB,aAAa,GACd,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC"}

package/src/index.js

@@ -0,0 +1,11 @@
+// @adhd/agent-store-runtime — public barrel
+export { SessionStore, estimateTokens, windowMessages } from './store/session-store.js';
+export { TaskStore } from './store/task-store.js';
+export { UsageClient } from './runtime/usage-client.js';
+export * from './db/schema.js';
+export { generateId } from './utils/ids.js';
+export { nowIso } from './utils/timestamps.js';
+export { ToolError } from './validation/errors.js';
+export { sessionSchema, agentDefinitionStoredSchema, taskSchema, } from './validation/schemas.js';
+export { logger } from './logger.js';
+//# sourceMappingURL=index.js.map

package/src/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../packages/agent/agent-store-runtime/src/index.ts"],"names":[],"mappings":"AAAA,4CAA4C;AAE5C,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AACxF,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAGxD,cAAc,gBAAgB,CAAC;AAE/B,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AAE/C,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAEnD,OAAO,EACL,aAAa,EACb,2BAA2B,EAC3B,UAAU,GACX,MAAM,yBAAyB,CAAC;AAMjC,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC"}

package/src/logger.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Minimal structured logger written to stderr.
+ * Uses console.error so MCP stdio (stdout) is never corrupted.
+ */
+export declare const logger: {
+    info(obj: Record<string, unknown>, msg?: string): void;
+    warn(obj: Record<string, unknown>, msg?: string): void;
+    error(obj: Record<string, unknown>, msg?: string): void;
+};
+//# sourceMappingURL=logger.d.ts.map

package/src/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../../../../packages/agent/agent-store-runtime/src/logger.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,eAAO,MAAM,MAAM;cACP,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,QAAQ,MAAM,GAAG,IAAI;cAM5C,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,QAAQ,MAAM,GAAG,IAAI;eAM3C,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,QAAQ,MAAM,GAAG,IAAI;CAMxD,CAAC"}

package/src/logger.js

@@ -0,0 +1,19 @@
+/**
+ * Minimal structured logger written to stderr.
+ * Uses console.error so MCP stdio (stdout) is never corrupted.
+ */
+export const logger = {
+    info(obj, msg) {
+        const ts = new Date().toISOString();
+        console.error(JSON.stringify({ level: 'info', time: ts, ...obj, ...(msg ? { msg } : {}) }));
+    },
+    warn(obj, msg) {
+        const ts = new Date().toISOString();
+        console.error(JSON.stringify({ level: 'warn', time: ts, ...obj, ...(msg ? { msg } : {}) }));
+    },
+    error(obj, msg) {
+        const ts = new Date().toISOString();
+        console.error(JSON.stringify({ level: 'error', time: ts, ...obj, ...(msg ? { msg } : {}) }));
+    },
+};
+//# sourceMappingURL=logger.js.map

package/src/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../../../../../packages/agent/agent-store-runtime/src/logger.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,MAAM,CAAC,MAAM,MAAM,GAAG;IACpB,IAAI,CAAC,GAA4B,EAAE,GAAY;QAC7C,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACpC,OAAO,CAAC,KAAK,CACX,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAC7E,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,GAA4B,EAAE,GAAY;QAC7C,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACpC,OAAO,CAAC,KAAK,CACX,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAC7E,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,GAA4B,EAAE,GAAY;QAC9C,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACpC,OAAO,CAAC,KAAK,CACX,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAC9E,CAAC;IACJ,CAAC;CACF,CAAC"}

package/src/runtime/usage-client.d.ts

@@ -0,0 +1,25 @@
+import type { ExecutionContext, TokenUsage } from '@adhd/agent-base-types';
+export type Scope = 'task' | 'session' | 'agent';
+export interface UsageTotals {
+    inputTokens: number;
+    outputTokens: number;
+    cacheTokens: number;
+    modelCalls: number;
+}
+export declare class UsageClient {
+    private readonly db;
+    private readonly accumulators;
+    constructor(db: unknown);
+    create(taskId: string, _ctx: ExecutionContext): void;
+    remove(taskId: string): void;
+    recordModelCall(taskId: string, usage: TokenUsage | undefined): void;
+    markModelCallStart(taskId: string): void;
+    markModelCallEnd(taskId: string): void;
+    recordToolCall(taskId: string, toolName: string): void;
+    getToolCallCount(taskId: string, toolName: string): number;
+    getWallClockMs(taskId: string): number;
+    getModelMs(taskId: string): number;
+    getTotals(taskId: string, scope: Scope, sessionId?: string, agentName?: string): UsageTotals;
+    getUsageInWindow(scope: Scope, id: string, windowMs?: number, excludeTaskId?: string): number;
+}
+//# sourceMappingURL=usage-client.d.ts.map

package/src/runtime/usage-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-client.d.ts","sourceRoot":"","sources":["../../../../../../packages/agent/agent-store-runtime/src/runtime/usage-client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAE3E,MAAM,MAAM,KAAK,GAAG,MAAM,GAAG,SAAS,GAAG,OAAO,CAAC;AAEjD,MAAM,WAAW,WAAW;IAC1B,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;CACpB;AAcD,qBAAa,WAAW;IAGV,OAAO,CAAC,QAAQ,CAAC,EAAE;IAF/B,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAkC;gBAElC,EAAE,EAAE,OAAO;IAIxC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,gBAAgB,GAAG,IAAI;IAapD,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAM5B,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,GAAG,SAAS,GAAG,IAAI;IAapE,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAKxC,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAOtC,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,IAAI;IAOtD,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM;IAI1D,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;IAKtC,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;IAMlC,SAAS,CACP,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,KAAK,EACZ,SAAS,CAAC,EAAE,MAAM,EAClB,SAAS,CAAC,EAAE,MAAM,GACjB,WAAW;IAwEd,gBAAgB,CACd,KAAK,EAAE,KAAK,EACZ,EAAE,EAAE,MAAM,EACV,QAAQ,GAAE,MAA4B,EACtC,aAAa,CAAC,EAAE,MAAM,GACrB,MAAM;CA0CV"}

package/src/runtime/usage-client.js

@@ -0,0 +1,156 @@
+export class UsageClient {
+    db;
+    accumulators = new Map();
+    constructor(db) {
+        this.db = db;
+    }
+    // ── Accumulator lifecycle ───────────────────────────────────────────
+    create(taskId, _ctx) {
+        this.accumulators.set(taskId, {
+            taskId,
+            startedAtMs: Date.now(),
+            inputTokens: 0,
+            outputTokens: 0,
+            cacheTokens: 0,
+            modelCalls: 0,
+            totalModelMs: 0,
+            toolCalls: new Map(),
+        });
+    }
+    remove(taskId) {
+        this.accumulators.delete(taskId);
+    }
+    // ── Recording ───────────────────────────────────────────────────────
+    recordModelCall(taskId, usage) {
+        const acc = this.accumulators.get(taskId);
+        if (!acc)
+            return;
+        if (usage) {
+            acc.inputTokens += usage.inputTokens ?? 0;
+            acc.outputTokens += usage.outputTokens ?? 0;
+            acc.cacheTokens +=
+                (usage.cacheReadTokens ?? 0) + (usage.cacheCreationTokens ?? 0);
+        }
+        acc.modelCalls += 1;
+    }
+    markModelCallStart(taskId) {
+        const acc = this.accumulators.get(taskId);
+        if (acc)
+            acc.modelCallStartMs = Date.now();
+    }
+    markModelCallEnd(taskId) {
+        const acc = this.accumulators.get(taskId);
+        if (!acc || acc.modelCallStartMs === undefined)
+            return;
+        acc.totalModelMs += Date.now() - acc.modelCallStartMs;
+        acc.modelCallStartMs = undefined;
+    }
+    recordToolCall(taskId, toolName) {
+        const acc = this.accumulators.get(taskId);
+        if (!acc)
+            return;
+        const count = acc.toolCalls.get(toolName) ?? 0;
+        acc.toolCalls.set(toolName, count + 1);
+    }
+    getToolCallCount(taskId, toolName) {
+        return this.accumulators.get(taskId)?.toolCalls.get(toolName) ?? 0;
+    }
+    getWallClockMs(taskId) {
+        const acc = this.accumulators.get(taskId);
+        return acc ? Date.now() - acc.startedAtMs : 0;
+    }
+    getModelMs(taskId) {
+        return this.accumulators.get(taskId)?.totalModelMs ?? 0;
+    }
+    // ── Scope-aware totals ──────────────────────────────────────────────
+    getTotals(taskId, scope, sessionId, agentName) {
+        const acc = this.accumulators.get(taskId);
+        const inMem = acc
+            ? {
+                inputTokens: acc.inputTokens,
+                outputTokens: acc.outputTokens,
+                cacheTokens: acc.cacheTokens,
+                modelCalls: acc.modelCalls,
+            }
+            : { inputTokens: 0, outputTokens: 0, cacheTokens: 0, modelCalls: 0 };
+        if (scope === 'task' || !this.db)
+            return inMem;
+        try {
+            const db = this.db;
+            let row;
+            if (scope === 'session' && sessionId) {
+                row = db
+                    .prepare(`SELECT
+                            COALESCE(SUM(tu.input_tokens), 0) AS input,
+                            COALESCE(SUM(tu.output_tokens), 0) AS output,
+                            COALESCE(SUM(COALESCE(tu.cache_read_input_tokens,0) + COALESCE(tu.cache_creation_input_tokens,0)), 0) AS cache,
+                            COALESCE(SUM(tu.model_calls), 0) AS calls
+                        FROM task_usage tu
+                        JOIN tasks t ON tu.task_id = t.id
+                        WHERE t.session_id = ? AND tu.task_id != ?`)
+                    .get(sessionId, taskId);
+            }
+            else if (scope === 'agent') {
+                row = db
+                    .prepare(`SELECT
+                            COALESCE(SUM(input_tokens), 0) AS input,
+                            COALESCE(SUM(output_tokens), 0) AS output,
+                            COALESCE(SUM(COALESCE(cache_read_input_tokens,0) + COALESCE(cache_creation_input_tokens,0)), 0) AS cache,
+                            COALESCE(SUM(model_calls), 0) AS calls
+                        FROM task_usage
+                        WHERE agent_name = ? AND task_id != ?`)
+                    .get(agentName, taskId);
+            }
+            if (row) {
+                return {
+                    inputTokens: (row.input ?? 0) + inMem.inputTokens,
+                    outputTokens: (row.output ?? 0) + inMem.outputTokens,
+                    cacheTokens: (row.cache ?? 0) + inMem.cacheTokens,
+                    modelCalls: (row.calls ?? 0) + inMem.modelCalls,
+                };
+            }
+        }
+        catch {
+            // DB query failed — fall back to in-memory
+        }
+        return inMem;
+    }
+    // ── Time-window query ──────────────────────────────────────────────
+    getUsageInWindow(scope, id, windowMs = 24 * 60 * 60 * 1000, excludeTaskId) {
+        if (!this.db)
+            return 0;
+        try {
+            const db = this.db;
+            const since = new Date(Date.now() - windowMs).toISOString();
+            let row;
+            if (scope === 'session') {
+                const excludeClause = excludeTaskId ? ' AND tu.task_id != ?' : '';
+                const params = [id, since];
+                if (excludeTaskId)
+                    params.push(excludeTaskId);
+                row = db
+                    .prepare(`SELECT COALESCE(SUM(tu.input_tokens + tu.output_tokens + COALESCE(tu.cache_read_input_tokens,0) + COALESCE(tu.cache_creation_input_tokens,0)), 0) AS total
+                         FROM task_usage tu
+                         JOIN tasks t ON tu.task_id = t.id
+                         WHERE t.session_id = ? AND tu.created_at >= ?${excludeClause}`)
+                    .get(...params);
+            }
+            else if (scope === 'agent') {
+                const excludeClause = excludeTaskId ? ' AND task_id != ?' : '';
+                const params = [id, since];
+                if (excludeTaskId)
+                    params.push(excludeTaskId);
+                row = db
+                    .prepare(`SELECT COALESCE(SUM(input_tokens + output_tokens + COALESCE(cache_read_input_tokens,0) + COALESCE(cache_creation_input_tokens,0)), 0) AS total
+                         FROM task_usage
+                         WHERE agent_name = ? AND created_at >= ?${excludeClause}`)
+                    .get(...params);
+            }
+            return row?.total ?? 0;
+        }
+        catch {
+            return 0;
+        }
+    }
+}
+//# sourceMappingURL=usage-client.js.map

package/src/runtime/usage-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-client.js","sourceRoot":"","sources":["../../../../../../packages/agent/agent-store-runtime/src/runtime/usage-client.ts"],"names":[],"mappings":"AAuBA,MAAM,OAAO,WAAW;IAGO;IAFZ,YAAY,GAAG,IAAI,GAAG,EAAuB,CAAC;IAE/D,YAA6B,EAAW;QAAX,OAAE,GAAF,EAAE,CAAS;IAAG,CAAC;IAE5C,uEAAuE;IAEvE,MAAM,CAAC,MAAc,EAAE,IAAsB;QAC3C,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE;YAC5B,MAAM;YACN,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;YACvB,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,CAAC;YACd,UAAU,EAAE,CAAC;YACb,YAAY,EAAE,CAAC;YACf,SAAS,EAAE,IAAI,GAAG,EAAE;SACrB,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,MAAc;QACnB,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED,uEAAuE;IAEvE,eAAe,CAAC,MAAc,EAAE,KAA6B;QAC3D,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,CAAC,GAAG;YAAE,OAAO;QAEjB,IAAI,KAAK,EAAE,CAAC;YACV,GAAG,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC;YAC1C,GAAG,CAAC,YAAY,IAAI,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;YAC5C,GAAG,CAAC,WAAW;gBACb,CAAC,KAAK,CAAC,eAAe,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,mBAAmB,IAAI,CAAC,CAAC,CAAC;QACpE,CAAC;QACD,GAAG,CAAC,UAAU,IAAI,CAAC,CAAC;IACtB,CAAC;IAED,kBAAkB,CAAC,MAAc;QAC/B,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,GAAG;YAAE,GAAG,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7C,CAAC;IAED,gBAAgB,CAAC,MAAc;QAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,gBAAgB,KAAK,SAAS;YAAE,OAAO;QACvD,GAAG,CAAC,YAAY,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,gBAAgB,CAAC;QACtD,GAAG,CAAC,gBAAgB,GAAG,SAAS,CAAC;IACnC,CAAC;IAED,cAAc,CAAC,MAAc,EAAE,QAAgB;QAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,MAAM,KAAK,GAAG,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC/C,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;IACzC,CAAC;IAED,gBAAgB,CAAC,MAAc,EAAE,QAAgB;QAC/C,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACrE,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,OAAO,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC;IAED,UAAU,CAAC,MAAc;QACvB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,YAAY,IAAI,CAAC,CAAC;IAC1D,CAAC;IAED,uEAAuE;IAEvE,SAAS,CACP,MAAc,EACd,KAAY,EACZ,SAAkB,EAClB,SAAkB;QAElB,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,MAAM,KAAK,GAAG,GAAG;YACf,CAAC,CAAC;gBACE,WAAW,EAAE,GAAG,CAAC,WAAW;gBAC5B,YAAY,EAAE,GAAG,CAAC,YAAY;gBAC9B,WAAW,EAAE,GAAG,CAAC,WAAW;gBAC5B,UAAU,EAAE,GAAG,CAAC,UAAU;aAC3B;YACH,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;QAEvE,IAAI,KAAK,KAAK,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,KAAK,CAAC;QAE/C,IAAI,CAAC;YACH,MAAM,EAAE,GAAG,IAAI,CAAC,EAQf,CAAC;YAEF,IAAI,GAES,CAAC;YAEd,IAAI,KAAK,KAAK,SAAS,IAAI,SAAS,EAAE,CAAC;gBACrC,GAAG,GAAG,EAAE;qBACL,OAAO,CACN;;;;;;;mEAOuD,CACxD;qBACA,GAAG,CAAC,SAAS,EAAE,MAAM,CAAe,CAAC;YAC1C,CAAC;iBAAM,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;gBAC7B,GAAG,GAAG,EAAE;qBACL,OAAO,CACN;;;;;;8DAMkD,CACnD;qBACA,GAAG,CAAC,SAAS,EAAE,MAAM,CAAe,CAAC;YAC1C,CAAC;YAED,IAAI,GAAG,EAAE,CAAC;gBACR,OAAO;oBACL,WAAW,EAAE,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,WAAW;oBACjD,YAAY,EAAE,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,YAAY;oBACpD,WAAW,EAAE,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,WAAW;oBACjD,UAAU,EAAE,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,UAAU;iBAChD,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,2CAA2C;QAC7C,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,sEAAsE;IAEtE,gBAAgB,CACd,KAAY,EACZ,EAAU,EACV,WAAmB,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,EACtC,aAAsB;QAEtB,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,CAAC,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,EAAE,GAAG,IAAI,CAAC,EAIf,CAAC;YACF,MAAM,KAAK,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAE5D,IAAI,GAAkC,CAAC;YAEvC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAM,aAAa,GAAG,aAAa,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,EAAE,CAAC;gBAClE,MAAM,MAAM,GAAc,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;gBACtC,IAAI,aAAa;oBAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBAC9C,GAAG,GAAG,EAAE;qBACL,OAAO,CACN;;;wEAG4D,aAAa,EAAE,CAC5E;qBACA,GAAG,CAAC,GAAG,MAAM,CAAe,CAAC;YAClC,CAAC;iBAAM,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;gBAC7B,MAAM,aAAa,GAAG,aAAa,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC/D,MAAM,MAAM,GAAc,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;gBACtC,IAAI,aAAa;oBAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBAC9C,GAAG,GAAG,EAAE;qBACL,OAAO,CACN;;mEAEuD,aAAa,EAAE,CACvE;qBACA,GAAG,CAAC,GAAG,MAAM,CAAe,CAAC;YAClC,CAAC;YAED,OAAO,GAAG,EAAE,KAAK,IAAI,CAAC,CAAC;QACzB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;CACF"}

package/src/store/session-store.d.ts

@@ -0,0 +1,24 @@
+import type { BetterSQLite3Database } from 'drizzle-orm/better-sqlite3';
+import type { AgentDefinition, Message, Session } from '../validation/schemas.js';
+import type { SessionListInput } from '../validation/schemas.js';
+import type { IHookRegistry } from '@adhd/agent-base-types';
+export declare class SessionStore {
+    private readonly db;
+    private readonly hooks?;
+    constructor(db: BetterSQLite3Database<Record<string, never>>, hooks?: IHookRegistry | undefined);
+    create(input: {
+        agentName: string;
+        agentDefinition: AgentDefinition;
+        composedPromptId?: string;
+    }): Session;
+    read(id: string): Session;
+    getAgentDefinition(sessionId: string): AgentDefinition;
+    list(input: SessionListInput): Session[];
+    close(id: string): Session;
+    clearMessages(sessionId: string): number;
+    appendMessage(sessionId: string, message: Message): void;
+    getMessages(sessionId: string): Message[];
+}
+export declare function estimateTokens(messages: Message[]): number;
+export declare function windowMessages(messages: Message[], tokenLimit: number): Message[];
+//# sourceMappingURL=session-store.d.ts.map

package/src/store/session-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-store.d.ts","sourceRoot":"","sources":["../../../../../../packages/agent/agent-store-runtime/src/store/session-store.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAIxE,OAAO,KAAK,EACV,eAAe,EACf,OAAO,EACP,OAAO,EACR,MAAM,0BAA0B,CAAC;AAKlC,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAIjE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAE5D,qBAAa,YAAY;IAErB,OAAO,CAAC,QAAQ,CAAC,EAAE;IACnB,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;gBADN,EAAE,EAAE,qBAAqB,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,EAChD,KAAK,CAAC,EAAE,aAAa,YAAA;IAGxC,MAAM,CAAC,KAAK,EAAE;QACZ,SAAS,EAAE,MAAM,CAAC;QAClB,eAAe,EAAE,eAAe,CAAC;QACjC,gBAAgB,CAAC,EAAE,MAAM,CAAC;KAC3B,GAAG,OAAO;IA4BX,IAAI,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAsBzB,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,eAAe;IAiBtD,IAAI,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,EAAE;IAiCxC,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAyB1B,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM;IAsBxC,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI;IAiBxD,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,EAAE;CAiB1C;AAED,wBAAgB,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CAW1D;AAED,wBAAgB,cAAc,CAC5B,QAAQ,EAAE,OAAO,EAAE,EACnB,UAAU,EAAE,MAAM,GACjB,OAAO,EAAE,CAsBX"}

package/src/store/session-store.js

@@ -0,0 +1,183 @@
+import { and, eq } from 'drizzle-orm';
+import { messagesTable, sessionsTable } from '../db/schema.js';
+import { logger } from '../logger.js';
+import { agentDefinitionStoredSchema, sessionSchema, } from '../validation/schemas.js';
+import { ToolError } from '../validation/errors.js';
+import { generateId } from '../utils/ids.js';
+import { nowIso } from '../utils/timestamps.js';
+export class SessionStore {
+    db;
+    hooks;
+    constructor(db, hooks) {
+        this.db = db;
+        this.hooks = hooks;
+    }
+    create(input) {
+        const now = nowIso();
+        const id = generateId();
+        this.db
+            .insert(sessionsTable)
+            .values({
+            id,
+            agentName: input.agentName,
+            agentVersion: input.agentDefinition.version,
+            agentData: JSON.stringify(input.agentDefinition),
+            status: 'active',
+            createdAt: now,
+            updatedAt: now,
+            composedPromptId: input.composedPromptId ?? null,
+        })
+            .run();
+        logger.info({ sessionId: id, agentName: input.agentName }, 'Session created');
+        const session = this.read(id);
+        void this.hooks?.emit('session:created', { session });
+        return session;
+    }
+    read(id) {
+        const row = this.db
+            .select()
+            .from(sessionsTable)
+            .where(eq(sessionsTable.id, id))
+            .get();
+        if (!row) {
+            throw new ToolError('SESSION_NOT_FOUND', `Session '${id}' not found`);
+        }
+        return sessionSchema.parse({
+            id: row.id,
+            agentName: row.agentName,
+            agentVersion: row.agentVersion,
+            status: row.status,
+            createdAt: row.createdAt,
+            updatedAt: row.updatedAt,
+            closedAt: row.closedAt ?? undefined,
+        });
+    }
+    getAgentDefinition(sessionId) {
+        const row = this.db
+            .select()
+            .from(sessionsTable)
+            .where(eq(sessionsTable.id, sessionId))
+            .get();
+        if (!row) {
+            throw new ToolError('SESSION_NOT_FOUND', `Session '${sessionId}' not found`);
+        }
+        return agentDefinitionStoredSchema.parse(JSON.parse(row.agentData));
+    }
+    list(input) {
+        const conditions = [];
+        if (input.agentName) {
+            conditions.push(eq(sessionsTable.agentName, input.agentName));
+        }
+        if (input.status) {
+            conditions.push(eq(sessionsTable.status, input.status));
+        }
+        const rows = conditions.length > 0
+            ? this.db
+                .select()
+                .from(sessionsTable)
+                .where(and(...conditions))
+                .all()
+            : this.db.select().from(sessionsTable).all();
+        return rows.map((row) => sessionSchema.parse({
+            id: row.id,
+            agentName: row.agentName,
+            agentVersion: row.agentVersion,
+            status: row.status,
+            createdAt: row.createdAt,
+            updatedAt: row.updatedAt,
+            closedAt: row.closedAt ?? undefined,
+        }));
+    }
+    close(id) {
+        const session = this.read(id);
+        if (session.status === 'closed') {
+            throw new ToolError('SESSION_CLOSED', `Session '${id}' is already closed`);
+        }
+        const now = nowIso();
+        this.db
+            .update(sessionsTable)
+            .set({
+            status: 'closed',
+            closedAt: now,
+            updatedAt: now,
+        })
+            .where(eq(sessionsTable.id, id))
+            .run();
+        logger.info({ sessionId: id }, 'Session closed');
+        return this.read(id);
+    }
+    clearMessages(sessionId) {
+        const session = this.read(sessionId);
+        if (session.status === 'closed') {
+            throw new ToolError('SESSION_CLOSED', `Session '${sessionId}' is closed; cannot clear context`);
+        }
+        const result = this.db
+            .delete(messagesTable)
+            .where(eq(messagesTable.sessionId, sessionId))
+            .run();
+        logger.info({ sessionId, cleared: result.changes }, 'Session context cleared');
+        return result.changes;
+    }
+    appendMessage(sessionId, message) {
+        this.db
+            .insert(messagesTable)
+            .values({
+            id: message.id,
+            sessionId,
+            role: message.role,
+            content: message.content ?? null,
+            toolCalls: message.toolCalls ? JSON.stringify(message.toolCalls) : null,
+            toolResults: message.toolResults
+                ? JSON.stringify(message.toolResults)
+                : null,
+            createdAt: message.createdAt,
+        })
+            .run();
+    }
+    getMessages(sessionId) {
+        const rows = this.db
+            .select()
+            .from(messagesTable)
+            .where(eq(messagesTable.sessionId, sessionId))
+            .all();
+        return rows.map((row) => ({
+            id: row.id,
+            sessionId: row.sessionId,
+            role: row.role,
+            content: row.content ?? undefined,
+            toolCalls: row.toolCalls ? JSON.parse(row.toolCalls) : undefined,
+            toolResults: row.toolResults ? JSON.parse(row.toolResults) : undefined,
+            createdAt: row.createdAt,
+        }));
+    }
+}
+export function estimateTokens(messages) {
+    return Math.ceil(messages.reduce((sum, m) => {
+        return (sum +
+            (m.content?.length ?? 0) +
+            (m.toolCalls ? JSON.stringify(m.toolCalls).length : 0) +
+            (m.toolResults ? JSON.stringify(m.toolResults).length : 0));
+    }, 0) / 4);
+}
+export function windowMessages(messages, tokenLimit) {
+    if (tokenLimit <= 0)
+        return messages;
+    if (estimateTokens(messages) <= tokenLimit)
+        return messages;
+    const systemMessages = messages.filter((m) => m.role === 'system');
+    const nonSystemMessages = messages.filter((m) => m.role !== 'system');
+    const systemBudget = estimateTokens(systemMessages);
+    const remaining = Math.max(0, tokenLimit - systemBudget);
+    const selected = [];
+    let used = 0;
+    for (let i = nonSystemMessages.length - 1; i >= 0; i--) {
+        const msg = nonSystemMessages[i];
+        const cost = estimateTokens([msg]);
+        if (used + cost > remaining && selected.length > 0)
+            break;
+        selected.unshift(msg);
+        used += cost;
+    }
+    return [...systemMessages, ...selected];
+}
+//# sourceMappingURL=session-store.js.map