@mysten-incubation/memwal 0.1.0-dev.0 → 0.1.0-dev.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.
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../../src/ai/middleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;GAqBG;AAKH,OAAO,KAAK,EAAE,YAAY,EAAgB,MAAM,aAAa,CAAC;AAU9D;;;;;GAKG;AAEH,KAAK,gBAAgB,GAAG,GAAG,CAAC;AAE5B,MAAM,WAAW,iBAAkB,SAAQ,YAAY;IACnD,sDAAsD;IACtD,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,4DAA4D;IAC5D,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,uEAAuE;IACvE,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,4CAA4C;IAC5C,KAAK,CAAC,EAAE,OAAO,CAAC;CACnB;AAMD;;;;;;;;;;;GAWG;AACH,wBAAgB,UAAU,CACtB,KAAK,EAAE,gBAAgB,EACvB,OAAO,EAAE,iBAAiB,
|
|
1
|
+
{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../../src/ai/middleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;GAqBG;AAKH,OAAO,KAAK,EAAE,YAAY,EAAgB,MAAM,aAAa,CAAC;AAU9D;;;;;GAKG;AAEH,KAAK,gBAAgB,GAAG,GAAG,CAAC;AAE5B,MAAM,WAAW,iBAAkB,SAAQ,YAAY;IACnD,sDAAsD;IACtD,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,4DAA4D;IAC5D,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,uEAAuE;IACvE,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,4CAA4C;IAC5C,KAAK,CAAC,EAAE,OAAO,CAAC;CACnB;AAMD;;;;;;;;;;;GAWG;AACH,wBAAgB,UAAU,CACtB,KAAK,EAAE,gBAAgB,EACvB,OAAO,EAAE,iBAAiB,OAoG7B;AA0BD,wBAAgB,mBAAmB,CAC/B,MAAM,EAAE,OAAO,EACf,aAAa,EAAE,MAAM,GACtB,OAAO,CAuCT"}
|
package/dist/ai/middleware.js
CHANGED
|
@@ -47,7 +47,18 @@ export function withMemWal(model, options) {
|
|
|
47
47
|
const log = debug
|
|
48
48
|
? (...args) => console.warn("[Walrus Memory]", ...args)
|
|
49
49
|
: () => { };
|
|
50
|
-
|
|
50
|
+
// Fire-and-forget auto-save writes are tracked here so short-lived
|
|
51
|
+
// callers (CLI scripts, serverless handlers) can await them via
|
|
52
|
+
// flush() before the process exits, instead of losing them silently.
|
|
53
|
+
const pendingSaves = new Set();
|
|
54
|
+
function saveInBackground(userMessage) {
|
|
55
|
+
const savePromise = memwal
|
|
56
|
+
.analyze(userMessage)
|
|
57
|
+
.catch((err) => log("Auto-save failed:", err));
|
|
58
|
+
pendingSaves.add(savePromise);
|
|
59
|
+
savePromise.finally(() => pendingSaves.delete(savePromise));
|
|
60
|
+
}
|
|
61
|
+
const wrapped = wrapLanguageModel({
|
|
51
62
|
model,
|
|
52
63
|
middleware: {
|
|
53
64
|
specificationVersion: 'v3', // Required by ai SDK v6+; ignored by v4/v5
|
|
@@ -82,7 +93,7 @@ export function withMemWal(model, options) {
|
|
|
82
93
|
if (autoSave) {
|
|
83
94
|
const userMessage = findLastUserMessage(params.prompt);
|
|
84
95
|
if (userMessage) {
|
|
85
|
-
|
|
96
|
+
saveInBackground(userMessage);
|
|
86
97
|
}
|
|
87
98
|
}
|
|
88
99
|
return result;
|
|
@@ -93,13 +104,19 @@ export function withMemWal(model, options) {
|
|
|
93
104
|
if (autoSave) {
|
|
94
105
|
const userMessage = findLastUserMessage(params.prompt);
|
|
95
106
|
if (userMessage) {
|
|
96
|
-
|
|
107
|
+
saveInBackground(userMessage);
|
|
97
108
|
}
|
|
98
109
|
}
|
|
99
110
|
return result;
|
|
100
111
|
},
|
|
101
112
|
},
|
|
102
113
|
});
|
|
114
|
+
// Lets short-lived callers await outstanding auto-save writes before
|
|
115
|
+
// exiting, e.g. `await model.flush()` right before `process.exit()`.
|
|
116
|
+
wrapped.flush = async () => {
|
|
117
|
+
await Promise.allSettled([...pendingSaves]);
|
|
118
|
+
};
|
|
119
|
+
return wrapped;
|
|
103
120
|
}
|
|
104
121
|
// ============================================================
|
|
105
122
|
// Helpers
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../src/ai/middleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;GAqBG;AAGH,OAAO,EAAE,iBAAiB,EAAE,MAAM,IAAI,CAAC;AACvC,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,OAAO,EACH,uBAAuB,EACvB,mCAAmC,GACtC,MAAM,uBAAuB,CAAC;AA0B/B,+DAA+D;AAC/D,aAAa;AACb,+DAA+D;AAE/D;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,UAAU,CACtB,KAAuB,EACvB,OAA0B;IAE1B,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,CAAC,CAAC;IAC7C,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC;IAC1C,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,GAAG,CAAC;IACjD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,KAAK,CAAC;IAErC,MAAM,GAAG,GAAG,KAAK;QACb,CAAC,CAAC,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC;QAClE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAEhB,
|
|
1
|
+
{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../src/ai/middleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;GAqBG;AAGH,OAAO,EAAE,iBAAiB,EAAE,MAAM,IAAI,CAAC;AACvC,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,OAAO,EACH,uBAAuB,EACvB,mCAAmC,GACtC,MAAM,uBAAuB,CAAC;AA0B/B,+DAA+D;AAC/D,aAAa;AACb,+DAA+D;AAE/D;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,UAAU,CACtB,KAAuB,EACvB,OAA0B;IAE1B,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,CAAC,CAAC;IAC7C,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC;IAC1C,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,GAAG,CAAC;IACjD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,KAAK,CAAC;IAErC,MAAM,GAAG,GAAG,KAAK;QACb,CAAC,CAAC,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC;QAClE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAEhB,mEAAmE;IACnE,gEAAgE;IAChE,qEAAqE;IACrE,MAAM,YAAY,GAAG,IAAI,GAAG,EAAoB,CAAC;IAEjD,SAAS,gBAAgB,CAAC,WAAmB;QACzC,MAAM,WAAW,GAAG,MAAM;aACrB,OAAO,CAAC,WAAW,CAAC;aACpB,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE,CAAC,GAAG,CAAC,mBAAmB,EAAE,GAAG,CAAC,CAAC,CAAC;QAC5D,YAAY,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAC9B,WAAW,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,MAAM,OAAO,GAAI,iBAAyB,CAAC;QACvC,KAAK;QACL,UAAU,EAAE;YACR,oBAAoB,EAAE,IAAI,EAAE,2CAA2C;YACvE,+DAA+D;YAC/D,+CAA+C;YAC/C,+DAA+D;YAC/D,eAAe,EAAE,KAAK,EAAE,EAAE,MAAM,EAAO,EAAE,EAAE;gBACvC,IAAI,CAAC;oBACD,MAAM,eAAe,GAAG,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBAC3D,IAAI,CAAC,eAAe;wBAAE,OAAO,MAAM,CAAC;oBAEpC,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC;oBAEvE,4DAA4D;oBAC5D,MAAM,QAAQ,GAAG,YAAY,CAAC,OAAO,CAAC,MAAM,CACxC,CAAC,CAAe,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,IAAI,YAAY,CACxD,CAAC;oBAEF,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;wBAAE,OAAO,MAAM,CAAC;oBAEzC,MAAM,aAAa,GAAG,uBAAuB,CAAC,QAAQ,CAAC,CAAC;oBACxD,MAAM,cAAc,GAAG,mBAAmB,CACtC,MAAM,CAAC,MAAM,EACb,aAAa,CAChB,CAAC;oBAEF,GAAG,CAAC,YAAY,QAAQ,CAAC,MAAM,oBAAoB,CAAC,CAAC;oBAErD,OAAO,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;gBACjD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACb,GAAG,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAC;oBACpC,OAAO,MAAM,CAAC;gBAClB,CAAC;YACL,CAAC;YAED,+DAA+D;YAC/D,4DAA4D;YAC5D,+DAA+D;YAC/D,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,MAAM,EAAO,EAAE,EAAE;gBAChD,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;gBAElC,IAAI,QAAQ,EAAE,CAAC;oBACX,MAAM,WAAW,GAAG,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBACvD,IAAI,WAAW,EAAE,CAAC;wBACd,gBAAgB,CAAC,WAAW,CAAC,CAAC;oBAClC,CAAC;gBACL,CAAC;gBAED,OAAO,MAAM,CAAC;YAClB,CAAC;YAED,2CAA2C;YAC3C,UAAU,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAO,EAAE,EAAE;gBAC5C,MAAM,MAAM,GAAG,MAAM,QAAQ,EAAE,CAAC;gBAEhC,IAAI,QAAQ,EAAE,CAAC;oBACX,MAAM,WAAW,GAAG,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBACvD,IAAI,WAAW,EAAE,CAAC;wBACd,gBAAgB,CAAC,WAAW,CAAC,CAAC;oBAClC,CAAC;gBACL,CAAC;gBAED,OAAO,MAAM,CAAC;YAClB,CAAC;SACJ;KACJ,CAAC,CAAC;IAEH,qEAAqE;IACrE,qEAAqE;IACrE,OAAO,CAAC,KAAK,GAAG,KAAK,IAAmB,EAAE;QACtC,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,GAAG,YAAY,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC;IAEF,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,+DAA+D;AAC/D,UAAU;AACV,+DAA+D;AAE/D,SAAS,mBAAmB,CACxB,MAAe;IAEf,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO,IAAI,CAAC;IAExC,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1C,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,CAAyC,CAAC;QAC9D,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACtB,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ;gBAAE,OAAO,GAAG,CAAC,OAAO,CAAC;YACxD,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC7B,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO;qBACxB,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;qBACrC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBAC7B,OAAO,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC;YACvC,CAAC;QACL,CAAC;IACL,CAAC;IACD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAC/B,MAAe,EACf,aAAqB;IAErB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO,MAAM,CAAC;IAE1C,2EAA2E;IAC3E,wEAAwE;IACxE,YAAY;IACZ,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAC/B,CAAC,GAAW,EAAE,CAAM,EAAE,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EACjE,CAAC,CAAC,CACL,CAAC;IAEF,IAAI,aAAa,GAAG,CAAC,EAAE,CAAC;QACpB,MAAM,MAAM,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CACT,aAAa,EACb,CAAC,EACD;YACI,IAAI,EAAE,QAAiB;YACvB,OAAO,EAAE,mCAAmC;SAC/C,EACD;YACI,IAAI,EAAE,MAAe;YACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;SAC5D,CACJ,CAAC;QACF,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,OAAO;QACH;YACI,IAAI,EAAE,QAAiB;YACvB,OAAO,EAAE,mCAAmC;SAC/C;QACD;YACI,IAAI,EAAE,MAAe;YACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;SAC5D;QACD,GAAG,MAAM;KACZ,CAAC;AACN,CAAC"}
|
package/package.json
CHANGED