@cc-soul/openclaw 1.0.0

package/cc-soul/federation.js

@@ -0,0 +1,207 @@
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+import { createHash } from "crypto";
+import { SYNC_CONFIG_PATH, saveJson } from "./persistence.ts";
+import { memoryState, addMemory } from "./memory.ts";
+import { evalMetrics } from "./quality.ts";
+import { getSyncConfig, getInstanceId } from "./sync.ts";
+const PII_PATTERNS = [
+  /\b1[3-9]\d{9}\b/g,
+  // Chinese phone numbers
+  /\b\d{17}[\dXx]\b/g,
+  // Chinese ID card
+  /\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b/g,
+  // Email
+  /\b(?:sk-|api[_-]?key|token|secret|password)[=:]\s*\S+/gi,
+  // API keys/secrets
+  /\b\d{4}[\s-]?\d{4}[\s-]?\d{4}[\s-]?\d{4}\b/g,
+  // Credit card numbers
+  /\b(?:ssh-rsa|ssh-ed25519)\s+\S+/g
+  // SSH keys
+];
+function stripPII(text) {
+  let clean = text;
+  for (const pattern of PII_PATTERNS) {
+    clean = clean.replace(pattern, "[REDACTED]");
+  }
+  return clean;
+}
+__name(stripPII, "stripPII");
+function hasPII(text) {
+  return PII_PATTERNS.some((p) => {
+    p.lastIndex = 0;
+    return p.test(text);
+  });
+}
+__name(hasPII, "hasPII");
+function contentHash(content) {
+  return createHash("sha256").update(content).digest("hex").slice(0, 16);
+}
+__name(contentHash, "contentHash");
+let autoRegisterAttempted = false;
+async function autoRegisterIfNeeded() {
+  const config = getSyncConfig();
+  if (config.hubApiKey || !config.hubUrl || !config.federationEnabled || autoRegisterAttempted) {
+    return Boolean(config.hubApiKey);
+  }
+  autoRegisterAttempted = true;
+  const instanceId = getInstanceId();
+  const instanceName = config.instanceName || "unnamed";
+  try {
+    const resp = await fetch(`${config.hubUrl}/federation/auto-register`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ instanceId, instanceName })
+    });
+    if (!resp.ok) {
+      console.error(`[cc-soul][federation] auto-register failed: ${resp.status}`);
+      return false;
+    }
+    const data = await resp.json();
+    if (data.key) {
+      config.hubApiKey = data.key;
+      saveJson(SYNC_CONFIG_PATH, config);
+      console.log(`[cc-soul][federation] auto-registered! key: ${data.key.slice(0, 10)}...`);
+      return true;
+    }
+  } catch (e) {
+    console.error(`[cc-soul][federation] auto-register error: ${e.message}`);
+  }
+  return false;
+}
+__name(autoRegisterIfNeeded, "autoRegisterIfNeeded");
+let lastUpload = 0;
+const UPLOAD_COOLDOWN = 6 * 36e5;
+async function uploadToHub() {
+  const config = getSyncConfig();
+  if (!config.federationEnabled || !config.hubUrl) return 0;
+  if (!config.hubApiKey) {
+    const ok = await autoRegisterIfNeeded();
+    if (!ok) return 0;
+  }
+  const now = Date.now();
+  if (now - lastUpload < UPLOAD_COOLDOWN) return 0;
+  lastUpload = now;
+  const shareable = memoryState.memories.filter(
+    (m) => (m.scope === "fact" || m.scope === "discovery" || m.scope === "consolidated") && m.visibility === "global" && m.content.length > 10 && !m.content.startsWith("[\u7F51\u7EDC\u77E5\u8BC6") && !hasPII(m.content)
+  ).map((m) => ({
+    content: stripPII(m.content),
+    scope: m.scope,
+    tags: m.tags,
+    sourceInstance: getInstanceId(),
+    sourceQuality: evalMetrics.avgQuality,
+    timestamp: m.ts,
+    contentHash: contentHash(m.content)
+  }));
+  if (shareable.length === 0) return 0;
+  try {
+    const resp = await fetch(`${config.hubUrl}/federation/upload`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "Authorization": `Bearer ${config.hubApiKey}`,
+        "X-Instance-Id": getInstanceId(),
+        "X-Quality-Score": String(evalMetrics.avgQuality)
+      },
+      body: JSON.stringify({ memories: shareable })
+    });
+    if (resp.ok) {
+      const result = await resp.json();
+      console.log(`[cc-soul][federation] uploaded ${shareable.length} memories, accepted: ${result.accepted || "?"}`);
+      return result.accepted || shareable.length;
+    }
+    console.error(`[cc-soul][federation] upload failed: ${resp.status}`);
+  } catch (e) {
+    console.error(`[cc-soul][federation] upload error: ${e.message}`);
+  }
+  return 0;
+}
+__name(uploadToHub, "uploadToHub");
+let lastDownload = 0;
+const DOWNLOAD_COOLDOWN = 6 * 36e5;
+async function downloadFromHub() {
+  const config = getSyncConfig();
+  if (!config.federationEnabled || !config.hubUrl) return 0;
+  if (!config.hubApiKey) {
+    const ok = await autoRegisterIfNeeded();
+    if (!ok) return 0;
+  }
+  const now = Date.now();
+  if (now - lastDownload < DOWNLOAD_COOLDOWN) return 0;
+  lastDownload = now;
+  try {
+    const resp = await fetch(
+      `${config.hubUrl}/federation/download?since=${config.lastSync}&instance=${getInstanceId()}`,
+      {
+        headers: {
+          "Authorization": `Bearer ${config.hubApiKey}`,
+          "X-Instance-Id": getInstanceId()
+        }
+      }
+    );
+    if (!resp.ok) {
+      console.error(`[cc-soul][federation] download failed: ${resp.status}`);
+      return 0;
+    }
+    const data = await resp.json();
+    if (!data.memories || data.memories.length === 0) return 0;
+    const existingHashes = new Set(
+      memoryState.memories.map((m) => contentHash(m.content))
+    );
+    let imported = 0;
+    for (const fm of data.memories) {
+      if (fm.sourceInstance === getInstanceId()) continue;
+      if (existingHashes.has(fm.contentHash)) continue;
+      if (fm.sourceQuality < 5) continue;
+      if (hasPII(fm.content)) continue;
+      const agedays = (Date.now() - fm.timestamp) / 864e5;
+      const freshness = Math.max(0.3, 1 - agedays / 365);
+      const trustScore = fm.sourceQuality / 10 * freshness;
+      if (agedays > 180 && fm.sourceQuality < 7) continue;
+      const trustLabel = trustScore > 0.7 ? "\u9AD8\u53EF\u4FE1" : trustScore > 0.4 ? "\u5F85\u9A8C\u8BC1" : "\u4F4E\u53EF\u4FE1";
+      const content = `[\u7F51\u7EDC\u77E5\u8BC6|${trustLabel}] ${fm.content}`;
+      addMemory(content, fm.scope, void 0, "global");
+      existingHashes.add(fm.contentHash);
+      imported++;
+    }
+    console.log(`[cc-soul][federation] downloaded ${data.memories.length}, imported ${imported} (dedup + quality filter)`);
+    return imported;
+  } catch (e) {
+    console.error(`[cc-soul][federation] download error: ${e.message}`);
+  }
+  return 0;
+}
+__name(downloadFromHub, "downloadFromHub");
+function autoFederate() {
+  const config = getSyncConfig();
+  if (!config.federationEnabled) return;
+  uploadToHub().catch((e) => console.error(`[cc-soul][federation] auto-upload: ${e}`));
+  downloadFromHub().catch((e) => console.error(`[cc-soul][federation] auto-download: ${e}`));
+}
+__name(autoFederate, "autoFederate");
+async function reportBadKnowledge(memoryContent) {
+  const config = getSyncConfig();
+  if (!config.federationEnabled || !config.hubUrl || !config.hubApiKey) return;
+  const hash = contentHash(memoryContent.replace(/^\[网络知识[|｜][^\]]*\]\s*/, ""));
+  try {
+    await fetch(`${config.hubUrl}/federation/report`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "Authorization": `Bearer ${config.hubApiKey}`
+      },
+      body: JSON.stringify({ contentHash: hash })
+    });
+    console.log(`[cc-soul][federation] reported bad knowledge: ${memoryContent.slice(0, 60)}`);
+  } catch (e) {
+    console.error(`[cc-soul][federation] report failed: ${e.message}`);
+  }
+}
+__name(reportBadKnowledge, "reportBadKnowledge");
+export {
+  autoFederate,
+  autoRegisterIfNeeded,
+  downloadFromHub,
+  reportBadKnowledge,
+  uploadToHub
+};

