@byte5ai/palaia 2.3.5 → 2.5.1

package/src/hooks/session.ts

@@ -0,0 +1,464 @@
+/**
+ * Session lifecycle hooks for palaia v3.
+ *
+ * Handles session_start, session_end, before_reset, llm_input, llm_output,
+ * and after_tool_call hooks to provide:
+ * - Session briefing on session start (context restoration)
+ * - Automatic session summaries on session end/reset
+ * - LLM model switch detection
+ * - Tool observation tracking
+ * - Token usage tracking
+ */
+
+import type { OpenClawPluginApi } from "../types.js";
+import type { PalaiaPluginConfig } from "../config.js";
+import { run, type RunnerOpts } from "../runner.js";
+import {
+  getOrCreateSessionState,
+  deleteSessionState,
+  type PendingBriefing,
+  type ToolObservation,
+} from "./state.js";
+import {
+  stripPalaiaInjectedContext,
+  stripPrivateBlocks,
+  trimToRecentExchanges,
+  extractWithLLM,
+} from "./capture.js";
+import { extractMessageTexts } from "./recall.js";
+import { loadPriorities, resolvePriorities } from "../priorities.js";
+
+/**
+ * Resolve effective capture scope from priorities (per-agent override)
+ * falling back to plugin config, then to "team".
+ */
+async function getEffectiveCaptureScope(config: PalaiaPluginConfig): Promise<string> {
+  try {
+    const prio = await loadPriorities(config.workspace || "");
+    const agentId = process.env.PALAIA_AGENT || undefined;
+    const resolved = resolvePriorities(prio, {
+      recallTypeWeight: config.recallTypeWeight,
+      recallMinScore: config.recallMinScore,
+      maxInjectedChars: config.maxInjectedChars,
+      tier: config.tier,
+    }, agentId);
+    if (resolved.captureScope) return resolved.captureScope;
+  } catch {
+    // Fall through to config default
+  }
+  return config.captureScope || "team";
+}
+
+function buildRunnerOpts(config: PalaiaPluginConfig): RunnerOpts {
+  return {
+    binaryPath: config.binaryPath,
+    workspace: config.workspace,
+    timeoutMs: config.timeoutMs,
+  };
+}
+
+// ── Session Briefing ────────────────────────────────────────���───────────
+
+/**
+ * Load session briefing data (last session summary + open tasks).
+ * Called during session_start to prepare context for the first turn.
+ */
+export async function loadSessionBriefing(
+  config: PalaiaPluginConfig,
+  logger: { info(...a: unknown[]): void; warn(...a: unknown[]): void },
+): Promise<PendingBriefing> {
+  const opts = buildRunnerOpts(config);
+  let summary: string | null = null;
+  let summaryCreated: number = Date.now();
+  const openTasks: string[] = [];
+
+  // Load last session summary and open tasks in parallel
+  const { runJson } = await import("../runner.js");
+  const [summaryResult, tasksResult] = await Promise.allSettled([
+    runJson<{ results: Array<{ title: string; body: string; created?: string }> }>(
+      ["query", "session-summary", "--limit", "1", "--tags", "session-summary"],
+      { ...opts, timeoutMs: 5000 },
+    ),
+    runJson<{ results: Array<{ title: string; body: string; priority?: string }> }>(
+      ["list", "--type", "task", "--status", "open", "--limit", "5"],
+      { ...opts, timeoutMs: 5000 },
+    ),
+  ]);
+
+  if (summaryResult.status === "fulfilled" && summaryResult.value?.results?.[0]) {
+    summary = summaryResult.value.results[0].body;
+    const created = summaryResult.value.results[0].created;
+    if (created) {
+      const parsed = new Date(created).getTime();
+      if (!isNaN(parsed)) summaryCreated = parsed;
+    }
+  }
+
+  if (tasksResult.status === "fulfilled" && tasksResult.value?.results) {
+    for (const task of tasksResult.value.results) {
+      const prio = task.priority ? ` (${task.priority})` : "";
+      openTasks.push(`${task.title}${prio}`);
+    }
+  }
+
+  return { summary, openTasks, timestamp: summaryCreated };
+}
+
+/**
+ * Format a pending briefing into injectable text.
+ */
+export function formatBriefing(briefing: PendingBriefing, maxChars: number): string {
+  if (maxChars <= 0) return "";
+  if (!briefing.summary && briefing.openTasks.length === 0) return "";
+
+  const parts: string[] = ["## Session Briefing (Palaia)\n"];
+
+  if (briefing.summary) {
+    const agoMs = Date.now() - briefing.timestamp;
+    const agoMin = Math.round(agoMs / 60_000);
+    const agoStr = agoMin < 1 ? "just now" :
+      agoMin < 60 ? `${agoMin}m ago` :
+      `${Math.round(agoMin / 60)}h ago`;
+    parts.push(`Last session (${agoStr}):`);
+    parts.push(briefing.summary);
+    parts.push("");
+  }
+
+  if (briefing.openTasks.length > 0) {
+    parts.push("Open tasks:");
+    for (const task of briefing.openTasks) {
+      parts.push(`- ${task}`);
+    }
+    parts.push("");
+  }
+
+  const text = parts.join("\n");
+  return text.length <= maxChars ? text : text.slice(0, maxChars - 3) + "...";
+}
+
+// ── Session Summary Capture ─────────────────────────────────────────────
+
+/**
+ * Extract and save a session summary from conversation messages.
+ * Called during before_reset (with messages) or session_end (without).
+ */
+export async function captureSessionSummary(
+  messages: unknown[] | undefined,
+  sessionKey: string,
+  api: OpenClawPluginApi,
+  config: PalaiaPluginConfig,
+  logger: { info(...a: unknown[]): void; warn(...a: unknown[]): void },
+): Promise<void> {
+  const opts = buildRunnerOpts(config);
+  const state = getOrCreateSessionState(sessionKey);
+
+  // Guard: prevent double-save (before_reset + session_end race)
+  if (state.summarySaved) return;
+  let summaryText: string | null = null;
+
+  if (messages && messages.length > 0) {
+    // Try LLM-based summary extraction
+    try {
+      const results = await extractWithLLM(messages, api.config, {
+        captureModel: config.captureModel,
+      }, []);
+
+      if (results.length > 0) {
+        summaryText = results[0].content;
+      }
+    } catch {
+      // Fall through to rule-based
+    }
+
+    // Rule-based fallback: last 3 user+assistant pairs
+    if (!summaryText) {
+      const allTexts = extractMessageTexts(messages);
+      const cleaned = allTexts.map((t: { role: string; text: string }) => ({
+        ...t,
+        text: stripPrivateBlocks(
+          t.role === "user" ? stripPalaiaInjectedContext(t.text) : t.text
+        ),
+      }));
+      const recent = trimToRecentExchanges(cleaned, 3);
+      if (recent.length > 0) {
+        summaryText = recent
+          .map(t => `[${t.role}]: ${t.text.slice(0, 200)}`)
+          .join("\n")
+          .slice(0, 500);
+      }
+    }
+  } else {
+    // No messages available (session_end without before_reset).
+    // Use accumulated tool observations as summary.
+    if (state.toolObservations.length > 0) {
+      summaryText = state.toolObservations
+        .map(o => `${o.toolName}: ${o.resultSummary}`)
+        .join("\n")
+        .slice(0, 500);
+    }
+  }
+
+  if (!summaryText) return;
+
+  // Save session summary
+  try {
+    const scope = await getEffectiveCaptureScope(config);
+    const args: string[] = [
+      "write", summaryText,
+      "--type", "memory",
+      "--tags", "session-summary,auto-capture",
+      "--scope", scope,
+    ];
+    if (state.autoSessionId) args.push("--instance", state.autoSessionId);
+    const agentName = process.env.PALAIA_AGENT || undefined;
+    if (agentName) args.push("--agent", agentName);
+
+    await run(args, { ...opts, timeoutMs: 10_000 });
+    state.summarySaved = true;  // Only mark after successful write
+    logger.info(`[palaia] Session summary saved (${summaryText.length} chars)`);
+  } catch (error) {
+    // Don't set summarySaved — allow retry from session_end if before_reset failed
+    logger.warn(`[palaia] Failed to save session summary: ${error}`);
+  }
+}
+
+// ── Tool Observations ─────────────────────────────────────────────────���─
+
+/** Tools worth tracking for observations. */
+const TRACKED_TOOLS = new Set([
+  // File operations
+  "memory_search", "memory_get", "memory_write",
+  "read_file", "write_file", "edit_file",
+  "search_files", "list_files",
+  // Shell
+  "bash", "shell", "terminal",
+  // Web
+  "web_search", "fetch_url",
+]);
+
+/**
+ * Process an after_tool_call event into a tool observation.
+ * Returns null if the tool isn't worth tracking.
+ */
+export function processToolCall(
+  toolName: string,
+  params: Record<string, unknown>,
+  result: unknown,
+  durationMs: number,
+): ToolObservation | null {
+  // Only track relevant tools (skip internal/framework tools)
+  if (!TRACKED_TOOLS.has(toolName) && !toolName.startsWith("palaia_")) {
+    return null;
+  }
+
+  // Skip errored tool calls (only null/undefined, not falsy 0/"")
+  if (result === undefined || result === null) return null;
+
+  // Summarize params (keep it short)
+  const paramKeys = Object.keys(params).slice(0, 3);
+  const paramsSummary = paramKeys
+    .map(k => {
+      const v = params[k];
+      try {
+        const s = typeof v === "string" ? v.slice(0, 80) : JSON.stringify(v)?.slice(0, 80) ?? "";
+        return `${k}=${s}`;
+      } catch {
+        return `${k}=[object]`;
+      }
+    })
+    .join(", ");
+
+  // Summarize result
+  let resultSummary: string;
+  if (typeof result === "string") {
+    resultSummary = result.slice(0, 200);
+  } else if (typeof result === "object" && result !== null) {
+    resultSummary = JSON.stringify(result).slice(0, 200);
+  } else {
+    resultSummary = String(result).slice(0, 200);
+  }
+
+  return {
+    toolName,
+    paramsSummary,
+    resultSummary,
+    durationMs,
+    timestamp: Date.now(),
+  };
+}
+
+// ── Hook Registration ───────────────────────────────────────────────────
+
+/**
+ * Register all session lifecycle hooks.
+ * Called from hooks/index.ts during plugin registration.
+ */
+export function registerSessionHooks(
+  api: OpenClawPluginApi,
+  config: PalaiaPluginConfig,
+): void {
+  const logger = api.logger;
+
+  // ── session_start: Load briefing for context restoration ──────────
+  if (config.sessionBriefing) {
+    api.on("session_start", async (event: any, ctx: any) => {
+      const sessionKey = ctx?.sessionKey || ctx?.sessionId;
+      if (!sessionKey) return;
+      const state = getOrCreateSessionState(sessionKey);
+
+      // Create a promise that before_prompt_build can await
+      let resolve: () => void;
+      state.briefingReady = new Promise<void>(r => { resolve = r; });
+      state.briefingReadyResolve = resolve!;
+
+      try {
+        state.pendingBriefing = await loadSessionBriefing(config, logger);
+        if (state.pendingBriefing.summary || state.pendingBriefing.openTasks.length > 0) {
+          logger.info(`[palaia] Session briefing loaded for ${sessionKey}`);
+        }
+      } catch (error) {
+        logger.warn(`[palaia] Failed to load session briefing: ${error}`);
+      } finally {
+        state.briefingReadyResolve?.();
+      }
+    });
+  }
+
+  // ── session_end: Save session summary ─────────────────────────────
+  if (config.sessionSummary) {
+    api.on("session_end", async (event: any, ctx: any) => {
+      const sessionKey = ctx?.sessionKey || ctx?.sessionId;
+      if (!sessionKey) return;
+
+      try {
+        // Only save summary if session had meaningful interaction.
+        // Guard: if OpenClaw omits messageCount, fall back to tool observations
+        // as evidence of interaction rather than silently skipping.
+        const messageCount = (event as any)?.messageCount;
+        const state = getOrCreateSessionState(sessionKey);
+        const hasInteraction = (typeof messageCount === "number" && messageCount >= 4)
+          || (messageCount === undefined && state.toolObservations.length > 0);
+        if (hasInteraction) {
+          await captureSessionSummary(undefined, sessionKey, api, config, logger);
+        }
+      } catch (error) {
+        logger.warn(`[palaia] session_end summary failed: ${error}`);
+      } finally {
+        deleteSessionState(sessionKey);
+      }
+    });
+  }
+
+  // ── before_reset: Save session summary with messages (priority) ───
+  if (config.sessionSummary) {
+    api.on("before_reset", async (event: any, ctx: any) => {
+      const sessionKey = ctx?.sessionKey || ctx?.sessionId;
+      if (!sessionKey) return;
+
+      try {
+        const messages = (event as any)?.messages;
+        if (messages && Array.isArray(messages) && messages.length >= 4) {
+          await captureSessionSummary(messages, sessionKey, api, config, logger);
+        }
+      } catch (error) {
+        logger.warn(`[palaia] before_reset summary failed: ${error}`);
+      }
+    });
+  }
+
+  // ── llm_input: Model switch detection ─────────────────────────────
+  api.on("llm_input", (event: any, ctx: any) => {
+    const sessionKey = ctx?.sessionKey || ctx?.sessionId;
+    if (!sessionKey) return;
+    const state = getOrCreateSessionState(sessionKey);
+
+    const model = (event as any)?.model;
+    if (!model) return;
+
+    if (state.lastModel && model !== state.lastModel) {
+      state.modelSwitchDetected = true;
+      logger.info(`[palaia] Model switch detected: ${state.lastModel} → ${model}`);
+
+      // Re-load briefing for next assemble
+      if (config.sessionBriefing && !state.pendingBriefing) {
+        loadSessionBriefing(config, logger).then(briefing => {
+          state.pendingBriefing = briefing;
+          state.briefingDelivered = false;
+        }).catch(() => {});
+      }
+    }
+    state.lastModel = model;
+  });
+
+  // ── after_tool_call: Tool observation tracking ────────────────��───
+  if (config.captureToolObservations) {
+    api.on("after_tool_call", (event: any, ctx: any) => {
+      const sessionKey = ctx?.sessionKey || ctx?.sessionId;
+      if (!sessionKey) return;
+      const state = getOrCreateSessionState(sessionKey);
+
+      const obs = processToolCall(
+        (event as any)?.toolName || "",
+        (event as any)?.params || {},
+        (event as any)?.result,
+        (event as any)?.durationMs || 0,
+      );
+
+      if (obs) {
+        state.toolObservations.push(obs);
+        // Keep max 50 observations per session to prevent memory bloat
+        if (state.toolObservations.length > 50) {
+          state.toolObservations.shift();
+        }
+      }
+    });
+  }
+
+  // ── subagent_spawning: Log sub-agent spawn for context tracking ─
+  api.on("subagent_spawning", async (event: any, _ctx: any) => {
+    try {
+      const childKey = (event as any)?.childSessionKey;
+      if (!childKey) return;
+      logger.info(`[palaia] Sub-agent spawning: ${childKey}`);
+    } catch {
+      // Non-fatal
+    }
+  });
+
+  // ── subagent_ended: Capture sub-agent results ─────────────────
+  if (config.autoCapture) {
+    api.on("subagent_ended", async (event: any, _ctx: any) => {
+      try {
+        const outcome = (event as any)?.outcome;
+        if (outcome !== "ok") return;
+
+        const childKey = (event as any)?.targetSessionKey;
+        if (!childKey || !api.runtime?.subagent?.getSessionMessages) return;
+
+        const result = await api.runtime.subagent.getSessionMessages({
+          sessionKey: childKey,
+          limit: 10,
+        });
+        if (!result?.messages?.length) return;
+
+        const texts = extractMessageTexts(result.messages);
+        const summaryParts = texts.slice(-4).map(
+          (t: { role: string; text: string }) => `[${t.role}]: ${t.text.slice(0, 200)}`
+        );
+        if (summaryParts.length === 0) return;
+
+        const summaryText = `Sub-agent result:\n${summaryParts.join("\n")}`.slice(0, 500);
+        const subagentScope = await getEffectiveCaptureScope(config);
+        await run([
+          "write", summaryText,
+          "--type", "memory",
+          "--tags", "subagent-result,auto-capture",
+          "--scope", subagentScope,
+        ], { ...buildRunnerOpts(config), timeoutMs: 10_000 });
+        logger.info(`[palaia] Sub-agent result captured (${summaryText.length} chars)`);
+      } catch (error) {
+        logger.warn(`[palaia] Sub-agent result capture failed: ${error}`);
+      }
+    });
+  }
+}

