@aliyunrds/ctxdb 1.0.1-beta.2 → 1.0.1

package/dist/hooks/hermes-post-llm-call.js

@@ -0,0 +1,220 @@
+#!/usr/bin/env node
+import {
+  captureParsedMessages
+} from "../chunk-IMYLU5C2.js";
+import "../chunk-TGVURF54.js";
+import {
+  HttpClient,
+  agentFromArgvWithFallback,
+  debug,
+  load,
+  setDebug
+} from "../chunk-6FZL67GH.js";
+import {
+  shouldSkipHooks
+} from "../chunk-UEKR2Z3S.js";
+
+// src/hooks/hermes-post-llm-call.ts
+import { pathToFileURL } from "url";
+function asRecord(value) {
+  return value && typeof value === "object" && !Array.isArray(value) ? value : {};
+}
+function firstString(...values) {
+  for (const value of values) {
+    if (typeof value === "string") return value;
+  }
+  return "";
+}
+function textContent(content) {
+  if (typeof content === "string") return content.trim();
+  if (!Array.isArray(content)) return "";
+  const chunks = [];
+  for (const rawBlock of content) {
+    if (typeof rawBlock === "string") {
+      if (rawBlock.trim()) chunks.push(rawBlock.trim());
+      continue;
+    }
+    const block = asRecord(rawBlock);
+    const text = firstString(block.text, block.content);
+    if (text.trim()) chunks.push(text.trim());
+  }
+  return chunks.join("\n").trim();
+}
+function currentTurnHistory(history) {
+  const records = history.map(asRecord);
+  let start = -1;
+  for (let i = records.length - 1; i >= 0; i--) {
+    if (records[i].role === "user") {
+      start = i;
+      break;
+    }
+  }
+  return start >= 0 ? records.slice(start) : records;
+}
+function parsedHistoryMessages(history) {
+  const messages = [];
+  for (const item of history) {
+    const role = item.role;
+    if (role !== "user" && role !== "assistant") continue;
+    const content = textContent(item.content);
+    if (!content) continue;
+    messages.push({
+      role,
+      content,
+      index: messages.length,
+      isSummary: false
+    });
+  }
+  return messages;
+}
+function parseToolArguments(value) {
+  if (value && typeof value === "object" && !Array.isArray(value)) {
+    return value;
+  }
+  if (typeof value !== "string") return {};
+  try {
+    const parsed = JSON.parse(value);
+    return parsed && typeof parsed === "object" && !Array.isArray(parsed) ? parsed : {};
+  } catch {
+    return {};
+  }
+}
+function normalizeToolCall(rawCall) {
+  const call = asRecord(rawCall);
+  const fn = asRecord(call.function);
+  const name = firstString(fn.name, call.name);
+  if (!name) return null;
+  const detectionName = name.toLowerCase() === "terminal" ? "bash" : name;
+  const args = parseToolArguments(fn.arguments ?? call.arguments ?? call.input);
+  return {
+    type: "toolCall",
+    name: detectionName,
+    arguments: args
+  };
+}
+function toHermesKbDetectionMessages(history) {
+  const messages = [];
+  for (const item of history) {
+    if (item.role !== "assistant") continue;
+    const blocks = [];
+    if (Array.isArray(item.content)) {
+      blocks.push(...item.content.filter((block) => block && typeof block === "object"));
+    }
+    if (Array.isArray(item.tool_calls)) {
+      for (const rawCall of item.tool_calls) {
+        const block = normalizeToolCall(rawCall);
+        if (block) blocks.push(block);
+      }
+    }
+    if (blocks.length > 0) messages.push({ role: "assistant", content: blocks });
+  }
+  return messages;
+}
+function extractHermesPostLlmCall(event) {
+  const extra = asRecord(event.extra);
+  const rawHistory = Array.isArray(extra.conversation_history) ? extra.conversation_history : Array.isArray(event.conversation_history) ? event.conversation_history : [];
+  const turnHistory = currentTurnHistory(rawHistory);
+  const userMessage = firstString(extra.user_message, event.user_message).trim();
+  const assistantResponse = firstString(
+    extra.assistant_response,
+    event.assistant_response
+  ).trim();
+  const explicitMessages = [];
+  if (userMessage) {
+    explicitMessages.push({
+      role: "user",
+      content: userMessage,
+      index: explicitMessages.length,
+      isSummary: false
+    });
+  }
+  if (assistantResponse) {
+    explicitMessages.push({
+      role: "assistant",
+      content: assistantResponse,
+      index: explicitMessages.length,
+      isSummary: false
+    });
+  }
+  const sessionId = firstString(event.session_id, extra.session_id) || void 0;
+  const historyMessages = parsedHistoryMessages(turnHistory);
+  return {
+    sessionId,
+    messages: userMessage && assistantResponse ? explicitMessages : historyMessages.length > 0 ? historyMessages : explicitMessages,
+    kbDetectionMessages: toHermesKbDetectionMessages(turnHistory)
+  };
+}
+async function readStdinJson() {
+  let raw = "";
+  for await (const chunk of process.stdin) raw += chunk;
+  if (!raw.trim()) return {};
+  try {
+    const parsed = JSON.parse(raw);
+    return parsed && typeof parsed === "object" && !Array.isArray(parsed) ? parsed : {};
+  } catch {
+    return {};
+  }
+}
+async function main() {
+  try {
+    if (shouldSkipHooks()) return 0;
+    const event = await readStdinJson();
+    const { agent, fellBack } = agentFromArgvWithFallback();
+    if (fellBack) {
+      process.stderr.write(
+        `ctxdb hermes post_llm_call: agent unspecified, defaulting to ${agent}
+`
+      );
+    }
+    const input = extractHermesPostLlmCall(event);
+    const cfg = load({ agent });
+    setDebug(cfg.debug);
+    debug("hermes.post_llm_call", "start", {
+      sessionId: input.sessionId,
+      messageCount: input.messages.length,
+      userId: cfg.userId
+    });
+    if (!cfg.apiKey || !cfg.baseUrl) {
+      debug("hermes.post_llm_call", "skip (config incomplete)");
+      return 0;
+    }
+    const client = new HttpClient({
+      baseUrl: cfg.baseUrl,
+      apiKey: cfg.apiKey,
+      timeoutMs: 5e3
+    });
+    const result = await captureParsedMessages(
+      input.messages,
+      cfg,
+      client,
+      agent,
+      input.sessionId,
+      { kbDetectionMessages: input.kbDetectionMessages }
+    );
+    if (result.captured) {
+      debug("hermes.post_llm_call", `ok (${result.messageCount} msgs)`, result);
+      process.stderr.write(`ctxdb hermes capture: ok (${result.messageCount} msgs)
+`);
+    } else {
+      debug("hermes.post_llm_call", `skip (${result.reason})`, result);
+      process.stderr.write(`ctxdb hermes capture: skip (${result.reason})
+`);
+    }
+    return 0;
+  } catch (err) {
+    process.stderr.write(
+      `ctxdb hermes post_llm_call: unexpected error: ${err?.message ?? err}
+`
+    );
+    return 0;
+  }
+}
+if (import.meta.url === pathToFileURL(process.argv[1] ?? "").href) {
+  main().then((code) => {
+    process.exitCode = code;
+  });
+}
+export {
+  extractHermesPostLlmCall,
+  toHermesKbDetectionMessages
+};