package/cc-soul/fingerprint.js

@@ -0,0 +1 @@
+import{resolve}from"path";import{DATA_DIR,loadJson,debouncedSave}from"./persistence.ts";const FINGERPRINT_PATH=resolve(DATA_DIR,"fingerprint.json");let fingerprint=loadJson(FINGERPRINT_PATH,{i:0,t:0,p:0,o:0,u:0,F:0,h:0,I:0});function saveFingerprint(){debouncedSave(FINGERPRINT_PATH,fingerprint)}function updateFingerprint(n){if(!n||n.length<10)return;const r=fingerprint.h<50?.2:.05,i=n.length,e=n.split(/[。！？!?.\n]+/).filter(n=>n.trim().length>0),t=e.length>0?i/e.length:i,g=/[？?]/.test(n)?1:0,a=n.includes("```")?1:0,f=/[\u{1F300}-\u{1FAFF}]|[😀-🙏]/u.test(n)?1:0,p=(n.match(/我/g)||[]).length/Math.max(1,n.length/100);fingerprint.i=fingerprint.i*(1-r)+i*r,fingerprint.t=fingerprint.t*(1-r)+t*r,fingerprint.p=fingerprint.p*(1-r)+g*r,fingerprint.o=fingerprint.o*(1-r)+a*r,fingerprint.u=fingerprint.u*(1-r)+f*r,fingerprint.F=fingerprint.F*(1-r)+p*r,fingerprint.h++,fingerprint.I=Date.now(),saveFingerprint()}function checkPersonaConsistency(n){if(fingerprint.h<30)return"";const r=[];n.length>3*fingerprint.i&&r.push("回复异常长"),n.length<.15*fingerprint.i&&n.length>5&&r.push("回复异常短"),/作为一个?AI|作为语言模型|作为人工智能|I am an AI|as an AI/i.test(n)&&r.push("人设泄露：提到了AI身份");return(n.match(/可能|也许|或许|不太确定|我不确定|大概/g)||[]).length>5&&r.push("过度犹豫"),0===r.length?"":`[风格偏离] ${r.join("；")}`}function getFingerprintSummary(){return fingerprint.h<20?"":`[回复风格基线] 平均长度${Math.round(fingerprint.i)}字 | 提问率${(100*fingerprint.p).toFixed(0)}% | 代码率${(100*fingerprint.o).toFixed(0)}% | 样本${fingerprint.h}条`}function loadFingerprint(){const n=loadJson(FINGERPRINT_PATH,fingerprint);Object.assign(fingerprint,n)}let cachedDriftWarning="";function getCachedDriftWarning(){const n=cachedDriftWarning;return cachedDriftWarning="",n}function setCachedDriftWarning(n){cachedDriftWarning=n}export{checkPersonaConsistency,getCachedDriftWarning,getFingerprintSummary,loadFingerprint,setCachedDriftWarning,updateFingerprint};

