@langchain/core 1.0.0-alpha.4 → 1.0.0-alpha.5

package/dist/utils/testing/embeddings.cjs

@@ -0,0 +1,90 @@
+const require_embeddings = require('../../embeddings.cjs');
+
+//#region src/utils/testing/embeddings.ts
+/**
+* A class that provides synthetic embeddings by overriding the
+* embedDocuments and embedQuery methods to generate embeddings based on
+* the input documents. The embeddings are generated by converting each
+* document into chunks, calculating a numerical value for each chunk, and
+* returning an array of these values as the embedding.
+*/
+var SyntheticEmbeddings = class extends require_embeddings.Embeddings {
+	vectorSize;
+	constructor(params) {
+		super(params ?? {});
+		this.vectorSize = params?.vectorSize ?? 4;
+	}
+	/**
+	* Generates synthetic embeddings for a list of documents.
+	* @param documents List of documents to generate embeddings for.
+	* @returns A promise that resolves with a list of synthetic embeddings for each document.
+	*/
+	async embedDocuments(documents) {
+		return Promise.all(documents.map((doc) => this.embedQuery(doc)));
+	}
+	/**
+	* Generates a synthetic embedding for a document. The document is
+	* converted into chunks, a numerical value is calculated for each chunk,
+	* and an array of these values is returned as the embedding.
+	* @param document The document to generate an embedding for.
+	* @returns A promise that resolves with a synthetic embedding for the document.
+	*/
+	async embedQuery(document) {
+		let doc = document;
+		doc = doc.toLowerCase().replaceAll(/[^a-z ]/g, "");
+		const padMod = doc.length % this.vectorSize;
+		const padGapSize = padMod === 0 ? 0 : this.vectorSize - padMod;
+		const padSize = doc.length + padGapSize;
+		doc = doc.padEnd(padSize, " ");
+		const chunkSize = doc.length / this.vectorSize;
+		const docChunk = [];
+		for (let co = 0; co < doc.length; co += chunkSize) docChunk.push(doc.slice(co, co + chunkSize));
+		const ret = docChunk.map((s) => {
+			let sum = 0;
+			for (let co = 0; co < s.length; co += 1) sum += s === " " ? 0 : s.charCodeAt(co);
+			const ret$1 = sum % 26 / 26;
+			return ret$1;
+		});
+		return ret;
+	}
+};
+/**
+* A class that provides fake embeddings by overriding the embedDocuments
+* and embedQuery methods to return fixed values.
+*/
+var FakeEmbeddings = class extends require_embeddings.Embeddings {
+	constructor(params) {
+		super(params ?? {});
+	}
+	/**
+	* Generates fixed embeddings for a list of documents.
+	* @param documents List of documents to generate embeddings for.
+	* @returns A promise that resolves with a list of fixed embeddings for each document.
+	*/
+	embedDocuments(documents) {
+		return Promise.resolve(documents.map(() => [
+			.1,
+			.2,
+			.3,
+			.4
+		]));
+	}
+	/**
+	* Generates a fixed embedding for a query.
+	* @param _ The query to generate an embedding for.
+	* @returns A promise that resolves with a fixed embedding for the query.
+	*/
+	embedQuery(_) {
+		return Promise.resolve([
+			.1,
+			.2,
+			.3,
+			.4
+		]);
+	}
+};
+
+//#endregion
+exports.FakeEmbeddings = FakeEmbeddings;
+exports.SyntheticEmbeddings = SyntheticEmbeddings;
+//# sourceMappingURL=embeddings.cjs.map

