audrey 0.17.0 → 0.20.0
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/README.md +129 -374
- package/dist/mcp-server/config.d.ts +20 -0
- package/dist/mcp-server/config.d.ts.map +1 -0
- package/dist/mcp-server/config.js +125 -0
- package/dist/mcp-server/config.js.map +1 -0
- package/dist/mcp-server/index.d.ts +100 -0
- package/dist/mcp-server/index.d.ts.map +1 -0
- package/dist/mcp-server/index.js +1113 -0
- package/dist/mcp-server/index.js.map +1 -0
- package/dist/src/adaptive.d.ts +7 -0
- package/dist/src/adaptive.d.ts.map +1 -0
- package/dist/src/adaptive.js +49 -0
- package/dist/src/adaptive.js.map +1 -0
- package/dist/src/affect.d.ts +19 -0
- package/dist/src/affect.d.ts.map +1 -0
- package/dist/src/affect.js +72 -0
- package/dist/src/affect.js.map +1 -0
- package/dist/src/audrey.d.ts +140 -0
- package/dist/src/audrey.d.ts.map +1 -0
- package/dist/src/audrey.js +564 -0
- package/dist/src/audrey.js.map +1 -0
- package/dist/src/capsule.d.ts +68 -0
- package/dist/src/capsule.d.ts.map +1 -0
- package/dist/src/capsule.js +311 -0
- package/dist/src/capsule.js.map +1 -0
- package/dist/src/causal.d.ts +28 -0
- package/dist/src/causal.d.ts.map +1 -0
- package/dist/src/causal.js +65 -0
- package/dist/src/causal.js.map +1 -0
- package/dist/src/confidence.d.ts +12 -0
- package/dist/src/confidence.d.ts.map +1 -0
- package/dist/src/confidence.js +63 -0
- package/dist/src/confidence.js.map +1 -0
- package/dist/src/consolidate.d.ts +8 -0
- package/dist/src/consolidate.d.ts.map +1 -0
- package/dist/src/consolidate.js +218 -0
- package/dist/src/consolidate.js.map +1 -0
- package/dist/src/context.d.ts +3 -0
- package/dist/src/context.d.ts.map +1 -0
- package/dist/src/context.js +19 -0
- package/dist/src/context.js.map +1 -0
- package/dist/src/db.d.ts +12 -0
- package/dist/src/db.d.ts.map +1 -0
- package/dist/src/db.js +380 -0
- package/dist/src/db.js.map +1 -0
- package/dist/src/decay.d.ts +7 -0
- package/dist/src/decay.d.ts.map +1 -0
- package/dist/src/decay.js +68 -0
- package/dist/src/decay.js.map +1 -0
- package/dist/src/embedding.d.ts +57 -0
- package/dist/src/embedding.d.ts.map +1 -0
- package/dist/src/embedding.js +254 -0
- package/dist/src/embedding.js.map +1 -0
- package/dist/src/encode.d.ts +15 -0
- package/dist/src/encode.d.ts.map +1 -0
- package/dist/src/encode.js +36 -0
- package/dist/src/encode.js.map +1 -0
- package/dist/src/events.d.ts +69 -0
- package/dist/src/events.d.ts.map +1 -0
- package/dist/src/events.js +149 -0
- package/dist/src/events.js.map +1 -0
- package/dist/src/export.d.ts +3 -0
- package/dist/src/export.d.ts.map +1 -0
- package/dist/src/export.js +46 -0
- package/dist/src/export.js.map +1 -0
- package/dist/src/forget.d.ts +11 -0
- package/dist/src/forget.d.ts.map +1 -0
- package/dist/src/forget.js +105 -0
- package/dist/src/forget.js.map +1 -0
- package/dist/src/fts.d.ts +34 -0
- package/dist/src/fts.d.ts.map +1 -0
- package/dist/src/fts.js +117 -0
- package/dist/src/fts.js.map +1 -0
- package/dist/src/hybrid-recall.d.ts +37 -0
- package/dist/src/hybrid-recall.d.ts.map +1 -0
- package/dist/src/hybrid-recall.js +213 -0
- package/dist/src/hybrid-recall.js.map +1 -0
- package/dist/src/import.d.ts +4 -0
- package/dist/src/import.d.ts.map +1 -0
- package/dist/src/import.js +127 -0
- package/dist/src/import.js.map +1 -0
- package/dist/src/index.d.ts +22 -0
- package/dist/src/index.d.ts.map +1 -0
- package/{src → dist/src}/index.js +5 -13
- package/dist/src/index.js.map +1 -0
- package/dist/src/interference.d.ts +13 -0
- package/dist/src/interference.d.ts.map +1 -0
- package/dist/src/interference.js +45 -0
- package/dist/src/interference.js.map +1 -0
- package/dist/src/introspect.d.ts +4 -0
- package/dist/src/introspect.d.ts.map +1 -0
- package/dist/src/introspect.js +40 -0
- package/dist/src/introspect.js.map +1 -0
- package/dist/src/llm.d.ts +38 -0
- package/dist/src/llm.d.ts.map +1 -0
- package/dist/src/llm.js +167 -0
- package/dist/src/llm.js.map +1 -0
- package/dist/src/migrate.d.ts +6 -0
- package/dist/src/migrate.d.ts.map +1 -0
- package/dist/src/migrate.js +51 -0
- package/dist/src/migrate.js.map +1 -0
- package/dist/src/promote.d.ts +40 -0
- package/dist/src/promote.d.ts.map +1 -0
- package/dist/src/promote.js +200 -0
- package/dist/src/promote.js.map +1 -0
- package/dist/src/prompts.d.ts +16 -0
- package/dist/src/prompts.d.ts.map +1 -0
- package/{src → dist/src}/prompts.js +172 -203
- package/dist/src/prompts.js.map +1 -0
- package/dist/src/recall.d.ts +9 -0
- package/dist/src/recall.d.ts.map +1 -0
- package/dist/src/recall.js +432 -0
- package/dist/src/recall.js.map +1 -0
- package/dist/src/redact.d.ts +27 -0
- package/dist/src/redact.d.ts.map +1 -0
- package/dist/src/redact.js +228 -0
- package/dist/src/redact.js.map +1 -0
- package/dist/src/rollback.d.ts +8 -0
- package/dist/src/rollback.d.ts.map +1 -0
- package/dist/src/rollback.js +33 -0
- package/dist/src/rollback.js.map +1 -0
- package/dist/src/routes.d.ts +7 -0
- package/dist/src/routes.d.ts.map +1 -0
- package/dist/src/routes.js +226 -0
- package/dist/src/routes.js.map +1 -0
- package/dist/src/rules-compiler.d.ts +20 -0
- package/dist/src/rules-compiler.d.ts.map +1 -0
- package/dist/src/rules-compiler.js +143 -0
- package/dist/src/rules-compiler.js.map +1 -0
- package/dist/src/server.d.ts +12 -0
- package/dist/src/server.d.ts.map +1 -0
- package/dist/src/server.js +22 -0
- package/dist/src/server.js.map +1 -0
- package/dist/src/tool-trace.d.ts +37 -0
- package/dist/src/tool-trace.d.ts.map +1 -0
- package/dist/src/tool-trace.js +142 -0
- package/dist/src/tool-trace.js.map +1 -0
- package/dist/src/types.d.ts +446 -0
- package/dist/src/types.d.ts.map +1 -0
- package/dist/src/types.js +6 -0
- package/dist/src/types.js.map +1 -0
- package/dist/src/ulid.d.ts +3 -0
- package/dist/src/ulid.d.ts.map +1 -0
- package/dist/src/ulid.js +11 -0
- package/dist/src/ulid.js.map +1 -0
- package/dist/src/utils.d.ts +10 -0
- package/dist/src/utils.d.ts.map +1 -0
- package/dist/src/utils.js +41 -0
- package/dist/src/utils.js.map +1 -0
- package/dist/src/validate.d.ts +22 -0
- package/dist/src/validate.d.ts.map +1 -0
- package/dist/src/validate.js +109 -0
- package/dist/src/validate.js.map +1 -0
- package/docs/production-readiness.md +28 -0
- package/examples/fintech-ops-demo.js +1 -1
- package/examples/healthcare-ops-demo.js +1 -1
- package/examples/stripe-demo.js +1 -1
- package/package.json +34 -13
- package/benchmarks/baselines.js +0 -169
- package/benchmarks/cases.js +0 -421
- package/benchmarks/reference-results.js +0 -70
- package/benchmarks/report.js +0 -255
- package/benchmarks/run.js +0 -514
- package/mcp-server/config.js +0 -133
- package/mcp-server/index.js +0 -1265
- package/mcp-server/serve.js +0 -482
- package/src/adaptive.js +0 -53
- package/src/affect.js +0 -64
- package/src/audrey.js +0 -642
- package/src/causal.js +0 -95
- package/src/confidence.js +0 -120
- package/src/consolidate.js +0 -281
- package/src/context.js +0 -15
- package/src/db.js +0 -391
- package/src/decay.js +0 -84
- package/src/embedding.js +0 -260
- package/src/encode.js +0 -69
- package/src/export.js +0 -67
- package/src/forget.js +0 -111
- package/src/fts.js +0 -134
- package/src/import.js +0 -273
- package/src/interference.js +0 -51
- package/src/introspect.js +0 -48
- package/src/llm.js +0 -249
- package/src/migrate.js +0 -58
- package/src/recall.js +0 -573
- package/src/rollback.js +0 -42
- package/src/ulid.js +0 -18
- package/src/utils.js +0 -63
- package/src/validate.js +0 -172
- package/types/index.d.ts +0 -434
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"db.js","sourceRoot":"","sources":["../../src/db.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,KAAK,SAAS,MAAM,YAAY,CAAC;AACxC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAExD,MAAM,MAAM,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAqJd,CAAC;AAgCF,MAAM,UAAU,gBAAgB,CAAC,EAAqB,EAAE,UAAkB;IACxE,EAAE,CAAC,IAAI,CAAC;;;wBAGc,UAAU;;;;GAI/B,CAAC,CAAC;IACH,EAAE,CAAC,IAAI,CAAC;;;wBAGc,UAAU;;;GAG/B,CAAC,CAAC;IACH,EAAE,CAAC,IAAI,CAAC;;;wBAGc,UAAU;;;GAG/B,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,EAAqB;IAClD,EAAE,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;IAC7C,EAAE,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;IAC9C,EAAE,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;AACjD,CAAC;AAED,SAAS,YAAY,CAAC,EAAqB,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,YAAY,EAAE,SAAS,EAAE,UAAU,EAAuB;IAC/I,MAAM,KAAK,GAAI,EAAE,CAAC,OAAO,CAAC,6BAA6B,MAAM,EAAE,CAAC,CAAC,GAAG,EAAe,CAAC,CAAC,CAAC;IACtF,IAAI,KAAK,GAAG,CAAC;QAAE,OAAO;IAEtB,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,UAAU,UAAU,SAAS,MAAM,8BAA8B,CAAC,CAAC,GAAG,EAAoB,CAAC;IACnH,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IAE9B,MAAM,aAAa,GAAG,UAAU,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACzD,MAAM,MAAM,GAAG,EAAE,CAAC,OAAO,CAAC,eAAe,MAAM,IAAI,UAAU,aAAa,YAAY,GAAG,CAAC,CAAC;IAC3F,MAAM,EAAE,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;QAC7B,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,IAAI,aAAa,IAAI,GAAG,CAAC,SAAS,CAAC,UAAU,KAAK,aAAa;gBAAE,SAAS;YAC1E,MAAM,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CAAC,CAAC;IACH,EAAE,EAAE,CAAC;AACP,CAAC;AAED,SAAS,uBAAuB,CAAC,EAAqB,EAAE,UAAkB;IACxE,YAAY,CAAC,EAAE,EAAE;QACf,MAAM,EAAE,UAAU;QAClB,MAAM,EAAE,cAAc;QACtB,UAAU,EAAE,qCAAqC;QACjD,UAAU,EAAE,qCAAqC;QACjD,YAAY,EAAE,YAAY;QAC1B,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,CAAC,CAAC;QACtF,UAAU;KACX,CAAC,CAAC;IAEH,YAAY,CAAC,EAAE,EAAE;QACf,MAAM,EAAE,WAAW;QACnB,MAAM,EAAE,eAAe;QACvB,UAAU,EAAE,sBAAsB;QAClC,UAAU,EAAE,sBAAsB;QAClC,YAAY,EAAE,SAAS;QACvB,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,KAAK,CAAC;QACtD,UAAU;KACX,CAAC,CAAC;IAEH,YAAY,CAAC,EAAE,EAAE;QACf,MAAM,EAAE,YAAY;QACpB,MAAM,EAAE,gBAAgB;QACxB,UAAU,EAAE,sBAAsB;QAClC,UAAU,EAAE,sBAAsB;QAClC,YAAY,EAAE,SAAS;QACvB,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,KAAK,CAAC;QACtD,UAAU;KACX,CAAC,CAAC;AACL,CAAC;AAWD,SAAS,sBAAsB,CAAC,EAAqB;IACnD,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,aAAa,GAAG,CAAC,CAAC;IAEtB,IAAI,CAAC;QACH,WAAW,GAAI,EAAE,CAAC,OAAO,CAAC,wCAAwC,CAAC,CAAC,GAAG,EAAe,CAAC,CAAC,CAAC;QACzF,YAAY,GAAI,EAAE,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC,GAAG,EAAe,CAAC,CAAC,CAAC;QAC3F,aAAa,GAAI,EAAE,CAAC,OAAO,CAAC,0CAA0C,CAAC,CAAC,GAAG,EAAe,CAAC,CAAC,CAAC;IAC/F,CAAC;IAAC,MAAM,CAAC;QACP,+BAA+B;IACjC,CAAC;IAED,MAAM,QAAQ,GAAI,EAAE,CAAC,OAAO,CAAC,gEAAgE,CAAC,CAAC,GAAG,EAAe,CAAC,CAAC,CAAC;IACpH,MAAM,SAAS,GAAI,EAAE,CAAC,OAAO,CAAC,iEAAiE,CAAC,CAAC,GAAG,EAAe,CAAC,CAAC,CAAC;IACtH,MAAM,UAAU,GAAI,EAAE,CAAC,OAAO,CAAC,kEAAkE,CAAC,CAAC,GAAG,EAAe,CAAC,CAAC,CAAC;IAExH,OAAO;QACL,QAAQ;QACR,WAAW;QACX,SAAS;QACT,YAAY;QACZ,UAAU;QACV,aAAa;KACd,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,EAAqB,EAAE,KAAa,EAAE,MAAc,EAAE,UAAkB;IAClG,MAAM,OAAO,GAAG,EAAE,CAAC,MAAM,CAAC,cAAc,KAAK,GAAG,CAAmB,CAAC;IACpE,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;IACxD,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,EAAE,CAAC,IAAI,CAAC,eAAe,KAAK,eAAe,MAAM,IAAI,UAAU,EAAE,CAAC,CAAC;IACrE,CAAC;AACH,CAAC;AAED,MAAM,cAAc,GAAG,EAAE,CAAC;AAE1B,MAAM,UAAU,GAA2D;IACzE,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,kBAAkB,CAAC,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,mBAAmB,CAAC,CAAC,CAAC,CAAC,EAAE;IAC9F,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,kBAAkB,CAAC,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,mBAAmB,CAAC,CAAC,CAAC,CAAC,EAAE;IAC7F,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,kBAAkB,CAAC,EAAE,EAAE,WAAW,EAAE,oBAAoB,EAAE,mBAAmB,CAAC,CAAC,CAAC,CAAC,EAAE;IAC1G,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,kBAAkB,CAAC,EAAE,EAAE,WAAW,EAAE,UAAU,EAAE,kBAAkB,CAAC,CAAC,CAAC,CAAC,EAAE;IAC/F,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,kBAAkB,CAAC,EAAE,EAAE,YAAY,EAAE,oBAAoB,EAAE,mBAAmB,CAAC,CAAC,CAAC,CAAC,EAAE;IAC3G,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,kBAAkB,CAAC,EAAE,EAAE,YAAY,EAAE,UAAU,EAAE,kBAAkB,CAAC,CAAC,CAAC,CAAC,EAAE;IAChG,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,kBAAkB,CAAC,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,mBAAmB,CAAC,CAAC,CAAC,CAAC,EAAE;IAC9F,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE;YACjB,kBAAkB,CAAC,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE,wBAAwB,CAAC,CAAC;YACtE,kBAAkB,CAAC,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,wBAAwB,CAAC,CAAC;YACvE,kBAAkB,CAAC,EAAE,EAAE,YAAY,EAAE,OAAO,EAAE,wBAAwB,CAAC,CAAC;YACxE,EAAE,CAAC,IAAI,CAAC,kEAAkE,CAAC,CAAC;YAC5E,EAAE,CAAC,IAAI,CAAC,oEAAoE,CAAC,CAAC;YAC9E,EAAE,CAAC,IAAI,CAAC,sEAAsE,CAAC,CAAC;QAClF,CAAC,EAAC;IACF,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE;YACjB,eAAe,CAAC,EAAE,CAAC,CAAC;YACpB,WAAW,CAAC,EAAE,CAAC,CAAC;QAClB,CAAC,EAAC;IACF,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE;YAClB,kBAAkB,CAAC,EAAE,EAAE,UAAU,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAAC;YACvE,kBAAkB,CAAC,EAAE,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,CAAC,CAAC;YAC3D,kBAAkB,CAAC,EAAE,EAAE,WAAW,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAAC;YACxE,kBAAkB,CAAC,EAAE,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,CAAC,CAAC;YAC5D,kBAAkB,CAAC,EAAE,EAAE,YAAY,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAAC;YACzE,kBAAkB,CAAC,EAAE,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,CAAC,CAAC;QAC/D,CAAC,EAAC;IACF,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CAAC,GAAG;YACnB,oEAAoE;YACpE,0EAA0E;YAC1E,wEAAwE;QAC1E,CAAC,EAAC;CACH,CAAC;AAEF,SAAS,aAAa,CAAC,EAAqB;IAC1C,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,8DAA8D,CAAC,CAAC,GAAG,EAA2B,CAAC;IACtH,MAAM,cAAc,GAAG,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAEnD,IAAI,cAAc,IAAI,cAAc;QAAE,OAAO;IAE7C,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,GAAG,cAAc,CAAC,CAAC;IACnE,KAAK,MAAM,SAAS,IAAI,OAAO,EAAE,CAAC;QAChC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IACnB,CAAC;IAED,EAAE,CAAC,OAAO,CACR;2DACuD,CACxD,CAAC,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,cAAc,CAC5B,OAAe,EACf,UAAmC,EAAE;IAErC,IAAI,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;IAC7B,IAAI,QAAQ,GAAG,KAAK,CAAC;IAErB,SAAS,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACxC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAC1C,MAAM,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;IAChC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;IAChC,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;IAC/B,EAAE,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC;IACjC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAChB,aAAa,CAAC,EAAE,CAAC,CAAC;IAElB,IAAI,UAAU,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,EAAE,CAAC,OAAO,CAAC,0DAA0D,CAAC,CAAC,GAAG,EAA2B,CAAC;QACrH,IAAI,MAAM,EAAE,CAAC;YACX,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,IAAI,UAAU,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,+CAA+C,UAAU,EAAE,CAAC,CAAC;QAC/E,CAAC;QAED,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEnB,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CACzB,0DAA0D,CAC3D,CAAC,GAAG,EAA2B,CAAC;QAEjC,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,UAAU,GAAG,QAAQ,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YAChD,IAAI,UAAU,KAAK,UAAU,EAAE,CAAC;gBAC9B,cAAc,CAAC,EAAE,CAAC,CAAC;gBACnB,EAAE,CAAC,OAAO,CACR,6DAA6D,CAC9D,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;gBAC1B,QAAQ,GAAG,IAAI,CAAC;YAClB,CAAC;QACH,CAAC;aAAM,CAAC;YACN,EAAE,CAAC,OAAO,CACR,iEAAiE,CAClE,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;QAC5B,CAAC;QAED,gBAAgB,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QAEjC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,uBAAuB,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YACxC,MAAM,IAAI,GAAG,sBAAsB,CAAC,EAAE,CAAC,CAAC;YACxC,IACE,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,WAAW;mBAC/B,IAAI,CAAC,SAAS,KAAK,IAAI,CAAC,YAAY;mBACpC,IAAI,CAAC,UAAU,KAAK,IAAI,CAAC,aAAa,EACzC,CAAC;gBACD,QAAQ,GAAG,IAAI,CAAC;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,OAAe;IAClD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAC1C,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;QAAE,OAAO,IAAI,CAAC;IACrC,MAAM,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IACpD,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,0DAA0D,CAAC,CAAC,GAAG,EAA2B,CAAC;QAClH,OAAO,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAC9C,CAAC;IAAC,OAAO,GAAY,EAAE,CAAC;QACtB,IAAI,GAAG,YAAY,KAAK,IAAI,GAAG,CAAC,OAAO,EAAE,QAAQ,CAAC,eAAe,CAAC;YAAE,OAAO,IAAI,CAAC;QAChF,MAAM,GAAG,CAAC;IACZ,CAAC;YAAS,CAAC;QACT,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,EAAqB;IACjD,IAAI,EAAE,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;QAClB,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import Database from 'better-sqlite3';
|
|
2
|
+
import type { DecayResult, HalfLives } from './types.js';
|
|
3
|
+
export declare function applyDecay(db: Database.Database, { dormantThreshold, halfLives }?: {
|
|
4
|
+
dormantThreshold?: number;
|
|
5
|
+
halfLives?: Partial<HalfLives>;
|
|
6
|
+
}): DecayResult;
|
|
7
|
+
//# sourceMappingURL=decay.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"decay.d.ts","sourceRoot":"","sources":["../../src/decay.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AA2BzD,wBAAgB,UAAU,CACxB,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,EAAE,gBAAsB,EAAE,SAAS,EAAE,GAAE;IAAE,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAA;CAAO,GACxG,WAAW,CA0Eb"}
|
|
@@ -0,0 +1,68 @@
|
|
|
1
|
+
import { computeConfidence, DEFAULT_HALF_LIVES, salienceModifier } from './confidence.js';
|
|
2
|
+
import { interferenceModifier } from './interference.js';
|
|
3
|
+
import { daysBetween } from './utils.js';
|
|
4
|
+
export function applyDecay(db, { dormantThreshold = 0.1, halfLives } = {}) {
|
|
5
|
+
const now = new Date();
|
|
6
|
+
let totalEvaluated = 0;
|
|
7
|
+
let transitionedToDormant = 0;
|
|
8
|
+
const semantics = db.prepare(`
|
|
9
|
+
SELECT id, supporting_count, contradicting_count, created_at,
|
|
10
|
+
last_reinforced_at, retrieval_count, interference_count, salience
|
|
11
|
+
FROM semantics WHERE state = 'active'
|
|
12
|
+
`).all();
|
|
13
|
+
const markDormantSem = db.prepare('UPDATE semantics SET state = ? WHERE id = ?');
|
|
14
|
+
for (const sem of semantics) {
|
|
15
|
+
totalEvaluated++;
|
|
16
|
+
const ageDays = daysBetween(sem.created_at, now);
|
|
17
|
+
const daysSinceRetrieval = sem.last_reinforced_at
|
|
18
|
+
? daysBetween(sem.last_reinforced_at, now)
|
|
19
|
+
: ageDays;
|
|
20
|
+
let confidence = computeConfidence({
|
|
21
|
+
sourceType: 'tool-result',
|
|
22
|
+
supportingCount: sem.supporting_count || 0,
|
|
23
|
+
contradictingCount: sem.contradicting_count || 0,
|
|
24
|
+
ageDays,
|
|
25
|
+
halfLifeDays: halfLives?.semantic ?? DEFAULT_HALF_LIVES.semantic,
|
|
26
|
+
retrievalCount: sem.retrieval_count || 0,
|
|
27
|
+
daysSinceRetrieval,
|
|
28
|
+
});
|
|
29
|
+
confidence *= interferenceModifier(sem.interference_count || 0);
|
|
30
|
+
confidence *= salienceModifier(sem.salience ?? 0.5);
|
|
31
|
+
confidence = Math.max(0, Math.min(1, confidence));
|
|
32
|
+
if (confidence < dormantThreshold) {
|
|
33
|
+
markDormantSem.run('dormant', sem.id);
|
|
34
|
+
transitionedToDormant++;
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
const procedures = db.prepare(`
|
|
38
|
+
SELECT id, success_count, failure_count, created_at,
|
|
39
|
+
last_reinforced_at, retrieval_count, interference_count, salience
|
|
40
|
+
FROM procedures WHERE state = 'active'
|
|
41
|
+
`).all();
|
|
42
|
+
const markDormantProc = db.prepare('UPDATE procedures SET state = ? WHERE id = ?');
|
|
43
|
+
for (const proc of procedures) {
|
|
44
|
+
totalEvaluated++;
|
|
45
|
+
const ageDays = daysBetween(proc.created_at, now);
|
|
46
|
+
const daysSinceRetrieval = proc.last_reinforced_at
|
|
47
|
+
? daysBetween(proc.last_reinforced_at, now)
|
|
48
|
+
: ageDays;
|
|
49
|
+
let confidence = computeConfidence({
|
|
50
|
+
sourceType: 'tool-result',
|
|
51
|
+
supportingCount: proc.success_count || 0,
|
|
52
|
+
contradictingCount: proc.failure_count || 0,
|
|
53
|
+
ageDays,
|
|
54
|
+
halfLifeDays: halfLives?.procedural ?? DEFAULT_HALF_LIVES.procedural,
|
|
55
|
+
retrievalCount: proc.retrieval_count || 0,
|
|
56
|
+
daysSinceRetrieval,
|
|
57
|
+
});
|
|
58
|
+
confidence *= interferenceModifier(proc.interference_count || 0);
|
|
59
|
+
confidence *= salienceModifier(proc.salience ?? 0.5);
|
|
60
|
+
confidence = Math.max(0, Math.min(1, confidence));
|
|
61
|
+
if (confidence < dormantThreshold) {
|
|
62
|
+
markDormantProc.run('dormant', proc.id);
|
|
63
|
+
transitionedToDormant++;
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
return { totalEvaluated, transitionedToDormant, timestamp: now.toISOString() };
|
|
67
|
+
}
|
|
68
|
+
//# sourceMappingURL=decay.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"decay.js","sourceRoot":"","sources":["../../src/decay.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAC1F,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAwBzC,MAAM,UAAU,UAAU,CACxB,EAAqB,EACrB,EAAE,gBAAgB,GAAG,GAAG,EAAE,SAAS,KAAoE,EAAE;IAEzG,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IACvB,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,IAAI,qBAAqB,GAAG,CAAC,CAAC;IAE9B,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,CAAC;;;;GAI5B,CAAC,CAAC,GAAG,EAAwB,CAAC;IAE/B,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC,6CAA6C,CAAC,CAAC;IAEjF,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;QAC5B,cAAc,EAAE,CAAC;QACjB,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;QACjD,MAAM,kBAAkB,GAAG,GAAG,CAAC,kBAAkB;YAC/C,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,kBAAkB,EAAE,GAAG,CAAC;YAC1C,CAAC,CAAC,OAAO,CAAC;QAEZ,IAAI,UAAU,GAAG,iBAAiB,CAAC;YACjC,UAAU,EAAE,aAAa;YACzB,eAAe,EAAE,GAAG,CAAC,gBAAgB,IAAI,CAAC;YAC1C,kBAAkB,EAAE,GAAG,CAAC,mBAAmB,IAAI,CAAC;YAChD,OAAO;YACP,YAAY,EAAE,SAAS,EAAE,QAAQ,IAAI,kBAAkB,CAAC,QAAQ;YAChE,cAAc,EAAE,GAAG,CAAC,eAAe,IAAI,CAAC;YACxC,kBAAkB;SACnB,CAAC,CAAC;QACH,UAAU,IAAI,oBAAoB,CAAC,GAAG,CAAC,kBAAkB,IAAI,CAAC,CAAC,CAAC;QAChE,UAAU,IAAI,gBAAgB,CAAC,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,CAAC;QACpD,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC;QAElD,IAAI,UAAU,GAAG,gBAAgB,EAAE,CAAC;YAClC,cAAc,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;YACtC,qBAAqB,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC;;;;GAI7B,CAAC,CAAC,GAAG,EAA0B,CAAC;IAEjC,MAAM,eAAe,GAAG,EAAE,CAAC,OAAO,CAAC,8CAA8C,CAAC,CAAC;IAEnF,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,cAAc,EAAE,CAAC;QACjB,MAAM,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;QAClD,MAAM,kBAAkB,GAAG,IAAI,CAAC,kBAAkB;YAChD,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,CAAC;YAC3C,CAAC,CAAC,OAAO,CAAC;QAEZ,IAAI,UAAU,GAAG,iBAAiB,CAAC;YACjC,UAAU,EAAE,aAAa;YACzB,eAAe,EAAE,IAAI,CAAC,aAAa,IAAI,CAAC;YACxC,kBAAkB,EAAE,IAAI,CAAC,aAAa,IAAI,CAAC;YAC3C,OAAO;YACP,YAAY,EAAE,SAAS,EAAE,UAAU,IAAI,kBAAkB,CAAC,UAAU;YACpE,cAAc,EAAE,IAAI,CAAC,eAAe,IAAI,CAAC;YACzC,kBAAkB;SACnB,CAAC,CAAC;QACH,UAAU,IAAI,oBAAoB,CAAC,IAAI,CAAC,kBAAkB,IAAI,CAAC,CAAC,CAAC;QACjE,UAAU,IAAI,gBAAgB,CAAC,IAAI,CAAC,QAAQ,IAAI,GAAG,CAAC,CAAC;QACrD,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC;QAElD,IAAI,UAAU,GAAG,gBAAgB,EAAE,CAAC;YAClC,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;YACxC,qBAAqB,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,OAAO,EAAE,cAAc,EAAE,qBAAqB,EAAE,SAAS,EAAE,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC;AACjF,CAAC"}
|
|
@@ -0,0 +1,57 @@
|
|
|
1
|
+
import type { EmbeddingConfig, EmbeddingProvider } from './types.js';
|
|
2
|
+
export declare class MockEmbeddingProvider implements EmbeddingProvider {
|
|
3
|
+
dimensions: number;
|
|
4
|
+
modelName: string;
|
|
5
|
+
modelVersion: string;
|
|
6
|
+
constructor({ dimensions }?: Partial<EmbeddingConfig>);
|
|
7
|
+
embed(text: string): Promise<number[]>;
|
|
8
|
+
embedBatch(texts: string[]): Promise<number[][]>;
|
|
9
|
+
vectorToBuffer(vector: number[]): Buffer;
|
|
10
|
+
bufferToVector(buffer: Buffer): number[];
|
|
11
|
+
}
|
|
12
|
+
export declare class OpenAIEmbeddingProvider implements EmbeddingProvider {
|
|
13
|
+
apiKey: string | undefined;
|
|
14
|
+
model: string;
|
|
15
|
+
dimensions: number;
|
|
16
|
+
timeout: number;
|
|
17
|
+
modelName: string;
|
|
18
|
+
modelVersion: string;
|
|
19
|
+
constructor({ apiKey, model, dimensions, timeout }?: Partial<EmbeddingConfig>);
|
|
20
|
+
embed(text: string): Promise<number[]>;
|
|
21
|
+
embedBatch(texts: string[]): Promise<number[][]>;
|
|
22
|
+
vectorToBuffer(vector: number[]): Buffer;
|
|
23
|
+
bufferToVector(buffer: Buffer): number[];
|
|
24
|
+
}
|
|
25
|
+
export declare class LocalEmbeddingProvider implements EmbeddingProvider {
|
|
26
|
+
model: string;
|
|
27
|
+
dimensions: number;
|
|
28
|
+
modelName: string;
|
|
29
|
+
modelVersion: string;
|
|
30
|
+
device: string;
|
|
31
|
+
batchSize: number;
|
|
32
|
+
pipelineFactory: ((task: string, model: string, options?: Record<string, unknown>) => Promise<unknown>) | null;
|
|
33
|
+
_pipeline: any;
|
|
34
|
+
_readyPromise: Promise<void> | null;
|
|
35
|
+
_actualDevice: string | null;
|
|
36
|
+
constructor({ model, device, batchSize, pipelineFactory }?: Partial<EmbeddingConfig>);
|
|
37
|
+
ready(): Promise<void>;
|
|
38
|
+
embed(text: string): Promise<number[]>;
|
|
39
|
+
embedBatch(texts: string[]): Promise<number[][]>;
|
|
40
|
+
vectorToBuffer(vector: number[]): Buffer;
|
|
41
|
+
bufferToVector(buffer: Buffer): number[];
|
|
42
|
+
}
|
|
43
|
+
export declare class GeminiEmbeddingProvider implements EmbeddingProvider {
|
|
44
|
+
apiKey: string | undefined;
|
|
45
|
+
model: string;
|
|
46
|
+
dimensions: number;
|
|
47
|
+
timeout: number;
|
|
48
|
+
modelName: string;
|
|
49
|
+
modelVersion: string;
|
|
50
|
+
constructor({ apiKey, model, timeout }?: Partial<EmbeddingConfig>);
|
|
51
|
+
embed(text: string): Promise<number[]>;
|
|
52
|
+
embedBatch(texts: string[]): Promise<number[][]>;
|
|
53
|
+
vectorToBuffer(vector: number[]): Buffer;
|
|
54
|
+
bufferToVector(buffer: Buffer): number[];
|
|
55
|
+
}
|
|
56
|
+
export declare function createEmbeddingProvider(config: EmbeddingConfig): EmbeddingProvider;
|
|
57
|
+
//# sourceMappingURL=embedding.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embedding.d.ts","sourceRoot":"","sources":["../../src/embedding.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAGrE,qBAAa,qBAAsB,YAAW,iBAAiB;IAC7D,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;gBAET,EAAE,UAAe,EAAE,GAAE,OAAO,CAAC,eAAe,CAAM;IAMxD,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAUtC,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAItD,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM;IAIxC,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE;CAGzC;AAED,qBAAa,uBAAwB,YAAW,iBAAiB;IAC/D,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;gBAET,EAAE,MAAM,EAAE,KAAgC,EAAE,UAAiB,EAAE,OAAe,EAAE,GAAE,OAAO,CAAC,eAAe,CAAM;IASrH,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAsBtC,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAsBtD,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM;IAIxC,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE;CAGzC;AAED,qBAAa,sBAAuB,YAAW,iBAAiB;IAC9D,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,EAAE,CAAC,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAC/G,SAAS,EAAE,GAAG,CAAC;IACf,aAAa,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACpC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;gBAEjB,EAAE,KAAiC,EAAE,MAAc,EAAE,SAAc,EAAE,eAAsB,EAAE,GAAE,OAAO,CAAC,eAAe,CAAM;IAaxI,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAoBhB,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAMtC,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAYtD,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM;IAIxC,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE;CAGzC;AAED,qBAAa,uBAAwB,YAAW,iBAAiB;IAC/D,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;gBAET,EAAE,MAAM,EAAE,KAA8B,EAAE,OAAe,EAAE,GAAE,OAAO,CAAC,eAAe,CAAM;IAShG,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAsBtC,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAiCtD,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM;IAIxC,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE;CAGzC;AAED,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,eAAe,GAAG,iBAAiB,CAalF"}
|
|
@@ -0,0 +1,254 @@
|
|
|
1
|
+
import { createHash } from 'node:crypto';
|
|
2
|
+
import { describeHttpError, requireApiKey } from './utils.js';
|
|
3
|
+
export class MockEmbeddingProvider {
|
|
4
|
+
dimensions;
|
|
5
|
+
modelName;
|
|
6
|
+
modelVersion;
|
|
7
|
+
constructor({ dimensions = 64 } = {}) {
|
|
8
|
+
this.dimensions = dimensions ?? 64;
|
|
9
|
+
this.modelName = 'mock-embedding';
|
|
10
|
+
this.modelVersion = '1.0.0';
|
|
11
|
+
}
|
|
12
|
+
async embed(text) {
|
|
13
|
+
const hash = createHash('sha256').update(text).digest();
|
|
14
|
+
const vector = new Array(this.dimensions);
|
|
15
|
+
for (let i = 0; i < this.dimensions; i++) {
|
|
16
|
+
vector[i] = (hash[i % hash.length] / 255) * 2 - 1;
|
|
17
|
+
}
|
|
18
|
+
const magnitude = Math.sqrt(vector.reduce((sum, v) => sum + v * v, 0));
|
|
19
|
+
return vector.map(v => v / magnitude);
|
|
20
|
+
}
|
|
21
|
+
async embedBatch(texts) {
|
|
22
|
+
return Promise.all(texts.map(t => this.embed(t)));
|
|
23
|
+
}
|
|
24
|
+
vectorToBuffer(vector) {
|
|
25
|
+
return Buffer.from(new Float32Array(vector).buffer);
|
|
26
|
+
}
|
|
27
|
+
bufferToVector(buffer) {
|
|
28
|
+
return Array.from(new Float32Array(buffer.buffer, buffer.byteOffset, buffer.byteLength / 4));
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
export class OpenAIEmbeddingProvider {
|
|
32
|
+
apiKey;
|
|
33
|
+
model;
|
|
34
|
+
dimensions;
|
|
35
|
+
timeout;
|
|
36
|
+
modelName;
|
|
37
|
+
modelVersion;
|
|
38
|
+
constructor({ apiKey, model = 'text-embedding-3-small', dimensions = 1536, timeout = 30000 } = {}) {
|
|
39
|
+
this.apiKey = apiKey || process.env.OPENAI_API_KEY;
|
|
40
|
+
this.model = model ?? 'text-embedding-3-small';
|
|
41
|
+
this.dimensions = dimensions ?? 1536;
|
|
42
|
+
this.timeout = timeout ?? 30000;
|
|
43
|
+
this.modelName = this.model;
|
|
44
|
+
this.modelVersion = 'latest';
|
|
45
|
+
}
|
|
46
|
+
async embed(text) {
|
|
47
|
+
requireApiKey(this.apiKey, 'OpenAI embedding', 'OPENAI_API_KEY');
|
|
48
|
+
const controller = new AbortController();
|
|
49
|
+
const timer = setTimeout(() => controller.abort(), this.timeout);
|
|
50
|
+
try {
|
|
51
|
+
const response = await fetch('https://api.openai.com/v1/embeddings', {
|
|
52
|
+
method: 'POST',
|
|
53
|
+
headers: {
|
|
54
|
+
'Authorization': `Bearer ${this.apiKey}`,
|
|
55
|
+
'Content-Type': 'application/json',
|
|
56
|
+
},
|
|
57
|
+
body: JSON.stringify({ input: text, model: this.model, dimensions: this.dimensions }),
|
|
58
|
+
signal: controller.signal,
|
|
59
|
+
});
|
|
60
|
+
if (!response.ok)
|
|
61
|
+
throw new Error(`OpenAI embedding failed: ${await describeHttpError(response)}`);
|
|
62
|
+
const data = await response.json();
|
|
63
|
+
return data.data[0].embedding;
|
|
64
|
+
}
|
|
65
|
+
finally {
|
|
66
|
+
clearTimeout(timer);
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
async embedBatch(texts) {
|
|
70
|
+
requireApiKey(this.apiKey, 'OpenAI embedding', 'OPENAI_API_KEY');
|
|
71
|
+
const controller = new AbortController();
|
|
72
|
+
const timer = setTimeout(() => controller.abort(), this.timeout);
|
|
73
|
+
try {
|
|
74
|
+
const response = await fetch('https://api.openai.com/v1/embeddings', {
|
|
75
|
+
method: 'POST',
|
|
76
|
+
headers: {
|
|
77
|
+
'Authorization': `Bearer ${this.apiKey}`,
|
|
78
|
+
'Content-Type': 'application/json',
|
|
79
|
+
},
|
|
80
|
+
body: JSON.stringify({ input: texts, model: this.model, dimensions: this.dimensions }),
|
|
81
|
+
signal: controller.signal,
|
|
82
|
+
});
|
|
83
|
+
if (!response.ok)
|
|
84
|
+
throw new Error(`OpenAI embedding failed: ${await describeHttpError(response)}`);
|
|
85
|
+
const data = await response.json();
|
|
86
|
+
return data.data.map(d => d.embedding);
|
|
87
|
+
}
|
|
88
|
+
finally {
|
|
89
|
+
clearTimeout(timer);
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
vectorToBuffer(vector) {
|
|
93
|
+
return Buffer.from(new Float32Array(vector).buffer);
|
|
94
|
+
}
|
|
95
|
+
bufferToVector(buffer) {
|
|
96
|
+
return Array.from(new Float32Array(buffer.buffer, buffer.byteOffset, buffer.byteLength / 4));
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
export class LocalEmbeddingProvider {
|
|
100
|
+
model;
|
|
101
|
+
dimensions;
|
|
102
|
+
modelName;
|
|
103
|
+
modelVersion;
|
|
104
|
+
device;
|
|
105
|
+
batchSize;
|
|
106
|
+
pipelineFactory;
|
|
107
|
+
_pipeline; // eslint-disable-line @typescript-eslint/no-explicit-any
|
|
108
|
+
_readyPromise;
|
|
109
|
+
_actualDevice;
|
|
110
|
+
constructor({ model = 'Xenova/all-MiniLM-L6-v2', device = 'gpu', batchSize = 64, pipelineFactory = null } = {}) {
|
|
111
|
+
this.model = model ?? 'Xenova/all-MiniLM-L6-v2';
|
|
112
|
+
this.dimensions = 384;
|
|
113
|
+
this.modelName = this.model;
|
|
114
|
+
this.modelVersion = '1.0.0';
|
|
115
|
+
this.device = device ?? 'gpu';
|
|
116
|
+
this.batchSize = batchSize ?? 64;
|
|
117
|
+
this.pipelineFactory = pipelineFactory ?? null;
|
|
118
|
+
this._pipeline = null;
|
|
119
|
+
this._readyPromise = null;
|
|
120
|
+
this._actualDevice = null;
|
|
121
|
+
}
|
|
122
|
+
ready() {
|
|
123
|
+
if (!this._readyPromise) {
|
|
124
|
+
this._readyPromise = (async () => {
|
|
125
|
+
const pipeline = this.pipelineFactory || (await import('@huggingface/transformers')).pipeline;
|
|
126
|
+
try {
|
|
127
|
+
this._pipeline = await pipeline('feature-extraction', this.model, {
|
|
128
|
+
dtype: 'fp32', device: this.device,
|
|
129
|
+
});
|
|
130
|
+
this._actualDevice = this.device;
|
|
131
|
+
}
|
|
132
|
+
catch {
|
|
133
|
+
this._pipeline = await pipeline('feature-extraction', this.model, {
|
|
134
|
+
dtype: 'fp32', device: 'cpu',
|
|
135
|
+
});
|
|
136
|
+
this._actualDevice = 'cpu';
|
|
137
|
+
}
|
|
138
|
+
})();
|
|
139
|
+
}
|
|
140
|
+
return this._readyPromise;
|
|
141
|
+
}
|
|
142
|
+
async embed(text) {
|
|
143
|
+
await this.ready();
|
|
144
|
+
const output = await this._pipeline(text, { pooling: 'mean', normalize: true });
|
|
145
|
+
return Array.from(output.data);
|
|
146
|
+
}
|
|
147
|
+
async embedBatch(texts) {
|
|
148
|
+
if (texts.length === 0)
|
|
149
|
+
return [];
|
|
150
|
+
await this.ready();
|
|
151
|
+
const results = [];
|
|
152
|
+
for (let i = 0; i < texts.length; i += this.batchSize) {
|
|
153
|
+
const chunk = texts.slice(i, i + this.batchSize);
|
|
154
|
+
const output = await this._pipeline(chunk, { pooling: 'mean', normalize: true });
|
|
155
|
+
results.push(...output.tolist());
|
|
156
|
+
}
|
|
157
|
+
return results;
|
|
158
|
+
}
|
|
159
|
+
vectorToBuffer(vector) {
|
|
160
|
+
return Buffer.from(new Float32Array(vector).buffer);
|
|
161
|
+
}
|
|
162
|
+
bufferToVector(buffer) {
|
|
163
|
+
return Array.from(new Float32Array(buffer.buffer, buffer.byteOffset, buffer.byteLength / 4));
|
|
164
|
+
}
|
|
165
|
+
}
|
|
166
|
+
export class GeminiEmbeddingProvider {
|
|
167
|
+
apiKey;
|
|
168
|
+
model;
|
|
169
|
+
dimensions;
|
|
170
|
+
timeout;
|
|
171
|
+
modelName;
|
|
172
|
+
modelVersion;
|
|
173
|
+
constructor({ apiKey, model = 'gemini-embedding-001', timeout = 30000 } = {}) {
|
|
174
|
+
this.apiKey = apiKey || process.env.GOOGLE_API_KEY;
|
|
175
|
+
this.model = model ?? 'gemini-embedding-001';
|
|
176
|
+
this.dimensions = 3072;
|
|
177
|
+
this.timeout = timeout ?? 30000;
|
|
178
|
+
this.modelName = this.model;
|
|
179
|
+
this.modelVersion = 'latest';
|
|
180
|
+
}
|
|
181
|
+
async embed(text) {
|
|
182
|
+
requireApiKey(this.apiKey, 'Gemini embedding', 'GOOGLE_API_KEY');
|
|
183
|
+
const controller = new AbortController();
|
|
184
|
+
const timer = setTimeout(() => controller.abort(), this.timeout);
|
|
185
|
+
try {
|
|
186
|
+
const response = await fetch(`https://generativelanguage.googleapis.com/v1beta/models/${this.model}:embedContent`, {
|
|
187
|
+
method: 'POST',
|
|
188
|
+
headers: { 'Content-Type': 'application/json', 'x-goog-api-key': this.apiKey },
|
|
189
|
+
body: JSON.stringify({ model: `models/${this.model}`, content: { parts: [{ text }] } }),
|
|
190
|
+
signal: controller.signal,
|
|
191
|
+
});
|
|
192
|
+
if (!response.ok)
|
|
193
|
+
throw new Error(`Gemini embedding failed: ${await describeHttpError(response)}`);
|
|
194
|
+
const data = await response.json();
|
|
195
|
+
return data.embedding.values;
|
|
196
|
+
}
|
|
197
|
+
finally {
|
|
198
|
+
clearTimeout(timer);
|
|
199
|
+
}
|
|
200
|
+
}
|
|
201
|
+
async embedBatch(texts) {
|
|
202
|
+
if (texts.length === 0)
|
|
203
|
+
return [];
|
|
204
|
+
requireApiKey(this.apiKey, 'Gemini embedding', 'GOOGLE_API_KEY');
|
|
205
|
+
const results = [];
|
|
206
|
+
for (let i = 0; i < texts.length; i += 100) {
|
|
207
|
+
const chunk = texts.slice(i, i + 100);
|
|
208
|
+
const controller = new AbortController();
|
|
209
|
+
const timer = setTimeout(() => controller.abort(), this.timeout);
|
|
210
|
+
try {
|
|
211
|
+
const response = await fetch(`https://generativelanguage.googleapis.com/v1beta/models/${this.model}:batchEmbedContents`, {
|
|
212
|
+
method: 'POST',
|
|
213
|
+
headers: { 'Content-Type': 'application/json', 'x-goog-api-key': this.apiKey },
|
|
214
|
+
body: JSON.stringify({
|
|
215
|
+
requests: chunk.map(text => ({
|
|
216
|
+
model: `models/${this.model}`,
|
|
217
|
+
content: { parts: [{ text }] },
|
|
218
|
+
})),
|
|
219
|
+
}),
|
|
220
|
+
signal: controller.signal,
|
|
221
|
+
});
|
|
222
|
+
if (!response.ok)
|
|
223
|
+
throw new Error(`Gemini batch embedding failed: ${await describeHttpError(response)}`);
|
|
224
|
+
const data = await response.json();
|
|
225
|
+
results.push(...data.embeddings.map(e => e.values));
|
|
226
|
+
}
|
|
227
|
+
finally {
|
|
228
|
+
clearTimeout(timer);
|
|
229
|
+
}
|
|
230
|
+
}
|
|
231
|
+
return results;
|
|
232
|
+
}
|
|
233
|
+
vectorToBuffer(vector) {
|
|
234
|
+
return Buffer.from(new Float32Array(vector).buffer);
|
|
235
|
+
}
|
|
236
|
+
bufferToVector(buffer) {
|
|
237
|
+
return Array.from(new Float32Array(buffer.buffer, buffer.byteOffset, buffer.byteLength / 4));
|
|
238
|
+
}
|
|
239
|
+
}
|
|
240
|
+
export function createEmbeddingProvider(config) {
|
|
241
|
+
switch (config.provider) {
|
|
242
|
+
case 'mock':
|
|
243
|
+
return new MockEmbeddingProvider(config);
|
|
244
|
+
case 'openai':
|
|
245
|
+
return new OpenAIEmbeddingProvider(config);
|
|
246
|
+
case 'local':
|
|
247
|
+
return new LocalEmbeddingProvider(config);
|
|
248
|
+
case 'gemini':
|
|
249
|
+
return new GeminiEmbeddingProvider(config);
|
|
250
|
+
default:
|
|
251
|
+
throw new Error(`Unknown embedding provider: ${config.provider}. Valid: mock, openai, local, gemini`);
|
|
252
|
+
}
|
|
253
|
+
}
|
|
254
|
+
//# sourceMappingURL=embedding.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embedding.js","sourceRoot":"","sources":["../../src/embedding.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EAAE,iBAAiB,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAE9D,MAAM,OAAO,qBAAqB;IAChC,UAAU,CAAS;IACnB,SAAS,CAAS;IAClB,YAAY,CAAS;IAErB,YAAY,EAAE,UAAU,GAAG,EAAE,KAA+B,EAAE;QAC5D,IAAI,CAAC,UAAU,GAAG,UAAU,IAAI,EAAE,CAAC;QACnC,IAAI,CAAC,SAAS,GAAG,gBAAgB,CAAC;QAClC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,MAAM,IAAI,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;QACxD,MAAM,MAAM,GAAG,IAAI,KAAK,CAAS,IAAI,CAAC,UAAU,CAAC,CAAC;QAClD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAE,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAE,GAAG,CAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QACzE,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAE,GAAG,SAAS,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,OAAO,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC;IAED,cAAc,CAAC,MAAgB;QAC7B,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC;IAC/F,CAAC;CACF;AAED,MAAM,OAAO,uBAAuB;IAClC,MAAM,CAAqB;IAC3B,KAAK,CAAS;IACd,UAAU,CAAS;IACnB,OAAO,CAAS;IAChB,SAAS,CAAS;IAClB,YAAY,CAAS;IAErB,YAAY,EAAE,MAAM,EAAE,KAAK,GAAG,wBAAwB,EAAE,UAAU,GAAG,IAAI,EAAE,OAAO,GAAG,KAAK,KAA+B,EAAE;QACzH,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QACnD,IAAI,CAAC,KAAK,GAAG,KAAK,IAAI,wBAAwB,CAAC;QAC/C,IAAI,CAAC,UAAU,GAAG,UAAU,IAAI,IAAI,CAAC;QACrC,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,KAAK,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC;QAC5B,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,kBAAkB,EAAE,gBAAgB,CAAC,CAAC;QACjE,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACjE,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,sCAAsC,EAAE;gBACnE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;oBACxC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC;gBACrF,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,EAAE;gBAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,MAAM,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YACnG,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAyC,CAAC;YAC1E,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,SAAS,CAAC;QACjC,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,kBAAkB,EAAE,gBAAgB,CAAC,CAAC;QACjE,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACjE,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,sCAAsC,EAAE;gBACnE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;oBACxC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC;gBACtF,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,EAAE;gBAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,MAAM,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YACnG,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAyC,CAAC;YAC1E,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED,cAAc,CAAC,MAAgB;QAC7B,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC;IAC/F,CAAC;CACF;AAED,MAAM,OAAO,sBAAsB;IACjC,KAAK,CAAS;IACd,UAAU,CAAS;IACnB,SAAS,CAAS;IAClB,YAAY,CAAS;IACrB,MAAM,CAAS;IACf,SAAS,CAAS;IAClB,eAAe,CAAgG;IAC/G,SAAS,CAAM,CAAC,yDAAyD;IACzE,aAAa,CAAuB;IACpC,aAAa,CAAgB;IAE7B,YAAY,EAAE,KAAK,GAAG,yBAAyB,EAAE,MAAM,GAAG,KAAK,EAAE,SAAS,GAAG,EAAE,EAAE,eAAe,GAAG,IAAI,KAA+B,EAAE;QACtI,IAAI,CAAC,KAAK,GAAG,KAAK,IAAI,yBAAyB,CAAC;QAChD,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC;QAC5B,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,KAAK,CAAC;QAC9B,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,EAAE,CAAC;QACjC,IAAI,CAAC,eAAe,GAAG,eAAe,IAAI,IAAI,CAAC;QAC/C,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC1B,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;IAC5B,CAAC;IAED,KAAK;QACH,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YACxB,IAAI,CAAC,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;gBAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,IAAI,CAAC,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC,CAAC,QAAQ,CAAC;gBAC9F,IAAI,CAAC;oBACH,IAAI,CAAC,SAAS,GAAG,MAAM,QAAQ,CAAC,oBAAoB,EAAE,IAAI,CAAC,KAAK,EAAE;wBAChE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,MAAuB;qBACpD,CAAC,CAAC;oBACH,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC;gBACnC,CAAC;gBAAC,MAAM,CAAC;oBACP,IAAI,CAAC,SAAS,GAAG,MAAM,QAAQ,CAAC,oBAAoB,EAAE,IAAI,CAAC,KAAK,EAAE;wBAChE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK;qBAC7B,CAAC,CAAC;oBACH,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;gBAC7B,CAAC;YACH,CAAC,CAAC,EAAE,CAAC;QACP,CAAC;QACD,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAChF,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAoB,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAClC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACnB,MAAM,OAAO,GAAe,EAAE,CAAC;QAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;YACjD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACjF,OAAO,CAAC,IAAI,CAAC,GAAI,MAAM,CAAC,MAAM,EAAiB,CAAC,CAAC;QACnD,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,cAAc,CAAC,MAAgB;QAC7B,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC;IAC/F,CAAC;CACF;AAED,MAAM,OAAO,uBAAuB;IAClC,MAAM,CAAqB;IAC3B,KAAK,CAAS;IACd,UAAU,CAAS;IACnB,OAAO,CAAS;IAChB,SAAS,CAAS;IAClB,YAAY,CAAS;IAErB,YAAY,EAAE,MAAM,EAAE,KAAK,GAAG,sBAAsB,EAAE,OAAO,GAAG,KAAK,KAA+B,EAAE;QACpG,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QACnD,IAAI,CAAC,KAAK,GAAG,KAAK,IAAI,sBAAsB,CAAC;QAC7C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,KAAK,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC;QAC5B,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,kBAAkB,EAAE,gBAAgB,CAAC,CAAC;QACjE,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACjE,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,2DAA2D,IAAI,CAAC,KAAK,eAAe,EACpF;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE;gBAC9E,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,UAAU,IAAI,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvF,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CACF,CAAC;YACF,IAAI,CAAC,QAAQ,CAAC,EAAE;gBAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,MAAM,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YACnG,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAyC,CAAC;YAC1E,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;QAC/B,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAClC,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,kBAAkB,EAAE,gBAAgB,CAAC,CAAC;QACjE,MAAM,OAAO,GAAe,EAAE,CAAC;QAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC;YAC3C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC;YACtC,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YACjE,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,2DAA2D,IAAI,CAAC,KAAK,qBAAqB,EAC1F;oBACE,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE;oBAC9E,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wBACnB,QAAQ,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;4BAC3B,KAAK,EAAE,UAAU,IAAI,CAAC,KAAK,EAAE;4BAC7B,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE;yBAC/B,CAAC,CAAC;qBACJ,CAAC;oBACF,MAAM,EAAE,UAAU,CAAC,MAAM;iBAC1B,CACF,CAAC;gBACF,IAAI,CAAC,QAAQ,CAAC,EAAE;oBAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,MAAM,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;gBACzG,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAA4C,CAAC;gBAC7E,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YACtD,CAAC;oBAAS,CAAC;gBACT,YAAY,CAAC,KAAK,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,cAAc,CAAC,MAAgB;QAC7B,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC;IAC/F,CAAC;CACF;AAED,MAAM,UAAU,uBAAuB,CAAC,MAAuB;IAC7D,QAAQ,MAAM,CAAC,QAAQ,EAAE,CAAC;QACxB,KAAK,MAAM;YACT,OAAO,IAAI,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAC3C,KAAK,QAAQ;YACX,OAAO,IAAI,uBAAuB,CAAC,MAAM,CAAC,CAAC;QAC7C,KAAK,OAAO;YACV,OAAO,IAAI,sBAAsB,CAAC,MAAM,CAAC,CAAC;QAC5C,KAAK,QAAQ;YACX,OAAO,IAAI,uBAAuB,CAAC,MAAM,CAAC,CAAC;QAC7C;YACE,MAAM,IAAI,KAAK,CAAC,+BAAgC,MAA0B,CAAC,QAAQ,sCAAsC,CAAC,CAAC;IAC/H,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import Database from 'better-sqlite3';
|
|
2
|
+
import type { Affect, CausalParams, EmbeddingProvider, SourceType } from './types.js';
|
|
3
|
+
export declare function encodeEpisode(db: Database.Database, embeddingProvider: EmbeddingProvider, { content, source, salience, causal, tags, supersedes, context, affect, arousalWeight, private: isPrivate, }: {
|
|
4
|
+
content: string;
|
|
5
|
+
source: SourceType;
|
|
6
|
+
salience?: number;
|
|
7
|
+
causal?: CausalParams;
|
|
8
|
+
tags?: string[];
|
|
9
|
+
supersedes?: string;
|
|
10
|
+
context?: Record<string, string>;
|
|
11
|
+
affect?: Partial<Affect>;
|
|
12
|
+
arousalWeight?: number;
|
|
13
|
+
private?: boolean;
|
|
14
|
+
}): Promise<string>;
|
|
15
|
+
//# sourceMappingURL=encode.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"encode.d.ts","sourceRoot":"","sources":["../../src/encode.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,KAAK,EAAE,MAAM,EAAE,YAAY,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAMtF,wBAAsB,aAAa,CACjC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,iBAAiB,EAAE,iBAAiB,EACpC,EACE,OAAO,EACP,MAAM,EACN,QAAc,EACd,MAAM,EACN,IAAI,EACJ,UAAU,EACV,OAAY,EACZ,MAAW,EACX,aAAmB,EACnB,OAAO,EAAE,SAAiB,GAC3B,EAAE;IACD,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,UAAU,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,YAAY,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,MAAM,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IACzB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB,GACA,OAAO,CAAC,MAAM,CAAC,CA0CjB"}
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
import { generateId } from './ulid.js';
|
|
2
|
+
import { sourceReliability } from './confidence.js';
|
|
3
|
+
import { arousalSalienceBoost } from './affect.js';
|
|
4
|
+
import { insertFTSEpisode } from './fts.js';
|
|
5
|
+
export async function encodeEpisode(db, embeddingProvider, { content, source, salience = 0.5, causal, tags, supersedes, context = {}, affect = {}, arousalWeight = 0.3, private: isPrivate = false, }) {
|
|
6
|
+
if (!content || typeof content !== 'string')
|
|
7
|
+
throw new Error('content must be a non-empty string');
|
|
8
|
+
if (salience < 0 || salience > 1)
|
|
9
|
+
throw new Error('salience must be between 0 and 1');
|
|
10
|
+
if (tags && !Array.isArray(tags))
|
|
11
|
+
throw new Error('tags must be an array');
|
|
12
|
+
const reliability = sourceReliability(source);
|
|
13
|
+
const vector = await embeddingProvider.embed(content);
|
|
14
|
+
const embeddingBuffer = embeddingProvider.vectorToBuffer(vector);
|
|
15
|
+
const id = generateId();
|
|
16
|
+
const now = new Date().toISOString();
|
|
17
|
+
const boost = arousalSalienceBoost(affect.arousal);
|
|
18
|
+
const effectiveSalience = Math.min(1.0, salience + (boost * arousalWeight));
|
|
19
|
+
const insertAndLink = db.transaction(() => {
|
|
20
|
+
db.prepare(`
|
|
21
|
+
INSERT INTO episodes (
|
|
22
|
+
id, content, embedding, source, source_reliability, salience, context, affect,
|
|
23
|
+
tags, causal_trigger, causal_consequence, created_at,
|
|
24
|
+
embedding_model, embedding_version, supersedes, "private"
|
|
25
|
+
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
26
|
+
`).run(id, content, embeddingBuffer, source, reliability, effectiveSalience, JSON.stringify(context), JSON.stringify(affect), tags ? JSON.stringify(tags) : null, causal?.trigger || null, causal?.consequence || null, now, embeddingProvider.modelName, embeddingProvider.modelVersion, supersedes || null, isPrivate ? 1 : 0);
|
|
27
|
+
db.prepare('INSERT INTO vec_episodes(id, embedding, source, consolidated) VALUES (?, ?, ?, ?)').run(id, embeddingBuffer, source, BigInt(0));
|
|
28
|
+
insertFTSEpisode(db, id, content, tags ?? null);
|
|
29
|
+
if (supersedes) {
|
|
30
|
+
db.prepare('UPDATE episodes SET superseded_by = ? WHERE id = ?').run(id, supersedes);
|
|
31
|
+
}
|
|
32
|
+
});
|
|
33
|
+
insertAndLink();
|
|
34
|
+
return id;
|
|
35
|
+
}
|
|
36
|
+
//# sourceMappingURL=encode.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"encode.js","sourceRoot":"","sources":["../../src/encode.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,EAAqB,EACrB,iBAAoC,EACpC,EACE,OAAO,EACP,MAAM,EACN,QAAQ,GAAG,GAAG,EACd,MAAM,EACN,IAAI,EACJ,UAAU,EACV,OAAO,GAAG,EAAE,EACZ,MAAM,GAAG,EAAE,EACX,aAAa,GAAG,GAAG,EACnB,OAAO,EAAE,SAAS,GAAG,KAAK,GAY3B;IAED,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;IACnG,IAAI,QAAQ,GAAG,CAAC,IAAI,QAAQ,GAAG,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;IACtF,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAE3E,MAAM,WAAW,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAC9C,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACtD,MAAM,eAAe,GAAG,iBAAiB,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IACjE,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;IACxB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAErC,MAAM,KAAK,GAAG,oBAAoB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACnD,MAAM,iBAAiB,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,QAAQ,GAAG,CAAC,KAAK,GAAG,aAAa,CAAC,CAAC,CAAC;IAE5E,MAAM,aAAa,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;QACxC,EAAE,CAAC,OAAO,CAAC;;;;;;KAMV,CAAC,CAAC,GAAG,CACJ,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,WAAW,EAAE,iBAAiB,EACpE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EACvB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EACtB,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,EAClC,MAAM,EAAE,OAAO,IAAI,IAAI,EAAE,MAAM,EAAE,WAAW,IAAI,IAAI,EACpD,GAAG,EAAE,iBAAiB,CAAC,SAAS,EAAE,iBAAiB,CAAC,YAAY,EAChE,UAAU,IAAI,IAAI,EAClB,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAClB,CAAC;QACF,EAAE,CAAC,OAAO,CACR,mFAAmF,CACpF,CAAC,GAAG,CAAC,EAAE,EAAE,eAAe,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9C,gBAAgB,CAAC,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,IAAI,IAAI,CAAC,CAAC;QAChD,IAAI,UAAU,EAAE,CAAC;YACf,EAAE,CAAC,OAAO,CAAC,oDAAoD,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QACvF,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,aAAa,EAAE,CAAC;IAChB,OAAO,EAAE,CAAC;AACZ,CAAC"}
|
|
@@ -0,0 +1,69 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* memory_events CRUD. Thin wrapper — business logic (hashing, redaction,
|
|
3
|
+
* summarization) lives in tool-trace.ts.
|
|
4
|
+
*/
|
|
5
|
+
import Database from 'better-sqlite3';
|
|
6
|
+
export type EventType = 'PreToolUse' | 'PostToolUse' | 'PostToolUseFailure' | 'PreCompact' | 'PostCompact' | 'SessionStart' | 'SessionStop' | 'SubagentStart' | 'SubagentStop' | 'Observation';
|
|
7
|
+
export type EventOutcome = 'succeeded' | 'failed' | 'blocked' | 'skipped' | 'unknown';
|
|
8
|
+
export type RedactionState = 'unreviewed' | 'redacted' | 'clean' | 'quarantined';
|
|
9
|
+
export interface MemoryEvent {
|
|
10
|
+
id: string;
|
|
11
|
+
session_id: string | null;
|
|
12
|
+
event_type: EventType | string;
|
|
13
|
+
source: string;
|
|
14
|
+
actor_agent: string | null;
|
|
15
|
+
tool_name: string | null;
|
|
16
|
+
input_hash: string | null;
|
|
17
|
+
output_hash: string | null;
|
|
18
|
+
outcome: EventOutcome | null;
|
|
19
|
+
error_summary: string | null;
|
|
20
|
+
cwd: string | null;
|
|
21
|
+
file_fingerprints: string | null;
|
|
22
|
+
redaction_state: RedactionState;
|
|
23
|
+
metadata: string | null;
|
|
24
|
+
created_at: string;
|
|
25
|
+
}
|
|
26
|
+
export interface EventInsert {
|
|
27
|
+
id?: string;
|
|
28
|
+
sessionId?: string | null;
|
|
29
|
+
eventType: EventType | string;
|
|
30
|
+
source: string;
|
|
31
|
+
actorAgent?: string | null;
|
|
32
|
+
toolName?: string | null;
|
|
33
|
+
inputHash?: string | null;
|
|
34
|
+
outputHash?: string | null;
|
|
35
|
+
outcome?: EventOutcome | null;
|
|
36
|
+
errorSummary?: string | null;
|
|
37
|
+
cwd?: string | null;
|
|
38
|
+
fileFingerprints?: string[] | null;
|
|
39
|
+
redactionState?: RedactionState;
|
|
40
|
+
metadata?: Record<string, unknown> | null;
|
|
41
|
+
createdAt?: string;
|
|
42
|
+
}
|
|
43
|
+
export interface EventQuery {
|
|
44
|
+
sessionId?: string;
|
|
45
|
+
toolName?: string;
|
|
46
|
+
eventType?: string;
|
|
47
|
+
outcome?: EventOutcome;
|
|
48
|
+
since?: string;
|
|
49
|
+
limit?: number;
|
|
50
|
+
}
|
|
51
|
+
export declare function insertEvent(db: Database.Database, input: EventInsert): MemoryEvent;
|
|
52
|
+
export declare function listEvents(db: Database.Database, query?: EventQuery): MemoryEvent[];
|
|
53
|
+
export declare function countEvents(db: Database.Database, query?: EventQuery): number;
|
|
54
|
+
export interface FailurePattern {
|
|
55
|
+
tool_name: string;
|
|
56
|
+
failure_count: number;
|
|
57
|
+
last_error_summary: string | null;
|
|
58
|
+
last_failed_at: string;
|
|
59
|
+
}
|
|
60
|
+
/**
|
|
61
|
+
* Tools that have failed recently, most recent first. Feeds PreToolUse
|
|
62
|
+
* preflight warnings: "this command failed last time — here's what fixed it."
|
|
63
|
+
*/
|
|
64
|
+
export declare function recentFailures(db: Database.Database, options?: {
|
|
65
|
+
since?: string;
|
|
66
|
+
limit?: number;
|
|
67
|
+
}): FailurePattern[];
|
|
68
|
+
export declare function deleteEventsBefore(db: Database.Database, cutoffIso: string): number;
|
|
69
|
+
//# sourceMappingURL=events.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"events.d.ts","sourceRoot":"","sources":["../../src/events.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAGtC,MAAM,MAAM,SAAS,GACjB,YAAY,GACZ,aAAa,GACb,oBAAoB,GACpB,YAAY,GACZ,aAAa,GACb,cAAc,GACd,aAAa,GACb,eAAe,GACf,cAAc,GACd,aAAa,CAAC;AAElB,MAAM,MAAM,YAAY,GAAG,WAAW,GAAG,QAAQ,GAAG,SAAS,GAAG,SAAS,GAAG,SAAS,CAAC;AACtF,MAAM,MAAM,cAAc,GAAG,YAAY,GAAG,UAAU,GAAG,OAAO,GAAG,aAAa,CAAC;AAEjF,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,UAAU,EAAE,SAAS,GAAG,MAAM,CAAC;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,OAAO,EAAE,YAAY,GAAG,IAAI,CAAC;IAC7B,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,GAAG,EAAE,MAAM,GAAG,IAAI,CAAC;IACnB,iBAAiB,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,eAAe,EAAE,cAAc,CAAC;IAChC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,SAAS,EAAE,SAAS,GAAG,MAAM,CAAC;IAC9B,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,OAAO,CAAC,EAAE,YAAY,GAAG,IAAI,CAAC;IAC9B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,GAAG,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,gBAAgB,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IACnC,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IAC1C,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,UAAU;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,YAAY,CAAC;IACvB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAOD,wBAAgB,WAAW,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,KAAK,EAAE,WAAW,GAAG,WAAW,CAsDlF;AAED,wBAAgB,UAAU,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,KAAK,GAAE,UAAe,GAAG,WAAW,EAAE,CA+BvF;AAED,wBAAgB,WAAW,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,KAAK,GAAE,UAAe,GAAG,MAAM,CAYjF;AAED,MAAM,WAAW,cAAc;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,kBAAkB,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,cAAc,EAAE,MAAM,CAAC;CACxB;AAED;;;GAGG;AACH,wBAAgB,cAAc,CAC5B,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,OAAO,GAAE;IAAE,KAAK,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAO,GAC/C,cAAc,EAAE,CAuBlB;AAED,wBAAgB,kBAAkB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAGnF"}
|