package/cc-soul/flow.js

@@ -0,0 +1,199 @@
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+import { spawnCLI } from "./cli.ts";
+import { memoryState, addMemory } from "./memory.ts";
+const lastResolvedFlows = /* @__PURE__ */ new Map();
+function createEmptyFlow() {
+  return {
+    topic: "",
+    turnCount: 0,
+    frustration: 0,
+    resolved: false,
+    depth: "shallow",
+    lastMsgLengths: [],
+    topicKeywords: [],
+    lastUpdate: Date.now()
+  };
+}
+__name(createEmptyFlow, "createEmptyFlow");
+const flows = /* @__PURE__ */ new Map();
+const MAX_FLOWS = 50;
+let onSessionResolved = null;
+function setOnSessionResolved(cb) {
+  onSessionResolved = cb;
+}
+__name(setOnSessionResolved, "setOnSessionResolved");
+function updateFlow(userMsg, botResponse, flowKey) {
+  let flow = flows.get(flowKey) || createEmptyFlow();
+  const msgWords = (userMsg.match(/[\u4e00-\u9fff]{2,}|[a-z]{3,}/gi) || []).map((w) => w.toLowerCase());
+  const overlap = flow.topicKeywords.filter((w) => msgWords.includes(w)).length;
+  const isSameTopic = overlap >= 2 || flow.turnCount > 0 && overlap >= 1 && userMsg.length < 50;
+  if (isSameTopic) {
+    flow.turnCount++;
+    for (const w of msgWords.slice(0, 5)) {
+      if (!flow.topicKeywords.includes(w)) flow.topicKeywords.push(w);
+    }
+    if (flow.topicKeywords.length > 15) flow.topicKeywords = flow.topicKeywords.slice(-10);
+  } else {
+    flow = {
+      topic: userMsg.slice(0, 50),
+      turnCount: 1,
+      frustration: 0,
+      resolved: false,
+      depth: "shallow",
+      lastMsgLengths: [],
+      topicKeywords: msgWords.slice(0, 8),
+      lastUpdate: Date.now()
+    };
+  }
+  flow.lastMsgLengths.push(userMsg.length);
+  if (flow.lastMsgLengths.length > 5) flow.lastMsgLengths.shift();
+  if (flow.lastMsgLengths.length >= 3) {
+    const lengths = flow.lastMsgLengths;
+    const trend = lengths[lengths.length - 1] - lengths[0];
+    if (trend < -50) flow.frustration = Math.min(1, flow.frustration + 0.2);
+    if (userMsg.length < 10 && flow.turnCount > 3) flow.frustration = Math.min(1, flow.frustration + 0.15);
+  }
+  if (["\u7B97\u4E86", "\u4E0D\u5BF9", "\u8FD8\u662F\u4E0D\u884C", "\u600E\u4E48\u53C8", "\u8BF4\u4E86\u591A\u5C11\u904D"].some((w) => userMsg.includes(w))) {
+    flow.frustration = Math.min(1, flow.frustration + 0.3);
+  }
+  if (["\u641E\u5B9A", "\u53EF\u4EE5\u4E86", "\u597D\u4E86", "\u89E3\u51B3\u4E86", "\u8C22\u8C22", "thanks", "\u6210\u529F\u4E86"].some((w) => userMsg.toLowerCase().includes(w))) {
+    flow.resolved = true;
+    flow.frustration = Math.max(0, flow.frustration - 0.3);
+    if (typeof onSessionResolved === "function") onSessionResolved();
+    if (!lastResolvedFlows.has(flowKey)) {
+      lastResolvedFlows.set(flowKey, {
+        resolvedAt: Date.now(),
+        summarized: false,
+        topic: flow.topic,
+        turnCount: flow.turnCount
+      });
+    }
+  }
+  if (flow.turnCount <= 2) flow.depth = "shallow";
+  else if (flow.turnCount <= 6 && flow.frustration < 0.5) flow.depth = "deep";
+  else if (flow.turnCount > 6 || flow.frustration >= 0.5) flow.depth = "stuck";
+  flow.lastUpdate = Date.now();
+  flows.set(flowKey, flow);
+  if (flows.size > MAX_FLOWS) {
+    const oldest = [...flows.entries()].sort((a, b) => a[1].lastUpdate - b[1].lastUpdate)[0];
+    if (oldest) flows.delete(oldest[0]);
+  }
+  return flow;
+}
+__name(updateFlow, "updateFlow");
+function getFlowHints(flowKey) {
+  const flow = flows.get(flowKey);
+  if (!flow) return [];
+  const hints = [];
+  if (flow.depth === "stuck") {
+    hints.push(`\u5DF2\u7ECF\u8BA8\u8BBA${flow.turnCount}\u8F6E\u4E86\uFF0C\u53EF\u80FD\u9677\u5165\u50F5\u5C40\u3002\u8BD5\u8BD5\u6362\u4E2A\u601D\u8DEF\u6216\u76F4\u63A5\u7ED9\u6700\u7EC8\u65B9\u6848`);
+  }
+  if (flow.frustration >= 0.6) {
+    hints.push("\u7528\u6237\u53EF\u80FD\u8D8A\u6765\u8D8A\u4E0D\u8010\u70E6\u4E86\uFF0C\u7B80\u5316\u56DE\u7B54\uFF0C\u76F4\u63A5\u7ED9\u65B9\u6848");
+  }
+  if (flow.resolved) {
+    hints.push("\u95EE\u9898\u4F3C\u4E4E\u5DF2\u89E3\u51B3\uFF0C\u81EA\u7136\u6536\u5C3E\u5373\u53EF");
+  }
+  if (flow.turnCount >= 4 && !flow.resolved && flow.frustration < 0.3) {
+    hints.push(`\u8BA8\u8BBA\u5DF2${flow.turnCount}\u8F6E\uFF0C\u7528\u6237\u5F88\u6709\u8010\u5FC3\uFF0C\u53EF\u4EE5\u7EE7\u7EED\u6DF1\u5165`);
+  }
+  return hints;
+}
+__name(getFlowHints, "getFlowHints");
+function getFlowContext(flowKey) {
+  const flow = flows.get(flowKey);
+  if (!flow || flow.turnCount <= 1) return "";
+  return `[\u5BF9\u8BDD\u6D41] \u5F53\u524D\u8BDD\u9898\u5DF2${flow.turnCount}\u8F6E | \u6DF1\u5EA6:${flow.depth} | \u7528\u6237\u8010\u5FC3:${(1 - flow.frustration).toFixed(1)} | ${flow.resolved ? "\u5DF2\u89E3\u51B3" : "\u8FDB\u884C\u4E2D"}`;
+}
+__name(getFlowContext, "getFlowContext");
+function getCurrentFlowDepth() {
+  let worst = "shallow";
+  for (const flow of flows.values()) {
+    if (flow.depth === "stuck") return "stuck";
+    if (flow.depth === "deep") worst = "deep";
+  }
+  return worst;
+}
+__name(getCurrentFlowDepth, "getCurrentFlowDepth");
+function resetFlow(flowKey) {
+  if (flowKey) {
+    flows.delete(flowKey);
+  } else {
+    flows.clear();
+  }
+}
+__name(resetFlow, "resetFlow");
+function checkSessionEnd(flowKey) {
+  const resolved = lastResolvedFlows.get(flowKey);
+  if (!resolved) return null;
+  if (resolved.summarized) return null;
+  if (Date.now() - resolved.resolvedAt > 5 * 60 * 1e3) {
+    resolved.summarized = true;
+    return { ended: true, topic: resolved.topic, turnCount: resolved.turnCount };
+  }
+  return null;
+}
+__name(checkSessionEnd, "checkSessionEnd");
+function checkAllSessionEnds() {
+  const ended = [];
+  for (const [key] of lastResolvedFlows) {
+    const result = checkSessionEnd(key);
+    if (result) {
+      ended.push({ flowKey: key, topic: result.topic, turnCount: result.turnCount });
+    }
+  }
+  return ended;
+}
+__name(checkAllSessionEnds, "checkAllSessionEnds");
+function generateSessionSummary(topic, turnCount, _flowKey) {
+  const chatHistory = memoryState.chatHistory;
+  const recentHistory = chatHistory.slice(-(turnCount * 2));
+  const historyText = recentHistory.map(
+    (t) => `\u7528\u6237: ${t.user.slice(0, 100)}
+\u52A9\u624B: ${t.assistant.slice(0, 100)}`
+  ).join("\n\n");
+  spawnCLI(
+    `\u4EE5\u4E0B\u662F\u4E00\u6BB5${turnCount}\u8F6E\u7684\u5BF9\u8BDD\uFF08\u8BDD\u9898: ${topic}\uFF09\u3002\u8BF7\u603B\u7ED3\uFF1A
+1. \u804A\u4E86\u4EC0\u4E48
+2. \u7528\u6237\u6EE1\u610F\u5417
+3. \u6709\u6CA1\u6709\u9057\u7559\u95EE\u9898
+4. \u503C\u5F97\u8BB0\u4F4F\u7684\u4E8B\u5B9E/\u504F\u597D
+
+${historyText.slice(0, 2e3)}
+
+\u683C\u5F0F: {"summary":"\u4E00\u6BB5\u8BDD\u603B\u7ED3","facts":["\u503C\u5F97\u8BB0\u4F4F\u7684\u4E8B\u5B9E"],"satisfied":true/false,"pending":"\u9057\u7559\u95EE\u9898\u6216null"}`,
+    (output) => {
+      try {
+        const m = output.match(/\{[\s\S]*?\}/);
+        if (m) {
+          const result = JSON.parse(m[0]);
+          if (result.summary) {
+            addMemory(`[\u4F1A\u8BDD\u603B\u7ED3] ${topic}: ${result.summary}`, "consolidated");
+          }
+          for (const fact of result.facts || []) {
+            addMemory(fact, "fact");
+          }
+          if (result.pending) {
+            addMemory(`[\u9057\u7559\u95EE\u9898] ${result.pending}`, "task");
+          }
+          console.log(`[cc-soul][session] summarized: ${topic} (${turnCount} turns, satisfied=${result.satisfied})`);
+        }
+      } catch (e) {
+        console.error(`[cc-soul][session] summary parse error: ${e.message}`);
+      }
+    }
+  );
+}
+__name(generateSessionSummary, "generateSessionSummary");
+export {
+  checkAllSessionEnds,
+  checkSessionEnd,
+  generateSessionSummary,
+  getCurrentFlowDepth,
+  getFlowContext,
+  getFlowHints,
+  resetFlow,
+  setOnSessionResolved,
+  updateFlow
+};