package/dist/utils/testing/embeddings.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.cjs","names":["Embeddings","params?: SyntheticEmbeddingsParams","documents: string[]","document: string","ret: number[]","ret","params?: EmbeddingsParams","_: string"],"sources":["../../../src/utils/testing/embeddings.ts"],"sourcesContent":["import { Embeddings, EmbeddingsParams } from \"../../embeddings.js\";\n\n/**\n * An interface that defines additional parameters specific to the\n * SyntheticEmbeddings class.\n */\ninterface SyntheticEmbeddingsParams extends EmbeddingsParams {\n  vectorSize: number;\n}\n\n/**\n * A class that provides synthetic embeddings by overriding the\n * embedDocuments and embedQuery methods to generate embeddings based on\n * the input documents. The embeddings are generated by converting each\n * document into chunks, calculating a numerical value for each chunk, and\n * returning an array of these values as the embedding.\n */\nexport class SyntheticEmbeddings\n  extends Embeddings\n  implements SyntheticEmbeddingsParams\n{\n  vectorSize: number;\n\n  constructor(params?: SyntheticEmbeddingsParams) {\n    super(params ?? {});\n    this.vectorSize = params?.vectorSize ?? 4;\n  }\n\n  /**\n   * Generates synthetic embeddings for a list of documents.\n   * @param documents List of documents to generate embeddings for.\n   * @returns A promise that resolves with a list of synthetic embeddings for each document.\n   */\n  async embedDocuments(documents: string[]): Promise<number[][]> {\n    return Promise.all(documents.map((doc) => this.embedQuery(doc)));\n  }\n\n  /**\n   * Generates a synthetic embedding for a document. The document is\n   * converted into chunks, a numerical value is calculated for each chunk,\n   * and an array of these values is returned as the embedding.\n   * @param document The document to generate an embedding for.\n   * @returns A promise that resolves with a synthetic embedding for the document.\n   */\n  async embedQuery(document: string): Promise<number[]> {\n    let doc = document;\n\n    // Only use the letters (and space) from the document, and make them lower case\n    doc = doc.toLowerCase().replaceAll(/[^a-z ]/g, \"\");\n\n    // Pad the document to make sure it has a divisible number of chunks\n    const padMod = doc.length % this.vectorSize;\n    const padGapSize = padMod === 0 ? 0 : this.vectorSize - padMod;\n    const padSize = doc.length + padGapSize;\n    doc = doc.padEnd(padSize, \" \");\n\n    // Break it into chunks\n    const chunkSize = doc.length / this.vectorSize;\n    const docChunk = [];\n    for (let co = 0; co < doc.length; co += chunkSize) {\n      docChunk.push(doc.slice(co, co + chunkSize));\n    }\n\n    // Turn each chunk into a number\n    const ret: number[] = docChunk.map((s) => {\n      let sum = 0;\n      // Get a total value by adding the value of each character in the string\n      for (let co = 0; co < s.length; co += 1) {\n        sum += s === \" \" ? 0 : s.charCodeAt(co);\n      }\n      // Reduce this to a number between 0 and 25 inclusive\n      // Then get the fractional number by dividing it by 26\n      const ret = (sum % 26) / 26;\n      return ret;\n    });\n\n    return ret;\n  }\n}\n\n/**\n * A class that provides fake embeddings by overriding the embedDocuments\n * and embedQuery methods to return fixed values.\n */\nexport class FakeEmbeddings extends Embeddings {\n  constructor(params?: EmbeddingsParams) {\n    super(params ?? {});\n  }\n\n  /**\n   * Generates fixed embeddings for a list of documents.\n   * @param documents List of documents to generate embeddings for.\n   * @returns A promise that resolves with a list of fixed embeddings for each document.\n   */\n  embedDocuments(documents: string[]): Promise<number[][]> {\n    return Promise.resolve(documents.map(() => [0.1, 0.2, 0.3, 0.4]));\n  }\n\n  /**\n   * Generates a fixed embedding for a query.\n   * @param _ The query to generate an embedding for.\n   * @returns A promise that resolves with a fixed embedding for the query.\n   */\n  embedQuery(_: string): Promise<number[]> {\n    return Promise.resolve([0.1, 0.2, 0.3, 0.4]);\n  }\n}\n"],"mappings":";;;;;;;;;;AAiBA,IAAa,sBAAb,cACUA,8BAEV;CACE;CAEA,YAAYC,QAAoC;EAC9C,MAAM,UAAU,CAAE,EAAC;EACnB,KAAK,aAAa,QAAQ,cAAc;CACzC;;;;;;CAOD,MAAM,eAAeC,WAA0C;AAC7D,SAAO,QAAQ,IAAI,UAAU,IAAI,CAAC,QAAQ,KAAK,WAAW,IAAI,CAAC,CAAC;CACjE;;;;;;;;CASD,MAAM,WAAWC,UAAqC;EACpD,IAAI,MAAM;EAGV,MAAM,IAAI,aAAa,CAAC,WAAW,YAAY,GAAG;EAGlD,MAAM,SAAS,IAAI,SAAS,KAAK;EACjC,MAAM,aAAa,WAAW,IAAI,IAAI,KAAK,aAAa;EACxD,MAAM,UAAU,IAAI,SAAS;EAC7B,MAAM,IAAI,OAAO,SAAS,IAAI;EAG9B,MAAM,YAAY,IAAI,SAAS,KAAK;EACpC,MAAM,WAAW,CAAE;AACnB,OAAK,IAAI,KAAK,GAAG,KAAK,IAAI,QAAQ,MAAM,WACtC,SAAS,KAAK,IAAI,MAAM,IAAI,KAAK,UAAU,CAAC;EAI9C,MAAMC,MAAgB,SAAS,IAAI,CAAC,MAAM;GACxC,IAAI,MAAM;AAEV,QAAK,IAAI,KAAK,GAAG,KAAK,EAAE,QAAQ,MAAM,GACpC,OAAO,MAAM,MAAM,IAAI,EAAE,WAAW,GAAG;GAIzC,MAAMC,QAAO,MAAM,KAAM;AACzB,UAAOA;EACR,EAAC;AAEF,SAAO;CACR;AACF;;;;;AAMD,IAAa,iBAAb,cAAoCL,8BAAW;CAC7C,YAAYM,QAA2B;EACrC,MAAM,UAAU,CAAE,EAAC;CACpB;;;;;;CAOD,eAAeJ,WAA0C;AACvD,SAAO,QAAQ,QAAQ,UAAU,IAAI,MAAM;GAAC;GAAK;GAAK;GAAK;EAAI,EAAC,CAAC;CAClE;;;;;;CAOD,WAAWK,GAA8B;AACvC,SAAO,QAAQ,QAAQ;GAAC;GAAK;GAAK;GAAK;EAAI,EAAC;CAC7C;AACF"}

