@langchain/core 0.1.14 → 0.1.16

package/dist/utils/testing/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.FakeTool = exports.FakeTracer = exports.FakeListChatMessageHistory = exports.FakeChatMessageHistory = exports.FakeListChatModel = exports.FakeRetriever = exports.FakeChatModel = exports.FakeStreamingLLM = exports.FakeLLM = exports.FakeRunnable = exports.FakeSplitIntoListParser = void 0;
+exports.SyntheticEmbeddings = exports.FakeEmbeddings = exports.FakeTool = exports.FakeTracer = exports.FakeListChatMessageHistory = exports.FakeChatMessageHistory = exports.FakeListChatModel = exports.FakeRetriever = exports.FakeChatModel = exports.FakeStreamingLLM = exports.FakeLLM = exports.FakeRunnable = exports.FakeSplitIntoListParser = void 0;
 const chat_history_js_1 = require("../../chat_history.cjs");
 const document_js_1 = require("../../documents/document.cjs");
 const chat_models_js_1 = require("../../language_models/chat_models.cjs");
@@ -12,6 +12,7 @@ const retrievers_js_1 = require("../../retrievers.cjs");
 const base_js_2 = require("../../runnables/base.cjs");
 const tools_js_1 = require("../../tools.cjs");
 const base_js_3 = require("../../tracers/base.cjs");
+const embeddings_js_1 = require("../../embeddings.cjs");
 /**
  * Parser for comma-separated values. It splits the input text by commas
  * and trims the resulting values.
@@ -99,16 +100,27 @@ class FakeStreamingLLM extends llms_js_1.LLM {
             writable: true,
             value: 50
         });
+        Object.defineProperty(this, "responses", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.sleep = fields.sleep ?? this.sleep;
+        this.responses = fields.responses;
     }
     _llmType() {
         return "fake";
     }
     async _call(prompt) {
-        return prompt;
+        const response = this.responses?.[0];
+        this.responses = this.responses?.slice(1);
+        return response ?? prompt;
     }
     async *_streamResponseChunks(input) {
-        for (const c of input) {
+        const response = this.responses?.[0];
+        this.responses = this.responses?.slice(1);
+        for (const c of response ?? input) {
             await new Promise((resolve) => setTimeout(resolve, this.sleep));
             yield { text: c, generationInfo: {} };
         }
@@ -395,3 +407,93 @@ class FakeTool extends tools_js_1.StructuredTool {
     }
 }
 exports.FakeTool = FakeTool;
+/**
+ * A class that provides fake embeddings by overriding the embedDocuments
+ * and embedQuery methods to return fixed values.
+ */
+class FakeEmbeddings extends embeddings_js_1.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(() => [0.1, 0.2, 0.3, 0.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([0.1, 0.2, 0.3, 0.4]);
+    }
+}
+exports.FakeEmbeddings = FakeEmbeddings;
+/**
+ * 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.
+ */
+class SyntheticEmbeddings extends embeddings_js_1.Embeddings {
+    constructor(params) {
+        super(params ?? {});
+        Object.defineProperty(this, "vectorSize", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        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;
+        // Only use the letters (and space) from the document, and make them lower case
+        doc = doc.toLowerCase().replaceAll(/[^a-z ]/g, "");
+        // Pad the document to make sure it has a divisible number of chunks
+        const padMod = doc.length % this.vectorSize;
+        const padGapSize = padMod === 0 ? 0 : this.vectorSize - padMod;
+        const padSize = doc.length + padGapSize;
+        doc = doc.padEnd(padSize, " ");
+        // Break it into chunks
+        const chunkSize = doc.length / this.vectorSize;
+        const docChunk = [];
+        for (let co = 0; co < doc.length; co += chunkSize) {
+            docChunk.push(doc.slice(co, co + chunkSize));
+        }
+        // Turn each chunk into a number
+        const ret = docChunk.map((s) => {
+            let sum = 0;
+            // Get a total value by adding the value of each character in the string
+            for (let co = 0; co < s.length; co += 1) {
+                sum += s === " " ? 0 : s.charCodeAt(co);
+            }
+            // Reduce this to a number between 0 and 25 inclusive
+            // Then get the fractional number by dividing it by 26
+            const ret = (sum % 26) / 26;
+            return ret;
+        });
+        return ret;
+    }
+}
+exports.SyntheticEmbeddings = SyntheticEmbeddings;

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

@@ -11,6 +11,7 @@ import { BaseRetriever } from "../../retrievers.js";
 import { Runnable } from "../../runnables/base.js";
 import { StructuredTool, ToolParams } from "../../tools.js";
 import { BaseTracer, Run } from "../../tracers/base.js";
+import { Embeddings, EmbeddingsParams } from "../../embeddings.js";
 /**
  * Parser for comma-separated values. It splits the input text by commas
  * and trims the resulting values.
@@ -40,8 +41,10 @@ export declare class FakeLLM extends LLM {
 }
 export declare class FakeStreamingLLM extends LLM {
     sleep?: number;
+    responses?: string[];
     constructor(fields: {
         sleep?: number;
+        responses?: string[];
     } & BaseLLMParams);
     _llmType(): string;
     _call(prompt: string): Promise<string>;
@@ -144,3 +147,55 @@ export declare class FakeTool<T extends z.ZodObject<any, any, any, any> = z.ZodO
     constructor(fields: FakeToolParams<T>);
     protected _call(arg: z.output<T>, _runManager?: CallbackManagerForToolRun): Promise<string>;
 }
+/**
+ * A class that provides fake embeddings by overriding the embedDocuments
+ * and embedQuery methods to return fixed values.
+ */
+export 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[]>;
+}
+/**
+ * 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.
+ */
+export 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[]>;
+}
+export {};

