openhermes 1.12.1 → 2.5.0

package/lib/ohc/notify.mjs

@@ -0,0 +1,150 @@
+function formatTokenCount(tokens) {
+  if (tokens >= 1000) return `${(tokens / 1000).toFixed(1)}K`.replace(".0K", "K")
+  return String(tokens)
+}
+
+function buildProgressBar(prunedCount, visibleCount, width) {
+  width = width || 30
+  const total = prunedCount + visibleCount
+  if (total === 0) return `\u2502${"\u2591".repeat(width)}\u2502  0% active`
+  const activeRatio = visibleCount / total
+  const activeW = Math.round(activeRatio * width)
+  const prunedW = width - activeW
+  const bar = "\u2588".repeat(Math.min(activeW, width)) + "\u2591".repeat(Math.min(prunedW, width))
+  return `\u2502${bar.slice(0, width)}\u2502  ${Math.round(activeRatio * 100)}% active`
+}
+
+function buildMinimal(count, tokensRemoved, savedTotal, blockCount) {
+  const label = "Compression"
+  return `\u25A3 OHC | ~${formatTokenCount(savedTotal)} saved total \u2014 ${label}`
+}
+
+function buildDetailed(count, tokensRemoved, savedTotal, blockCount, prunedCount, visibleCount, summary) {
+  const label = "Compression"
+  let msg = `\u25A3 OHC | ~${formatTokenCount(savedTotal)} saved total`
+  if (prunedCount + visibleCount > 0) {
+    msg += `\n\n${buildProgressBar(prunedCount, visibleCount)}`
+  }
+  msg += `\n\n\u25A3 ${label} #${blockCount}`
+  msg += `\n\u2192 ${count} message${count === 1 ? "" : "s"} removed`
+  if (summary) msg += `\n\u2192 Summary: ${summary}`
+  return msg
+}
+
+function buildStrategyNotification(strategy, count, detail) {
+  return `\u25A3 OHC | ${strategy}: ${count} pruned${detail ? ` (${detail})` : ""}`
+}
+
+export async function sendCompressNotification(client, sessionId, config, count, summary, tokensRemoved, savedTotal, blockCount, prunedCount, visibleCount) {
+  if (count === 0) return false
+
+  const notifType = config.notification ?? "toast"
+  const notifMode = config.notificationMode ?? "minimal"
+
+  if (notifType === "off") return false
+
+  if (notifType === "toast") {
+    const message = notifMode === "minimal"
+      ? buildMinimal(count, tokensRemoved, savedTotal, blockCount)
+      : buildDetailed(count, tokensRemoved, savedTotal, blockCount, prunedCount, visibleCount, summary)
+    try {
+      await client.tui.showToast({
+        body: {
+          title: "OHC: Compression",
+          message,
+          variant: "info",
+          duration: 5000,
+        },
+      })
+    } catch {}
+    return true
+  }
+
+  try {
+    await client.session.prompt({
+      path: { id: sessionId },
+      body: {
+        noReply: true,
+        parts: [{ type: "text", text: buildDetailed(count, tokensRemoved, savedTotal, blockCount, prunedCount, visibleCount, summary), ignored: true }],
+      },
+    })
+  } catch {}
+  return true
+}
+
+export async function sendStrategyNotification(client, sessionId, config, strategy, count, detail) {
+  if (count === 0) return false
+  const notifType = config.notification ?? "toast"
+  if (notifType === "off") return false
+
+  const message = buildStrategyNotification(strategy, count, detail)
+
+  if (notifType === "toast") {
+    try {
+      await client.tui.showToast({
+        body: {
+          title: `OHC: ${strategy}`,
+          message,
+          variant: "info",
+          duration: 3000,
+        },
+      })
+    } catch {}
+    return true
+  }
+
+  try {
+    await client.session.prompt({
+      path: { id: sessionId },
+      body: {
+        noReply: true,
+        parts: [{ type: "text", text: message, ignored: true }],
+      },
+    })
+  } catch {}
+  return true
+}
+
+export async function sendMemoryNotification(client, sessionId, config, action, cls, id, summary) {
+  const notifType = config.notification ?? "toast"
+  const notifMode = config.notificationMode ?? "minimal"
+
+  if (notifType === "off") return false
+
+  const buildHeader = (action, summary) => `[Memory] ${summary}`
+  const buildDetailed = (action, cls, id, summary) => {
+    let msg = `\u25A3 Memory ${action}`
+    if (cls) msg += `\n\u2192 Class: ${cls}`
+    if (id) msg += `\n\u2192 ID: ${id}`
+    if (summary) msg += `\n\u2192 ${summary}`
+    return msg
+  }
+
+  if (notifType === "toast") {
+    const message = notifMode === "minimal"
+      ? buildHeader(action, summary)
+      : buildDetailed(action, cls, id, summary)
+    try {
+      await client.tui.showToast({
+        body: {
+          title: "Memory",
+          message,
+          variant: "info",
+          duration: 4000,
+        },
+      })
+    } catch {}
+    return true
+  }
+
+  try {
+    await client.session.prompt({
+      path: { id: sessionId },
+      body: {
+        noReply: true,
+        parts: [{ type: "text", text: buildDetailed(action, cls, id, summary), ignored: true }],
+      },
+    })
+  } catch {}
+  return true
+}

