langchain 0.0.212 → 0.0.213

package/dist/chains/retrieval.cjs

@@ -0,0 +1,60 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createRetrievalChain = void 0;
+const runnables_1 = require("@langchain/core/runnables");
+function isBaseRetriever(x) {
+    return (!!x &&
+        typeof x.getRelevantDocuments === "function");
+}
+/**
+ * Create a retrieval chain that retrieves documents and then passes them on.
+ * @param {CreateRetrievalChainParams} params A params object
+ *     containing a retriever and a combineDocsChain.
+ * @returns An LCEL Runnable which returns a an object
+ *     containing at least `context` and `answer` keys.
+ * @example
+ * ```typescript
+ * // yarn add langchain @langchain/openai
+ *
+ * import { ChatOpenAI } from "@langchain/openai";
+ * import { pull } from "langchain/hub";
+ * import { createRetrievalChain } from "langchain/chains/retrieval";
+ * import { createStuffDocumentsChain } from "langchain/chains/combine_documents";
+ *
+ * const retrievalQAChatPrompt = await pull("langchain-ai/retrieval-qa-chat");
+ * const llm = new ChatOpenAI({});
+ * const retriever = ...
+ * const combineDocsChain = await createStuffDocumentsChain(...);
+ * const retrievalChain = await createRetrievalChain({
+ *   retriever,
+ *   combineDocsChain,
+ * });
+ * const response = await chain.invoke({ input: "..." });
+ * ```
+ */
+async function createRetrievalChain({ retriever, combineDocsChain, }) {
+    let retrieveDocumentsChain;
+    if (isBaseRetriever(retriever)) {
+        retrieveDocumentsChain = runnables_1.RunnableSequence.from([
+            (input) => input.input,
+            retriever,
+        ]);
+    }
+    else {
+        // TODO: Fix typing by adding withConfig to core RunnableInterface
+        retrieveDocumentsChain = retriever;
+    }
+    const retrievalChain = runnables_1.RunnableSequence.from([
+        runnables_1.RunnablePassthrough.assign({
+            context: retrieveDocumentsChain.withConfig({
+                runName: "retrieve_documents",
+            }),
+            chat_history: (input) => input.chat_history ?? [],
+        }),
+        runnables_1.RunnablePassthrough.assign({
+            answer: combineDocsChain,
+        }),
+    ]).withConfig({ runName: "retrieval_chain" });
+    return retrievalChain;
+}
+exports.createRetrievalChain = createRetrievalChain;

package/dist/chains/retrieval.d.ts

@@ -0,0 +1,65 @@
+import type { BaseRetrieverInterface } from "@langchain/core/retrievers";
+import { type RunnableInterface } from "@langchain/core/runnables";
+import type { BaseMessage } from "@langchain/core/messages";
+import type { DocumentInterface } from "@langchain/core/documents";
+/**
+ * Parameters for the createRetrievalChain method.
+ */
+export type CreateRetrievalChainParams<RunOutput> = {
+    /**
+     * Retriever-like object that returns list of documents. Should
+     * either be a subclass of BaseRetriever or a Runnable that returns
+     * a list of documents. If a subclass of BaseRetriever, then it
+     * is expected that an `input` key be passed in - this is what
+     * is will be used to pass into the retriever. If this is NOT a
+     * subclass of BaseRetriever, then all the inputs will be passed
+     * into this runnable, meaning that runnable should take a object
+     * as input.
+     */
+    retriever: BaseRetrieverInterface | RunnableInterface<Record<string, any>, DocumentInterface[]>;
+    /**
+     * Runnable that takes inputs and produces a string output.
+     * The inputs to this will be any original inputs to this chain, a new
+     * context key with the retrieved documents, and chat_history (if not present
+     * in the inputs) with a value of `[]` (to easily enable conversational
+     * retrieval).
+     */
+    combineDocsChain: RunnableInterface<Record<string, any>, RunOutput>;
+};
+/**
+ * Create a retrieval chain that retrieves documents and then passes them on.
+ * @param {CreateRetrievalChainParams} params A params object
+ *     containing a retriever and a combineDocsChain.
+ * @returns An LCEL Runnable which returns a an object
+ *     containing at least `context` and `answer` keys.
+ * @example
+ * ```typescript
+ * // yarn add langchain @langchain/openai
+ *
+ * import { ChatOpenAI } from "@langchain/openai";
+ * import { pull } from "langchain/hub";
+ * import { createRetrievalChain } from "langchain/chains/retrieval";
+ * import { createStuffDocumentsChain } from "langchain/chains/combine_documents";
+ *
+ * const retrievalQAChatPrompt = await pull("langchain-ai/retrieval-qa-chat");
+ * const llm = new ChatOpenAI({});
+ * const retriever = ...
+ * const combineDocsChain = await createStuffDocumentsChain(...);
+ * const retrievalChain = await createRetrievalChain({
+ *   retriever,
+ *   combineDocsChain,
+ * });
+ * const response = await chain.invoke({ input: "..." });
+ * ```
+ */
+export declare function createRetrievalChain<RunOutput>({ retriever, combineDocsChain, }: CreateRetrievalChainParams<RunOutput>): Promise<RunnableInterface<{
+    input: string;
+    chat_history?: BaseMessage[] | string;
+} & {
+    [key: string]: unknown;
+}, {
+    context: string;
+    answer: RunOutput;
+} & {
+    [key: string]: unknown;
+}>>;

