@deeplake/hivemind 0.6.47

package/cursor/bundle/commands/auth-login.js

@@ -0,0 +1,862 @@
+#!/usr/bin/env node
+
+// dist/src/commands/auth.js
+import { readFileSync, writeFileSync, existsSync, mkdirSync, unlinkSync } from "node:fs";
+import { join } from "node:path";
+import { homedir } from "node:os";
+import { execSync } from "node:child_process";
+var CONFIG_DIR = join(homedir(), ".deeplake");
+var CREDS_PATH = join(CONFIG_DIR, "credentials.json");
+var DEFAULT_API_URL = "https://api.deeplake.ai";
+function loadCredentials() {
+  if (!existsSync(CREDS_PATH))
+    return null;
+  try {
+    return JSON.parse(readFileSync(CREDS_PATH, "utf-8"));
+  } catch {
+    return null;
+  }
+}
+function saveCredentials(creds) {
+  if (!existsSync(CONFIG_DIR))
+    mkdirSync(CONFIG_DIR, { recursive: true, mode: 448 });
+  writeFileSync(CREDS_PATH, JSON.stringify({ ...creds, savedAt: (/* @__PURE__ */ new Date()).toISOString() }, null, 2), { mode: 384 });
+}
+function deleteCredentials() {
+  if (existsSync(CREDS_PATH)) {
+    unlinkSync(CREDS_PATH);
+    return true;
+  }
+  return false;
+}
+async function apiGet(path, token, apiUrl, orgId) {
+  const headers = {
+    Authorization: `Bearer ${token}`,
+    "Content-Type": "application/json"
+  };
+  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}`,
+    "Content-Type": "application/json"
+  };
+  if (orgId)
+    headers["X-Activeloop-Org-Id"] = orgId;
+  const resp = await fetch(`${apiUrl}${path}`, { method: "POST", headers, body: JSON.stringify(body) });
+  if (!resp.ok)
+    throw new Error(`API ${resp.status}: ${await resp.text().catch(() => "")}`);
+  return resp.json();
+}
+async function apiDelete(path, token, apiUrl, orgId) {
+  const headers = {
+    Authorization: `Bearer ${token}`,
+    "Content-Type": "application/json"
+  };
+  if (orgId)
+    headers["X-Activeloop-Org-Id"] = orgId;
+  const resp = await fetch(`${apiUrl}${path}`, { method: "DELETE", headers });
+  if (!resp.ok)
+    throw new Error(`API ${resp.status}: ${await resp.text().catch(() => "")}`);
+}
+async function requestDeviceCode(apiUrl = DEFAULT_API_URL) {
+  const resp = await fetch(`${apiUrl}/auth/device/code`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" }
+  });
+  if (!resp.ok)
+    throw new Error(`Device flow unavailable: HTTP ${resp.status}`);
+  return resp.json();
+}
+async function pollForToken(deviceCode, apiUrl = DEFAULT_API_URL) {
+  const resp = await fetch(`${apiUrl}/auth/device/token`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({ device_code: deviceCode })
+  });
+  if (resp.ok)
+    return resp.json();
+  if (resp.status === 400) {
+    const err = await resp.json().catch(() => null);
+    if (err?.error === "authorization_pending" || err?.error === "slow_down")
+      return null;
+    if (err?.error === "expired_token")
+      throw new Error("Device code expired. Try again.");
+    if (err?.error === "access_denied")
+      throw new Error("Authorization denied.");
+  }
+  throw new Error(`Token polling failed: HTTP ${resp.status}`);
+}
+function openBrowser(url) {
+  try {
+    const cmd = process.platform === "darwin" ? `open "${url}"` : process.platform === "win32" ? `start "${url}"` : `xdg-open "${url}" 2>/dev/null`;
+    execSync(cmd, { stdio: "ignore", timeout: 5e3 });
+    return true;
+  } catch {
+    return false;
+  }
+}
+async function deviceFlowLogin(apiUrl = DEFAULT_API_URL) {
+  const code = await requestDeviceCode(apiUrl);
+  const opened = openBrowser(code.verification_uri_complete);
+  const msg = [
+    "\nDeeplake Authentication",
+    "\u2500".repeat(40),
+    `
+Open this URL: ${code.verification_uri_complete}`,
+    `Or visit ${code.verification_uri} and enter code: ${code.user_code}`,
+    opened ? "\nBrowser opened. Waiting for sign in..." : "\nWaiting for sign in..."
+  ].join("\n");
+  process.stderr.write(msg + "\n");
+  const interval = Math.max(code.interval || 5, 5) * 1e3;
+  const deadline = Date.now() + code.expires_in * 1e3;
+  while (Date.now() < deadline) {
+    await new Promise((r) => setTimeout(r, interval));
+    const result = await pollForToken(code.device_code, apiUrl);
+    if (result) {
+      process.stderr.write("\nAuthentication successful!\n");
+      return { token: result.access_token, expiresIn: result.expires_in };
+    }
+  }
+  throw new Error("Device code expired.");
+}
+async function listOrgs(token, apiUrl = DEFAULT_API_URL) {
+  const data = await apiGet("/organizations", token, apiUrl);
+  return Array.isArray(data) ? data : [];
+}
+async function switchOrg(orgId, orgName) {
+  const creds = loadCredentials();
+  if (!creds)
+    throw new Error("Not logged in. Run deeplake login first.");
+  saveCredentials({ ...creds, orgId, orgName });
+}
+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 : [];
+}
+async function switchWorkspace(workspaceId) {
+  const creds = loadCredentials();
+  if (!creds)
+    throw new Error("Not logged in. Run deeplake login first.");
+  saveCredentials({ ...creds, workspaceId });
+}
+async function inviteMember(username, accessMode, token, orgId, apiUrl = DEFAULT_API_URL) {
+  await apiPost(`/organizations/${orgId}/members/invite`, { username, access_mode: accessMode }, token, apiUrl, orgId);
+}
+async function listMembers(token, orgId, apiUrl = DEFAULT_API_URL) {
+  const data = await apiGet(`/organizations/${orgId}/members`, token, apiUrl, orgId);
+  return data.members ?? [];
+}
+async function removeMember(userId, token, orgId, apiUrl = DEFAULT_API_URL) {
+  await apiDelete(`/organizations/${orgId}/members/${userId}`, token, apiUrl, orgId);
+}
+async function login(apiUrl = DEFAULT_API_URL) {
+  const { token: authToken } = await deviceFlowLogin(apiUrl);
+  const user = await apiGet("/me", authToken, apiUrl);
+  const userName = user.name || (user.email ? user.email.split("@")[0] : "unknown");
+  process.stderr.write(`
+Logged in as: ${userName}
+`);
+  const orgs = await listOrgs(authToken, apiUrl);
+  let orgId;
+  let orgName;
+  if (orgs.length === 1) {
+    orgId = orgs[0].id;
+    orgName = orgs[0].name;
+    process.stderr.write(`Organization: ${orgName}
+`);
+  } else {
+    process.stderr.write("\nOrganizations:\n");
+    orgs.forEach((org, i) => process.stderr.write(`  ${i + 1}. ${org.name}
+`));
+    orgId = orgs[0].id;
+    orgName = orgs[0].name;
+    process.stderr.write(`
+Using: ${orgName}
+`);
+  }
+  const tokenName = `deeplake-plugin-${(/* @__PURE__ */ new Date()).toISOString().slice(0, 10)}`;
+  const tokenData = await apiPost("/users/me/tokens", {
+    name: tokenName,
+    duration: 365 * 24 * 3600,
+    organization_id: orgId
+  }, authToken, apiUrl);
+  const apiToken = tokenData.token.token;
+  const creds = {
+    token: apiToken,
+    orgId,
+    orgName,
+    userName,
+    workspaceId: "default",
+    apiUrl,
+    savedAt: (/* @__PURE__ */ new Date()).toISOString()
+  };
+  saveCredentials(creds);
+  return creds;
+}
+
+// dist/src/config.js
+import { readFileSync as readFileSync2, existsSync as existsSync2 } from "node:fs";
+import { join as join2 } from "node:path";
+import { homedir as homedir2, userInfo } from "node:os";
+function loadConfig() {
+  const home = homedir2();
+  const credPath = join2(home, ".deeplake", "credentials.json");
+  let creds = null;
+  if (existsSync2(credPath)) {
+    try {
+      creds = JSON.parse(readFileSync2(credPath, "utf-8"));
+    } catch {
+      return null;
+    }
+  }
+  const token = process.env.HIVEMIND_TOKEN ?? creds?.token;
+  const orgId = process.env.HIVEMIND_ORG_ID ?? creds?.orgId;
+  if (!token || !orgId)
+    return null;
+  return {
+    token,
+    orgId,
+    orgName: creds?.orgName ?? orgId,
+    userName: creds?.userName || userInfo().username || "unknown",
+    workspaceId: process.env.HIVEMIND_WORKSPACE_ID ?? creds?.workspaceId ?? "default",
+    apiUrl: process.env.HIVEMIND_API_URL ?? creds?.apiUrl ?? "https://api.deeplake.ai",
+    tableName: process.env.HIVEMIND_TABLE ?? "memory",
+    sessionsTableName: process.env.HIVEMIND_SESSIONS_TABLE ?? "sessions",
+    memoryPath: process.env.HIVEMIND_MEMORY_PATH ?? join2(home, ".deeplake", "memory")
+  };
+}
+
+// dist/src/deeplake-api.js
+import { randomUUID } from "node:crypto";
+import { existsSync as existsSync3, mkdirSync as mkdirSync2, readFileSync as readFileSync3, writeFileSync as writeFileSync2 } from "node:fs";
+import { join as join4 } from "node:path";
+import { tmpdir } from "node:os";
+
+// dist/src/utils/debug.js
+import { appendFileSync } from "node:fs";
+import { join as join3 } from "node:path";
+import { homedir as homedir3 } from "node:os";
+var DEBUG = process.env.HIVEMIND_DEBUG === "1";
+var LOG = join3(homedir3(), ".deeplake", "hook-debug.log");
+function log(tag, msg) {
+  if (!DEBUG)
+    return;
+  appendFileSync(LOG, `${(/* @__PURE__ */ new Date()).toISOString()} [${tag}] ${msg}
+`);
+}
+
+// dist/src/utils/sql.js
+function sqlStr(value) {
+  return value.replace(/\\/g, "\\\\").replace(/'/g, "''").replace(/\0/g, "").replace(/[\x01-\x08\x0b\x0c\x0e-\x1f\x7f]/g, "");
+}
+
+// dist/src/deeplake-api.js
+var log2 = (msg) => log("sdk", msg);
+function summarizeSql(sql, maxLen = 220) {
+  const compact = sql.replace(/\s+/g, " ").trim();
+  return compact.length > maxLen ? `${compact.slice(0, maxLen)}...` : compact;
+}
+function traceSql(msg) {
+  const traceEnabled = process.env.HIVEMIND_TRACE_SQL === "1" || process.env.HIVEMIND_DEBUG === "1";
+  if (!traceEnabled)
+    return;
+  process.stderr.write(`[deeplake-sql] ${msg}
+`);
+  if (process.env.HIVEMIND_DEBUG === "1")
+    log2(msg);
+}
+var RETRYABLE_CODES = /* @__PURE__ */ new Set([429, 500, 502, 503, 504]);
+var MAX_RETRIES = 3;
+var BASE_DELAY_MS = 500;
+var MAX_CONCURRENCY = 5;
+var QUERY_TIMEOUT_MS = Number(process.env.HIVEMIND_QUERY_TIMEOUT_MS ?? 1e4);
+var INDEX_MARKER_TTL_MS = Number(process.env.HIVEMIND_INDEX_MARKER_TTL_MS ?? 6 * 60 * 6e4);
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+function isTimeoutError(error) {
+  const name = error instanceof Error ? error.name.toLowerCase() : "";
+  const message = error instanceof Error ? error.message.toLowerCase() : String(error).toLowerCase();
+  return name.includes("timeout") || name === "aborterror" || message.includes("timeout") || message.includes("timed out");
+}
+function isDuplicateIndexError(error) {
+  const message = error instanceof Error ? error.message.toLowerCase() : String(error).toLowerCase();
+  return message.includes("duplicate key value violates unique constraint") || message.includes("pg_class_relname_nsp_index") || message.includes("already exists");
+}
+function isSessionInsertQuery(sql) {
+  return /^\s*insert\s+into\s+"[^"]+"\s*\(\s*id\s*,\s*path\s*,\s*filename\s*,\s*message\s*,/i.test(sql);
+}
+function isTransientHtml403(text) {
+  const body = text.toLowerCase();
+  return body.includes("<html") || body.includes("403 forbidden") || body.includes("cloudflare") || body.includes("nginx");
+}
+function getIndexMarkerDir() {
+  return process.env.HIVEMIND_INDEX_MARKER_DIR ?? join4(tmpdir(), "hivemind-deeplake-indexes");
+}
+var Semaphore = class {
+  max;
+  waiting = [];
+  active = 0;
+  constructor(max) {
+    this.max = max;
+  }
+  async acquire() {
+    if (this.active < this.max) {
+      this.active++;
+      return;
+    }
+    await new Promise((resolve) => this.waiting.push(resolve));
+  }
+  release() {
+    this.active--;
+    const next = this.waiting.shift();
+    if (next) {
+      this.active++;
+      next();
+    }
+  }
+};
+var DeeplakeApi = class {
+  token;
+  apiUrl;
+  orgId;
+  workspaceId;
+  tableName;
+  _pendingRows = [];
+  _sem = new Semaphore(MAX_CONCURRENCY);
+  _tablesCache = null;
+  constructor(token, apiUrl, orgId, workspaceId, tableName) {
+    this.token = token;
+    this.apiUrl = apiUrl;
+    this.orgId = orgId;
+    this.workspaceId = workspaceId;
+    this.tableName = tableName;
+  }
+  /** Execute SQL with retry on transient errors and bounded concurrency. */
+  async query(sql) {
+    const startedAt = Date.now();
+    const summary = summarizeSql(sql);
+    traceSql(`query start: ${summary}`);
+    await this._sem.acquire();
+    try {
+      const rows = await this._queryWithRetry(sql);
+      traceSql(`query ok (${Date.now() - startedAt}ms, rows=${rows.length}): ${summary}`);
+      return rows;
+    } catch (e) {
+      const message = e instanceof Error ? e.message : String(e);
+      traceSql(`query fail (${Date.now() - startedAt}ms): ${summary} :: ${message}`);
+      throw e;
+    } finally {
+      this._sem.release();
+    }
+  }
+  async _queryWithRetry(sql) {
+    let lastError;
+    for (let attempt = 0; attempt <= MAX_RETRIES; attempt++) {
+      let resp;
+      try {
+        const signal = AbortSignal.timeout(QUERY_TIMEOUT_MS);
+        resp = await fetch(`${this.apiUrl}/workspaces/${this.workspaceId}/tables/query`, {
+          method: "POST",
+          headers: {
+            Authorization: `Bearer ${this.token}`,
+            "Content-Type": "application/json",
+            "X-Activeloop-Org-Id": this.orgId
+          },
+          signal,
+          body: JSON.stringify({ query: sql })
+        });
+      } catch (e) {
+        if (isTimeoutError(e)) {
+          lastError = new Error(`Query timeout after ${QUERY_TIMEOUT_MS}ms`);
+          throw lastError;
+        }
+        lastError = e instanceof Error ? e : new Error(String(e));
+        if (attempt < MAX_RETRIES) {
+          const delay = BASE_DELAY_MS * Math.pow(2, attempt) + Math.random() * 200;
+          log2(`query retry ${attempt + 1}/${MAX_RETRIES} (fetch error: ${lastError.message}) in ${delay.toFixed(0)}ms`);
+          await sleep(delay);
+          continue;
+        }
+        throw lastError;
+      }
+      if (resp.ok) {
+        const raw = await resp.json();
+        if (!raw?.rows || !raw?.columns)
+          return [];
+        return raw.rows.map((row) => Object.fromEntries(raw.columns.map((col, i) => [col, row[i]])));
+      }
+      const text = await resp.text().catch(() => "");
+      const retryable403 = isSessionInsertQuery(sql) && (resp.status === 401 || resp.status === 403 && (text.length === 0 || isTransientHtml403(text)));
+      if (attempt < MAX_RETRIES && (RETRYABLE_CODES.has(resp.status) || retryable403)) {
+        const delay = BASE_DELAY_MS * Math.pow(2, attempt) + Math.random() * 200;
+        log2(`query retry ${attempt + 1}/${MAX_RETRIES} (${resp.status}) in ${delay.toFixed(0)}ms`);
+        await sleep(delay);
+        continue;
+      }
+      throw new Error(`Query failed: ${resp.status}: ${text.slice(0, 200)}`);
+    }
+    throw lastError ?? new Error("Query failed: max retries exceeded");
+  }
+  // ── Writes ──────────────────────────────────────────────────────────────────
+  /** Queue rows for writing. Call commit() to flush. */
+  appendRows(rows) {
+    this._pendingRows.push(...rows);
+  }
+  /** Flush pending rows via SQL. */
+  async commit() {
+    if (this._pendingRows.length === 0)
+      return;
+    const rows = this._pendingRows;
+    this._pendingRows = [];
+    const CONCURRENCY = 10;
+    for (let i = 0; i < rows.length; i += CONCURRENCY) {
+      const chunk = rows.slice(i, i + CONCURRENCY);
+      await Promise.allSettled(chunk.map((r) => this.upsertRowSql(r)));
+    }
+    log2(`commit: ${rows.length} rows`);
+  }
+  async upsertRowSql(row) {
+    const ts = (/* @__PURE__ */ new Date()).toISOString();
+    const cd = row.creationDate ?? ts;
+    const lud = row.lastUpdateDate ?? ts;
+    const exists = await this.query(`SELECT path FROM "${this.tableName}" WHERE path = '${sqlStr(row.path)}' LIMIT 1`);
+    if (exists.length > 0) {
+      let setClauses = `summary = E'${sqlStr(row.contentText)}', mime_type = '${sqlStr(row.mimeType)}', size_bytes = ${row.sizeBytes}, last_update_date = '${lud}'`;
+      if (row.project !== void 0)
+        setClauses += `, project = '${sqlStr(row.project)}'`;
+      if (row.description !== void 0)
+        setClauses += `, description = '${sqlStr(row.description)}'`;
+      await this.query(`UPDATE "${this.tableName}" SET ${setClauses} WHERE path = '${sqlStr(row.path)}'`);
+    } else {
+      const id = randomUUID();
+      let cols = "id, path, filename, summary, mime_type, size_bytes, creation_date, last_update_date";
+      let vals = `'${id}', '${sqlStr(row.path)}', '${sqlStr(row.filename)}', E'${sqlStr(row.contentText)}', '${sqlStr(row.mimeType)}', ${row.sizeBytes}, '${cd}', '${lud}'`;
+      if (row.project !== void 0) {
+        cols += ", project";
+        vals += `, '${sqlStr(row.project)}'`;
+      }
+      if (row.description !== void 0) {
+        cols += ", description";
+        vals += `, '${sqlStr(row.description)}'`;
+      }
+      await this.query(`INSERT INTO "${this.tableName}" (${cols}) VALUES (${vals})`);
+    }
+  }
+  /** Update specific columns on a row by path. */
+  async updateColumns(path, columns) {
+    const setClauses = Object.entries(columns).map(([col, val]) => typeof val === "number" ? `${col} = ${val}` : `${col} = '${sqlStr(String(val))}'`).join(", ");
+    await this.query(`UPDATE "${this.tableName}" SET ${setClauses} WHERE path = '${sqlStr(path)}'`);
+  }
+  // ── Convenience ─────────────────────────────────────────────────────────────
+  /** Create a BM25 search index on a column. */
+  async createIndex(column) {
+    await this.query(`CREATE INDEX IF NOT EXISTS idx_${sqlStr(column)}_bm25 ON "${this.tableName}" USING deeplake_index ("${column}")`);
+  }
+  buildLookupIndexName(table, suffix) {
+    return `idx_${table}_${suffix}`.replace(/[^a-zA-Z0-9_]/g, "_");
+  }
+  getLookupIndexMarkerPath(table, suffix) {
+    const markerKey = [
+      this.workspaceId,
+      this.orgId,
+      table,
+      suffix
+    ].join("__").replace(/[^a-zA-Z0-9_.-]/g, "_");
+    return join4(getIndexMarkerDir(), `${markerKey}.json`);
+  }
+  hasFreshLookupIndexMarker(table, suffix) {
+    const markerPath = this.getLookupIndexMarkerPath(table, suffix);
+    if (!existsSync3(markerPath))
+      return false;
+    try {
+      const raw = JSON.parse(readFileSync3(markerPath, "utf-8"));
+      const updatedAt = raw.updatedAt ? new Date(raw.updatedAt).getTime() : NaN;
+      if (!Number.isFinite(updatedAt) || Date.now() - updatedAt > INDEX_MARKER_TTL_MS)
+        return false;
+      return true;
+    } catch {
+      return false;
+    }
+  }
+  markLookupIndexReady(table, suffix) {
+    mkdirSync2(getIndexMarkerDir(), { recursive: true });
+    writeFileSync2(this.getLookupIndexMarkerPath(table, suffix), JSON.stringify({ updatedAt: (/* @__PURE__ */ new Date()).toISOString() }), "utf-8");
+  }
+  async ensureLookupIndex(table, suffix, columnsSql) {
+    if (this.hasFreshLookupIndexMarker(table, suffix))
+      return;
+    const indexName = this.buildLookupIndexName(table, suffix);
+    try {
+      await this.query(`CREATE INDEX IF NOT EXISTS "${indexName}" ON "${table}" ${columnsSql}`);
+      this.markLookupIndexReady(table, suffix);
+    } catch (e) {
+      if (isDuplicateIndexError(e)) {
+        this.markLookupIndexReady(table, suffix);
+        return;
+      }
+      log2(`index "${indexName}" skipped: ${e.message}`);
+    }
+  }
+  /** List all tables in the workspace (with retry). */
+  async listTables(forceRefresh = false) {
+    if (!forceRefresh && this._tablesCache)
+      return [...this._tablesCache];
+    const { tables, cacheable } = await this._fetchTables();
+    if (cacheable)
+      this._tablesCache = [...tables];
+    return tables;
+  }
+  async _fetchTables() {
+    for (let attempt = 0; attempt <= MAX_RETRIES; attempt++) {
+      try {
+        const resp = await fetch(`${this.apiUrl}/workspaces/${this.workspaceId}/tables`, {
+          headers: {
+            Authorization: `Bearer ${this.token}`,
+            "X-Activeloop-Org-Id": this.orgId
+          }
+        });
+        if (resp.ok) {
+          const data = await resp.json();
+          return {
+            tables: (data.tables ?? []).map((t) => t.table_name),
+            cacheable: true
+          };
+        }
+        if (attempt < MAX_RETRIES && RETRYABLE_CODES.has(resp.status)) {
+          await sleep(BASE_DELAY_MS * Math.pow(2, attempt) + Math.random() * 200);
+          continue;
+        }
+        return { tables: [], cacheable: false };
+      } catch {
+        if (attempt < MAX_RETRIES) {
+          await sleep(BASE_DELAY_MS * Math.pow(2, attempt));
+          continue;
+        }
+        return { tables: [], cacheable: false };
+      }
+    }
+    return { tables: [], cacheable: false };
+  }
+  /** Create the memory table if it doesn't already exist. Migrate columns on existing tables. */
+  async ensureTable(name) {
+    const tbl = name ?? this.tableName;
+    const tables = await this.listTables();
+    if (!tables.includes(tbl)) {
+      log2(`table "${tbl}" not found, creating`);
+      await this.query(`CREATE TABLE IF NOT EXISTS "${tbl}" (id TEXT NOT NULL DEFAULT '', path TEXT NOT NULL DEFAULT '', filename TEXT NOT NULL DEFAULT '', summary TEXT NOT NULL DEFAULT '', author TEXT NOT NULL DEFAULT '', mime_type TEXT NOT NULL DEFAULT 'text/plain', size_bytes BIGINT NOT NULL DEFAULT 0, project TEXT NOT NULL DEFAULT '', description TEXT NOT NULL DEFAULT '', agent TEXT NOT NULL DEFAULT '', creation_date TEXT NOT NULL DEFAULT '', last_update_date TEXT NOT NULL DEFAULT '') USING deeplake`);
+      log2(`table "${tbl}" created`);
+      if (!tables.includes(tbl))
+        this._tablesCache = [...tables, tbl];
+    }
+  }
+  /** Create the sessions table (uses JSONB for message since every row is a JSON event). */
+  async ensureSessionsTable(name) {
+    const tables = await this.listTables();
+    if (!tables.includes(name)) {
+      log2(`table "${name}" not found, creating`);
+      await this.query(`CREATE TABLE IF NOT EXISTS "${name}" (id TEXT NOT NULL DEFAULT '', path TEXT NOT NULL DEFAULT '', filename TEXT NOT NULL DEFAULT '', message JSONB, author TEXT NOT NULL DEFAULT '', mime_type TEXT NOT NULL DEFAULT 'application/json', size_bytes BIGINT NOT NULL DEFAULT 0, project TEXT NOT NULL DEFAULT '', description TEXT NOT NULL DEFAULT '', agent TEXT NOT NULL DEFAULT '', creation_date TEXT NOT NULL DEFAULT '', last_update_date TEXT NOT NULL DEFAULT '') USING deeplake`);
+      log2(`table "${name}" created`);
+      if (!tables.includes(name))
+        this._tablesCache = [...tables, name];
+    }
+    await this.ensureLookupIndex(name, "path_creation_date", `("path", "creation_date")`);
+  }
+};
+
+// dist/src/commands/session-prune.js
+import { createInterface } from "node:readline";
+function parseArgs(argv) {
+  let before;
+  let sessionId;
+  let all = false;
+  let yes = false;
+  for (let i = 0; i < argv.length; i++) {
+    const arg = argv[i];
+    if (arg === "--before" && argv[i + 1]) {
+      before = argv[++i];
+    } else if (arg === "--session-id" && argv[i + 1]) {
+      sessionId = argv[++i];
+    } else if (arg === "--all") {
+      all = true;
+    } else if (arg === "--yes" || arg === "-y") {
+      yes = true;
+    }
+  }
+  return { before, sessionId, all, yes };
+}
+function confirm(message) {
+  const rl = createInterface({ input: process.stdin, output: process.stderr });
+  return new Promise((resolve) => {
+    rl.question(`${message} [y/N] `, (answer) => {
+      rl.close();
+      resolve(answer.trim().toLowerCase() === "y");
+    });
+  });
+}
+function extractSessionId(path) {
+  const m = path.match(/\/sessions\/[^/]+\/[^/]+_([^.]+)\.jsonl$/);
+  return m ? m[1] : path.split("/").pop()?.replace(/\.jsonl$/, "") ?? path;
+}
+async function listSessions(api, sessionsTable, author) {
+  const rows = await api.query(`SELECT path, COUNT(*) as cnt, MIN(creation_date) as first_event, MAX(creation_date) as last_event, MAX(project) as project FROM "${sessionsTable}" WHERE author = '${sqlStr(author)}' GROUP BY path ORDER BY first_event DESC`);
+  return rows.map((r) => ({
+    path: String(r.path),
+    rowCount: Number(r.cnt),
+    firstEvent: String(r.first_event),
+    lastEvent: String(r.last_event),
+    project: String(r.project ?? "")
+  }));
+}
+async function deleteSessions(config, sessionPaths) {
+  if (sessionPaths.length === 0)
+    return { sessionsDeleted: 0, summariesDeleted: 0 };
+  const sessionsApi = new DeeplakeApi(config.token, config.apiUrl, config.orgId, config.workspaceId, config.sessionsTableName);
+  const memoryApi = new DeeplakeApi(config.token, config.apiUrl, config.orgId, config.workspaceId, config.tableName);
+  let sessionsDeleted = 0;
+  let summariesDeleted = 0;
+  for (const sessionPath of sessionPaths) {
+    await sessionsApi.query(`DELETE FROM "${config.sessionsTableName}" WHERE path = '${sqlStr(sessionPath)}'`);
+    sessionsDeleted++;
+    const sessionId = extractSessionId(sessionPath);
+    const summaryPath = `/summaries/${config.userName}/${sessionId}.md`;
+    const existing = await memoryApi.query(`SELECT path FROM "${config.tableName}" WHERE path = '${sqlStr(summaryPath)}' LIMIT 1`);
+    if (existing.length > 0) {
+      await memoryApi.query(`DELETE FROM "${config.tableName}" WHERE path = '${sqlStr(summaryPath)}'`);
+      summariesDeleted++;
+    }
+  }
+  return { sessionsDeleted, summariesDeleted };
+}
+async function sessionPrune(argv) {
+  const config = loadConfig();
+  if (!config) {
+    console.error("Not logged in. Run: deeplake login");
+    process.exit(1);
+  }
+  const { before, sessionId, all, yes } = parseArgs(argv);
+  const author = config.userName;
+  const sessionsApi = new DeeplakeApi(config.token, config.apiUrl, config.orgId, config.workspaceId, config.sessionsTableName);
+  const sessions = await listSessions(sessionsApi, config.sessionsTableName, author);
+  if (sessions.length === 0) {
+    console.log(`No sessions found for author "${author}".`);
+    return;
+  }
+  let targets;
+  if (sessionId) {
+    targets = sessions.filter((s) => extractSessionId(s.path) === sessionId);
+    if (targets.length === 0) {
+      console.error(`Session not found: ${sessionId}`);
+      console.error(`
+Your sessions:`);
+      for (const s of sessions.slice(0, 10)) {
+        console.error(`  ${extractSessionId(s.path)}  ${s.firstEvent.slice(0, 10)}  ${s.project}`);
+      }
+      process.exit(1);
+    }
+  } else if (before) {
+    const cutoff = new Date(before);
+    if (isNaN(cutoff.getTime())) {
+      console.error(`Invalid date: ${before}`);
+      process.exit(1);
+    }
+    targets = sessions.filter((s) => new Date(s.lastEvent) < cutoff);
+  } else if (all) {
+    targets = sessions;
+  } else {
+    console.log(`Sessions for "${author}" (${sessions.length} total):
+`);
+    console.log("  Session ID".padEnd(42) + "Date".padEnd(14) + "Events".padEnd(10) + "Project");
+    console.log("  " + "\u2500".repeat(80));
+    for (const s of sessions) {
+      const id = extractSessionId(s.path);
+      const date = s.firstEvent.slice(0, 10);
+      console.log(`  ${id.padEnd(40)}${date.padEnd(14)}${String(s.rowCount).padEnd(10)}${s.project}`);
+    }
+    console.log(`
+To delete, use: --all, --before <date>, or --session-id <id>`);
+    return;
+  }
+  if (targets.length === 0) {
+    console.log("No sessions match the given criteria.");
+    return;
+  }
+  console.log(`Will delete ${targets.length} session(s) for "${author}":
+`);
+  for (const s of targets) {
+    const id = extractSessionId(s.path);
+    console.log(`  ${id}  ${s.firstEvent.slice(0, 10)}  ${s.rowCount} events  ${s.project}`);
+  }
+  console.log();
+  if (!yes) {
+    const ok = await confirm("Proceed with deletion?");
+    if (!ok) {
+      console.log("Aborted.");
+      return;
+    }
+  }
+  const { sessionsDeleted, summariesDeleted } = await deleteSessions(config, targets.map((t) => t.path));
+  console.log(`Deleted ${sessionsDeleted} session(s) and ${summariesDeleted} summary file(s).`);
+}
+
+// dist/src/commands/auth-login.js
+async function main() {
+  const args = process.argv.slice(2);
+  const cmd = args[0] ?? "whoami";
+  const creds = loadCredentials();
+  const apiUrl = creds?.apiUrl ?? "https://api.deeplake.ai";
+  switch (cmd) {
+    case "login": {
+      await login(apiUrl);
+      break;
+    }
+    case "whoami": {
+      if (!creds) {
+        console.log("Not logged in. Run: node auth-login.js login");
+        break;
+      }
+      console.log(`User org: ${creds.orgName ?? creds.orgId}`);
+      console.log(`Workspace: ${creds.workspaceId ?? "default"}`);
+      console.log(`API: ${creds.apiUrl ?? "https://api.deeplake.ai"}`);
+      break;
+    }
+    case "org": {
+      if (!creds) {
+        console.log("Not logged in.");
+        process.exit(1);
+      }
+      const sub = args[1];
+      if (sub === "list") {
+        const orgs = await listOrgs(creds.token, apiUrl);
+        orgs.forEach((o) => console.log(`${o.id}  ${o.name}`));
+      } else if (sub === "switch") {
+        const target = args[2];
+        if (!target) {
+          console.log("Usage: org switch <org-name-or-id>");
+          process.exit(1);
+        }
+        const orgs = await listOrgs(creds.token, apiUrl);
+        const match = orgs.find((o) => o.id === target || o.name.toLowerCase() === target.toLowerCase());
+        if (!match) {
+          console.log(`Org not found: ${target}`);
+          process.exit(1);
+        }
+        await switchOrg(match.id, match.name);
+        console.log(`Switched to org: ${match.name}`);
+      } else {
+        console.log("Usage: org list | org switch <name-or-id>");
+      }
+      break;
+    }
+    case "workspaces": {
+      if (!creds) {
+        console.log("Not logged in.");
+        process.exit(1);
+      }
+      const ws = await listWorkspaces(creds.token, apiUrl, creds.orgId);
+      ws.forEach((w) => console.log(`${w.id}  ${w.name}`));
+      break;
+    }
+    case "workspace": {
+      if (!creds) {
+        console.log("Not logged in.");
+        process.exit(1);
+      }
+      const wsId = args[1];
+      if (!wsId) {
+        console.log("Usage: workspace <id>");
+        process.exit(1);
+      }
+      await switchWorkspace(wsId);
+      console.log(`Switched to workspace: ${wsId}`);
+      break;
+    }
+    case "invite": {
+      if (!creds) {
+        console.log("Not logged in.");
+        process.exit(1);
+      }
+      const email = args[1];
+      const mode = args[2]?.toUpperCase() ?? "WRITE";
+      if (!email) {
+        console.log("Usage: invite <email> [ADMIN|WRITE|READ]");
+        process.exit(1);
+      }
+      await inviteMember(email, mode, creds.token, creds.orgId, apiUrl);
+      console.log(`Invited ${email} with ${mode} access`);
+      break;
+    }
+    case "members": {
+      if (!creds) {
+        console.log("Not logged in.");
+        process.exit(1);
+      }
+      const members = await listMembers(creds.token, creds.orgId, apiUrl);
+      members.forEach((m) => console.log(`${m.role.padEnd(8)} ${m.email ?? m.name}`));
+      break;
+    }
+    case "remove": {
+      if (!creds) {
+        console.log("Not logged in.");
+        process.exit(1);
+      }
+      const userId = args[1];
+      if (!userId) {
+        console.log("Usage: remove <user-id>");
+        process.exit(1);
+      }
+      await removeMember(userId, creds.token, creds.orgId, apiUrl);
+      console.log(`Removed user ${userId}`);
+      break;
+    }
+    case "sessions": {
+      const sub = args[1];
+      if (sub === "prune") {
+        await sessionPrune(args.slice(2));
+      } else {
+        console.log("Usage: sessions prune [--all | --before <date> | --session-id <id>] [--yes]");
+      }
+      break;
+    }
+    case "autoupdate": {
+      if (!creds) {
+        console.log("Not logged in.");
+        process.exit(1);
+      }
+      const val = args[1]?.toLowerCase();
+      if (val === "on" || val === "true") {
+        saveCredentials({ ...creds, autoupdate: true });
+        console.log("Autoupdate enabled. Plugin will update automatically on session start.");
+      } else if (val === "off" || val === "false") {
+        saveCredentials({ ...creds, autoupdate: false });
+        console.log("Autoupdate disabled. You'll see a notice when updates are available.");
+      } else {
+        const current = creds.autoupdate !== false ? "on" : "off";
+        console.log(`Autoupdate is currently: ${current}`);
+        console.log("Usage: autoupdate [on|off]");
+      }
+      break;
+    }
+    case "logout": {
+      if (deleteCredentials()) {
+        console.log("Logged out. Credentials removed.");
+      } else {
+        console.log("Not logged in.");
+      }
+      break;
+    }
+    default:
+      console.log("Commands: login, logout, whoami, org list, org switch, workspaces, workspace, sessions prune, invite, members, remove, autoupdate");
+  }
+}
+main().catch((e) => {
+  console.error(e.message);
+  process.exit(1);
+});