langchain 0.1.36 → 0.2.0-rc.1

package/dist/prompts/index.js

@@ -1,15 +1,6 @@
-import { logVersion010MigrationWarning } from "../util/entrypoint_deprecation.js";
-/* #__PURE__ */ logVersion010MigrationWarning({
-    oldEntrypointName: "prompts",
-    newEntrypointName: "prompts",
-    newPackageName: "@langchain/core",
-});
-export { BasePromptTemplate, StringPromptValue, BaseStringPromptTemplate, BaseExampleSelector, } from "./base.js";
-export { PromptTemplate } from "./prompt.js";
-export { BasePromptSelector, ConditionalPromptSelector, isChatModel, isLLM, } from "./selectors/conditional.js";
-export { LengthBasedExampleSelector, } from "./selectors/LengthBasedExampleSelector.js";
-export { SemanticSimilarityExampleSelector, } from "./selectors/SemanticSimilarityExampleSelector.js";
-export { FewShotPromptTemplate, FewShotChatMessagePromptTemplate, } from "./few_shot.js";
-export { ChatPromptTemplate, HumanMessagePromptTemplate, AIMessagePromptTemplate, SystemMessagePromptTemplate, ChatMessagePromptTemplate, MessagesPlaceholder, BaseChatPromptTemplate, } from "./chat.js";
-export { parseTemplate, renderTemplate, checkValidTemplate, } from "./template.js";
-export { PipelinePromptTemplate, } from "./pipeline.js";
+/* #__PURE__ */ console.warn([
+    `[ERROR]: The "langchain/prompts" entrypoint was removed in LangChain version 0.2.0.`,
+    ``,
+    `Instead, please import from "@langchain/core/prompts".`,
+].join("\n"));
+export {};

package/dist/retrievers/self_query/chroma.cjs

@@ -2,6 +2,12 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ChromaTranslator = void 0;
 const structured_query_1 = require("@langchain/core/structured_query");
