@ax-llm/ax 9.0.61 → 10.0.2
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/ai/base.js +2 -3
- package/ai/base.js.map +1 -1
- package/ai/types.d.ts +2 -0
- package/cjs/ai/base.js +2 -3
- package/cjs/ai/base.js.map +1 -1
- package/cjs/ai/types.d.ts +2 -0
- package/cjs/docs/manager.js +2 -2
- package/cjs/docs/manager.js.map +1 -1
- package/cjs/docs/reranker.d.ts +4 -4
- package/cjs/docs/reranker.js +5 -5
- package/cjs/docs/reranker.js.map +1 -1
- package/cjs/docs/rewriter.d.ts +3 -3
- package/cjs/docs/rewriter.js +3 -3
- package/cjs/docs/rewriter.js.map +1 -1
- package/cjs/dsp/evaluate.d.ts +4 -1
- package/cjs/dsp/evaluate.js +4 -2
- package/cjs/dsp/evaluate.js.map +1 -1
- package/cjs/dsp/functions.js +2 -1
- package/cjs/dsp/functions.js.map +1 -1
- package/cjs/dsp/generate.d.ts +8 -6
- package/cjs/dsp/generate.js +47 -42
- package/cjs/dsp/generate.js.map +1 -1
- package/cjs/dsp/index.d.ts +0 -1
- package/cjs/dsp/index.js +0 -1
- package/cjs/dsp/index.js.map +1 -1
- package/cjs/dsp/optimize.d.ts +4 -1
- package/cjs/dsp/optimize.js +4 -2
- package/cjs/dsp/optimize.js.map +1 -1
- package/cjs/dsp/program.d.ts +7 -4
- package/cjs/dsp/program.js +10 -3
- package/cjs/dsp/program.js.map +1 -1
- package/cjs/dsp/prompt.js +19 -8
- package/cjs/dsp/prompt.js.map +1 -1
- package/cjs/dsp/util.js +6 -3
- package/cjs/dsp/util.js.map +1 -1
- package/cjs/funcs/code.js +1 -1
- package/cjs/funcs/code.js.map +1 -1
- package/cjs/prompts/agent.d.ts +8 -4
- package/cjs/prompts/agent.js +23 -11
- package/cjs/prompts/agent.js.map +1 -1
- package/cjs/prompts/cot.d.ts +3 -4
- package/cjs/prompts/cot.js +3 -3
- package/cjs/prompts/cot.js.map +1 -1
- package/cjs/prompts/rag.d.ts +4 -4
- package/cjs/prompts/rag.js +6 -6
- package/cjs/prompts/rag.js.map +1 -1
- package/cjs/prompts/react.d.ts +2 -3
- package/cjs/prompts/react.js +2 -2
- package/cjs/prompts/react.js.map +1 -1
- package/cjs/util/apicall.js +2 -2
- package/cjs/util/apicall.js.map +1 -1
- package/cjs/util/sse.d.ts +12 -0
- package/cjs/util/sse.js +97 -0
- package/cjs/util/sse.js.map +1 -0
- package/cjs/util/transform.d.ts +0 -3
- package/cjs/util/transform.js +1 -32
- package/cjs/util/transform.js.map +1 -1
- package/docs/manager.js +2 -2
- package/docs/manager.js.map +1 -1
- package/docs/reranker.d.ts +4 -4
- package/docs/reranker.js +6 -6
- package/docs/reranker.js.map +1 -1
- package/docs/rewriter.d.ts +3 -3
- package/docs/rewriter.js +4 -4
- package/docs/rewriter.js.map +1 -1
- package/dsp/evaluate.d.ts +4 -1
- package/dsp/evaluate.js +4 -2
- package/dsp/evaluate.js.map +1 -1
- package/dsp/functions.js +2 -1
- package/dsp/functions.js.map +1 -1
- package/dsp/generate.d.ts +8 -6
- package/dsp/generate.js +44 -40
- package/dsp/generate.js.map +1 -1
- package/dsp/index.d.ts +0 -1
- package/dsp/index.js +0 -1
- package/dsp/index.js.map +1 -1
- package/dsp/optimize.d.ts +4 -1
- package/dsp/optimize.js +4 -2
- package/dsp/optimize.js.map +1 -1
- package/dsp/program.d.ts +7 -4
- package/dsp/program.js +10 -3
- package/dsp/program.js.map +1 -1
- package/dsp/prompt.js +19 -8
- package/dsp/prompt.js.map +1 -1
- package/dsp/util.js +6 -3
- package/dsp/util.js.map +1 -1
- package/funcs/code.js +1 -1
- package/funcs/code.js.map +1 -1
- package/package.json +1 -1
- package/prompts/agent.d.ts +8 -4
- package/prompts/agent.js +23 -11
- package/prompts/agent.js.map +1 -1
- package/prompts/cot.d.ts +3 -4
- package/prompts/cot.js +4 -4
- package/prompts/cot.js.map +1 -1
- package/prompts/rag.d.ts +4 -4
- package/prompts/rag.js +7 -7
- package/prompts/rag.js.map +1 -1
- package/prompts/react.d.ts +2 -3
- package/prompts/react.js +2 -2
- package/prompts/react.js.map +1 -1
- package/util/apicall.js +2 -2
- package/util/apicall.js.map +1 -1
- package/util/sse.d.ts +12 -0
- package/util/sse.js +93 -0
- package/util/sse.js.map +1 -0
- package/util/transform.d.ts +0 -3
- package/util/transform.js +0 -30
- package/util/transform.js.map +1 -1
- package/cjs/dsp/gen2.d.ts +0 -8
- package/cjs/dsp/gen2.js +0 -15
- package/cjs/dsp/gen2.js.map +0 -1
- package/dsp/gen2.d.ts +0 -8
- package/dsp/gen2.js +0 -11
- package/dsp/gen2.js.map +0 -1
package/cjs/dsp/util.js
CHANGED
|
@@ -14,7 +14,7 @@ msg) => {
|
|
|
14
14
|
const filledBar = colorLog.blueBright('█'.repeat(filledBarLength));
|
|
15
15
|
const emptyBar = ' '.repeat(emptyBarLength);
|
|
16
16
|
const itemsPerSecond = elapsedTime > 0 ? (current / elapsedTime).toFixed(2) : '0.00';
|
|
17
|
-
process.stdout.write(`\r${msg}: ${current} / ${total}
|
|
17
|
+
process.stdout.write(`\r${msg}: ${current} / ${total} (${colorLog.yellow(percentage)}%): 100%|${filledBar}${emptyBar}| Success: ${success}/${total} [${colorLog.red(elapsedTime.toFixed(2))}, ${itemsPerSecond}it/s]`);
|
|
18
18
|
};
|
|
19
19
|
exports.updateProgressBar = updateProgressBar;
|
|
20
20
|
const validateValue = (field, value) => {
|
|
@@ -36,7 +36,10 @@ const validateValue = (field, value) => {
|
|
|
36
36
|
}
|
|
37
37
|
};
|
|
38
38
|
const validImage = (val) => {
|
|
39
|
-
if (
|
|
39
|
+
if (!val ||
|
|
40
|
+
typeof val !== 'object' ||
|
|
41
|
+
!('mimeType' in val) ||
|
|
42
|
+
!('data' in val)) {
|
|
40
43
|
return false;
|
|
41
44
|
}
|
|
42
45
|
return true;
|
|
@@ -77,7 +80,7 @@ const validateValue = (field, value) => {
|
|
|
77
80
|
isValid = validateSingleValue(ft.name, value);
|
|
78
81
|
}
|
|
79
82
|
if (!isValid) {
|
|
80
|
-
throw new Error(`Validation failed: Expected '${field.name}' to be a ${
|
|
83
|
+
throw new Error(`Validation failed: Expected '${field.name}' to be a ${field.type?.isArray ? 'an array of ' : ''}${ft.name} instead got '${typeof value}' (${value})`);
|
|
81
84
|
}
|
|
82
85
|
};
|
|
83
86
|
exports.validateValue = validateValue;
|
package/cjs/dsp/util.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"util.js","sourceRoot":"","sources":["../../../dsp/util.ts"],"names":[],"mappings":";;;
|
|
1
|
+
{"version":3,"file":"util.js","sourceRoot":"","sources":["../../../dsp/util.ts"],"names":[],"mappings":";;;AAgHA,8CAmBC;AAnID,yDAAyD;AACzD,2CAA0C;AAK1C,MAAM,QAAQ,GAAG,IAAI,iBAAQ,EAAE,CAAC;AAEzB,MAAM,iBAAiB,GAAG,CAC/B,OAAe,EACf,KAAa,EACb,OAAe,EACf,WAAmB,EAAE,aAAa;AAClC,mBAA2B,EAAE,EAAE,oCAAoC;AACnE,GAAW,EACL,EAAE;IACR,MAAM,UAAU,GAAG,CAAC,CAAC,OAAO,GAAG,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACxD,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,gBAAgB,GAAG,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC;IACzE,MAAM,cAAc,GAAG,gBAAgB,GAAG,eAAe,CAAC;IAC1D,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC;IACnE,MAAM,QAAQ,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;IAC5C,MAAM,cAAc,GAClB,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,GAAG,WAAW,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IAEhE,OAAO,CAAC,MAAM,CAAC,KAAK,CAClB,KAAK,GAAG,KAAK,OAAO,MAAM,KAAK,KAAK,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,SAAS,GAAG,QAAQ,cAAc,OAAO,IAAI,KAAK,KAAK,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,cAAc,OAAO,CACjM,CAAC;AACJ,CAAC,CAAC;AAnBW,QAAA,iBAAiB,qBAmB5B;AAEK,MAAM,aAAa,GAAG,CAC3B,KAAwB,EACxB,KAA6B,EACvB,EAAE;IACR,MAAM,EAAE,GAAG,KAAK,CAAC,IAAI,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAE5D,MAAM,mBAAmB,GAAG,CAC1B,YAAoB,EACpB,GAA2B,EAClB,EAAE;QACX,QAAQ,YAAY,EAAE,CAAC;YACrB,KAAK,QAAQ;gBACX,OAAO,OAAO,GAAG,KAAK,QAAQ,CAAC;YACjC,KAAK,QAAQ;gBACX,OAAO,OAAO,GAAG,KAAK,QAAQ,CAAC;YACjC,KAAK,SAAS;gBACZ,OAAO,OAAO,GAAG,KAAK,SAAS,CAAC;YAClC,KAAK,MAAM;gBACT,OAAO,GAAG,YAAY,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ,CAAC;YACxD,KAAK,UAAU;gBACb,OAAO,GAAG,YAAY,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ,CAAC;YACxD;gBACE,OAAO,KAAK,CAAC,CAAC,8BAA8B;QAChD,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,UAAU,GAAG,CAAC,GAA2B,EAAW,EAAE;QAC1D,IACE,CAAC,GAAG;YACJ,OAAO,GAAG,KAAK,QAAQ;YACvB,CAAC,CAAC,UAAU,IAAI,GAAG,CAAC;YACpB,CAAC,CAAC,MAAM,IAAI,GAAG,CAAC,EAChB,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IAEF,IAAI,KAAK,CAAC,IAAI,EAAE,IAAI,KAAK,OAAO,EAAE,CAAC;QACjC,IAAI,GAAG,CAAC;QACR,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;oBACtB,GAAG,GAAG,6CAA6C,CAAC;oBACpD,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;aAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,GAAG,GAAG,6CAA6C,CAAC;QACtD,CAAC;QAED,IAAI,GAAG,EAAE,CAAC;YACR,MAAM,IAAI,KAAK,CACb,gCAAgC,KAAK,CAAC,IAAI,aAAa,GAAG,iBAAiB,KAAK,GAAG,CACpF,CAAC;QACJ,CAAC;QACD,OAAO;IACT,CAAC;IAED,IAAI,OAAO,GAAG,IAAI,CAAC;IAEnB,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;QACf,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,GAAG,KAAK,CAAC;QAClB,CAAC;aAAM,CAAC;YACN,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC,mBAAmB,CAAC,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC;oBACxC,OAAO,GAAG,KAAK,CAAC;oBAChB,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,mBAAmB,CAAC,EAAE,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC;IAED,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CACb,gCAAgC,KAAK,CAAC,IAAI,aAAa,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,IAAI,iBAAiB,OAAO,KAAK,MAAM,KAAK,GAAG,CACtJ,CAAC;IACJ,CAAC;AACH,CAAC,CAAC;AAjFW,QAAA,aAAa,iBAiFxB;AAEF,SAAgB,iBAAiB,CAC/B,MAAiC;IAEjC,MAAM,QAAQ,GAAsC,EAAE,CAAC;IAEvD,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;QACvB,MAAM,GAAG,GAAG,GAAG,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QAEzC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACnB,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC;YAC7B,OAAO;QACT,CAAC;QAED,QAAQ,CAAC,GAAG,CAAE,CAAC,YAAY,IAAI,KAAK,CAAC,YAAY,CAAC;QAClD,QAAQ,CAAC,GAAG,CAAE,CAAC,gBAAgB,IAAI,KAAK,CAAC,gBAAgB,CAAC;QAC1D,QAAQ,CAAC,GAAG,CAAE,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;AACjC,CAAC"}
|
package/cjs/funcs/code.js
CHANGED
|
@@ -67,7 +67,7 @@ class AxJSInterpreter {
|
|
|
67
67
|
toFunction() {
|
|
68
68
|
return {
|
|
69
69
|
name: 'javascriptInterpreter',
|
|
70
|
-
description: 'Use this function to run Javascript code and get any expected return value
|
|
70
|
+
description: 'Use this function to run Javascript code and get any expected return value',
|
|
71
71
|
parameters: {
|
|
72
72
|
type: 'object',
|
|
73
73
|
properties: {
|
package/cjs/funcs/code.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"code.js","sourceRoot":"","sources":["../../../funcs/code.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,gDAAkC;AAClC,4CAA8B;AAC9B,8CAAgC;AAChC,6CAA+B;AAC/B,wCAA0B;AAC1B,kDAAoC;AACpC,2BAAqC;AAIrC,IAAY,yBAMX;AAND,WAAY,yBAAyB;IACnC,2CAAc,CAAA;IACd,wCAAW,CAAA;IACX,sCAAS,CAAA;IACT,8CAAiB,CAAA;IACjB,gDAAmB,CAAA;AACrB,CAAC,EANW,yBAAyB,yCAAzB,yBAAyB,QAMpC;AAYD,MAAa,eAAe;IAClB,WAAW,CAAuC;IAE1D,YAAY,EACV,WAAW,GAAG,EAAE,KAGF,EAAE;QAChB,IAAI,CAAC,WAAW,GAAG,WAAW,IAAI,EAAE,CAAC;IACvC,CAAC;IAEO,yBAAyB,CAAC,IAAY;QAC5C,MAAM,OAAO,GAAqB,EAAE,OAAO,EAAE,CAAC;QAE9C,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,CAAC,EAAE,CAAC;YAC5D,OAAO,CAAC,EAAE,GAAG,GAAG,CAAC;QACnB,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,GAAG,CAAC,EAAE,CAAC;YAC7D,OAAO,CAAC,IAAI,GAAG,KAAK,CAAC;YACrB,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC;QACzB,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,CAAC,EAAE,CAAC;YAC5D,OAAO,CAAC,EAAE,GAAG,GAAG,CAAC;QACnB,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,MAAM,CAAC,EAAE,CAAC;YAChE,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC;QAC3B,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,OAAO,CAAC,EAAE,CAAC;YACjE,OAAO,CAAC,OAAO,GAAG,QAAQ,CAAC;QAC7B,CAAC;QAED,OAAO,IAAA,oBAAe,EAAC,iBAAiB,IAAI,OAAO,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC;IAEM,UAAU;QACf,OAAO;YACL,IAAI,EAAE,uBAAuB;YAC7B,WAAW,EACT,
|
|
1
|
+
{"version":3,"file":"code.js","sourceRoot":"","sources":["../../../funcs/code.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,gDAAkC;AAClC,4CAA8B;AAC9B,8CAAgC;AAChC,6CAA+B;AAC/B,wCAA0B;AAC1B,kDAAoC;AACpC,2BAAqC;AAIrC,IAAY,yBAMX;AAND,WAAY,yBAAyB;IACnC,2CAAc,CAAA;IACd,wCAAW,CAAA;IACX,sCAAS,CAAA;IACT,8CAAiB,CAAA;IACjB,gDAAmB,CAAA;AACrB,CAAC,EANW,yBAAyB,yCAAzB,yBAAyB,QAMpC;AAYD,MAAa,eAAe;IAClB,WAAW,CAAuC;IAE1D,YAAY,EACV,WAAW,GAAG,EAAE,KAGF,EAAE;QAChB,IAAI,CAAC,WAAW,GAAG,WAAW,IAAI,EAAE,CAAC;IACvC,CAAC;IAEO,yBAAyB,CAAC,IAAY;QAC5C,MAAM,OAAO,GAAqB,EAAE,OAAO,EAAE,CAAC;QAE9C,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,CAAC,EAAE,CAAC;YAC5D,OAAO,CAAC,EAAE,GAAG,GAAG,CAAC;QACnB,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,GAAG,CAAC,EAAE,CAAC;YAC7D,OAAO,CAAC,IAAI,GAAG,KAAK,CAAC;YACrB,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC;QACzB,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,CAAC,EAAE,CAAC;YAC5D,OAAO,CAAC,EAAE,GAAG,GAAG,CAAC;QACnB,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,MAAM,CAAC,EAAE,CAAC;YAChE,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC;QAC3B,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,OAAO,CAAC,EAAE,CAAC;YACjE,OAAO,CAAC,OAAO,GAAG,QAAQ,CAAC;QAC7B,CAAC;QAED,OAAO,IAAA,oBAAe,EAAC,iBAAiB,IAAI,OAAO,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC;IAEM,UAAU;QACf,OAAO;YACL,IAAI,EAAE,uBAAuB;YAC7B,WAAW,EACT,4EAA4E;YAC9E,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,IAAI,EAAE;wBACJ,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,yCAAyC;qBACvD;iBACF;gBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;aACnB;YAED,IAAI,EAAE,CAAC,EAAE,IAAI,EAA8B,EAAE,EAAE,CAC7C,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC;SACvC,CAAC;IACJ,CAAC;CACF;AA1DD,0CA0DC"}
|
package/cjs/prompts/agent.d.ts
CHANGED
|
@@ -1,18 +1,22 @@
|
|
|
1
1
|
import type { AxAIService, AxFunction } from '../ai/types.js';
|
|
2
|
-
import { type
|
|
2
|
+
import { type AxGenOptions, AxSignature } from '../dsp/index.js';
|
|
3
3
|
import { type AxGenIn, type AxGenOut, type AxProgramDemos, type AxProgramExamples, type AxProgramForwardOptions, type AxTunable, type AxUsable } from '../dsp/program.js';
|
|
4
4
|
export interface AxAgentic extends AxTunable, AxUsable {
|
|
5
5
|
getFunction(): AxFunction;
|
|
6
6
|
}
|
|
7
|
-
export type AxAgentOptions = Omit<
|
|
7
|
+
export type AxAgentOptions = Omit<AxGenOptions, 'functions' | 'functionCall'>;
|
|
8
8
|
export declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implements AxAgentic {
|
|
9
|
+
private ai?;
|
|
9
10
|
private signature;
|
|
10
11
|
private program;
|
|
12
|
+
private functions?;
|
|
13
|
+
private agents?;
|
|
11
14
|
private name;
|
|
12
15
|
private description;
|
|
13
16
|
private subAgentList?;
|
|
14
17
|
private func;
|
|
15
|
-
constructor(ai
|
|
18
|
+
constructor({ ai, name, description, signature, agents, functions }: Readonly<{
|
|
19
|
+
ai?: Readonly<AxAIService>;
|
|
16
20
|
name: string;
|
|
17
21
|
description: string;
|
|
18
22
|
signature: AxSignature | string;
|
|
@@ -30,5 +34,5 @@ export declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implement
|
|
|
30
34
|
})[];
|
|
31
35
|
resetUsage(): void;
|
|
32
36
|
getFunction(): AxFunction;
|
|
33
|
-
forward(values: IN, options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
|
|
37
|
+
forward(ai: Readonly<AxAIService>, values: IN, options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
|
|
34
38
|
}
|
package/cjs/prompts/agent.js
CHANGED
|
@@ -6,26 +6,29 @@ const index_js_2 = require("../trace/index.js");
|
|
|
6
6
|
const cot_js_1 = require("./cot.js");
|
|
7
7
|
const react_js_1 = require("./react.js");
|
|
8
8
|
class AxAgent {
|
|
9
|
+
ai;
|
|
9
10
|
signature;
|
|
10
11
|
program;
|
|
12
|
+
functions;
|
|
13
|
+
agents;
|
|
11
14
|
name;
|
|
12
15
|
description;
|
|
13
16
|
subAgentList;
|
|
14
17
|
func;
|
|
15
|
-
constructor(ai,
|
|
18
|
+
constructor({ ai, name, description, signature, agents, functions }, options) {
|
|
19
|
+
this.ai = ai;
|
|
16
20
|
this.signature = new index_js_1.AxSignature(signature);
|
|
21
|
+
this.functions = functions;
|
|
22
|
+
this.agents = agents;
|
|
17
23
|
const funcs = [
|
|
18
24
|
...(functions ?? []),
|
|
19
25
|
...(agents?.map((a) => a.getFunction()) ?? [])
|
|
20
26
|
];
|
|
21
|
-
const opt = {
|
|
22
|
-
...options,
|
|
23
|
-
functions: funcs
|
|
24
|
-
};
|
|
27
|
+
const opt = { ...options, functions: funcs };
|
|
25
28
|
this.program =
|
|
26
29
|
funcs.length > 0
|
|
27
|
-
? new react_js_1.AxReAct(
|
|
28
|
-
: new cot_js_1.AxChainOfThought(
|
|
30
|
+
? new react_js_1.AxReAct(this.signature, opt)
|
|
31
|
+
: new cot_js_1.AxChainOfThought(this.signature, opt);
|
|
29
32
|
if (!name || name.length < 5) {
|
|
30
33
|
throw new Error('Agent name must be at least 10 characters (more descriptive): ' + name);
|
|
31
34
|
}
|
|
@@ -69,14 +72,23 @@ class AxAgent {
|
|
|
69
72
|
}
|
|
70
73
|
getFunction() {
|
|
71
74
|
const boundFunc = this.forward.bind(this);
|
|
75
|
+
// Create a wrapper function that excludes the 'ai' parameter
|
|
76
|
+
const wrappedFunc = (values, options) => {
|
|
77
|
+
const ai = this.ai ?? options?.ai;
|
|
78
|
+
if (!ai) {
|
|
79
|
+
throw new Error('AI service is required to run the agent');
|
|
80
|
+
}
|
|
81
|
+
return boundFunc(ai, values, options);
|
|
82
|
+
};
|
|
72
83
|
return {
|
|
73
84
|
...this.func,
|
|
74
|
-
func:
|
|
85
|
+
func: wrappedFunc
|
|
75
86
|
};
|
|
76
87
|
}
|
|
77
|
-
async forward(values, options) {
|
|
88
|
+
async forward(ai, values, options) {
|
|
89
|
+
const _ai = this.ai ?? ai;
|
|
78
90
|
if (!options?.tracer) {
|
|
79
|
-
return await this.program.forward(values, options);
|
|
91
|
+
return await this.program.forward(_ai, values, options);
|
|
80
92
|
}
|
|
81
93
|
const attributes = {
|
|
82
94
|
['agent.name']: this.name,
|
|
@@ -87,7 +99,7 @@ class AxAgent {
|
|
|
87
99
|
kind: index_js_2.AxSpanKind.SERVER,
|
|
88
100
|
attributes
|
|
89
101
|
}, async (span) => {
|
|
90
|
-
const res = await this.program.forward(values, options);
|
|
102
|
+
const res = await this.program.forward(_ai, values, options);
|
|
91
103
|
span.end();
|
|
92
104
|
return res;
|
|
93
105
|
});
|
package/cjs/prompts/agent.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../prompts/agent.ts"],"names":[],"mappings":";;;AACA,
|
|
1
|
+
{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../prompts/agent.ts"],"names":[],"mappings":";;;AACA,8CAAiE;AAWjE,gDAA+C;AAE/C,qCAA4C;AAC5C,yCAAqC;AAQrC,MAAa,OAAO;IAGV,EAAE,CAAe;IACjB,SAAS,CAAc;IACvB,OAAO,CAAkC;IACzC,SAAS,CAAgB;IACzB,MAAM,CAAe;IAErB,IAAI,CAAS;IACb,WAAW,CAAS;IACpB,YAAY,CAAU;IACtB,IAAI,CAAa;IAEzB,YACE,EACE,EAAE,EACF,IAAI,EACJ,WAAW,EACX,SAAS,EACT,MAAM,EACN,SAAS,EAQT,EACF,OAAkC;QAElC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,SAAS,GAAG,IAAI,sBAAW,CAAC,SAAS,CAAC,CAAC;QAC5C,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,MAAM,KAAK,GAAiB;YAC1B,GAAG,CAAC,SAAS,IAAI,EAAE,CAAC;YACpB,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;SAC/C,CAAC;QAEF,MAAM,GAAG,GAAG,EAAE,GAAG,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAE7C,IAAI,CAAC,OAAO;YACV,KAAK,CAAC,MAAM,GAAG,CAAC;gBACd,CAAC,CAAC,IAAI,kBAAO,CAAU,IAAI,CAAC,SAAS,EAAE,GAAG,CAAC;gBAC3C,CAAC,CAAC,IAAI,yBAAgB,CAAU,IAAI,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;QAEzD,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CACb,gEAAgE,GAAG,IAAI,CACxE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CACb,4FAA4F;gBAC1F,WAAW,CACd,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAExE,IAAI,CAAC,IAAI,GAAG;YACV,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;YAC5B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE;YACzC,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO;SACzB,CAAC;QAEF,KAAK,MAAM,KAAK,IAAI,MAAM,IAAI,EAAE,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAEM,WAAW,CAAC,QAAqC;QACtD,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACrC,CAAC;IAEM,KAAK,CAAC,EAAU;QACrB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACzB,CAAC;IAEM,WAAW,CAAC,QAAgB;QACjC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACrC,CAAC;IAEM,SAAS;QACd,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;IAClC,CAAC;IAEM,QAAQ,CAAC,KAAgC;QAC9C,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAEM,QAAQ;QACb,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;IACjC,CAAC;IAEM,UAAU;QACf,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IAC5B,CAAC;IAEM,WAAW;QAChB,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE1C,6DAA6D;QAC7D,MAAM,WAAW,GAAG,CAClB,MAAU,EACV,OAA2C,EAC3C,EAAE;YACF,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,IAAI,OAAO,EAAE,EAAE,CAAC;YAClC,IAAI,CAAC,EAAE,EAAE,CAAC;gBACR,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;YAC7D,CAAC;YACD,OAAO,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QACxC,CAAC,CAAC;QAEF,OAAO;YACL,GAAG,IAAI,CAAC,IAAI;YACZ,IAAI,EAAE,WAAW;SAClB,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,OAAO,CAClB,EAAyB,EACzB,MAAU,EACV,OAA2C;QAE3C,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC;QAE1B,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YACrB,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,UAAU,GAAG;YACjB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,IAAI;YACzB,CAAC,mBAAmB,CAAC,EAAE,IAAI,CAAC,WAAW;YACvC,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,YAAY,IAAI,MAAM;SACjD,CAAC;QAEF,OAAO,MAAM,OAAO,EAAE,MAAM,CAAC,eAAe,CAC1C,OAAO,EACP;YACE,IAAI,EAAE,qBAAU,CAAC,MAAM;YACvB,UAAU;SACX,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAC7D,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF;AA7JD,0BA6JC;AAED,SAAS,WAAW,CAAC,WAAmB;IACtC,8FAA8F;IAC9F,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAEhD,wFAAwF;IACxF,MAAM,eAAe,GAAG,KAAK;SAC1B,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QACnB,qEAAqE;QACrE,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAErC,gEAAgE;QAChE,IAAI,KAAK,GAAG,CAAC,IAAI,SAAS,IAAI,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3C,OAAO,SAAS,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;SACD,IAAI,CAAC,EAAE,CAAC,CAAC;IAEZ,OAAO,eAAe,CAAC;AACzB,CAAC"}
|
package/cjs/prompts/cot.d.ts
CHANGED
|
@@ -1,9 +1,8 @@
|
|
|
1
|
-
import type
|
|
2
|
-
import { AxGenerate, type AxGenerateOptions } from '../dsp/generate.js';
|
|
1
|
+
import { AxGen, type AxGenOptions } from '../dsp/generate.js';
|
|
3
2
|
import type { AxGenIn, AxGenOut } from '../dsp/program.js';
|
|
4
3
|
import { AxSignature } from '../dsp/sig.js';
|
|
5
|
-
export declare class AxChainOfThought<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends
|
|
4
|
+
export declare class AxChainOfThought<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends AxGen<IN, OUT & {
|
|
6
5
|
reason: string;
|
|
7
6
|
}> {
|
|
8
|
-
constructor(
|
|
7
|
+
constructor(signature: Readonly<AxSignature | string>, options?: Readonly<AxGenOptions>);
|
|
9
8
|
}
|
package/cjs/prompts/cot.js
CHANGED
|
@@ -3,8 +3,8 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
3
3
|
exports.AxChainOfThought = void 0;
|
|
4
4
|
const generate_js_1 = require("../dsp/generate.js");
|
|
5
5
|
const sig_js_1 = require("../dsp/sig.js");
|
|
6
|
-
class AxChainOfThought extends generate_js_1.
|
|
7
|
-
constructor(
|
|
6
|
+
class AxChainOfThought extends generate_js_1.AxGen {
|
|
7
|
+
constructor(signature, options) {
|
|
8
8
|
const sig = new sig_js_1.AxSignature(signature);
|
|
9
9
|
const description = `Let's work this out in a step by step way in order to ensure we have the right answer.`;
|
|
10
10
|
sig.setOutputFields([
|
|
@@ -14,7 +14,7 @@ class AxChainOfThought extends generate_js_1.AxGenerate {
|
|
|
14
14
|
},
|
|
15
15
|
...sig.getOutputFields()
|
|
16
16
|
]);
|
|
17
|
-
super(
|
|
17
|
+
super(sig, options);
|
|
18
18
|
}
|
|
19
19
|
}
|
|
20
20
|
exports.AxChainOfThought = AxChainOfThought;
|
package/cjs/prompts/cot.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"cot.js","sourceRoot":"","sources":["../../../prompts/cot.ts"],"names":[],"mappings":";;;
|
|
1
|
+
{"version":3,"file":"cot.js","sourceRoot":"","sources":["../../../prompts/cot.ts"],"names":[],"mappings":";;;AAAA,oDAA8D;AAE9D,0CAA4C;AAE5C,MAAa,gBAGX,SAAQ,mBAAmC;IAC3C,YACE,SAAyC,EACzC,OAAgC;QAEhC,MAAM,GAAG,GAAG,IAAI,oBAAW,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,WAAW,GAAG,wFAAwF,CAAC;QAE7G,GAAG,CAAC,eAAe,CAAC;YAClB;gBACE,IAAI,EAAE,QAAQ;gBACd,WAAW;aACZ;YACD,GAAG,GAAG,CAAC,eAAe,EAAE;SACzB,CAAC,CAAC;QAEH,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IACtB,CAAC;CACF;AArBD,4CAqBC"}
|
package/cjs/prompts/rag.d.ts
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import type
|
|
2
|
-
import { type AxGenerateOptions } from '../dsp/index.js';
|
|
1
|
+
import { type AxGenOptions } from '../dsp/index.js';
|
|
3
2
|
import { type AxProgramForwardOptions } from '../dsp/program.js';
|
|
3
|
+
import type { AxAIService } from '../index.js';
|
|
4
4
|
import { AxChainOfThought } from './cot.js';
|
|
5
5
|
export declare class AxRAG extends AxChainOfThought<{
|
|
6
6
|
context: string[];
|
|
@@ -11,10 +11,10 @@ export declare class AxRAG extends AxChainOfThought<{
|
|
|
11
11
|
private genQuery;
|
|
12
12
|
private queryFn;
|
|
13
13
|
private maxHops;
|
|
14
|
-
constructor(
|
|
14
|
+
constructor(queryFn: (query: string) => Promise<string>, options: Readonly<AxGenOptions & {
|
|
15
15
|
maxHops?: number;
|
|
16
16
|
}>);
|
|
17
|
-
forward({ question }: Readonly<{
|
|
17
|
+
forward(ai: Readonly<AxAIService>, { question }: Readonly<{
|
|
18
18
|
question: string;
|
|
19
19
|
}>, options?: Readonly<AxProgramForwardOptions>): Promise<{
|
|
20
20
|
answer: string;
|
package/cjs/prompts/rag.js
CHANGED
|
@@ -7,26 +7,26 @@ class AxRAG extends cot_js_1.AxChainOfThought {
|
|
|
7
7
|
genQuery;
|
|
8
8
|
queryFn;
|
|
9
9
|
maxHops;
|
|
10
|
-
constructor(
|
|
10
|
+
constructor(queryFn, options) {
|
|
11
11
|
const sig = '"Answer questions with short factoid answers." context:string[] "may contain relevant facts", question -> answer';
|
|
12
|
-
super(
|
|
12
|
+
super(sig, options);
|
|
13
13
|
this.maxHops = options?.maxHops ?? 3;
|
|
14
14
|
const qsig = new index_js_1.AxSignature('"Write a simple search query that will help answer a complex question." context?:string[] "may contain relevant facts", question -> query "question to further our understanding"');
|
|
15
|
-
this.genQuery = new index_js_1.
|
|
15
|
+
this.genQuery = new index_js_1.AxGen(qsig);
|
|
16
16
|
this.queryFn = queryFn;
|
|
17
17
|
this.register(this.genQuery);
|
|
18
18
|
}
|
|
19
|
-
async forward({ question }, options) {
|
|
19
|
+
async forward(ai, { question }, options) {
|
|
20
20
|
let context = [];
|
|
21
21
|
for (let i = 0; i < this.maxHops; i++) {
|
|
22
|
-
const { query } = await this.genQuery.forward({
|
|
22
|
+
const { query } = await this.genQuery.forward(ai, {
|
|
23
23
|
context,
|
|
24
24
|
question
|
|
25
25
|
}, options);
|
|
26
26
|
const val = await this.queryFn(query);
|
|
27
27
|
context = index_js_1.axStringUtil.dedup([...context, val]);
|
|
28
28
|
}
|
|
29
|
-
return super.forward({ context, question }, options);
|
|
29
|
+
return super.forward(ai, { context, question }, options);
|
|
30
30
|
}
|
|
31
31
|
}
|
|
32
32
|
exports.AxRAG = AxRAG;
|
package/cjs/prompts/rag.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"rag.js","sourceRoot":"","sources":["../../../prompts/rag.ts"],"names":[],"mappings":";;;
|
|
1
|
+
{"version":3,"file":"rag.js","sourceRoot":"","sources":["../../../prompts/rag.ts"],"names":[],"mappings":";;;AAAA,8CAKyB;AAIzB,qCAA4C;AAE5C,MAAa,KAAM,SAAQ,yBAG1B;IACS,QAAQ,CAGd;IACM,OAAO,CAAqC;IAC5C,OAAO,CAAS;IAExB,YACE,OAA2C,EAC3C,OAAsD;QAEtD,MAAM,GAAG,GACP,kHAAkH,CAAC;QACrH,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAEpB,IAAI,CAAC,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,CAAC,CAAC;QAErC,MAAM,IAAI,GAAG,IAAI,sBAAW,CAC1B,mLAAmL,CACpL,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,gBAAK,CAGvB,IAAI,CAAC,CAAC;QACR,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAEe,KAAK,CAAC,OAAO,CAC3B,EAAyB,EACzB,EAAE,QAAQ,EAAkC,EAC5C,OAA2C;QAE3C,IAAI,OAAO,GAAa,EAAE,CAAC;QAE3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAC3C,EAAE,EACF;gBACE,OAAO;gBACP,QAAQ;aACT,EACD,OAAO,CACR,CAAC;YACF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtC,OAAO,GAAG,uBAAY,CAAC,KAAK,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QAClD,CAAC;QAED,OAAO,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,OAAO,CAAC,CAAC;IAC3D,CAAC;CACF;AAtDD,sBAsDC"}
|
package/cjs/prompts/react.d.ts
CHANGED
|
@@ -1,8 +1,7 @@
|
|
|
1
|
-
import type {
|
|
2
|
-
import type { AxGenerateOptions } from '../dsp/generate.js';
|
|
1
|
+
import type { AxGenOptions } from '../dsp/generate.js';
|
|
3
2
|
import type { AxGenIn, AxGenOut } from '../dsp/program.js';
|
|
4
3
|
import { AxSignature } from '../dsp/sig.js';
|
|
5
4
|
import { AxChainOfThought } from './cot.js';
|
|
6
5
|
export declare class AxReAct<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends AxChainOfThought<IN, OUT> {
|
|
7
|
-
constructor(
|
|
6
|
+
constructor(signature: Readonly<AxSignature | string>, options: Readonly<AxGenOptions>);
|
|
8
7
|
}
|
package/cjs/prompts/react.js
CHANGED
|
@@ -4,7 +4,7 @@ exports.AxReAct = void 0;
|
|
|
4
4
|
const sig_js_1 = require("../dsp/sig.js");
|
|
5
5
|
const cot_js_1 = require("./cot.js");
|
|
6
6
|
class AxReAct extends cot_js_1.AxChainOfThought {
|
|
7
|
-
constructor(
|
|
7
|
+
constructor(signature, options) {
|
|
8
8
|
if (!options?.functions || options.functions.length === 0) {
|
|
9
9
|
throw new Error('No functions provided');
|
|
10
10
|
}
|
|
@@ -24,7 +24,7 @@ class AxReAct extends cot_js_1.AxChainOfThought {
|
|
|
24
24
|
// type: { name: 'string', isArray: true }
|
|
25
25
|
// });
|
|
26
26
|
sig.setOutputFields(sig.getOutputFields().map((v) => ({ ...v, isOptional: true })));
|
|
27
|
-
super(
|
|
27
|
+
super(sig, options);
|
|
28
28
|
}
|
|
29
29
|
}
|
|
30
30
|
exports.AxReAct = AxReAct;
|
package/cjs/prompts/react.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"react.js","sourceRoot":"","sources":["../../../prompts/react.ts"],"names":[],"mappings":";;;
|
|
1
|
+
{"version":3,"file":"react.js","sourceRoot":"","sources":["../../../prompts/react.ts"],"names":[],"mappings":";;;AAEA,0CAA4C;AAE5C,qCAA4C;AAE5C,MAAa,OAGX,SAAQ,yBAAyB;IACjC,YACE,SAAyC,EACzC,OAA+B;QAE/B,IAAI,CAAC,OAAO,EAAE,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1D,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YAC1C,IAAI,YAAY,IAAI,CAAC,EAAE,CAAC;gBACtB,OAAO,CAAC,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC;YAC7B,CAAC;YACD,OAAO,CAAC,CAAC,IAAI,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEjE,MAAM,GAAG,GAAG,IAAI,oBAAW,CAAC,SAAS,CAAC,CAAC;QACvC,GAAG,CAAC,cAAc,CAChB,8BAA8B,QAAQ,qDAAqD,CAC5F,CAAC;QAEF,sBAAsB;QACtB,yBAAyB;QACzB,wDAAwD;QACxD,sBAAsB;QACtB,4CAA4C;QAC5C,MAAM;QAEN,GAAG,CAAC,eAAe,CACjB,GAAG,CAAC,eAAe,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAC/D,CAAC;QAEF,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IACtB,CAAC;CACF;AAvCD,0BAuCC"}
|
package/cjs/util/apicall.js
CHANGED
|
@@ -6,8 +6,8 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
6
6
|
exports.apiCall = void 0;
|
|
7
7
|
const path_1 = __importDefault(require("path"));
|
|
8
8
|
const web_1 = require("stream/web");
|
|
9
|
+
const sse_js_1 = require("./sse.js");
|
|
9
10
|
const stream_js_1 = require("./stream.js");
|
|
10
|
-
const transform_js_1 = require("./transform.js");
|
|
11
11
|
const textDecoderStream = web_1.TextDecoderStream ?? stream_js_1.TextDecoderStreamPolyfill;
|
|
12
12
|
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
13
13
|
const apiCall = async (api, json) => {
|
|
@@ -42,7 +42,7 @@ const apiCall = async (api, json) => {
|
|
|
42
42
|
}
|
|
43
43
|
const st = res.body
|
|
44
44
|
.pipeThrough(new textDecoderStream())
|
|
45
|
-
.pipeThrough(new
|
|
45
|
+
.pipeThrough(new sse_js_1.SSEParser());
|
|
46
46
|
return st;
|
|
47
47
|
}
|
|
48
48
|
catch (e) {
|
package/cjs/util/apicall.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"apicall.js","sourceRoot":"","sources":["../../../util/apicall.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAwB;AACxB,oCAGoB;AAIpB,
|
|
1
|
+
{"version":3,"file":"apicall.js","sourceRoot":"","sources":["../../../util/apicall.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAwB;AACxB,oCAGoB;AAIpB,qCAAqC;AACrC,2CAAwD;AAaxD,MAAM,iBAAiB,GAAG,uBAAuB,IAAI,qCAAyB,CAAC;AAE/E,gEAAgE;AACzD,MAAM,OAAO,GAAG,KAAK,EAC1B,GAQC,EACD,IAAc,EACkC,EAAE;IAClD,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IACzD,MAAM,OAAO,GAAG,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,IAAI,GAAG,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IAC7E,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAEzC,IAAI,GAAG,CAAC,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;QAC5B,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC;YACrB,qBAAqB,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM;YAC/C,UAAU,EAAE,MAAM,CAAC,IAAI;SACxB,CAAC,CAAC;IACL,CAAC;IAED,IAAI,GAAqD,CAAC;IAE1D,IAAI,CAAC;QACH,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,KAAK,CAAC,CAAC,MAAM,EAAE;YACvC,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM;YAChC,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,GAAG,CAAC,OAAO;aACf;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;YACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAC3D,MAAM,IAAI,KAAK,CACb,GAAG,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,UAAU,qBAAqB,QAAQ,EAAE,CAChE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAc,CAAC;QACzC,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,EAAE,GAAG,GAAG,CAAC,IAAI;aAChB,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC;aACpC,WAAW,CAAC,IAAI,kBAAS,EAAa,CAAC,CAAC;QAE3C,OAAO,EAAE,CAAC;IACZ,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,GAAG,CAAC,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;YAC5B,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAU,CAAC,CAAC;QAC7C,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC9C,MAAM,IAAI,KAAK,CACb,cAAc,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,OAAO,EAAE,CAC5D,CAAC;IACJ,CAAC;AACH,CAAC,CAAC;AAhEW,QAAA,OAAO,WAgElB"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import { TransformStream } from 'stream/web';
|
|
2
|
+
export declare class SSEParser<T = unknown> extends TransformStream<string, T> {
|
|
3
|
+
private readonly dataParser;
|
|
4
|
+
private buffer;
|
|
5
|
+
private currentEvent;
|
|
6
|
+
constructor(dataParser?: (data: string) => T);
|
|
7
|
+
private handleChunk;
|
|
8
|
+
private handleFlush;
|
|
9
|
+
private processBuffer;
|
|
10
|
+
private parseLine;
|
|
11
|
+
private emitEvent;
|
|
12
|
+
}
|
package/cjs/util/sse.js
ADDED
|
@@ -0,0 +1,97 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.SSEParser = void 0;
|
|
4
|
+
const web_1 = require("stream/web");
|
|
5
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
6
|
+
class SSEParser extends web_1.TransformStream {
|
|
7
|
+
dataParser;
|
|
8
|
+
buffer = '';
|
|
9
|
+
currentEvent = { rawData: '' };
|
|
10
|
+
constructor(dataParser = JSON.parse) {
|
|
11
|
+
super({
|
|
12
|
+
transform: (chunk, controller) => this.handleChunk(chunk, controller),
|
|
13
|
+
flush: (controller) => this.handleFlush(controller)
|
|
14
|
+
});
|
|
15
|
+
this.dataParser = dataParser;
|
|
16
|
+
}
|
|
17
|
+
handleChunk(chunk, controller) {
|
|
18
|
+
this.buffer += chunk;
|
|
19
|
+
this.processBuffer(controller);
|
|
20
|
+
}
|
|
21
|
+
handleFlush(controller) {
|
|
22
|
+
this.processBuffer(controller);
|
|
23
|
+
if (this.currentEvent.rawData) {
|
|
24
|
+
this.emitEvent(controller);
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
processBuffer(controller) {
|
|
28
|
+
const lines = this.buffer.split(/\r\n|\r|\n/);
|
|
29
|
+
this.buffer = lines.pop() || '';
|
|
30
|
+
for (const line of lines) {
|
|
31
|
+
if (line.trim() === '') {
|
|
32
|
+
this.emitEvent(controller);
|
|
33
|
+
}
|
|
34
|
+
else {
|
|
35
|
+
this.parseLine(line);
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
parseLine(line) {
|
|
40
|
+
const colonIndex = line.indexOf(':');
|
|
41
|
+
if (colonIndex === -1) {
|
|
42
|
+
// If there's no colon, treat the whole line as data
|
|
43
|
+
this.currentEvent.rawData += this.currentEvent.rawData
|
|
44
|
+
? '\n' + line.trim()
|
|
45
|
+
: line.trim();
|
|
46
|
+
return;
|
|
47
|
+
}
|
|
48
|
+
const field = line.slice(0, colonIndex).trim();
|
|
49
|
+
const value = line.slice(colonIndex + 1).trim();
|
|
50
|
+
switch (field) {
|
|
51
|
+
case 'event':
|
|
52
|
+
this.currentEvent.event = value;
|
|
53
|
+
break;
|
|
54
|
+
case 'data':
|
|
55
|
+
this.currentEvent.rawData += this.currentEvent.rawData
|
|
56
|
+
? '\n' + value
|
|
57
|
+
: value;
|
|
58
|
+
break;
|
|
59
|
+
case 'id':
|
|
60
|
+
this.currentEvent.id = value;
|
|
61
|
+
break;
|
|
62
|
+
case 'retry': {
|
|
63
|
+
const retryValue = parseInt(value, 10);
|
|
64
|
+
if (!isNaN(retryValue)) {
|
|
65
|
+
this.currentEvent.retry = retryValue;
|
|
66
|
+
}
|
|
67
|
+
break;
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
emitEvent(controller) {
|
|
72
|
+
if (this.currentEvent.rawData) {
|
|
73
|
+
// Check for special "[DONE]" message or other non-JSON data
|
|
74
|
+
if (this.currentEvent.rawData.trim() === '[DONE]' ||
|
|
75
|
+
this.currentEvent.rawData.trim().startsWith('[')) {
|
|
76
|
+
return;
|
|
77
|
+
}
|
|
78
|
+
else {
|
|
79
|
+
try {
|
|
80
|
+
// Attempt to parse the data using the provided dataParser
|
|
81
|
+
const parsedData = this.dataParser(this.currentEvent.rawData);
|
|
82
|
+
// Emit only if we successfully parsed the data
|
|
83
|
+
controller.enqueue(parsedData);
|
|
84
|
+
}
|
|
85
|
+
catch (e) {
|
|
86
|
+
// If parsing fails, log the error without emitting
|
|
87
|
+
console.warn('Failed to parse event data:', e);
|
|
88
|
+
console.log('Raw data that failed to parse:', this.currentEvent.rawData);
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
// Reset the current event
|
|
92
|
+
this.currentEvent = { rawData: '' };
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
exports.SSEParser = SSEParser;
|
|
97
|
+
//# sourceMappingURL=sse.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sse.js","sourceRoot":"","sources":["../../../util/sse.ts"],"names":[],"mappings":";;;AAAA,oCAA+E;AAS/E,gEAAgE;AAChE,MAAa,SAAuB,SAAQ,qBAA0B;IAIvC;IAHrB,MAAM,GAAW,EAAE,CAAC;IACpB,YAAY,GAAsB,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;IAE1D,YAA6B,aAAkC,IAAI,CAAC,KAAK;QACvE,KAAK,CAAC;YACJ,SAAS,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,UAAU,CAAC;YACrE,KAAK,EAAE,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC;SACpD,CAAC,CAAC;QAJwB,eAAU,GAAV,UAAU,CAAkC;IAKzE,CAAC;IAEO,WAAW,CACjB,KAAa,EACb,UAA+C;QAE/C,IAAI,CAAC,MAAM,IAAI,KAAK,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IACjC,CAAC;IAEO,WAAW,CAAC,UAA+C;QACjE,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAC/B,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;YAC9B,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,UAA+C;QACnE,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;QAEhC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;gBACvB,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YAC7B,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,IAAY;QAC5B,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,UAAU,KAAK,CAAC,CAAC,EAAE,CAAC;YACtB,oDAAoD;YACpD,IAAI,CAAC,YAAY,CAAC,OAAO,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO;gBACpD,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE;gBACpB,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAEhD,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,OAAO;gBACV,IAAI,CAAC,YAAY,CAAC,KAAK,GAAG,KAAK,CAAC;gBAChC,MAAM;YACR,KAAK,MAAM;gBACT,IAAI,CAAC,YAAY,CAAC,OAAO,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO;oBACpD,CAAC,CAAC,IAAI,GAAG,KAAK;oBACd,CAAC,CAAC,KAAK,CAAC;gBACV,MAAM;YACR,KAAK,IAAI;gBACP,IAAI,CAAC,YAAY,CAAC,EAAE,GAAG,KAAK,CAAC;gBAC7B,MAAM;YACR,KAAK,OAAO,CAAC,CAAC,CAAC;gBACb,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBACvC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;oBACvB,IAAI,CAAC,YAAY,CAAC,KAAK,GAAG,UAAU,CAAC;gBACvC,CAAC;gBACD,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,UAA+C;QAC/D,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;YAC9B,4DAA4D;YAC5D,IACE,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,QAAQ;gBAC7C,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAChD,CAAC;gBACD,OAAO;YACT,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC;oBACH,0DAA0D;oBAC1D,MAAM,UAAU,GAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;oBACjE,+CAA+C;oBAC/C,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBACjC,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,mDAAmD;oBACnD,OAAO,CAAC,IAAI,CAAC,6BAA6B,EAAE,CAAC,CAAC,CAAC;oBAC/C,OAAO,CAAC,GAAG,CACT,gCAAgC,EAChC,IAAI,CAAC,YAAY,CAAC,OAAO,CAC1B,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,0BAA0B;YAC1B,IAAI,CAAC,YAAY,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACtC,CAAC;IACH,CAAC;CACF;AAtGD,8BAsGC"}
|
package/cjs/util/transform.d.ts
CHANGED
|
@@ -1,7 +1,4 @@
|
|
|
1
1
|
import { TransformStream } from 'stream/web';
|
|
2
|
-
export declare class JSONStringifyStream<O> extends TransformStream<string, O> {
|
|
3
|
-
constructor();
|
|
4
|
-
}
|
|
5
2
|
export declare class RespTransformStream<I, O> extends TransformStream<I, O> {
|
|
6
3
|
constructor(transformFn: (arg0: I) => O, doneCallback?: (args0: readonly O[]) => Promise<void>);
|
|
7
4
|
}
|
package/cjs/util/transform.js
CHANGED
|
@@ -1,30 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.RespTransformStream =
|
|
3
|
+
exports.RespTransformStream = void 0;
|
|
4
4
|
const web_1 = require("stream/web");
|
|
5
|
-
class JSONTransformer {
|
|
6
|
-
async transform(obj, controller) {
|
|
7
|
-
(obj.split('\n') ?? [])
|
|
8
|
-
.map(extractJson)
|
|
9
|
-
.map((v) => {
|
|
10
|
-
try {
|
|
11
|
-
return v && v.length > 0 ? JSON.parse(v) : null;
|
|
12
|
-
}
|
|
13
|
-
catch (e) {
|
|
14
|
-
return null;
|
|
15
|
-
}
|
|
16
|
-
})
|
|
17
|
-
.filter((v) => v)
|
|
18
|
-
.forEach((v) => v && controller.enqueue(v));
|
|
19
|
-
}
|
|
20
|
-
}
|
|
21
|
-
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
22
|
-
class JSONStringifyStream extends web_1.TransformStream {
|
|
23
|
-
constructor() {
|
|
24
|
-
super(new JSONTransformer());
|
|
25
|
-
}
|
|
26
|
-
}
|
|
27
|
-
exports.JSONStringifyStream = JSONStringifyStream;
|
|
28
5
|
class TypeTransformer {
|
|
29
6
|
buffer;
|
|
30
7
|
doneCallback;
|
|
@@ -53,12 +30,4 @@ class RespTransformStream extends web_1.TransformStream {
|
|
|
53
30
|
}
|
|
54
31
|
}
|
|
55
32
|
exports.RespTransformStream = RespTransformStream;
|
|
56
|
-
const extractJson = (str) => {
|
|
57
|
-
const startIndex = str.indexOf('{');
|
|
58
|
-
const endIndex = str.lastIndexOf('}');
|
|
59
|
-
if (startIndex !== -1 && endIndex !== -1) {
|
|
60
|
-
return str.substring(startIndex, endIndex + 1);
|
|
61
|
-
}
|
|
62
|
-
return null;
|
|
63
|
-
};
|
|
64
33
|
//# sourceMappingURL=transform.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"transform.js","sourceRoot":"","sources":["../../../util/transform.ts"],"names":[],"mappings":";;;AAAA,oCAIoB;AAEpB,MAAM,eAAe;
|
|
1
|
+
{"version":3,"file":"transform.js","sourceRoot":"","sources":["../../../util/transform.ts"],"names":[],"mappings":";;;AAAA,oCAIoB;AAEpB,MAAM,eAAe;IACX,MAAM,CAAO;IACb,YAAY,CAA0C;IACtD,WAAW,CAAiB;IAEpC,YACE,WAA2B,EAC3B,YAAqD;QAErD,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,GAAM,EAAE,UAA+C;QACrE,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,GAAG,EAAE,CAAC;YACR,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACxB,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,UAA+C;QACzD,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAC7C,UAAU,CAAC,SAAS,EAAE,CAAC;IACzB,CAAC;CACF;AAED,gEAAgE;AAChE,MAAa,mBAA0B,SAAQ,qBAAqB;IAClE,YACE,WAA2B,EAC3B,YAAqD;QAErD,KAAK,CAAC,IAAI,eAAe,CAAO,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;IAC9D,CAAC;CACF;AAPD,kDAOC"}
|
package/docs/manager.js
CHANGED
|
@@ -58,7 +58,7 @@ export class AxDBManager {
|
|
|
58
58
|
const texts = Array.isArray(query) ? query : [query];
|
|
59
59
|
if (typeof texts[0] === 'string' && this.rewriter) {
|
|
60
60
|
for (const [i, text] of texts.entries()) {
|
|
61
|
-
const { rewrittenQuery } = await this.rewriter.forward({
|
|
61
|
+
const { rewrittenQuery } = await this.rewriter.forward(this.ai, {
|
|
62
62
|
query: text
|
|
63
63
|
});
|
|
64
64
|
texts[i] = rewrittenQuery;
|
|
@@ -84,7 +84,7 @@ export class AxDBManager {
|
|
|
84
84
|
const tp = topPercent && topPercent > 1 ? topPercent / 100 : topPercent;
|
|
85
85
|
const resultItems = tp ? getTopInPercent(m, tp) : m;
|
|
86
86
|
if (this.reranker) {
|
|
87
|
-
const { rankedItems } = await this.reranker.forward({
|
|
87
|
+
const { rankedItems } = await this.reranker.forward(this.ai, {
|
|
88
88
|
query: texts[0],
|
|
89
89
|
items: resultItems.map((item) => item.text)
|
|
90
90
|
});
|
package/docs/manager.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../docs/manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAKN,MAAM,aAAa,CAAC;AAyBrB,MAAM,KAAK,GAAG,WAAW,CAAC;AAE1B,MAAM,OAAO,WAAW;IACd,EAAE,CAAc;IAChB,EAAE,CAAc;IAChB,OAAO,CAA6B;IACpC,QAAQ,CAAwC;IAChD,QAAQ,CAA0C;IAE1D,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,EAA6B;QACvD,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,OAAO,GAAG,MAAM,EAAE,OAAO,IAAI,IAAI,CAAC,cAAc,CAAC;QACtD,IAAI,CAAC,QAAQ,GAAG,MAAM,EAAE,QAAQ,CAAC;QACjC,IAAI,CAAC,QAAQ,GAAG,MAAM,EAAE,QAAQ,CAAC;IACnC,CAAC;IAEO,cAAc,GAAG,CAAC,IAAY,EAAY,EAAE;QAClD,iCAAiC;QACjC,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC7B,CAAC,CAAC;IAEF,MAAM,GAAG,KAAK,EACZ,IAAiC,EACjC,OAIE,EACa,EAAE;QACjB,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;gBACtC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;gBACnB,CAAC,CAAE,IAAe,CAAC;YAErB,kEAAkE;YAClE,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,MAAM,CACrD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAC5B,CAAC;YAEF,MAAM,gBAAgB,GAAG,OAAO,EAAE,gBAAgB,CAAC;YACnD,MAAM,gBAAgB,GAAG,OAAO,EAAE,gBAAgB,CAAC;YAEnD,MAAM,MAAM,GAAG,aAAa,CAAC;gBAC3B,aAAa;gBACb,gBAAgB;gBAChB,gBAAgB;aACjB,CAAC,CAAC;YAEH,MAAM,EAAE,GAAG,OAAO,EAAE,SAAS,IAAI,EAAE,CAAC;YAEpC,kCAAkC;YAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;gBAEtC,qEAAqE;gBACrE,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;gBAElD,gCAAgC;gBAChC,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU;qBAC9B,GAAG,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;oBAC1B,EAAE,EAAE,SAAS,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,EAAE,EAAE,8CAA8C;oBACjF,KAAK;oBACL,MAAM,EAAE,SAAS;oBACjB,QAAQ,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE;iBACvC,CAAC,CAAC;qBACF,MAAM,CACL,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,CAC/D,CAAC;gBAEJ,0BAA0B;gBAC1B,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,EAAE,CAAC,CAAC;QACrD,CAAC;IACH,CAAC,CAAC;IAEF,KAAK,GAAG,KAAK,EACX,KAAsD,EACtD,EAAE,UAAU,KAAoD,EAAE,EAC1C,EAAE;QAC1B,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAErD,IAAI,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClD,KAAK,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;gBACxC,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;
|
|
1
|
+
{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../docs/manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAKN,MAAM,aAAa,CAAC;AAyBrB,MAAM,KAAK,GAAG,WAAW,CAAC;AAE1B,MAAM,OAAO,WAAW;IACd,EAAE,CAAc;IAChB,EAAE,CAAc;IAChB,OAAO,CAA6B;IACpC,QAAQ,CAAwC;IAChD,QAAQ,CAA0C;IAE1D,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,EAA6B;QACvD,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,OAAO,GAAG,MAAM,EAAE,OAAO,IAAI,IAAI,CAAC,cAAc,CAAC;QACtD,IAAI,CAAC,QAAQ,GAAG,MAAM,EAAE,QAAQ,CAAC;QACjC,IAAI,CAAC,QAAQ,GAAG,MAAM,EAAE,QAAQ,CAAC;IACnC,CAAC;IAEO,cAAc,GAAG,CAAC,IAAY,EAAY,EAAE;QAClD,iCAAiC;QACjC,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC7B,CAAC,CAAC;IAEF,MAAM,GAAG,KAAK,EACZ,IAAiC,EACjC,OAIE,EACa,EAAE;QACjB,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;gBACtC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;gBACnB,CAAC,CAAE,IAAe,CAAC;YAErB,kEAAkE;YAClE,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,MAAM,CACrD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAC5B,CAAC;YAEF,MAAM,gBAAgB,GAAG,OAAO,EAAE,gBAAgB,CAAC;YACnD,MAAM,gBAAgB,GAAG,OAAO,EAAE,gBAAgB,CAAC;YAEnD,MAAM,MAAM,GAAG,aAAa,CAAC;gBAC3B,aAAa;gBACb,gBAAgB;gBAChB,gBAAgB;aACjB,CAAC,CAAC;YAEH,MAAM,EAAE,GAAG,OAAO,EAAE,SAAS,IAAI,EAAE,CAAC;YAEpC,kCAAkC;YAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;gBAEtC,qEAAqE;gBACrE,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;gBAElD,gCAAgC;gBAChC,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU;qBAC9B,GAAG,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;oBAC1B,EAAE,EAAE,SAAS,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,EAAE,EAAE,8CAA8C;oBACjF,KAAK;oBACL,MAAM,EAAE,SAAS;oBACjB,QAAQ,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE;iBACvC,CAAC,CAAC;qBACF,MAAM,CACL,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,CAC/D,CAAC;gBAEJ,0BAA0B;gBAC1B,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,EAAE,CAAC,CAAC;QACrD,CAAC;IACH,CAAC,CAAC;IAEF,KAAK,GAAG,KAAK,EACX,KAAsD,EACtD,EAAE,UAAU,KAAoD,EAAE,EAC1C,EAAE;QAC1B,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAErD,IAAI,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClD,KAAK,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;gBACxC,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE;oBAC9D,KAAK,EAAE,IAAI;iBACZ,CAAC,CAAC;gBACH,KAAK,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC;YAC5B,CAAC;QACH,CAAC;QAED,IAAI,OAAqC,CAAC;QAE1C,IAAI,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;YACjC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;YACpD,OAAO,GAAG,YAAY,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAC/C,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CACjC,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,OAAO,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAChD,MAAM,GAAG,GAAkB,EAAE,CAAC;QAE9B,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,YAAY,EAAE,CAAC;YACvC,MAAM,CAAC,GAAG,OAAO;iBACd,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;iBACtE,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC7B,KAAK;gBACL,IAAI,EAAE,QAAQ,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE;aAC/B,CAAC,CAAC,CAAC;YAEN,MAAM,EAAE,GAAG,UAAU,IAAI,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,GAAG,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC;YACxE,MAAM,WAAW,GAAG,EAAE,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAEpD,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE;oBAC3D,KAAK,EAAE,KAAK,CAAC,CAAC,CAAW;oBACzB,KAAK,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;iBAC5C,CAAC,CAAC;gBAEH,MAAM,KAAK,GAAG,WAAW;qBACtB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;qBACvD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAgB,CAAC;gBAEjD,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAClB,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QAED,OAAO,GAAG,CAAC;IACb,CAAC,CAAC;CACH;AAED,MAAM,aAAa,GAAG,CAAC,EACrB,aAAa,EACb,gBAAgB,GAAG,GAAG,EACtB,gBAAgB,GAAG,GAAG,EAKtB,EAAY,EAAE;IACd,MAAM,MAAM,GAAa,EAAE,CAAC;IAE5B,IAAI,YAAY,GAAG,EAAE,CAAC;IACtB,IAAI,gBAAgB,GAAG,CAAC,CAAC;IAEzB,aAAa,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;QAC9B,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,6BAA6B;QAC/D,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,mCAAmC;QAEnE,IAAI,gBAAgB,GAAG,SAAS,IAAI,gBAAgB,EAAE,CAAC;YACrD,wDAAwD;YACxD,YAAY,IAAI,KAAK,GAAG,MAAM,CAAC;YAC/B,gBAAgB,IAAI,SAAS,CAAC;QAChC,CAAC;aAAM,IACL,gBAAgB,GAAG,CAAC;YACpB,gBAAgB,GAAG,SAAS,IAAI,gBAAgB,GAAG,GAAG,EACtD,CAAC;YACD,0FAA0F;YAC1F,YAAY,IAAI,KAAK,GAAG,MAAM,CAAC;YAC/B,gBAAgB,IAAI,SAAS,CAAC;QAChC,CAAC;aAAM,CAAC;YACN,wFAAwF;YACxF,IAAI,gBAAgB,GAAG,gBAAgB,EAAE,CAAC;gBACxC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;gBACjC,YAAY,GAAG,EAAE,CAAC;gBAClB,gBAAgB,GAAG,CAAC,CAAC;YACvB,CAAC;YACD,kEAAkE;YAClE,IAAI,SAAS,GAAG,gBAAgB,EAAE,CAAC;gBACjC,MAAM,cAAc,GAAG,KAAK,CAAC;gBAC7B,OAAO,cAAc,CAAC,MAAM,GAAG,gBAAgB,GAAG,GAAG,EAAE,CAAC;oBACtD,MAAM,KAAK,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;oBACzD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;gBAC/B,CAAC;gBACD,yFAAyF;gBACzF,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC9B,YAAY,IAAI,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;oBAClD,gBAAgB,IAAI,cAAc,CAAC,MAAM,CAAC;gBAC5C,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,+DAA+D;gBAC/D,YAAY,GAAG,KAAK,GAAG,MAAM,CAAC;gBAC9B,gBAAgB,GAAG,SAAS,CAAC;YAC/B,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,yEAAyE;IACzE,IAAI,gBAAgB,GAAG,gBAAgB,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/D,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;IACnC,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC;AAEF,MAAM,eAAe,GAAG,CACtB,OAA6B,EAC7B,UAAkB,GAAG,EACR,EAAE;IACf,2CAA2C;IAC3C,MAAM,aAAa,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IAErE,oDAAoD;IACpD,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,OAAO,CAAC,CAAC;IAErE,gCAAgC;IAChC,OAAO,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,kBAAkB,CAAC,CAAC;AACpD,CAAC,CAAC"}
|
package/docs/reranker.d.ts
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import type { AxAIService } from '../ai/types.js';
|
|
2
|
-
import {
|
|
2
|
+
import { AxGen, type AxGenOptions, type AxProgramForwardOptions } from '../dsp/index.js';
|
|
3
3
|
import type { AxRerankerIn, AxRerankerOut } from './manager.js';
|
|
4
|
-
export declare class AxDefaultResultReranker extends
|
|
5
|
-
constructor(
|
|
6
|
-
forward: (input: Readonly<AxRerankerIn>, options?: Readonly<AxProgramForwardOptions>) => Promise<AxRerankerOut>;
|
|
4
|
+
export declare class AxDefaultResultReranker extends AxGen<AxRerankerIn, AxRerankerOut> {
|
|
5
|
+
constructor(options?: Readonly<AxGenOptions>);
|
|
6
|
+
forward: (ai: Readonly<AxAIService>, input: Readonly<AxRerankerIn>, options?: Readonly<AxProgramForwardOptions>) => Promise<AxRerankerOut>;
|
|
7
7
|
}
|