package/dist/chains/retrieval.js

@@ -0,0 +1,56 @@
+import { RunnableSequence, RunnablePassthrough, } from "@langchain/core/runnables";
+function isBaseRetriever(x) {
+    return (!!x &&
+        typeof x.getRelevantDocuments === "function");
+}
+/**
+ * Create a retrieval chain that retrieves documents and then passes them on.
+ * @param {CreateRetrievalChainParams} params A params object
+ *     containing a retriever and a combineDocsChain.
+ * @returns An LCEL Runnable which returns a an object
+ *     containing at least `context` and `answer` keys.
+ * @example
+ * ```typescript
+ * // yarn add langchain @langchain/openai
+ *
+ * import { ChatOpenAI } from "@langchain/openai";
+ * import { pull } from "langchain/hub";
+ * import { createRetrievalChain } from "langchain/chains/retrieval";
+ * import { createStuffDocumentsChain } from "langchain/chains/combine_documents";
+ *
+ * const retrievalQAChatPrompt = await pull("langchain-ai/retrieval-qa-chat");
+ * const llm = new ChatOpenAI({});
+ * const retriever = ...
+ * const combineDocsChain = await createStuffDocumentsChain(...);
+ * const retrievalChain = await createRetrievalChain({
+ *   retriever,
+ *   combineDocsChain,
+ * });
+ * const response = await chain.invoke({ input: "..." });
+ * ```
+ */
+export async function createRetrievalChain({ retriever, combineDocsChain, }) {
+    let retrieveDocumentsChain;
+    if (isBaseRetriever(retriever)) {
+        retrieveDocumentsChain = RunnableSequence.from([
+            (input) => input.input,
+            retriever,
+        ]);
+    }
+    else {
+        // TODO: Fix typing by adding withConfig to core RunnableInterface
+        retrieveDocumentsChain = retriever;
+    }
+    const retrievalChain = RunnableSequence.from([
+        RunnablePassthrough.assign({
+            context: retrieveDocumentsChain.withConfig({
+                runName: "retrieve_documents",
+            }),
+            chat_history: (input) => input.chat_history ?? [],
+        }),
+        RunnablePassthrough.assign({
+            answer: combineDocsChain,
+        }),
+    ]).withConfig({ runName: "retrieval_chain" });
+    return retrievalChain;
+}

package/dist/load/import_map.cjs