package/dist/utils/testing/embeddings.d.cts

@@ -0,0 +1,58 @@
+import { Embeddings, EmbeddingsParams } from "../../embeddings.cjs";
+
+//#region src/utils/testing/embeddings.d.ts
+
+/**
+ * An interface that defines additional parameters specific to the
+ * SyntheticEmbeddings class.
+ */
+interface SyntheticEmbeddingsParams extends EmbeddingsParams {
+  vectorSize: number;
+}
+/**
+ * A class that provides synthetic embeddings by overriding the
+ * embedDocuments and embedQuery methods to generate embeddings based on
+ * the input documents. The embeddings are generated by converting each
+ * document into chunks, calculating a numerical value for each chunk, and
+ * returning an array of these values as the embedding.
+ */
+declare class SyntheticEmbeddings extends Embeddings implements SyntheticEmbeddingsParams {
+  vectorSize: number;
+  constructor(params?: SyntheticEmbeddingsParams);
+  /**
+   * Generates synthetic embeddings for a list of documents.
+   * @param documents List of documents to generate embeddings for.
+   * @returns A promise that resolves with a list of synthetic embeddings for each document.
+   */
+  embedDocuments(documents: string[]): Promise<number[][]>;
+  /**
+   * Generates a synthetic embedding for a document. The document is
+   * converted into chunks, a numerical value is calculated for each chunk,
+   * and an array of these values is returned as the embedding.
+   * @param document The document to generate an embedding for.
+   * @returns A promise that resolves with a synthetic embedding for the document.
+   */
+  embedQuery(document: string): Promise<number[]>;
+}
+/**
+ * A class that provides fake embeddings by overriding the embedDocuments
+ * and embedQuery methods to return fixed values.
+ */
+declare class FakeEmbeddings extends Embeddings {
+  constructor(params?: EmbeddingsParams);
+  /**
+   * Generates fixed embeddings for a list of documents.
+   * @param documents List of documents to generate embeddings for.
+   * @returns A promise that resolves with a list of fixed embeddings for each document.
+   */
+  embedDocuments(documents: string[]): Promise<number[][]>;
+  /**
+   * Generates a fixed embedding for a query.
+   * @param _ The query to generate an embedding for.
+   * @returns A promise that resolves with a fixed embedding for the query.
+   */
+  embedQuery(_: string): Promise<number[]>;
+}
+//#endregion
+export { FakeEmbeddings, SyntheticEmbeddings };
+//# sourceMappingURL=embeddings.d.cts.map