package/dist/hooks/hermes-pre-llm-call.js

@@ -0,0 +1,145 @@
+#!/usr/bin/env node
+import {
+  HOOK_TIMEOUT_MS,
+  composeSessionStart
+} from "../chunk-RZONPKY4.js";
+import {
+  composeUserPromptSubmit
+} from "../chunk-FD3AMXVU.js";
+import {
+  fetchKbCatalogBlock
+} from "../chunk-EUQ3OFCQ.js";
+import "../chunk-UH7AJF6F.js";
+import {
+  isCircuitOpen
+} from "../chunk-TGVURF54.js";
+import {
+  HttpClient,
+  agentFromArgvWithFallback,
+  debug,
+  isComplete,
+  load,
+  setDebug
+} from "../chunk-6FZL67GH.js";
+import {
+  shouldSkipHooks
+} from "../chunk-UEKR2Z3S.js";
+
+// src/hooks/hermes-pre-llm-call.ts
+import { pathToFileURL } from "url";
+function asRecord(v) {
+  return v && typeof v === "object" && !Array.isArray(v) ? v : {};
+}
+function firstString(...values) {
+  for (const v of values) {
+    if (typeof v === "string") return v;
+  }
+  return "";
+}
+function extractHermesPreLlmCall(event) {
+  const extra = asRecord(event.extra);
+  return {
+    prompt: firstString(extra.user_message, event.user_message, event.prompt),
+    cwd: firstString(event.cwd, extra.cwd),
+    isFirstTurn: extra.is_first_turn === true || event.is_first_turn === true
+  };
+}
+function formatHermesPreLlmCallStdout(ctx) {
+  return JSON.stringify({ context: ctx }) + "\n";
+}
+async function composeHermesPreLlmCall(cfg, agent, client, input) {
+  const pieces = [];
+  const sessionStartNeeded = input.isFirstTurn && Boolean(input.cwd) && (cfg.warmupRecall || cfg.kbCatalogInjection === "session_start");
+  const sessionStart = sessionStartNeeded ? await composeSessionStart(cfg, agent, client, input.cwd, HOOK_TIMEOUT_MS) : void 0;
+  if (sessionStart?.ctx) pieces.push(sessionStart.ctx);
+  const repeatSessionCatalog = cfg.kbCatalogInjection === "session_start" && !sessionStartNeeded;
+  const recallNeeded = cfg.autoRecall && Boolean(input.prompt.trim());
+  const [kbCatalogBlock, recall] = await Promise.all([
+    repeatSessionCatalog ? fetchKbCatalogBlock(client, agent).catch(() => "") : Promise.resolve(""),
+    recallNeeded ? composeUserPromptSubmit(cfg, agent, client, input.prompt) : Promise.resolve(void 0)
+  ]);
+  if (kbCatalogBlock) pieces.push(kbCatalogBlock);
+  if (recall?.ctx) pieces.push(recall.ctx);
+  return { ctx: pieces.join("\n\n"), recall, sessionStart };
+}
+async function readStdinJson() {
+  let raw = "";
+  for await (const chunk of process.stdin) raw += chunk;
+  if (!raw.trim()) return {};
+  try {
+    const obj = JSON.parse(raw);
+    return obj && typeof obj === "object" && !Array.isArray(obj) ? obj : {};
+  } catch {
+    return {};
+  }
+}
+async function main() {
+  try {
+    if (shouldSkipHooks()) return 0;
+    const event = await readStdinJson();
+    const { agent, fellBack } = agentFromArgvWithFallback();
+    if (fellBack) {
+      process.stderr.write(
+        `ctxdb hermes pre_llm_call: agent unspecified, defaulting to ${agent}
+`
+      );
+    }
+    const input = extractHermesPreLlmCall(event);
+    const cfg = load({ agent });
+    setDebug(cfg.debug);
+    debug("hermes.pre_llm_call", "start", {
+      prompt: input.prompt.slice(0, 200),
+      cwd: input.cwd,
+      isFirstTurn: input.isFirstTurn,
+      userId: cfg.userId
+    });
+    const wantsRecall = cfg.autoRecall && Boolean(input.prompt.trim());
+    const wantsSessionStart = input.isFirstTurn && Boolean(input.cwd) && (cfg.warmupRecall || cfg.kbCatalogInjection === "session_start");
+    const wantsRepeatedSessionCatalog = cfg.kbCatalogInjection === "session_start";
+    if (!isComplete(cfg) || !wantsRecall && !wantsSessionStart && !wantsRepeatedSessionCatalog) {
+      debug("hermes.pre_llm_call", "skip (config incomplete or no enabled injection)");
+      return 0;
+    }
+    if (isCircuitOpen(agent, cfg.baseUrl)) {
+      debug("hermes.pre_llm_call", "skip (circuit open)");
+      process.stderr.write(`ctxdb hermes pre_llm_call: skip (circuit_open: ${cfg.baseUrl})
+`);
+      return 0;
+    }
+    const client = new HttpClient({
+      baseUrl: cfg.baseUrl,
+      apiKey: cfg.apiKey,
+      timeoutMs: HOOK_TIMEOUT_MS
+    });
+    const composed = await composeHermesPreLlmCall(cfg, agent, client, input);
+    if (composed.sessionStart?.warmupTimedOut) {
+      process.stderr.write("ctxdb hermes warmup: timeout\n");
+    }
+    if (composed.recall && !composed.recall.recall.ok && !composed.recall.ctx) {
+      const reason = composed.recall.recall.reason ?? "no_context";
+      debug("hermes.pre_llm_call", `recall no result: ${reason}`, composed.recall.recall);
+      if (reason.startsWith("http_error:")) {
+        process.stderr.write(`ctxdb hermes recall: ${reason}
+`);
+      }
+    }
+    if (!composed.ctx) return 0;
+    debug("hermes.pre_llm_call", `ok, context length=${composed.ctx.length}`);
+    process.stdout.write(formatHermesPreLlmCallStdout(composed.ctx));
+    return 0;
+  } catch (err) {
+    process.stderr.write(`ctxdb hermes pre_llm_call: unexpected error: ${err?.message ?? err}
+`);
+    return 0;
+  }
+}
+if (import.meta.url === pathToFileURL(process.argv[1] ?? "").href) {
+  main().then((code) => {
+    process.exitCode = code;
+  });
+}
+export {
+  composeHermesPreLlmCall,
+  extractHermesPreLlmCall,
+  formatHermesPreLlmCallStdout
+};