@@ -24,9 +24,9 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.document_transformers__openai_functions = exports.document_loaders__web__sort_xyz_blockchain = exports.document_loaders__web__serpapi = exports.document_loaders__web__searchapi = exports.document_loaders__base = exports.document = exports.memory = exports.text_splitter = exports.vectorstores__xata = exports.vectorstores__vectara = exports.vectorstores__prisma = exports.vectorstores__memory = exports.vectorstores__base = exports.prompts = exports.llms__fake = exports.llms__yandex = exports.llms__fireworks = exports.llms__ollama = exports.llms__cloudflare_workersai = exports.llms__aleph_alpha = exports.llms__ai21 = exports.llms__openai = exports.llms__base = exports.embeddings__voyage = exports.embeddings__minimax = exports.embeddings__openai = exports.embeddings__ollama = exports.embeddings__fake = exports.embeddings__cache_backed = exports.embeddings__base = exports.chains__openai_functions = exports.chains__combine_documents__reduce = exports.chains = exports.tools__google_places = exports.tools__render = exports.tools__connery = exports.tools = exports.base_language = exports.agents__openai__output_parser = exports.agents__xml__output_parser = exports.agents__react__output_parser = exports.agents__format_scratchpad__log_to_message = exports.agents__format_scratchpad__xml = exports.agents__format_scratchpad__log = exports.agents__format_scratchpad__openai_tools = exports.agents__format_scratchpad = exports.agents__toolkits__connery = exports.agents__toolkits = exports.agents = exports.load__serializable = void 0;
-exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__openai_files = exports.experimental__openai_assistant = exports.experimental__autogpt = exports.util__time = exports.util__math = exports.util__document = exports.storage__in_memory = exports.storage__encoder_backed = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = exports.retrievers__hyde = exports.retrievers__document_compressors__embeddings_filter = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__tavily_search_api = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = exports.retrievers__document_compressors = exports.retrievers__contextual_compression = exports.retrievers__databerry = exports.retrievers__chaindesk = exports.retrievers__remote = exports.output_parsers = exports.callbacks = exports.schema__storage = exports.schema__runnable = exports.schema__retriever = exports.schema__query_constructor = exports.schema__prompt_template = exports.schema__output_parser = exports.schema__document = exports.schema = exports.chat_models__fake = exports.chat_models__yandex = exports.chat_models__minimax = exports.chat_models__ollama = exports.chat_models__baiduwenxin = exports.chat_models__fireworks = exports.chat_models__cloudflare_workersai = exports.chat_models__anthropic = exports.chat_models__openai = exports.chat_models__base = void 0;
-exports.runnables__remote = exports.runnables = exports.evaluation = exports.experimental__masking = exports.experimental__chains__violation_of_expectations = exports.experimental__chat_models__ollama_functions = exports.experimental__chat_models__bittensor = void 0;
+exports.document_loaders__base = exports.document = exports.memory = exports.text_splitter = exports.vectorstores__xata = exports.vectorstores__vectara = exports.vectorstores__prisma = exports.vectorstores__memory = exports.vectorstores__base = exports.prompts = exports.llms__fake = exports.llms__yandex = exports.llms__fireworks = exports.llms__ollama = exports.llms__cloudflare_workersai = exports.llms__aleph_alpha = exports.llms__ai21 = exports.llms__openai = exports.llms__base = exports.embeddings__voyage = exports.embeddings__minimax = exports.embeddings__openai = exports.embeddings__ollama = exports.embeddings__fake = exports.embeddings__cache_backed = exports.embeddings__base = exports.chains__retrieval = exports.chains__openai_functions = exports.chains__history_aware_retriever = exports.chains__combine_documents__reduce = exports.chains__combine_documents = exports.chains = exports.tools__google_places = exports.tools__retriever = exports.tools__render = exports.tools__connery = exports.tools = exports.base_language = exports.agents__openai__output_parser = exports.agents__xml__output_parser = exports.agents__react__output_parser = exports.agents__format_scratchpad__log_to_message = exports.agents__format_scratchpad__xml = exports.agents__format_scratchpad__log = exports.agents__format_scratchpad__openai_tools = exports.agents__format_scratchpad = exports.agents__toolkits__connery = exports.agents__toolkits = exports.agents = exports.load__serializable = void 0;
+exports.experimental__openai_assistant = exports.experimental__autogpt = exports.util__time = exports.util__math = exports.util__document = exports.storage__in_memory = exports.storage__encoder_backed = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = exports.retrievers__hyde = exports.retrievers__document_compressors__embeddings_filter = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__tavily_search_api = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = exports.retrievers__document_compressors = exports.retrievers__contextual_compression = exports.retrievers__databerry = exports.retrievers__chaindesk = exports.retrievers__remote = exports.output_parsers = exports.callbacks = exports.schema__storage = exports.schema__runnable = exports.schema__retriever = exports.schema__query_constructor = exports.schema__prompt_template = exports.schema__output_parser = exports.schema__document = exports.schema = exports.chat_models__fake = exports.chat_models__yandex = exports.chat_models__minimax = exports.chat_models__ollama = exports.chat_models__baiduwenxin = exports.chat_models__fireworks = exports.chat_models__cloudflare_workersai = exports.chat_models__anthropic = exports.chat_models__openai = exports.chat_models__base = exports.document_transformers__openai_functions = exports.document_loaders__web__sort_xyz_blockchain = exports.document_loaders__web__serpapi = exports.document_loaders__web__searchapi = void 0;
+exports.runnables__remote = exports.runnables = exports.evaluation = exports.experimental__masking = exports.experimental__chains__violation_of_expectations = exports.experimental__chat_models__ollama_functions = exports.experimental__chat_models__bittensor = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__openai_files = void 0;
 exports.load__serializable = __importStar(require("../load/serializable.cjs"));
 exports.agents = __importStar(require("../agents/index.cjs"));
 exports.agents__toolkits = __importStar(require("../agents/toolkits/index.cjs"));
