refactor: move types next to schema

src/brain/index.test.ts

@@ -16,6 +16,7 @@ let customDailySlots: Array<{
   start: string;
   end: string;
   activity: string;
+  notes: string;
 }> | null = null;
 let customAvailability: Array<{
   start: string;
@@ -27,8 +28,14 @@ function build48Slots(): Array<{
   start: string;
   end: string;
   activity: string;
+  notes: string;
 }> {
-  const slots: Array<{ start: string; end: string; activity: string }> = [];
+  const slots: Array<{
+    start: string;
+    end: string;
+    activity: string;
+    notes: string;
+  }> = [];
   for (let i = 0; i < 48; i++) {
     const startHour = Math.floor(i / 2);
     const startMin = (i % 2) * 30;
@@ -41,7 +48,7 @@ function build48Slots(): Array<{
       const endMin = ((i + 1) % 2) * 30;
       end = `${String(endHour).padStart(2, "0")}:${String(endMin).padStart(2, "0")}`;
     }
-    slots.push({ start, end, activity: `slot-${i}` });
+    slots.push({ start, end, activity: `slot-${i}`, notes: "" });
   }
   return slots;
 }
@@ -70,7 +77,7 @@ function buildAvailability(): Array<{
 const mockCall = mock(async <T>(model: unknown, options: any): Promise<T> => {
   llmCalls.push({ model, options });
   if (options.jsonSchemaName === "daily-schedule") {
-    return (customDailySlots ?? build48Slots()) as unknown as T;
+    return { items: customDailySlots ?? build48Slots() } as unknown as T;
   }
   if (options.jsonSchemaName === "monthly-schedule") {
     if (customMonthlyDays) return customMonthlyDays as unknown as T;
@@ -150,16 +157,18 @@ describe("Brain.createDailySchedule", () => {
     const result = await brain.createDailySchedule(today, "focus on writing");
 
     expect(result).not.toBeNull();
-    expect(result).toHaveLength(48);
-    expect(result![0]).toEqual({
+    expect(result!.items).toHaveLength(48);
+    expect(result!.items[0]).toEqual({
       start: "00:00",
       end: "00:30",
       activity: "slot-0",
+      notes: "",
     });
-    expect(result![47]).toEqual({
+    expect(result!.items[47]).toEqual({
       start: "23:30",
       end: "24:00",
       activity: "slot-47",
+      notes: "",
     });
 
     const llmCall = llmCalls.find(
@@ -177,7 +186,7 @@ describe("Brain.createDailySchedule", () => {
       },
     );
     expect(facts).toHaveLength(1);
-    expect(JSON.parse(facts[0]!.statement)).toHaveLength(48);
+    expect(JSON.parse(facts[0]!.statement).items).toHaveLength(48);
   });
 
   test("S4: month wrap (June 30 -> July 1)", async () => {
@@ -306,7 +315,7 @@ describe("Brain.getTodayScheduledAvailability", () => {
     const todayKey = formatDateKey(today);
     await brain.db.addFact({
       spaceName: brain.space.name,
-      statement: JSON.stringify(build48Slots()),
+      statement: JSON.stringify({ items: build48Slots() }),
       summary: "test daily",
       source: "test",
       confidence: 1.0,
@@ -357,7 +366,7 @@ describe("Brain.removeScheduledAvailability", () => {
     const todayKey = formatDateKey(today);
     await brain.db.addFact({
       spaceName: brain.space.name,
-      statement: JSON.stringify(build48Slots()),
+      statement: JSON.stringify({ items: build48Slots() }),
       summary: "test daily",
       source: "test",
       confidence: 1.0,
@@ -426,7 +435,7 @@ describe("Brain.createDebug", () => {
 
     const schedule = await brain.createDailySchedule(today, "msg");
     expect(schedule).not.toBeNull();
-    expect(schedule).toHaveLength(48);
+    expect(schedule!.items).toHaveLength(48);
 
     const facts = await brain.db.getTopicFacts(`daily-schedule:${tomorrowKey}`, {
       spaceName: brain.space.name,

src/brain/index.ts

@@ -7,14 +7,14 @@ import {
   availabilitySchema,
   dailyScheduleSchema,
   monthlyScheduleSchema,
+  type Availability,
+  type DailySchedule,
+  type MonthlySchedule,
 } from "@/openrouter/schema";
 import { logger } from "@/utils/logger";
 import { factExtractor } from "./factExtractor";
 import { brainManager, type BrainItem } from "./manager";
 import {
-  type Availability,
-  type DailySchedule,
-  type MonthlySchedule,
   formatDateKey,
   formatMonthKey,
   nextDay,
@@ -72,7 +72,7 @@ export class Brain {
         await this.db.addFact({
           spaceName: this.space.name,
           statement: JSON.stringify(schedule),
-          summary: `Daily schedule for ${dateKey} (${schedule.length} slots)`,
+          summary: `Daily schedule for ${dateKey} (${schedule.items.length} slots)`,
           source: "createDailySchedule",
           confidence: 1.0,
           topics: [
@@ -291,11 +291,11 @@ export class Brain {
         "PERSONA_BASE_SYSTEM_PROMPT",
       );
       const generatedBaseSystemPrompt = await llm.call<string>(
-llm.models.identity,
-{
-        instruction: personaSystemInstruction,
-        message: description,
-      },
+        llm.models.identity,
+        {
+          instruction: personaSystemInstruction,
+          message: description,
+        },
       );
 
       const personaSystemFixed = await loadPrompt(

src/brain/schedule.ts

@@ -1,25 +1,3 @@
-export type DailySlot = {
-  start: string;
-  end: string;
-  activity: string;
-  notes?: string;
-};
-export type DailySchedule = DailySlot[];
-
-export type MonthlyDay = {
-  day: number;
-  summary: string;
-};
-export type MonthlySchedule = MonthlyDay[];
-
-export type AvailabilityStatus = "online" | "do-not-disturb" | "offline";
-export type Availability = {
-  start: string;
-  end: string;
-  status: AvailabilityStatus;
-};
-export type AvailabilityWindows = Availability[];
-
 export function pad2(n: number): string {
   return n < 10 ? `0${n}` : `${n}`;
 }