package/dist/hooks/session-start.js

@@ -1,183 +1,14 @@
 #!/usr/bin/env node
 import {
-  fetchKbCatalogBlock,
-  recallTurn
-} from "../chunk-CYCD234A.js";
-import "../chunk-6S5RJYBC.js";
-import {
-  isCircuitOpen
-} from "../chunk-UULWJJT4.js";
-import {
-  HttpClient,
-  agentFromArgvWithFallback,
-  debug,
-  isComplete,
-  load,
-  setDebug
-} from "../chunk-TKMIWM6Q.js";
-import {
-  shouldSkipHooks
-} from "../chunk-UEKR2Z3S.js";
-
-// src/hooks/session-start.ts
-import { pathToFileURL } from "url";
-
-// src/lib/warmup-recall.ts
-import { execSync } from "child_process";
-import { basename } from "path";
-function collectGitSignals(cwd) {
-  const result = { branch: "", recentCommits: [] };
-  try {
-    result.branch = execSync("git rev-parse --abbrev-ref HEAD", {
-      cwd,
-      timeout: 500,
-      encoding: "utf-8",
-      stdio: ["ignore", "pipe", "ignore"]
-    }).trim();
-  } catch {
-  }
-  try {
-    const log = execSync("git log --oneline -3 --no-decorate", {
-      cwd,
-      timeout: 500,
-      encoding: "utf-8",
-      stdio: ["ignore", "pipe", "ignore"]
-    }).trim();
-    if (log) {
-      result.recentCommits = log.split("\n").map((l) => {
-        const idx = l.indexOf(" ");
-        return idx > 0 ? l.slice(idx + 1) : l;
-      });
-    }
-  } catch {
-  }
-  return result;
-}
-function buildWarmupQuery(cwd, git) {
-  const project = basename(cwd) || "unknown";
-  const parts = [`project: ${project}`];
-  if (git.branch) {
-    parts.push(`branch: ${git.branch}`);
-  }
-  if (git.recentCommits.length > 0) {
-    parts.push(`recent work: ${git.recentCommits.join("; ")}`);
-  }
-  return parts.join(", ");
-}
-async function warmupRecall(cwd, cfg, client) {
-  if (!cfg.warmupRecall) {
-    return { ok: false, reason: "warmup_recall_disabled", additionalContext: "", memoryCount: 0, knowledgeChunkCount: 0 };
-  }
-  const git = collectGitSignals(cwd);
-  const query = buildWarmupQuery(cwd, git);
-  return recallTurn(query, cfg, client);
-}
-
-// src/hooks/session-start.ts
-var HOOK_TIMEOUT_MS = 5e3;
-function timeout(ms) {
-  return new Promise((resolve) => setTimeout(() => resolve(null), ms));
-}
-async function composeSessionStart(cfg, agent, client, cwd, timeoutMs = HOOK_TIMEOUT_MS) {
-  const kbInjectHere = cfg.kbCatalogInjection === "session_start";
-  const warmupPromise = cfg.warmupRecall ? Promise.race([warmupRecall(cwd, cfg, client), timeout(timeoutMs).then(() => null)]) : Promise.resolve(null);
-  const kbPromise = kbInjectHere ? Promise.race([fetchKbCatalogBlock(client, agent).catch(() => ""), timeout(timeoutMs).then(() => "")]) : Promise.resolve("");
-  const [result, kbBlock] = await Promise.all([warmupPromise, kbPromise]);
-  const warmupTimedOut = cfg.warmupRecall && result === null;
-  const warmupCtx = result && result.ok ? result.additionalContext || "" : "";
-  if (!warmupCtx && result && !result.ok) {
-    debug("warmup", `no result: ${result.reason}`);
-  }
-  let ctx = warmupCtx;
-  if (kbBlock) ctx = ctx ? `${ctx}
-
-${kbBlock}` : kbBlock;
-  return {
-    ctx,
-    memoryCount: result?.ok ? result.memoryCount : 0,
-    kbChunkCount: result?.ok ? result.knowledgeChunkCount : 0,
-    kbCatalogLines: kbBlock ? kbBlock.split("\n").length : 0,
-    warmupTimedOut
-  };
-}
-function formatSessionStartStdout(agent, ctx) {
-  if (agent === "codex") return ctx + "\n";
-  const out = {
-    hookSpecificOutput: {
-      hookEventName: "SessionStart",
-      additionalContext: ctx
-    }
-  };
-  return JSON.stringify(out) + "\n";
-}
-async function readStdinJson() {
-  let raw = "";
-  for await (const chunk of process.stdin) raw += chunk;
-  if (!raw.trim()) return {};
-  try {
-    const obj = JSON.parse(raw);
-    return obj && typeof obj === "object" && !Array.isArray(obj) ? obj : {};
-  } catch {
-    return {};
-  }
-}
-async function main() {
-  try {
-    if (shouldSkipHooks()) return 0;
-    const event = await readStdinJson();
-    const { agent, fellBack } = agentFromArgvWithFallback();
-    if (fellBack) {
-      process.stderr.write(
-        `ctxdb warmup: agent unspecified, defaulting to ${agent}
-`
-      );
-    }
-    const cwd = typeof event.cwd === "string" ? event.cwd : "";
-    if (!cwd) return 0;
-    const cfg = load({ agent });
-    setDebug(cfg.debug);
-    debug("warmup", "start", { cwd, userId: cfg.userId });
-    const kbInjectHere = cfg.kbCatalogInjection === "session_start";
-    if (!isComplete(cfg) || !cfg.warmupRecall && !kbInjectHere) {
-      debug("warmup", "skip (config incomplete or both warmup+kb off)");
-      return 0;
-    }
-    if (isCircuitOpen(agent, cfg.baseUrl)) {
-      debug("warmup", "skip (circuit open)");
-      return 0;
-    }
-    const client = new HttpClient({
-      baseUrl: cfg.baseUrl,
-      apiKey: cfg.apiKey,
-      timeoutMs: HOOK_TIMEOUT_MS
-    });
-    const composed = await composeSessionStart(cfg, agent, client, cwd);
-    if (composed.warmupTimedOut) {
-      process.stderr.write("ctxdb warmup: timeout\n");
-    }
-    if (!composed.ctx) return 0;
-    const { memoryCount, kbChunkCount, kbCatalogLines, ctx } = composed;
-    debug(
-      "warmup",
-      `ok, memories=${memoryCount} kb=${kbChunkCount} kb_catalog=${kbCatalogLines}`
-    );
-    process.stderr.write(
-      `ctxdb warmup: ok (${memoryCount} memories, ${kbChunkCount} kb chunks, kb_catalog=${kbCatalogLines} lines)
-`
-    );
-    process.stdout.write(formatSessionStartStdout(agent, ctx));
-    return 0;
-  } catch (err) {
-    process.stderr.write(`ctxdb warmup: unexpected error: ${err?.message ?? err}
-`);
-    return 0;
-  }
-}
-if (import.meta.url === pathToFileURL(process.argv[1] ?? "").href) {
-  main().then((code) => {
-    process.exitCode = code;
-  });
-}
+  HOOK_TIMEOUT_MS,
+  composeSessionStart,
+  formatSessionStartStdout
+} from "../chunk-RZONPKY4.js";
+import "../chunk-EUQ3OFCQ.js";
+import "../chunk-UH7AJF6F.js";
+import "../chunk-TGVURF54.js";
+import "../chunk-6FZL67GH.js";
+import "../chunk-UEKR2Z3S.js";
 export {
   HOOK_TIMEOUT_MS,
   composeSessionStart,