package/lib/ohc/protected-patterns.mjs

@@ -0,0 +1,54 @@
+const DEFAULT_PROTECTED_TOOLS = new Set([
+  "task", "skill", "todowrite", "todoread",
+  "compress", "batch", "plan_enter", "plan_exit",
+  "write", "edit",
+])
+
+export function isToolNameProtected(toolName, extraProtected = []) {
+  if (DEFAULT_PROTECTED_TOOLS.has(toolName)) return true
+  return extraProtected.includes(toolName)
+}
+
+export function getFilePathsFromParameters(tool, params) {
+  if (!params || typeof params !== "object") return []
+  if (params.filePath) return [params.filePath]
+  if (params.file_path) return [params.file_path]
+  if (params.path) return [params.path]
+  if (params.target) return [params.target]
+  if (params.directory) return [params.directory]
+  if (tool === "glob" && params.pattern) return [params.pattern]
+  return []
+}
+
+export function isFilePathProtected(filePaths, protectedPatterns = []) {
+  if (!protectedPatterns?.length) return false
+  if (!filePaths?.length) return false
+  for (const fp of filePaths) {
+    for (const pattern of protectedPatterns) {
+      if (globMatch(fp, pattern)) return true
+    }
+  }
+  return false
+}
+
+function globMatch(filePath, pattern) {
+  const regexStr = pattern
+    .replace(/[.+^${}()|[\]\\]/g, "\\$&")
+    .replace(/\*/g, ".*")
+    .replace(/\?/g, ".")
+  try {
+    return new RegExp(`^${regexStr}$`, "i").test(filePath)
+  } catch {
+    return filePath.toLowerCase() === pattern.toLowerCase()
+  }
+}
+
+export function getTurnProtectionTags(state) {
+  const config = state.protectedTurns || {}
+  if (!config.enabled || !config.turns) return []
+  const threshold = state.currentTurn - config.turns
+  return [...state.toolIdList].filter(id => {
+    const entry = state.toolParameters.get(id)
+    return entry && entry.turn > threshold
+  })
+}

package/lib/ohc/prune-apply.mjs