@@ -43,10 +43,14 @@ exports.base_language = __importStar(require("../base_language/index.cjs"));
 exports.tools = __importStar(require("../tools/index.cjs"));
 exports.tools__connery = __importStar(require("../tools/connery.cjs"));
 exports.tools__render = __importStar(require("../tools/render.cjs"));
+exports.tools__retriever = __importStar(require("../tools/retriever.cjs"));
 exports.tools__google_places = __importStar(require("../tools/google_places.cjs"));
 exports.chains = __importStar(require("../chains/index.cjs"));
+exports.chains__combine_documents = __importStar(require("../chains/combine_documents/index.cjs"));
 exports.chains__combine_documents__reduce = __importStar(require("../chains/combine_documents/reduce.cjs"));
+exports.chains__history_aware_retriever = __importStar(require("../chains/history_aware_retriever.cjs"));
 exports.chains__openai_functions = __importStar(require("../chains/openai_functions/index.cjs"));
+exports.chains__retrieval = __importStar(require("../chains/retrieval.cjs"));
 exports.embeddings__base = __importStar(require("../embeddings/base.cjs"));
 exports.embeddings__cache_backed = __importStar(require("../embeddings/cache_backed.cjs"));
 exports.embeddings__fake = __importStar(require("../embeddings/fake.cjs"));

package/dist/load/import_map.d.ts

@@ -14,10 +14,14 @@ export * as base_language from "../base_language/index.js";
 export * as tools from "../tools/index.js";
 export * as tools__connery from "../tools/connery.js";
 export * as tools__render from "../tools/render.js";
+export * as tools__retriever from "../tools/retriever.js";
 export * as tools__google_places from "../tools/google_places.js";
 export * as chains from "../chains/index.js";
+export * as chains__combine_documents from "../chains/combine_documents/index.js";
 export * as chains__combine_documents__reduce from "../chains/combine_documents/reduce.js";
+export * as chains__history_aware_retriever from "../chains/history_aware_retriever.js";
 export * as chains__openai_functions from "../chains/openai_functions/index.js";
+export * as chains__retrieval from "../chains/retrieval.js";
 export * as embeddings__base from "../embeddings/base.js";
 export * as embeddings__cache_backed from "../embeddings/cache_backed.js";
 export * as embeddings__fake from "../embeddings/fake.js";

package/dist/load/import_map.js

@@ -15,10 +15,14 @@ export * as base_language from "../base_language/index.js";
 export * as tools from "../tools/index.js";
 export * as tools__connery from "../tools/connery.js";
 export * as tools__render from "../tools/render.js";
+export * as tools__retriever from "../tools/retriever.js";
 export * as tools__google_places from "../tools/google_places.js";
 export * as chains from "../chains/index.js";