package/src/hooks/state.ts

@@ -127,6 +127,8 @@ export function pruneStaleEntries(): void {
       lastInboundMessageByChannel.delete(key);
     }
   }
+  // Also prune stale session state (orphaned sessions)
+  pruneStaleSessionState();
 }
 
 /**
@@ -154,6 +156,106 @@ export function resetTurnState(): void {
   lastInboundMessageByChannel.clear();
 }
 
+// ============================================================================
+// Session Continuity State (v3.0)
+// ============================================================================
+
+/** Accumulated tool observation from after_tool_call hooks. */
+export interface ToolObservation {
+  toolName: string;
+  paramsSummary: string;
+  resultSummary: string;
+  durationMs: number;
+  timestamp: number;
+}
+
+/** Pending session briefing, cached between session_start and first before_prompt_build. */
+export interface PendingBriefing {
+  summary: string | null;
+  openTasks: string[];
+  timestamp: number;
+}
+
+/** Session-level state that persists across turns within a session. */
+export interface SessionState {
+  /** Auto-generated session ID for instance tracking. */
+  autoSessionId: string;
+  /** Last known LLM model (for switch detection). */
+  lastModel: string | null;
+  /** True when model switch detected, cleared after briefing re-injection. */
+  modelSwitchDetected: boolean;
+  /** Pending briefing to inject on next before_prompt_build. */
+  pendingBriefing: PendingBriefing | null;
+  /** Accumulated tool observations for the session (used in session summary). */
+  toolObservations: ToolObservation[];
+  /** Timestamp of session start. */
+  startedAt: number;
+  /** Whether the first turn briefing has been delivered. */
+  briefingDelivered: boolean;
+  /** Whether a session summary has already been saved (prevents double-save). */
+  summarySaved: boolean;
+  /** Promise that resolves when session_start briefing load is complete. */
+  briefingReady: Promise<void> | null;
+  /** Resolver for briefingReady promise. */
+  briefingReadyResolve: (() => void) | null;
+}
+
+/** Session state map. Keyed by sessionKey. */
+export const sessionStateByKey = new Map<string, SessionState>();
+
+/** Maximum age for session state entries before pruning (1 hour). */
+const SESSION_STATE_TTL_MS = 60 * 60 * 1000;
+
+/**
+ * Remove stale entries from sessionStateByKey.
+ * Called alongside pruneStaleEntries() to prevent memory leaks from
+ * sessions that ended without firing session_end (crash/kill).
+ */
+export function pruneStaleSessionState(): void {
+  const now = Date.now();
+  for (const [key, state] of sessionStateByKey) {
+    if (now - state.startedAt > SESSION_STATE_TTL_MS) {
+      sessionStateByKey.delete(key);
+    }
+  }
+}
+
+/** Generate an auto session ID. */
+export function generateAutoSessionId(): string {
+  const now = new Date();
+  const pad = (n: number) => String(n).padStart(2, "0");
+  const date = `${now.getFullYear()}${pad(now.getMonth() + 1)}${pad(now.getDate())}`;
+  const time = `${pad(now.getHours())}${pad(now.getMinutes())}`;
+  const hex = Math.floor(Math.random() * 0xffff).toString(16).padStart(4, "0");
+  return `auto-${date}-${time}-${hex}`;
+}
+
+/** Get or create session state for a session key. */
+export function getOrCreateSessionState(sessionKey: string): SessionState {
+  let state = sessionStateByKey.get(sessionKey);
+  if (!state) {
+    state = {
+      autoSessionId: generateAutoSessionId(),
+      lastModel: null,
+      modelSwitchDetected: false,
+      pendingBriefing: null,
+      toolObservations: [],
+      startedAt: Date.now(),
+      briefingDelivered: false,
+      summarySaved: false,
+      briefingReady: null,
+      briefingReadyResolve: null,
+    };
+    sessionStateByKey.set(sessionKey, state);
+  }
+  return state;
+}
+
+/** Delete session state (cleanup). */
+export function deleteSessionState(sessionKey: string): void {
+  sessionStateByKey.delete(sessionKey);
+}
+
 // ============================================================================
 // Session Key Helpers
 // ============================================================================

