@treeseed/sdk 0.1.1 → 0.3.0

package/dist/d1-store.js

@@ -1,8 +1,12 @@
 import crypto from "node:crypto";
 import { applyFilters, applySort } from "./sdk-filters.js";
+import { normalizeFilterFields, normalizeMutationData, normalizeRecordToCanonicalShape, normalizeSortFields } from "./sdk-fields.js";
+import { assertExpectedVersion } from "./sdk-version.js";
+import { resolveModelDefinition } from "./model-registry.js";
 import { CursorStore } from "./stores/cursor-store.js";
 import { LeaseStore } from "./stores/lease-store.js";
 import { MessageStore } from "./stores/message-store.js";
+import { OperationalStore } from "./stores/operational-store.js";
 import { RunStore } from "./stores/run-store.js";
 import { SubscriptionStore } from "./stores/subscription-store.js";
 function nowIso() {
@@ -11,6 +15,16 @@ function nowIso() {
 function nextLeaseToken() {
   return crypto.randomUUID();
 }
+function pickSortForRequest(request, defaultField) {
+  switch (request.strategy) {
+    case "oldest":
+      return [{ field: defaultField, direction: "asc" }];
+    case "highest_priority":
+    case "latest":
+    default:
+      return [{ field: defaultField, direction: "desc" }];
+  }
+}
 function filterSinceField(model) {
   switch (model) {
     case "message":
@@ -32,6 +46,12 @@ class MemoryAgentDatabase {
   runs = /* @__PURE__ */ new Map();
   contentLeases = /* @__PURE__ */ new Map();
   cursors = /* @__PURE__ */ new Map();
+  workDays = /* @__PURE__ */ new Map();
+  tasks = /* @__PURE__ */ new Map();
+  taskEvents = /* @__PURE__ */ new Map();
+  taskOutputs = /* @__PURE__ */ new Map();
+  graphRuns = /* @__PURE__ */ new Map();
+  reports = /* @__PURE__ */ new Map();
   messageId = 0;
   constructor(seed) {
     for (const item of seed?.subscriptions ?? []) {
@@ -82,6 +102,18 @@ class MemoryAgentDatabase {
         token: lease.token
       }));
     }
+    if (model === "work_day") {
+      return [...this.workDays.values()];
+    }
+    if (model === "task") {
+      return [...this.tasks.values()];
+    }
+    if (model === "graph_run") {
+      return [...this.graphRuns.values()];
+    }
+    if (model === "report") {
+      return [...this.reports.values()];
+    }
     throw new Error(`Unsupported D1 model "${model}".`);
   }
   async get(request) {
@@ -110,13 +142,29 @@ class MemoryAgentDatabase {
         token: lease.token
       } : null;
     }
+    if (request.model === "work_day") {
+      return this.workDays.get(key) ?? null;
+    }
+    if (request.model === "task") {
+      return this.tasks.get(key) ?? null;
+    }
+    if (request.model === "graph_run") {
+      return this.graphRuns.get(key) ?? null;
+    }
+    if (request.model === "report") {
+      return this.reports.get(key) ?? null;
+    }
     return this.rowsForModel(request.model).find(
       (row) => [row.id, row.email, row.runId].map((value) => String(value ?? "")).includes(key)
     ) ?? null;
   }
   async search(request) {
-    const filtered = applyFilters(this.rowsForModel(request.model), request.filters);
-    const sorted = applySort(filtered, request.sort);
+    const definition = resolveModelDefinition(request.model);
+    const rows = this.rowsForModel(request.model).map(
+      (row) => normalizeRecordToCanonicalShape(definition, row)
+    );
+    const filtered = applyFilters(rows, normalizeFilterFields(definition, request.filters), definition);
+    const sorted = applySort(filtered, normalizeSortFields(definition, request.sort), definition);
     return sorted.slice(0, request.limit ?? sorted.length);
   }
   async follow(request) {
@@ -138,14 +186,35 @@ class MemoryAgentDatabase {
   }
   async pick(request) {
     if (request.model === "message") {
-      const item = await this.claimMessage({
-        workerId: request.workerId,
-        messageTypes: request.filters?.filter((filter) => filter.field === "type" && filter.op === "in").flatMap((filter) => Array.isArray(filter.value) ? filter.value.map(String) : []),
-        leaseSeconds: request.leaseSeconds
+      const candidates = [...this.messages.values()].filter(
+        (message) => (message.status === "pending" || message.status === "failed") && new Date(message.availableAt).valueOf() <= Date.now() && (!request.filters?.filter((filter) => filter.field === "type" && filter.op === "in").flatMap((filter) => Array.isArray(filter.value) ? filter.value.map(String) : []).length || request.filters.filter((filter) => filter.field === "type" && filter.op === "in").flatMap((filter) => Array.isArray(filter.value) ? filter.value.map(String) : []).includes(message.type))
+      ).sort((left, right) => {
+        if (request.strategy === "oldest") {
+          return left.availableAt.localeCompare(right.availableAt) || right.priority - left.priority;
+        }
+        if (request.strategy === "latest") {
+          return right.availableAt.localeCompare(left.availableAt) || right.priority - left.priority;
+        }
+        return right.priority - left.priority || left.availableAt.localeCompare(right.availableAt);
       });
+      const pending = candidates[0];
+      if (!pending) {
+        return { item: null, leaseToken: null };
+      }
+      const claimedAt = nowIso();
+      const next = {
+        ...pending,
+        status: "claimed",
+        claimedBy: request.workerId,
+        claimedAt,
+        leaseExpiresAt: new Date(Date.now() + request.leaseSeconds * 1e3).toISOString(),
+        attempts: pending.attempts + 1,
+        updatedAt: claimedAt
+      };
+      this.messages.set(next.id, next);
       return {
-        item,
-        leaseToken: item ? nextLeaseToken() : null
+        item: next,
+        leaseToken: nextLeaseToken()
       };
     }
     if (request.model === "content_lease") {
@@ -163,7 +232,7 @@ class MemoryAgentDatabase {
     const items = await this.search({
       model: request.model,
       filters: request.filters,
-      sort: [{ field: filterSinceField(request.model), direction: "desc" }]
+      sort: pickSortForRequest(request, filterSinceField(request.model))
     });
     return {
       item: items[0] ?? null,
@@ -171,38 +240,40 @@ class MemoryAgentDatabase {
     };
   }
   async create(request) {
+    const definition = resolveModelDefinition(request.model);
+    const data = normalizeMutationData(definition, request.data);
     switch (request.model) {
       case "message":
         return await this.createMessage({
-          type: String(request.data.type ?? "message.created"),
-          payload: request.data.payload ?? request.data,
-          relatedModel: typeof request.data.relatedModel === "string" ? request.data.relatedModel : null,
-          relatedId: typeof request.data.relatedId === "string" ? request.data.relatedId : null,
-          priority: Number(request.data.priority ?? 0),
-          maxAttempts: Number(request.data.maxAttempts ?? 3),
+          type: String(data.type ?? "message.created"),
+          payload: data.payload ?? data,
+          relatedModel: typeof data.related_model === "string" ? data.related_model : null,
+          relatedId: typeof data.related_id === "string" ? data.related_id : null,
+          priority: Number(data.priority ?? 0),
+          maxAttempts: Number(data.maxAttempts ?? 3),
           actor: request.actor
         });
       case "subscription": {
         const record = {
           id: this.subscriptions.size + 1,
-          email: String(request.data.email ?? ""),
-          name: request.data.name ? String(request.data.name) : null,
-          status: String(request.data.status ?? "active"),
-          source: String(request.data.source ?? "sdk"),
-          consent_at: String(request.data.consent_at ?? nowIso()),
-          created_at: String(request.data.created_at ?? nowIso()),
-          updated_at: String(request.data.updated_at ?? nowIso()),
-          ip_hash: String(request.data.ip_hash ?? "")
+          email: String(data.email ?? ""),
+          name: data.name ? String(data.name) : null,
+          status: String(data.status ?? "active"),
+          source: String(data.source ?? "sdk"),
+          consent_at: String(data.consent_at ?? nowIso()),
+          created_at: String(data.created_at ?? nowIso()),
+          updated_at: String(data.updated_at ?? nowIso()),
+          ip_hash: String(data.ip_hash ?? "")
         };
         this.subscriptions.set(String(record.id), record);
         return record;
       }
       case "agent_run":
-        return this.recordRun({ run: request.data });
+        return this.recordRun({ run: data });
       case "agent_cursor": {
-        const agentSlug = String(request.data.agentSlug ?? "");
-        const cursorKey = String(request.data.cursorKey ?? "");
-        const cursorValue = String(request.data.cursorValue ?? "");
+        const agentSlug = String(data.agent_slug ?? "");
+        const cursorKey = String(data.cursor_key ?? "");
+        const cursorValue = String(data.cursor_value ?? "");
         this.cursors.set(`${agentSlug}:${cursorKey}`, cursorValue);
         return {
           agentSlug,
@@ -213,68 +284,182 @@ class MemoryAgentDatabase {
       }
       case "content_lease": {
         const token = await this.tryClaimContentLease({
-          model: String(request.data.model ?? ""),
-          itemKey: String(request.data.itemKey ?? ""),
-          claimedBy: String(request.data.claimedBy ?? request.actor),
-          leaseSeconds: Number(request.data.leaseSeconds ?? 300)
+          model: String(data.model ?? ""),
+          itemKey: String(data.item_key ?? ""),
+          claimedBy: String(data.claimed_by ?? request.actor),
+          leaseSeconds: Number(data.leaseSeconds ?? 300)
         });
-        const lease = this.contentLeases.get(`${request.data.model}:${request.data.itemKey}`);
+        const lease = this.contentLeases.get(`${data.model}:${data.item_key}`);
         return {
-          model: String(request.data.model ?? ""),
-          itemKey: String(request.data.itemKey ?? ""),
-          claimedBy: String(request.data.claimedBy ?? request.actor),
+          model: String(data.model ?? ""),
+          itemKey: String(data.item_key ?? ""),
+          claimedBy: String(data.claimed_by ?? request.actor),
           claimedAt: String(lease?.claimedAt ?? nowIso()),
           leaseExpiresAt: String(lease?.leaseExpiresAt ?? nowIso()),
           token: String(token ?? lease?.token ?? "")
         };
       }
+      case "work_day":
+        return this.startWorkDay({
+          id: typeof data.id === "string" ? data.id : void 0,
+          projectId: String(data.projectId ?? data.project_id ?? ""),
+          capacityBudget: Number(data.capacityBudget ?? data.capacity_budget ?? 0),
+          graphVersion: typeof (data.graphVersion ?? data.graph_version) === "string" ? String(data.graphVersion ?? data.graph_version) : null,
+          summary: data.summary ?? null,
+          actor: request.actor
+        });
+      case "task":
+        return this.createTask({
+          id: typeof data.id === "string" ? data.id : void 0,
+          workDayId: String(data.workDayId ?? data.work_day_id ?? ""),
+          agentId: String(data.agentId ?? data.agent_id ?? ""),
+          type: String(data.type ?? ""),
+          state: typeof data.state === "string" ? data.state : "pending",
+          priority: Number(data.priority ?? 0),
+          idempotencyKey: String(data.idempotencyKey ?? data.idempotency_key ?? ""),
+          payload: data.payload ?? data,
+          payloadHash: typeof (data.payloadHash ?? data.payload_hash) === "string" ? String(data.payloadHash ?? data.payload_hash) : null,
+          maxAttempts: Number(data.maxAttempts ?? data.max_attempts ?? 3),
+          availableAt: typeof (data.availableAt ?? data.available_at) === "string" ? String(data.availableAt ?? data.available_at) : void 0,
+          graphVersion: typeof (data.graphVersion ?? data.graph_version) === "string" ? String(data.graphVersion ?? data.graph_version) : null,
+          parentTaskId: typeof (data.parentTaskId ?? data.parent_task_id) === "string" ? String(data.parentTaskId ?? data.parent_task_id) : null,
+          actor: request.actor
+        });
+      case "graph_run": {
+        const record = {
+          id: String(data.id ?? crypto.randomUUID()),
+          workDayId: String(data.workDayId ?? data.work_day_id ?? ""),
+          corpusHash: String(data.corpusHash ?? data.corpus_hash ?? ""),
+          graphVersion: String(data.graphVersion ?? data.graph_version ?? ""),
+          queryJson: typeof (data.queryJson ?? data.query_json) === "string" ? String(data.queryJson ?? data.query_json) : null,
+          seedIdsJson: typeof (data.seedIdsJson ?? data.seed_ids_json) === "string" ? String(data.seedIdsJson ?? data.seed_ids_json) : null,
+          selectedNodeIdsJson: typeof (data.selectedNodeIdsJson ?? data.selected_node_ids_json) === "string" ? String(data.selectedNodeIdsJson ?? data.selected_node_ids_json) : null,
+          statsJson: typeof (data.statsJson ?? data.stats_json) === "string" ? String(data.statsJson ?? data.stats_json) : null,
+          snapshotRef: typeof (data.snapshotRef ?? data.snapshot_ref) === "string" ? String(data.snapshotRef ?? data.snapshot_ref) : null,
+          createdAt: String(data.createdAt ?? data.created_at ?? nowIso())
+        };
+        this.graphRuns.set(record.id, record);
+        return record;
+      }
+      case "report":
+        return this.createReport({
+          id: typeof data.id === "string" ? data.id : void 0,
+          workDayId: String(data.workDayId ?? data.work_day_id ?? ""),
+          kind: String(data.kind ?? "workday_summary"),
+          body: data.body ?? data,
+          renderedRef: typeof (data.renderedRef ?? data.rendered_ref) === "string" ? String(data.renderedRef ?? data.rendered_ref) : null,
+          sentAt: typeof (data.sentAt ?? data.sent_at) === "string" ? String(data.sentAt ?? data.sent_at) : null,
+          actor: request.actor
+        });
       default:
         throw new Error(`Unsupported D1 create model "${request.model}".`);
     }
   }
   async update(request) {
+    const definition = resolveModelDefinition(request.model);
+    const data = normalizeMutationData(definition, request.data);
     switch (request.model) {
       case "message": {
-        const current = this.messages.get(Number(request.id ?? request.key ?? request.data.id ?? 0));
+        const current = this.messages.get(Number(request.id ?? request.key ?? data.id ?? 0));
         if (!current) {
           return null;
         }
+        assertExpectedVersion(request.expectedVersion, current, `message ${current.id}`);
         const next = {
           ...current,
-          ...request.data,
+          ...data,
           updatedAt: nowIso()
         };
         this.messages.set(next.id, next);
         return next;
       }
       case "subscription": {
-        const key = String(request.id ?? request.key ?? request.data.email ?? "");
+        const key = String(request.id ?? request.key ?? data.email ?? "");
         const current = await this.get({ model: "subscription", key });
         if (!current) {
           return null;
         }
+        assertExpectedVersion(request.expectedVersion, current, `subscription "${current.email}"`);
         const next = {
           ...current,
-          ...request.data,
+          ...data,
           updated_at: nowIso()
         };
         this.subscriptions.set(String(next.id ?? next.email), next);
         return next;
       }
       case "agent_run":
-        return this.recordRun({ run: { ...request.data, runId: request.id ?? request.key ?? request.data.runId } });
+        assertExpectedVersion(
+          request.expectedVersion,
+          await this.get({ model: "agent_run", key: String(request.id ?? request.key ?? data.run_id ?? "") }),
+          `agent_run "${String(request.id ?? request.key ?? data.run_id ?? "")}"`
+        );
+        return this.recordRun({ run: { ...data, runId: request.id ?? request.key ?? data.run_id } });
       case "agent_cursor":
+        assertExpectedVersion(
+          request.expectedVersion,
+          await this.get({
+            model: "agent_cursor",
+            key: `${String(data.agent_slug ?? request.id ?? request.key ?? "")}:${String(data.cursor_key ?? request.slug ?? "")}`
+          }),
+          `agent_cursor "${String(data.agent_slug ?? request.id ?? request.key ?? "")}:${String(data.cursor_key ?? request.slug ?? "")}"`
+        );
         return this.create({
           model: "agent_cursor",
-          data: request.data,
+          data,
           actor: request.actor
         });
       case "content_lease":
+        assertExpectedVersion(
+          request.expectedVersion,
+          await this.get({
+            model: "content_lease",
+            key: `${String(data.model ?? request.id ?? "")}:${String(data.item_key ?? request.slug ?? request.key ?? "")}`
+          }),
+          `content_lease "${String(data.model ?? request.id ?? "")}:${String(data.item_key ?? request.slug ?? request.key ?? "")}"`
+        );
         return this.create({
           model: "content_lease",
-          data: request.data,
+          data,
           actor: request.actor
         });
+      case "work_day":
+        return this.closeWorkDay({
+          id: String(request.id ?? request.key ?? data.id ?? ""),
+          state: data.state ?? "completed",
+          summary: data.summary ?? null,
+          actor: request.actor
+        });
+      case "task": {
+        const taskId = String(request.id ?? request.key ?? data.id ?? "");
+        if (data.state === "completed") {
+          return this.completeTask({
+            id: taskId,
+            output: data.output ?? null,
+            outputRef: typeof data.outputRef === "string" ? data.outputRef : null,
+            summary: data.summary ?? null,
+            actor: request.actor
+          });
+        }
+        if (data.state === "failed") {
+          return this.failTask({
+            id: taskId,
+            errorCode: typeof data.errorCode === "string" ? data.errorCode : null,
+            errorMessage: String(data.errorMessage ?? "Task failed"),
+            retryable: Boolean(data.retryable),
+            nextVisibleAt: typeof data.nextVisibleAt === "string" ? data.nextVisibleAt : null,
+            actor: request.actor
+          });
+        }
+        return this.recordTaskProgress({
+          id: taskId,
+          workerId: typeof data.workerId === "string" ? data.workerId : null,
+          state: typeof data.state === "string" ? data.state : void 0,
+          appendEvent: data.appendEvent,
+          patch: data.patch ?? void 0,
+          actor: request.actor
+        });
+      }
       default:
         throw new Error(`Unsupported D1 update model "${request.model}".`);
     }
@@ -374,6 +559,197 @@ class MemoryAgentDatabase {
   inspectLeases() {
     return [...this.contentLeases.values()];
   }
+  async startWorkDay(request) {
+    const timestamp = nowIso();
+    const record = {
+      id: request.id ?? crypto.randomUUID(),
+      projectId: request.projectId,
+      state: "active",
+      capacityBudget: Number(request.capacityBudget ?? 0),
+      capacityUsed: 0,
+      graphVersion: request.graphVersion ?? null,
+      summaryJson: request.summary ? JSON.stringify(request.summary) : null,
+      startedAt: timestamp,
+      endedAt: null,
+      createdAt: timestamp,
+      updatedAt: timestamp
+    };
+    this.workDays.set(record.id, record);
+    return record;
+  }
+  async closeWorkDay(request) {
+    const existing = this.workDays.get(request.id);
+    if (!existing) return null;
+    const next = {
+      ...existing,
+      state: request.state ?? "completed",
+      summaryJson: request.summary ? JSON.stringify(request.summary) : existing.summaryJson,
+      endedAt: nowIso(),
+      updatedAt: nowIso()
+    };
+    this.workDays.set(next.id, next);
+    return next;
+  }
+  async createTask(request) {
+    const timestamp = nowIso();
+    const record = {
+      id: request.id ?? crypto.randomUUID(),
+      workDayId: request.workDayId,
+      agentId: request.agentId,
+      type: request.type,
+      state: request.state ?? "pending",
+      priority: Number(request.priority ?? 0),
+      idempotencyKey: request.idempotencyKey,
+      payloadJson: JSON.stringify(request.payload),
+      payloadHash: request.payloadHash ?? null,
+      attemptCount: 0,
+      maxAttempts: Number(request.maxAttempts ?? 3),
+      claimedBy: null,
+      leaseExpiresAt: null,
+      availableAt: request.availableAt ?? timestamp,
+      lastErrorCode: null,
+      lastErrorMessage: null,
+      graphVersion: request.graphVersion ?? null,
+      parentTaskId: request.parentTaskId ?? null,
+      createdAt: timestamp,
+      startedAt: null,
+      completedAt: null,
+      updatedAt: timestamp
+    };
+    this.tasks.set(record.id, record);
+    return record;
+  }
+  async claimTask(request) {
+    const existing = this.tasks.get(request.id);
+    if (!existing) return null;
+    const next = {
+      ...existing,
+      state: "claimed",
+      claimedBy: request.workerId,
+      leaseExpiresAt: new Date(Date.now() + request.leaseSeconds * 1e3).toISOString(),
+      attemptCount: existing.attemptCount + 1,
+      startedAt: existing.startedAt ?? nowIso(),
+      updatedAt: nowIso()
+    };
+    this.tasks.set(next.id, next);
+    return next;
+  }
+  async recordTaskProgress(request) {
+    const existing = this.tasks.get(request.id);
+    if (!existing) return null;
+    const nextPayload = {
+      ...JSON.parse(existing.payloadJson),
+      ...request.patch ?? {}
+    };
+    const next = {
+      ...existing,
+      state: request.state ?? existing.state,
+      claimedBy: request.workerId ?? existing.claimedBy,
+      payloadJson: JSON.stringify(nextPayload),
+      updatedAt: nowIso()
+    };
+    this.tasks.set(next.id, next);
+    if (request.appendEvent?.kind) {
+      await this.appendTaskEvent({
+        taskId: request.id,
+        kind: request.appendEvent.kind,
+        data: request.appendEvent.data,
+        actor: request.actor
+      });
+    }
+    return next;
+  }
+  async completeTask(request) {
+    const existing = this.tasks.get(request.id);
+    if (!existing) return null;
+    const next = {
+      ...existing,
+      state: "completed",
+      completedAt: nowIso(),
+      leaseExpiresAt: null,
+      updatedAt: nowIso()
+    };
+    this.tasks.set(next.id, next);
+    if (request.output) {
+      const outputs = this.taskOutputs.get(request.id) ?? [];
+      outputs.push({
+        id: crypto.randomUUID(),
+        taskId: request.id,
+        outputJson: JSON.stringify(request.output),
+        outputRef: request.outputRef ?? null,
+        createdAt: nowIso()
+      });
+      this.taskOutputs.set(request.id, outputs);
+    }
+    if (request.summary) {
+      await this.appendTaskEvent({
+        taskId: request.id,
+        kind: "completed",
+        data: request.summary,
+        actor: request.actor
+      });
+    }
+    return next;
+  }
+  async failTask(request) {
+    const existing = this.tasks.get(request.id);
+    if (!existing) return null;
+    const next = {
+      ...existing,
+      state: request.retryable ? "pending" : "failed",
+      availableAt: request.nextVisibleAt ?? existing.availableAt,
+      lastErrorCode: request.errorCode ?? null,
+      lastErrorMessage: request.errorMessage,
+      leaseExpiresAt: null,
+      updatedAt: nowIso()
+    };
+    this.tasks.set(next.id, next);
+    return next;
+  }
+  async appendTaskEvent(request) {
+    const events = this.taskEvents.get(request.taskId) ?? [];
+    const next = {
+      id: crypto.randomUUID(),
+      taskId: request.taskId,
+      seq: events.length + 1,
+      kind: request.kind,
+      dataJson: JSON.stringify({ ...request.data ?? {}, actor: request.actor }),
+      createdAt: nowIso()
+    };
+    events.push(next);
+    this.taskEvents.set(request.taskId, events);
+    return next;
+  }
+  async searchTasks(request) {
+    return [...this.tasks.values()].filter((task) => !request.workDayId || task.workDayId === request.workDayId).filter((task) => !request.agentId || task.agentId === request.agentId).filter((task) => {
+      if (!request.state) return true;
+      const states = Array.isArray(request.state) ? request.state : [request.state];
+      return states.includes(task.state);
+    }).sort((left, right) => right.priority - left.priority || left.availableAt.localeCompare(right.availableAt)).slice(0, request.limit ?? 50);
+  }
+  async createReport(request) {
+    const record = {
+      id: request.id ?? crypto.randomUUID(),
+      workDayId: request.workDayId,
+      kind: request.kind,
+      bodyJson: JSON.stringify(request.body),
+      renderedRef: request.renderedRef ?? null,
+      sentAt: request.sentAt ?? null,
+      createdAt: nowIso()
+    };
+    this.reports.set(record.id, record);
+    return record;
+  }
+  async getManagerContext(taskId) {
+    const task = this.tasks.get(taskId) ?? null;
+    const workDay = task ? this.workDays.get(task.workDayId) ?? null : null;
+    return {
+      task,
+      workDay,
+      agent: null,
+      graph: workDay?.graphVersion ? { graphVersion: workDay.graphVersion } : null
+    };
+  }
 }
 class CloudflareD1AgentDatabase {
   constructor(db) {
@@ -383,6 +759,7 @@ class CloudflareD1AgentDatabase {
     this.runs = new RunStore(db);
     this.cursors = new CursorStore(db);
     this.leases = new LeaseStore(db);
+    this.operational = new OperationalStore(db);
   }
   db;
   subscriptions;
@@ -390,6 +767,7 @@ class CloudflareD1AgentDatabase {
   runs;
   cursors;
   leases;
+  operational;
   async get(request) {
     if (request.model === "subscription") {
       return this.subscriptions.getByKey(String(request.id ?? request.slug ?? request.key ?? ""));
@@ -406,23 +784,52 @@ class CloudflareD1AgentDatabase {
     if (request.model === "content_lease") {
       return this.leases.getByKey(String(request.id ?? request.key ?? request.slug ?? ""));
     }
+    if (request.model === "work_day") {
+      return this.operational.getWorkDay(String(request.id ?? request.key ?? request.slug ?? ""));
+    }
+    if (request.model === "task") {
+      return this.operational.getTask(String(request.id ?? request.key ?? request.slug ?? ""));
+    }
+    if (request.model === "report") {
+      return this.operational.getReport(String(request.id ?? request.key ?? request.slug ?? ""));
+    }
     throw new Error(`Unsupported D1 get model "${request.model}".`);
   }
   async search(request) {
+    const definition = resolveModelDefinition(request.model);
+    const normalizedRequest = {
+      ...request,
+      filters: normalizeFilterFields(definition, request.filters),
+      sort: normalizeSortFields(definition, request.sort)
+    };
     if (request.model === "subscription") {
-      return this.subscriptions.search(request);
+      return this.subscriptions.search(normalizedRequest);
     }
     if (request.model === "message") {
-      return this.messages.search(request);
+      return this.messages.search(normalizedRequest);
     }
     if (request.model === "agent_run") {
-      return this.runs.search(request);
+      return this.runs.search(normalizedRequest);
     }
     if (request.model === "agent_cursor") {
-      return this.cursors.search(request);
+      return this.cursors.search(normalizedRequest);
     }
     if (request.model === "content_lease") {
-      return this.leases.search(request);
+      return this.leases.search(normalizedRequest);
+    }
+    if (request.model === "work_day") {
+      return this.operational.searchWorkDays(request.limit);
+    }
+    if (request.model === "task") {
+      return this.operational.searchTasks({
+        workDayId: request.filters?.find((entry) => entry.field === "workDayId" || entry.field === "work_day_id")?.value,
+        agentId: request.filters?.find((entry) => entry.field === "agentId" || entry.field === "agent_id")?.value,
+        state: request.filters?.find((entry) => entry.field === "state")?.value,
+        limit: request.limit
+      });
+    }
+    if (request.model === "report") {
+      return [];
     }
     throw new Error(`Unsupported D1 search model "${request.model}".`);
   }
@@ -442,7 +849,7 @@ class CloudflareD1AgentDatabase {
         workerId: request.workerId,
         messageTypes: request.filters?.filter((filter) => filter.field === "type" && filter.op === "in").flatMap((filter) => Array.isArray(filter.value) ? filter.value.map(String) : []),
         leaseSeconds: request.leaseSeconds
-      });
+      }, request.strategy);
       return {
         item: claimed,
         leaseToken: claimed ? nextLeaseToken() : null
@@ -452,7 +859,7 @@ class CloudflareD1AgentDatabase {
       const items = await this.leases.search({
         model: "content_lease",
         filters: request.filters,
-        sort: [{ field: "lease_expires_at", direction: "desc" }],
+        sort: pickSortForRequest(request, "lease_expires_at"),
         limit: 1
       });
       return {
@@ -460,58 +867,171 @@ class CloudflareD1AgentDatabase {
         leaseToken: items[0]?.token ?? null
       };
     }
+    if (request.model === "task") {
+      const items = await this.operational.searchTasks({
+        state: "pending",
+        limit: 1
+      });
+      return { item: items[0] ?? null, leaseToken: null };
+    }
     return {
-      item: null,
+      item: (await this.search({
+        model: request.model,
+        filters: request.filters,
+        sort: pickSortForRequest(request, filterSinceField(request.model)),
+        limit: 1
+      }))[0] ?? null,
       leaseToken: null
     };
   }
   async create(request) {
+    const definition = resolveModelDefinition(request.model);
+    const normalizedRequest = {
+      ...request,
+      data: normalizeMutationData(definition, request.data)
+    };
     if (request.model === "message") {
       return await this.createMessage({
-        type: String(request.data.type ?? "message.created"),
-        payload: request.data.payload ?? request.data,
-        relatedModel: typeof request.data.relatedModel === "string" ? request.data.relatedModel : null,
-        relatedId: typeof request.data.relatedId === "string" ? request.data.relatedId : null,
-        priority: Number(request.data.priority ?? 0),
-        maxAttempts: Number(request.data.maxAttempts ?? 3),
+        type: String(normalizedRequest.data.type ?? "message.created"),
+        payload: normalizedRequest.data.payload ?? normalizedRequest.data,
+        relatedModel: typeof normalizedRequest.data.related_model === "string" ? normalizedRequest.data.related_model : null,
+        relatedId: typeof normalizedRequest.data.related_id === "string" ? normalizedRequest.data.related_id : null,
+        priority: Number(normalizedRequest.data.priority ?? 0),
+        maxAttempts: Number(normalizedRequest.data.maxAttempts ?? 3),
         actor: request.actor
       });
     }
     if (request.model === "subscription") {
-      return await this.subscriptions.create(request);
+      return await this.subscriptions.create(normalizedRequest);
     }
     if (request.model === "agent_run") {
-      return await this.runs.record({ run: request.data });
+      return await this.runs.record({ run: normalizedRequest.data });
     }
     if (request.model === "agent_cursor") {
       return await this.cursors.update({
-        ...request,
+        ...normalizedRequest,
         model: "agent_cursor"
       });
     }
     if (request.model === "content_lease") {
       return await this.leases.update({
-        ...request,
+        ...normalizedRequest,
         model: "content_lease"
       });
     }
+    if (request.model === "work_day") {
+      return await this.startWorkDay({
+        id: typeof normalizedRequest.data.id === "string" ? normalizedRequest.data.id : void 0,
+        projectId: String(normalizedRequest.data.projectId ?? normalizedRequest.data.project_id ?? ""),
+        capacityBudget: Number(normalizedRequest.data.capacityBudget ?? normalizedRequest.data.capacity_budget ?? 0),
+        graphVersion: typeof (normalizedRequest.data.graphVersion ?? normalizedRequest.data.graph_version) === "string" ? String(normalizedRequest.data.graphVersion ?? normalizedRequest.data.graph_version) : null,
+        summary: normalizedRequest.data.summary,
+        actor: request.actor
+      });
+    }
+    if (request.model === "task") {
+      return await this.createTask({
+        id: typeof normalizedRequest.data.id === "string" ? normalizedRequest.data.id : void 0,
+        workDayId: String(normalizedRequest.data.workDayId ?? normalizedRequest.data.work_day_id ?? ""),
+        agentId: String(normalizedRequest.data.agentId ?? normalizedRequest.data.agent_id ?? ""),
+        type: String(normalizedRequest.data.type ?? ""),
+        state: typeof normalizedRequest.data.state === "string" ? normalizedRequest.data.state : "pending",
+        priority: Number(normalizedRequest.data.priority ?? 0),
+        idempotencyKey: String(normalizedRequest.data.idempotencyKey ?? normalizedRequest.data.idempotency_key ?? ""),
+        payload: normalizedRequest.data.payload ?? normalizedRequest.data,
+        payloadHash: typeof (normalizedRequest.data.payloadHash ?? normalizedRequest.data.payload_hash) === "string" ? String(normalizedRequest.data.payloadHash ?? normalizedRequest.data.payload_hash) : null,
+        maxAttempts: Number(normalizedRequest.data.maxAttempts ?? normalizedRequest.data.max_attempts ?? 3),
+        availableAt: typeof (normalizedRequest.data.availableAt ?? normalizedRequest.data.available_at) === "string" ? String(normalizedRequest.data.availableAt ?? normalizedRequest.data.available_at) : void 0,
+        graphVersion: typeof (normalizedRequest.data.graphVersion ?? normalizedRequest.data.graph_version) === "string" ? String(normalizedRequest.data.graphVersion ?? normalizedRequest.data.graph_version) : null,
+        parentTaskId: typeof (normalizedRequest.data.parentTaskId ?? normalizedRequest.data.parent_task_id) === "string" ? String(normalizedRequest.data.parentTaskId ?? normalizedRequest.data.parent_task_id) : null,
+        actor: request.actor
+      });
+    }
+    if (request.model === "graph_run") {
+      return await this.operational.createGraphRun({
+        id: String(normalizedRequest.data.id ?? crypto.randomUUID()),
+        workDayId: String(normalizedRequest.data.workDayId ?? normalizedRequest.data.work_day_id ?? ""),
+        corpusHash: String(normalizedRequest.data.corpusHash ?? normalizedRequest.data.corpus_hash ?? ""),
+        graphVersion: String(normalizedRequest.data.graphVersion ?? normalizedRequest.data.graph_version ?? ""),
+        queryJson: typeof (normalizedRequest.data.queryJson ?? normalizedRequest.data.query_json) === "string" ? String(normalizedRequest.data.queryJson ?? normalizedRequest.data.query_json) : null,
+        seedIdsJson: typeof (normalizedRequest.data.seedIdsJson ?? normalizedRequest.data.seed_ids_json) === "string" ? String(normalizedRequest.data.seedIdsJson ?? normalizedRequest.data.seed_ids_json) : null,
+        selectedNodeIdsJson: typeof (normalizedRequest.data.selectedNodeIdsJson ?? normalizedRequest.data.selected_node_ids_json) === "string" ? String(normalizedRequest.data.selectedNodeIdsJson ?? normalizedRequest.data.selected_node_ids_json) : null,
+        statsJson: typeof (normalizedRequest.data.statsJson ?? normalizedRequest.data.stats_json) === "string" ? String(normalizedRequest.data.statsJson ?? normalizedRequest.data.stats_json) : null,
+        snapshotRef: typeof (normalizedRequest.data.snapshotRef ?? normalizedRequest.data.snapshot_ref) === "string" ? String(normalizedRequest.data.snapshotRef ?? normalizedRequest.data.snapshot_ref) : null,
+        createdAt: typeof (normalizedRequest.data.createdAt ?? normalizedRequest.data.created_at) === "string" ? String(normalizedRequest.data.createdAt ?? normalizedRequest.data.created_at) : void 0
+      });
+    }
+    if (request.model === "report") {
+      return await this.createReport({
+        id: typeof normalizedRequest.data.id === "string" ? normalizedRequest.data.id : void 0,
+        workDayId: String(normalizedRequest.data.workDayId ?? normalizedRequest.data.work_day_id ?? ""),
+        kind: String(normalizedRequest.data.kind ?? "workday_summary"),
+        body: normalizedRequest.data.body ?? normalizedRequest.data,
+        renderedRef: typeof (normalizedRequest.data.renderedRef ?? normalizedRequest.data.rendered_ref) === "string" ? String(normalizedRequest.data.renderedRef ?? normalizedRequest.data.rendered_ref) : null,
+        sentAt: typeof (normalizedRequest.data.sentAt ?? normalizedRequest.data.sent_at) === "string" ? String(normalizedRequest.data.sentAt ?? normalizedRequest.data.sent_at) : null,
+        actor: request.actor
+      });
+    }
     throw new Error(`Unsupported D1 create model "${request.model}".`);
   }
   async update(request) {
+    const definition = resolveModelDefinition(request.model);
+    const normalizedRequest = {
+      ...request,
+      data: normalizeMutationData(definition, request.data)
+    };
     if (request.model === "message") {
-      return this.messages.update(request);
+      return this.messages.update(normalizedRequest);
     }
     if (request.model === "subscription") {
-      return this.subscriptions.update(request);
+      return this.subscriptions.update(normalizedRequest);
     }
     if (request.model === "agent_run") {
-      return this.runs.update(request);
+      return this.runs.update(normalizedRequest);
     }
     if (request.model === "agent_cursor") {
-      return this.cursors.update(request);
+      return this.cursors.update(normalizedRequest);
     }
     if (request.model === "content_lease") {
-      return this.leases.update(request);
+      return this.leases.update(normalizedRequest);
+    }
+    if (request.model === "work_day") {
+      return this.closeWorkDay({
+        id: String(request.id ?? request.key ?? normalizedRequest.data.id ?? ""),
+        state: normalizedRequest.data.state ?? "completed",
+        summary: normalizedRequest.data.summary,
+        actor: request.actor
+      });
+    }
+    if (request.model === "task") {
+      const taskId = String(request.id ?? request.key ?? normalizedRequest.data.id ?? "");
+      if (normalizedRequest.data.state === "completed") {
+        return this.completeTask({
+          id: taskId,
+          output: normalizedRequest.data.output,
+          outputRef: typeof normalizedRequest.data.outputRef === "string" ? normalizedRequest.data.outputRef : null,
+          summary: normalizedRequest.data.summary,
+          actor: request.actor
+        });
+      }
+      if (normalizedRequest.data.state === "failed") {
+        return this.failTask({
+          id: taskId,
+          errorCode: typeof normalizedRequest.data.errorCode === "string" ? normalizedRequest.data.errorCode : null,
+          errorMessage: String(normalizedRequest.data.errorMessage ?? "Task failed"),
+          retryable: Boolean(normalizedRequest.data.retryable),
+          nextVisibleAt: typeof normalizedRequest.data.nextVisibleAt === "string" ? normalizedRequest.data.nextVisibleAt : null,
+          actor: request.actor
+        });
+      }
+      return this.recordTaskProgress({
+        id: taskId,
+        workerId: typeof normalizedRequest.data.workerId === "string" ? normalizedRequest.data.workerId : null,
+        state: typeof normalizedRequest.data.state === "string" ? normalizedRequest.data.state : void 0,
+        appendEvent: normalizedRequest.data.appendEvent,
+        patch: normalizedRequest.data.patch,
+        actor: request.actor
+      });
     }
     throw new Error(`Unsupported D1 update model "${request.model}".`);
   }
@@ -542,6 +1062,46 @@ class CloudflareD1AgentDatabase {
   releaseAllLeases() {
     return this.leases.releaseAll();
   }
+  startWorkDay(request) {
+    return this.operational.startWorkDay(request);
+  }
+  closeWorkDay(request) {
+    return this.operational.closeWorkDay(request);
+  }
+  createTask(request) {
+    return this.operational.createTask(request);
+  }
+  claimTask(request) {
+    return this.operational.claimTask(request);
+  }
+  recordTaskProgress(request) {
+    return this.operational.recordTaskProgress(request);
+  }
+  completeTask(request) {
+    return this.operational.completeTask(request);
+  }
+  failTask(request) {
+    return this.operational.failTask(request);
+  }
+  appendTaskEvent(request) {
+    return this.operational.appendTaskEvent(request);
+  }
+  searchTasks(request) {
+    return this.operational.searchTasks(request);
+  }
+  createReport(request) {
+    return this.operational.createReport(request);
+  }
+  async getManagerContext(taskId) {
+    const task = await this.operational.getTask(taskId);
+    const workDay = task ? await this.operational.getWorkDay(task.workDayId) : null;
+    return {
+      task,
+      workDay,
+      agent: null,
+      graph: workDay?.graphVersion ? { graphVersion: workDay.graphVersion } : null
+    };
+  }
 }
 export {
   CloudflareD1AgentDatabase,