@@ -0,0 +1,134 @@
+const PRUNED_OUTPUT_PLACEHOLDER = "[Output removed — information superseded or no longer needed]"
+const PRUNED_ERROR_INPUT_PLACEHOLDER = "[input removed — failed tool call]"
+const PRUNED_QUESTION_INPUT_PLACEHOLDER = "[questions removed — see output for answers]"
+
+export function applyPruneTools(state, messages) {
+  if (!state.prune.tools.size) return 0
+
+  let prunedCount = 0
+
+  for (const msg of messages) {
+    if (!Array.isArray(msg.parts)) continue
+
+    for (const part of msg.parts) {
+      if (part.type !== "tool" || !part.callID) continue
+      if (!state.prune.tools.has(part.callID)) continue
+
+      prunedCount++
+
+      if (part.state?.status === "completed") {
+        if (part.tool === "question") {
+          if (part.state.input?.questions !== undefined) {
+            part.state.input.questions = PRUNED_QUESTION_INPUT_PLACEHOLDER
+          }
+        } else if (part.tool !== "edit" && part.tool !== "write") {
+          part.state = {
+            ...part.state,
+            output: PRUNED_OUTPUT_PLACEHOLDER,
+          }
+        }
+      }
+
+      if (part.state?.status === "error") {
+        const input = part.state.input
+        if (input && typeof input === "object") {
+          for (const key of Object.keys(input)) {
+            if (typeof input[key] === "string") {
+              input[key] = PRUNED_ERROR_INPUT_PLACEHOLDER
+            }
+          }
+        }
+      }
+    }
+  }
+
+  return prunedCount
+}
+
+export function filterCompressedBlocks(state, messages) {
+  const ms = state.prune?.messages
+  if (!ms || !ms.activeBlockIds?.size || !ms.blocksById?.size) return { removed: 0, injected: 0 }
+
+  const coveredMessageIds = new Set()
+  for (const [rawId, entry] of ms.byMessageId) {
+    const hasActive = Array.isArray(entry.activeBlockIds) && entry.activeBlockIds.some(id => ms.activeBlockIds.has(id))
+    if (hasActive) coveredMessageIds.add(rawId)
+  }
+
+  if (!coveredMessageIds.size) return { removed: 0, injected: 0 }
+
+  const activeBlocks = [...ms.activeBlockIds].map(id => ms.blocksById.get(id)).filter(Boolean)
+  const summaryByAnchor = new Map()
+  for (const block of activeBlocks) {
+    if (block.anchorMessageId && !summaryByAnchor.has(block.anchorMessageId)) {
+      summaryByAnchor.set(block.anchorMessageId, block)
+    }
+  }
+
+  let removed = 0
+  let injected = 0
+  const result = []
+  const blocksDeployed = new Set()
+
+  for (const msg of messages) {
+    const mid = msg.info?.id
+
+    if (mid && coveredMessageIds.has(mid)) {
+      if (!blocksDeployed.has(mid)) {
+        blocksDeployed.add(mid)
+        const block = summaryByAnchor.get(mid)
+        if (block) {
+          result.push({
+            parts: [{ type: "text", text: block.summary }],
+            info: { role: "system" },
+          })
+          injected++
+        }
+      }
+      removed++
+      continue
+    }
+
+    result.push(msg)
+  }
+
+  messages.length = 0
+  messages.push(...result)
+
+  return { removed, injected }
+}
+
+export function applyFullToolRemoval(state, messages) {
+  if (!state.prune.tools.size) return 0
+
+  const removed = []
+
+  for (const msg of messages) {
+    if (!Array.isArray(msg.parts)) continue
+
+    const toRemove = []
+    for (const part of msg.parts) {
+      if (part.type !== "tool" || !part.callID) continue
+      if (!state.prune.tools.has(part.callID)) continue
+      if (part.tool !== "edit" && part.tool !== "write") continue
+      toRemove.push(part.callID)
+    }
+
+    if (!toRemove.length) continue
+
+    const before = msg.parts.length
+    msg.parts = msg.parts.filter(p => p.type !== "tool" || !toRemove.includes(p.callID))
+    const after = msg.parts.length
+    if (after === 0 && before > 0) {
+      removed.push(msg.info.id)
+    }
+  }
+
+  if (removed.length) {
+    const result = messages.filter(m => !removed.includes(m.info.id))
+    messages.length = 0
+    messages.push(...result)
+  }
+
+  return removed.length
+}