package/cc-soul/graph.js

@@ -0,0 +1,85 @@
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+import { GRAPH_PATH, loadJson, debouncedSave } from "./persistence.ts";
+const graphState = {
+  entities: [],
+  relations: []
+};
+function loadGraph() {
+  const data = loadJson(GRAPH_PATH, { entities: [], relations: [] });
+  graphState.entities.length = 0;
+  graphState.entities.push(...data.entities || []);
+  graphState.relations.length = 0;
+  graphState.relations.push(...data.relations || []);
+}
+__name(loadGraph, "loadGraph");
+function saveGraph() {
+  debouncedSave(GRAPH_PATH, { entities: graphState.entities, relations: graphState.relations });
+}
+__name(saveGraph, "saveGraph");
+function addEntity(name, type, attrs = []) {
+  if (!name || name.length < 2) return;
+  const existing = graphState.entities.find((e) => e.name === name);
+  if (existing) {
+    existing.mentions++;
+    for (const a of attrs) {
+      if (!existing.attrs.includes(a)) existing.attrs.push(a);
+    }
+  } else {
+    graphState.entities.push({ name, type, attrs, firstSeen: Date.now(), mentions: 1 });
+  }
+  if (graphState.entities.length > 500) {
+    graphState.entities.sort((a, b) => b.mentions - a.mentions);
+    const trimmed = graphState.entities.slice(0, 400);
+    graphState.entities.length = 0;
+    graphState.entities.push(...trimmed);
+  }
+  saveGraph();
+}
+__name(addEntity, "addEntity");
+function addRelation(source, target, type) {
+  if (!source || !target) return;
+  const exists = graphState.relations.some((r) => r.source === source && r.target === target && r.type === type);
+  if (exists) return;
+  graphState.relations.push({ source, target, type, ts: Date.now() });
+  if (graphState.relations.length > 1e3) {
+    const trimmed = graphState.relations.slice(-800);
+    graphState.relations.length = 0;
+    graphState.relations.push(...trimmed);
+  }
+  saveGraph();
+}
+__name(addRelation, "addRelation");
+function addEntitiesFromAnalysis(entities) {
+  for (const e of entities) {
+    if (e.name && e.name.length >= 2) {
+      addEntity(e.name, e.type);
+      if (e.relation) addRelation(e.name, "\u7528\u6237", e.relation.slice(0, 30));
+    }
+  }
+}
+__name(addEntitiesFromAnalysis, "addEntitiesFromAnalysis");
+function queryEntityContext(msg) {
+  const results = [];
+  for (const entity of graphState.entities) {
+    if (msg.includes(entity.name)) {
+      const rels = graphState.relations.filter((r) => r.source === entity.name || r.target === entity.name);
+      if (rels.length > 0) {
+        const relStr = rels.map((r) => `${r.source} ${r.type} ${r.target}`).join(", ");
+        results.push(`[${entity.type}] ${entity.name}: ${relStr}`);
+      } else if (entity.attrs.length > 0) {
+        results.push(`[${entity.type}] ${entity.name}: ${entity.attrs.join(", ")}`);
+      }
+    }
+  }
+  return results.slice(0, 3);
+}
+__name(queryEntityContext, "queryEntityContext");
+export {
+  addEntitiesFromAnalysis,
+  addEntity,
+  addRelation,
+  graphState,
+  loadGraph,
+  queryEntityContext
+};