+export * as chains__combine_documents from "../chains/combine_documents/index.js";
 export * as chains__combine_documents__reduce from "../chains/combine_documents/reduce.js";
+export * as chains__history_aware_retriever from "../chains/history_aware_retriever.js";
 export * as chains__openai_functions from "../chains/openai_functions/index.js";
+export * as chains__retrieval from "../chains/retrieval.js";
 export * as embeddings__base from "../embeddings/base.js";
 export * as embeddings__cache_backed from "../embeddings/cache_backed.js";
 export * as embeddings__fake from "../embeddings/fake.js";

package/dist/output_parsers/json.cjs

@@ -1,81 +1,5 @@
 "use strict";
-// Adapted from https://github.com/KillianLucas/open-interpreter/blob/main/interpreter/utils/parse_partial_json.py
-// MIT License
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.parsePartialJson = void 0;
-function parsePartialJson(s) {
-    // If the input is undefined, return null to indicate failure.
-    if (typeof s === "undefined") {
-        return null;
-    }
-    // Attempt to parse the string as-is.
-    try {
-        return JSON.parse(s);
-    }
-    catch (error) {
-        // Pass
-    }
-    // Initialize variables.
-    let new_s = "";
-    const stack = [];
-    let isInsideString = false;
-    let escaped = false;
-    // Process each character in the string one at a time.
-    for (let char of s) {
-        if (isInsideString) {
-            if (char === '"' && !escaped) {
-                isInsideString = false;
-            }
-            else if (char === "\n" && !escaped) {
-                char = "\\n"; // Replace the newline character with the escape sequence.
-            }
-            else if (char === "\\") {
-                escaped = !escaped;
-            }
-            else {
-                escaped = false;
-            }
-        }
-        else {
-            if (char === '"') {
-                isInsideString = true;
-                escaped = false;
-            }
-            else if (char === "{") {
-                stack.push("}");
-            }
-            else if (char === "[") {
-                stack.push("]");
-            }
-            else if (char === "}" || char === "]") {
-                if (stack && stack[stack.length - 1] === char) {
-                    stack.pop();
-                }
-                else {
-                    // Mismatched closing character; the input is malformed.
-                    return null;
-                }
-            }
-        }
-        // Append the processed character to the new string.
-        new_s += char;
-    }
-    // If we're still inside a string at the end of processing,
-    // we need to close the string.
-    if (isInsideString) {
-        new_s += '"';
-    }
-    // Close any remaining open structures in the reverse order that they were opened.
-    for (let i = stack.length - 1; i >= 0; i -= 1) {
-        new_s += stack[i];
-    }
-    // Attempt to parse the modified string as JSON.
-    try {
-        return JSON.parse(new_s);
-    }
-    catch (error) {
-        // If we still can't parse the string as JSON, return null to indicate failure.
-        return null;
-    }
-}
-exports.parsePartialJson = parsePartialJson;
+var output_parsers_1 = require("@langchain/core/output_parsers");
+Object.defineProperty(exports, "parsePartialJson", { enumerable: true, get: function () { return output_parsers_1.parsePartialJson; } });

package/dist/output_parsers/json.d.ts

@@ -1 +1 @@
-export declare function parsePartialJson(s: string): any;
+export { parsePartialJson } from "@langchain/core/output_parsers";

package/dist/output_parsers/json.js