package/src/priorities.ts

@@ -32,6 +32,8 @@ export interface AgentPriorityOverride {
   recallMinScore?: number;
   maxInjectedChars?: number;
   tier?: string;
+  scopeVisibility?: string[];  // Issue #145: agent isolation
+  captureScope?: string;       // Issue #147: per-agent write scope
 }
 
 export interface ProjectPriorityOverride {
@@ -45,6 +47,8 @@ export interface ResolvedPriorities {
   recallMinScore: number;
   maxInjectedChars: number;
   tier: string;
+  scopeVisibility: string[] | null;  // Issue #145: agent isolation
+  captureScope: string | null;       // Issue #147: per-agent write scope
 }
 
 // ============================================================================
@@ -136,6 +140,8 @@ export function resolvePriorities(
     recallMinScore: defaults.recallMinScore ?? DEFAULT_MIN_SCORE,
     maxInjectedChars: defaults.maxInjectedChars ?? DEFAULT_MAX_CHARS,
     tier: defaults.tier ?? DEFAULT_TIER,
+    scopeVisibility: null,
+    captureScope: null,
   };
 
   if (!prio) return resolved;
@@ -176,6 +182,12 @@ export function resolvePriorities(
       if (agentCfg.tier !== undefined) {
         resolved.tier = agentCfg.tier;
       }
+      if (agentCfg.scopeVisibility) {
+        resolved.scopeVisibility = [...agentCfg.scopeVisibility];
+      }
+      if (agentCfg.captureScope) {
+        resolved.captureScope = agentCfg.captureScope;
+      }
     }
   }
 