package/dist/utils/testing/embeddings.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.d.cts","names":["Embeddings","EmbeddingsParams","SyntheticEmbeddingsParams","SyntheticEmbeddings","Promise","FakeEmbeddings"],"sources":["../../../src/utils/testing/embeddings.d.ts"],"sourcesContent":["import { Embeddings, EmbeddingsParams } from \"../../embeddings.js\";\n/**\n * An interface that defines additional parameters specific to the\n * SyntheticEmbeddings class.\n */\ninterface SyntheticEmbeddingsParams extends EmbeddingsParams {\n    vectorSize: number;\n}\n/**\n * A class that provides synthetic embeddings by overriding the\n * embedDocuments and embedQuery methods to generate embeddings based on\n * the input documents. The embeddings are generated by converting each\n * document into chunks, calculating a numerical value for each chunk, and\n * returning an array of these values as the embedding.\n */\nexport declare class SyntheticEmbeddings extends Embeddings implements SyntheticEmbeddingsParams {\n    vectorSize: number;\n    constructor(params?: SyntheticEmbeddingsParams);\n    /**\n     * Generates synthetic embeddings for a list of documents.\n     * @param documents List of documents to generate embeddings for.\n     * @returns A promise that resolves with a list of synthetic embeddings for each document.\n     */\n    embedDocuments(documents: string[]): Promise<number[][]>;\n    /**\n     * Generates a synthetic embedding for a document. The document is\n     * converted into chunks, a numerical value is calculated for each chunk,\n     * and an array of these values is returned as the embedding.\n     * @param document The document to generate an embedding for.\n     * @returns A promise that resolves with a synthetic embedding for the document.\n     */\n    embedQuery(document: string): Promise<number[]>;\n}\n/**\n * A class that provides fake embeddings by overriding the embedDocuments\n * and embedQuery methods to return fixed values.\n */\nexport declare class FakeEmbeddings extends Embeddings {\n    constructor(params?: EmbeddingsParams);\n    /**\n     * Generates fixed embeddings for a list of documents.\n     * @param documents List of documents to generate embeddings for.\n     * @returns A promise that resolves with a list of fixed embeddings for each document.\n     */\n    embedDocuments(documents: string[]): Promise<number[][]>;\n    /**\n     * Generates a fixed embedding for a query.\n     * @param _ The query to generate an embedding for.\n     * @returns A promise that resolves with a fixed embedding for the query.\n     */\n    embedQuery(_: string): Promise<number[]>;\n}\nexport {};\n"],"mappings":";;;;;;AAAmE;AAenE;UAVUE,yBAAAA,SAAkCD,gBAUJ,CAAA;EAAA,UAEfC,EAAAA,MAAAA;;;;;AAFuE;AAsBhG;;;AAOyCE,cA7BpBD,mBAAAA,SAA4BH,UAAAA,YAAsBE,yBA6B9BE,CAAAA;EAAO,UAMrBA,EAAAA,MAAAA;EAAO,WAbUJ,CAAAA,MAAAA,CAAAA,EApBnBE,yBAoBmBF;EAAU;;;;;uCAdbI;;;;;;;;gCAQPA;;;;;;cAMbC,cAAAA,SAAuBL,UAAAA;uBACnBC;;;;;;uCAMgBG;;;;;;yBAMdA"}