+const entrypoint_deprecation_js_1 = require("../../util/entrypoint_deprecation.cjs");
+/* #__PURE__ */ (0, entrypoint_deprecation_js_1.logVersion020MigrationWarning)({
+    oldEntrypointName: "retrievers/self_query/chroma",
+    newEntrypointName: "structured_query/chroma",
+    newPackageName: "@langchain/community",
+});
 /**
  * Specialized translator for the Chroma vector database. It extends the
  * BasicTranslator class and translates internal query language elements
@@ -23,6 +29,7 @@ const structured_query_1 = require("@langchain/core/structured_query");
  * );
  * ```
  */
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
 class ChromaTranslator extends structured_query_1.BasicTranslator {
     constructor() {
         super({

package/dist/retrievers/self_query/chroma.d.ts

@@ -1,5 +1,5 @@
-import { Chroma } from "@langchain/community/vectorstores/chroma";
 import { BasicTranslator } from "@langchain/core/structured_query";
+import { VectorStore } from "@langchain/core/vectorstores";
 /**
  * Specialized translator for the Chroma vector database. It extends the
  * BasicTranslator class and translates internal query language elements
@@ -21,6 +21,6 @@ import { BasicTranslator } from "@langchain/core/structured_query";
  * );
  * ```
  */
-export declare class ChromaTranslator<T extends Chroma> extends BasicTranslator<T> {
+export declare class ChromaTranslator<T extends VectorStore> extends BasicTranslator<T> {
     constructor();
 }

package/dist/retrievers/self_query/chroma.js

@@ -1,4 +1,10 @@
 import { BasicTranslator, Comparators, Operators, } from "@langchain/core/structured_query";
+import { logVersion020MigrationWarning } from "../../util/entrypoint_deprecation.js";
+/* #__PURE__ */ logVersion020MigrationWarning({
+    oldEntrypointName: "retrievers/self_query/chroma",
+    newEntrypointName: "structured_query/chroma",
+    newPackageName: "@langchain/community",
+});
 /**
  * Specialized translator for the Chroma vector database. It extends the
  * BasicTranslator class and translates internal query language elements
@@ -20,6 +26,7 @@ import { BasicTranslator, Comparators, Operators, } from "@langchain/core/struct
  * );
  * ```
  */
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
 export class ChromaTranslator extends BasicTranslator {
     constructor() {
         super({

package/dist/retrievers/self_query/index.d.ts

@@ -3,7 +3,7 @@ import { BaseRetriever, type BaseRetrieverInput } from "@langchain/core/retrieve
 import { Document } from "@langchain/core/documents";
 import { VectorStore } from "@langchain/core/vectorstores";
 import { BaseTranslator, BasicTranslator, FunctionalTranslator, StructuredQuery } from "@langchain/core/structured_query";
-import { CallbackManagerForRetrieverRun } from "../../callbacks/manager.js";
+import { CallbackManagerForRetrieverRun } from "@langchain/core/callbacks/manager";
 import { QueryConstructorRunnableOptions } from "../../chains/query_constructor/index.js";
 export { BaseTranslator, BasicTranslator, FunctionalTranslator };
 /**

package/dist/retrievers/self_query/pinecone.cjs

@@ -2,6 +2,12 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PineconeTranslator = void 0;
 const structured_query_1 = require("@langchain/core/structured_query");
+const entrypoint_deprecation_js_1 = require("../../util/entrypoint_deprecation.cjs");
+/* #__PURE__ */ (0, entrypoint_deprecation_js_1.logVersion020MigrationWarning)({
+    oldEntrypointName: "retrievers/self_query/pinecone",
+    newEntrypointName: "",
+    newPackageName: "@langchain/pinecone",
+});
 /**
  * Specialized translator class that extends the BasicTranslator. It is
  * designed to work with PineconeStore, a type of vector store in

package/dist/retrievers/self_query/pinecone.js

@@ -1,4 +1,10 @@
 import { BasicTranslator, Comparators, Operators, } from "@langchain/core/structured_query";
+import { logVersion020MigrationWarning } from "../../util/entrypoint_deprecation.js";
+/* #__PURE__ */ logVersion020MigrationWarning({
+    oldEntrypointName: "retrievers/self_query/pinecone",
+    newEntrypointName: "",
+    newPackageName: "@langchain/pinecone",
+});
 /**
  * Specialized translator class that extends the BasicTranslator. It is
  * designed to work with PineconeStore, a type of vector store in

package/dist/retrievers/self_query/supabase.cjs

@@ -1,8 +1,15 @@
 "use strict";
+/* eslint-disable @typescript-eslint/no-explicit-any */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SupabaseTranslator = void 0;
 const structured_query_1 = require("@langchain/core/structured_query");
 const supabase_utils_js_1 = require("./supabase_utils.cjs");
+const entrypoint_deprecation_js_1 = require("../../util/entrypoint_deprecation.cjs");
+/* #__PURE__ */ (0, entrypoint_deprecation_js_1.logVersion020MigrationWarning)({
+    oldEntrypointName: "retrievers/self_query/supabase",
+    newEntrypointName: "structured_query/supabase",
+    newPackageName: "@langchain/community",
+});
 /**
  * A specialized translator designed to work with Supabase, extending the
  * BaseTranslator class. It translates structured queries into a format

package/dist/retrievers/self_query/supabase.d.ts

@@ -1,5 +1,5 @@
-import type { SupabaseFilterRPCCall, SupabaseMetadata, SupabaseVectorStore } from "@langchain/community/vectorstores/supabase";
 import { BaseTranslator, Comparator, Comparison, Operation, Operator, StructuredQuery } from "@langchain/core/structured_query";
+import { VectorStore } from "@langchain/core/vectorstores";
 /**
  * Represents the possible values that can be used in a comparison in a
  * structured query. It can be a string or a number.
@@ -12,6 +12,8 @@ type ValueType = {
     gt: string | number;
     gte: string | number;
 };
+type SupabaseFilterRPCCall = any;
+type SupabaseMetadata = any;
 /**
  * A specialized translator designed to work with Supabase, extending the
  * BaseTranslator class. It translates structured queries into a format
@@ -31,7 +33,7 @@ type ValueType = {
  * );
  * ```
  */
-export declare class SupabaseTranslator<T extends SupabaseVectorStore> extends BaseTranslator<T> {
+export declare class SupabaseTranslator<T extends VectorStore> extends BaseTranslator<T> {
     VisitOperationOutput: SupabaseFilterRPCCall;
     VisitComparisonOutput: SupabaseFilterRPCCall;
     allowedOperators: Operator[];

package/dist/retrievers/self_query/supabase.js

@@ -1,5 +1,12 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
 import { isFilterEmpty, isFloat, isInt, isObject, isString, BaseTranslator, Comparators, Operators, } from "@langchain/core/structured_query";
 import { ProxyParamsDuplicator, convertObjectFilterToStructuredQuery, } from "./supabase_utils.js";
+import { logVersion020MigrationWarning } from "../../util/entrypoint_deprecation.js";
+/* #__PURE__ */ logVersion020MigrationWarning({
+    oldEntrypointName: "retrievers/self_query/supabase",
+    newEntrypointName: "structured_query/supabase",
+    newPackageName: "@langchain/community",
+});
 /**
  * A specialized translator designed to work with Supabase, extending the
  * BaseTranslator class. It translates structured queries into a format

package/dist/retrievers/self_query/supabase_utils.cjs

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.convertObjectFilterToStructuredQuery = exports.ProxyParamsDuplicator = void 0;
+/* eslint-disable @typescript-eslint/no-explicit-any */
 const structured_query_1 = require("@langchain/core/structured_query");
 /**
  * Utility class used to duplicate parameters for a proxy object,
@@ -236,7 +237,11 @@ class ProxyParamsDuplicator {
      * @param filter The `SupabaseFilterRPCCall` object.
      * @returns The flattened parameters as a string.
      */
-    static getFlattenedParams(rpc, filter) {
+    static getFlattenedParams(
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    rpc, 
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    filter) {
         const proxiedDuplicator = new ProxyParamsDuplicator();
         const proxiedRpc = new Proxy(rpc, proxiedDuplicator.buildProxyHandler());
         void filter(proxiedRpc);
@@ -249,7 +254,11 @@ exports.ProxyParamsDuplicator = ProxyParamsDuplicator;
  * The function creates a new `StructuredQuery` object and uses the
  * `Operation` and `Comparison` classes to build the query.
  */
-function convertObjectFilterToStructuredQuery(objFilter) {
-    return new structured_query_1.StructuredQuery("", new structured_query_1.Operation(structured_query_1.Operators.and, Object.entries(objFilter).map(([column, value]) => new structured_query_1.Comparison(structured_query_1.Comparators.eq, column, value))));
+function convertObjectFilterToStructuredQuery(
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+objFilter) {
+    return new structured_query_1.StructuredQuery("", new structured_query_1.Operation(structured_query_1.Operators.and, Object.entries(objFilter).map(
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    ([column, value]) => new structured_query_1.Comparison(structured_query_1.Comparators.eq, column, value))));
 }
 exports.convertObjectFilterToStructuredQuery = convertObjectFilterToStructuredQuery;

package/dist/retrievers/self_query/supabase_utils.d.ts

@@ -1,4 +1,3 @@
-import type { SupabaseFilter, SupabaseFilterRPCCall, SupabaseMetadata } from "@langchain/community/vectorstores/supabase";
 import { StructuredQuery } from "@langchain/core/structured_query";
 /**
  * Utility class used to duplicate parameters for a proxy object,
@@ -16,7 +15,7 @@ export declare class ProxyParamsDuplicator {
      * property being accessed.
      * @returns A proxy handler for a `SupabaseFilter` object.
      */
-    buildProxyHandler(): ProxyHandler<SupabaseFilter>;
+    buildProxyHandler(): ProxyHandler<any>;
     /**
      * Removes type annotations from a value string.
      * @param value The value string to clean.
@@ -92,11 +91,11 @@ export declare class ProxyParamsDuplicator {
      * @param filter The `SupabaseFilterRPCCall` object.
      * @returns The flattened parameters as a string.
      */
-    static getFlattenedParams(rpc: SupabaseFilter, filter: SupabaseFilterRPCCall): string;
+    static getFlattenedParams(rpc: any, filter: any): string;
 }
 /**
  * Converts a `SupabaseMetadata` object into a `StructuredQuery` object.
  * The function creates a new `StructuredQuery` object and uses the
  * `Operation` and `Comparison` classes to build the query.
  */
-export declare function convertObjectFilterToStructuredQuery(objFilter: SupabaseMetadata): StructuredQuery;
+export declare function convertObjectFilterToStructuredQuery(objFilter: any): StructuredQuery;

package/dist/retrievers/self_query/supabase_utils.js

@@ -1,3 +1,4 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
 import { Comparators, Comparison, Operation, Operators, StructuredQuery, } from "@langchain/core/structured_query";
 /**
  * Utility class used to duplicate parameters for a proxy object,
@@ -233,7 +234,11 @@ export class ProxyParamsDuplicator {
      * @param filter The `SupabaseFilterRPCCall` object.
      * @returns The flattened parameters as a string.
      */
-    static getFlattenedParams(rpc, filter) {
+    static getFlattenedParams(
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    rpc, 
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    filter) {
         const proxiedDuplicator = new ProxyParamsDuplicator();
         const proxiedRpc = new Proxy(rpc, proxiedDuplicator.buildProxyHandler());
         void filter(proxiedRpc);
@@ -245,6 +250,10 @@ export class ProxyParamsDuplicator {
  * The function creates a new `StructuredQuery` object and uses the
  * `Operation` and `Comparison` classes to build the query.
  */
-export function convertObjectFilterToStructuredQuery(objFilter) {
-    return new StructuredQuery("", new Operation(Operators.and, Object.entries(objFilter).map(([column, value]) => new Comparison(Comparators.eq, column, value))));
+export function convertObjectFilterToStructuredQuery(
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+objFilter) {
+    return new StructuredQuery("", new Operation(Operators.and, Object.entries(objFilter).map(
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    ([column, value]) => new Comparison(Comparators.eq, column, value))));
 }

package/dist/retrievers/self_query/vectara.cjs

@@ -1,7 +1,14 @@
 "use strict";
+/* eslint-disable @typescript-eslint/no-explicit-any */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.VectaraTranslator = void 0;
 const structured_query_1 = require("@langchain/core/structured_query");
+const entrypoint_deprecation_js_1 = require("../../util/entrypoint_deprecation.cjs");
+/* #__PURE__ */ (0, entrypoint_deprecation_js_1.logVersion020MigrationWarning)({
+    oldEntrypointName: "retrievers/self_query/vectara",
+    newEntrypointName: "structured_query/vectara",
+    newPackageName: "@langchain/community",
+});
 function processValue(value) {
     /** Convert a value to a string and add single quotes if it is a string. */
     if (typeof value === "string") {

package/dist/retrievers/self_query/vectara.d.ts

@@ -1,5 +1,5 @@
-import { VectaraFilter, VectaraStore } from "@langchain/community/vectorstores/vectara";
 import { BaseTranslator, Comparator, Comparison, Operation, Operator, StructuredQuery } from "@langchain/core/structured_query";
+import { VectorStore } from "@langchain/core/vectorstores";
 export type VectaraVisitorResult = VectaraOperationResult | VectaraComparisonResult | VectaraVisitorStructuredQueryResult;
 export type VectaraOperationResult = String;
 export type VectaraComparisonResult = String;
@@ -8,7 +8,8 @@ export type VectaraVisitorStructuredQueryResult = {
         filter?: VectaraOperationResult | VectaraComparisonResult;
     };
 };
-export declare class VectaraTranslator<T extends VectaraStore> extends BaseTranslator<T> {
+type VectaraFilter = any;
+export declare class VectaraTranslator<T extends VectorStore> extends BaseTranslator<T> {
     VisitOperationOutput: VectaraOperationResult;
     VisitComparisonOutput: VectaraComparisonResult;
     allowedOperators: Operator[];
@@ -38,3 +39,4 @@ export declare class VectaraTranslator<T extends VectaraStore> extends BaseTrans
     visitStructuredQuery(query: StructuredQuery): this["VisitStructuredQueryOutput"];
     mergeFilters(defaultFilter: VectaraFilter | undefined, generatedFilter: VectaraFilter | undefined, mergeType?: string, forceDefaultFilter?: boolean): VectaraFilter | undefined;
 }
+export {};

package/dist/retrievers/self_query/vectara.js

@@ -1,4 +1,11 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
 import { BaseTranslator, isFilterEmpty, Comparators, Operators, } from "@langchain/core/structured_query";
+import { logVersion020MigrationWarning } from "../../util/entrypoint_deprecation.js";
+/* #__PURE__ */ logVersion020MigrationWarning({
+    oldEntrypointName: "retrievers/self_query/vectara",
+    newEntrypointName: "structured_query/vectara",
+    newPackageName: "@langchain/community",
+});
 function processValue(value) {
     /** Convert a value to a string and add single quotes if it is a string. */
     if (typeof value === "string") {

package/dist/retrievers/self_query/weaviate.cjs

@@ -1,7 +1,14 @@
 "use strict";
+/* eslint-disable @typescript-eslint/no-explicit-any */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.WeaviateTranslator = void 0;
 const structured_query_1 = require("@langchain/core/structured_query");
+const entrypoint_deprecation_js_1 = require("../../util/entrypoint_deprecation.cjs");
+/* #__PURE__ */ (0, entrypoint_deprecation_js_1.logVersion020MigrationWarning)({
+    oldEntrypointName: "retrievers/self_query/weaviate",
+    newEntrypointName: "",
+    newPackageName: "@langchain/weaviate",
+});
 /**
  * A class that translates or converts data into a format that can be used
  * with Weaviate, a vector search engine. It extends the `BaseTranslator`

package/dist/retrievers/self_query/weaviate.d.ts

@@ -1,11 +1,12 @@
-import { WeaviateFilter, WeaviateStore } from "@langchain/community/vectorstores/weaviate";
 import { BaseTranslator, Comparator, Comparison, Operation, Operator, StructuredQuery } from "@langchain/core/structured_query";
+import { VectorStore } from "@langchain/core/vectorstores";
 type WeaviateOperatorValues = {
     valueText: string;
     valueInt: number;
     valueNumber: number;
     valueBoolean: boolean;
 };
+type WeaviateFilter = any;
 type WeaviateOperatorKeys = keyof WeaviateOperatorValues;
 type ExclusiveOperatorValue = {
     [L in WeaviateOperatorKeys]: {
@@ -47,7 +48,7 @@ export type WeaviateStructuredQueryResult = {
  * );
  * ```
  */
-export declare class WeaviateTranslator<T extends WeaviateStore> extends BaseTranslator<T> {
+export declare class WeaviateTranslator<T extends VectorStore> extends BaseTranslator<T> {
     VisitOperationOutput: WeaviateOperationResult;
     VisitComparisonOutput: WeaviateComparisonResult;
     allowedOperators: Operator[];

package/dist/retrievers/self_query/weaviate.js

@@ -1,4 +1,11 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
 import { isFilterEmpty, isString, isInt, isFloat, BaseTranslator, Comparators, Operators, } from "@langchain/core/structured_query";
+import { logVersion020MigrationWarning } from "../../util/entrypoint_deprecation.js";
+/* #__PURE__ */ logVersion020MigrationWarning({
+    oldEntrypointName: "retrievers/self_query/weaviate",
+    newEntrypointName: "",
+    newPackageName: "@langchain/weaviate",
+});
 /**
  * A class that translates or converts data into a format that can be used
  * with Weaviate, a vector search engine. It extends the `BaseTranslator`

package/dist/smith/config.d.ts

@@ -16,10 +16,10 @@ export type EvaluatorInputFormatter = ({ rawInput, rawPrediction, rawReferenceOu
     rawReferenceOutput?: any;
     run: Run;
 }) => EvaluatorInputs;
-export type DynamicRunEvaluatorParams = {
-    input: Record<string, unknown>;
-    prediction?: Record<string, unknown>;
-    reference?: Record<string, unknown>;
+export type DynamicRunEvaluatorParams<Input extends Record<string, any> = Record<string, unknown>, Prediction extends Record<string, any> = Record<string, unknown>, Reference extends Record<string, any> = Record<string, unknown>> = {
+    input: Input;
+    prediction?: Prediction;
+    reference?: Reference;
     run: Run;
     example?: Example;
 };

package/dist/stores/doc/base.cjs

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Docstore = void 0;
+/**
+ * Abstract class for a document store. All document stores should extend
+ * this class.
+ */
+class Docstore {
+}
+exports.Docstore = Docstore;

package/dist/stores/doc/base.d.ts

@@ -0,0 +1,9 @@
+import { Document } from "@langchain/core/documents";
+/**
+ * Abstract class for a document store. All document stores should extend
+ * this class.
+ */
+export declare abstract class Docstore {
+    abstract search(search: string): Promise<Document>;
+    abstract add(texts: Record<string, Document>): Promise<void>;
+}

package/dist/stores/doc/base.js

@@ -0,0 +1,6 @@
+/**
+ * Abstract class for a document store. All document stores should extend
+ * this class.
+ */
+export class Docstore {
+}

package/dist/stores/doc/in_memory.cjs

@@ -1,17 +1,107 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("@langchain/community/stores/doc/in_memory"), exports);
+exports.SynchronousInMemoryDocstore = exports.InMemoryDocstore = void 0;
+const base_js_1 = require("./base.cjs");
+/**
+ * Class for storing and retrieving documents in memory asynchronously.
+ * Extends the Docstore class.
+ */
+class InMemoryDocstore extends base_js_1.Docstore {
+    constructor(docs) {
+        super();
+        Object.defineProperty(this, "_docs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this._docs = docs ?? new Map();
+    }
+    /**
+     * Searches for a document in the store based on its ID.
+     * @param search The ID of the document to search for.
+     * @returns The document with the given ID.
+     */
+    async search(search) {
+        const result = this._docs.get(search);
+        if (!result) {
+            throw new Error(`ID ${search} not found.`);
+        }
+        else {
+            return result;
+        }
+    }
+    /**
+     * Adds new documents to the store.
+     * @param texts An object where the keys are document IDs and the values are the documents themselves.
+     * @returns Void
+     */
+    async add(texts) {
+        const keys = [...this._docs.keys()];
+        const overlapping = Object.keys(texts).filter((x) => keys.includes(x));
+        if (overlapping.length > 0) {
+            throw new Error(`Tried to add ids that already exist: ${overlapping}`);
+        }
+        for (const [key, value] of Object.entries(texts)) {
+            this._docs.set(key, value);
+        }
+    }
+    async mget(keys) {
+        return Promise.all(keys.map((key) => this.search(key)));
+    }
+    async mset(keyValuePairs) {
+        await Promise.all(keyValuePairs.map(([key, value]) => this.add({ [key]: value })));
+    }
+    async mdelete(_keys) {
+        throw new Error("Not implemented.");
+    }
+    // eslint-disable-next-line require-yield
+    async *yieldKeys(_prefix) {
+        throw new Error("Not implemented");
+    }
+}
+exports.InMemoryDocstore = InMemoryDocstore;
+/**
+ * Class for storing and retrieving documents in memory synchronously.
+ */
+class SynchronousInMemoryDocstore {
+    constructor(docs) {
+        Object.defineProperty(this, "_docs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this._docs = docs ?? new Map();
+    }
+    /**
+     * Searches for a document in the store based on its ID.
+     * @param search The ID of the document to search for.
+     * @returns The document with the given ID.
+     */
+    search(search) {
+        const result = this._docs.get(search);
+        if (!result) {
+            throw new Error(`ID ${search} not found.`);
+        }
+        else {
+            return result;
+        }
+    }
+    /**
+     * Adds new documents to the store.
+     * @param texts An object where the keys are document IDs and the values are the documents themselves.
+     * @returns Void
+     */
+    add(texts) {
+        const keys = [...this._docs.keys()];
+        const overlapping = Object.keys(texts).filter((x) => keys.includes(x));
+        if (overlapping.length > 0) {
+            throw new Error(`Tried to add ids that already exist: ${overlapping}`);
+        }
+        for (const [key, value] of Object.entries(texts)) {
+            this._docs.set(key, value);
+        }
+    }
+}
+exports.SynchronousInMemoryDocstore = SynchronousInMemoryDocstore;

package/dist/stores/doc/in_memory.d.ts

@@ -1 +1,46 @@
-export * from "@langchain/community/stores/doc/in_memory";
+import { Document } from "@langchain/core/documents";
+import { BaseStoreInterface } from "@langchain/core/stores";
+import { Docstore } from "./base.js";
+/**
+ * Class for storing and retrieving documents in memory asynchronously.
+ * Extends the Docstore class.
+ */
+export declare class InMemoryDocstore extends Docstore implements BaseStoreInterface<string, Document> {
+    _docs: Map<string, Document>;
+    constructor(docs?: Map<string, Document>);
+    /**
+     * Searches for a document in the store based on its ID.
+     * @param search The ID of the document to search for.
+     * @returns The document with the given ID.
+     */
+    search(search: string): Promise<Document>;
+    /**
+     * Adds new documents to the store.
+     * @param texts An object where the keys are document IDs and the values are the documents themselves.
+     * @returns Void
+     */
+    add(texts: Record<string, Document>): Promise<void>;
+    mget(keys: string[]): Promise<Document[]>;
+    mset(keyValuePairs: [string, Document][]): Promise<void>;
+    mdelete(_keys: string[]): Promise<void>;
+    yieldKeys(_prefix?: string): AsyncGenerator<string>;
+}
+/**
+ * Class for storing and retrieving documents in memory synchronously.
+ */
+export declare class SynchronousInMemoryDocstore {
+    _docs: Map<string, Document>;
+    constructor(docs?: Map<string, Document>);
+    /**
+     * Searches for a document in the store based on its ID.
+     * @param search The ID of the document to search for.
+     * @returns The document with the given ID.
+     */
+    search(search: string): Document;
+    /**
+     * Adds new documents to the store.
+     * @param texts An object where the keys are document IDs and the values are the documents themselves.
+     * @returns Void
+     */
+    add(texts: Record<string, Document>): void;
+}