langchain 0.0.149 → 0.0.151

package/dist/experimental/chat_models/bittensor.cjs

@@ -0,0 +1,141 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NIBittensorChatModel = void 0;
+const base_js_1 = require("../../chat_models/base.cjs");
+const index_js_1 = require("../../schema/index.cjs");
+/**
+ * Class representing the Neural Internet chat model powerd by Bittensor, a decentralized network
+ * full of different AI models.s
+ * To analyze API_KEYS and logs of you usage visit
+ *      https://api.neuralinternet.ai/api-keys
+ *      https://api.neuralinternet.ai/logs
+ */
+class NIBittensorChatModel extends base_js_1.BaseChatModel {
+    static lc_name() {
+        return "NIBittensorLLM";
+    }
+    constructor(fields) {
+        super(fields ?? {});
+        Object.defineProperty(this, "systemPrompt", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.systemPrompt =
+            fields?.systemPrompt ??
+                "You are an assistant which is created by Neural Internet(NI) in decentralized network named as a Bittensor. Your task is to provide accurate response based on user prompt";
+    }
+    _combineLLMOutput() {
+        return [];
+    }
+    _llmType() {
+        return "NIBittensorLLM";
+    }
+    messageToOpenAIRole(message) {
+        const type = message._getType();
+        switch (type) {
+            case "system":
+                return "system";
+            case "ai":
+                return "assistant";
+            case "human":
+                return "user";
+            default:
+                return "user";
+        }
+    }
+    stringToChatMessage(message) {
+        return new index_js_1.ChatMessage(message, "assistant");
+    }
+    /** Call out to NIBittensorChatModel's complete endpoint.
+     Args:
+         messages: The messages to pass into the model.
+  
+         Returns: The chat response by the model.
+  
+     Example:
+      const chat = new NIBittensorChatModel();
+      const message = new HumanMessage('What is bittensor?');
+      const res = await chat.call([message]);
+     */
+    async _generate(messages) {
+        const processed_messages = messages.map((message) => ({
+            role: this.messageToOpenAIRole(message),
+            content: message.content,
+        }));
+        const generations = [];
+        try {
+            // Retrieve API KEY
+            const apiKeyResponse = await fetch("https://test.neuralinternet.ai/admin/api-keys/");
+            if (!apiKeyResponse.ok) {
+                throw new Error("Network response was not ok");
+            }
+            const apiKeysData = await apiKeyResponse.json();
+            const apiKey = apiKeysData[0].api_key;
+            const headers = {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${apiKey}`,
+                "Endpoint-Version": "2023-05-19",
+            };
+            const minerResponse = await fetch("https://test.neuralinternet.ai/top_miner_uids", { headers });
+            if (!minerResponse.ok) {
+                throw new Error("Network response was not ok");
+            }
+            const uids = await minerResponse.json();
+            if (Array.isArray(uids) && uids.length) {
+                for (const uid of uids) {
+                    try {
+                        const payload = {
+                            uids: [uid],
+                            messages: [
+                                { role: "system", content: this.systemPrompt },
+                                ...processed_messages,
+                            ],
+                        };
+                        const response = await fetch("https://test.neuralinternet.ai/chat", {
+                            method: "POST",
+                            headers,
+                            body: JSON.stringify(payload),
+                        });
+                        if (!response.ok) {
+                            throw new Error("Network response was not ok");
+                        }
+                        const chatData = await response.json();
+                        if (chatData.choices) {
+                            const generation = {
+                                text: chatData.choices[0].message.content,
+                                message: this.stringToChatMessage(chatData.choices[0].message.content),
+                            };
+                            generations.push(generation);
+                            return { generations, llmOutput: {} };
+                        }
+                    }
+                    catch (error) {
+                        continue;
+                    }
+                }
+            }
+        }
+        catch (error) {
+            const generation = {
+                text: "Sorry I am unable to provide response now, Please try again later.",
+                message: this.stringToChatMessage("Sorry I am unable to provide response now, Please try again later."),
+            };
+            generations.push(generation);
+            return { generations, llmOutput: {} };
+        }
+        const generation = {
+            text: "Sorry I am unable to provide response now, Please try again later.",
+            message: this.stringToChatMessage("Sorry I am unable to provide response now, Please try again later."),
+        };
+        generations.push(generation);
+        return { generations, llmOutput: {} };
+    }
+    identifyingParams() {
+        return {
+            systemPrompt: this.systemPrompt,
+        };
+    }
+}
+exports.NIBittensorChatModel = NIBittensorChatModel;

package/dist/experimental/chat_models/bittensor.d.ts

@@ -0,0 +1,36 @@
+import { BaseChatModel, BaseChatModelParams } from "../../chat_models/base.js";
+import { BaseMessage, ChatResult } from "../../schema/index.js";
+export interface BittensorInput extends BaseChatModelParams {
+    systemPrompt?: string | null | undefined;
+}
+/**
+ * Class representing the Neural Internet chat model powerd by Bittensor, a decentralized network
+ * full of different AI models.s
+ * To analyze API_KEYS and logs of you usage visit
+ *      https://api.neuralinternet.ai/api-keys
+ *      https://api.neuralinternet.ai/logs
+ */
+export declare class NIBittensorChatModel extends BaseChatModel implements BittensorInput {
+    static lc_name(): string;
+    systemPrompt: string;
+    constructor(fields?: BittensorInput);
+    _combineLLMOutput(): never[];
+    _llmType(): string;
+    messageToOpenAIRole(message: BaseMessage): "system" | "user" | "assistant";
+    stringToChatMessage(message: string): BaseMessage;
+    /** Call out to NIBittensorChatModel's complete endpoint.
+     Args:
+         messages: The messages to pass into the model.
+  
+         Returns: The chat response by the model.
+  
+     Example:
+      const chat = new NIBittensorChatModel();
+      const message = new HumanMessage('What is bittensor?');
+      const res = await chat.call([message]);
+     */
+    _generate(messages: BaseMessage[]): Promise<ChatResult>;
+    identifyingParams(): {
+        systemPrompt: string | null | undefined;
+    };
+}

package/dist/experimental/chat_models/bittensor.js

@@ -0,0 +1,137 @@
+import { BaseChatModel } from "../../chat_models/base.js";
+import { ChatMessage, } from "../../schema/index.js";
+/**
+ * Class representing the Neural Internet chat model powerd by Bittensor, a decentralized network
+ * full of different AI models.s
+ * To analyze API_KEYS and logs of you usage visit
+ *      https://api.neuralinternet.ai/api-keys
+ *      https://api.neuralinternet.ai/logs
+ */
+export class NIBittensorChatModel extends BaseChatModel {
+    static lc_name() {
+        return "NIBittensorLLM";
+    }
+    constructor(fields) {
+        super(fields ?? {});
+        Object.defineProperty(this, "systemPrompt", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.systemPrompt =
+            fields?.systemPrompt ??
+                "You are an assistant which is created by Neural Internet(NI) in decentralized network named as a Bittensor. Your task is to provide accurate response based on user prompt";
+    }
+    _combineLLMOutput() {
+        return [];
+    }
+    _llmType() {
+        return "NIBittensorLLM";
+    }
+    messageToOpenAIRole(message) {
+        const type = message._getType();
+        switch (type) {
+            case "system":
+                return "system";
+            case "ai":
+                return "assistant";
+            case "human":
+                return "user";
+            default:
+                return "user";
+        }
+    }
+    stringToChatMessage(message) {
+        return new ChatMessage(message, "assistant");
+    }
+    /** Call out to NIBittensorChatModel's complete endpoint.
+     Args:
+         messages: The messages to pass into the model.
+  
+         Returns: The chat response by the model.
+  
+     Example:
+      const chat = new NIBittensorChatModel();
+      const message = new HumanMessage('What is bittensor?');
+      const res = await chat.call([message]);
+     */
+    async _generate(messages) {
+        const processed_messages = messages.map((message) => ({
+            role: this.messageToOpenAIRole(message),
+            content: message.content,
+        }));
+        const generations = [];
+        try {
+            // Retrieve API KEY
+            const apiKeyResponse = await fetch("https://test.neuralinternet.ai/admin/api-keys/");
+            if (!apiKeyResponse.ok) {
+                throw new Error("Network response was not ok");
+            }
+            const apiKeysData = await apiKeyResponse.json();
+            const apiKey = apiKeysData[0].api_key;
+            const headers = {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${apiKey}`,
+                "Endpoint-Version": "2023-05-19",
+            };
+            const minerResponse = await fetch("https://test.neuralinternet.ai/top_miner_uids", { headers });
+            if (!minerResponse.ok) {
+                throw new Error("Network response was not ok");
+            }
+            const uids = await minerResponse.json();
+            if (Array.isArray(uids) && uids.length) {
+                for (const uid of uids) {
+                    try {
+                        const payload = {
+                            uids: [uid],
+                            messages: [
+                                { role: "system", content: this.systemPrompt },
+                                ...processed_messages,
+                            ],
+                        };
+                        const response = await fetch("https://test.neuralinternet.ai/chat", {
+                            method: "POST",
+                            headers,
+                            body: JSON.stringify(payload),
+                        });
+                        if (!response.ok) {
+                            throw new Error("Network response was not ok");
+                        }
+                        const chatData = await response.json();
+                        if (chatData.choices) {
+                            const generation = {
+                                text: chatData.choices[0].message.content,
+                                message: this.stringToChatMessage(chatData.choices[0].message.content),
+                            };
+                            generations.push(generation);
+                            return { generations, llmOutput: {} };
+                        }
+                    }
+                    catch (error) {
+                        continue;
+                    }
+                }
+            }
+        }
+        catch (error) {
+            const generation = {
+                text: "Sorry I am unable to provide response now, Please try again later.",
+                message: this.stringToChatMessage("Sorry I am unable to provide response now, Please try again later."),
+            };
+            generations.push(generation);
+            return { generations, llmOutput: {} };
+        }
+        const generation = {
+            text: "Sorry I am unable to provide response now, Please try again later.",
+            message: this.stringToChatMessage("Sorry I am unable to provide response now, Please try again later."),
+        };
+        generations.push(generation);
+        return { generations, llmOutput: {} };
+    }
+    identifyingParams() {
+        return {
+            systemPrompt: this.systemPrompt,
+        };
+    }
+}

package/dist/llms/openai.cjs

@@ -56,9 +56,10 @@ class OpenAI extends base_js_1.BaseLLM {
     constructor(fields, 
     /** @deprecated */
     configuration) {
-        if (fields?.modelName?.startsWith("gpt-3.5-turbo") ||
+        if ((fields?.modelName?.startsWith("gpt-3.5-turbo") ||
             fields?.modelName?.startsWith("gpt-4") ||
-            fields?.modelName?.startsWith("gpt-4-32k")) {
+            fields?.modelName?.startsWith("gpt-4-32k")) &&
+            !fields?.modelName.endsWith("-instruct")) {
             // eslint-disable-next-line no-constructor-return, @typescript-eslint/no-explicit-any
             return new openai_chat_js_1.OpenAIChat(fields, configuration);
         }

package/dist/llms/openai.js

@@ -53,9 +53,10 @@ export class OpenAI extends BaseLLM {
     constructor(fields, 
     /** @deprecated */
     configuration) {
-        if (fields?.modelName?.startsWith("gpt-3.5-turbo") ||
+        if ((fields?.modelName?.startsWith("gpt-3.5-turbo") ||
             fields?.modelName?.startsWith("gpt-4") ||
-            fields?.modelName?.startsWith("gpt-4-32k")) {
+            fields?.modelName?.startsWith("gpt-4-32k")) &&
+            !fields?.modelName.endsWith("-instruct")) {
             // eslint-disable-next-line no-constructor-return, @typescript-eslint/no-explicit-any
             return new OpenAIChat(fields, configuration);
         }

package/dist/llms/replicate.cjs

@@ -44,6 +44,12 @@ class Replicate extends base_js_1.LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "promptKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         const apiKey = fields?.apiKey ??
             (0, env_js_1.getEnvironmentVariable)("REPLICATE_API_KEY") ?? // previous environment variable for backwards compatibility
             (0, env_js_1.getEnvironmentVariable)("REPLICATE_API_TOKEN"); // current environment variable, matching the Python library
@@ -53,6 +59,7 @@ class Replicate extends base_js_1.LLM {
         this.apiKey = apiKey;
         this.model = fields.model;
         this.input = fields.input ?? {};
+        this.promptKey = fields.promptKey;
     }
     _llmType() {
         return "replicate";
@@ -64,11 +71,30 @@ class Replicate extends base_js_1.LLM {
             userAgent: "langchain",
             auth: this.apiKey,
         });
+        if (this.promptKey === undefined) {
+            const [modelString, versionString] = this.model.split(":");
+            const version = await replicate.models.versions.get(modelString.split("/")[0], modelString.split("/")[1], versionString);
+            const openapiSchema = version.openapi_schema;
+            const inputProperties = 
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            openapiSchema?.components?.schemas?.Input?.properties;
+            if (inputProperties === undefined) {
+                this.promptKey = "prompt";
+            }
+            else {
+                const sortedInputProperties = Object.entries(inputProperties).sort(([_keyA, valueA], [_keyB, valueB]) => {
+                    const orderA = valueA["x-order"] || 0;
+                    const orderB = valueB["x-order"] || 0;
+                    return orderA - orderB;
+                });
+                this.promptKey = sortedInputProperties[0][0] ?? "prompt";
+            }
+        }
         const output = await this.caller.callWithOptions({ signal: options.signal }, () => replicate.run(this.model, {
-            wait: true,
             input: {
+                // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                [this.promptKey]: prompt,
                 ...this.input,
-                prompt,
             },
         }));
         if (typeof output === "string") {

package/dist/llms/replicate.d.ts

@@ -10,6 +10,8 @@ export interface ReplicateInput {
         [key: string]: string | number | boolean;
     };
     apiKey?: string;
+    /** The key used to pass prompts to the model. */
+    promptKey?: string;
 }
 /**
  * Class responsible for managing the interaction with the Replicate API.
@@ -26,6 +28,7 @@ export declare class Replicate extends LLM implements ReplicateInput {
     model: ReplicateInput["model"];
     input: ReplicateInput["input"];
     apiKey: string;
+    promptKey?: string;
     constructor(fields: ReplicateInput & BaseLLMParams);
     _llmType(): string;
     /** @ignore */

package/dist/llms/replicate.js

@@ -41,6 +41,12 @@ export class Replicate extends LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "promptKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         const apiKey = fields?.apiKey ??
             getEnvironmentVariable("REPLICATE_API_KEY") ?? // previous environment variable for backwards compatibility
             getEnvironmentVariable("REPLICATE_API_TOKEN"); // current environment variable, matching the Python library
@@ -50,6 +56,7 @@ export class Replicate extends LLM {
         this.apiKey = apiKey;
         this.model = fields.model;
         this.input = fields.input ?? {};
+        this.promptKey = fields.promptKey;
     }
     _llmType() {
         return "replicate";
@@ -61,11 +68,30 @@ export class Replicate extends LLM {
             userAgent: "langchain",
             auth: this.apiKey,
         });
+        if (this.promptKey === undefined) {
+            const [modelString, versionString] = this.model.split(":");
+            const version = await replicate.models.versions.get(modelString.split("/")[0], modelString.split("/")[1], versionString);
+            const openapiSchema = version.openapi_schema;
+            const inputProperties = 
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            openapiSchema?.components?.schemas?.Input?.properties;
+            if (inputProperties === undefined) {
+                this.promptKey = "prompt";
+            }
+            else {
+                const sortedInputProperties = Object.entries(inputProperties).sort(([_keyA, valueA], [_keyB, valueB]) => {
+                    const orderA = valueA["x-order"] || 0;
+                    const orderB = valueB["x-order"] || 0;
+                    return orderA - orderB;
+                });
+                this.promptKey = sortedInputProperties[0][0] ?? "prompt";
+            }
+        }
         const output = await this.caller.callWithOptions({ signal: options.signal }, () => replicate.run(this.model, {
-            wait: true,
             input: {
+                // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                [this.promptKey]: prompt,
                 ...this.input,
-                prompt,
             },
         }));
         if (typeof output === "string") {

package/dist/load/import_constants.cjs

@@ -48,6 +48,7 @@ exports.optionalImportEntrypoints = [
     "langchain/vectorstores/qdrant",
     "langchain/vectorstores/supabase",
     "langchain/vectorstores/opensearch",
+    "langchain/vectorstores/pgvector",
     "langchain/vectorstores/milvus",
     "langchain/vectorstores/typeorm",
     "langchain/vectorstores/myscale",

package/dist/load/import_constants.js

@@ -45,6 +45,7 @@ export const optionalImportEntrypoints = [
     "langchain/vectorstores/qdrant",
     "langchain/vectorstores/supabase",
     "langchain/vectorstores/opensearch",
+    "langchain/vectorstores/pgvector",
     "langchain/vectorstores/milvus",
     "langchain/vectorstores/typeorm",
     "langchain/vectorstores/myscale",

package/dist/load/import_map.cjs

@@ -25,7 +25,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.retrievers__document_compressors = exports.retrievers__contextual_compression = exports.retrievers__databerry = exports.retrievers__remote = exports.output_parsers = exports.callbacks = exports.schema__storage = exports.schema__runnable = exports.schema__retriever = exports.schema__query_constructor = exports.schema__output_parser = exports.schema__document = exports.schema = exports.chat_models__minimax = exports.chat_models__ollama = exports.chat_models__baiduwenxin = 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__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__ollama = exports.llms__aleph_alpha = exports.llms__ai21 = exports.llms__openai = exports.llms__base = 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 = exports.tools = exports.base_language = exports.agents__toolkits = exports.agents = exports.load__serializable = void 0;
-exports.evaluation = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__math = exports.storage__in_memory = 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__chain_extract = exports.retrievers__time_weighted = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = void 0;
+exports.evaluation = exports.experimental__chat_models__bittensor = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__math = exports.storage__in_memory = 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__chain_extract = exports.retrievers__time_weighted = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = 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"));
@@ -94,4 +94,5 @@ exports.experimental__autogpt = __importStar(require("../experimental/autogpt/in
 exports.experimental__babyagi = __importStar(require("../experimental/babyagi/index.cjs"));
 exports.experimental__generative_agents = __importStar(require("../experimental/generative_agents/index.cjs"));
 exports.experimental__plan_and_execute = __importStar(require("../experimental/plan_and_execute/index.cjs"));
+exports.experimental__chat_models__bittensor = __importStar(require("../experimental/chat_models/bittensor.cjs"));
 exports.evaluation = __importStar(require("../evaluation/index.cjs"));

package/dist/load/import_map.d.ts

@@ -66,4 +66,5 @@ export * as experimental__autogpt from "../experimental/autogpt/index.js";
 export * as experimental__babyagi from "../experimental/babyagi/index.js";
 export * as experimental__generative_agents from "../experimental/generative_agents/index.js";
 export * as experimental__plan_and_execute from "../experimental/plan_and_execute/index.js";
+export * as experimental__chat_models__bittensor from "../experimental/chat_models/bittensor.js";
 export * as evaluation from "../evaluation/index.js";

package/dist/load/import_map.js

@@ -67,4 +67,5 @@ export * as experimental__autogpt from "../experimental/autogpt/index.js";
 export * as experimental__babyagi from "../experimental/babyagi/index.js";
 export * as experimental__generative_agents from "../experimental/generative_agents/index.js";
 export * as experimental__plan_and_execute from "../experimental/plan_and_execute/index.js";
+export * as experimental__chat_models__bittensor from "../experimental/chat_models/bittensor.js";
 export * as evaluation from "../evaluation/index.js";

package/dist/prompts/prompt.cjs

@@ -99,6 +99,8 @@ class PromptTemplate extends base_js_1.BaseStringPromptTemplate {
             }
         });
         return new PromptTemplate({
+            // Rely on extracted types
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
             inputVariables: [...names],
             templateFormat,
             template,

package/dist/prompts/prompt.d.ts

@@ -32,7 +32,7 @@ type NonAlphanumeric = " " | "\t" | "\n" | "\r" | '"' | "'" | "{" | "[" | "(" |
  */
 type ExtractTemplateParamsRecursive<T extends string, Result extends string[] = []> = T extends `${string}{${infer Param}}${infer Rest}` ? Param extends `${NonAlphanumeric}${string}` ? ExtractTemplateParamsRecursive<Rest, Result> : ExtractTemplateParamsRecursive<Rest, [...Result, Param]> : Result;
 export type ParamsFromFString<T extends string> = {
-    [Key in ExtractTemplateParamsRecursive<T>[number]]: string;
+    [Key in ExtractTemplateParamsRecursive<T>[number] | (string & Record<never, never>)]: string;
 };
 /**
  * Schema to represent a basic prompt for an LLM.

package/dist/prompts/prompt.js

@@ -96,6 +96,8 @@ export class PromptTemplate extends BaseStringPromptTemplate {
             }
         });
         return new PromptTemplate({
+            // Rely on extracted types
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
             inputVariables: [...names],
             templateFormat,
             template,

package/dist/schema/runnable/base.cjs

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.RunnableWithFallbacks = exports.RunnableLambda = exports.RunnableMap = exports.RunnableSequence = exports.RunnableRetry = exports.RunnableEach = exports.RunnableBinding = exports.Runnable = void 0;
+exports._coerceToRunnable = exports.RunnableWithFallbacks = exports.RunnableLambda = exports.RunnableMap = exports.RunnableSequence = exports.RunnableRetry = exports.RunnableEach = exports.RunnableBinding = exports.Runnable = void 0;
 const p_retry_1 = __importDefault(require("p-retry"));
 const manager_js_1 = require("../../callbacks/manager.cjs");
 const serializable_js_1 = require("../../load/serializable.cjs");
@@ -700,6 +700,7 @@ class RunnableSequence extends Runnable {
     static isRunnableSequence(thing) {
         return Array.isArray(thing.middle) && Runnable.isRunnable(thing);
     }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     static from([first, ...runnables]) {
         return new RunnableSequence({
             first: _coerceToRunnable(first),
@@ -789,8 +790,15 @@ class RunnableLambda extends Runnable {
         });
         this.func = fields.func;
     }
+    async _invoke(input, config, runManager) {
+        let output = await this.func(input);
+        if (output && Runnable.isRunnable(output)) {
+            output = await output.invoke(input, this._patchConfig(config, runManager?.getChild()));
+        }
+        return output;
+    }
     async invoke(input, options) {
-        return this._callWithConfig(async (input) => this.func(input), input, options);
+        return this._callWithConfig(this._invoke, input, options);
     }
 }
 exports.RunnableLambda = RunnableLambda;
@@ -908,3 +916,4 @@ function _coerceToRunnable(coerceable) {
         throw new Error(`Expected a Runnable, function or object.\nInstead got an unsupported type.`);
     }
 }
+exports._coerceToRunnable = _coerceToRunnable;

package/dist/schema/runnable/base.d.ts

@@ -242,7 +242,7 @@ export declare class RunnableSequence<RunInput = any, RunOutput = any> extends R
     _streamIterator(input: RunInput, options?: RunnableConfig): AsyncGenerator<RunOutput>;
     pipe<NewRunOutput>(coerceable: RunnableLike<RunOutput, NewRunOutput>): RunnableSequence<RunInput, Exclude<NewRunOutput, Error>>;
     static isRunnableSequence(thing: any): thing is RunnableSequence;
-    static from<RunInput, RunOutput>([first, ...runnables]: [
+    static from<RunInput = any, RunOutput = any>([first, ...runnables]: [
         RunnableLike<RunInput>,
         ...RunnableLike[],
         RunnableLike<any, RunOutput>
@@ -272,6 +272,7 @@ export declare class RunnableLambda<RunInput, RunOutput> extends Runnable<RunInp
     constructor(fields: {
         func: RunnableFunc<RunInput, RunOutput>;
     });
+    _invoke(input: RunInput, config?: Partial<BaseCallbackConfig>, runManager?: CallbackManagerForChainRun): Promise<RunOutput>;
     invoke(input: RunInput, options?: Partial<BaseCallbackConfig>): Promise<RunOutput>;
 }
 /**
@@ -297,4 +298,5 @@ export declare class RunnableWithFallbacks<RunInput, RunOutput> extends Runnable
     }): Promise<(RunOutput | Error)[]>;
     batch(inputs: RunInput[], options?: Partial<BaseCallbackConfig> | Partial<BaseCallbackConfig>[], batchOptions?: RunnableBatchOptions): Promise<(RunOutput | Error)[]>;
 }
+export declare function _coerceToRunnable<RunInput, RunOutput>(coerceable: RunnableLike<RunInput, RunOutput>): Runnable<RunInput, Exclude<RunOutput, Error>>;
 export {};

package/dist/schema/runnable/base.js

@@ -690,6 +690,7 @@ export class RunnableSequence extends Runnable {
     static isRunnableSequence(thing) {
         return Array.isArray(thing.middle) && Runnable.isRunnable(thing);
     }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     static from([first, ...runnables]) {
         return new RunnableSequence({
             first: _coerceToRunnable(first),
@@ -777,8 +778,15 @@ export class RunnableLambda extends Runnable {
         });
         this.func = fields.func;
     }
+    async _invoke(input, config, runManager) {
+        let output = await this.func(input);
+        if (output && Runnable.isRunnable(output)) {
+            output = await output.invoke(input, this._patchConfig(config, runManager?.getChild()));
+        }
+        return output;
+    }
     async invoke(input, options) {
-        return this._callWithConfig(async (input) => this.func(input), input, options);
+        return this._callWithConfig(this._invoke, input, options);
     }
 }
 /**
@@ -874,7 +882,7 @@ export class RunnableWithFallbacks extends Runnable {
     }
 }
 // TODO: Figure out why the compiler needs help eliminating Error as a RunOutput type
-function _coerceToRunnable(coerceable) {
+export function _coerceToRunnable(coerceable) {
     if (typeof coerceable === "function") {
         return new RunnableLambda({ func: coerceable });
     }