package/dist/utils/testing/embeddings.d.ts

@@ -0,0 +1,58 @@
+import { Embeddings, EmbeddingsParams } from "../../embeddings.js";
+
+//#region src/utils/testing/embeddings.d.ts
+
+/**
+ * An interface that defines additional parameters specific to the
+ * SyntheticEmbeddings class.
+ */
+interface SyntheticEmbeddingsParams extends EmbeddingsParams {
+  vectorSize: number;
+}
+/**
+ * A class that provides synthetic embeddings by overriding the
+ * embedDocuments and embedQuery methods to generate embeddings based on
+ * the input documents. The embeddings are generated by converting each
+ * document into chunks, calculating a numerical value for each chunk, and
+ * returning an array of these values as the embedding.
+ */
+declare class SyntheticEmbeddings extends Embeddings implements SyntheticEmbeddingsParams {
+  vectorSize: number;
+  constructor(params?: SyntheticEmbeddingsParams);
+  /**
+   * Generates synthetic embeddings for a list of documents.
+   * @param documents List of documents to generate embeddings for.
+   * @returns A promise that resolves with a list of synthetic embeddings for each document.
+   */
+  embedDocuments(documents: string[]): Promise<number[][]>;
+  /**
+   * Generates a synthetic embedding for a document. The document is
+   * converted into chunks, a numerical value is calculated for each chunk,
+   * and an array of these values is returned as the embedding.
+   * @param document The document to generate an embedding for.
+   * @returns A promise that resolves with a synthetic embedding for the document.
+   */
+  embedQuery(document: string): Promise<number[]>;
+}
+/**
+ * A class that provides fake embeddings by overriding the embedDocuments
+ * and embedQuery methods to return fixed values.
+ */
+declare class FakeEmbeddings extends Embeddings {
+  constructor(params?: EmbeddingsParams);
+  /**
+   * Generates fixed embeddings for a list of documents.
+   * @param documents List of documents to generate embeddings for.
+   * @returns A promise that resolves with a list of fixed embeddings for each document.
+   */
+  embedDocuments(documents: string[]): Promise<number[][]>;
+  /**
+   * Generates a fixed embedding for a query.
+   * @param _ The query to generate an embedding for.
+   * @returns A promise that resolves with a fixed embedding for the query.
+   */
+  embedQuery(_: string): Promise<number[]>;
+}
+//#endregion
+export { FakeEmbeddings, SyntheticEmbeddings };
+//# sourceMappingURL=embeddings.d.ts.map

