langchain 0.0.49 → 0.0.51
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/cache.cjs +1 -0
- package/cache.d.ts +1 -0
- package/cache.js +1 -0
- package/dist/agents/agent.cjs +2 -0
- package/dist/agents/agent.js +2 -0
- package/dist/agents/agent.js.map +1 -1
- package/dist/callbacks/utils.cjs +2 -0
- package/dist/callbacks/utils.js +2 -0
- package/dist/callbacks/utils.js.map +1 -1
- package/dist/chains/prompt_selector.cjs +2 -4
- package/dist/chains/prompt_selector.d.ts +2 -1
- package/dist/chains/prompt_selector.js +2 -4
- package/dist/chains/prompt_selector.js.map +1 -1
- package/dist/document.cjs +0 -1
- package/dist/document.d.ts +5 -5
- package/dist/document.js +0 -1
- package/dist/document.js.map +1 -1
- package/dist/llms/base.cjs +1 -1
- package/dist/llms/base.js +2 -2
- package/dist/llms/base.js.map +1 -1
- package/dist/memory/base.cjs +4 -5
- package/dist/memory/base.js +4 -5
- package/dist/memory/base.js.map +1 -1
- package/dist/memory/chat_memory.d.ts +3 -3
- package/dist/memory/chat_memory.js.map +1 -1
- package/dist/memory/index.cjs +3 -1
- package/dist/memory/index.d.ts +1 -0
- package/dist/memory/index.js +1 -0
- package/dist/memory/index.js.map +1 -1
- package/dist/memory/motorhead_memory.cjs +106 -0
- package/dist/memory/motorhead_memory.d.ts +25 -0
- package/dist/memory/motorhead_memory.js +102 -0
- package/dist/memory/motorhead_memory.js.map +1 -0
- package/dist/output_parsers/fix.cjs +1 -0
- package/dist/output_parsers/fix.js +1 -0
- package/dist/output_parsers/fix.js.map +1 -1
- package/dist/output_parsers/structured.cjs +1 -0
- package/dist/output_parsers/structured.js +1 -0
- package/dist/output_parsers/structured.js.map +1 -1
- package/dist/util/async_caller.d.ts +1 -2
- package/dist/util/async_caller.js.map +1 -1
- package/dist/util/axios-fetch-adapter.cjs +4 -0
- package/dist/util/axios-fetch-adapter.js +4 -0
- package/dist/util/axios-fetch-adapter.js.map +1 -1
- package/dist/vectorstores/chroma.d.ts +1 -1
- package/dist/vectorstores/hnswlib.d.ts +1 -1
- package/dist/vectorstores/index.cjs +3 -1
- package/dist/vectorstores/index.d.ts +1 -0
- package/dist/vectorstores/index.js +1 -0
- package/dist/vectorstores/index.js.map +1 -1
- package/dist/vectorstores/prisma.cjs +178 -0
- package/dist/vectorstores/prisma.d.ts +90 -0
- package/dist/vectorstores/prisma.js +174 -0
- package/dist/vectorstores/prisma.js.map +1 -0
- package/package.json +10 -8
package/cache.cjs
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
module.exports = require('./dist/cache.cjs');
|
package/cache.d.ts
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export * from './dist/cache.js'
|
package/cache.js
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export * from './dist/cache.js'
|
package/dist/agents/agent.cjs
CHANGED
|
@@ -226,6 +226,8 @@ class Agent extends BaseSingleActionAgent {
|
|
|
226
226
|
return { returnValues: { output: action.log }, log: action.log };
|
|
227
227
|
}
|
|
228
228
|
catch (err) {
|
|
229
|
+
// fine to use instanceof because we're in the same module
|
|
230
|
+
// eslint-disable-next-line no-instanceof/no-instanceof
|
|
229
231
|
if (!(err instanceof ParseError)) {
|
|
230
232
|
throw err;
|
|
231
233
|
}
|
package/dist/agents/agent.js
CHANGED
|
@@ -221,6 +221,8 @@ export class Agent extends BaseSingleActionAgent {
|
|
|
221
221
|
return { returnValues: { output: action.log }, log: action.log };
|
|
222
222
|
}
|
|
223
223
|
catch (err) {
|
|
224
|
+
// fine to use instanceof because we're in the same module
|
|
225
|
+
// eslint-disable-next-line no-instanceof/no-instanceof
|
|
224
226
|
if (!(err instanceof ParseError)) {
|
|
225
227
|
throw err;
|
|
226
228
|
}
|
package/dist/agents/agent.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/agents/agent.ts"],"names":[],"mappings":"AAkBA,MAAM,UAAW,SAAQ,KAAK;IAG5B,YAAY,GAAW,EAAE,MAAc;QACrC,KAAK,CAAC,GAAG,CAAC,CAAC;QAHb;;;;;WAAe;QAIb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAgB,qBAAqB;IAGzC,IAAI,YAAY;QACd,OAAO,CAAC,QAAQ,CAAC,CAAC;IACpB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,UAAU;QACR,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAeD;;OAEG;IACH,qBAAqB,CACnB,mBAAmC,EACnC,MAAmB,EACnB,OAAoB;QAEpB,IAAI,mBAAmB,KAAK,OAAO,EAAE;YACnC,OAAO,OAAO,CAAC,OAAO,CAAC;gBACrB,YAAY,EAAE,EAAE,MAAM,EAAE,sCAAsC,EAAE;gBAChE,GAAG,EAAE,EAAE;aACR,CAAC,CAAC;SACJ;QAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,mBAAmB,EAAE,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CACpB,aAA0C,EAC1C,MAAmB;QAEnB,OAAO,EAAE,CAAC;IACZ,CAAC;CACF;AAQD,MAAM,OAAO,oBAAqB,SAAQ,qBAAqB;IAO7D,YAAY,KAAgC;QAC1C,KAAK,EAAE,CAAC;QAPV;;;;;WAAmB;QAEnB;;;;;WAAsC;QAEtC;;;;;WAAgB;QAId,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACvB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;IACzC,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;IACjC,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,IAAI,CACR,KAAkB,EAClB,MAAmB;QAEnB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACtC,kBAAkB,EAAE,KAAK;YACzB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,GAAG,MAAM;SACV,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;IAClE,CAAC;CACF;AAED;;;;;;GAMG;AACH,MAAM,OAAgB,KAAM,SAAQ,qBAAqB;IAKvD,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,kBAAkB,CAAC,CAAC;IACzE,CAAC;IAED,YAAY,KAAiB;QAC3B,KAAK,EAAE,CAAC;QAbV;;;;;WAAmB;QAEnB;;;;mBAAmC,SAAS;WAAC;QAY3C,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,YAAY,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB,CACvB,MAAc;QAEd,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAiBD;;;;;;;SAOK;IACL,MAAM,CAAC,YAAY,CACjB,MAAc;IACd,8DAA8D;IAC9D,OAA6B;QAE7B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED,yDAAyD;IACzD,MAAM,CAAC,eAAe,CACpB,IAAuB,EACvB,MAAc;IACd,8DAA8D;IAC9D,KAA2B;QAE3B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,aAAa,CAAC,MAAc,IAAS,CAAC;IAE7C,KAAK;QACH,OAAO,CAAC,KAAK,IAAI,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,cAAc,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,mBAAmB,CAAC,KAAkB;QACpC,OAAO,KAAK,CAAC,MAAM,CACjB,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,QAAQ;YACR;gBACE,MAAM,CAAC,GAAG;gBACV,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,WAAW,EAAE;gBAC3C,IAAI,CAAC,SAAS,EAAE;aACjB,CAAC,IAAI,CAAC,IAAI,CAAC,EACd,EAAE,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,KAAK,CACjB,KAAkB,EAClB,MAAmB,EACnB,MAAe;QAEf,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,SAAS,GAAgB;YAC7B,GAAG,MAAM;YACT,gBAAgB,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC,QAAQ;SAC7D,CAAC;QAEF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE;YAC7B,SAAS,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;SAC/B;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QACtD,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,UAAU,CAAC,mBAAmB,MAAM,EAAE,EAAE,MAAM,CAAC,CAAC;SAC3D;QACD,MAAM,MAAM,GAAG;YACb,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,SAAS,EAAE,MAAM,CAAC,KAAK;YACvB,GAAG,EAAE,MAAM;SACZ,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,cAAc,EAAE,EAAE;YACzC,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;SACxE;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;OAOG;IACH,IAAI,CACF,KAAkB,EAClB,MAAmB;QAEnB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,qBAAqB,CACzB,mBAAmC,EACnC,KAAkB,EAClB,MAAmB;QAEnB,IAAI,mBAAmB,KAAK,OAAO,EAAE;YACnC,OAAO;gBACL,YAAY,EAAE,EAAE,MAAM,EAAE,sCAAsC,EAAE;gBAChE,GAAG,EAAE,EAAE;aACR,CAAC;SACH;QAED,IAAI,mBAAmB,KAAK,UAAU,EAAE;YACtC,IAAI;gBACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B,KAAK,EACL,MAAM,EACN,sEAAsE,CACvE,CAAC;gBACF,IAAI,cAAc,IAAI,MAAM,EAAE;oBAC5B,OAAO,MAAM,CAAC;iBACf;gBAED,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;aAClE;YAAC,OAAO,GAAG,EAAE;gBACZ,IAAI,CAAC,CAAC,GAAG,YAAY,UAAU,CAAC,EAAE;oBAChC,MAAM,GAAG,CAAC;iBACX;gBACD,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC;aAClE;SACF;QAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,mBAAmB,EAAE,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,IAAmE;QAEnE,QAAQ,IAAI,CAAC,KAAK,EAAE;YAClB,KAAK,6BAA6B,CAAC,CAAC;gBAClC,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;gBAC1D,OAAO,aAAa,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;aACxC;YACD;gBACE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;SACzC;IACH,CAAC;CACF"}
|
|
1
|
+
{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/agents/agent.ts"],"names":[],"mappings":"AAkBA,MAAM,UAAW,SAAQ,KAAK;IAG5B,YAAY,GAAW,EAAE,MAAc;QACrC,KAAK,CAAC,GAAG,CAAC,CAAC;QAHb;;;;;WAAe;QAIb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAgB,qBAAqB;IAGzC,IAAI,YAAY;QACd,OAAO,CAAC,QAAQ,CAAC,CAAC;IACpB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,UAAU;QACR,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAeD;;OAEG;IACH,qBAAqB,CACnB,mBAAmC,EACnC,MAAmB,EACnB,OAAoB;QAEpB,IAAI,mBAAmB,KAAK,OAAO,EAAE;YACnC,OAAO,OAAO,CAAC,OAAO,CAAC;gBACrB,YAAY,EAAE,EAAE,MAAM,EAAE,sCAAsC,EAAE;gBAChE,GAAG,EAAE,EAAE;aACR,CAAC,CAAC;SACJ;QAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,mBAAmB,EAAE,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CACpB,aAA0C,EAC1C,MAAmB;QAEnB,OAAO,EAAE,CAAC;IACZ,CAAC;CACF;AAQD,MAAM,OAAO,oBAAqB,SAAQ,qBAAqB;IAO7D,YAAY,KAAgC;QAC1C,KAAK,EAAE,CAAC;QAPV;;;;;WAAmB;QAEnB;;;;;WAAsC;QAEtC;;;;;WAAgB;QAId,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACvB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;IACzC,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;IACjC,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,IAAI,CACR,KAAkB,EAClB,MAAmB;QAEnB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACtC,kBAAkB,EAAE,KAAK;YACzB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,GAAG,MAAM;SACV,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;IAClE,CAAC;CACF;AAED;;;;;;GAMG;AACH,MAAM,OAAgB,KAAM,SAAQ,qBAAqB;IAKvD,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,kBAAkB,CAAC,CAAC;IACzE,CAAC;IAED,YAAY,KAAiB;QAC3B,KAAK,EAAE,CAAC;QAbV;;;;;WAAmB;QAEnB;;;;mBAAmC,SAAS;WAAC;QAY3C,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,YAAY,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB,CACvB,MAAc;QAEd,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAiBD;;;;;;;SAOK;IACL,MAAM,CAAC,YAAY,CACjB,MAAc;IACd,8DAA8D;IAC9D,OAA6B;QAE7B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED,yDAAyD;IACzD,MAAM,CAAC,eAAe,CACpB,IAAuB,EACvB,MAAc;IACd,8DAA8D;IAC9D,KAA2B;QAE3B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,aAAa,CAAC,MAAc,IAAS,CAAC;IAE7C,KAAK;QACH,OAAO,CAAC,KAAK,IAAI,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,cAAc,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,mBAAmB,CAAC,KAAkB;QACpC,OAAO,KAAK,CAAC,MAAM,CACjB,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,QAAQ;YACR;gBACE,MAAM,CAAC,GAAG;gBACV,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,WAAW,EAAE;gBAC3C,IAAI,CAAC,SAAS,EAAE;aACjB,CAAC,IAAI,CAAC,IAAI,CAAC,EACd,EAAE,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,KAAK,CACjB,KAAkB,EAClB,MAAmB,EACnB,MAAe;QAEf,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,SAAS,GAAgB;YAC7B,GAAG,MAAM;YACT,gBAAgB,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC,QAAQ;SAC7D,CAAC;QAEF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE;YAC7B,SAAS,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;SAC/B;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QACtD,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,UAAU,CAAC,mBAAmB,MAAM,EAAE,EAAE,MAAM,CAAC,CAAC;SAC3D;QACD,MAAM,MAAM,GAAG;YACb,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,SAAS,EAAE,MAAM,CAAC,KAAK;YACvB,GAAG,EAAE,MAAM;SACZ,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,cAAc,EAAE,EAAE;YACzC,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;SACxE;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;OAOG;IACH,IAAI,CACF,KAAkB,EAClB,MAAmB;QAEnB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,qBAAqB,CACzB,mBAAmC,EACnC,KAAkB,EAClB,MAAmB;QAEnB,IAAI,mBAAmB,KAAK,OAAO,EAAE;YACnC,OAAO;gBACL,YAAY,EAAE,EAAE,MAAM,EAAE,sCAAsC,EAAE;gBAChE,GAAG,EAAE,EAAE;aACR,CAAC;SACH;QAED,IAAI,mBAAmB,KAAK,UAAU,EAAE;YACtC,IAAI;gBACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B,KAAK,EACL,MAAM,EACN,sEAAsE,CACvE,CAAC;gBACF,IAAI,cAAc,IAAI,MAAM,EAAE;oBAC5B,OAAO,MAAM,CAAC;iBACf;gBAED,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;aAClE;YAAC,OAAO,GAAG,EAAE;gBACZ,0DAA0D;gBAC1D,uDAAuD;gBACvD,IAAI,CAAC,CAAC,GAAG,YAAY,UAAU,CAAC,EAAE;oBAChC,MAAM,GAAG,CAAC;iBACX;gBACD,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC;aAClE;SACF;QAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,mBAAmB,EAAE,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,IAAmE;QAEnE,QAAQ,IAAI,CAAC,KAAK,EAAE;YAClB,KAAK,6BAA6B,CAAC,CAAC;gBAClC,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;gBAC1D,OAAO,aAAa,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;aACxC;YACD;gBACE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;SACzC;IACH,CAAC;CACF"}
|
package/dist/callbacks/utils.cjs
CHANGED
|
@@ -27,6 +27,8 @@ function getCallbackManager() {
|
|
|
27
27
|
exports.getCallbackManager = getCallbackManager;
|
|
28
28
|
async function setTracerSession(options, callbackManager = getCallbackManager()) {
|
|
29
29
|
for (const handler of callbackManager.handlers) {
|
|
30
|
+
// fine to use instanceof here because we're in the same package
|
|
31
|
+
// eslint-disable-next-line no-instanceof/no-instanceof
|
|
30
32
|
if (handler instanceof tracers_js_1.LangChainTracer) {
|
|
31
33
|
const sessionName = options?.sessionName;
|
|
32
34
|
if (sessionName) {
|
package/dist/callbacks/utils.js
CHANGED
|
@@ -22,6 +22,8 @@ export function getCallbackManager() {
|
|
|
22
22
|
}
|
|
23
23
|
export async function setTracerSession(options, callbackManager = getCallbackManager()) {
|
|
24
24
|
for (const handler of callbackManager.handlers) {
|
|
25
|
+
// fine to use instanceof here because we're in the same package
|
|
26
|
+
// eslint-disable-next-line no-instanceof/no-instanceof
|
|
25
27
|
if (handler instanceof LangChainTracer) {
|
|
26
28
|
const sessionName = options?.sessionName;
|
|
27
29
|
if (sessionName) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/callbacks/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,sBAAsB,EAAE,MAAM,WAAW,CAAC;AAEpE,MAAM,OAAO,wBAAyB,SAAQ,eAAe;IAG3D;QACE,KAAK,EAAE,CAAC;IACV,CAAC;IAED,MAAM,CAAC,WAAW;QAChB,IAAI,CAAC,wBAAwB,CAAC,QAAQ,EAAE;YACtC,wBAAwB,CAAC,QAAQ,GAAG,IAAI,wBAAwB,EAAE,CAAC;YACnE,wBAAwB,CAAC,QAAQ,CAAC,UAAU,CAC1C,IAAI,sBAAsB,EAAE,CAC7B,CAAC;YACF,IACE,OAAO,OAAO,KAAK,WAAW;gBAC9B,0CAA0C;gBAC1C,OAAO,CAAC,GAAG,CAAC,iBAAiB,KAAK,WAAW,EAC7C;gBACA,wBAAwB,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,eAAe,EAAE,CAAC,CAAC;aACrE;SACF;QAED,OAAO,wBAAwB,CAAC,QAAQ,CAAC;IAC3C,CAAC;CACF;AAED,MAAM,UAAU,kBAAkB;IAChC,OAAO,wBAAwB,CAAC,WAAW,EAAE,CAAC;AAChD,CAAC;AAMD,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,OAAuB,EACvB,kBAAmC,kBAAkB,EAAE;IAEvD,KAAK,MAAM,OAAO,IAAI,eAAe,CAAC,QAAQ,EAAE;QAC9C,IAAI,OAAO,YAAY,eAAe,EAAE;YACtC,MAAM,WAAW,GAAG,OAAO,EAAE,WAAW,CAAC;YACzC,IAAI,WAAW,EAAE;gBACf,MAAM,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;aACxC;iBAAM;gBACL,MAAM,OAAO,CAAC,kBAAkB,EAAE,CAAC;aACpC;SACF;KACF;AACH,CAAC"}
|
|
1
|
+
{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/callbacks/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,sBAAsB,EAAE,MAAM,WAAW,CAAC;AAEpE,MAAM,OAAO,wBAAyB,SAAQ,eAAe;IAG3D;QACE,KAAK,EAAE,CAAC;IACV,CAAC;IAED,MAAM,CAAC,WAAW;QAChB,IAAI,CAAC,wBAAwB,CAAC,QAAQ,EAAE;YACtC,wBAAwB,CAAC,QAAQ,GAAG,IAAI,wBAAwB,EAAE,CAAC;YACnE,wBAAwB,CAAC,QAAQ,CAAC,UAAU,CAC1C,IAAI,sBAAsB,EAAE,CAC7B,CAAC;YACF,IACE,OAAO,OAAO,KAAK,WAAW;gBAC9B,0CAA0C;gBAC1C,OAAO,CAAC,GAAG,CAAC,iBAAiB,KAAK,WAAW,EAC7C;gBACA,wBAAwB,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,eAAe,EAAE,CAAC,CAAC;aACrE;SACF;QAED,OAAO,wBAAwB,CAAC,QAAQ,CAAC;IAC3C,CAAC;CACF;AAED,MAAM,UAAU,kBAAkB;IAChC,OAAO,wBAAwB,CAAC,WAAW,EAAE,CAAC;AAChD,CAAC;AAMD,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,OAAuB,EACvB,kBAAmC,kBAAkB,EAAE;IAEvD,KAAK,MAAM,OAAO,IAAI,eAAe,CAAC,QAAQ,EAAE;QAC9C,gEAAgE;QAChE,uDAAuD;QACvD,IAAI,OAAO,YAAY,eAAe,EAAE;YACtC,MAAM,WAAW,GAAG,OAAO,EAAE,WAAW,CAAC;YACzC,IAAI,WAAW,EAAE;gBACf,MAAM,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;aACxC;iBAAM;gBACL,MAAM,OAAO,CAAC,kBAAkB,EAAE,CAAC;aACpC;SACF;KACF;AACH,CAAC"}
|
|
@@ -1,8 +1,6 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.isChatModel = exports.isLLM = exports.ConditionalPromptSelector = exports.BasePromptSelector = void 0;
|
|
4
|
-
const base_js_1 = require("../chat_models/base.cjs");
|
|
5
|
-
const index_js_1 = require("../base_language/index.cjs");
|
|
6
4
|
class BasePromptSelector {
|
|
7
5
|
}
|
|
8
6
|
exports.BasePromptSelector = BasePromptSelector;
|
|
@@ -35,11 +33,11 @@ class ConditionalPromptSelector extends BasePromptSelector {
|
|
|
35
33
|
}
|
|
36
34
|
exports.ConditionalPromptSelector = ConditionalPromptSelector;
|
|
37
35
|
function isLLM(llm) {
|
|
38
|
-
return llm
|
|
36
|
+
return llm._modelType() === "base_llm";
|
|
39
37
|
}
|
|
40
38
|
exports.isLLM = isLLM;
|
|
41
39
|
function isChatModel(llm) {
|
|
42
|
-
return llm
|
|
40
|
+
return llm._modelType() === "base_chat_model";
|
|
43
41
|
}
|
|
44
42
|
exports.isChatModel = isChatModel;
|
|
45
43
|
//# sourceMappingURL=prompt_selector.js.map
|
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
import { BaseChatModel } from "../chat_models/base.js";
|
|
2
2
|
import { BasePromptTemplate } from "../prompts/base.js";
|
|
3
3
|
import { BaseLanguageModel } from "../base_language/index.js";
|
|
4
|
+
import { BaseLLM } from "../llms/base.js";
|
|
4
5
|
export declare abstract class BasePromptSelector {
|
|
5
6
|
abstract getPrompt(llm: BaseLanguageModel): BasePromptTemplate;
|
|
6
7
|
}
|
|
@@ -16,5 +17,5 @@ export declare class ConditionalPromptSelector extends BasePromptSelector {
|
|
|
16
17
|
]>);
|
|
17
18
|
getPrompt(llm: BaseLanguageModel): BasePromptTemplate;
|
|
18
19
|
}
|
|
19
|
-
export declare function isLLM(llm: BaseLanguageModel): llm is
|
|
20
|
+
export declare function isLLM(llm: BaseLanguageModel): llm is BaseLLM;
|
|
20
21
|
export declare function isChatModel(llm: BaseLanguageModel): llm is BaseChatModel;
|
|
@@ -1,5 +1,3 @@
|
|
|
1
|
-
import { BaseChatModel } from "../chat_models/base.js";
|
|
2
|
-
import { BaseLanguageModel } from "../base_language/index.js";
|
|
3
1
|
export class BasePromptSelector {
|
|
4
2
|
}
|
|
5
3
|
export class ConditionalPromptSelector extends BasePromptSelector {
|
|
@@ -30,9 +28,9 @@ export class ConditionalPromptSelector extends BasePromptSelector {
|
|
|
30
28
|
}
|
|
31
29
|
}
|
|
32
30
|
export function isLLM(llm) {
|
|
33
|
-
return llm
|
|
31
|
+
return llm._modelType() === "base_llm";
|
|
34
32
|
}
|
|
35
33
|
export function isChatModel(llm) {
|
|
36
|
-
return llm
|
|
34
|
+
return llm._modelType() === "base_chat_model";
|
|
37
35
|
}
|
|
38
36
|
//# sourceMappingURL=prompt_selector.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"prompt_selector.js","sourceRoot":"","sources":["../../src/chains/prompt_selector.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"prompt_selector.js","sourceRoot":"","sources":["../../src/chains/prompt_selector.ts"],"names":[],"mappings":"AAKA,MAAM,OAAgB,kBAAkB;CAEvC;AAED,MAAM,OAAO,yBAA0B,SAAQ,kBAAkB;IAO/D,YACE,cAAkC,EAClC,eAKI,EAAE;QAEN,KAAK,EAAE,CAAC;QAfV;;;;;WAAkC;QAElC;;;;;WAEE;QAYA,IAAI,CAAC,aAAa,GAAG,cAAc,CAAC;QACpC,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;IAED,SAAS,CAAC,GAAsB;QAC9B,KAAK,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE;YACnD,IAAI,SAAS,CAAC,GAAG,CAAC,EAAE;gBAClB,OAAO,MAAM,CAAC;aACf;SACF;QACD,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;CACF;AAED,MAAM,UAAU,KAAK,CAAC,GAAsB;IAC1C,OAAO,GAAG,CAAC,UAAU,EAAE,KAAK,UAAU,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,GAAsB;IAChD,OAAO,GAAG,CAAC,UAAU,EAAE,KAAK,iBAAiB,CAAC;AAChD,CAAC"}
|
package/dist/document.cjs
CHANGED
package/dist/document.d.ts
CHANGED
|
@@ -1,12 +1,12 @@
|
|
|
1
|
-
export interface DocumentParams {
|
|
1
|
+
export interface DocumentParams<Metadata extends Record<string, any> = Record<string, any>> {
|
|
2
2
|
pageContent: string;
|
|
3
|
-
metadata:
|
|
3
|
+
metadata: Metadata;
|
|
4
4
|
}
|
|
5
5
|
/**
|
|
6
6
|
* Interface for interacting with a document.
|
|
7
7
|
*/
|
|
8
|
-
export declare class Document implements DocumentParams {
|
|
8
|
+
export declare class Document<Metadata extends Record<string, any> = Record<string, any>> implements DocumentParams {
|
|
9
9
|
pageContent: string;
|
|
10
|
-
metadata:
|
|
11
|
-
constructor(fields?: Partial<DocumentParams
|
|
10
|
+
metadata: Metadata;
|
|
11
|
+
constructor(fields?: Partial<DocumentParams<Metadata>>);
|
|
12
12
|
}
|
package/dist/document.js
CHANGED
package/dist/document.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"document.js","sourceRoot":"","sources":["../src/document.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"document.js","sourceRoot":"","sources":["../src/document.ts"],"names":[],"mappings":"AASA;;GAEG;AACH,MAAM,OAAO,QAAQ;IASnB,YAAY,MAA0C;QAJtD;;;;;WAAoB;QAEpB;;;;;WAAmB;QAGjB,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,MAAM,EAAE,QAAQ,IAAK,EAAe,CAAC;IACvD,CAAC;CACF"}
|
package/dist/llms/base.cjs
CHANGED
package/dist/llms/base.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { InMemoryCache } from "../cache.js";
|
|
2
2
|
import { BaseLanguageModel, } from "../base_language/index.js";
|
|
3
3
|
/**
|
|
4
4
|
* LLM Wrapper. Provides an {@link call} (an {@link generate}) function that takes in a prompt (or prompts) and returns a string.
|
|
@@ -21,7 +21,7 @@ export class BaseLLM extends BaseLanguageModel {
|
|
|
21
21
|
writable: true,
|
|
22
22
|
value: void 0
|
|
23
23
|
});
|
|
24
|
-
if (cache
|
|
24
|
+
if (typeof cache === "object") {
|
|
25
25
|
this.cache = cache;
|
|
26
26
|
}
|
|
27
27
|
else if (cache) {
|
package/dist/llms/base.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/llms/base.ts"],"names":[],"mappings":"AAAA,OAAO,
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/llms/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAa,aAAa,EAAE,MAAM,aAAa,CAAC;AAEvD,OAAO,EACL,iBAAiB,GAElB,MAAM,2BAA2B,CAAC;AAgBnC;;GAEG;AACH,MAAM,OAAgB,OAAQ,SAAQ,iBAAiB;IAQrD,YAAY,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,IAAI,EAAiB;QACxD,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,WAAW,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QARvE;;WAEG;QACH;;;;;WAAa;QAEb;;;;;WAAkB;QAIhB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;YAC7B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;SACpB;aAAM,IAAI,KAAK,EAAE;YAChB,IAAI,CAAC,KAAK,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC;SACrC;aAAM;YACL,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;SACxB;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,YAA+B,EAC/B,IAAe;QAEf,MAAM,OAAO,GAAa,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CACzD,WAAW,CAAC,QAAQ,EAAE,CACvB,CAAC;QACF,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IACtC,CAAC;IAOD,cAAc;IACd,KAAK,CAAC,iBAAiB,CACrB,OAAiB,EACjB,IAAe;QAEf,MAAM,IAAI,CAAC,eAAe,CAAC,cAAc,CACvC,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EACzB,OAAO,EACP,IAAI,CAAC,OAAO,CACb,CAAC;QACF,IAAI,MAAM,CAAC;QACX,IAAI;YACF,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;SAC9C;QAAC,OAAO,GAAG,EAAE;YACZ,MAAM,IAAI,CAAC,eAAe,CAAC,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YAC7D,MAAM,GAAG,CAAC;SACX;QAED,MAAM,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,OAAiB,EAAE,IAAe;QAC/C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YAC3B,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;SACpE;QAED,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;YACf,OAAO,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;SAC9C;QAED,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;QAEnB,MAAM,YAAY,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;QACxD,MAAM,oBAAoB,GAAa,EAAE,CAAC;QAC1C,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,GAAG,CACnC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE;YAClC,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YACxD,IAAI,CAAC,MAAM,EAAE;gBACX,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;aAClC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CACH,CAAC;QAEF,IAAI,SAAS,GAAG,EAAE,CAAC;QACnB,IAAI,oBAAoB,CAAC,MAAM,GAAG,CAAC,EAAE;YACnC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAC1C,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAC3C,IAAI,CACL,CAAC;YACF,MAAM,OAAO,CAAC,GAAG,CACf,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE;gBAClD,MAAM,WAAW,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;gBAChD,WAAW,CAAC,WAAW,CAAC,GAAG,UAAU,CAAC;gBACtC,OAAO,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;YACtE,CAAC,CAAC,CACH,CAAC;YACF,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,EAAE,CAAC;SACrC;QAED,OAAO,EAAE,WAAW,EAAE,SAAS,EAAe,CAAC;IACjD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,MAAc,EAAE,IAAe;QACxC,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;QAC5D,OAAO,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,8DAA8D;IAC9D,kBAAkB;QAChB,OAAO,EAAE,CAAC;IACZ,CAAC;IAOD;;OAEG;IACH,SAAS;QACP,OAAO;YACL,GAAG,IAAI,CAAC,kBAAkB,EAAE;YAC5B,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE;YACtB,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE;SAC1B,CAAC;IACJ,CAAC;IAED,UAAU;QACR,OAAO,UAAmB,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAmB;QAC1C,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,CAAC;QACxC,IAAI,MAAM,IAAI,MAAM,KAAK,UAAU,EAAE;YACnC,MAAM,IAAI,KAAK,CAAC,8BAA8B,MAAM,EAAE,CAAC,CAAC;SACzD;QACD,MAAM,GAAG,GAAG;YACV,MAAM,EAAE,CAAC,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,MAAM;SAC7C,CAAC,KAAK,CAAC,CAAC;QACT,IAAI,GAAG,KAAK,SAAS,EAAE;YACrB,MAAM,IAAI,KAAK,CAAC,8BAA8B,KAAK,EAAE,CAAC,CAAC;SACxD;QACD,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;CAGF;AAED;;;;;;GAMG;AACH,MAAM,OAAgB,GAAI,SAAQ,OAAO;IAMvC,KAAK,CAAC,SAAS,CAAC,OAAiB,EAAE,IAAe;QAChD,MAAM,WAAW,GAAG,EAAE,CAAC;QACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YAC1C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;YAChD,WAAW,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;SAC9B;QACD,OAAO,EAAE,WAAW,EAAE,CAAC;IACzB,CAAC;CACF"}
|
package/dist/memory/base.cjs
CHANGED
|
@@ -1,7 +1,6 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.getBufferString = exports.getInputValue = exports.BaseMemory = void 0;
|
|
4
|
-
const index_js_1 = require("../schema/index.cjs");
|
|
5
4
|
class BaseMemory {
|
|
6
5
|
}
|
|
7
6
|
exports.BaseMemory = BaseMemory;
|
|
@@ -20,16 +19,16 @@ function getBufferString(messages, human_prefix = "Human", ai_prefix = "AI") {
|
|
|
20
19
|
const string_messages = [];
|
|
21
20
|
for (const m of messages) {
|
|
22
21
|
let role;
|
|
23
|
-
if (m
|
|
22
|
+
if (m._getType() === "human") {
|
|
24
23
|
role = human_prefix;
|
|
25
24
|
}
|
|
26
|
-
else if (m
|
|
25
|
+
else if (m._getType() === "ai") {
|
|
27
26
|
role = ai_prefix;
|
|
28
27
|
}
|
|
29
|
-
else if (m
|
|
28
|
+
else if (m._getType() === "system") {
|
|
30
29
|
role = "System";
|
|
31
30
|
}
|
|
32
|
-
else if (m
|
|
31
|
+
else if (m._getType() === "generic") {
|
|
33
32
|
role = m.role;
|
|
34
33
|
}
|
|
35
34
|
else {
|
package/dist/memory/base.js
CHANGED
|
@@ -1,4 +1,3 @@
|
|
|
1
|
-
import { HumanChatMessage, AIChatMessage, SystemChatMessage, ChatMessage, } from "../schema/index.js";
|
|
2
1
|
export class BaseMemory {
|
|
3
2
|
}
|
|
4
3
|
export const getInputValue = (inputValues, inputKey) => {
|
|
@@ -15,16 +14,16 @@ export function getBufferString(messages, human_prefix = "Human", ai_prefix = "A
|
|
|
15
14
|
const string_messages = [];
|
|
16
15
|
for (const m of messages) {
|
|
17
16
|
let role;
|
|
18
|
-
if (m
|
|
17
|
+
if (m._getType() === "human") {
|
|
19
18
|
role = human_prefix;
|
|
20
19
|
}
|
|
21
|
-
else if (m
|
|
20
|
+
else if (m._getType() === "ai") {
|
|
22
21
|
role = ai_prefix;
|
|
23
22
|
}
|
|
24
|
-
else if (m
|
|
23
|
+
else if (m._getType() === "system") {
|
|
25
24
|
role = "System";
|
|
26
25
|
}
|
|
27
|
-
else if (m
|
|
26
|
+
else if (m._getType() === "generic") {
|
|
28
27
|
role = m.role;
|
|
29
28
|
}
|
|
30
29
|
else {
|
package/dist/memory/base.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/memory/base.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/memory/base.ts"],"names":[],"mappings":"AAQA,MAAM,OAAgB,UAAU;CAO/B;AAED,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,WAAwB,EAAE,QAAiB,EAAE,EAAE;IAC3E,IAAI,QAAQ,KAAK,SAAS,EAAE;QAC1B,OAAO,WAAW,CAAC,QAAQ,CAAC,CAAC;KAC9B;IACD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACtC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE;QACrB,OAAO,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;KAC7B;IACD,MAAM,IAAI,KAAK,CACb,kFAAkF,IAAI,EAAE,CACzF,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,UAAU,eAAe,CAC7B,QAA2B,EAC3B,YAAY,GAAG,OAAO,EACtB,SAAS,GAAG,IAAI;IAEhB,MAAM,eAAe,GAAa,EAAE,CAAC;IACrC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE;QACxB,IAAI,IAAY,CAAC;QACjB,IAAI,CAAC,CAAC,QAAQ,EAAE,KAAK,OAAO,EAAE;YAC5B,IAAI,GAAG,YAAY,CAAC;SACrB;aAAM,IAAI,CAAC,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;YAChC,IAAI,GAAG,SAAS,CAAC;SAClB;aAAM,IAAI,CAAC,CAAC,QAAQ,EAAE,KAAK,QAAQ,EAAE;YACpC,IAAI,GAAG,QAAQ,CAAC;SACjB;aAAM,IAAI,CAAC,CAAC,QAAQ,EAAE,KAAK,SAAS,EAAE;YACrC,IAAI,GAAI,CAAiB,CAAC,IAAI,CAAC;SAChC;aAAM;YACL,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,EAAE,CAAC,CAAC;SACvD;QACD,eAAe,CAAC,IAAI,CAAC,GAAG,IAAI,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;KAC5C;IACD,OAAO,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACpC,CAAC"}
|
|
@@ -7,8 +7,8 @@ export declare class ChatMessageHistory extends BaseChatMessageHistory {
|
|
|
7
7
|
addAIChatMessage(message: string): void;
|
|
8
8
|
}
|
|
9
9
|
export interface BaseMemoryInput {
|
|
10
|
-
chatHistory
|
|
11
|
-
returnMessages
|
|
10
|
+
chatHistory?: ChatMessageHistory;
|
|
11
|
+
returnMessages?: boolean;
|
|
12
12
|
inputKey?: string;
|
|
13
13
|
outputKey?: string;
|
|
14
14
|
}
|
|
@@ -17,6 +17,6 @@ export declare abstract class BaseChatMemory extends BaseMemory {
|
|
|
17
17
|
returnMessages: boolean;
|
|
18
18
|
inputKey?: string;
|
|
19
19
|
outputKey?: string;
|
|
20
|
-
constructor(fields?:
|
|
20
|
+
constructor(fields?: BaseMemoryInput);
|
|
21
21
|
saveContext(inputValues: InputValues, outputValues: OutputValues): Promise<void>;
|
|
22
22
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"chat_memory.js","sourceRoot":"","sources":["../../src/memory/chat_memory.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,gBAAgB,EAChB,aAAa,EAEb,sBAAsB,GACvB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,UAAU,EAGV,aAAa,GACd,MAAM,WAAW,CAAC;AAEnB,MAAM,OAAO,kBAAmB,SAAQ,sBAAsB;IAG5D,YAAY,QAA4B;QACtC,KAAK,EAAE,CAAC;QAHV;;;;mBAA8B,EAAE;WAAC;QAI/B,IAAI,CAAC,QAAQ,GAAG,QAAQ,IAAI,EAAE,CAAC;IACjC,CAAC;IAED,cAAc,CAAC,OAAe;QAC5B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;IACpD,CAAC;IAED,gBAAgB,CAAC,OAAe;QAC9B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;IACjD,CAAC;CACF;AASD,MAAM,OAAgB,cAAe,SAAQ,UAAU;IASrD,YAAY,
|
|
1
|
+
{"version":3,"file":"chat_memory.js","sourceRoot":"","sources":["../../src/memory/chat_memory.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,gBAAgB,EAChB,aAAa,EAEb,sBAAsB,GACvB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,UAAU,EAGV,aAAa,GACd,MAAM,WAAW,CAAC;AAEnB,MAAM,OAAO,kBAAmB,SAAQ,sBAAsB;IAG5D,YAAY,QAA4B;QACtC,KAAK,EAAE,CAAC;QAHV;;;;mBAA8B,EAAE;WAAC;QAI/B,IAAI,CAAC,QAAQ,GAAG,QAAQ,IAAI,EAAE,CAAC;IACjC,CAAC;IAED,cAAc,CAAC,OAAe;QAC5B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;IACpD,CAAC;IAED,gBAAgB,CAAC,OAAe;QAC9B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;IACjD,CAAC;CACF;AASD,MAAM,OAAgB,cAAe,SAAQ,UAAU;IASrD,YAAY,MAAwB;QAClC,KAAK,EAAE,CAAC;QATV;;;;;WAAgC;QAEhC;;;;mBAAiB,KAAK;WAAC;QAEvB;;;;;WAAkB;QAElB;;;;;WAAmB;QAIjB,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,IAAI,kBAAkB,EAAE,CAAC;QACnE,IAAI,CAAC,cAAc,GAAG,MAAM,EAAE,cAAc,IAAI,IAAI,CAAC,cAAc,CAAC;QACpE,IAAI,CAAC,QAAQ,GAAG,MAAM,EAAE,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC;QAClD,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,WAAW,CACf,WAAwB,EACxB,YAA0B;QAE1B,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,aAAa,CAAC,WAAW,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC3E,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAC/B,aAAa,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,CAC5C,CAAC;IACJ,CAAC;CACF"}
|
package/dist/memory/index.cjs
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.ChatMessageHistory = exports.BaseChatMemory = exports.BufferWindowMemory = exports.BaseMemory = exports.BufferMemory = void 0;
|
|
3
|
+
exports.MotorheadMemory = exports.ChatMessageHistory = exports.BaseChatMemory = exports.BufferWindowMemory = exports.BaseMemory = exports.BufferMemory = void 0;
|
|
4
4
|
var buffer_memory_js_1 = require("./buffer_memory.cjs");
|
|
5
5
|
Object.defineProperty(exports, "BufferMemory", { enumerable: true, get: function () { return buffer_memory_js_1.BufferMemory; } });
|
|
6
6
|
var base_js_1 = require("./base.cjs");
|
|
@@ -10,4 +10,6 @@ Object.defineProperty(exports, "BufferWindowMemory", { enumerable: true, get: fu
|
|
|
10
10
|
var chat_memory_js_1 = require("./chat_memory.cjs");
|
|
11
11
|
Object.defineProperty(exports, "BaseChatMemory", { enumerable: true, get: function () { return chat_memory_js_1.BaseChatMemory; } });
|
|
12
12
|
Object.defineProperty(exports, "ChatMessageHistory", { enumerable: true, get: function () { return chat_memory_js_1.ChatMessageHistory; } });
|
|
13
|
+
var motorhead_memory_js_1 = require("./motorhead_memory.cjs");
|
|
14
|
+
Object.defineProperty(exports, "MotorheadMemory", { enumerable: true, get: function () { return motorhead_memory_js_1.MotorheadMemory; } });
|
|
13
15
|
//# sourceMappingURL=index.js.map
|
package/dist/memory/index.d.ts
CHANGED
|
@@ -2,3 +2,4 @@ export { BufferMemory } from "./buffer_memory.js";
|
|
|
2
2
|
export { BaseMemory } from "./base.js";
|
|
3
3
|
export { BufferWindowMemory } from "./buffer_window_memory.js";
|
|
4
4
|
export { BaseChatMemory, ChatMessageHistory } from "./chat_memory.js";
|
|
5
|
+
export { MotorheadMemory } from "./motorhead_memory.js";
|
package/dist/memory/index.js
CHANGED
|
@@ -2,4 +2,5 @@ export { BufferMemory } from "./buffer_memory.js";
|
|
|
2
2
|
export { BaseMemory } from "./base.js";
|
|
3
3
|
export { BufferWindowMemory } from "./buffer_window_memory.js";
|
|
4
4
|
export { BaseChatMemory, ChatMessageHistory } from "./chat_memory.js";
|
|
5
|
+
export { MotorheadMemory } from "./motorhead_memory.js";
|
|
5
6
|
//# sourceMappingURL=index.js.map
|
package/dist/memory/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/memory/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC"}
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/memory/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtE,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC"}
|
|
@@ -0,0 +1,106 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.MotorheadMemory = void 0;
|
|
4
|
+
const chat_memory_js_1 = require("./chat_memory.cjs");
|
|
5
|
+
const base_js_1 = require("./base.cjs");
|
|
6
|
+
const index_js_1 = require("../util/index.cjs");
|
|
7
|
+
const async_caller_js_1 = require("../util/async_caller.cjs");
|
|
8
|
+
class MotorheadMemory extends chat_memory_js_1.BaseChatMemory {
|
|
9
|
+
constructor(fields) {
|
|
10
|
+
const { sessionId, motorheadURL, memoryKey, timeout, returnMessages, inputKey, outputKey, chatHistory, ...rest } = fields;
|
|
11
|
+
super({ returnMessages, inputKey, outputKey, chatHistory });
|
|
12
|
+
Object.defineProperty(this, "motorheadURL", {
|
|
13
|
+
enumerable: true,
|
|
14
|
+
configurable: true,
|
|
15
|
+
writable: true,
|
|
16
|
+
value: "localhost:8080"
|
|
17
|
+
});
|
|
18
|
+
Object.defineProperty(this, "timeout", {
|
|
19
|
+
enumerable: true,
|
|
20
|
+
configurable: true,
|
|
21
|
+
writable: true,
|
|
22
|
+
value: 3000
|
|
23
|
+
});
|
|
24
|
+
Object.defineProperty(this, "memoryKey", {
|
|
25
|
+
enumerable: true,
|
|
26
|
+
configurable: true,
|
|
27
|
+
writable: true,
|
|
28
|
+
value: "history"
|
|
29
|
+
});
|
|
30
|
+
Object.defineProperty(this, "sessionId", {
|
|
31
|
+
enumerable: true,
|
|
32
|
+
configurable: true,
|
|
33
|
+
writable: true,
|
|
34
|
+
value: void 0
|
|
35
|
+
});
|
|
36
|
+
Object.defineProperty(this, "context", {
|
|
37
|
+
enumerable: true,
|
|
38
|
+
configurable: true,
|
|
39
|
+
writable: true,
|
|
40
|
+
value: void 0
|
|
41
|
+
});
|
|
42
|
+
Object.defineProperty(this, "caller", {
|
|
43
|
+
enumerable: true,
|
|
44
|
+
configurable: true,
|
|
45
|
+
writable: true,
|
|
46
|
+
value: void 0
|
|
47
|
+
});
|
|
48
|
+
this.caller = new async_caller_js_1.AsyncCaller(rest);
|
|
49
|
+
this.sessionId = sessionId;
|
|
50
|
+
this.motorheadURL = motorheadURL ?? this.motorheadURL;
|
|
51
|
+
this.memoryKey = memoryKey ?? this.memoryKey;
|
|
52
|
+
this.timeout = timeout ?? this.timeout;
|
|
53
|
+
}
|
|
54
|
+
async init() {
|
|
55
|
+
const res = await this.caller.call(index_js_1.fetchWithTimeout, `${this.motorheadURL}/sessions/${this.sessionId}/memory`, {
|
|
56
|
+
timeout: this.timeout,
|
|
57
|
+
headers: {
|
|
58
|
+
"Content-Type": "application/json",
|
|
59
|
+
},
|
|
60
|
+
});
|
|
61
|
+
const { messages = [], context = "NONE" } = await res.json();
|
|
62
|
+
messages.forEach((message) => {
|
|
63
|
+
if (message.role === "AI") {
|
|
64
|
+
this.chatHistory.addAIChatMessage(message.content);
|
|
65
|
+
}
|
|
66
|
+
else {
|
|
67
|
+
this.chatHistory.addUserMessage(message.content);
|
|
68
|
+
}
|
|
69
|
+
});
|
|
70
|
+
if (context && context !== "NONE") {
|
|
71
|
+
this.context = context;
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
async loadMemoryVariables(_values) {
|
|
75
|
+
if (this.returnMessages) {
|
|
76
|
+
const result = {
|
|
77
|
+
[this.memoryKey]: this.chatHistory.messages,
|
|
78
|
+
};
|
|
79
|
+
return result;
|
|
80
|
+
}
|
|
81
|
+
const result = {
|
|
82
|
+
[this.memoryKey]: (0, base_js_1.getBufferString)(this.chatHistory.messages),
|
|
83
|
+
};
|
|
84
|
+
return result;
|
|
85
|
+
}
|
|
86
|
+
async saveContext(inputValues, outputValues) {
|
|
87
|
+
await Promise.all([
|
|
88
|
+
this.caller.call(index_js_1.fetchWithTimeout, `${this.motorheadURL}/sessions/${this.sessionId}/memory`, {
|
|
89
|
+
timeout: this.timeout,
|
|
90
|
+
method: "POST",
|
|
91
|
+
body: JSON.stringify({
|
|
92
|
+
messages: [
|
|
93
|
+
{ role: "Human", content: `${inputValues.input}` },
|
|
94
|
+
{ role: "AI", content: `${outputValues.response}` },
|
|
95
|
+
],
|
|
96
|
+
}),
|
|
97
|
+
headers: {
|
|
98
|
+
"Content-Type": "application/json",
|
|
99
|
+
},
|
|
100
|
+
}),
|
|
101
|
+
super.saveContext(inputValues, outputValues),
|
|
102
|
+
]);
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
exports.MotorheadMemory = MotorheadMemory;
|
|
106
|
+
//# sourceMappingURL=motorhead_memory.js.map
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
import { BaseChatMemory, BaseMemoryInput } from "./chat_memory.js";
|
|
2
|
+
import { InputValues, OutputValues, MemoryVariables } from "./base.js";
|
|
3
|
+
import { AsyncCaller, AsyncCallerParams } from "../util/async_caller.js";
|
|
4
|
+
export interface MotorheadMemoryMessage {
|
|
5
|
+
role: string;
|
|
6
|
+
content: string;
|
|
7
|
+
}
|
|
8
|
+
export type MotorheadMemoryInput = BaseMemoryInput & AsyncCallerParams & {
|
|
9
|
+
sessionId: string;
|
|
10
|
+
motorheadURL?: string;
|
|
11
|
+
memoryKey?: string;
|
|
12
|
+
timeout?: number;
|
|
13
|
+
};
|
|
14
|
+
export declare class MotorheadMemory extends BaseChatMemory {
|
|
15
|
+
motorheadURL: string;
|
|
16
|
+
timeout: number;
|
|
17
|
+
memoryKey: string;
|
|
18
|
+
sessionId: string;
|
|
19
|
+
context?: string;
|
|
20
|
+
caller: AsyncCaller;
|
|
21
|
+
constructor(fields: MotorheadMemoryInput);
|
|
22
|
+
init(): Promise<void>;
|
|
23
|
+
loadMemoryVariables(_values: InputValues): Promise<MemoryVariables>;
|
|
24
|
+
saveContext(inputValues: InputValues, outputValues: OutputValues): Promise<void>;
|
|
25
|
+
}
|
|
@@ -0,0 +1,102 @@
|
|
|
1
|
+
import { BaseChatMemory } from "./chat_memory.js";
|
|
2
|
+
import { getBufferString, } from "./base.js";
|
|
3
|
+
import { fetchWithTimeout } from "../util/index.js";
|
|
4
|
+
import { AsyncCaller } from "../util/async_caller.js";
|
|
5
|
+
export class MotorheadMemory extends BaseChatMemory {
|
|
6
|
+
constructor(fields) {
|
|
7
|
+
const { sessionId, motorheadURL, memoryKey, timeout, returnMessages, inputKey, outputKey, chatHistory, ...rest } = fields;
|
|
8
|
+
super({ returnMessages, inputKey, outputKey, chatHistory });
|
|
9
|
+
Object.defineProperty(this, "motorheadURL", {
|
|
10
|
+
enumerable: true,
|
|
11
|
+
configurable: true,
|
|
12
|
+
writable: true,
|
|
13
|
+
value: "localhost:8080"
|
|
14
|
+
});
|
|
15
|
+
Object.defineProperty(this, "timeout", {
|
|
16
|
+
enumerable: true,
|
|
17
|
+
configurable: true,
|
|
18
|
+
writable: true,
|
|
19
|
+
value: 3000
|
|
20
|
+
});
|
|
21
|
+
Object.defineProperty(this, "memoryKey", {
|
|
22
|
+
enumerable: true,
|
|
23
|
+
configurable: true,
|
|
24
|
+
writable: true,
|
|
25
|
+
value: "history"
|
|
26
|
+
});
|
|
27
|
+
Object.defineProperty(this, "sessionId", {
|
|
28
|
+
enumerable: true,
|
|
29
|
+
configurable: true,
|
|
30
|
+
writable: true,
|
|
31
|
+
value: void 0
|
|
32
|
+
});
|
|
33
|
+
Object.defineProperty(this, "context", {
|
|
34
|
+
enumerable: true,
|
|
35
|
+
configurable: true,
|
|
36
|
+
writable: true,
|
|
37
|
+
value: void 0
|
|
38
|
+
});
|
|
39
|
+
Object.defineProperty(this, "caller", {
|
|
40
|
+
enumerable: true,
|
|
41
|
+
configurable: true,
|
|
42
|
+
writable: true,
|
|
43
|
+
value: void 0
|
|
44
|
+
});
|
|
45
|
+
this.caller = new AsyncCaller(rest);
|
|
46
|
+
this.sessionId = sessionId;
|
|
47
|
+
this.motorheadURL = motorheadURL ?? this.motorheadURL;
|
|
48
|
+
this.memoryKey = memoryKey ?? this.memoryKey;
|
|
49
|
+
this.timeout = timeout ?? this.timeout;
|
|
50
|
+
}
|
|
51
|
+
async init() {
|
|
52
|
+
const res = await this.caller.call(fetchWithTimeout, `${this.motorheadURL}/sessions/${this.sessionId}/memory`, {
|
|
53
|
+
timeout: this.timeout,
|
|
54
|
+
headers: {
|
|
55
|
+
"Content-Type": "application/json",
|
|
56
|
+
},
|
|
57
|
+
});
|
|
58
|
+
const { messages = [], context = "NONE" } = await res.json();
|
|
59
|
+
messages.forEach((message) => {
|
|
60
|
+
if (message.role === "AI") {
|
|
61
|
+
this.chatHistory.addAIChatMessage(message.content);
|
|
62
|
+
}
|
|
63
|
+
else {
|
|
64
|
+
this.chatHistory.addUserMessage(message.content);
|
|
65
|
+
}
|
|
66
|
+
});
|
|
67
|
+
if (context && context !== "NONE") {
|
|
68
|
+
this.context = context;
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
async loadMemoryVariables(_values) {
|
|
72
|
+
if (this.returnMessages) {
|
|
73
|
+
const result = {
|
|
74
|
+
[this.memoryKey]: this.chatHistory.messages,
|
|
75
|
+
};
|
|
76
|
+
return result;
|
|
77
|
+
}
|
|
78
|
+
const result = {
|
|
79
|
+
[this.memoryKey]: getBufferString(this.chatHistory.messages),
|
|
80
|
+
};
|
|
81
|
+
return result;
|
|
82
|
+
}
|
|
83
|
+
async saveContext(inputValues, outputValues) {
|
|
84
|
+
await Promise.all([
|
|
85
|
+
this.caller.call(fetchWithTimeout, `${this.motorheadURL}/sessions/${this.sessionId}/memory`, {
|
|
86
|
+
timeout: this.timeout,
|
|
87
|
+
method: "POST",
|
|
88
|
+
body: JSON.stringify({
|
|
89
|
+
messages: [
|
|
90
|
+
{ role: "Human", content: `${inputValues.input}` },
|
|
91
|
+
{ role: "AI", content: `${outputValues.response}` },
|
|
92
|
+
],
|
|
93
|
+
}),
|
|
94
|
+
headers: {
|
|
95
|
+
"Content-Type": "application/json",
|
|
96
|
+
},
|
|
97
|
+
}),
|
|
98
|
+
super.saveContext(inputValues, outputValues),
|
|
99
|
+
]);
|
|
100
|
+
}
|
|
101
|
+
}
|
|
102
|
+
//# sourceMappingURL=motorhead_memory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"motorhead_memory.js","sourceRoot":"","sources":["../../src/memory/motorhead_memory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAmB,MAAM,kBAAkB,CAAC;AACnE,OAAO,EAIL,eAAe,GAChB,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAqB,MAAM,yBAAyB,CAAC;AAezE,MAAM,OAAO,eAAgB,SAAQ,cAAc;IAajD,YAAY,MAA4B;QACtC,MAAM,EACJ,SAAS,EACT,YAAY,EACZ,SAAS,EACT,OAAO,EACP,cAAc,EACd,QAAQ,EACR,SAAS,EACT,WAAW,EACX,GAAG,IAAI,EACR,GAAG,MAAM,CAAC;QACX,KAAK,CAAC,EAAE,cAAc,EAAE,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAC;QAxB9D;;;;mBAAe,gBAAgB;WAAC;QAEhC;;;;mBAAU,IAAI;WAAC;QAEf;;;;mBAAY,SAAS;WAAC;QAEtB;;;;;WAAkB;QAElB;;;;;WAAiB;QAEjB;;;;;WAAoB;QAgBlB,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,YAAY,GAAG,YAAY,IAAI,IAAI,CAAC,YAAY,CAAC;QACtD,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QAC7C,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,IAAI;QACR,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAChC,gBAAgB,EAChB,GAAG,IAAI,CAAC,YAAY,aAAa,IAAI,CAAC,SAAS,SAAS,EACxD;YACE,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;aACnC;SACF,CACF,CAAC;QAEF,MAAM,EAAE,QAAQ,GAAG,EAAE,EAAE,OAAO,GAAG,MAAM,EAAE,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAE7D,QAAQ,CAAC,OAAO,CAAC,CAAC,OAA+B,EAAE,EAAE;YACnD,IAAI,OAAO,CAAC,IAAI,KAAK,IAAI,EAAE;gBACzB,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;aACpD;iBAAM;gBACL,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;aAClD;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,OAAO,IAAI,OAAO,KAAK,MAAM,EAAE;YACjC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;SACxB;IACH,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,OAAoB;QAC5C,IAAI,IAAI,CAAC,cAAc,EAAE;YACvB,MAAM,MAAM,GAAG;gBACb,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ;aAC5C,CAAC;YACF,OAAO,MAAM,CAAC;SACf;QACD,MAAM,MAAM,GAAG;YACb,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC;SAC7D,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,WAAW,CACf,WAAwB,EACxB,YAA0B;QAE1B,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,gBAAgB,EAChB,GAAG,IAAI,CAAC,YAAY,aAAa,IAAI,CAAC,SAAS,SAAS,EACxD;gBACE,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,QAAQ,EAAE;wBACR,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,WAAW,CAAC,KAAK,EAAE,EAAE;wBAClD,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,YAAY,CAAC,QAAQ,EAAE,EAAE;qBACpD;iBACF,CAAC;gBACF,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;aACF,CACF;YACD,KAAK,CAAC,WAAW,CAAC,WAAW,EAAE,YAAY,CAAC;SAC7C,CAAC,CAAC;IACL,CAAC;CACF"}
|
|
@@ -32,6 +32,7 @@ class OutputFixingParser extends index_js_1.BaseOutputParser {
|
|
|
32
32
|
return await this.parser.parse(completion);
|
|
33
33
|
}
|
|
34
34
|
catch (e) {
|
|
35
|
+
// eslint-disable-next-line no-instanceof/no-instanceof
|
|
35
36
|
if (e instanceof index_js_1.OutputParserException) {
|
|
36
37
|
const result = await this.retryChain.call({
|
|
37
38
|
instructions: this.parser.getFormatInstructions(),
|
|
@@ -29,6 +29,7 @@ export class OutputFixingParser extends BaseOutputParser {
|
|
|
29
29
|
return await this.parser.parse(completion);
|
|
30
30
|
}
|
|
31
31
|
catch (e) {
|
|
32
|
+
// eslint-disable-next-line no-instanceof/no-instanceof
|
|
32
33
|
if (e instanceof OutputParserException) {
|
|
33
34
|
const result = await this.retryChain.call({
|
|
34
35
|
instructions: this.parser.getFormatInstructions(),
|