package/src/tools.ts

@@ -10,6 +10,7 @@ import { Type } from "@sinclair/typebox";
 import { run, runJson, type RunnerOpts } from "./runner.js";
 import type { PalaiaPluginConfig } from "./config.js";
 import { sanitizeScope, isValidScope } from "./hooks/index.js";
+import { loadPriorities, resolvePriorities } from "./priorities.js";
 import type { OpenClawPluginApi } from "./types.js";
 
 /** Shape returned by `palaia query --json` */
@@ -81,11 +82,6 @@ export function registerTools(api: OpenClawPluginApi, config: PalaiaPluginConfig
           description: "hot|warm|all (default: hot+warm)",
         })
       ),
-      scope: Type.Optional(
-        Type.String({
-          description: "Filter by scope: private|team|shared:X|public",
-        })
-      ),
       type: Type.Optional(
         Type.String({
           description: "Filter by entry type: memory|process|task",
@@ -98,10 +94,25 @@ export function registerTools(api: OpenClawPluginApi, config: PalaiaPluginConfig
         query: string;
         maxResults?: number;
         tier?: string;
-        scope?: string;
         type?: string;
       }
     ) {
+      // Load scope visibility from priorities (Issue #145: agent isolation)
+      let scopeVisibility: string[] | null = null;
+      try {
+        const prio = await loadPriorities(config.workspace || "");
+        const agentId = process.env.PALAIA_AGENT || undefined;
+        const resolvedPrio = resolvePriorities(prio, {
+          recallTypeWeight: config.recallTypeWeight,
+          recallMinScore: config.recallMinScore,
+          maxInjectedChars: config.maxInjectedChars,
+          tier: config.tier,
+        }, agentId);
+        scopeVisibility = resolvedPrio.scopeVisibility;
+      } catch {
+        // Non-fatal: proceed without scope filtering
+      }
+
       const limit = params.maxResults || config.maxResults || 5;
       const args: string[] = ["query", params.query, "--limit", String(limit)];
 
@@ -117,8 +128,21 @@ export function registerTools(api: OpenClawPluginApi, config: PalaiaPluginConfig
 
       const result = await runJson<QueryResult>(args, opts);
 
+      // Apply scope visibility filter (Issue #145: agent isolation)
+      let filteredResults = result.results || [];
+      if (scopeVisibility) {
+        filteredResults = filteredResults.filter((r) => {
+          const scope = r.scope || "team";
+          return scopeVisibility!.some((allowed) => {
+            if (scope === allowed) return true;
+            if (allowed === "shared" && scope.startsWith("shared:")) return true;
+            return false;
+          });
+        });
+      }
+
       // Format as memory_search compatible output
-      const snippets = (result.results || []).map((r) => {
+      const snippets = filteredResults.map((r) => {
         const body = r.content || r.body || "";
         const path = r.path || `${r.tier}/${r.id}.md`;
         return {