whale-code 6.5.5 → 6.5.6

package/dist/server/lib/session-checkpoint.js

@@ -6,11 +6,18 @@
  * (fire-and-forget to avoid blocking the agent loop) and can be loaded to
  * resume an interrupted conversation.
  */
+
+// ============================================================================
+// TYPES
+// ============================================================================
+
 /** How often (in turns) to persist a checkpoint. */
 const CHECKPOINT_INTERVAL = 5;
+
 // ============================================================================
 // SAVE CHECKPOINT
 // ============================================================================
+
 /**
  * Upsert a checkpoint row for the given conversation.
  *
@@ -21,83 +28,86 @@ const CHECKPOINT_INTERVAL = 5;
  * agent loop is never blocked.
  */
 export async function saveCheckpoint(supabase, conversationId, turn, messages, tokensSoFar, costSoFar, toolsUsed) {
-    // Only checkpoint at the configured interval
-    if (turn % CHECKPOINT_INTERVAL !== 0)
-        return;
-    try {
-        const checkpoint = {
-            conversation_id: conversationId,
-            turn,
-            messages: JSON.stringify(messages),
-            tokens_used: { ...tokensSoFar },
-            cost_so_far: costSoFar,
-            tools_used: [...new Set(toolsUsed)],
-            created_at: new Date().toISOString(),
-        };
-        const { error } = await supabase
-            .from("ai_conversation_checkpoints")
-            .upsert({
-            conversation_id: checkpoint.conversation_id,
-            turn: checkpoint.turn,
-            messages: checkpoint.messages,
-            tokens_used: checkpoint.tokens_used,
-            cost_so_far: checkpoint.cost_so_far,
-            tools_used: checkpoint.tools_used,
-            created_at: checkpoint.created_at,
-        }, { onConflict: "conversation_id" });
-        if (error) {
-            // Table may not exist yet — log once and move on
-            console.warn(`[checkpoint] Save failed for ${conversationId} turn ${turn}: ${error.message}`);
-        }
-        else {
-            console.log(`[checkpoint] Saved conversation ${conversationId} at turn ${turn}`);
-        }
-    }
-    catch (err) {
-        // Graceful degradation — never crash the agent loop
-        console.warn(`[checkpoint] Unexpected error saving ${conversationId}: ${err.message}`);
+  // Only checkpoint at the configured interval
+  if (turn % CHECKPOINT_INTERVAL !== 0) return;
+  try {
+    const checkpoint = {
+      conversation_id: conversationId,
+      turn,
+      messages: JSON.stringify(messages),
+      tokens_used: {
+        ...tokensSoFar
+      },
+      cost_so_far: costSoFar,
+      tools_used: [...new Set(toolsUsed)],
+      created_at: new Date().toISOString()
+    };
+    const {
+      error
+    } = await supabase.from("ai_conversation_checkpoints").upsert({
+      conversation_id: checkpoint.conversation_id,
+      turn: checkpoint.turn,
+      messages: checkpoint.messages,
+      tokens_used: checkpoint.tokens_used,
+      cost_so_far: checkpoint.cost_so_far,
+      tools_used: checkpoint.tools_used,
+      created_at: checkpoint.created_at
+    }, {
+      onConflict: "conversation_id"
+    });
+    if (error) {
+      // Table may not exist yet — log once and move on
+      console.warn(`[checkpoint] Save failed for ${conversationId} turn ${turn}: ${error.message}`);
+    } else {
+      console.log(`[checkpoint] Saved conversation ${conversationId} at turn ${turn}`);
     }
+  } catch (err) {
+    // Graceful degradation — never crash the agent loop
+    console.warn(`[checkpoint] Unexpected error saving ${conversationId}: ${err.message}`);
+  }
 }
+
 // ============================================================================
 // LOAD CHECKPOINT
 // ============================================================================
+
 /**
  * Load the most recent checkpoint for a conversation.
  * Returns null if no checkpoint exists or the table hasn't been created yet.
  */
 export async function loadCheckpoint(supabase, conversationId) {
-    try {
-        const { data, error } = await supabase
-            .from("ai_conversation_checkpoints")
-            .select("conversation_id, turn, messages, tokens_used, cost_so_far, tools_used, created_at")
-            .eq("conversation_id", conversationId)
-            .order("turn", { ascending: false })
-            .limit(1)
-            .single();
-        if (error || !data) {
-            if (error && !error.message.includes("0 rows")) {
-                console.warn(`[checkpoint] Load failed for ${conversationId}: ${error.message}`);
-            }
-            return null;
-        }
-        return {
-            conversation_id: data.conversation_id,
-            turn: data.turn,
-            messages: data.messages,
-            tokens_used: data.tokens_used,
-            cost_so_far: data.cost_so_far,
-            tools_used: data.tools_used ?? [],
-            created_at: data.created_at,
-        };
-    }
-    catch (err) {
-        console.warn(`[checkpoint] Unexpected error loading ${conversationId}: ${err.message}`);
-        return null;
+  try {
+    const {
+      data,
+      error
+    } = await supabase.from("ai_conversation_checkpoints").select("conversation_id, turn, messages, tokens_used, cost_so_far, tools_used, created_at").eq("conversation_id", conversationId).order("turn", {
+      ascending: false
+    }).limit(1).single();
+    if (error || !data) {
+      if (error && !error.message.includes("0 rows")) {
+        console.warn(`[checkpoint] Load failed for ${conversationId}: ${error.message}`);
+      }
+      return null;
     }
+    return {
+      conversation_id: data.conversation_id,
+      turn: data.turn,
+      messages: data.messages,
+      tokens_used: data.tokens_used,
+      cost_so_far: data.cost_so_far,
+      tools_used: data.tools_used ?? [],
+      created_at: data.created_at
+    };
+  } catch (err) {
+    console.warn(`[checkpoint] Unexpected error loading ${conversationId}: ${err.message}`);
+    return null;
+  }
 }
+
 // ============================================================================
 // MARK ORPHANED CONVERSATIONS
 // ============================================================================
+
 /**
  * Find conversations that were in_progress before the given server start time
  * and mark them as resumable.  Returns the count of conversations marked.
@@ -106,40 +116,39 @@ export async function loadCheckpoint(supabase, conversationId) {
  * that were abandoned when the previous server instance went down.
  */
 export async function markOrphaned(supabase, serverStartTime) {
-    try {
-        // Find checkpoints whose created_at is before this server boot AND whose
-        // conversation is still in an "active" / "in_progress" state (no final
-        // assistant message after the checkpoint timestamp).
-        const { data: orphans, error: selectError } = await supabase
-            .from("ai_conversation_checkpoints")
-            .select("conversation_id, turn, created_at")
-            .lt("created_at", serverStartTime.toISOString());
-        if (selectError) {
-            console.warn(`[checkpoint] markOrphaned select failed: ${selectError.message}`);
-            return 0;
-        }
-        if (!orphans || orphans.length === 0)
-            return 0;
-        const orphanIds = orphans.map((o) => o.conversation_id);
-        // Tag conversations as resumable in their metadata
-        const { error: updateError } = await supabase
-            .from("ai_conversations")
-            .update({
-            metadata: {
-                resumable: true,
-                orphaned_at: serverStartTime.toISOString(),
-            },
-        })
-            .in("id", orphanIds);
-        if (updateError) {
-            console.warn(`[checkpoint] markOrphaned update failed: ${updateError.message}`);
-            return 0;
-        }
-        console.log(`[checkpoint] Marked ${orphanIds.length} orphaned conversation(s) as resumable`);
-        return orphanIds.length;
+  try {
+    // Find checkpoints whose created_at is before this server boot AND whose
+    // conversation is still in an "active" / "in_progress" state (no final
+    // assistant message after the checkpoint timestamp).
+    const {
+      data: orphans,
+      error: selectError
+    } = await supabase.from("ai_conversation_checkpoints").select("conversation_id, turn, created_at").lt("created_at", serverStartTime.toISOString());
+    if (selectError) {
+      console.warn(`[checkpoint] markOrphaned select failed: ${selectError.message}`);
+      return 0;
     }
-    catch (err) {
-        console.warn(`[checkpoint] markOrphaned unexpected error: ${err.message}`);
-        return 0;
+    if (!orphans || orphans.length === 0) return 0;
+    const orphanIds = orphans.map(o => o.conversation_id);
+
+    // Tag conversations as resumable in their metadata
+    const {
+      error: updateError
+    } = await supabase.from("ai_conversations").update({
+      metadata: {
+        resumable: true,
+        orphaned_at: serverStartTime.toISOString()
+      }
+    }).in("id", orphanIds);
+    if (updateError) {
+      console.warn(`[checkpoint] markOrphaned update failed: ${updateError.message}`);
+      return 0;
     }
+    console.log(`[checkpoint] Marked ${orphanIds.length} orphaned conversation(s) as resumable`);
+    return orphanIds.length;
+  } catch (err) {
+    console.warn(`[checkpoint] markOrphaned unexpected error: ${err.message}`);
+    return 0;
+  }
 }
+//# sourceMappingURL=session-checkpoint.js.map

package/dist/server/lib/session-checkpoint.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-checkpoint.js","names":["CHECKPOINT_INTERVAL","saveCheckpoint","supabase","conversationId","turn","messages","tokensSoFar","costSoFar","toolsUsed","checkpoint","conversation_id","JSON","stringify","tokens_used","cost_so_far","tools_used","Set","created_at","Date","toISOString","error","from","upsert","onConflict","console","warn","message","log","err","loadCheckpoint","data","select","eq","order","ascending","limit","single","includes","markOrphaned","serverStartTime","orphans","selectError","lt","length","orphanIds","map","o","updateError","update","metadata","resumable","orphaned_at","in"],"sources":["../../../src/server/lib/session-checkpoint.ts"],"sourcesContent":["/**\n * Session Checkpoint Manager — Phase 4.2\n *\n * Periodically saves conversation state to Supabase so that server restarts\n * don't lose in-progress conversations.  Checkpoints are written every 5 turns\n * (fire-and-forget to avoid blocking the agent loop) and can be loaded to\n * resume an interrupted conversation.\n */\n\nimport type { SupabaseClient } from \"@supabase/supabase-js\";\n\n// ============================================================================\n// TYPES\n// ============================================================================\n\nexport interface Checkpoint {\n  conversation_id: string;\n  turn: number;\n  messages: string; // JSON-serialized message array\n  tokens_used: {\n    input: number;\n    output: number;\n    cacheRead: number;\n    cacheCreation: number;\n  };\n  cost_so_far: number;\n  tools_used: string[];\n  created_at: string;\n}\n\n/** How often (in turns) to persist a checkpoint. */\nconst CHECKPOINT_INTERVAL = 5;\n\n// ============================================================================\n// SAVE CHECKPOINT\n// ============================================================================\n\n/**\n * Upsert a checkpoint row for the given conversation.\n *\n * Only actually writes when `turn % CHECKPOINT_INTERVAL === 0` so callers can\n * invoke this every turn without worrying about write frequency.\n *\n * The write is fire-and-forget — errors are logged but never thrown so the\n * agent loop is never blocked.\n */\nexport async function saveCheckpoint(\n  supabase: SupabaseClient,\n  conversationId: string,\n  turn: number,\n  messages: unknown[],\n  tokensSoFar: { input: number; output: number; cacheRead: number; cacheCreation: number },\n  costSoFar: number,\n  toolsUsed: string[],\n): Promise<void> {\n  // Only checkpoint at the configured interval\n  if (turn % CHECKPOINT_INTERVAL !== 0) return;\n\n  try {\n    const checkpoint: Checkpoint = {\n      conversation_id: conversationId,\n      turn,\n      messages: JSON.stringify(messages),\n      tokens_used: { ...tokensSoFar },\n      cost_so_far: costSoFar,\n      tools_used: [...new Set(toolsUsed)],\n      created_at: new Date().toISOString(),\n    };\n\n    const { error } = await supabase\n      .from(\"ai_conversation_checkpoints\")\n      .upsert(\n        {\n          conversation_id: checkpoint.conversation_id,\n          turn: checkpoint.turn,\n          messages: checkpoint.messages,\n          tokens_used: checkpoint.tokens_used,\n          cost_so_far: checkpoint.cost_so_far,\n          tools_used: checkpoint.tools_used,\n          created_at: checkpoint.created_at,\n        },\n        { onConflict: \"conversation_id\" },\n      );\n\n    if (error) {\n      // Table may not exist yet — log once and move on\n      console.warn(`[checkpoint] Save failed for ${conversationId} turn ${turn}: ${error.message}`);\n    } else {\n      console.log(`[checkpoint] Saved conversation ${conversationId} at turn ${turn}`);\n    }\n  } catch (err) {\n    // Graceful degradation — never crash the agent loop\n    console.warn(`[checkpoint] Unexpected error saving ${conversationId}: ${(err as Error).message}`);\n  }\n}\n\n// ============================================================================\n// LOAD CHECKPOINT\n// ============================================================================\n\n/**\n * Load the most recent checkpoint for a conversation.\n * Returns null if no checkpoint exists or the table hasn't been created yet.\n */\nexport async function loadCheckpoint(\n  supabase: SupabaseClient,\n  conversationId: string,\n): Promise<Checkpoint | null> {\n  try {\n    const { data, error } = await supabase\n      .from(\"ai_conversation_checkpoints\")\n      .select(\"conversation_id, turn, messages, tokens_used, cost_so_far, tools_used, created_at\")\n      .eq(\"conversation_id\", conversationId)\n      .order(\"turn\", { ascending: false })\n      .limit(1)\n      .single();\n\n    if (error || !data) {\n      if (error && !error.message.includes(\"0 rows\")) {\n        console.warn(`[checkpoint] Load failed for ${conversationId}: ${error.message}`);\n      }\n      return null;\n    }\n\n    return {\n      conversation_id: data.conversation_id,\n      turn: data.turn,\n      messages: data.messages,\n      tokens_used: data.tokens_used,\n      cost_so_far: data.cost_so_far,\n      tools_used: data.tools_used ?? [],\n      created_at: data.created_at,\n    } as Checkpoint;\n  } catch (err) {\n    console.warn(`[checkpoint] Unexpected error loading ${conversationId}: ${(err as Error).message}`);\n    return null;\n  }\n}\n\n// ============================================================================\n// MARK ORPHANED CONVERSATIONS\n// ============================================================================\n\n/**\n * Find conversations that were in_progress before the given server start time\n * and mark them as resumable.  Returns the count of conversations marked.\n *\n * This should be called once during server startup to flag any conversations\n * that were abandoned when the previous server instance went down.\n */\nexport async function markOrphaned(\n  supabase: SupabaseClient,\n  serverStartTime: Date,\n): Promise<number> {\n  try {\n    // Find checkpoints whose created_at is before this server boot AND whose\n    // conversation is still in an \"active\" / \"in_progress\" state (no final\n    // assistant message after the checkpoint timestamp).\n    const { data: orphans, error: selectError } = await supabase\n      .from(\"ai_conversation_checkpoints\")\n      .select(\"conversation_id, turn, created_at\")\n      .lt(\"created_at\", serverStartTime.toISOString());\n\n    if (selectError) {\n      console.warn(`[checkpoint] markOrphaned select failed: ${selectError.message}`);\n      return 0;\n    }\n\n    if (!orphans || orphans.length === 0) return 0;\n\n    const orphanIds = orphans.map((o: { conversation_id: string }) => o.conversation_id);\n\n    // Tag conversations as resumable in their metadata\n    const { error: updateError } = await supabase\n      .from(\"ai_conversations\")\n      .update({\n        metadata: {\n          resumable: true,\n          orphaned_at: serverStartTime.toISOString(),\n        },\n      })\n      .in(\"id\", orphanIds);\n\n    if (updateError) {\n      console.warn(`[checkpoint] markOrphaned update failed: ${updateError.message}`);\n      return 0;\n    }\n\n    console.log(`[checkpoint] Marked ${orphanIds.length} orphaned conversation(s) as resumable`);\n    return orphanIds.length;\n  } catch (err) {\n    console.warn(`[checkpoint] markOrphaned unexpected error: ${(err as Error).message}`);\n    return 0;\n  }\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAIA;AACA;AACA;;AAiBA;AACA,MAAMA,mBAAmB,GAAG,CAAC;;AAE7B;AACA;AACA;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,eAAeC,cAAcA,CAClCC,QAAwB,EACxBC,cAAsB,EACtBC,IAAY,EACZC,QAAmB,EACnBC,WAAwF,EACxFC,SAAiB,EACjBC,SAAmB,EACJ;EACf;EACA,IAAIJ,IAAI,GAAGJ,mBAAmB,KAAK,CAAC,EAAE;EAEtC,IAAI;IACF,MAAMS,UAAsB,GAAG;MAC7BC,eAAe,EAAEP,cAAc;MAC/BC,IAAI;MACJC,QAAQ,EAAEM,IAAI,CAACC,SAAS,CAACP,QAAQ,CAAC;MAClCQ,WAAW,EAAE;QAAE,GAAGP;MAAY,CAAC;MAC/BQ,WAAW,EAAEP,SAAS;MACtBQ,UAAU,EAAE,CAAC,GAAG,IAAIC,GAAG,CAACR,SAAS,CAAC,CAAC;MACnCS,UAAU,EAAE,IAAIC,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC;IACrC,CAAC;IAED,MAAM;MAAEC;IAAM,CAAC,GAAG,MAAMlB,QAAQ,CAC7BmB,IAAI,CAAC,6BAA6B,CAAC,CACnCC,MAAM,CACL;MACEZ,eAAe,EAAED,UAAU,CAACC,eAAe;MAC3CN,IAAI,EAAEK,UAAU,CAACL,IAAI;MACrBC,QAAQ,EAAEI,UAAU,CAACJ,QAAQ;MAC7BQ,WAAW,EAAEJ,UAAU,CAACI,WAAW;MACnCC,WAAW,EAAEL,UAAU,CAACK,WAAW;MACnCC,UAAU,EAAEN,UAAU,CAACM,UAAU;MACjCE,UAAU,EAAER,UAAU,CAACQ;IACzB,CAAC,EACD;MAAEM,UAAU,EAAE;IAAkB,CAClC,CAAC;IAEH,IAAIH,KAAK,EAAE;MACT;MACAI,OAAO,CAACC,IAAI,CAAC,gCAAgCtB,cAAc,SAASC,IAAI,KAAKgB,KAAK,CAACM,OAAO,EAAE,CAAC;IAC/F,CAAC,MAAM;MACLF,OAAO,CAACG,GAAG,CAAC,mCAAmCxB,cAAc,YAAYC,IAAI,EAAE,CAAC;IAClF;EACF,CAAC,CAAC,OAAOwB,GAAG,EAAE;IACZ;IACAJ,OAAO,CAACC,IAAI,CAAC,wCAAwCtB,cAAc,KAAMyB,GAAG,CAAWF,OAAO,EAAE,CAAC;EACnG;AACF;;AAEA;AACA;AACA;;AAEA;AACA;AACA;AACA;AACA,OAAO,eAAeG,cAAcA,CAClC3B,QAAwB,EACxBC,cAAsB,EACM;EAC5B,IAAI;IACF,MAAM;MAAE2B,IAAI;MAAEV;IAAM,CAAC,GAAG,MAAMlB,QAAQ,CACnCmB,IAAI,CAAC,6BAA6B,CAAC,CACnCU,MAAM,CAAC,mFAAmF,CAAC,CAC3FC,EAAE,CAAC,iBAAiB,EAAE7B,cAAc,CAAC,CACrC8B,KAAK,CAAC,MAAM,EAAE;MAAEC,SAAS,EAAE;IAAM,CAAC,CAAC,CACnCC,KAAK,CAAC,CAAC,CAAC,CACRC,MAAM,CAAC,CAAC;IAEX,IAAIhB,KAAK,IAAI,CAACU,IAAI,EAAE;MAClB,IAAIV,KAAK,IAAI,CAACA,KAAK,CAACM,OAAO,CAACW,QAAQ,CAAC,QAAQ,CAAC,EAAE;QAC9Cb,OAAO,CAACC,IAAI,CAAC,gCAAgCtB,cAAc,KAAKiB,KAAK,CAACM,OAAO,EAAE,CAAC;MAClF;MACA,OAAO,IAAI;IACb;IAEA,OAAO;MACLhB,eAAe,EAAEoB,IAAI,CAACpB,eAAe;MACrCN,IAAI,EAAE0B,IAAI,CAAC1B,IAAI;MACfC,QAAQ,EAAEyB,IAAI,CAACzB,QAAQ;MACvBQ,WAAW,EAAEiB,IAAI,CAACjB,WAAW;MAC7BC,WAAW,EAAEgB,IAAI,CAAChB,WAAW;MAC7BC,UAAU,EAAEe,IAAI,CAACf,UAAU,IAAI,EAAE;MACjCE,UAAU,EAAEa,IAAI,CAACb;IACnB,CAAC;EACH,CAAC,CAAC,OAAOW,GAAG,EAAE;IACZJ,OAAO,CAACC,IAAI,CAAC,yCAAyCtB,cAAc,KAAMyB,GAAG,CAAWF,OAAO,EAAE,CAAC;IAClG,OAAO,IAAI;EACb;AACF;;AAEA;AACA;AACA;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,eAAeY,YAAYA,CAChCpC,QAAwB,EACxBqC,eAAqB,EACJ;EACjB,IAAI;IACF;IACA;IACA;IACA,MAAM;MAAET,IAAI,EAAEU,OAAO;MAAEpB,KAAK,EAAEqB;IAAY,CAAC,GAAG,MAAMvC,QAAQ,CACzDmB,IAAI,CAAC,6BAA6B,CAAC,CACnCU,MAAM,CAAC,mCAAmC,CAAC,CAC3CW,EAAE,CAAC,YAAY,EAAEH,eAAe,CAACpB,WAAW,CAAC,CAAC,CAAC;IAElD,IAAIsB,WAAW,EAAE;MACfjB,OAAO,CAACC,IAAI,CAAC,4CAA4CgB,WAAW,CAACf,OAAO,EAAE,CAAC;MAC/E,OAAO,CAAC;IACV;IAEA,IAAI,CAACc,OAAO,IAAIA,OAAO,CAACG,MAAM,KAAK,CAAC,EAAE,OAAO,CAAC;IAE9C,MAAMC,SAAS,GAAGJ,OAAO,CAACK,GAAG,CAAEC,CAA8B,IAAKA,CAAC,CAACpC,eAAe,CAAC;;IAEpF;IACA,MAAM;MAAEU,KAAK,EAAE2B;IAAY,CAAC,GAAG,MAAM7C,QAAQ,CAC1CmB,IAAI,CAAC,kBAAkB,CAAC,CACxB2B,MAAM,CAAC;MACNC,QAAQ,EAAE;QACRC,SAAS,EAAE,IAAI;QACfC,WAAW,EAAEZ,eAAe,CAACpB,WAAW,CAAC;MAC3C;IACF,CAAC,CAAC,CACDiC,EAAE,CAAC,IAAI,EAAER,SAAS,CAAC;IAEtB,IAAIG,WAAW,EAAE;MACfvB,OAAO,CAACC,IAAI,CAAC,4CAA4CsB,WAAW,CAACrB,OAAO,EAAE,CAAC;MAC/E,OAAO,CAAC;IACV;IAEAF,OAAO,CAACG,GAAG,CAAC,uBAAuBiB,SAAS,CAACD,MAAM,wCAAwC,CAAC;IAC5F,OAAOC,SAAS,CAACD,MAAM;EACzB,CAAC,CAAC,OAAOf,GAAG,EAAE;IACZJ,OAAO,CAACC,IAAI,CAAC,+CAAgDG,GAAG,CAAWF,OAAO,EAAE,CAAC;IACrF,OAAO,CAAC;EACV;AACF","ignoreList":[]}