@@ -1,77 +1 @@
-// Adapted from https://github.com/KillianLucas/open-interpreter/blob/main/interpreter/utils/parse_partial_json.py
-// MIT License
-export function parsePartialJson(s) {
-    // If the input is undefined, return null to indicate failure.
-    if (typeof s === "undefined") {
-        return null;
-    }
-    // Attempt to parse the string as-is.
-    try {
-        return JSON.parse(s);
-    }
-    catch (error) {
-        // Pass
-    }
-    // Initialize variables.
-    let new_s = "";
-    const stack = [];
-    let isInsideString = false;
-    let escaped = false;
-    // Process each character in the string one at a time.
-    for (let char of s) {
-        if (isInsideString) {
-            if (char === '"' && !escaped) {
-                isInsideString = false;
-            }
-            else if (char === "\n" && !escaped) {
-                char = "\\n"; // Replace the newline character with the escape sequence.
-            }
-            else if (char === "\\") {
-                escaped = !escaped;
-            }
-            else {
-                escaped = false;
-            }
-        }
-        else {
-            if (char === '"') {
-                isInsideString = true;
-                escaped = false;
-            }
-            else if (char === "{") {
-                stack.push("}");
-            }
-            else if (char === "[") {
-                stack.push("]");
-            }
-            else if (char === "}" || char === "]") {
-                if (stack && stack[stack.length - 1] === char) {
-                    stack.pop();
-                }
-                else {
-                    // Mismatched closing character; the input is malformed.
-                    return null;
-                }
-            }
-        }
-        // Append the processed character to the new string.
-        new_s += char;
-    }
-    // If we're still inside a string at the end of processing,
-    // we need to close the string.
-    if (isInsideString) {
-        new_s += '"';
-    }
-    // Close any remaining open structures in the reverse order that they were opened.
-    for (let i = stack.length - 1; i >= 0; i -= 1) {
-        new_s += stack[i];
-    }
-    // Attempt to parse the modified string as JSON.
-    try {
-        return JSON.parse(new_s);
-    }
-    catch (error) {
-        // If we still can't parse the string as JSON, return null to indicate failure.
-        return null;
-    }
-}
+export { parsePartialJson } from "@langchain/core/output_parsers";

package/dist/output_parsers/openai_functions.d.ts

@@ -40,7 +40,7 @@ export declare class JsonOutputFunctionsParser extends BaseCumulativeTransformOu
     constructor(config?: {
         argsOnly?: boolean;
     } & BaseCumulativeTransformOutputParserInput);
