@askexenow/exe-os 0.8.0

package/dist/hooks/exe-heartbeat-hook.js

@@ -0,0 +1,232 @@
+// src/adapters/claude/hooks/exe-heartbeat-hook.ts
+import { execFileSync } from "child_process";
+import path3 from "path";
+import { fileURLToPath } from "url";
+
+// src/adapters/claude/active-agent.ts
+import { readFileSync as readFileSync2, writeFileSync, mkdirSync, unlinkSync, readdirSync } from "fs";
+import { execSync as execSync2 } from "child_process";
+import path2 from "path";
+
+// src/lib/config.ts
+import { readFile, writeFile, mkdir } from "fs/promises";
+import { readFileSync, existsSync, renameSync } from "fs";
+import path from "path";
+import os from "os";
+function resolveDataDir() {
+  if (process.env.EXE_OS_DIR) return process.env.EXE_OS_DIR;
+  if (process.env.EXE_MEM_DIR) return process.env.EXE_MEM_DIR;
+  const newDir = path.join(os.homedir(), ".exe-os");
+  const legacyDir = path.join(os.homedir(), ".exe-mem");
+  if (!existsSync(newDir) && existsSync(legacyDir)) {
+    try {
+      renameSync(legacyDir, newDir);
+      process.stderr.write(`[exe-os] Migrated data directory: ~/.exe-mem \u2192 ~/.exe-os
+`);
+    } catch {
+      return legacyDir;
+    }
+  }
+  return newDir;
+}
+var EXE_AI_DIR = resolveDataDir();
+var DB_PATH = path.join(EXE_AI_DIR, "memories.db");
+var MODELS_DIR = path.join(EXE_AI_DIR, "models");
+var CONFIG_PATH = path.join(EXE_AI_DIR, "config.json");
+var LEGACY_LANCE_PATH = path.join(EXE_AI_DIR, "local.lance");
+var CURRENT_CONFIG_VERSION = 1;
+var DEFAULT_CONFIG = {
+  config_version: CURRENT_CONFIG_VERSION,
+  dbPath: DB_PATH,
+  modelFile: "jina-embeddings-v5-small-q4_k_m.gguf",
+  embeddingDim: 1024,
+  batchSize: 20,
+  flushIntervalMs: 1e4,
+  autoIngestion: true,
+  autoRetrieval: true,
+  searchMode: "hybrid",
+  hookSearchMode: "hybrid",
+  fileGrepEnabled: true,
+  splashEffect: true,
+  consolidationEnabled: true,
+  consolidationIntervalMs: 6 * 60 * 60 * 1e3,
+  consolidationModel: "claude-haiku-4-5-20251001",
+  consolidationMaxCallsPerRun: 20,
+  selfQueryRouter: true,
+  selfQueryModel: "claude-haiku-4-5-20251001",
+  rerankerEnabled: true,
+  scalingRoadmap: {
+    rerankerAutoTrigger: {
+      enabled: true,
+      broadQueryMinCardinality: 5e4,
+      fetchTopK: 150,
+      returnTopK: 5
+    }
+  },
+  graphRagEnabled: true,
+  wikiEnabled: false,
+  wikiUrl: "",
+  wikiApiKey: "",
+  wikiSyncIntervalMs: 30 * 60 * 1e3,
+  wikiWorkspaceMapping: {
+    exe: "Executive",
+    yoshi: "Engineering",
+    mari: "Marketing",
+    tom: "Engineering",
+    sasha: "Production"
+  },
+  wikiAutoUpdate: true,
+  wikiAutoUpdateThreshold: 0.5,
+  wikiAutoUpdateCreateNew: true,
+  skillLearning: true,
+  skillThreshold: 3,
+  skillModel: "claude-haiku-4-5-20251001",
+  exeHeartbeat: {
+    enabled: true,
+    intervalSeconds: 60,
+    staleInProgressThresholdHours: 2
+  },
+  sessionLifecycle: {
+    idleKillEnabled: true,
+    idleKillTicksRequired: 3,
+    idleKillIntercomAckWindowMs: 1e4,
+    maxAutoInstances: 10
+  }
+};
+
+// src/lib/session-key.ts
+import { execSync } from "child_process";
+var _cached = null;
+function getSessionKey() {
+  if (_cached) return _cached;
+  let pid = process.ppid;
+  for (let i = 0; i < 10; i++) {
+    try {
+      const info = execSync(`ps -p ${pid} -o ppid=,comm=`, {
+        encoding: "utf8",
+        timeout: 2e3
+      }).trim();
+      const match = info.match(/^\s*(\d+)\s+(.+)$/);
+      if (!match) break;
+      const [, ppid, cmd] = match;
+      if (cmd === "claude" || cmd.endsWith("/claude")) {
+        _cached = String(pid);
+        return _cached;
+      }
+      pid = parseInt(ppid, 10);
+      if (pid <= 1) break;
+    } catch {
+      break;
+    }
+  }
+  _cached = process.env.CLAUDE_CODE_SSE_PORT ?? String(process.ppid);
+  return _cached;
+}
+
+// src/adapters/claude/active-agent.ts
+var CACHE_DIR = path2.join(EXE_AI_DIR, "session-cache");
+var STALE_MS = 24 * 60 * 60 * 1e3;
+function getMarkerPath() {
+  return path2.join(CACHE_DIR, `active-agent-${getSessionKey()}.json`);
+}
+function getActiveAgent() {
+  try {
+    const markerPath = getMarkerPath();
+    const raw = readFileSync2(markerPath, "utf8");
+    const data = JSON.parse(raw);
+    if (data.agentId) {
+      if (data.startedAt) {
+        const age = Date.now() - new Date(data.startedAt).getTime();
+        if (age > STALE_MS) {
+          try {
+            unlinkSync(markerPath);
+          } catch {
+          }
+        } else {
+          return {
+            agentId: data.agentId,
+            agentRole: data.agentRole || "employee"
+          };
+        }
+      } else {
+        return {
+          agentId: data.agentId,
+          agentRole: data.agentRole || "employee"
+        };
+      }
+    }
+  } catch {
+  }
+  try {
+    const sessionName = execSync2(
+      "tmux display-message -p '#{session_name}' 2>/dev/null",
+      { encoding: "utf8", timeout: 2e3 }
+    ).trim();
+    const empMatch = sessionName.match(/^(\w+)-exe\d+$/);
+    if (empMatch && empMatch[1] !== "exe") {
+      return { agentId: empMatch[1], agentRole: "employee" };
+    }
+    if (/^exe\d+$/.test(sessionName)) {
+      return { agentId: "exe", agentRole: "COO" };
+    }
+  } catch {
+  }
+  return {
+    agentId: process.env.AGENT_ID || "default",
+    agentRole: process.env.AGENT_ROLE || "employee"
+  };
+}
+
+// src/adapters/claude/hooks/exe-heartbeat-hook.ts
+var HOOK_TIMEOUT_MS = 5e3;
+var CLI_TIMEOUT_MS = 4e3;
+var EXE_AGENT_ID = "exe";
+var watchdog = setTimeout(() => process.exit(0), HOOK_TIMEOUT_MS);
+watchdog.unref();
+var input = "";
+process.stdin.setEncoding("utf8");
+process.stdin.on("data", (chunk) => {
+  input += chunk;
+});
+process.stdin.on("end", () => {
+  void input;
+  try {
+    const agent = getActiveAgent();
+    if (agent.agentId !== EXE_AGENT_ID) {
+      cleanExit();
+      return;
+    }
+    const cliPath = path3.resolve(
+      path3.dirname(fileURLToPath(import.meta.url)),
+      "..",
+      "bin",
+      "exe-heartbeat.js"
+    );
+    let output = "";
+    try {
+      output = execFileSync(process.execPath, [cliPath], {
+        encoding: "utf8",
+        timeout: CLI_TIMEOUT_MS
+      });
+    } catch {
+      cleanExit();
+      return;
+    }
+    if (output.trim().length === 0) {
+      cleanExit();
+      return;
+    }
+    process.stdout.write(JSON.stringify({
+      hookSpecificOutput: {
+        hookEventName: "UserPromptSubmit",
+        additionalContext: output
+      }
+    }));
+  } catch {
+  }
+  cleanExit();
+});
+function cleanExit() {
+  clearTimeout(watchdog);
+  process.exit(0);
+}