package/dist/utils/testing/index.js

@@ -9,6 +9,7 @@ import { BaseRetriever } from "../../retrievers.js";
 import { Runnable } from "../../runnables/base.js";
 import { StructuredTool } from "../../tools.js";
 import { BaseTracer } from "../../tracers/base.js";
+import { Embeddings } from "../../embeddings.js";
 /**
  * Parser for comma-separated values. It splits the input text by commas
  * and trims the resulting values.
@@ -93,16 +94,27 @@ export class FakeStreamingLLM extends LLM {
             writable: true,
             value: 50
         });
+        Object.defineProperty(this, "responses", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.sleep = fields.sleep ?? this.sleep;
+        this.responses = fields.responses;
     }
     _llmType() {
         return "fake";
     }
     async _call(prompt) {
-        return prompt;
+        const response = this.responses?.[0];
+        this.responses = this.responses?.slice(1);
+        return response ?? prompt;
     }
     async *_streamResponseChunks(input) {
-        for (const c of input) {
+        const response = this.responses?.[0];
+        this.responses = this.responses?.slice(1);
+        for (const c of response ?? input) {
             await new Promise((resolve) => setTimeout(resolve, this.sleep));
             yield { text: c, generationInfo: {} };
         }
@@ -381,3 +393,91 @@ export class FakeTool extends StructuredTool {
         return JSON.stringify(arg);
     }
 }
+/**
+ * A class that provides fake embeddings by overriding the embedDocuments
+ * and embedQuery methods to return fixed values.
+ */
+export class FakeEmbeddings 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(() => [0.1, 0.2, 0.3, 0.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([0.1, 0.2, 0.3, 0.4]);
+    }
+}
+/**
+ * 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.
+ */
+export class SyntheticEmbeddings extends Embeddings {
+    constructor(params) {
+        super(params ?? {});
+        Object.defineProperty(this, "vectorSize", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        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;
+        // Only use the letters (and space) from the document, and make them lower case
+        doc = doc.toLowerCase().replaceAll(/[^a-z ]/g, "");
+        // Pad the document to make sure it has a divisible number of chunks
+        const padMod = doc.length % this.vectorSize;
+        const padGapSize = padMod === 0 ? 0 : this.vectorSize - padMod;
+        const padSize = doc.length + padGapSize;
+        doc = doc.padEnd(padSize, " ");
+        // Break it into chunks
+        const chunkSize = doc.length / this.vectorSize;
+        const docChunk = [];
+        for (let co = 0; co < doc.length; co += chunkSize) {
+            docChunk.push(doc.slice(co, co + chunkSize));
+        }
+        // Turn each chunk into a number
+        const ret = docChunk.map((s) => {
+            let sum = 0;
+            // Get a total value by adding the value of each character in the string
+            for (let co = 0; co < s.length; co += 1) {
+                sum += s === " " ? 0 : s.charCodeAt(co);
+            }
+            // Reduce this to a number between 0 and 25 inclusive
+            // Then get the fractional number by dividing it by 26
+            const ret = (sum % 26) / 26;
+            return ret;
+        });
+        return ret;
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/core",
-  "version": "0.1.14",
+  "version": "0.1.16",
   "description": "Core LangChain.js abstractions and schemas",
   "type": "module",
   "engines": {
@@ -18,7 +18,7 @@
     "build:cjs": "NODE_OPTIONS=--max-old-space-size=4096 tsc --outDir dist-cjs/ -p tsconfig.cjs.json && node scripts/move-cjs-to-dist.js && rimraf dist-cjs",
     "build:watch": "node scripts/create-entrypoints.js && tsc --outDir dist/ --watch",
     "build:scripts": "node scripts/create-entrypoints.js && node scripts/check-tree-shaking.js",
-    "lint:eslint": "NODE_OPTIONS=--max-old-space-size=4096 eslint src",
+    "lint:eslint": "NODE_OPTIONS=--max-old-space-size=4096 eslint --cache --ext .ts,.js src/",
     "lint:dpdm": "dpdm --exit-code circular:1 --no-warning --no-tree src/*.ts src/**/*.ts",
     "lint": "yarn lint:eslint && yarn lint:dpdm",
     "lint:fix": "yarn lint:eslint --fix && yarn lint:dpdm",
@@ -43,8 +43,8 @@
     "p-queue": "^6.6.2",
     "p-retry": "4",
     "uuid": "^9.0.0",
-    "zod": "^3.22.3",
-    "zod-to-json-schema": "3.20.3"
+    "zod": "^3.22.4",
+    "zod-to-json-schema": "^3.22.3"
   },
   "devDependencies": {
     "@jest/globals": "^29.5.0",