-    protected _diff(prev: JSONPatchOperation | undefined, next: JSONPatchOperation): object | undefined;
+    protected _diff(prev: unknown | undefined, next: unknown): JSONPatchOperation[] | undefined;
     parsePartialResult(generations: ChatGeneration[]): Promise<object | undefined>;
     /**
      * Parses the output and returns a JSON object. If `argsOnly` is true,

package/dist/tools/retriever.cjs

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createRetrieverTool = void 0;
+const zod_1 = require("zod");
+const dynamic_1 = require("@langchain/community/tools/dynamic");
+const document_js_1 = require("../util/document.cjs");
+function createRetrieverTool(retriever, input) {
+    const func = async ({ query }, runManager) => {
+        const docs = await retriever.getRelevantDocuments(query, runManager?.getChild("retriever"));
+        return (0, document_js_1.formatDocumentsAsString)(docs);
+    };
+    const schema = zod_1.z.object({
+        query: zod_1.z.string().describe("query to look up in retriever"),
+    });
+    return new dynamic_1.DynamicStructuredTool({ ...input, func, schema });
+}
+exports.createRetrieverTool = createRetrieverTool;

package/dist/tools/retriever.d.ts

@@ -0,0 +1,10 @@
+import type { BaseRetrieverInterface } from "@langchain/core/retrievers";
+import { z } from "zod";
+import { DynamicStructuredTool, type DynamicStructuredToolInput } from "@langchain/community/tools/dynamic";
+export declare function createRetrieverTool(retriever: BaseRetrieverInterface, input: Omit<DynamicStructuredToolInput, "func" | "schema">): DynamicStructuredTool<z.ZodObject<{
+    query: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    query: string;
+}, {
+    query: string;
+}>>;

package/dist/tools/retriever.js

@@ -0,0 +1,13 @@
+import { z } from "zod";
+import { DynamicStructuredTool, } from "@langchain/community/tools/dynamic";
+import { formatDocumentsAsString } from "../util/document.js";
+export function createRetrieverTool(retriever, input) {
+    const func = async ({ query }, runManager) => {
+        const docs = await retriever.getRelevantDocuments(query, runManager?.getChild("retriever"));
+        return formatDocumentsAsString(docs);
+    };
+    const schema = z.object({
+        query: z.string().describe("query to look up in retriever"),
+    });
+    return new DynamicStructuredTool({ ...input, func, schema });
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langchain",
-  "version": "0.0.212",
+  "version": "0.0.213",
   "description": "Typescript bindings for langchain",
   "type": "module",
   "engines": {
@@ -79,6 +79,9 @@
     "tools/render.cjs",
     "tools/render.js",
     "tools/render.d.ts",
+    "tools/retriever.cjs",
+    "tools/retriever.js",
+    "tools/retriever.d.ts",
     "tools/sql.cjs",
     "tools/sql.js",
     "tools/sql.d.ts",
@@ -97,9 +100,15 @@
     "chains.cjs",
     "chains.js",
     "chains.d.ts",
+    "chains/combine_documents.cjs",
+    "chains/combine_documents.js",
+    "chains/combine_documents.d.ts",
     "chains/combine_documents/reduce.cjs",
     "chains/combine_documents/reduce.js",
     "chains/combine_documents/reduce.d.ts",
+    "chains/history_aware_retriever.cjs",
+    "chains/history_aware_retriever.js",
+    "chains/history_aware_retriever.d.ts",
     "chains/load.cjs",
     "chains/load.js",
     "chains/load.d.ts",
@@ -112,6 +121,9 @@
     "chains/query_constructor/ir.cjs",
     "chains/query_constructor/ir.js",
     "chains/query_constructor/ir.d.ts",
+    "chains/retrieval.cjs",
+    "chains/retrieval.js",
+    "chains/retrieval.d.ts",
     "chains/sql_db.cjs",
     "chains/sql_db.js",
     "chains/sql_db.d.ts",
@@ -1178,9 +1190,9 @@
   },
   "dependencies": {
     "@anthropic-ai/sdk": "^0.9.1",
-    "@langchain/community": "~0.0.8",
-    "@langchain/core": "~0.1.3",
-    "@langchain/openai": "~0.0.7",
+    "@langchain/community": "~0.0.12",
+    "@langchain/core": "~0.1.5",
+    "@langchain/openai": "~0.0.9",
     "binary-extensions": "^2.2.0",
     "expr-eval": "^2.0.2",
     "js-tiktoken": "^1.0.7",
@@ -1329,6 +1341,11 @@
       "import": "./tools/render.js",
       "require": "./tools/render.cjs"
     },
+    "./tools/retriever": {
+      "types": "./tools/retriever.d.ts",
+      "import": "./tools/retriever.js",
+      "require": "./tools/retriever.cjs"
+    },
     "./tools/sql": {
       "types": "./tools/sql.d.ts",
       "import": "./tools/sql.js",
@@ -1359,11 +1376,21 @@
       "import": "./chains.js",
       "require": "./chains.cjs"
     },
+    "./chains/combine_documents": {
+      "types": "./chains/combine_documents.d.ts",
+      "import": "./chains/combine_documents.js",
+      "require": "./chains/combine_documents.cjs"
+    },
     "./chains/combine_documents/reduce": {
       "types": "./chains/combine_documents/reduce.d.ts",
       "import": "./chains/combine_documents/reduce.js",
       "require": "./chains/combine_documents/reduce.cjs"
     },
+    "./chains/history_aware_retriever": {
+      "types": "./chains/history_aware_retriever.d.ts",
+      "import": "./chains/history_aware_retriever.js",
+      "require": "./chains/history_aware_retriever.cjs"
+    },
     "./chains/load": {
       "types": "./chains/load.d.ts",
       "import": "./chains/load.js",
@@ -1384,6 +1411,11 @@
       "import": "./chains/query_constructor/ir.js",
       "require": "./chains/query_constructor/ir.cjs"
     },
+    "./chains/retrieval": {
+      "types": "./chains/retrieval.d.ts",
+      "import": "./chains/retrieval.js",
+      "require": "./chains/retrieval.cjs"
+    },
     "./chains/sql_db": {
       "types": "./chains/sql_db.d.ts",
       "import": "./chains/sql_db.js",

package/tools/retriever.cjs

@@ -0,0 +1 @@
+module.exports = require('../dist/tools/retriever.cjs');

package/tools/retriever.d.ts

@@ -0,0 +1 @@
+export * from '../dist/tools/retriever.js'

package/tools/retriever.js

@@ -0,0 +1 @@
+export * from '../dist/tools/retriever.js'