@mingxy/cerebro 1.10.6 → 1.10.8

package/src/index.ts

@@ -3,7 +3,7 @@ import { readFileSync, writeFileSync } from "node:fs";
 import { join, dirname } from "node:path";
 import { tmpdir } from "node:os";
 import { fileURLToPath } from "node:url";
-import { OmemClient } from "./client.js";
+import { CerebroClient } from "./client.js";
 import { autoRecallHook, compactingHook, keywordDetectionHook, sessionIdleHook } from "./hooks.js";
 import { getUserTag, getProjectTag } from "./tags.js";
 import { buildTools } from "./tools.js";
@@ -74,24 +74,24 @@ const OmemPlugin: Plugin = async (input) => {
   let overrides: Record<string, unknown> = {};
   try {
     const ocCfg = JSON.parse(readFileSync(join(directory, "opencode.json"), "utf-8"));
-    const pc = ocCfg?.plugin_config?.["@mingxy/omem"] || ocCfg?.plugin_config?.["@ourmem/opencode"];
+    const pc = ocCfg?.plugin_config?.["@mingxy/cerebro"];
     if (pc) overrides = pc;
   } catch {}
 
   const config = loadPluginConfig(overrides as any);
 
-  const omemClient = new OmemClient(config.apiUrl, config.apiKey, config);
+  const cerebroClient = new CerebroClient(config.connection.apiUrl, config.connection.apiKey, config);
 
   // 启动时检测连接状态
   try {
-    await omemClient.getStats();
+    await cerebroClient.getStats();
     showToast(tui, "🧠 Cerebro · Connected", `Version v${pluginVersion}`, "success", 6000);
-    logInfo(`Connected to ${config.apiUrl}`);
+    logInfo(`Connected to ${config.connection.apiUrl}`);
   } catch (err) {
     const errMsg = err instanceof Error ? err.message : String(err);
     logError(`Connection failed: ${errMsg}`);
-    if (errMsg.includes("[omem]")) {
-      const cleanMsg = errMsg.replace(/^\[omem\]\s*/, "");
+    if (errMsg.includes("[cerebro]")) {
+      const cleanMsg = errMsg.replace(/^\[cerebro\]\s*/, "");
       showToast(
         tui,
         `🧠 Cerebro v${pluginVersion} · Server Error`,
@@ -103,7 +103,7 @@ const OmemPlugin: Plugin = async (input) => {
       showToast(
         tui,
         `🧠 Cerebro v${pluginVersion} · Connection Failed`,
-        `Unable to reach ${config.apiUrl}`,
+        `Unable to reach ${config.connection.apiUrl}`,
         "error",
         8000
       );
@@ -117,7 +117,7 @@ const OmemPlugin: Plugin = async (input) => {
 
   let currentSessionId: string | undefined;
 
-  const recallHook = autoRecallHook(omemClient, containerTags, tui, config);
+  const recallHook = autoRecallHook(cerebroClient, containerTags, tui, config);
 
   return {
     config: async (cfg: any) => {
@@ -131,10 +131,10 @@ const OmemPlugin: Plugin = async (input) => {
       if (input.sessionID) currentSessionId = input.sessionID;
       return recallHook(input, output);
     },
-    "chat.message": keywordDetectionHook(omemClient, containerTags, config.autoCaptureThreshold, tui, config.ingestMode),
-    "experimental.session.compacting": compactingHook(omemClient, containerTags, tui, config.ingestMode, isAutoStoreEnabled, () => currentSessionId, client),
-    tool: buildTools(omemClient, containerTags, { agentId, getSessionId: () => currentSessionId }),
-    event: sessionIdleHook(omemClient, containerTags, tui, client, config.ingestMode, config.autoCaptureThreshold, () => currentSessionId, isAutoStoreEnabled, agentId),
+    "chat.message": keywordDetectionHook(cerebroClient, containerTags, config.ingest.autoCaptureThreshold, tui, config.ingest.ingestMode, config, agentId),
+    "experimental.session.compacting": compactingHook(cerebroClient, containerTags, tui, config.ingest.ingestMode, isAutoStoreEnabled, () => currentSessionId, client, config, agentId),
+    tool: buildTools(cerebroClient, containerTags, { agentId, getSessionId: () => currentSessionId }),
+    event: sessionIdleHook(cerebroClient, containerTags, tui, client, config.ingest.ingestMode, config.ingest.autoCaptureThreshold, () => currentSessionId, isAutoStoreEnabled, agentId, config),
     "shell.env": async (_input: any, output: any) => {
       if (directory) {
         output.env.OMEM_PROJECT_DIR = directory;

package/src/logger.ts

@@ -1,65 +1,65 @@
-import { appendFileSync, mkdirSync, existsSync } from "node:fs";
-import { join } from "node:path";
-import { loadPluginConfig } from "./config.js";
-
-const LEVEL_MAP: Record<string, number> = {
-  DEBUG: 0,
-  INFO: 1,
-  WARN: 2,
-  ERROR: 3,
-};
-
-const cfg = loadPluginConfig();
-const MIN_LEVEL = LEVEL_MAP[cfg.logLevel] ?? LEVEL_MAP.INFO;
-const LOG_DIR = cfg.logDir;
-const LOG_FILE = join(LOG_DIR, "plugin.log");
-const LOG_ENABLED = cfg.logEnabled;
-
-let lastLogTime = Date.now();
-
-function ensureLogDir(): void {
-  if (!existsSync(LOG_DIR)) {
-    try {
-      mkdirSync(LOG_DIR, { recursive: true });
-    } catch {}
-  }
-}
-
-function writeLog(level: string, message: string, fields?: Record<string, unknown>): void {
-  if (!LOG_ENABLED) return;
-  const lvl = LEVEL_MAP[level] ?? 0;
-  if (lvl < MIN_LEVEL) return;
-  ensureLogDir();
-  const now = new Date();
-  const nowMs = now.getTime();
-  const delta = ((nowMs - lastLogTime) / 1000).toFixed(2);
-  lastLogTime = nowMs;
-  const ts = now.toISOString().replace("T", " ").replace(/\.\d+Z$/, "");
-  const parts = [`${level.padEnd(5)} ${ts} +${delta}s service=cerebro`];
-  if (fields) {
-    for (const [k, v] of Object.entries(fields)) {
-      const val = typeof v === "string" ? v : JSON.stringify(v);
-      parts.push(`${k}=${val}`);
-    }
-  }
-  parts.push(message);
-  try {
-    appendFileSync(LOG_FILE, parts.join(" ") + "\n");
-  } catch {}
-}
-
-export function logInfo(message: string, fields?: Record<string, unknown>): void {
-  writeLog("INFO", message, fields);
-}
-
-export function logWarn(message: string, fields?: Record<string, unknown>): void {
-  writeLog("WARN", message, fields);
-}
-
-export function logError(message: string, fields?: Record<string, unknown>): void {
-  writeLog("ERROR", message, fields);
-}
-
-export function logDebug(message: string, fields?: Record<string, unknown>): void {
-  writeLog("DEBUG", message, fields);
-}
+import { appendFileSync, mkdirSync, existsSync } from "node:fs";
+import { join } from "node:path";
+import { loadPluginConfig } from "./config.js";
+
+const LEVEL_MAP: Record<string, number> = {
+  DEBUG: 0,
+  INFO: 1,
+  WARN: 2,
+  ERROR: 3,
+};
+
+const cfg = loadPluginConfig();
+const MIN_LEVEL = LEVEL_MAP[cfg.logging.logLevel] ?? LEVEL_MAP.INFO;
+const LOG_DIR = cfg.logging.logDir;
+const LOG_FILE = join(LOG_DIR, "plugin.log");
+const LOG_ENABLED = cfg.logging.logEnabled;
+
+let lastLogTime = Date.now();
+
+function ensureLogDir(): void {
+  if (!existsSync(LOG_DIR)) {
+    try {
+      mkdirSync(LOG_DIR, { recursive: true });
+    } catch {}
+  }
+}
+
+function writeLog(level: string, message: string, fields?: Record<string, unknown>): void {
+  if (!LOG_ENABLED) return;
+  const lvl = LEVEL_MAP[level] ?? 0;
+  if (lvl < MIN_LEVEL) return;
+  ensureLogDir();
+  const now = new Date();
+  const nowMs = now.getTime();
+  const delta = ((nowMs - lastLogTime) / 1000).toFixed(2);
+  lastLogTime = nowMs;
+  const ts = now.toISOString().replace("T", " ").replace(/\.\d+Z$/, "");
+  const parts = [`${level.padEnd(5)} ${ts} +${delta}s service=cerebro`];
+  if (fields) {
+    for (const [k, v] of Object.entries(fields)) {
+      const val = typeof v === "string" ? v : JSON.stringify(v);
+      parts.push(`${k}=${val}`);
+    }
+  }
+  parts.push(message);
+  try {
+    appendFileSync(LOG_FILE, parts.join(" ") + "\n");
+  } catch {}
+}
+
+export function logInfo(message: string, fields?: Record<string, unknown>): void {
+  writeLog("INFO", message, fields);
+}
+
+export function logWarn(message: string, fields?: Record<string, unknown>): void {
+  writeLog("WARN", message, fields);
+}
+
+export function logError(message: string, fields?: Record<string, unknown>): void {
+  writeLog("ERROR", message, fields);
+}
+
+export function logDebug(message: string, fields?: Record<string, unknown>): void {
+  writeLog("DEBUG", message, fields);
+}

package/src/tools.ts

@@ -1,5 +1,5 @@
 import { tool } from "@opencode-ai/plugin";
-import type { OmemClient } from "./client.js";
+import type { CerebroClient } from "./client.js";
 import { isAutoStoreEnabled, setAutoStoreEnabled } from "./index.js";
 
 function extractMemoryIds(result: unknown): string[] {
@@ -26,7 +26,7 @@ export interface ToolContext {
   getSessionId: () => string | undefined;
 }
 
-export function buildTools(client: OmemClient, containerTags: string[], context: ToolContext) {
+export function buildTools(client: CerebroClient, containerTags: string[], context: ToolContext) {
   return {
     memory_store: tool({
       description:
@@ -95,7 +95,7 @@ export function buildTools(client: OmemClient, containerTags: string[], context:
           args.visibility,
           args.category,
         );
-        if (!result) return JSON.stringify({ ok: false, error: "The omem server may be unavailable." });
+        if (!result) return JSON.stringify({ ok: false, error: "The Cerebro server may be unavailable." });
         return JSON.stringify({ ok: true, id: result.id, tags: result.tags });
       },
     }),

package/omem.example.jsonc

@@ -1,22 +0,0 @@
-{
-  "apiUrl": "https://www.mengxy.cc",
-  "apiKey": "your-tenant-id",
-
-  "requestTimeoutMs": 15000,
-
-  "maxQueryLength": 200,
-  "maxContentChars": 30000,
-  "maxContentLength": 500,
-
-  "autoCaptureThreshold": 5,
-  "ingestMode": "smart",
-
-  "similarityThreshold": 0.4,
-  "maxRecallResults": 10,
-
-  "toastDelayMs": 7000,
-
-  "logEnabled": true,
-  "logLevel": "INFO",
-  "logDir": "~/.config/cerebro"
-}