@mingxy/cerebro 1.14.5 → 1.15.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/config.d.ts +1 -0
- package/dist/config.d.ts.map +1 -1
- package/dist/config.js +2 -0
- package/dist/config.js.map +1 -1
- package/dist/hooks.d.ts +10 -1
- package/dist/hooks.d.ts.map +1 -1
- package/dist/hooks.js +403 -19
- package/dist/hooks.js.map +1 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +26 -8
- package/dist/index.js.map +1 -1
- package/package.json +1 -1
- package/src/config.ts +3 -0
- package/src/hooks.ts +442 -17
- package/src/index.ts +29 -10
package/dist/config.d.ts
CHANGED
|
@@ -41,6 +41,7 @@ export interface OmemPluginConfig {
|
|
|
41
41
|
};
|
|
42
42
|
agentMemoryPolicy?: Record<string, "none" | "readonly" | "readwrite">;
|
|
43
43
|
defaultPolicy?: "none" | "readonly" | "readwrite";
|
|
44
|
+
injectionStrategy?: "parts" | "system";
|
|
44
45
|
}
|
|
45
46
|
declare const DEFAULTS: OmemPluginConfig;
|
|
46
47
|
export declare function loadPluginConfig(overrides?: Partial<OmemPluginConfig>): OmemPluginConfig;
|
package/dist/config.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAMA,MAAM,WAAW,gBAAgB;IAC/B,UAAU,EAAE;QACV,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;QACf,gBAAgB,EAAE,MAAM,CAAC;KAC1B,CAAC;IACF,OAAO,EAAE;QACP,cAAc,EAAE,MAAM,CAAC;QACvB,eAAe,EAAE,MAAM,CAAC;QACxB,gBAAgB,EAAE,MAAM,CAAC;KAC1B,CAAC;IACF,MAAM,EAAE;QACN,oBAAoB,EAAE,MAAM,CAAC;QAC7B,UAAU,EAAE,OAAO,GAAG,KAAK,CAAC;KAC7B,CAAC;IACF,MAAM,EAAE;QACN,mBAAmB,EAAE,MAAM,CAAC;QAC5B,gBAAgB,EAAE,MAAM,CAAC;QACzB,eAAe,EAAE,MAAM,CAAC;QACxB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,mBAAmB,EAAE,MAAM,CAAC;QAC5B,gBAAgB,EAAE,MAAM,CAAC;QACzB,gBAAgB,EAAE,MAAM,CAAC;QACzB,UAAU,EAAE,MAAM,CAAC;QACnB,cAAc,EAAE,QAAQ,GAAG,UAAU,GAAG,OAAO,CAAC;QAChD,iBAAiB,EAAE,MAAM,CAAC;KAC3B,CAAC;IACF,OAAO,EAAE;QACP,UAAU,EAAE,OAAO,CAAC;QACpB,QAAQ,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;QAC9C,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,EAAE,EAAE;QACF,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,WAAW,CAAC,EAAE;QACZ,OAAO,EAAE,OAAO,CAAC;QACjB,KAAK,EAAE,MAAM,EAAE,CAAC;QAChB,YAAY,EAAE,MAAM,EAAE,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,iBAAiB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,UAAU,GAAG,WAAW,CAAC,CAAC;IACtE,aAAa,CAAC,EAAE,MAAM,GAAG,UAAU,GAAG,WAAW,CAAC;
|
|
1
|
+
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAMA,MAAM,WAAW,gBAAgB;IAC/B,UAAU,EAAE;QACV,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;QACf,gBAAgB,EAAE,MAAM,CAAC;KAC1B,CAAC;IACF,OAAO,EAAE;QACP,cAAc,EAAE,MAAM,CAAC;QACvB,eAAe,EAAE,MAAM,CAAC;QACxB,gBAAgB,EAAE,MAAM,CAAC;KAC1B,CAAC;IACF,MAAM,EAAE;QACN,oBAAoB,EAAE,MAAM,CAAC;QAC7B,UAAU,EAAE,OAAO,GAAG,KAAK,CAAC;KAC7B,CAAC;IACF,MAAM,EAAE;QACN,mBAAmB,EAAE,MAAM,CAAC;QAC5B,gBAAgB,EAAE,MAAM,CAAC;QACzB,eAAe,EAAE,MAAM,CAAC;QACxB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,mBAAmB,EAAE,MAAM,CAAC;QAC5B,gBAAgB,EAAE,MAAM,CAAC;QACzB,gBAAgB,EAAE,MAAM,CAAC;QACzB,UAAU,EAAE,MAAM,CAAC;QACnB,cAAc,EAAE,QAAQ,GAAG,UAAU,GAAG,OAAO,CAAC;QAChD,iBAAiB,EAAE,MAAM,CAAC;KAC3B,CAAC;IACF,OAAO,EAAE;QACP,UAAU,EAAE,OAAO,CAAC;QACpB,QAAQ,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;QAC9C,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,EAAE,EAAE;QACF,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,WAAW,CAAC,EAAE;QACZ,OAAO,EAAE,OAAO,CAAC;QACjB,KAAK,EAAE,MAAM,EAAE,CAAC;QAChB,YAAY,EAAE,MAAM,EAAE,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,iBAAiB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,UAAU,GAAG,WAAW,CAAC,CAAC;IACtE,aAAa,CAAC,EAAE,MAAM,GAAG,UAAU,GAAG,WAAW,CAAC;IAClD,iBAAiB,CAAC,EAAE,OAAO,GAAG,QAAQ,CAAC;CACxC;AAID,QAAA,MAAM,QAAQ,EAAE,gBA0Cf,CAAC;AAgIF,wBAAgB,gBAAgB,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,GAAG,gBAAgB,CA+CxF;AAID,MAAM,MAAM,WAAW,GAAG,MAAM,GAAG,UAAU,GAAG,WAAW,CAAC;AAE5D,wBAAgB,kBAAkB,CAChC,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,OAAO,CAAC,gBAAgB,CAAC,GAChC,WAAW,CAeb;AAED,OAAO,EAAE,QAAQ,EAAE,CAAC"}
|
package/dist/config.js
CHANGED
|
@@ -43,6 +43,7 @@ const DEFAULTS = {
|
|
|
43
43
|
excludeTools: ["memory_store", "memory_search", "memory_get", "memory_toggle", "memory_ingest"],
|
|
44
44
|
maxToolNames: 3,
|
|
45
45
|
},
|
|
46
|
+
injectionStrategy: "parts",
|
|
46
47
|
};
|
|
47
48
|
function isFlatConfig(cfg) {
|
|
48
49
|
return "apiUrl" in cfg && !("connection" in cfg);
|
|
@@ -100,6 +101,7 @@ function deepMerge(base, overrides) {
|
|
|
100
101
|
result.agentMemoryPolicy = overrides.agentMemoryPolicy;
|
|
101
102
|
if (overrides.defaultPolicy)
|
|
102
103
|
result.defaultPolicy = overrides.defaultPolicy;
|
|
104
|
+
result.injectionStrategy = overrides.injectionStrategy ?? base.injectionStrategy;
|
|
103
105
|
return result;
|
|
104
106
|
}
|
|
105
107
|
// ── Load config ──────────────────────────────────────────────────────
|
package/dist/config.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAClE,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;
|
|
1
|
+
{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAClE,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAkDjC,wEAAwE;AAExE,MAAM,QAAQ,GAAqB;IACjC,UAAU,EAAE;QACV,MAAM,EAAE,uBAAuB;QAC/B,MAAM,EAAE,EAAE;QACV,gBAAgB,EAAE,KAAK;KACxB;IACD,OAAO,EAAE;QACP,cAAc,EAAE,GAAG;QACnB,eAAe,EAAE,KAAK;QACtB,gBAAgB,EAAE,GAAG;KACtB;IACD,MAAM,EAAE;QACN,oBAAoB,EAAE,CAAC;QACvB,UAAU,EAAE,OAAO;KACpB;IACD,MAAM,EAAE;QACN,mBAAmB,EAAE,GAAG;QACxB,gBAAgB,EAAE,EAAE;QACpB,eAAe,EAAE,CAAC;QAClB,iBAAiB,EAAE,CAAC;QACpB,mBAAmB,EAAE,IAAI;QACzB,gBAAgB,EAAE,GAAG;QACrB,gBAAgB,EAAE,CAAC;QACnB,UAAU,EAAE,EAAE;QACd,cAAc,EAAE,UAAU;QAC1B,iBAAiB,EAAE,GAAG;KACvB;IACD,OAAO,EAAE;QACP,UAAU,EAAE,IAAI;QAChB,QAAQ,EAAE,MAAM;QAChB,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,SAAS,CAAC;KAC9C;IACD,EAAE,EAAE;QACF,YAAY,EAAE,IAAI;KACnB;IACD,WAAW,EAAE;QACX,OAAO,EAAE,IAAI;QACb,KAAK,EAAE,CAAC,GAAG,CAAC;QACZ,YAAY,EAAE,CAAC,cAAc,EAAE,eAAe,EAAE,YAAY,EAAE,eAAe,EAAE,eAAe,CAAC;QAC/F,YAAY,EAAE,CAAC;KAChB;IACD,iBAAiB,EAAE,OAAgB;CACpC,CAAC;AAwBF,SAAS,YAAY,CAAC,GAA4B;IAChD,OAAO,QAAQ,IAAI,GAAG,IAAI,CAAC,CAAC,YAAY,IAAI,GAAG,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAgB;IAC3C,OAAO;QACL,UAAU,EAAE;YACV,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM;YACjD,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM;YACjD,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,IAAI,QAAQ,CAAC,UAAU,CAAC,gBAAgB;SAChF;QACD,OAAO,EAAE;YACP,cAAc,EAAE,IAAI,CAAC,cAAc,IAAI,QAAQ,CAAC,OAAO,CAAC,cAAc;YACtE,eAAe,EAAE,IAAI,CAAC,eAAe,IAAI,QAAQ,CAAC,OAAO,CAAC,eAAe;YACzE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,IAAI,QAAQ,CAAC,OAAO,CAAC,gBAAgB;SAC7E;QACD,MAAM,EAAE;YACN,oBAAoB,EAAE,IAAI,CAAC,oBAAoB,IAAI,QAAQ,CAAC,MAAM,CAAC,oBAAoB;YACvF,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,QAAQ,CAAC,MAAM,CAAC,UAAU;SAC1D;QACD,MAAM,EAAE;YACN,mBAAmB,EAAE,IAAI,CAAC,mBAAmB,IAAI,QAAQ,CAAC,MAAM,CAAC,mBAAmB;YACpF,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,IAAI,QAAQ,CAAC,MAAM,CAAC,gBAAgB;YAC3E,eAAe,EAAE,QAAQ,CAAC,MAAM,CAAC,eAAe;YAChD,iBAAiB,EAAE,QAAQ,CAAC,MAAM,CAAC,iBAAiB;YACpD,mBAAmB,EAAE,QAAQ,CAAC,MAAM,CAAC,mBAAmB;YACxD,gBAAgB,EAAE,QAAQ,CAAC,MAAM,CAAC,gBAAgB;YAClD,gBAAgB,EAAE,QAAQ,CAAC,MAAM,CAAC,gBAAgB;YAClD,UAAU,EAAE,QAAQ,CAAC,MAAM,CAAC,UAAU;YACtC,cAAc,EAAE,QAAQ,CAAC,MAAM,CAAC,cAAc;YAC9C,iBAAiB,EAAE,QAAQ,CAAC,MAAM,CAAC,iBAAiB;SACrD;QACD,OAAO,EAAE;YACP,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,QAAQ,CAAC,OAAO,CAAC,UAAU;YAC1D,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,OAAO,CAAC,QAAQ;YACpD,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM;SAC/C;QACD,EAAE,EAAE;YACF,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,QAAQ,CAAC,EAAE,CAAC,YAAY;SAC5D;KACF,CAAC;AACJ,CAAC;AAKD,MAAM,YAAY,GAAwB,IAAI,GAAG,CAAa,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;AAEhF,SAAS,SAAS,CAAC,IAAsB,EAAE,SAAoC;IAC7E,MAAM,MAAM,GAAqB;QAC/B,UAAU,EAAE,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,GAAG,SAAS,CAAC,UAAU,EAAE;QAC3D,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE;QAClD,MAAM,EAAE,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC,MAAM,EAAE;QAC/C,MAAM,EAAE,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC,MAAM,EAAE;QAC/C,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE;QAClD,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC,EAAE,EAAE,GAAG,SAAS,CAAC,EAAE,EAAE;KACpC,CAAC;IACF,MAAM,CAAC,WAAW,GAAG,EAAE,GAAG,IAAI,CAAC,WAAY,EAAE,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;IACxE,IAAI,SAAS,CAAC,iBAAiB;QAAE,MAAM,CAAC,iBAAiB,GAAG,SAAS,CAAC,iBAAiB,CAAC;IACxF,IAAI,SAAS,CAAC,aAAa;QAAE,MAAM,CAAC,aAAa,GAAG,SAAS,CAAC,aAAa,CAAC;IAC5E,MAAM,CAAC,iBAAiB,GAAG,SAAS,CAAC,iBAAiB,IAAI,IAAI,CAAC,iBAAiB,CAAC;IACjF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,wEAAwE;AAExE,MAAM,SAAS,GAA2B,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;AAEnF,SAAS,sBAAsB;IAC7B,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QACrE,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAA4B,CAAC;QAClF,MAAM,MAAM,GAAI,GAAG,EAAE,OAAmC,EAAE,QAA8B,CAAC;QACzF,MAAM,IAAI,GAAG,GAAG,EAAE,QAA8B,CAAC;QACjD,MAAM,KAAK,GAAG,MAAM,IAAI,IAAI,IAAI,MAAM,CAAC;QACvC,OAAO,SAAS,CAAC,KAAK,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC;IAC5C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC,IAAI,CAAC;IACxB,CAAC;AACH,CAAC;AAED,MAAM,oBAAoB,GAAG,sBAAsB,EAAE,CAAC;AAEtD,2FAA2F;AAC3F,SAAS,SAAS,CAAC,OAAe,EAAE,MAAgC,EAAE,QAAgB,MAAM;IAC1F,MAAM,GAAG,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAClC,IAAI,GAAG,GAAG,oBAAoB;QAAE,OAAO;IACvC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QAC7D,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAC3C,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAC7E,MAAM,KAAK,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,oBAAoB,OAAO,EAAE,CAAC,CAAC;QACtE,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC5C,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACtE,CAAC;QACH,CAAC;QACD,SAAS,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACvC,cAAc,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;IAClD,CAAC;IAAC,OAAO,QAAQ,EAAE,CAAC;QAClB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,qCAAqC,QAAQ,YAAY,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACjI,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,SAAqC;IACpE,IAAI,MAAM,GAAqB,eAAe,CAAC,QAAQ,CAAC,CAAC;IAEzD,+BAA+B;IAC/B,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QACrE,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAA4B,CAAC;QAElF,2BAA2B;QAC3B,MAAM,MAAM,GAAqB,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAiB,CAAC,CAAC,CAAC,CAAC,GAAkC,CAAC;QAEjI,+CAA+C;QAC/C,MAAM,GAAG,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACrC,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,SAAS,CAAC,yCAAyC,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAC7E,CAAC;IAED,oEAAoE;IACpE,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY;QAAE,MAAM,CAAC,UAAU,CAAC,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;IAClF,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY;QAAE,MAAM,CAAC,UAAU,CAAC,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;IAClF,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,CAAC;QACxC,MAAM,CAAC,UAAU,CAAC,gBAAgB,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,EAAE,CAAC,IAAI,QAAQ,CAAC,UAAU,CAAC,gBAAgB,CAAC;IACjI,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,oBAAoB,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE,EAAE,CAAC,IAAI,QAAQ,CAAC,MAAM,CAAC,oBAAoB,CAAC;IACrI,CAAC;IACD,IAAI,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,EAAE,CAAC,EAAE,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,gBAA8B,CAAC;IACxE,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,yBAAyB,EAAE,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,mBAAmB,GAAG,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,IAAI,QAAQ,CAAC,MAAM,CAAC,mBAAmB,CAAC;IAC/H,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,gBAAgB,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,EAAE,CAAC,IAAI,QAAQ,CAAC,MAAM,CAAC,gBAAgB,CAAC;IACzH,CAAC;IAED,gDAAgD;IAChD,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,GAAG,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACxC,CAAC;IAED,uCAAuC;IACvC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC3C,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IACzE,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAMD,MAAM,UAAU,kBAAkB,CAChC,SAAiB,EACjB,MAAiC;IAEjC,MAAM,QAAQ,GAAG,MAAM,CAAC,iBAAiB,CAAC;IAC1C,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,KAAK,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,KAAK;YAAE,OAAO,KAAK,CAAC;QACxB,MAAM,KAAK,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;QACtC,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YACrD,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,IAAI,GAAG,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC/E,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;IACH,CAAC;IACD,IAAI,MAAM,CAAC,aAAa;QAAE,OAAO,MAAM,CAAC,aAAa,CAAC;IACtD,SAAS,CAAC,6CAA6C,EAAE,EAAE,SAAS,EAAE,EAAE,OAAO,CAAC,CAAC;IACjF,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,OAAO,EAAE,QAAQ,EAAE,CAAC"}
|
package/dist/hooks.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import type { Model, UserMessage, Part } from "@opencode-ai/sdk";
|
|
2
|
-
import type { CerebroClient } from "./client.js";
|
|
2
|
+
import type { CerebroClient, SearchResult } from "./client.js";
|
|
3
3
|
import { type OmemPluginConfig } from "./config.js";
|
|
4
4
|
export declare function autoRecallHook(client: CerebroClient, containerTags: string[], tui: any, config?: Partial<OmemPluginConfig>, getAgentName?: () => string, directory?: string): (input: {
|
|
5
5
|
sessionID?: string;
|
|
@@ -7,6 +7,14 @@ export declare function autoRecallHook(client: CerebroClient, containerTags: str
|
|
|
7
7
|
}, output: {
|
|
8
8
|
system: string[];
|
|
9
9
|
}) => Promise<void>;
|
|
10
|
+
export declare function memoryInjectionHook(client: CerebroClient, containerTags: string[], tui: any, config?: Partial<OmemPluginConfig>, getAgentName?: () => string, directory?: string): (input: {
|
|
11
|
+
sessionID?: string;
|
|
12
|
+
messageID?: string;
|
|
13
|
+
model: Model;
|
|
14
|
+
}, output: {
|
|
15
|
+
message: UserMessage;
|
|
16
|
+
parts: Part[];
|
|
17
|
+
}) => Promise<void>;
|
|
10
18
|
export declare function keywordDetectionHook(_client: CerebroClient, _containerTags: string[], threshold: number, _tui: any, _ingestMode?: "smart" | "raw", config?: Partial<OmemPluginConfig>, agentId?: string): (input: {
|
|
11
19
|
sessionID: string;
|
|
12
20
|
messageID?: string;
|
|
@@ -14,6 +22,7 @@ export declare function keywordDetectionHook(_client: CerebroClient, _containerT
|
|
|
14
22
|
message: UserMessage;
|
|
15
23
|
parts: Part[];
|
|
16
24
|
}) => Promise<void>;
|
|
25
|
+
export declare function createCerebroCompactionPrompt(context: string[], projectMemories: SearchResult[]): string;
|
|
17
26
|
export declare function compactingHook(client: CerebroClient, containerTags: string[], tui: any, ingestMode?: "smart" | "raw", isAutoStoreEnabled?: (sessionId: string | undefined) => boolean, getMainSessionId?: () => string | undefined, sdkClient?: any, config?: Partial<OmemPluginConfig>, agentId?: string, directory?: string): (input: {
|
|
18
27
|
sessionID?: string;
|
|
19
28
|
}, output: {
|
package/dist/hooks.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"hooks.d.ts","sourceRoot":"","sources":["../src/hooks.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AACjE,OAAO,KAAK,EAAE,aAAa,
|
|
1
|
+
{"version":3,"file":"hooks.d.ts","sourceRoot":"","sources":["../src/hooks.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AACjE,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC/D,OAAO,EAAE,KAAK,gBAAgB,EAAsB,MAAM,aAAa,CAAC;AA6TxE,wBAAgB,cAAc,CAAC,MAAM,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,GAAE,OAAO,CAAC,gBAAgB,CAAM,EAAE,YAAY,CAAC,EAAE,MAAM,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,IAgB5K,OAAO;IAAE,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,KAAK,CAAA;CAAE,EAC3C,QAAQ;IAAE,MAAM,EAAE,MAAM,EAAE,CAAA;CAAE,mBAuR/B;AAED,wBAAgB,mBAAmB,CACjC,MAAM,EAAE,aAAa,EACrB,aAAa,EAAE,MAAM,EAAE,EACvB,GAAG,EAAE,GAAG,EACR,MAAM,GAAE,OAAO,CAAC,gBAAgB,CAAM,EACtC,YAAY,CAAC,EAAE,MAAM,MAAM,EAC3B,SAAS,CAAC,EAAE,MAAM,IAiBhB,OAAO;IAAE,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,KAAK,CAAA;CAAE,EAC/D,QAAQ;IAAE,OAAO,EAAE,WAAW,CAAC;IAAC,KAAK,EAAE,IAAI,EAAE,CAAA;CAAE,mBA4TlD;AAED,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,WAAW,GAAE,OAAO,GAAG,KAAe,EAAE,MAAM,GAAE,OAAO,CAAC,gBAAgB,CAAM,EAAE,OAAO,CAAC,EAAE,MAAM,IAGjN,OAAO;IAAE,SAAS,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,EAChD,QAAQ;IAAE,OAAO,EAAE,WAAW,CAAC;IAAC,KAAK,EAAE,IAAI,EAAE,CAAA;CAAE,mBAoClD;AAED,wBAAgB,6BAA6B,CAC3C,OAAO,EAAE,MAAM,EAAE,EACjB,eAAe,EAAE,YAAY,EAAE,GAC9B,MAAM,CA8CR;AAED,wBAAgB,cAAc,CAAC,MAAM,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,UAAU,GAAE,OAAO,GAAG,KAAe,EAAE,kBAAkB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,KAAK,OAAO,EAAE,gBAAgB,CAAC,EAAE,MAAM,MAAM,GAAG,SAAS,EAAE,SAAS,CAAC,EAAE,GAAG,EAAE,MAAM,GAAE,OAAO,CAAC,gBAAgB,CAAM,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,IAGvU,OAAO;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,EAC7B,QAAQ;IAAE,OAAO,EAAE,MAAM,EAAE,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,mBA6QjD;AAED,wBAAgB,gBAAgB,CAC9B,MAAM,EAAE,aAAa,EACrB,aAAa,EAAE,MAAM,EAAE,EACvB,GAAG,EAAE,GAAG,EACR,UAAU,GAAE,OAAO,GAAG,KAAe,EACrC,kBAAkB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,KAAK,OAAO,EAC/D,gBAAgB,CAAC,EAAE,MAAM,MAAM,GAAG,SAAS,EAC3C,SAAS,CAAC,EAAE,GAAG,EACf,MAAM,GAAE,OAAO,CAAC,gBAAgB,CAAM,EACtC,OAAO,CAAC,EAAE,MAAM,EAChB,SAAS,CAAC,EAAE,MAAM,IAIhB,OAAO;IACL,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,KAAK,CAAC;IACb,OAAO,EAAE,WAAW,CAAC;IACrB,QAAQ,EAAE,OAAO,CAAC;CACnB,EACD,SAAS;IAAE,OAAO,EAAE,OAAO,CAAA;CAAE,mBAuFhC;AAMD,eAAO,MAAM,gBAAgB;cAA2C,MAAM;eAAa,MAAM;GAAM,CAAC;AAExG,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,gBAAgB,IAC/C,OAAO;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,EAAE,SAAS;IAAE,IAAI,EAAE,GAAG,CAAA;CAAE,mBA+BjG;AAED,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAczF;AAED,wBAAgB,eAAe,CAC7B,aAAa,EAAE,aAAa,EAC5B,cAAc,EAAE,MAAM,EAAE,EACxB,GAAG,EAAE,GAAG,EACR,SAAS,EAAE,GAAG,EACd,WAAW,GAAE,OAAO,GAAG,KAAe,EACtC,SAAS,GAAE,MAAU,EACrB,gBAAgB,CAAC,EAAE,MAAM,MAAM,GAAG,SAAS,EAC3C,kBAAkB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,KAAK,OAAO,EAC/D,OAAO,CAAC,EAAE,MAAM,EAChB,MAAM,GAAE,OAAO,CAAC,gBAAgB,CAAM,EACtC,eAAe,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,EACxC,SAAS,CAAC,EAAE,MAAM,IAKJ,OAAO;IAAE,KAAK,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,GAAG,CAAA;KAAE,CAAA;CAAE,mBAiHnE"}
|
package/dist/hooks.js
CHANGED
|
@@ -124,7 +124,7 @@ function showToast(tui, title, message, variant = "info", delayMs = 7000) {
|
|
|
124
124
|
tui.showToast({ body: { title, message, variant, duration: 5000 } });
|
|
125
125
|
}
|
|
126
126
|
catch (err) {
|
|
127
|
-
|
|
127
|
+
logErr("showToast failed", { error: String(err) });
|
|
128
128
|
}
|
|
129
129
|
}, delayMs);
|
|
130
130
|
}
|
|
@@ -162,6 +162,17 @@ const injectedMemoryIds = new Map();
|
|
|
162
162
|
const firstMessages = new Map();
|
|
163
163
|
const sessionMessages = new Map();
|
|
164
164
|
const profileInjectedSessions = new Map();
|
|
165
|
+
const injectedSessions = new Set();
|
|
166
|
+
const compactingSummaryCooldown = new Map();
|
|
167
|
+
function hashString(str) {
|
|
168
|
+
let hash = 0;
|
|
169
|
+
for (let i = 0; i < str.length; i++) {
|
|
170
|
+
const char = str.charCodeAt(i);
|
|
171
|
+
hash = ((hash << 5) - hash) + char;
|
|
172
|
+
hash |= 0;
|
|
173
|
+
}
|
|
174
|
+
return hash.toString(36);
|
|
175
|
+
}
|
|
165
176
|
function formatRelativeAge(isoDate) {
|
|
166
177
|
const diffMs = Date.now() - new Date(isoDate).getTime();
|
|
167
178
|
const minutes = Math.floor(diffMs / 60_000);
|
|
@@ -223,7 +234,9 @@ function formatMemoryLine(r, maxContentLength) {
|
|
|
223
234
|
}
|
|
224
235
|
const FETCH_POLICY = [
|
|
225
236
|
"<cerebro-fetch-policy>",
|
|
226
|
-
"Each memory above is a condensed summary
|
|
237
|
+
"IMPORTANT: Each memory above is a condensed summary. The full version contains critical details that may change your response quality.",
|
|
238
|
+
"You MUST use memory_get(\"id\") to retrieve the complete content before making decisions based on any summary.",
|
|
239
|
+
"Do NOT rely on condensed summaries alone — depth of recall determines quality of response.",
|
|
227
240
|
"</cerebro-fetch-policy>",
|
|
228
241
|
].join("\n");
|
|
229
242
|
function buildContextBlock(results, maxContentLength = 500) {
|
|
@@ -425,6 +438,10 @@ export function autoRecallHook(client, containerTags, tui, config = {}, getAgent
|
|
|
425
438
|
}
|
|
426
439
|
};
|
|
427
440
|
if (!shouldRecallRes.should_recall) {
|
|
441
|
+
if (profileBlock) {
|
|
442
|
+
appendToSystem(output.system, profileBlock);
|
|
443
|
+
logDebug("autoRecallHook profile injected (no-recall path)", { sessionId: input.sessionID, outputSystemLength: output.system.length });
|
|
444
|
+
}
|
|
428
445
|
if (profileInjected && isFirstInjection) {
|
|
429
446
|
await createEventAndReturn(0, 0, 0);
|
|
430
447
|
showToast(tui, "👨 Profile Injected", `${profileCountText} · no memory recall needed`, "success", toastDelayMs);
|
|
@@ -437,6 +454,10 @@ export function autoRecallHook(client, containerTags, tui, config = {}, getAgent
|
|
|
437
454
|
const newResults = results.filter((r) => !existingIds.has(r.memory.id));
|
|
438
455
|
logDebug("autoRecallHook dedup", { totalResults: results.length, existingCount: existingIds.size, newCount: newResults.length });
|
|
439
456
|
if (newResults.length === 0) {
|
|
457
|
+
if (profileBlock) {
|
|
458
|
+
appendToSystem(output.system, profileBlock);
|
|
459
|
+
logDebug("autoRecallHook profile injected (dedup path)", { sessionId: input.sessionID, outputSystemLength: output.system.length });
|
|
460
|
+
}
|
|
440
461
|
if (profileInjected && isFirstInjection) {
|
|
441
462
|
showToast(tui, "👨 Profile Injected", `${profileCountText} · all memories already injected`, "success", toastDelayMs);
|
|
442
463
|
}
|
|
@@ -543,6 +564,315 @@ export function autoRecallHook(client, containerTags, tui, config = {}, getAgent
|
|
|
543
564
|
}
|
|
544
565
|
};
|
|
545
566
|
}
|
|
567
|
+
export function memoryInjectionHook(client, containerTags, tui, config = {}, getAgentName, directory) {
|
|
568
|
+
const similarityThreshold = config.recall?.similarityThreshold ?? 0.4;
|
|
569
|
+
const maxRecallResults = config.recall?.maxRecallResults ?? 10;
|
|
570
|
+
const fetchMultiplier = config.recall?.fetchMultiplier ?? 3;
|
|
571
|
+
const topkCapMultiplier = config.recall?.topkCapMultiplier ?? 2;
|
|
572
|
+
const mmrJaccardThreshold = config.recall?.mmrJaccardThreshold ?? 0.85;
|
|
573
|
+
const mmrPenaltyFactor = config.recall?.mmrPenaltyFactor ?? 0.5;
|
|
574
|
+
const phase2Multiplier = config.recall?.phase2Multiplier ?? 2;
|
|
575
|
+
const llmMaxEval = config.recall?.llmMaxEval ?? 15;
|
|
576
|
+
const refineStrategy = config.recall?.refineStrategy ?? "balanced";
|
|
577
|
+
const refineMediumChars = config.recall?.refineMediumChars ?? 200;
|
|
578
|
+
const maxContentLength = Math.max(MIN_CONTENT_LENGTH, config.content?.maxContentLength ?? 500);
|
|
579
|
+
const maxContentChars = Math.max(MIN_CONTENT_CHARS, config.content?.maxContentChars ?? 30000);
|
|
580
|
+
const toastDelayMs = config.ui?.toastDelayMs ?? 7000;
|
|
581
|
+
return async (input, output) => {
|
|
582
|
+
if (!input.sessionID)
|
|
583
|
+
return;
|
|
584
|
+
const agentId = getAgentName?.() || process.env.OMEM_AGENT_ID || "opencode";
|
|
585
|
+
const policy = resolveAgentPolicy(agentId, config);
|
|
586
|
+
if (policy === "none")
|
|
587
|
+
return;
|
|
588
|
+
const isFirstInjection = !injectedSessions.has(input.sessionID);
|
|
589
|
+
const isKeywordTriggered = keywordDetectedSessions.has(input.sessionID);
|
|
590
|
+
if (!isFirstInjection && !isKeywordTriggered)
|
|
591
|
+
return;
|
|
592
|
+
try {
|
|
593
|
+
logDebug("memoryInjectionHook start", { sessionId: input.sessionID, agentId, policy, isFirstInjection, isKeywordTriggered, similarityThreshold, maxRecallResults });
|
|
594
|
+
const messages = sessionMessages.get(input.sessionID) ?? [];
|
|
595
|
+
const userMessages = messages.filter((m) => m.role === "user");
|
|
596
|
+
// --- Profile Fetch ---
|
|
597
|
+
const profile = await client.getProfile();
|
|
598
|
+
let profileInjected = false;
|
|
599
|
+
let profileCountText = "";
|
|
600
|
+
let profileBlock = "";
|
|
601
|
+
const lastInjected = profileInjectedSessions.get(input.sessionID);
|
|
602
|
+
const ttlExpired = !lastInjected || (Date.now() - lastInjected > 30 * 60 * 1000);
|
|
603
|
+
const profileIsFirstInjection = !lastInjected;
|
|
604
|
+
if (profile && ttlExpired) {
|
|
605
|
+
const prefs = (profile?.static_facts ?? [])
|
|
606
|
+
.filter((sf) => {
|
|
607
|
+
const t = sf.tags ?? [];
|
|
608
|
+
return t.includes("preferences");
|
|
609
|
+
})
|
|
610
|
+
.map((sf) => sf.l2_content ?? sf.content ?? "")
|
|
611
|
+
.filter(Boolean);
|
|
612
|
+
const profileLines = prefs.length > 0
|
|
613
|
+
? prefs.map((c) => ` · ${c}`).join("\n")
|
|
614
|
+
: " · (preferences queuing, will populate on next refresh)";
|
|
615
|
+
profileBlock = [
|
|
616
|
+
"<cerebro-profile>",
|
|
617
|
+
profileLines,
|
|
618
|
+
"</cerebro-profile>",
|
|
619
|
+
].join("\n");
|
|
620
|
+
profileInjected = true;
|
|
621
|
+
profileInjectedSessions.set(input.sessionID, Date.now());
|
|
622
|
+
const p = profile;
|
|
623
|
+
const dynamicCount = p?.dynamic_context?.length ?? 0;
|
|
624
|
+
const staticCount = p?.static_facts?.length ?? 0;
|
|
625
|
+
profileCountText = `Dynamic(${dynamicCount}) · Static(${staticCount})`;
|
|
626
|
+
if (profileIsFirstInjection) {
|
|
627
|
+
logDebug("memoryInjectionHook profile ready (first)", { dynamicCount, staticCount });
|
|
628
|
+
}
|
|
629
|
+
else {
|
|
630
|
+
logDebug("memoryInjectionHook profile ready (TTL)", { dynamicCount, staticCount });
|
|
631
|
+
}
|
|
632
|
+
}
|
|
633
|
+
if (userMessages.length === 0) {
|
|
634
|
+
logDebug("memoryInjectionHook skipped: no user messages in session (post-compacting?)", { sessionId: input.sessionID });
|
|
635
|
+
return;
|
|
636
|
+
}
|
|
637
|
+
const rawQuery = userMessages[userMessages.length - 1]?.content || firstMessages.get(input.sessionID) || "";
|
|
638
|
+
const query_text = extractUserRequest(rawQuery);
|
|
639
|
+
if (!query_text) {
|
|
640
|
+
logDebug("memoryInjectionHook filtered system injection (profile already injected above)", { rawQueryPrefix: rawQuery.slice(0, 60) });
|
|
641
|
+
return;
|
|
642
|
+
}
|
|
643
|
+
const last_query_text = userMessages.length >= 2 ? userMessages[userMessages.length - 2].content : undefined;
|
|
644
|
+
const projectTags = containerTags.filter(t => t.startsWith("omem_project_"));
|
|
645
|
+
const conversationContext = userMessages.length >= 2
|
|
646
|
+
? userMessages.slice(-4, -1).map((m) => {
|
|
647
|
+
const stripped = stripPrivateContent(m.content);
|
|
648
|
+
return stripped.length > 200 ? stripped.slice(0, 200) : stripped;
|
|
649
|
+
})
|
|
650
|
+
: undefined;
|
|
651
|
+
const shouldRecallRes = await client.shouldRecall(query_text, last_query_text, input.sessionID, similarityThreshold, maxRecallResults, projectTags.length > 0 ? projectTags : undefined, conversationContext && conversationContext.length > 0 ? conversationContext : undefined, {
|
|
652
|
+
fetch_multiplier: fetchMultiplier,
|
|
653
|
+
topk_cap_multiplier: topkCapMultiplier,
|
|
654
|
+
mmr_jaccard_threshold: mmrJaccardThreshold,
|
|
655
|
+
mmr_penalty_factor: mmrPenaltyFactor,
|
|
656
|
+
phase2_multiplier: phase2Multiplier,
|
|
657
|
+
llm_max_eval: llmMaxEval,
|
|
658
|
+
refine_strategy: refineStrategy,
|
|
659
|
+
refine_medium_chars: refineMediumChars,
|
|
660
|
+
}, directory || process.env.OMEM_PROJECT_DIR);
|
|
661
|
+
if (!shouldRecallRes) {
|
|
662
|
+
showToast(tui, "🧠 Cerebro Service Unavailable", "Unable to reach memory API · check connection", "error", toastDelayMs);
|
|
663
|
+
return;
|
|
664
|
+
}
|
|
665
|
+
logDebug("memoryInjectionHook shouldRecall result", { shouldRecall: shouldRecallRes.should_recall, confidence: shouldRecallRes.confidence, memCount: shouldRecallRes.memories?.length ?? 0, discardedCount: shouldRecallRes.discarded?.length ?? 0, clustered: !!shouldRecallRes.clustered });
|
|
666
|
+
const storedMemoryIds = shouldRecallRes.memories?.map((r) => r.memory.id) ?? [];
|
|
667
|
+
const storedDiscardedIds = shouldRecallRes.discarded?.map((d) => d.memory_id) ?? [];
|
|
668
|
+
const maxScore = storedMemoryIds.length > 0
|
|
669
|
+
? Math.max(...(shouldRecallRes.memories?.map((r) => r.score) ?? [0]))
|
|
670
|
+
: 0;
|
|
671
|
+
const createEventAndReturn = async (injectedCount, keptCount, discardedCount, injectedContent) => {
|
|
672
|
+
try {
|
|
673
|
+
const items = [
|
|
674
|
+
...(shouldRecallRes.memories?.map((r) => ({
|
|
675
|
+
memory_id: r.memory.id,
|
|
676
|
+
score: r.score,
|
|
677
|
+
refine_relevance: r.refine_relevance,
|
|
678
|
+
refine_reasoning: r.refine_reasoning,
|
|
679
|
+
is_kept: true,
|
|
680
|
+
})) ?? []),
|
|
681
|
+
...(shouldRecallRes.discarded?.map((d) => ({
|
|
682
|
+
memory_id: d.memory_id,
|
|
683
|
+
score: d.score,
|
|
684
|
+
refine_relevance: d.refine_relevance,
|
|
685
|
+
refine_reasoning: d.refine_reasoning,
|
|
686
|
+
is_kept: false,
|
|
687
|
+
})) ?? []),
|
|
688
|
+
];
|
|
689
|
+
const result = await client.createRecallEvent({
|
|
690
|
+
session_id: input.sessionID,
|
|
691
|
+
recall_type: "auto",
|
|
692
|
+
query_text,
|
|
693
|
+
max_score: maxScore,
|
|
694
|
+
llm_confidence: shouldRecallRes.confidence ?? 0,
|
|
695
|
+
profile_injected: profileInjected,
|
|
696
|
+
kept_count: keptCount,
|
|
697
|
+
discarded_count: discardedCount,
|
|
698
|
+
injected_count: injectedCount,
|
|
699
|
+
profile_content: profileInjected && profileBlock ? profileBlock : undefined,
|
|
700
|
+
injected_content: injectedContent,
|
|
701
|
+
items: items.length > 0 ? items : undefined,
|
|
702
|
+
});
|
|
703
|
+
return result?.event_id;
|
|
704
|
+
}
|
|
705
|
+
catch (e) {
|
|
706
|
+
logErr("memoryInjectionHook createRecallEvent failed", { error: String(e) });
|
|
707
|
+
return undefined;
|
|
708
|
+
}
|
|
709
|
+
};
|
|
710
|
+
// --- no-recall path: inject profile only ---
|
|
711
|
+
if (!shouldRecallRes.should_recall) {
|
|
712
|
+
const partsToInject = [];
|
|
713
|
+
if (profileBlock)
|
|
714
|
+
partsToInject.push(profileBlock);
|
|
715
|
+
if (partsToInject.length > 0) {
|
|
716
|
+
const injectText = partsToInject.join("\n\n");
|
|
717
|
+
const contextPart = {
|
|
718
|
+
id: `prt_cerebro-context-${Date.now()}`,
|
|
719
|
+
sessionID: input.sessionID,
|
|
720
|
+
messageID: output.message.id,
|
|
721
|
+
type: "text",
|
|
722
|
+
text: injectText,
|
|
723
|
+
synthetic: true,
|
|
724
|
+
};
|
|
725
|
+
output.parts.unshift(contextPart);
|
|
726
|
+
logDebug("memoryInjectionHook profile injected (no-recall path)", { sessionId: input.sessionID });
|
|
727
|
+
}
|
|
728
|
+
injectedSessions.add(input.sessionID);
|
|
729
|
+
if (profileInjected && profileIsFirstInjection) {
|
|
730
|
+
await createEventAndReturn(0, 0, 0);
|
|
731
|
+
showToast(tui, "👨 Profile Injected", `${profileCountText} · no memory recall needed`, "success", toastDelayMs);
|
|
732
|
+
}
|
|
733
|
+
return;
|
|
734
|
+
}
|
|
735
|
+
const results = shouldRecallRes.memories ?? [];
|
|
736
|
+
const clustered = shouldRecallRes.clustered;
|
|
737
|
+
const existingIds = injectedMemoryIds.get(input.sessionID) ?? new Set();
|
|
738
|
+
const newResults = results.filter((r) => !existingIds.has(r.memory.id));
|
|
739
|
+
logDebug("memoryInjectionHook dedup", { totalResults: results.length, existingCount: existingIds.size, newCount: newResults.length });
|
|
740
|
+
// --- dedup path: inject profile only ---
|
|
741
|
+
if (newResults.length === 0) {
|
|
742
|
+
const partsToInject = [];
|
|
743
|
+
if (profileBlock)
|
|
744
|
+
partsToInject.push(profileBlock);
|
|
745
|
+
if (partsToInject.length > 0) {
|
|
746
|
+
const injectText = partsToInject.join("\n\n");
|
|
747
|
+
const contextPart = {
|
|
748
|
+
id: `prt_cerebro-context-${Date.now()}`,
|
|
749
|
+
sessionID: input.sessionID,
|
|
750
|
+
messageID: output.message.id,
|
|
751
|
+
type: "text",
|
|
752
|
+
text: injectText,
|
|
753
|
+
synthetic: true,
|
|
754
|
+
};
|
|
755
|
+
output.parts.unshift(contextPart);
|
|
756
|
+
logDebug("memoryInjectionHook profile injected (dedup path)", { sessionId: input.sessionID });
|
|
757
|
+
}
|
|
758
|
+
injectedSessions.add(input.sessionID);
|
|
759
|
+
if (profileInjected && profileIsFirstInjection) {
|
|
760
|
+
showToast(tui, "👨 Profile Injected", `${profileCountText} · all memories already injected`, "success", toastDelayMs);
|
|
761
|
+
}
|
|
762
|
+
return;
|
|
763
|
+
}
|
|
764
|
+
// --- Token Budget Calculation ---
|
|
765
|
+
const profileChars = profileInjected ? profileBlock.length : 0;
|
|
766
|
+
const budgetRemaining = maxContentChars - profileChars;
|
|
767
|
+
if (budgetRemaining < 0) {
|
|
768
|
+
logDebug("memoryInjectionHook budget overflow", { profileChars, maxContentChars, deficit: -budgetRemaining });
|
|
769
|
+
}
|
|
770
|
+
const itemCount = clustered
|
|
771
|
+
? (clustered.cluster_summaries.length + clustered.standalone_memories.length)
|
|
772
|
+
: newResults.length;
|
|
773
|
+
const dynamicMaxContentLength = itemCount > 0
|
|
774
|
+
? Math.min(maxContentLength, Math.max(MIN_ITEM_CONTENT_CHARS, Math.floor(budgetRemaining / itemCount)))
|
|
775
|
+
: maxContentLength;
|
|
776
|
+
logDebug("memoryInjectionHook budget", {
|
|
777
|
+
maxContentChars, profileChars, budgetRemaining, itemCount,
|
|
778
|
+
configuredMax: maxContentLength, dynamicMax: dynamicMaxContentLength,
|
|
779
|
+
});
|
|
780
|
+
const block = clustered
|
|
781
|
+
? buildClusteredContextBlock(clustered, dynamicMaxContentLength)
|
|
782
|
+
: buildContextBlock(newResults, dynamicMaxContentLength);
|
|
783
|
+
// ★★★ Core change: inject via output.parts.unshift + synthetic:true ★★★
|
|
784
|
+
const partsToInject = [];
|
|
785
|
+
if (profileBlock)
|
|
786
|
+
partsToInject.push(profileBlock);
|
|
787
|
+
if (block)
|
|
788
|
+
partsToInject.push(block);
|
|
789
|
+
if (block)
|
|
790
|
+
partsToInject.push(FETCH_POLICY);
|
|
791
|
+
if (isKeywordTriggered)
|
|
792
|
+
partsToInject.push(KEYWORD_NUDGE);
|
|
793
|
+
if (partsToInject.length > 0) {
|
|
794
|
+
const injectText = partsToInject.join("\n\n");
|
|
795
|
+
const contextPart = {
|
|
796
|
+
id: `prt_cerebro-context-${Date.now()}`,
|
|
797
|
+
sessionID: input.sessionID,
|
|
798
|
+
messageID: output.message.id,
|
|
799
|
+
type: "text",
|
|
800
|
+
text: injectText,
|
|
801
|
+
synthetic: true,
|
|
802
|
+
};
|
|
803
|
+
output.parts.unshift(contextPart);
|
|
804
|
+
logDebug("memoryInjectionHook block injected to output.parts", {
|
|
805
|
+
sessionId: input.sessionID,
|
|
806
|
+
injectTextLen: injectText.length,
|
|
807
|
+
blockPreview: block?.slice(0, 200),
|
|
808
|
+
});
|
|
809
|
+
}
|
|
810
|
+
else {
|
|
811
|
+
logDebug("memoryInjectionHook no content to inject", { sessionId: input.sessionID });
|
|
812
|
+
}
|
|
813
|
+
injectedSessions.add(input.sessionID);
|
|
814
|
+
if (isKeywordTriggered) {
|
|
815
|
+
keywordDetectedSessions.delete(input.sessionID);
|
|
816
|
+
}
|
|
817
|
+
const newIds = newResults.map((r) => r.memory.id);
|
|
818
|
+
injectedMemoryIds.set(input.sessionID, new Set([...existingIds, ...newIds]));
|
|
819
|
+
logDebug("memoryInjectionHook injection complete", { newIds: newIds.length, clustered: !!clustered, sessionId: input.sessionID });
|
|
820
|
+
await createEventAndReturn(newResults.length, storedMemoryIds.length, storedDiscardedIds.length, block || undefined);
|
|
821
|
+
const memDynamic = newResults.filter((r) => r.memory.memory_type === "fact" || r.memory.memory_type === "event").length;
|
|
822
|
+
const memStatic = newResults.filter((r) => r.memory.memory_type === "pinned" || r.memory.memory_type === "preference").length;
|
|
823
|
+
const memOther = newResults.length - memDynamic - memStatic;
|
|
824
|
+
let memCountMsg = "";
|
|
825
|
+
if (memDynamic > 0)
|
|
826
|
+
memCountMsg += `Dynamic(${memDynamic}) `;
|
|
827
|
+
if (memStatic > 0)
|
|
828
|
+
memCountMsg += `Static(${memStatic}) `;
|
|
829
|
+
if (memOther > 0)
|
|
830
|
+
memCountMsg += `Other(${memOther}) `;
|
|
831
|
+
const categories = categorize(newResults);
|
|
832
|
+
const catSummary = Array.from(categories.entries())
|
|
833
|
+
.map(([label, items]) => `${label}(${items.length})`)
|
|
834
|
+
.join(" · ");
|
|
835
|
+
let toastTitle;
|
|
836
|
+
let toastMessage;
|
|
837
|
+
if (clustered) {
|
|
838
|
+
const clusterCount = clustered.cluster_summaries.length;
|
|
839
|
+
const standaloneCount = clustered.standalone_memories.length;
|
|
840
|
+
toastTitle = `🧠 Context Injected · ${clusterCount} 主题簇${standaloneCount > 0 ? ` · ${standaloneCount} 补充` : ""}`;
|
|
841
|
+
toastMessage = profileInjected
|
|
842
|
+
? `Profile: ${profileCountText} · 聚合记忆展示`
|
|
843
|
+
: `聚合记忆展示`;
|
|
844
|
+
}
|
|
845
|
+
else {
|
|
846
|
+
toastTitle = `🧠 Context Injected · ${newResults.length} fragments`;
|
|
847
|
+
toastMessage = profileInjected
|
|
848
|
+
? `Profile: ${profileCountText} · Memories: ${memCountMsg.trim()}${catSummary ? ` · ${catSummary}` : ""}`
|
|
849
|
+
: `${memCountMsg.trim()}${catSummary ? ` · ${catSummary}` : ""}`;
|
|
850
|
+
}
|
|
851
|
+
showToast(tui, toastTitle, toastMessage, "success", toastDelayMs);
|
|
852
|
+
}
|
|
853
|
+
catch (err) {
|
|
854
|
+
const errMsg = err instanceof Error ? err.message : String(err);
|
|
855
|
+
if (errMsg.includes("[cerebro]")) {
|
|
856
|
+
const cleanMsg = errMsg.replace(/^\[cerebro\]\s*/, "");
|
|
857
|
+
if (cleanMsg.startsWith("500")) {
|
|
858
|
+
showToast(tui, "🧠 Cerebro Server Error", cleanMsg.substring(0, 200), "error");
|
|
859
|
+
}
|
|
860
|
+
else if (cleanMsg.includes("timed out")) {
|
|
861
|
+
showToast(tui, "🧠 Cerebro Service Timeout", cleanMsg.substring(0, 100), "error");
|
|
862
|
+
}
|
|
863
|
+
else {
|
|
864
|
+
showToast(tui, "🧠 Cerebro Error", cleanMsg.substring(0, 150), "error");
|
|
865
|
+
}
|
|
866
|
+
}
|
|
867
|
+
else if (errMsg.includes("fetch") || errMsg.includes("network")) {
|
|
868
|
+
showToast(tui, "🧠 Cerebro Service Unavailable", "Network error · check API connection", "error");
|
|
869
|
+
}
|
|
870
|
+
else {
|
|
871
|
+
showToast(tui, "🧠 Memory Recall Error", errMsg.substring(0, 100), "error");
|
|
872
|
+
}
|
|
873
|
+
}
|
|
874
|
+
};
|
|
875
|
+
}
|
|
546
876
|
export function keywordDetectionHook(_client, _containerTags, threshold, _tui, _ingestMode = "smart", config = {}, agentId) {
|
|
547
877
|
const effectiveAgentId = agentId || process.env.OMEM_AGENT_ID || "opencode";
|
|
548
878
|
return async (input, output) => {
|
|
@@ -576,6 +906,50 @@ export function keywordDetectionHook(_client, _containerTags, threshold, _tui, _
|
|
|
576
906
|
}
|
|
577
907
|
};
|
|
578
908
|
}
|
|
909
|
+
export function createCerebroCompactionPrompt(context, projectMemories) {
|
|
910
|
+
const sections = [
|
|
911
|
+
"[Cerebro Compaction Context]",
|
|
912
|
+
"",
|
|
913
|
+
"## 1. User's Original Request",
|
|
914
|
+
"Preserve the user's verbatim original request from the conversation above.",
|
|
915
|
+
"",
|
|
916
|
+
"## 2. Final Goal",
|
|
917
|
+
"What is the ultimate objective the user wants to achieve?",
|
|
918
|
+
"",
|
|
919
|
+
"## 3. Work Completed",
|
|
920
|
+
"List all completed work with file paths and technical decisions made.",
|
|
921
|
+
"",
|
|
922
|
+
"## 4. Remaining Tasks",
|
|
923
|
+
"What is still unfinished or pending?",
|
|
924
|
+
"",
|
|
925
|
+
"## 5. Prohibited Actions",
|
|
926
|
+
"Key constraints and forbidden operations to remember.",
|
|
927
|
+
"",
|
|
928
|
+
"## 6. Existing Project Knowledge",
|
|
929
|
+
];
|
|
930
|
+
if (projectMemories.length > 0) {
|
|
931
|
+
const memBlock = projectMemories
|
|
932
|
+
.slice(0, 10)
|
|
933
|
+
.map((r) => {
|
|
934
|
+
const content = r.memory.content ?? "";
|
|
935
|
+
const truncated = content.length > 200 ? content.slice(0, 200) + "..." : content;
|
|
936
|
+
return ` - [${r.memory.category ?? "general"}] ${truncated}`;
|
|
937
|
+
})
|
|
938
|
+
.join("\n");
|
|
939
|
+
sections.push(memBlock);
|
|
940
|
+
}
|
|
941
|
+
else {
|
|
942
|
+
sections.push(" (No project memories retrieved)");
|
|
943
|
+
}
|
|
944
|
+
if (context.length > 0) {
|
|
945
|
+
sections.push("");
|
|
946
|
+
sections.push("### Additional Context");
|
|
947
|
+
sections.push(...context);
|
|
948
|
+
}
|
|
949
|
+
sections.push("");
|
|
950
|
+
sections.push("IMPORTANT: Output must preserve the user's original language (Chinese/English/etc). Do not translate.");
|
|
951
|
+
return sections.join("\n");
|
|
952
|
+
}
|
|
579
953
|
export function compactingHook(client, containerTags, tui, ingestMode = "smart", isAutoStoreEnabled, getMainSessionId, sdkClient, config = {}, agentId, directory) {
|
|
580
954
|
const effectiveAgentId = agentId || process.env.OMEM_AGENT_ID || "opencode";
|
|
581
955
|
return async (input, output) => {
|
|
@@ -583,20 +957,21 @@ export function compactingHook(client, containerTags, tui, ingestMode = "smart",
|
|
|
583
957
|
// Search (read) always runs — even readonly agents need context during compacting
|
|
584
958
|
try {
|
|
585
959
|
const results = await client.searchMemories("*", 20, undefined, containerTags);
|
|
586
|
-
const
|
|
587
|
-
if (
|
|
588
|
-
|
|
589
|
-
|
|
590
|
-
|
|
591
|
-
|
|
592
|
-
|
|
593
|
-
|
|
594
|
-
|
|
595
|
-
|
|
596
|
-
|
|
597
|
-
|
|
598
|
-
|
|
599
|
-
|
|
960
|
+
const compactionPrompt = createCerebroCompactionPrompt(output.context, results);
|
|
961
|
+
if (output.prompt !== undefined) {
|
|
962
|
+
output.prompt = compactionPrompt;
|
|
963
|
+
}
|
|
964
|
+
else if (output.context.length > 0) {
|
|
965
|
+
output.context[output.context.length - 1] += "\n\n" + compactionPrompt;
|
|
966
|
+
}
|
|
967
|
+
else {
|
|
968
|
+
output.context.push(compactionPrompt);
|
|
969
|
+
}
|
|
970
|
+
if (output.context.length > 0) {
|
|
971
|
+
output.context[output.context.length - 1] += "\n\n" + FETCH_POLICY;
|
|
972
|
+
}
|
|
973
|
+
else {
|
|
974
|
+
output.context.push(FETCH_POLICY);
|
|
600
975
|
}
|
|
601
976
|
// 将compacting搜索结果的ID写入injectedMemoryIds，避免后续autoRecall重复注入
|
|
602
977
|
if (input.sessionID && results.length > 0) {
|
|
@@ -792,8 +1167,17 @@ export function compactingHook(client, containerTags, tui, ingestMode = "smart",
|
|
|
792
1167
|
logInfo("compactingHook: storing compact summary", {
|
|
793
1168
|
summaryLen: summaryText.length, msgId: compactMsg.info?.id,
|
|
794
1169
|
});
|
|
1170
|
+
// Dedup check: 30s cooldown per session+content hash
|
|
1171
|
+
const summaryHash = `${pollSessionId}:${hashString(summaryText)}`;
|
|
1172
|
+
const lastCompacting = compactingSummaryCooldown.get(summaryHash);
|
|
1173
|
+
if (lastCompacting && Date.now() - lastCompacting < 30000) {
|
|
1174
|
+
logDebug("compactingHook summary dedup", { sessionId: pollSessionId });
|
|
1175
|
+
break;
|
|
1176
|
+
}
|
|
1177
|
+
compactingSummaryCooldown.set(summaryHash, Date.now());
|
|
1178
|
+
const prefixedSummary = `[Session Summary] ${summaryText}`;
|
|
795
1179
|
try {
|
|
796
|
-
const result = await client.ingestMessages([{ role: "user", content:
|
|
1180
|
+
const result = await client.ingestMessages([{ role: "user", content: prefixedSummary }], {
|
|
797
1181
|
mode: ingestMode,
|
|
798
1182
|
tags: [...containerTags, "auto-capture", "compact-summary"],
|
|
799
1183
|
sessionId: pollEffectiveSessionId,
|
|
@@ -949,10 +1333,10 @@ export function buildWhisperText(toolNames, maxToolNames) {
|
|
|
949
1333
|
return null;
|
|
950
1334
|
const lines = ["<cerebro-memory-activation>"];
|
|
951
1335
|
if (toolNames.length <= maxToolNames) {
|
|
952
|
-
lines.push(`
|
|
1336
|
+
lines.push(`Before using ${toolNames.join(", ")}, memory_search() may surface relevant past decisions or patterns. Brief recall → better outcomes.`);
|
|
953
1337
|
}
|
|
954
1338
|
else {
|
|
955
|
-
lines.push("Before you act —
|
|
1339
|
+
lines.push("Before you act — memory_search() surfaces cross-session context: past decisions, user preferences, hard-won insights. The strongest responses are built on remembered context.");
|
|
956
1340
|
}
|
|
957
1341
|
lines.push("</cerebro-memory-activation>");
|
|
958
1342
|
return lines.join("\n");
|