@langchain/core 1.0.0-alpha.4 → 1.0.0-alpha.5
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/agents.d.ts.map +1 -1
- package/dist/caches/base.d.ts.map +1 -1
- package/dist/callbacks/base.d.ts.map +1 -1
- package/dist/language_models/chat_models.d.ts.map +1 -1
- package/dist/load/import_map.cjs +1 -1
- package/dist/load/import_map.js +1 -1
- package/dist/messages/ai.cjs +15 -10
- package/dist/messages/ai.cjs.map +1 -1
- package/dist/messages/ai.js +15 -10
- package/dist/messages/ai.js.map +1 -1
- package/dist/messages/base.cjs +8 -1
- package/dist/messages/base.cjs.map +1 -1
- package/dist/messages/base.js +8 -1
- package/dist/messages/base.js.map +1 -1
- package/dist/utils/testing/chat_models.cjs +245 -0
- package/dist/utils/testing/chat_models.cjs.map +1 -0
- package/dist/utils/testing/chat_models.d.cts +134 -0
- package/dist/utils/testing/chat_models.d.cts.map +1 -0
- package/dist/utils/testing/chat_models.d.ts +134 -0
- package/dist/utils/testing/chat_models.d.ts.map +1 -0
- package/dist/utils/testing/chat_models.js +243 -0
- package/dist/utils/testing/chat_models.js.map +1 -0
- package/dist/utils/testing/embeddings.cjs +90 -0
- package/dist/utils/testing/embeddings.cjs.map +1 -0
- package/dist/utils/testing/embeddings.d.cts +58 -0
- package/dist/utils/testing/embeddings.d.cts.map +1 -0
- package/dist/utils/testing/embeddings.d.ts +58 -0
- package/dist/utils/testing/embeddings.d.ts.map +1 -0
- package/dist/utils/testing/embeddings.js +89 -0
- package/dist/utils/testing/embeddings.js.map +1 -0
- package/dist/utils/testing/index.cjs +42 -655
- package/dist/utils/testing/index.cjs.map +1 -1
- package/dist/utils/testing/index.d.cts +11 -379
- package/dist/utils/testing/index.d.ts +11 -379
- package/dist/utils/testing/index.js +10 -623
- package/dist/utils/testing/index.js.map +1 -1
- package/dist/utils/testing/llms.cjs +59 -0
- package/dist/utils/testing/llms.cjs.map +1 -0
- package/dist/utils/testing/llms.d.cts +31 -0
- package/dist/utils/testing/llms.d.cts.map +1 -0
- package/dist/utils/testing/llms.d.ts +31 -0
- package/dist/utils/testing/llms.d.ts.map +1 -0
- package/dist/utils/testing/llms.js +58 -0
- package/dist/utils/testing/llms.js.map +1 -0
- package/dist/utils/testing/message_history.cjs +67 -0
- package/dist/utils/testing/message_history.cjs.map +1 -0
- package/dist/utils/testing/message_history.d.cts +31 -0
- package/dist/utils/testing/message_history.d.cts.map +1 -0
- package/dist/utils/testing/message_history.d.ts +31 -0
- package/dist/utils/testing/message_history.d.ts.map +1 -0
- package/dist/utils/testing/message_history.js +65 -0
- package/dist/utils/testing/message_history.js.map +1 -0
- package/dist/utils/testing/output_parsers.cjs +20 -0
- package/dist/utils/testing/output_parsers.cjs.map +1 -0
- package/dist/utils/testing/output_parsers.d.cts +16 -0
- package/dist/utils/testing/output_parsers.d.cts.map +1 -0
- package/dist/utils/testing/output_parsers.d.ts +16 -0
- package/dist/utils/testing/output_parsers.d.ts.map +1 -0
- package/dist/utils/testing/output_parsers.js +20 -0
- package/dist/utils/testing/output_parsers.js.map +1 -0
- package/dist/utils/testing/retrievers.cjs +19 -0
- package/dist/utils/testing/retrievers.cjs.map +1 -0
- package/dist/utils/testing/retrievers.d.cts +17 -0
- package/dist/utils/testing/retrievers.d.cts.map +1 -0
- package/dist/utils/testing/retrievers.d.ts +17 -0
- package/dist/utils/testing/retrievers.d.ts.map +1 -0
- package/dist/utils/testing/retrievers.js +19 -0
- package/dist/utils/testing/retrievers.js.map +1 -0
- package/dist/utils/testing/runnables.cjs +19 -0
- package/dist/utils/testing/runnables.cjs.map +1 -0
- package/dist/utils/testing/runnables.d.cts +15 -0
- package/dist/utils/testing/runnables.d.cts.map +1 -0
- package/dist/utils/testing/runnables.d.ts +15 -0
- package/dist/utils/testing/runnables.d.ts.map +1 -0
- package/dist/utils/testing/runnables.js +19 -0
- package/dist/utils/testing/runnables.js.map +1 -0
- package/dist/utils/testing/tools.cjs +21 -0
- package/dist/utils/testing/tools.cjs.map +1 -0
- package/dist/utils/testing/tools.d.cts +21 -0
- package/dist/utils/testing/tools.d.cts.map +1 -0
- package/dist/utils/testing/tools.d.ts +21 -0
- package/dist/utils/testing/tools.d.ts.map +1 -0
- package/dist/utils/testing/tools.js +21 -0
- package/dist/utils/testing/tools.js.map +1 -0
- package/dist/utils/testing/tracers.cjs +25 -0
- package/dist/utils/testing/tracers.cjs.map +1 -0
- package/dist/utils/testing/tracers.d.cts +15 -0
- package/dist/utils/testing/tracers.d.cts.map +1 -0
- package/dist/utils/testing/tracers.d.ts +15 -0
- package/dist/utils/testing/tracers.d.ts.map +1 -0
- package/dist/utils/testing/tracers.js +25 -0
- package/dist/utils/testing/tracers.js.map +1 -0
- package/dist/utils/testing/vectorstores.cjs +129 -0
- package/dist/utils/testing/vectorstores.cjs.map +1 -0
- package/dist/utils/testing/vectorstores.d.cts +101 -0
- package/dist/utils/testing/vectorstores.d.cts.map +1 -0
- package/dist/utils/testing/vectorstores.d.ts +101 -0
- package/dist/utils/testing/vectorstores.d.ts.map +1 -0
- package/dist/utils/testing/vectorstores.js +129 -0
- package/dist/utils/testing/vectorstores.js.map +1 -0
- package/package.json +1 -1
- package/dist/utils/testing/index.d.cts.map +0 -1
- package/dist/utils/testing/index.d.ts.map +0 -1
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { BaseOutputParser } from "../../output_parsers/base.js";
|
|
2
|
+
|
|
3
|
+
//#region src/utils/testing/output_parsers.ts
|
|
4
|
+
/**
|
|
5
|
+
* Parser for comma-separated values. It splits the input text by commas
|
|
6
|
+
* and trims the resulting values.
|
|
7
|
+
*/
|
|
8
|
+
var FakeSplitIntoListParser = class extends BaseOutputParser {
|
|
9
|
+
lc_namespace = ["tests", "fake"];
|
|
10
|
+
getFormatInstructions() {
|
|
11
|
+
return "";
|
|
12
|
+
}
|
|
13
|
+
async parse(text) {
|
|
14
|
+
return text.split(",").map((value) => value.trim());
|
|
15
|
+
}
|
|
16
|
+
};
|
|
17
|
+
|
|
18
|
+
//#endregion
|
|
19
|
+
export { FakeSplitIntoListParser };
|
|
20
|
+
//# sourceMappingURL=output_parsers.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"output_parsers.js","names":["text: string"],"sources":["../../../src/utils/testing/output_parsers.ts"],"sourcesContent":["import { BaseOutputParser } from \"../../output_parsers/base.js\";\n\n/**\n * Parser for comma-separated values. It splits the input text by commas\n * and trims the resulting values.\n */\nexport class FakeSplitIntoListParser extends BaseOutputParser<string[]> {\n lc_namespace = [\"tests\", \"fake\"];\n\n getFormatInstructions() {\n return \"\";\n }\n\n async parse(text: string): Promise<string[]> {\n return text.split(\",\").map((value) => value.trim());\n }\n}\n"],"mappings":";;;;;;;AAMA,IAAa,0BAAb,cAA6C,iBAA2B;CACtE,eAAe,CAAC,SAAS,MAAO;CAEhC,wBAAwB;AACtB,SAAO;CACR;CAED,MAAM,MAAMA,MAAiC;AAC3C,SAAO,KAAK,MAAM,IAAI,CAAC,IAAI,CAAC,UAAU,MAAM,MAAM,CAAC;CACpD;AACF"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
const require_retrievers_index = require('../../retrievers/index.cjs');
|
|
2
|
+
const require_document = require('../../documents/document.cjs');
|
|
3
|
+
|
|
4
|
+
//#region src/utils/testing/retrievers.ts
|
|
5
|
+
var FakeRetriever = class extends require_retrievers_index.BaseRetriever {
|
|
6
|
+
lc_namespace = ["test", "fake"];
|
|
7
|
+
output = [new require_document.Document({ pageContent: "foo" }), new require_document.Document({ pageContent: "bar" })];
|
|
8
|
+
constructor(fields) {
|
|
9
|
+
super();
|
|
10
|
+
this.output = fields?.output ?? this.output;
|
|
11
|
+
}
|
|
12
|
+
async _getRelevantDocuments(_query) {
|
|
13
|
+
return this.output;
|
|
14
|
+
}
|
|
15
|
+
};
|
|
16
|
+
|
|
17
|
+
//#endregion
|
|
18
|
+
exports.FakeRetriever = FakeRetriever;
|
|
19
|
+
//# sourceMappingURL=retrievers.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"retrievers.cjs","names":["BaseRetriever","Document","fields?: { output: Document[] }","_query: string"],"sources":["../../../src/utils/testing/retrievers.ts"],"sourcesContent":["import { Document } from \"../../documents/document.js\";\nimport { BaseRetriever } from \"../../retrievers/index.js\";\n\nexport class FakeRetriever extends BaseRetriever {\n lc_namespace = [\"test\", \"fake\"];\n\n output = [\n new Document({ pageContent: \"foo\" }),\n new Document({ pageContent: \"bar\" }),\n ];\n\n constructor(fields?: { output: Document[] }) {\n super();\n this.output = fields?.output ?? this.output;\n }\n\n async _getRelevantDocuments(\n _query: string\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n ): Promise<Document<Record<string, any>>[]> {\n return this.output;\n }\n}\n"],"mappings":";;;;AAGA,IAAa,gBAAb,cAAmCA,uCAAc;CAC/C,eAAe,CAAC,QAAQ,MAAO;CAE/B,SAAS,CACP,IAAIC,0BAAS,EAAE,aAAa,MAAO,IACnC,IAAIA,0BAAS,EAAE,aAAa,MAAO,EACpC;CAED,YAAYC,QAAiC;EAC3C,OAAO;EACP,KAAK,SAAS,QAAQ,UAAU,KAAK;CACtC;CAED,MAAM,sBACJC,QAE0C;AAC1C,SAAO,KAAK;CACb;AACF"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import { Document } from "../../documents/document.cjs";
|
|
2
|
+
import { BaseRetriever } from "../../retrievers/index.cjs";
|
|
3
|
+
|
|
4
|
+
//#region src/utils/testing/retrievers.d.ts
|
|
5
|
+
declare class FakeRetriever extends BaseRetriever {
|
|
6
|
+
lc_namespace: string[];
|
|
7
|
+
output: Document<Record<string, any>>[];
|
|
8
|
+
constructor(fields?: {
|
|
9
|
+
output: Document[];
|
|
10
|
+
});
|
|
11
|
+
_getRelevantDocuments(_query: string
|
|
12
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
13
|
+
): Promise<Document<Record<string, any>>[]>;
|
|
14
|
+
}
|
|
15
|
+
//#endregion
|
|
16
|
+
export { FakeRetriever };
|
|
17
|
+
//# sourceMappingURL=retrievers.d.cts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"retrievers.d.cts","names":["Document","BaseRetriever","FakeRetriever","Record","Promise"],"sources":["../../../src/utils/testing/retrievers.d.ts"],"sourcesContent":["import { Document } from \"../../documents/document.js\";\nimport { BaseRetriever } from \"../../retrievers/index.js\";\nexport declare class FakeRetriever extends BaseRetriever {\n lc_namespace: string[];\n output: Document<Record<string, any>>[];\n constructor(fields?: {\n output: Document[];\n });\n _getRelevantDocuments(_query: string\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n ): Promise<Document<Record<string, any>>[]>;\n}\n"],"mappings":";;;;cAEqBE,aAAAA,SAAsBD,aAAAA;;EAAtBC,MAAAA,EAETF,QAFsB,CAEbG,MAFa,CAAA,MAAA,EAAA,GAAA,CAAA,CAAA,EAAA;EAAA,WAAA,CAAA,MAEP,CAFO,EAAA;IAEbA,MAAAA,EAELH,QAFKG,EAAAA;EAAM,CAAA;EAAP,qBAEJH,CAAAA,MAAAA,EAAAA;EAAQ;EAAA,CAIM,EAAvBI,OAAQJ,CAAAA,QAAAA,CAASG,MAATH,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,CAAAA,EAAAA,CAAAA"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import { Document } from "../../documents/document.js";
|
|
2
|
+
import { BaseRetriever } from "../../retrievers/index.js";
|
|
3
|
+
|
|
4
|
+
//#region src/utils/testing/retrievers.d.ts
|
|
5
|
+
declare class FakeRetriever extends BaseRetriever {
|
|
6
|
+
lc_namespace: string[];
|
|
7
|
+
output: Document<Record<string, any>>[];
|
|
8
|
+
constructor(fields?: {
|
|
9
|
+
output: Document[];
|
|
10
|
+
});
|
|
11
|
+
_getRelevantDocuments(_query: string
|
|
12
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
13
|
+
): Promise<Document<Record<string, any>>[]>;
|
|
14
|
+
}
|
|
15
|
+
//#endregion
|
|
16
|
+
export { FakeRetriever };
|
|
17
|
+
//# sourceMappingURL=retrievers.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"retrievers.d.ts","names":["Document","BaseRetriever","FakeRetriever","Record","Promise"],"sources":["../../../src/utils/testing/retrievers.d.ts"],"sourcesContent":["import { Document } from \"../../documents/document.js\";\nimport { BaseRetriever } from \"../../retrievers/index.js\";\nexport declare class FakeRetriever extends BaseRetriever {\n lc_namespace: string[];\n output: Document<Record<string, any>>[];\n constructor(fields?: {\n output: Document[];\n });\n _getRelevantDocuments(_query: string\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n ): Promise<Document<Record<string, any>>[]>;\n}\n"],"mappings":";;;;cAEqBE,aAAAA,SAAsBD,aAAAA;;EAAtBC,MAAAA,EAETF,QAFsB,CAEbG,MAFa,CAAA,MAAA,EAAA,GAAA,CAAA,CAAA,EAAA;EAAA,WAAA,CAAA,MAEP,CAFO,EAAA;IAEbA,MAAAA,EAELH,QAFKG,EAAAA;EAAM,CAAA;EAAP,qBAEJH,CAAAA,MAAAA,EAAAA;EAAQ;EAAA,CAIM,EAAvBI,OAAQJ,CAAAA,QAAAA,CAASG,MAATH,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,CAAAA,EAAAA,CAAAA"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import { BaseRetriever } from "../../retrievers/index.js";
|
|
2
|
+
import { Document } from "../../documents/document.js";
|
|
3
|
+
|
|
4
|
+
//#region src/utils/testing/retrievers.ts
|
|
5
|
+
var FakeRetriever = class extends BaseRetriever {
|
|
6
|
+
lc_namespace = ["test", "fake"];
|
|
7
|
+
output = [new Document({ pageContent: "foo" }), new Document({ pageContent: "bar" })];
|
|
8
|
+
constructor(fields) {
|
|
9
|
+
super();
|
|
10
|
+
this.output = fields?.output ?? this.output;
|
|
11
|
+
}
|
|
12
|
+
async _getRelevantDocuments(_query) {
|
|
13
|
+
return this.output;
|
|
14
|
+
}
|
|
15
|
+
};
|
|
16
|
+
|
|
17
|
+
//#endregion
|
|
18
|
+
export { FakeRetriever };
|
|
19
|
+
//# sourceMappingURL=retrievers.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"retrievers.js","names":["fields?: { output: Document[] }","_query: string"],"sources":["../../../src/utils/testing/retrievers.ts"],"sourcesContent":["import { Document } from \"../../documents/document.js\";\nimport { BaseRetriever } from \"../../retrievers/index.js\";\n\nexport class FakeRetriever extends BaseRetriever {\n lc_namespace = [\"test\", \"fake\"];\n\n output = [\n new Document({ pageContent: \"foo\" }),\n new Document({ pageContent: \"bar\" }),\n ];\n\n constructor(fields?: { output: Document[] }) {\n super();\n this.output = fields?.output ?? this.output;\n }\n\n async _getRelevantDocuments(\n _query: string\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n ): Promise<Document<Record<string, any>>[]> {\n return this.output;\n }\n}\n"],"mappings":";;;;AAGA,IAAa,gBAAb,cAAmC,cAAc;CAC/C,eAAe,CAAC,QAAQ,MAAO;CAE/B,SAAS,CACP,IAAI,SAAS,EAAE,aAAa,MAAO,IACnC,IAAI,SAAS,EAAE,aAAa,MAAO,EACpC;CAED,YAAYA,QAAiC;EAC3C,OAAO;EACP,KAAK,SAAS,QAAQ,UAAU,KAAK;CACtC;CAED,MAAM,sBACJC,QAE0C;AAC1C,SAAO,KAAK;CACb;AACF"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
const require_base = require('../../runnables/base.cjs');
|
|
2
|
+
|
|
3
|
+
//#region src/utils/testing/runnables.ts
|
|
4
|
+
var FakeRunnable = class extends require_base.Runnable {
|
|
5
|
+
lc_namespace = ["tests", "fake"];
|
|
6
|
+
returnOptions;
|
|
7
|
+
constructor(fields) {
|
|
8
|
+
super(fields);
|
|
9
|
+
this.returnOptions = fields.returnOptions;
|
|
10
|
+
}
|
|
11
|
+
async invoke(input, options) {
|
|
12
|
+
if (this.returnOptions) return options ?? {};
|
|
13
|
+
return { input };
|
|
14
|
+
}
|
|
15
|
+
};
|
|
16
|
+
|
|
17
|
+
//#endregion
|
|
18
|
+
exports.FakeRunnable = FakeRunnable;
|
|
19
|
+
//# sourceMappingURL=runnables.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runnables.cjs","names":["Runnable","fields: { returnOptions?: boolean }","input: string","options?: Partial<BaseCallbackConfig>"],"sources":["../../../src/utils/testing/runnables.ts"],"sourcesContent":["import { BaseCallbackConfig } from \"../../callbacks/manager.js\";\nimport { Runnable } from \"../../runnables/base.js\";\n\nexport class FakeRunnable extends Runnable<string, Record<string, unknown>> {\n lc_namespace = [\"tests\", \"fake\"];\n\n returnOptions?: boolean;\n\n constructor(fields: { returnOptions?: boolean }) {\n super(fields);\n this.returnOptions = fields.returnOptions;\n }\n\n async invoke(\n input: string,\n options?: Partial<BaseCallbackConfig>\n ): Promise<Record<string, unknown>> {\n if (this.returnOptions) {\n return options ?? {};\n }\n return { input };\n }\n}\n"],"mappings":";;;AAGA,IAAa,eAAb,cAAkCA,sBAA0C;CAC1E,eAAe,CAAC,SAAS,MAAO;CAEhC;CAEA,YAAYC,QAAqC;EAC/C,MAAM,OAAO;EACb,KAAK,gBAAgB,OAAO;CAC7B;CAED,MAAM,OACJC,OACAC,SACkC;AAClC,MAAI,KAAK,cACP,QAAO,WAAW,CAAE;AAEtB,SAAO,EAAE,MAAO;CACjB;AACF"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import { BaseCallbackConfig } from "../../callbacks/manager.cjs";
|
|
2
|
+
import { Runnable } from "../../runnables/base.cjs";
|
|
3
|
+
|
|
4
|
+
//#region src/utils/testing/runnables.d.ts
|
|
5
|
+
declare class FakeRunnable extends Runnable<string, Record<string, unknown>> {
|
|
6
|
+
lc_namespace: string[];
|
|
7
|
+
returnOptions?: boolean;
|
|
8
|
+
constructor(fields: {
|
|
9
|
+
returnOptions?: boolean;
|
|
10
|
+
});
|
|
11
|
+
invoke(input: string, options?: Partial<BaseCallbackConfig>): Promise<Record<string, unknown>>;
|
|
12
|
+
}
|
|
13
|
+
//#endregion
|
|
14
|
+
export { FakeRunnable };
|
|
15
|
+
//# sourceMappingURL=runnables.d.cts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runnables.d.cts","names":["BaseCallbackConfig","Runnable","FakeRunnable","Record","Partial","Promise"],"sources":["../../../src/utils/testing/runnables.d.ts"],"sourcesContent":["import { BaseCallbackConfig } from \"../../callbacks/manager.js\";\nimport { Runnable } from \"../../runnables/base.js\";\nexport declare class FakeRunnable extends Runnable<string, Record<string, unknown>> {\n lc_namespace: string[];\n returnOptions?: boolean;\n constructor(fields: {\n returnOptions?: boolean;\n });\n invoke(input: string, options?: Partial<BaseCallbackConfig>): Promise<Record<string, unknown>>;\n}\n"],"mappings":";;;;cAEqBE,YAAAA,SAAqBD,iBAAiBE;;EAAtCD,aAAAA,CAAAA,EAAY,OAAA;EAAA,WAAA,CAAA,MAAA,EAAA;IAA0BC,aAAAA,CAAAA,EAAAA,OAAAA;EAAM,CAAA;EAMH,MAA1BC,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,OAAAA,CAAQJ,kBAARI,CAAAA,CAAAA,EAA8BC,OAA9BD,CAAsCD,MAAtCC,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,CAAAA"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import { BaseCallbackConfig } from "../../callbacks/manager.js";
|
|
2
|
+
import { Runnable } from "../../runnables/base.js";
|
|
3
|
+
|
|
4
|
+
//#region src/utils/testing/runnables.d.ts
|
|
5
|
+
declare class FakeRunnable extends Runnable<string, Record<string, unknown>> {
|
|
6
|
+
lc_namespace: string[];
|
|
7
|
+
returnOptions?: boolean;
|
|
8
|
+
constructor(fields: {
|
|
9
|
+
returnOptions?: boolean;
|
|
10
|
+
});
|
|
11
|
+
invoke(input: string, options?: Partial<BaseCallbackConfig>): Promise<Record<string, unknown>>;
|
|
12
|
+
}
|
|
13
|
+
//#endregion
|
|
14
|
+
export { FakeRunnable };
|
|
15
|
+
//# sourceMappingURL=runnables.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runnables.d.ts","names":["BaseCallbackConfig","Runnable","FakeRunnable","Record","Partial","Promise"],"sources":["../../../src/utils/testing/runnables.d.ts"],"sourcesContent":["import { BaseCallbackConfig } from \"../../callbacks/manager.js\";\nimport { Runnable } from \"../../runnables/base.js\";\nexport declare class FakeRunnable extends Runnable<string, Record<string, unknown>> {\n lc_namespace: string[];\n returnOptions?: boolean;\n constructor(fields: {\n returnOptions?: boolean;\n });\n invoke(input: string, options?: Partial<BaseCallbackConfig>): Promise<Record<string, unknown>>;\n}\n"],"mappings":";;;;cAEqBE,YAAAA,SAAqBD,iBAAiBE;;EAAtCD,aAAAA,CAAAA,EAAY,OAAA;EAAA,WAAA,CAAA,MAAA,EAAA;IAA0BC,aAAAA,CAAAA,EAAAA,OAAAA;EAAM,CAAA;EAMH,MAA1BC,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,OAAAA,CAAQJ,kBAARI,CAAAA,CAAAA,EAA8BC,OAA9BD,CAAsCD,MAAtCC,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,CAAAA"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import { Runnable } from "../../runnables/base.js";
|
|
2
|
+
|
|
3
|
+
//#region src/utils/testing/runnables.ts
|
|
4
|
+
var FakeRunnable = class extends Runnable {
|
|
5
|
+
lc_namespace = ["tests", "fake"];
|
|
6
|
+
returnOptions;
|
|
7
|
+
constructor(fields) {
|
|
8
|
+
super(fields);
|
|
9
|
+
this.returnOptions = fields.returnOptions;
|
|
10
|
+
}
|
|
11
|
+
async invoke(input, options) {
|
|
12
|
+
if (this.returnOptions) return options ?? {};
|
|
13
|
+
return { input };
|
|
14
|
+
}
|
|
15
|
+
};
|
|
16
|
+
|
|
17
|
+
//#endregion
|
|
18
|
+
export { FakeRunnable };
|
|
19
|
+
//# sourceMappingURL=runnables.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runnables.js","names":["fields: { returnOptions?: boolean }","input: string","options?: Partial<BaseCallbackConfig>"],"sources":["../../../src/utils/testing/runnables.ts"],"sourcesContent":["import { BaseCallbackConfig } from \"../../callbacks/manager.js\";\nimport { Runnable } from \"../../runnables/base.js\";\n\nexport class FakeRunnable extends Runnable<string, Record<string, unknown>> {\n lc_namespace = [\"tests\", \"fake\"];\n\n returnOptions?: boolean;\n\n constructor(fields: { returnOptions?: boolean }) {\n super(fields);\n this.returnOptions = fields.returnOptions;\n }\n\n async invoke(\n input: string,\n options?: Partial<BaseCallbackConfig>\n ): Promise<Record<string, unknown>> {\n if (this.returnOptions) {\n return options ?? {};\n }\n return { input };\n }\n}\n"],"mappings":";;;AAGA,IAAa,eAAb,cAAkC,SAA0C;CAC1E,eAAe,CAAC,SAAS,MAAO;CAEhC;CAEA,YAAYA,QAAqC;EAC/C,MAAM,OAAO;EACb,KAAK,gBAAgB,OAAO;CAC7B;CAED,MAAM,OACJC,OACAC,SACkC;AAClC,MAAI,KAAK,cACP,QAAO,WAAW,CAAE;AAEtB,SAAO,EAAE,MAAO;CACjB;AACF"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
const require_tools_index = require('../../tools/index.cjs');
|
|
2
|
+
|
|
3
|
+
//#region src/utils/testing/tools.ts
|
|
4
|
+
var FakeTool = class extends require_tools_index.StructuredTool {
|
|
5
|
+
name;
|
|
6
|
+
description;
|
|
7
|
+
schema;
|
|
8
|
+
constructor(fields) {
|
|
9
|
+
super(fields);
|
|
10
|
+
this.name = fields.name;
|
|
11
|
+
this.description = fields.description;
|
|
12
|
+
this.schema = fields.schema;
|
|
13
|
+
}
|
|
14
|
+
async _call(arg, _runManager) {
|
|
15
|
+
return JSON.stringify(arg);
|
|
16
|
+
}
|
|
17
|
+
};
|
|
18
|
+
|
|
19
|
+
//#endregion
|
|
20
|
+
exports.FakeTool = FakeTool;
|
|
21
|
+
//# sourceMappingURL=tools.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.cjs","names":["StructuredTool","fields: FakeToolParams<T>","arg: ToolInputSchemaOutputType<T>","_runManager?: CallbackManagerForToolRun"],"sources":["../../../src/utils/testing/tools.ts"],"sourcesContent":["import { CallbackManagerForToolRun } from \"../../callbacks/manager.js\";\nimport { StructuredTool, ToolParams } from \"../../tools/index.js\";\nimport { ToolInputSchemaOutputType } from \"../../tools/types.js\";\nimport { InteropZodObject } from \"../types/zod.js\";\n\nexport interface FakeToolParams<T extends InteropZodObject = InteropZodObject>\n extends ToolParams {\n name: string;\n description: string;\n schema: T;\n}\n\nexport class FakeTool<\n T extends InteropZodObject = InteropZodObject\n> extends StructuredTool<T> {\n name: string;\n\n description: string;\n\n schema: T;\n\n constructor(fields: FakeToolParams<T>) {\n super(fields);\n this.name = fields.name;\n this.description = fields.description;\n this.schema = fields.schema;\n }\n\n protected async _call(\n arg: ToolInputSchemaOutputType<T>,\n _runManager?: CallbackManagerForToolRun\n ): Promise<string> {\n return JSON.stringify(arg);\n }\n}\n"],"mappings":";;;AAYA,IAAa,WAAb,cAEUA,mCAAkB;CAC1B;CAEA;CAEA;CAEA,YAAYC,QAA2B;EACrC,MAAM,OAAO;EACb,KAAK,OAAO,OAAO;EACnB,KAAK,cAAc,OAAO;EAC1B,KAAK,SAAS,OAAO;CACtB;CAED,MAAgB,MACdC,KACAC,aACiB;AACjB,SAAO,KAAK,UAAU,IAAI;CAC3B;AACF"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import { InteropZodObject } from "../types/zod.cjs";
|
|
2
|
+
import { CallbackManagerForToolRun } from "../../callbacks/manager.cjs";
|
|
3
|
+
import { ToolInputSchemaOutputType, ToolParams } from "../../tools/types.cjs";
|
|
4
|
+
import { StructuredTool } from "../../tools/index.cjs";
|
|
5
|
+
|
|
6
|
+
//#region src/utils/testing/tools.d.ts
|
|
7
|
+
interface FakeToolParams<T extends InteropZodObject = InteropZodObject> extends ToolParams {
|
|
8
|
+
name: string;
|
|
9
|
+
description: string;
|
|
10
|
+
schema: T;
|
|
11
|
+
}
|
|
12
|
+
declare class FakeTool<T extends InteropZodObject = InteropZodObject> extends StructuredTool<T> {
|
|
13
|
+
name: string;
|
|
14
|
+
description: string;
|
|
15
|
+
schema: T;
|
|
16
|
+
constructor(fields: FakeToolParams<T>);
|
|
17
|
+
protected _call(arg: ToolInputSchemaOutputType<T>, _runManager?: CallbackManagerForToolRun): Promise<string>;
|
|
18
|
+
}
|
|
19
|
+
//#endregion
|
|
20
|
+
export { FakeTool, FakeToolParams };
|
|
21
|
+
//# sourceMappingURL=tools.d.cts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.d.cts","names":["CallbackManagerForToolRun","StructuredTool","ToolParams","ToolInputSchemaOutputType","InteropZodObject","FakeToolParams","T","FakeTool","Promise"],"sources":["../../../src/utils/testing/tools.d.ts"],"sourcesContent":["import { CallbackManagerForToolRun } from \"../../callbacks/manager.js\";\nimport { StructuredTool, ToolParams } from \"../../tools/index.js\";\nimport { ToolInputSchemaOutputType } from \"../../tools/types.js\";\nimport { InteropZodObject } from \"../types/zod.js\";\nexport interface FakeToolParams<T extends InteropZodObject = InteropZodObject> extends ToolParams {\n name: string;\n description: string;\n schema: T;\n}\nexport declare class FakeTool<T extends InteropZodObject = InteropZodObject> extends StructuredTool<T> {\n name: string;\n description: string;\n schema: T;\n constructor(fields: FakeToolParams<T>);\n protected _call(arg: ToolInputSchemaOutputType<T>, _runManager?: CallbackManagerForToolRun): Promise<string>;\n}\n"],"mappings":";;;;;;UAIiBK,yBAAyBD,mBAAmBA,0BAA0BF;;EAAtEG,WAAAA,EAAAA,MAAc;EAAA,MAAA,EAGnBC,CAHmB;;AAA8BF,cAKxCG,QALwCH,CAAAA,UAKrBA,gBALqBA,GAKFA,gBALEA,CAAAA,SAKwBH,cALxBG,CAKuCE,CALvCF,CAAAA,CAAAA;EAAgB,IAGjEE,EAAAA,MAAAA;EAAC,WAH0EJ,EAAAA,MAAAA;EAAU,MAAA,EAQrFI,CARqF;EAK5EC,WAAQ,CAAA,MAAA,EAILF,cAJK,CAIUC,CAJV,CAAA;EAAA,UAAA,KAAA,CAAA,GAAA,EAKJH,yBALI,CAKsBG,CALtB,CAAA,EAAA,WAAA,CAAA,EAKwCN,yBALxC,CAAA,EAKoEQ,OALpE,CAAA,MAAA,CAAA"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import { InteropZodObject } from "../types/zod.js";
|
|
2
|
+
import { CallbackManagerForToolRun } from "../../callbacks/manager.js";
|
|
3
|
+
import { ToolInputSchemaOutputType, ToolParams } from "../../tools/types.js";
|
|
4
|
+
import { StructuredTool } from "../../tools/index.js";
|
|
5
|
+
|
|
6
|
+
//#region src/utils/testing/tools.d.ts
|
|
7
|
+
interface FakeToolParams<T extends InteropZodObject = InteropZodObject> extends ToolParams {
|
|
8
|
+
name: string;
|
|
9
|
+
description: string;
|
|
10
|
+
schema: T;
|
|
11
|
+
}
|
|
12
|
+
declare class FakeTool<T extends InteropZodObject = InteropZodObject> extends StructuredTool<T> {
|
|
13
|
+
name: string;
|
|
14
|
+
description: string;
|
|
15
|
+
schema: T;
|
|
16
|
+
constructor(fields: FakeToolParams<T>);
|
|
17
|
+
protected _call(arg: ToolInputSchemaOutputType<T>, _runManager?: CallbackManagerForToolRun): Promise<string>;
|
|
18
|
+
}
|
|
19
|
+
//#endregion
|
|
20
|
+
export { FakeTool, FakeToolParams };
|
|
21
|
+
//# sourceMappingURL=tools.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.d.ts","names":["CallbackManagerForToolRun","StructuredTool","ToolParams","ToolInputSchemaOutputType","InteropZodObject","FakeToolParams","T","FakeTool","Promise"],"sources":["../../../src/utils/testing/tools.d.ts"],"sourcesContent":["import { CallbackManagerForToolRun } from \"../../callbacks/manager.js\";\nimport { StructuredTool, ToolParams } from \"../../tools/index.js\";\nimport { ToolInputSchemaOutputType } from \"../../tools/types.js\";\nimport { InteropZodObject } from \"../types/zod.js\";\nexport interface FakeToolParams<T extends InteropZodObject = InteropZodObject> extends ToolParams {\n name: string;\n description: string;\n schema: T;\n}\nexport declare class FakeTool<T extends InteropZodObject = InteropZodObject> extends StructuredTool<T> {\n name: string;\n description: string;\n schema: T;\n constructor(fields: FakeToolParams<T>);\n protected _call(arg: ToolInputSchemaOutputType<T>, _runManager?: CallbackManagerForToolRun): Promise<string>;\n}\n"],"mappings":";;;;;;UAIiBK,yBAAyBD,mBAAmBA,0BAA0BF;;EAAtEG,WAAAA,EAAAA,MAAc;EAAA,MAAA,EAGnBC,CAHmB;;AAA8BF,cAKxCG,QALwCH,CAAAA,UAKrBA,gBALqBA,GAKFA,gBALEA,CAAAA,SAKwBH,cALxBG,CAKuCE,CALvCF,CAAAA,CAAAA;EAAgB,IAGjEE,EAAAA,MAAAA;EAAC,WAH0EJ,EAAAA,MAAAA;EAAU,MAAA,EAQrFI,CARqF;EAK5EC,WAAQ,CAAA,MAAA,EAILF,cAJK,CAIUC,CAJV,CAAA;EAAA,UAAA,KAAA,CAAA,GAAA,EAKJH,yBALI,CAKsBG,CALtB,CAAA,EAAA,WAAA,CAAA,EAKwCN,yBALxC,CAAA,EAKoEQ,OALpE,CAAA,MAAA,CAAA"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import { StructuredTool } from "../../tools/index.js";
|
|
2
|
+
|
|
3
|
+
//#region src/utils/testing/tools.ts
|
|
4
|
+
var FakeTool = class extends StructuredTool {
|
|
5
|
+
name;
|
|
6
|
+
description;
|
|
7
|
+
schema;
|
|
8
|
+
constructor(fields) {
|
|
9
|
+
super(fields);
|
|
10
|
+
this.name = fields.name;
|
|
11
|
+
this.description = fields.description;
|
|
12
|
+
this.schema = fields.schema;
|
|
13
|
+
}
|
|
14
|
+
async _call(arg, _runManager) {
|
|
15
|
+
return JSON.stringify(arg);
|
|
16
|
+
}
|
|
17
|
+
};
|
|
18
|
+
|
|
19
|
+
//#endregion
|
|
20
|
+
export { FakeTool };
|
|
21
|
+
//# sourceMappingURL=tools.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.js","names":["fields: FakeToolParams<T>","arg: ToolInputSchemaOutputType<T>","_runManager?: CallbackManagerForToolRun"],"sources":["../../../src/utils/testing/tools.ts"],"sourcesContent":["import { CallbackManagerForToolRun } from \"../../callbacks/manager.js\";\nimport { StructuredTool, ToolParams } from \"../../tools/index.js\";\nimport { ToolInputSchemaOutputType } from \"../../tools/types.js\";\nimport { InteropZodObject } from \"../types/zod.js\";\n\nexport interface FakeToolParams<T extends InteropZodObject = InteropZodObject>\n extends ToolParams {\n name: string;\n description: string;\n schema: T;\n}\n\nexport class FakeTool<\n T extends InteropZodObject = InteropZodObject\n> extends StructuredTool<T> {\n name: string;\n\n description: string;\n\n schema: T;\n\n constructor(fields: FakeToolParams<T>) {\n super(fields);\n this.name = fields.name;\n this.description = fields.description;\n this.schema = fields.schema;\n }\n\n protected async _call(\n arg: ToolInputSchemaOutputType<T>,\n _runManager?: CallbackManagerForToolRun\n ): Promise<string> {\n return JSON.stringify(arg);\n }\n}\n"],"mappings":";;;AAYA,IAAa,WAAb,cAEU,eAAkB;CAC1B;CAEA;CAEA;CAEA,YAAYA,QAA2B;EACrC,MAAM,OAAO;EACb,KAAK,OAAO,OAAO;EACnB,KAAK,cAAc,OAAO;EAC1B,KAAK,SAAS,OAAO;CACtB;CAED,MAAgB,MACdC,KACAC,aACiB;AACjB,SAAO,KAAK,UAAU,IAAI;CAC3B;AACF"}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
const require_tracers_base = require('../../tracers/base.cjs');
|
|
2
|
+
|
|
3
|
+
//#region src/utils/testing/tracers.ts
|
|
4
|
+
var SingleRunExtractor = class extends require_tracers_base.BaseTracer {
|
|
5
|
+
runPromiseResolver;
|
|
6
|
+
runPromise;
|
|
7
|
+
/** The name of the callback handler. */
|
|
8
|
+
name = "single_run_extractor";
|
|
9
|
+
constructor() {
|
|
10
|
+
super();
|
|
11
|
+
this.runPromise = new Promise((extract) => {
|
|
12
|
+
this.runPromiseResolver = extract;
|
|
13
|
+
});
|
|
14
|
+
}
|
|
15
|
+
async persistRun(run) {
|
|
16
|
+
this.runPromiseResolver(run);
|
|
17
|
+
}
|
|
18
|
+
async extract() {
|
|
19
|
+
return this.runPromise;
|
|
20
|
+
}
|
|
21
|
+
};
|
|
22
|
+
|
|
23
|
+
//#endregion
|
|
24
|
+
exports.SingleRunExtractor = SingleRunExtractor;
|
|
25
|
+
//# sourceMappingURL=tracers.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tracers.cjs","names":["BaseTracer","run: Run"],"sources":["../../../src/utils/testing/tracers.ts"],"sourcesContent":["import { BaseTracer, Run } from \"../../tracers/base.js\";\n\nexport class SingleRunExtractor extends BaseTracer {\n runPromiseResolver: (run: Run) => void;\n\n runPromise: Promise<Run>;\n\n /** The name of the callback handler. */\n name = \"single_run_extractor\";\n\n constructor() {\n super();\n this.runPromise = new Promise<Run>((extract) => {\n this.runPromiseResolver = extract;\n });\n }\n\n async persistRun(run: Run) {\n this.runPromiseResolver(run);\n }\n\n async extract(): Promise<Run> {\n return this.runPromise;\n }\n}\n"],"mappings":";;;AAEA,IAAa,qBAAb,cAAwCA,gCAAW;CACjD;CAEA;;CAGA,OAAO;CAEP,cAAc;EACZ,OAAO;EACP,KAAK,aAAa,IAAI,QAAa,CAAC,YAAY;GAC9C,KAAK,qBAAqB;EAC3B;CACF;CAED,MAAM,WAAWC,KAAU;EACzB,KAAK,mBAAmB,IAAI;CAC7B;CAED,MAAM,UAAwB;AAC5B,SAAO,KAAK;CACb;AACF"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import { BaseTracer, Run } from "../../tracers/base.cjs";
|
|
2
|
+
|
|
3
|
+
//#region src/utils/testing/tracers.d.ts
|
|
4
|
+
declare class SingleRunExtractor extends BaseTracer {
|
|
5
|
+
runPromiseResolver: (run: Run) => void;
|
|
6
|
+
runPromise: Promise<Run>;
|
|
7
|
+
/** The name of the callback handler. */
|
|
8
|
+
name: string;
|
|
9
|
+
constructor();
|
|
10
|
+
persistRun(run: Run): Promise<void>;
|
|
11
|
+
extract(): Promise<Run>;
|
|
12
|
+
}
|
|
13
|
+
//#endregion
|
|
14
|
+
export { SingleRunExtractor };
|
|
15
|
+
//# sourceMappingURL=tracers.d.cts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tracers.d.cts","names":["BaseTracer","Run","SingleRunExtractor","Promise"],"sources":["../../../src/utils/testing/tracers.d.ts"],"sourcesContent":["import { BaseTracer, Run } from \"../../tracers/base.js\";\nexport declare class SingleRunExtractor extends BaseTracer {\n runPromiseResolver: (run: Run) => void;\n runPromise: Promise<Run>;\n /** The name of the callback handler. */\n name: string;\n constructor();\n persistRun(run: Run): Promise<void>;\n extract(): Promise<Run>;\n}\n"],"mappings":";;;cACqBE,kBAAAA,SAA2BF,UAAAA;4BAClBC;EADTC,UAAAA,EAELC,OAFKD,CAEGD,GAFe,CAAA;EAAA;EAAA,IACTA,EAAAA,MAAAA;EAAG,WACTA,CAAAA;EAAG,UAAXE,CAAAA,GAAAA,EAIIF,GAJJE,CAAAA,EAIUA,OAJVA,CAAAA,IAAAA,CAAAA;EAAO,OAIHF,CAAAA,CAAAA,EACLE,OADKF,CACGA,GADHA,CAAAA"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import { BaseTracer, Run } from "../../tracers/base.js";
|
|
2
|
+
|
|
3
|
+
//#region src/utils/testing/tracers.d.ts
|
|
4
|
+
declare class SingleRunExtractor extends BaseTracer {
|
|
5
|
+
runPromiseResolver: (run: Run) => void;
|
|
6
|
+
runPromise: Promise<Run>;
|
|
7
|
+
/** The name of the callback handler. */
|
|
8
|
+
name: string;
|
|
9
|
+
constructor();
|
|
10
|
+
persistRun(run: Run): Promise<void>;
|
|
11
|
+
extract(): Promise<Run>;
|
|
12
|
+
}
|
|
13
|
+
//#endregion
|
|
14
|
+
export { SingleRunExtractor };
|
|
15
|
+
//# sourceMappingURL=tracers.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tracers.d.ts","names":["BaseTracer","Run","SingleRunExtractor","Promise"],"sources":["../../../src/utils/testing/tracers.d.ts"],"sourcesContent":["import { BaseTracer, Run } from \"../../tracers/base.js\";\nexport declare class SingleRunExtractor extends BaseTracer {\n runPromiseResolver: (run: Run) => void;\n runPromise: Promise<Run>;\n /** The name of the callback handler. */\n name: string;\n constructor();\n persistRun(run: Run): Promise<void>;\n extract(): Promise<Run>;\n}\n"],"mappings":";;;cACqBE,kBAAAA,SAA2BF,UAAAA;4BAClBC;EADTC,UAAAA,EAELC,OAFKD,CAEGD,GAFe,CAAA;EAAA;EAAA,IACTA,EAAAA,MAAAA;EAAG,WACTA,CAAAA;EAAG,UAAXE,CAAAA,GAAAA,EAIIF,GAJJE,CAAAA,EAIUA,OAJVA,CAAAA,IAAAA,CAAAA;EAAO,OAIHF,CAAAA,CAAAA,EACLE,OADKF,CACGA,GADHA,CAAAA"}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
import { BaseTracer } from "../../tracers/base.js";
|
|
2
|
+
|
|
3
|
+
//#region src/utils/testing/tracers.ts
|
|
4
|
+
var SingleRunExtractor = class extends BaseTracer {
|
|
5
|
+
runPromiseResolver;
|
|
6
|
+
runPromise;
|
|
7
|
+
/** The name of the callback handler. */
|
|
8
|
+
name = "single_run_extractor";
|
|
9
|
+
constructor() {
|
|
10
|
+
super();
|
|
11
|
+
this.runPromise = new Promise((extract) => {
|
|
12
|
+
this.runPromiseResolver = extract;
|
|
13
|
+
});
|
|
14
|
+
}
|
|
15
|
+
async persistRun(run) {
|
|
16
|
+
this.runPromiseResolver(run);
|
|
17
|
+
}
|
|
18
|
+
async extract() {
|
|
19
|
+
return this.runPromise;
|
|
20
|
+
}
|
|
21
|
+
};
|
|
22
|
+
|
|
23
|
+
//#endregion
|
|
24
|
+
export { SingleRunExtractor };
|
|
25
|
+
//# sourceMappingURL=tracers.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tracers.js","names":["run: Run"],"sources":["../../../src/utils/testing/tracers.ts"],"sourcesContent":["import { BaseTracer, Run } from \"../../tracers/base.js\";\n\nexport class SingleRunExtractor extends BaseTracer {\n runPromiseResolver: (run: Run) => void;\n\n runPromise: Promise<Run>;\n\n /** The name of the callback handler. */\n name = \"single_run_extractor\";\n\n constructor() {\n super();\n this.runPromise = new Promise<Run>((extract) => {\n this.runPromiseResolver = extract;\n });\n }\n\n async persistRun(run: Run) {\n this.runPromiseResolver(run);\n }\n\n async extract(): Promise<Run> {\n return this.runPromise;\n }\n}\n"],"mappings":";;;AAEA,IAAa,qBAAb,cAAwC,WAAW;CACjD;CAEA;;CAGA,OAAO;CAEP,cAAc;EACZ,OAAO;EACP,KAAK,aAAa,IAAI,QAAa,CAAC,YAAY;GAC9C,KAAK,qBAAqB;EAC3B;CACF;CAED,MAAM,WAAWA,KAAU;EACzB,KAAK,mBAAmB,IAAI;CAC7B;CAED,MAAM,UAAwB;AAC5B,SAAO,KAAK;CACb;AACF"}
|
|
@@ -0,0 +1,129 @@
|
|
|
1
|
+
const require_vectorstores = require('../../vectorstores.cjs');
|
|
2
|
+
const require_document = require('../../documents/document.cjs');
|
|
3
|
+
const require_similarities = require('../ml-distance/similarities.cjs');
|
|
4
|
+
|
|
5
|
+
//#region src/utils/testing/vectorstores.ts
|
|
6
|
+
/**
|
|
7
|
+
* Class that extends `VectorStore` to store vectors in memory. Provides
|
|
8
|
+
* methods for adding documents, performing similarity searches, and
|
|
9
|
+
* creating instances from texts, documents, or an existing index.
|
|
10
|
+
*/
|
|
11
|
+
var FakeVectorStore = class FakeVectorStore extends require_vectorstores.VectorStore {
|
|
12
|
+
memoryVectors = [];
|
|
13
|
+
similarity;
|
|
14
|
+
_vectorstoreType() {
|
|
15
|
+
return "memory";
|
|
16
|
+
}
|
|
17
|
+
constructor(embeddings, { similarity,...rest } = {}) {
|
|
18
|
+
super(embeddings, rest);
|
|
19
|
+
this.similarity = similarity ?? require_similarities.cosine;
|
|
20
|
+
}
|
|
21
|
+
/**
|
|
22
|
+
* Method to add documents to the memory vector store. It extracts the
|
|
23
|
+
* text from each document, generates embeddings for them, and adds the
|
|
24
|
+
* resulting vectors to the store.
|
|
25
|
+
* @param documents Array of `Document` instances to be added to the store.
|
|
26
|
+
* @returns Promise that resolves when all documents have been added.
|
|
27
|
+
*/
|
|
28
|
+
async addDocuments(documents) {
|
|
29
|
+
const texts = documents.map(({ pageContent }) => pageContent);
|
|
30
|
+
return this.addVectors(await this.embeddings.embedDocuments(texts), documents);
|
|
31
|
+
}
|
|
32
|
+
/**
|
|
33
|
+
* Method to add vectors to the memory vector store. It creates
|
|
34
|
+
* `MemoryVector` instances for each vector and document pair and adds
|
|
35
|
+
* them to the store.
|
|
36
|
+
* @param vectors Array of vectors to be added to the store.
|
|
37
|
+
* @param documents Array of `Document` instances corresponding to the vectors.
|
|
38
|
+
* @returns Promise that resolves when all vectors have been added.
|
|
39
|
+
*/
|
|
40
|
+
async addVectors(vectors, documents) {
|
|
41
|
+
const memoryVectors = vectors.map((embedding, idx) => ({
|
|
42
|
+
content: documents[idx].pageContent,
|
|
43
|
+
embedding,
|
|
44
|
+
metadata: documents[idx].metadata
|
|
45
|
+
}));
|
|
46
|
+
this.memoryVectors = this.memoryVectors.concat(memoryVectors);
|
|
47
|
+
}
|
|
48
|
+
/**
|
|
49
|
+
* Method to perform a similarity search in the memory vector store. It
|
|
50
|
+
* calculates the similarity between the query vector and each vector in
|
|
51
|
+
* the store, sorts the results by similarity, and returns the top `k`
|
|
52
|
+
* results along with their scores.
|
|
53
|
+
* @param query Query vector to compare against the vectors in the store.
|
|
54
|
+
* @param k Number of top results to return.
|
|
55
|
+
* @param filter Optional filter function to apply to the vectors before performing the search.
|
|
56
|
+
* @returns Promise that resolves with an array of tuples, each containing a `Document` and its similarity score.
|
|
57
|
+
*/
|
|
58
|
+
async similaritySearchVectorWithScore(query, k, filter) {
|
|
59
|
+
const filterFunction = (memoryVector) => {
|
|
60
|
+
if (!filter) return true;
|
|
61
|
+
const doc = new require_document.Document({
|
|
62
|
+
metadata: memoryVector.metadata,
|
|
63
|
+
pageContent: memoryVector.content
|
|
64
|
+
});
|
|
65
|
+
return filter(doc);
|
|
66
|
+
};
|
|
67
|
+
const filteredMemoryVectors = this.memoryVectors.filter(filterFunction);
|
|
68
|
+
const searches = filteredMemoryVectors.map((vector, index) => ({
|
|
69
|
+
similarity: this.similarity(query, vector.embedding),
|
|
70
|
+
index
|
|
71
|
+
})).sort((a, b) => a.similarity > b.similarity ? -1 : 0).slice(0, k);
|
|
72
|
+
const result = searches.map((search) => [new require_document.Document({
|
|
73
|
+
metadata: filteredMemoryVectors[search.index].metadata,
|
|
74
|
+
pageContent: filteredMemoryVectors[search.index].content
|
|
75
|
+
}), search.similarity]);
|
|
76
|
+
return result;
|
|
77
|
+
}
|
|
78
|
+
/**
|
|
79
|
+
* Static method to create a `FakeVectorStore` instance from an array of
|
|
80
|
+
* texts. It creates a `Document` for each text and metadata pair, and
|
|
81
|
+
* adds them to the store.
|
|
82
|
+
* @param texts Array of texts to be added to the store.
|
|
83
|
+
* @param metadatas Array or single object of metadata corresponding to the texts.
|
|
84
|
+
* @param embeddings `Embeddings` instance used to generate embeddings for the texts.
|
|
85
|
+
* @param dbConfig Optional `FakeVectorStoreArgs` to configure the `FakeVectorStore` instance.
|
|
86
|
+
* @returns Promise that resolves with a new `FakeVectorStore` instance.
|
|
87
|
+
*/
|
|
88
|
+
static async fromTexts(texts, metadatas, embeddings, dbConfig) {
|
|
89
|
+
const docs = [];
|
|
90
|
+
for (let i = 0; i < texts.length; i += 1) {
|
|
91
|
+
const metadata = Array.isArray(metadatas) ? metadatas[i] : metadatas;
|
|
92
|
+
const newDoc = new require_document.Document({
|
|
93
|
+
pageContent: texts[i],
|
|
94
|
+
metadata
|
|
95
|
+
});
|
|
96
|
+
docs.push(newDoc);
|
|
97
|
+
}
|
|
98
|
+
return FakeVectorStore.fromDocuments(docs, embeddings, dbConfig);
|
|
99
|
+
}
|
|
100
|
+
/**
|
|
101
|
+
* Static method to create a `FakeVectorStore` instance from an array of
|
|
102
|
+
* `Document` instances. It adds the documents to the store.
|
|
103
|
+
* @param docs Array of `Document` instances to be added to the store.
|
|
104
|
+
* @param embeddings `Embeddings` instance used to generate embeddings for the documents.
|
|
105
|
+
* @param dbConfig Optional `FakeVectorStoreArgs` to configure the `FakeVectorStore` instance.
|
|
106
|
+
* @returns Promise that resolves with a new `FakeVectorStore` instance.
|
|
107
|
+
*/
|
|
108
|
+
static async fromDocuments(docs, embeddings, dbConfig) {
|
|
109
|
+
const instance = new this(embeddings, dbConfig);
|
|
110
|
+
await instance.addDocuments(docs);
|
|
111
|
+
return instance;
|
|
112
|
+
}
|
|
113
|
+
/**
|
|
114
|
+
* Static method to create a `FakeVectorStore` instance from an existing
|
|
115
|
+
* index. It creates a new `FakeVectorStore` instance without adding any
|
|
116
|
+
* documents or vectors.
|
|
117
|
+
* @param embeddings `Embeddings` instance used to generate embeddings for the documents.
|
|
118
|
+
* @param dbConfig Optional `FakeVectorStoreArgs` to configure the `FakeVectorStore` instance.
|
|
119
|
+
* @returns Promise that resolves with a new `FakeVectorStore` instance.
|
|
120
|
+
*/
|
|
121
|
+
static async fromExistingIndex(embeddings, dbConfig) {
|
|
122
|
+
const instance = new this(embeddings, dbConfig);
|
|
123
|
+
return instance;
|
|
124
|
+
}
|
|
125
|
+
};
|
|
126
|
+
|
|
127
|
+
//#endregion
|
|
128
|
+
exports.FakeVectorStore = FakeVectorStore;
|
|
129
|
+
//# sourceMappingURL=vectorstores.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"vectorstores.cjs","names":["VectorStore","embeddings: EmbeddingsInterface","cosine","documents: Document[]","vectors: number[][]","query: number[]","k: number","filter?: this[\"FilterType\"]","memoryVector: MemoryVector","Document","result: [Document, number][]","texts: string[]","metadatas: object[] | object","dbConfig?: FakeVectorStoreArgs","docs: Document[]"],"sources":["../../../src/utils/testing/vectorstores.ts"],"sourcesContent":["import { Document } from \"../../documents/document.js\";\nimport { EmbeddingsInterface } from \"../../embeddings.js\";\nimport { VectorStore } from \"../../vectorstores.js\";\nimport { cosine } from \"../ml-distance/similarities.js\";\n\n/**\n * Interface representing a vector in memory. It includes the content\n * (text), the corresponding embedding (vector), and any associated\n * metadata.\n */\ninterface MemoryVector {\n content: string;\n embedding: number[];\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n metadata: Record<string, any>;\n}\n\n/**\n * Interface for the arguments that can be passed to the\n * `FakeVectorStore` constructor. It includes an optional `similarity`\n * function.\n */\nexport interface FakeVectorStoreArgs {\n similarity?: typeof cosine;\n}\n\n/**\n * Class that extends `VectorStore` to store vectors in memory. Provides\n * methods for adding documents, performing similarity searches, and\n * creating instances from texts, documents, or an existing index.\n */\nexport class FakeVectorStore extends VectorStore {\n declare FilterType: (doc: Document) => boolean;\n\n memoryVectors: MemoryVector[] = [];\n\n similarity: typeof cosine;\n\n _vectorstoreType(): string {\n return \"memory\";\n }\n\n constructor(\n embeddings: EmbeddingsInterface,\n { similarity, ...rest }: FakeVectorStoreArgs = {}\n ) {\n super(embeddings, rest);\n\n this.similarity = similarity ?? cosine;\n }\n\n /**\n * Method to add documents to the memory vector store. It extracts the\n * text from each document, generates embeddings for them, and adds the\n * resulting vectors to the store.\n * @param documents Array of `Document` instances to be added to the store.\n * @returns Promise that resolves when all documents have been added.\n */\n async addDocuments(documents: Document[]): Promise<void> {\n const texts = documents.map(({ pageContent }) => pageContent);\n return this.addVectors(\n await this.embeddings.embedDocuments(texts),\n documents\n );\n }\n\n /**\n * Method to add vectors to the memory vector store. It creates\n * `MemoryVector` instances for each vector and document pair and adds\n * them to the store.\n * @param vectors Array of vectors to be added to the store.\n * @param documents Array of `Document` instances corresponding to the vectors.\n * @returns Promise that resolves when all vectors have been added.\n */\n async addVectors(vectors: number[][], documents: Document[]): Promise<void> {\n const memoryVectors = vectors.map((embedding, idx) => ({\n content: documents[idx].pageContent,\n embedding,\n metadata: documents[idx].metadata,\n }));\n\n this.memoryVectors = this.memoryVectors.concat(memoryVectors);\n }\n\n /**\n * Method to perform a similarity search in the memory vector store. It\n * calculates the similarity between the query vector and each vector in\n * the store, sorts the results by similarity, and returns the top `k`\n * results along with their scores.\n * @param query Query vector to compare against the vectors in the store.\n * @param k Number of top results to return.\n * @param filter Optional filter function to apply to the vectors before performing the search.\n * @returns Promise that resolves with an array of tuples, each containing a `Document` and its similarity score.\n */\n async similaritySearchVectorWithScore(\n query: number[],\n k: number,\n filter?: this[\"FilterType\"]\n ): Promise<[Document, number][]> {\n const filterFunction = (memoryVector: MemoryVector) => {\n if (!filter) {\n return true;\n }\n\n const doc = new Document({\n metadata: memoryVector.metadata,\n pageContent: memoryVector.content,\n });\n return filter(doc);\n };\n const filteredMemoryVectors = this.memoryVectors.filter(filterFunction);\n const searches = filteredMemoryVectors\n .map((vector, index) => ({\n similarity: this.similarity(query, vector.embedding),\n index,\n }))\n .sort((a, b) => (a.similarity > b.similarity ? -1 : 0))\n .slice(0, k);\n\n const result: [Document, number][] = searches.map((search) => [\n new Document({\n metadata: filteredMemoryVectors[search.index].metadata,\n pageContent: filteredMemoryVectors[search.index].content,\n }),\n search.similarity,\n ]);\n\n return result;\n }\n\n /**\n * Static method to create a `FakeVectorStore` instance from an array of\n * texts. It creates a `Document` for each text and metadata pair, and\n * adds them to the store.\n * @param texts Array of texts to be added to the store.\n * @param metadatas Array or single object of metadata corresponding to the texts.\n * @param embeddings `Embeddings` instance used to generate embeddings for the texts.\n * @param dbConfig Optional `FakeVectorStoreArgs` to configure the `FakeVectorStore` instance.\n * @returns Promise that resolves with a new `FakeVectorStore` instance.\n */\n static async fromTexts(\n texts: string[],\n metadatas: object[] | object,\n embeddings: EmbeddingsInterface,\n dbConfig?: FakeVectorStoreArgs\n ): Promise<FakeVectorStore> {\n const docs: Document[] = [];\n for (let i = 0; i < texts.length; i += 1) {\n const metadata = Array.isArray(metadatas) ? metadatas[i] : metadatas;\n const newDoc = new Document({\n pageContent: texts[i],\n metadata,\n });\n docs.push(newDoc);\n }\n return FakeVectorStore.fromDocuments(docs, embeddings, dbConfig);\n }\n\n /**\n * Static method to create a `FakeVectorStore` instance from an array of\n * `Document` instances. It adds the documents to the store.\n * @param docs Array of `Document` instances to be added to the store.\n * @param embeddings `Embeddings` instance used to generate embeddings for the documents.\n * @param dbConfig Optional `FakeVectorStoreArgs` to configure the `FakeVectorStore` instance.\n * @returns Promise that resolves with a new `FakeVectorStore` instance.\n */\n static async fromDocuments(\n docs: Document[],\n embeddings: EmbeddingsInterface,\n dbConfig?: FakeVectorStoreArgs\n ): Promise<FakeVectorStore> {\n const instance = new this(embeddings, dbConfig);\n await instance.addDocuments(docs);\n return instance;\n }\n\n /**\n * Static method to create a `FakeVectorStore` instance from an existing\n * index. It creates a new `FakeVectorStore` instance without adding any\n * documents or vectors.\n * @param embeddings `Embeddings` instance used to generate embeddings for the documents.\n * @param dbConfig Optional `FakeVectorStoreArgs` to configure the `FakeVectorStore` instance.\n * @returns Promise that resolves with a new `FakeVectorStore` instance.\n */\n static async fromExistingIndex(\n embeddings: EmbeddingsInterface,\n dbConfig?: FakeVectorStoreArgs\n ): Promise<FakeVectorStore> {\n const instance = new this(embeddings, dbConfig);\n return instance;\n }\n}\n"],"mappings":";;;;;;;;;;AA+BA,IAAa,kBAAb,MAAa,wBAAwBA,iCAAY;CAG/C,gBAAgC,CAAE;CAElC;CAEA,mBAA2B;AACzB,SAAO;CACR;CAED,YACEC,YACA,EAAE,WAAY,GAAG,MAA2B,GAAG,CAAE,GACjD;EACA,MAAM,YAAY,KAAK;EAEvB,KAAK,aAAa,cAAcC;CACjC;;;;;;;;CASD,MAAM,aAAaC,WAAsC;EACvD,MAAM,QAAQ,UAAU,IAAI,CAAC,EAAE,aAAa,KAAK,YAAY;AAC7D,SAAO,KAAK,WACV,MAAM,KAAK,WAAW,eAAe,MAAM,EAC3C,UACD;CACF;;;;;;;;;CAUD,MAAM,WAAWC,SAAqBD,WAAsC;EAC1E,MAAM,gBAAgB,QAAQ,IAAI,CAAC,WAAW,SAAS;GACrD,SAAS,UAAU,KAAK;GACxB;GACA,UAAU,UAAU,KAAK;EAC1B,GAAE;EAEH,KAAK,gBAAgB,KAAK,cAAc,OAAO,cAAc;CAC9D;;;;;;;;;;;CAYD,MAAM,gCACJE,OACAC,GACAC,QAC+B;EAC/B,MAAM,iBAAiB,CAACC,iBAA+B;AACrD,OAAI,CAAC,OACH,QAAO;GAGT,MAAM,MAAM,IAAIC,0BAAS;IACvB,UAAU,aAAa;IACvB,aAAa,aAAa;GAC3B;AACD,UAAO,OAAO,IAAI;EACnB;EACD,MAAM,wBAAwB,KAAK,cAAc,OAAO,eAAe;EACvE,MAAM,WAAW,sBACd,IAAI,CAAC,QAAQ,WAAW;GACvB,YAAY,KAAK,WAAW,OAAO,OAAO,UAAU;GACpD;EACD,GAAE,CACF,KAAK,CAAC,GAAG,MAAO,EAAE,aAAa,EAAE,aAAa,KAAK,EAAG,CACtD,MAAM,GAAG,EAAE;EAEd,MAAMC,SAA+B,SAAS,IAAI,CAAC,WAAW,CAC5D,IAAID,0BAAS;GACX,UAAU,sBAAsB,OAAO,OAAO;GAC9C,aAAa,sBAAsB,OAAO,OAAO;EAClD,IACD,OAAO,UACR,EAAC;AAEF,SAAO;CACR;;;;;;;;;;;CAYD,aAAa,UACXE,OACAC,WACAX,YACAY,UAC0B;EAC1B,MAAMC,OAAmB,CAAE;AAC3B,OAAK,IAAI,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK,GAAG;GACxC,MAAM,WAAW,MAAM,QAAQ,UAAU,GAAG,UAAU,KAAK;GAC3D,MAAM,SAAS,IAAIL,0BAAS;IAC1B,aAAa,MAAM;IACnB;GACD;GACD,KAAK,KAAK,OAAO;EAClB;AACD,SAAO,gBAAgB,cAAc,MAAM,YAAY,SAAS;CACjE;;;;;;;;;CAUD,aAAa,cACXK,MACAb,YACAY,UAC0B;EAC1B,MAAM,WAAW,IAAI,KAAK,YAAY;EACtC,MAAM,SAAS,aAAa,KAAK;AACjC,SAAO;CACR;;;;;;;;;CAUD,aAAa,kBACXZ,YACAY,UAC0B;EAC1B,MAAM,WAAW,IAAI,KAAK,YAAY;AACtC,SAAO;CACR;AACF"}
|