@deeplake/hivemind 0.7.75 → 0.7.77

package/hermes/bundle/session-start.js

@@ -53,7 +53,7 @@ var init_index_marker_store = __esm({
 
 // dist/src/hooks/hermes/session-start.js
 import { fileURLToPath as fileURLToPath2 } from "node:url";
-import { dirname as dirname7 } from "node:path";
+import { dirname as dirname10 } from "node:path";
 
 // dist/src/commands/auth.js
 import { execSync } from "node:child_process";
@@ -110,6 +110,19 @@ function decodeJwtPayload(token) {
     return null;
   }
 }
+async function apiGet(path, token, apiUrl, orgId) {
+  const headers = {
+    Authorization: `Bearer ${token}`,
+    "Content-Type": "application/json",
+    ...deeplakeClientHeader()
+  };
+  if (orgId)
+    headers["X-Activeloop-Org-Id"] = orgId;
+  const resp = await fetch(`${apiUrl}${path}`, { headers });
+  if (!resp.ok)
+    throw new Error(`API ${resp.status}: ${await resp.text().catch(() => "")}`);
+  return resp.json();
+}
 async function apiPost(path, body, token, apiUrl, orgId) {
   const headers = {
     Authorization: `Bearer ${token}`,
@@ -123,6 +136,10 @@ async function apiPost(path, body, token, apiUrl, orgId) {
     throw new Error(`API ${resp.status}: ${await resp.text().catch(() => "")}`);
   return resp.json();
 }
+async function listOrgs(token, apiUrl = DEFAULT_API_URL) {
+  const data = await apiGet("/organizations", token, apiUrl);
+  return Array.isArray(data) ? data : [];
+}
 async function healDriftedOrgToken(creds, log7 = () => {
 }) {
   if (!creds.token || !creds.orgId)
@@ -141,6 +158,33 @@ async function healDriftedOrgToken(creds, log7 = () => {
       organization_id: creds.orgId
     }, creds.token, apiUrl);
     const healed = { ...creds, token: tokenData.token.token };
+    try {
+      const orgs = await listOrgs(healed.token, apiUrl);
+      const matchedOrg = orgs.find((o) => o.id === creds.orgId);
+      if (matchedOrg && matchedOrg.name !== creds.orgName) {
+        log7(`orgName realigned: ${creds.orgName ?? "(unset)"} -> ${matchedOrg.name}`);
+        healed.orgName = matchedOrg.name;
+      }
+    } catch (e) {
+      log7(`orgName realign skipped: ${e.message}`);
+    }
+    const currentWs = creds.workspaceId ?? "default";
+    if (currentWs !== "default") {
+      try {
+        const wsList = await listWorkspaces(healed.token, apiUrl, creds.orgId);
+        const lcWs = currentWs.toLowerCase();
+        const wsMatch = wsList.find((w) => w.id === currentWs || w.name && w.name.toLowerCase() === lcWs);
+        if (!wsMatch) {
+          log7(`workspace '${currentWs}' not in org ${creds.orgId} \u2014 reset to default`);
+          healed.workspaceId = "default";
+        } else if (wsMatch.id !== currentWs) {
+          log7(`workspace '${currentWs}' resolved to id '${wsMatch.id}'`);
+          healed.workspaceId = wsMatch.id;
+        }
+      } catch (e) {
+        log7(`workspace realign skipped: ${e.message}`);
+      }
+    }
     saveCredentials(healed);
     log7(`token re-minted for org=${creds.orgId}`);
     return healed;
@@ -149,6 +193,11 @@ async function healDriftedOrgToken(creds, log7 = () => {
     return creds;
   }
 }
+async function listWorkspaces(token, apiUrl = DEFAULT_API_URL, orgId) {
+  const raw = await apiGet("/workspaces", token, apiUrl, orgId);
+  const data = raw.data ?? raw;
+  return Array.isArray(data) ? data : [];
+}
 
 // dist/src/config.js
 import { readFileSync as readFileSync3, existsSync } from "node:fs";
@@ -576,7 +625,7 @@ function getQueryTimeoutMs() {
   return Number(process.env.HIVEMIND_QUERY_TIMEOUT_MS ?? 1e4);
 }
 function sleep2(ms) {
-  return new Promise((resolve2) => setTimeout(resolve2, ms));
+  return new Promise((resolve3) => setTimeout(resolve3, ms));
 }
 function isTimeoutError(error) {
   const name = error instanceof Error ? error.name.toLowerCase() : "";
@@ -606,7 +655,7 @@ var Semaphore = class {
       this.active++;
       return;
     }
-    await new Promise((resolve2) => this.waiting.push(resolve2));
+    await new Promise((resolve3) => this.waiting.push(resolve3));
   }
   release() {
     this.active--;
@@ -1358,13 +1407,13 @@ function maybeAutoMineLocal() {
 
 // dist/src/utils/stdin.js
 function readStdin() {
-  return new Promise((resolve2, reject) => {
+  return new Promise((resolve3, reject) => {
     let data = "";
     process.stdin.setEncoding("utf-8");
     process.stdin.on("data", (chunk) => data += chunk);
     process.stdin.on("end", () => {
       try {
-        resolve2(JSON.parse(data));
+        resolve3(JSON.parse(data));
       } catch (err) {
         reject(new Error(`Failed to parse hook input: ${err}`));
       }
@@ -2205,9 +2254,213 @@ function spawnGraphPullWorker(cwd, bundleDir, deps = {}) {
   }
 }
 
+// dist/src/graph/session-context.js
+import { createHash as createHash3 } from "node:crypto";
+import { existsSync as existsSync13 } from "node:fs";
+import { join as join21 } from "node:path";
+
+// dist/src/graph/last-build.js
+import { existsSync as existsSync11, mkdirSync as mkdirSync10, readFileSync as readFileSync11, renameSync as renameSync5, writeFileSync as writeFileSync9 } from "node:fs";
+import { dirname as dirname7, join as join18 } from "node:path";
+function lastBuildPath(baseDir, worktreeId) {
+  if (worktreeId !== void 0) {
+    return join18(baseDir, "worktrees", worktreeId, ".last-build.json");
+  }
+  return join18(baseDir, ".last-build.json");
+}
+function readLastBuild(baseDir, worktreeId) {
+  let path = lastBuildPath(baseDir, worktreeId);
+  if (!existsSync11(path)) {
+    if (worktreeId === void 0)
+      return null;
+    const legacy = lastBuildPath(baseDir, void 0);
+    if (!existsSync11(legacy))
+      return null;
+    path = legacy;
+  }
+  let raw;
+  try {
+    raw = readFileSync11(path, "utf8");
+  } catch {
+    return null;
+  }
+  let parsed;
+  try {
+    parsed = JSON.parse(raw);
+  } catch {
+    return null;
+  }
+  if (parsed === null || typeof parsed !== "object")
+    return null;
+  const o = parsed;
+  if (typeof o.ts !== "number" || !Number.isFinite(o.ts))
+    return null;
+  if (o.commit_sha !== null && typeof o.commit_sha !== "string")
+    return null;
+  if (typeof o.snapshot_sha256 !== "string")
+    return null;
+  const out = { ts: o.ts, commit_sha: o.commit_sha, snapshot_sha256: o.snapshot_sha256 };
+  if (typeof o.node_count === "number" && Number.isFinite(o.node_count) && o.node_count >= 0) {
+    out.node_count = o.node_count;
+  }
+  if (typeof o.edge_count === "number" && Number.isFinite(o.edge_count) && o.edge_count >= 0) {
+    out.edge_count = o.edge_count;
+  }
+  return out;
+}
+
+// dist/src/graph/snapshot.js
+import { createHash } from "node:crypto";
+import { mkdirSync as mkdirSync12, renameSync as renameSync6, writeFileSync as writeFileSync10 } from "node:fs";
+import { homedir as homedir12 } from "node:os";
+import { dirname as dirname9, join as join20 } from "node:path";
+
+// dist/src/graph/history.js
+import { appendFileSync as appendFileSync2, existsSync as existsSync12, mkdirSync as mkdirSync11, readFileSync as readFileSync12 } from "node:fs";
+import { dirname as dirname8, join as join19 } from "node:path";
+
+// dist/src/graph/resolve/cross-file.js
+import { posix } from "node:path";
+
+// dist/src/graph/snapshot.js
+function graphsRoot() {
+  return process.env.HIVEMIND_GRAPHS_HOME ?? join20(homedir12(), ".hivemind", "graphs");
+}
+function repoDir(repoKey) {
+  return join20(graphsRoot(), repoKey);
+}
+
+// dist/src/utils/repo-identity.js
+import { execSync as execSync2 } from "node:child_process";
+import { createHash as createHash2 } from "node:crypto";
+import { basename as basename2, resolve as resolve2 } from "node:path";
+var DEFAULT_PORTS = {
+  http: "80",
+  https: "443",
+  ssh: "22",
+  git: "9418"
+};
+function normalizeGitRemoteUrl(url) {
+  let s = url.trim();
+  const schemeMatch = s.match(/^([a-z][a-z0-9+.-]*):\/\//i);
+  const scheme = schemeMatch ? schemeMatch[1].toLowerCase() : null;
+  if (schemeMatch)
+    s = s.slice(schemeMatch[0].length);
+  if (!scheme) {
+    const scp = s.match(/^(?:[^@/\s]+@)?([^:/\s]+):(.+)$/);
+    if (scp)
+      s = `${scp[1]}/${scp[2]}`;
+  }
+  s = s.replace(/^[^@/]+@/, "");
+  if (scheme && DEFAULT_PORTS[scheme]) {
+    s = s.replace(new RegExp(`^([^/]+):${DEFAULT_PORTS[scheme]}(/|$)`), "$1$2");
+  }
+  s = s.replace(/\.git\/?$/i, "");
+  s = s.replace(/\/+$/, "");
+  return s.toLowerCase();
+}
+function deriveProjectKey(cwd) {
+  const absCwd = resolve2(cwd);
+  const project = basename2(absCwd) || "unknown";
+  let signature = null;
+  try {
+    const raw = execSync2("git config --get remote.origin.url", {
+      cwd: absCwd,
+      encoding: "utf-8",
+      stdio: ["ignore", "pipe", "ignore"]
+    }).trim();
+    signature = raw ? normalizeGitRemoteUrl(raw) : null;
+  } catch {
+  }
+  const input = signature ?? absCwd;
+  const key = createHash2("sha1").update(input).digest("hex").slice(0, 16);
+  return { key, project };
+}
+
+// dist/src/graph/session-context.js
+function workTreeIdFor(cwd) {
+  return createHash3("sha256").update(cwd).digest("hex").slice(0, 16);
+}
+function graphContextLine(cwd, deps = {}) {
+  let key;
+  let snapshotsDir;
+  let baseDir;
+  try {
+    key = deriveProjectKey(cwd).key;
+    baseDir = repoDir(key);
+    snapshotsDir = join21(baseDir, "snapshots");
+  } catch {
+    return null;
+  }
+  if (!existsSync13(snapshotsDir))
+    return null;
+  const last = readLastBuild(baseDir, workTreeIdFor(cwd));
+  if (last === null)
+    return null;
+  if (last.commit_sha !== null && !/^[0-9a-f]{4,64}$/.test(last.commit_sha))
+    return null;
+  if (!/^[0-9a-f]{64}$/.test(last.snapshot_sha256))
+    return null;
+  const now = (deps.now ?? Date.now)();
+  const ageMs = Math.max(0, now - last.ts);
+  const nodesStr = last.node_count !== void 0 ? String(last.node_count) : "?";
+  const edgesStr = last.edge_count !== void 0 ? String(last.edge_count) : "?";
+  const commitStr = last.commit_sha !== null ? last.commit_sha.slice(0, 7) : "no-commit";
+  const ageStr = formatAge(ageMs);
+  const snapshotFile = last.commit_sha ?? last.snapshot_sha256;
+  const snapshotPath = join21(snapshotsDir, `${snapshotFile}.json`);
+  const STALE_WARN_MS = 60 * 60 * 1e3;
+  const STALE_HARD_MS = 24 * 60 * 60 * 1e3;
+  let staleness;
+  if (ageMs >= STALE_HARD_MS) {
+    staleness = "  \u26A0\uFE0F STALE: this snapshot is over a day old; the auto-rebuild may have stopped.\n     Prefer reading current source for any file you suspect has changed.";
+  } else if (ageMs >= STALE_WARN_MS) {
+    staleness = "  \u26A0\uFE0F Possibly out of date (> 1h since last build). For any file you've edited\n     in this session, fall back to reading the live source instead of the graph.";
+  } else {
+    staleness = "  Freshness: auto-rebuilds run on Stop/SessionEnd; if a file's mtime is newer\n  than the build timestamp above, prefer reading the live source for that file.";
+  }
+  return [
+    "",
+    "LOCAL CODE GRAPH (TypeScript / JavaScript / Python, AST-based):",
+    `  ${nodesStr} nodes, ${edgesStr} edges (commit ${commitStr}, built ${ageStr} ago)`,
+    "",
+    "  Use it as a fast INDEX to locate the few files/symbols that matter, then",
+    "  open them with Read to answer. It is NOT a substitute for the source: it",
+    "  omits instance-method calls (obj.method()), nested/inner functions, and",
+    "  dynamic dispatch \u2014 so confirm every claim against the file before stating it.",
+    "",
+    "  Query via the Deeplake mount (intercepted \u2014 use `cat`, not `ls`):",
+    "    cat ~/.deeplake/memory/graph/query/<pattern>   \u2190 start here",
+    "        search + 1-hop expand (callers, callees, imports). AND: query/<a>+<b>.",
+    "    cat ~/.deeplake/memory/graph/find/<pattern>     substring search \u2192 handles",
+    "    cat ~/.deeplake/memory/graph/show/<handle-or-pattern>   node + 1-hop neighbors",
+    "    cat ~/.deeplake/memory/graph/neighborhood/<file>        symbols + cross-file links",
+    "    Also: index.md \xB7 layers \xB7 tour \xB7 path/<from>/<to>",
+    "",
+    "  Then READ the files the graph points you to \u2014 don't answer from the graph",
+    "  alone. Cross-file calls/imports resolved for named imports across TS/JS/Python;",
+    "  bare (npm)/aliased/barrel/dynamic + instance-method dispatch stay unresolved.",
+    `  Raw snapshot (fallback): ${snapshotPath}`,
+    staleness
+  ].join("\n");
+}
+function formatAge(ms) {
+  const s = Math.floor(ms / 1e3);
+  if (s < 60)
+    return `${s}s`;
+  const m = Math.floor(s / 60);
+  if (m < 60)
+    return `${m}m`;
+  const h = Math.floor(m / 60);
+  if (h < 24)
+    return `${h}h`;
+  const d = Math.floor(h / 24);
+  return `${d}d`;
+}
+
 // dist/src/hooks/hermes/session-start.js
 var log6 = (msg) => log("hermes-session-start", msg);
-var __bundleDir = dirname7(fileURLToPath2(import.meta.url));
+var __bundleDir = dirname10(fileURLToPath2(import.meta.url));
 var context = `DEEPLAKE MEMORY: Persistent memory at ~/.deeplake/memory/ shared across sessions, users, and agents.
 
 Structure: index.md (start here) \u2192 summaries/*.md \u2192 sessions/*.jsonl (last resort). Do NOT jump straight to JSONL.
@@ -2314,9 +2567,10 @@ Not logged in to Deeplake. Run: hivemind login${localMinedNote}${versionNotice}`
   const baseWithGoals = creds?.token ? `${baseContext}
 
 ${GOALS_INSTRUCTIONS_CLI}` : baseContext;
-  const additional = rulesBlock ? `${baseWithGoals}
+  const graphNote = graphContextLine(cwd) ?? "";
+  const additional = (rulesBlock ? `${baseWithGoals}
 
-${rulesBlock}` : baseWithGoals;
+${rulesBlock}` : baseWithGoals) + graphNote;
   console.log(JSON.stringify({ context: additional }));
 }
 main().catch((e) => {