package/dist/utils/testing/embeddings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.d.ts","names":["Embeddings","EmbeddingsParams","SyntheticEmbeddingsParams","SyntheticEmbeddings","Promise","FakeEmbeddings"],"sources":["../../../src/utils/testing/embeddings.d.ts"],"sourcesContent":["import { Embeddings, EmbeddingsParams } from \"../../embeddings.js\";\n/**\n * An interface that defines additional parameters specific to the\n * SyntheticEmbeddings class.\n */\ninterface SyntheticEmbeddingsParams extends EmbeddingsParams {\n    vectorSize: number;\n}\n/**\n * A class that provides synthetic embeddings by overriding the\n * embedDocuments and embedQuery methods to generate embeddings based on\n * the input documents. The embeddings are generated by converting each\n * document into chunks, calculating a numerical value for each chunk, and\n * returning an array of these values as the embedding.\n */\nexport declare class SyntheticEmbeddings extends Embeddings implements SyntheticEmbeddingsParams {\n    vectorSize: number;\n    constructor(params?: SyntheticEmbeddingsParams);\n    /**\n     * Generates synthetic embeddings for a list of documents.\n     * @param documents List of documents to generate embeddings for.\n     * @returns A promise that resolves with a list of synthetic embeddings for each document.\n     */\n    embedDocuments(documents: string[]): Promise<number[][]>;\n    /**\n     * Generates a synthetic embedding for a document. The document is\n     * converted into chunks, a numerical value is calculated for each chunk,\n     * and an array of these values is returned as the embedding.\n     * @param document The document to generate an embedding for.\n     * @returns A promise that resolves with a synthetic embedding for the document.\n     */\n    embedQuery(document: string): Promise<number[]>;\n}\n/**\n * A class that provides fake embeddings by overriding the embedDocuments\n * and embedQuery methods to return fixed values.\n */\nexport declare class FakeEmbeddings extends Embeddings {\n    constructor(params?: EmbeddingsParams);\n    /**\n     * Generates fixed embeddings for a list of documents.\n     * @param documents List of documents to generate embeddings for.\n     * @returns A promise that resolves with a list of fixed embeddings for each document.\n     */\n    embedDocuments(documents: string[]): Promise<number[][]>;\n    /**\n     * Generates a fixed embedding for a query.\n     * @param _ The query to generate an embedding for.\n     * @returns A promise that resolves with a fixed embedding for the query.\n     */\n    embedQuery(_: string): Promise<number[]>;\n}\nexport {};\n"],"mappings":";;;;;;AAAmE;AAenE;UAVUE,yBAAAA,SAAkCD,gBAUJ,CAAA;EAAA,UAEfC,EAAAA,MAAAA;;;;;AAFuE;AAsBhG;;;AAOyCE,cA7BpBD,mBAAAA,SAA4BH,UAAAA,YAAsBE,yBA6B9BE,CAAAA;EAAO,UAMrBA,EAAAA,MAAAA;EAAO,WAbUJ,CAAAA,MAAAA,CAAAA,EApBnBE,yBAoBmBF;EAAU;;;;;uCAdbI;;;;;;;;gCAQPA;;;;;;cAMbC,cAAAA,SAAuBL,UAAAA;uBACnBC;;;;;;uCAMgBG;;;;;;yBAMdA"}

package/dist/utils/testing/embeddings.js

@@ -0,0 +1,89 @@
+import { Embeddings } from "../../embeddings.js";
+
+//#region src/utils/testing/embeddings.ts
+/**
+* A class that provides synthetic embeddings by overriding the
+* embedDocuments and embedQuery methods to generate embeddings based on
+* the input documents. The embeddings are generated by converting each
+* document into chunks, calculating a numerical value for each chunk, and
+* returning an array of these values as the embedding.
+*/
+var SyntheticEmbeddings = class extends Embeddings {
+	vectorSize;
+	constructor(params) {
+		super(params ?? {});
+		this.vectorSize = params?.vectorSize ?? 4;
+	}
+	/**
+	* Generates synthetic embeddings for a list of documents.
+	* @param documents List of documents to generate embeddings for.
+	* @returns A promise that resolves with a list of synthetic embeddings for each document.
+	*/
+	async embedDocuments(documents) {
+		return Promise.all(documents.map((doc) => this.embedQuery(doc)));
+	}
+	/**
+	* Generates a synthetic embedding for a document. The document is
+	* converted into chunks, a numerical value is calculated for each chunk,
+	* and an array of these values is returned as the embedding.
+	* @param document The document to generate an embedding for.
+	* @returns A promise that resolves with a synthetic embedding for the document.
+	*/
+	async embedQuery(document) {
+		let doc = document;
+		doc = doc.toLowerCase().replaceAll(/[^a-z ]/g, "");
+		const padMod = doc.length % this.vectorSize;
+		const padGapSize = padMod === 0 ? 0 : this.vectorSize - padMod;
+		const padSize = doc.length + padGapSize;
+		doc = doc.padEnd(padSize, " ");
+		const chunkSize = doc.length / this.vectorSize;
+		const docChunk = [];
+		for (let co = 0; co < doc.length; co += chunkSize) docChunk.push(doc.slice(co, co + chunkSize));
+		const ret = docChunk.map((s) => {
+			let sum = 0;
+			for (let co = 0; co < s.length; co += 1) sum += s === " " ? 0 : s.charCodeAt(co);
+			const ret$1 = sum % 26 / 26;
+			return ret$1;
+		});
+		return ret;
+	}
+};
+/**
+* A class that provides fake embeddings by overriding the embedDocuments
+* and embedQuery methods to return fixed values.
+*/
+var FakeEmbeddings = class extends Embeddings {
+	constructor(params) {
+		super(params ?? {});
+	}
+	/**
+	* Generates fixed embeddings for a list of documents.
+	* @param documents List of documents to generate embeddings for.
+	* @returns A promise that resolves with a list of fixed embeddings for each document.
+	*/
+	embedDocuments(documents) {
+		return Promise.resolve(documents.map(() => [
+			.1,
+			.2,
+			.3,
+			.4
+		]));
+	}
+	/**
+	* Generates a fixed embedding for a query.
+	* @param _ The query to generate an embedding for.
+	* @returns A promise that resolves with a fixed embedding for the query.
+	*/
+	embedQuery(_) {
+		return Promise.resolve([
+			.1,
+			.2,
+			.3,
+			.4
+		]);
+	}
+};
+
+//#endregion
+export { FakeEmbeddings, SyntheticEmbeddings };
+//# sourceMappingURL=embeddings.js.map

package/dist/utils/testing/embeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.js","names":["params?: SyntheticEmbeddingsParams","documents: string[]","document: string","ret: number[]","ret","params?: EmbeddingsParams","_: string"],"sources":["../../../src/utils/testing/embeddings.ts"],"sourcesContent":["import { Embeddings, EmbeddingsParams } from \"../../embeddings.js\";\n\n/**\n * An interface that defines additional parameters specific to the\n * SyntheticEmbeddings class.\n */\ninterface SyntheticEmbeddingsParams extends EmbeddingsParams {\n  vectorSize: number;\n}\n\n/**\n * A class that provides synthetic embeddings by overriding the\n * embedDocuments and embedQuery methods to generate embeddings based on\n * the input documents. The embeddings are generated by converting each\n * document into chunks, calculating a numerical value for each chunk, and\n * returning an array of these values as the embedding.\n */\nexport class SyntheticEmbeddings\n  extends Embeddings\n  implements SyntheticEmbeddingsParams\n{\n  vectorSize: number;\n\n  constructor(params?: SyntheticEmbeddingsParams) {\n    super(params ?? {});\n    this.vectorSize = params?.vectorSize ?? 4;\n  }\n\n  /**\n   * Generates synthetic embeddings for a list of documents.\n   * @param documents List of documents to generate embeddings for.\n   * @returns A promise that resolves with a list of synthetic embeddings for each document.\n   */\n  async embedDocuments(documents: string[]): Promise<number[][]> {\n    return Promise.all(documents.map((doc) => this.embedQuery(doc)));\n  }\n\n  /**\n   * Generates a synthetic embedding for a document. The document is\n   * converted into chunks, a numerical value is calculated for each chunk,\n   * and an array of these values is returned as the embedding.\n   * @param document The document to generate an embedding for.\n   * @returns A promise that resolves with a synthetic embedding for the document.\n   */\n  async embedQuery(document: string): Promise<number[]> {\n    let doc = document;\n\n    // Only use the letters (and space) from the document, and make them lower case\n    doc = doc.toLowerCase().replaceAll(/[^a-z ]/g, \"\");\n\n    // Pad the document to make sure it has a divisible number of chunks\n    const padMod = doc.length % this.vectorSize;\n    const padGapSize = padMod === 0 ? 0 : this.vectorSize - padMod;\n    const padSize = doc.length + padGapSize;\n    doc = doc.padEnd(padSize, \" \");\n\n    // Break it into chunks\n    const chunkSize = doc.length / this.vectorSize;\n    const docChunk = [];\n    for (let co = 0; co < doc.length; co += chunkSize) {\n      docChunk.push(doc.slice(co, co + chunkSize));\n    }\n\n    // Turn each chunk into a number\n    const ret: number[] = docChunk.map((s) => {\n      let sum = 0;\n      // Get a total value by adding the value of each character in the string\n      for (let co = 0; co < s.length; co += 1) {\n        sum += s === \" \" ? 0 : s.charCodeAt(co);\n      }\n      // Reduce this to a number between 0 and 25 inclusive\n      // Then get the fractional number by dividing it by 26\n      const ret = (sum % 26) / 26;\n      return ret;\n    });\n\n    return ret;\n  }\n}\n\n/**\n * A class that provides fake embeddings by overriding the embedDocuments\n * and embedQuery methods to return fixed values.\n */\nexport class FakeEmbeddings extends Embeddings {\n  constructor(params?: EmbeddingsParams) {\n    super(params ?? {});\n  }\n\n  /**\n   * Generates fixed embeddings for a list of documents.\n   * @param documents List of documents to generate embeddings for.\n   * @returns A promise that resolves with a list of fixed embeddings for each document.\n   */\n  embedDocuments(documents: string[]): Promise<number[][]> {\n    return Promise.resolve(documents.map(() => [0.1, 0.2, 0.3, 0.4]));\n  }\n\n  /**\n   * Generates a fixed embedding for a query.\n   * @param _ The query to generate an embedding for.\n   * @returns A promise that resolves with a fixed embedding for the query.\n   */\n  embedQuery(_: string): Promise<number[]> {\n    return Promise.resolve([0.1, 0.2, 0.3, 0.4]);\n  }\n}\n"],"mappings":";;;;;;;;;;AAiBA,IAAa,sBAAb,cACU,WAEV;CACE;CAEA,YAAYA,QAAoC;EAC9C,MAAM,UAAU,CAAE,EAAC;EACnB,KAAK,aAAa,QAAQ,cAAc;CACzC;;;;;;CAOD,MAAM,eAAeC,WAA0C;AAC7D,SAAO,QAAQ,IAAI,UAAU,IAAI,CAAC,QAAQ,KAAK,WAAW,IAAI,CAAC,CAAC;CACjE;;;;;;;;CASD,MAAM,WAAWC,UAAqC;EACpD,IAAI,MAAM;EAGV,MAAM,IAAI,aAAa,CAAC,WAAW,YAAY,GAAG;EAGlD,MAAM,SAAS,IAAI,SAAS,KAAK;EACjC,MAAM,aAAa,WAAW,IAAI,IAAI,KAAK,aAAa;EACxD,MAAM,UAAU,IAAI,SAAS;EAC7B,MAAM,IAAI,OAAO,SAAS,IAAI;EAG9B,MAAM,YAAY,IAAI,SAAS,KAAK;EACpC,MAAM,WAAW,CAAE;AACnB,OAAK,IAAI,KAAK,GAAG,KAAK,IAAI,QAAQ,MAAM,WACtC,SAAS,KAAK,IAAI,MAAM,IAAI,KAAK,UAAU,CAAC;EAI9C,MAAMC,MAAgB,SAAS,IAAI,CAAC,MAAM;GACxC,IAAI,MAAM;AAEV,QAAK,IAAI,KAAK,GAAG,KAAK,EAAE,QAAQ,MAAM,GACpC,OAAO,MAAM,MAAM,IAAI,EAAE,WAAW,GAAG;GAIzC,MAAMC,QAAO,MAAM,KAAM;AACzB,UAAOA;EACR,EAAC;AAEF,SAAO;CACR;AACF;;;;;AAMD,IAAa,iBAAb,cAAoC,WAAW;CAC7C,YAAYC,QAA2B;EACrC,MAAM,UAAU,CAAE,EAAC;CACpB;;;;;;CAOD,eAAeJ,WAA0C;AACvD,SAAO,QAAQ,QAAQ,UAAU,IAAI,MAAM;GAAC;GAAK;GAAK;GAAK;EAAI,EAAC,CAAC;CAClE;;;;;;CAOD,WAAWK,GAA8B;AACvC,SAAO,QAAQ,QAAQ;GAAC;GAAK;GAAK;GAAK;EAAI,EAAC;CAC7C;AACF"}