langchain 0.0.182-rc.1 → 0.0.182

package/dist/agents/openai/index.cjs

@@ -7,6 +7,7 @@ const prompt_js_1 = require("./prompt.cjs");
 const chat_js_1 = require("../../prompts/chat.cjs");
 const llm_chain_js_1 = require("../../chains/llm_chain.cjs");
 const output_parser_js_1 = require("./output_parser.cjs");
+const convert_to_openai_js_1 = require("../../tools/convert_to_openai.cjs");
 /**
  * Checks if the given action is a FunctionsAgentAction.
  * @param action The action to check.
@@ -140,7 +141,7 @@ class OpenAIAgent extends agent_js_1.Agent {
         const llm = this.llmChain.llm;
         const valuesForPrompt = { ...newInputs };
         const valuesForLLM = {
-            tools: this.tools,
+            functions: this.tools.map(convert_to_openai_js_1.formatToOpenAIFunction),
         };
         for (const key of this.llmChain.llm.callKeys) {
             if (key in inputs) {

package/dist/agents/openai/index.js

@@ -4,6 +4,7 @@ import { PREFIX } from "./prompt.js";
 import { ChatPromptTemplate, HumanMessagePromptTemplate, MessagesPlaceholder, SystemMessagePromptTemplate, } from "../../prompts/chat.js";
 import { LLMChain } from "../../chains/llm_chain.js";
 import { OpenAIFunctionsAgentOutputParser, } from "./output_parser.js";
+import { formatToOpenAIFunction } from "../../tools/convert_to_openai.js";
 /**
  * Checks if the given action is a FunctionsAgentAction.
  * @param action The action to check.
@@ -136,7 +137,7 @@ export class OpenAIAgent extends Agent {
         const llm = this.llmChain.llm;
         const valuesForPrompt = { ...newInputs };
         const valuesForLLM = {
-            tools: this.tools,
+            functions: this.tools.map(formatToOpenAIFunction),
         };
         for (const key of this.llmChain.llm.callKeys) {
             if (key in inputs) {

package/dist/cache/base.d.ts

@@ -12,7 +12,7 @@ import { Generation, StoredGeneration } from "../schema/index.js";
 export declare const getCacheKey: (...strings: string[]) => string;
 export declare function deserializeStoredGeneration(storedGeneration: StoredGeneration): {
     text: string;
-    message: import("../schema/index.js").ChatMessage | import("../schema/index.js").HumanMessage | import("../schema/index.js").AIMessage | import("../schema/index.js").SystemMessage | import("../schema/index.js").FunctionMessage;
+    message: import("../schema/index.js").ChatMessage | import("../schema/index.js").HumanMessage | import("../schema/index.js").AIMessage | import("../schema/index.js").SystemMessage | import("../schema/index.js").FunctionMessage | import("../schema/index.js").ToolMessage;
 } | {
     text: string;
     message?: undefined;

package/dist/chat_models/bedrock/web.cjs

@@ -2,7 +2,6 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ChatBedrock = exports.BedrockChat = exports.convertMessagesToPrompt = exports.convertMessagesToPromptAnthropic = void 0;
 const signature_v4_1 = require("@smithy/signature-v4");
-const credential_provider_node_1 = require("@aws-sdk/credential-provider-node");
 const protocol_http_1 = require("@smithy/protocol-http");
 const eventstream_codec_1 = require("@smithy/eventstream-codec");
 const util_utf8_1 = require("@smithy/util-utf8");
@@ -168,7 +167,7 @@ class BedrockChat extends base_js_1.SimpleChatModel {
             throw new Error("Please set the AWS_DEFAULT_REGION environment variable or pass it to the constructor as the region field.");
         }
         this.region = region;
-        const credentials = fields?.credentials ?? (0, credential_provider_node_1.defaultProvider)();
+        const credentials = fields?.credentials;
         if (!credentials) {
             throw new Error("Please set the AWS credentials in the 'credentials' field.");
         }

package/dist/chat_models/bedrock/web.js

@@ -1,5 +1,4 @@
 import { SignatureV4 } from "@smithy/signature-v4";
-import { defaultProvider } from "@aws-sdk/credential-provider-node";
 import { HttpRequest } from "@smithy/protocol-http";
 import { EventStreamCodec } from "@smithy/eventstream-codec";
 import { fromUtf8, toUtf8 } from "@smithy/util-utf8";
@@ -163,7 +162,7 @@ export class BedrockChat extends SimpleChatModel {
             throw new Error("Please set the AWS_DEFAULT_REGION environment variable or pass it to the constructor as the region field.");
         }
         this.region = region;
-        const credentials = fields?.credentials ?? defaultProvider();
+        const credentials = fields?.credentials;
         if (!credentials) {
             throw new Error("Please set the AWS credentials in the 'credentials' field.");
         }

package/dist/chat_models/openai.cjs

@@ -14,7 +14,8 @@ function extractGenericMessageCustomRole(message) {
     if (message.role !== "system" &&
         message.role !== "assistant" &&
         message.role !== "user" &&
-        message.role !== "function") {
+        message.role !== "function" &&
+        message.role !== "tool") {
         console.warn(`Unknown message role: ${message.role}`);
     }
     return message.role;
@@ -30,6 +31,8 @@ function messageToOpenAIRole(message) {
             return "user";
         case "function":
             return "function";
+        case "tool":
+            return "tool";
         case "generic": {
             if (!index_js_1.ChatMessage.isInstance(message))
                 throw new Error("Invalid generic chat message");
@@ -39,24 +42,12 @@ function messageToOpenAIRole(message) {
             throw new Error(`Unknown message type: ${type}`);
     }
 }
-function messageToOpenAIMessage(message) {
-    const msg = {
-        content: message.content || null,
-        name: message.name,
-        role: messageToOpenAIRole(message),
-        function_call: message.additional_kwargs.function_call,
-    };
-    if (msg.function_call?.arguments) {
-        // Remove spaces, new line characters etc.
-        msg.function_call.arguments = JSON.stringify(JSON.parse(msg.function_call.arguments));
-    }
-    return msg;
-}
 function openAIResponseToChatMessage(message) {
     switch (message.role) {
         case "assistant":
             return new index_js_1.AIMessage(message.content || "", {
                 function_call: message.function_call,
+                tool_calls: message.tool_calls,
             });
         default:
             return new index_js_1.ChatMessage(message.content || "", message.role ?? "unknown");
@@ -73,6 +64,11 @@ delta, defaultRole) {
             function_call: delta.function_call,
         };
     }
+    else if (delta.tool_calls) {
+        additional_kwargs = {
+            tool_calls: delta.tool_calls,
+        };
+    }
     else {
         additional_kwargs = {};
     }
@@ -92,10 +88,28 @@ delta, defaultRole) {
             name: delta.name,
         });
     }
+    else if (role === "tool") {
+        return new index_js_1.ToolMessageChunk({
+            content,
+            additional_kwargs,
+            tool_call_id: delta.tool_call_id,
+        });
+    }
     else {
         return new index_js_1.ChatMessageChunk({ content, role });
     }
 }
+function convertMessagesToOpenAIParams(messages) {
+    // TODO: Function messages do not support array content, fix cast
+    return messages.map((message) => ({
+        role: messageToOpenAIRole(message),
+        content: message.content,
+        name: message.name,
+        function_call: message.additional_kwargs.function_call,
+        tool_calls: message.additional_kwargs.tool_calls,
+        tool_call_id: message.tool_call_id,
+    }));
+}
 /**
  * Wrapper around OpenAI large language models that use the Chat endpoint.
  *
@@ -126,6 +140,7 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
             "function_call",
             "functions",
             "tools",
+            "tool_choice",
             "promptIndex",
             "response_format",
             "seed",
@@ -355,7 +370,11 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
      * Get the parameters used to invoke the model
      */
     invocationParams(options) {
-        return {
+        function isStructuredToolArray(tools) {
+            return (tools !== undefined &&
+                tools.every((tool) => Array.isArray(tool.lc_namespace)));
+        }
+        const params = {
             model: this.modelName,
             temperature: this.temperature,
             top_p: this.topP,
@@ -367,15 +386,17 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
             stop: options?.stop ?? this.stop,
             user: this.user,
             stream: this.streaming,
-            functions: options?.functions ??
-                (options?.tools
-                    ? options?.tools.map(convert_to_openai_js_1.formatToOpenAIFunction)
-                    : undefined),
+            functions: options?.functions,
             function_call: options?.function_call,
+            tools: isStructuredToolArray(options?.tools)
+                ? options?.tools.map(convert_to_openai_js_1.formatToOpenAITool)
+                : options?.tools,
+            tool_choice: options?.tool_choice,
             response_format: options?.response_format,
             seed: options?.seed,
             ...this.modelKwargs,
         };
+        return params;
     }
     /** @ignore */
     _identifyingParams() {
@@ -386,13 +407,7 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
         };
     }
     async *_streamResponseChunks(messages, options, runManager) {
-        const messagesMapped = messages.map((message) => ({
-            role: messageToOpenAIRole(message),
-            content: message.content,
-            name: message.name,
-            function_call: message.additional_kwargs
-                .function_call,
-        }));
+        const messagesMapped = convertMessagesToOpenAIParams(messages);
         const params = {
             ...this.invocationParams(options),
             messages: messagesMapped,
@@ -412,6 +427,10 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
                 prompt: options.promptIndex ?? 0,
                 completion: choice.index ?? 0,
             };
+            if (typeof chunk.content !== "string") {
+                console.log("[WARNING]: Received non-string content from OpenAI. This is currently not supported.");
+                continue;
+            }
             const generationChunk = new index_js_1.ChatGenerationChunk({
                 message: chunk,
                 text: chunk.content,
@@ -436,13 +455,7 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
     async _generate(messages, options, runManager) {
         const tokenUsage = {};
         const params = this.invocationParams(options);
-        const messagesMapped = messages.map((message) => ({
-            role: messageToOpenAIRole(message),
-            content: message.content,
-            name: message.name,
-            function_call: message.additional_kwargs
-                .function_call,
-        }));
+        const messagesMapped = convertMessagesToOpenAIParams(messages);
         if (params.stream) {
             const stream = this._streamResponseChunks(messages, options, runManager);
             const finalChunks = {};
@@ -461,7 +474,7 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
             const { functions, function_call } = this.invocationParams(options);
             // OpenAI does not support token usage report under stream mode,
             // fallback to estimation.
-            const promptTokenUsage = await this.getNumTokensFromPrompt(messages, functions, function_call);
+            const promptTokenUsage = await this.getEstimatedTokenCountFromPrompt(messages, functions, function_call);
             const completionTokenUsage = await this.getNumTokensFromGenerations(generations);
             tokenUsage.promptTokens = promptTokenUsage;
             tokenUsage.completionTokens = completionTokenUsage;
@@ -510,11 +523,9 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
      * Estimate the number of tokens a prompt will use.
      * Modified from: https://github.com/hmarr/openai-chat-tokens/blob/main/src/index.ts
      */
-    async getNumTokensFromPrompt(messages, functions, function_call) {
+    async getEstimatedTokenCountFromPrompt(messages, functions, function_call) {
         // It appears that if functions are present, the first system message is padded with a trailing newline. This
         // was inferred by trying lots of combinations of messages and functions and seeing what the token counts were.
-        // let paddedSystem = false;
-        const openaiMessages = messages.map((m) => messageToOpenAIMessage(m));
         let tokens = (await this.getNumTokensFromMessages(messages)).totalCount;
         // If there are functions, add the function definitions as they count towards token usage
         if (functions && function_call !== "auto") {
@@ -525,7 +536,7 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
         // If there's a system message _and_ functions are present, subtract four tokens. I assume this is because
         // functions typically add a system message, but reuse the first one if it's already there. This offsets
         // the extra 9 tokens added by the function definitions.
-        if (functions && openaiMessages.find((m) => m.role === "system")) {
+        if (functions && messages.find((m) => m._getType() === "system")) {
             tokens -= 4;
         }
         // If function_call is 'none', add one token.
@@ -544,8 +555,7 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
      */
     async getNumTokensFromGenerations(generations) {
         const generationUsages = await Promise.all(generations.map(async (generation) => {
-            const openAIMessage = messageToOpenAIMessage(generation.message);
-            if (openAIMessage.function_call) {
+            if (generation.message.additional_kwargs?.function_call) {
                 return (await this.getNumTokensFromMessages([generation.message]))
                     .countPerMessage[0];
             }
@@ -576,19 +586,20 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
                 : 0;
             let count = textCount + tokensPerMessage + roleCount + nameCount;
             // From: https://github.com/hmarr/openai-chat-tokens/blob/main/src/index.ts messageTokenEstimate
-            const openAIMessage = messageToOpenAIMessage(message);
-            if (openAIMessage.role === "function" ||
-                openAIMessage.role === "tool") {
+            const openAIMessage = message;
+            if (openAIMessage._getType() === "function") {
                 count -= 2;
             }
-            if (openAIMessage.function_call) {
+            if (openAIMessage.additional_kwargs?.function_call) {
                 count += 3;
             }
-            if (openAIMessage.function_call?.name) {
-                count += await this.getNumTokens(openAIMessage.function_call?.name);
+            if (openAIMessage?.additional_kwargs.function_call?.name) {
+                count += await this.getNumTokens(openAIMessage.additional_kwargs.function_call?.name);
             }
-            if (openAIMessage.function_call?.arguments) {
-                count += await this.getNumTokens(openAIMessage.function_call?.arguments);
+            if (openAIMessage.additional_kwargs.function_call?.arguments) {
+                count += await this.getNumTokens(
+                // Remove newlines and spaces
+                JSON.stringify(JSON.parse(openAIMessage.additional_kwargs.function_call?.arguments)));
             }
             totalCount += count;
             return count;

package/dist/chat_models/openai.d.ts

@@ -15,7 +15,8 @@ interface OpenAILLMOutput {
     tokenUsage: TokenUsage;
 }
 export interface ChatOpenAICallOptions extends OpenAICallOptions, BaseFunctionCallOptions {
-    tools?: StructuredTool[];
+    tools?: StructuredTool[] | OpenAIClient.ChatCompletionTool[];
+    tool_choice?: OpenAIClient.ChatCompletionToolChoiceOption;
     promptIndex?: number;
     response_format?: {
         type: "json_object";
@@ -98,7 +99,7 @@ export declare class ChatOpenAI<CallOptions extends ChatOpenAICallOptions = Chat
      * Estimate the number of tokens a prompt will use.
      * Modified from: https://github.com/hmarr/openai-chat-tokens/blob/main/src/index.ts
      */
-    private getNumTokensFromPrompt;
+    private getEstimatedTokenCountFromPrompt;
     /**
      * Estimate the number of tokens an array of generations have used.
      */

package/dist/chat_models/openai.js

@@ -1,6 +1,6 @@
 import { OpenAI as OpenAIClient } from "openai";
-import { AIMessage, AIMessageChunk, ChatGenerationChunk, ChatMessage, ChatMessageChunk, FunctionMessageChunk, HumanMessageChunk, SystemMessageChunk, } from "../schema/index.js";
-import { formatToOpenAIFunction } from "../tools/convert_to_openai.js";
+import { AIMessage, AIMessageChunk, ChatGenerationChunk, ChatMessage, ChatMessageChunk, FunctionMessageChunk, HumanMessageChunk, SystemMessageChunk, ToolMessageChunk, } from "../schema/index.js";
+import { formatToOpenAITool } from "../tools/convert_to_openai.js";
 import { getEndpoint } from "../util/azure.js";
 import { getEnvironmentVariable } from "../util/env.js";
 import { promptLayerTrackRequest } from "../util/prompt-layer.js";
@@ -11,7 +11,8 @@ function extractGenericMessageCustomRole(message) {
     if (message.role !== "system" &&
         message.role !== "assistant" &&
         message.role !== "user" &&
-        message.role !== "function") {
+        message.role !== "function" &&
+        message.role !== "tool") {
         console.warn(`Unknown message role: ${message.role}`);
     }
     return message.role;
@@ -27,6 +28,8 @@ function messageToOpenAIRole(message) {
             return "user";
         case "function":
             return "function";
+        case "tool":
+            return "tool";
         case "generic": {
             if (!ChatMessage.isInstance(message))
                 throw new Error("Invalid generic chat message");
@@ -36,24 +39,12 @@ function messageToOpenAIRole(message) {
             throw new Error(`Unknown message type: ${type}`);
     }
 }
-function messageToOpenAIMessage(message) {
-    const msg = {
-        content: message.content || null,
-        name: message.name,
-        role: messageToOpenAIRole(message),
-        function_call: message.additional_kwargs.function_call,
-    };
-    if (msg.function_call?.arguments) {
-        // Remove spaces, new line characters etc.
-        msg.function_call.arguments = JSON.stringify(JSON.parse(msg.function_call.arguments));
-    }
-    return msg;
-}
 function openAIResponseToChatMessage(message) {
     switch (message.role) {
         case "assistant":
             return new AIMessage(message.content || "", {
                 function_call: message.function_call,
+                tool_calls: message.tool_calls,
             });
         default:
             return new ChatMessage(message.content || "", message.role ?? "unknown");
@@ -70,6 +61,11 @@ delta, defaultRole) {
             function_call: delta.function_call,
         };
     }
+    else if (delta.tool_calls) {
+        additional_kwargs = {
+            tool_calls: delta.tool_calls,
+        };
+    }
     else {
         additional_kwargs = {};
     }
@@ -89,10 +85,28 @@ delta, defaultRole) {
             name: delta.name,
         });
     }
+    else if (role === "tool") {
+        return new ToolMessageChunk({
+            content,
+            additional_kwargs,
+            tool_call_id: delta.tool_call_id,
+        });
+    }
     else {
         return new ChatMessageChunk({ content, role });
     }
 }
+function convertMessagesToOpenAIParams(messages) {
+    // TODO: Function messages do not support array content, fix cast
+    return messages.map((message) => ({
+        role: messageToOpenAIRole(message),
+        content: message.content,
+        name: message.name,
+        function_call: message.additional_kwargs.function_call,
+        tool_calls: message.additional_kwargs.tool_calls,
+        tool_call_id: message.tool_call_id,
+    }));
+}
 /**
  * Wrapper around OpenAI large language models that use the Chat endpoint.
  *
@@ -123,6 +137,7 @@ export class ChatOpenAI extends BaseChatModel {
             "function_call",
             "functions",
             "tools",
+            "tool_choice",
             "promptIndex",
             "response_format",
             "seed",
@@ -352,7 +367,11 @@ export class ChatOpenAI extends BaseChatModel {
      * Get the parameters used to invoke the model
      */
     invocationParams(options) {
-        return {
+        function isStructuredToolArray(tools) {
+            return (tools !== undefined &&
+                tools.every((tool) => Array.isArray(tool.lc_namespace)));
+        }
+        const params = {
             model: this.modelName,
             temperature: this.temperature,
             top_p: this.topP,
@@ -364,15 +383,17 @@ export class ChatOpenAI extends BaseChatModel {
             stop: options?.stop ?? this.stop,
             user: this.user,
             stream: this.streaming,
-            functions: options?.functions ??
-                (options?.tools
-                    ? options?.tools.map(formatToOpenAIFunction)
-                    : undefined),
+            functions: options?.functions,
             function_call: options?.function_call,
+            tools: isStructuredToolArray(options?.tools)
+                ? options?.tools.map(formatToOpenAITool)
+                : options?.tools,
+            tool_choice: options?.tool_choice,
             response_format: options?.response_format,
             seed: options?.seed,
             ...this.modelKwargs,
         };
+        return params;
     }
     /** @ignore */
     _identifyingParams() {
@@ -383,13 +404,7 @@ export class ChatOpenAI extends BaseChatModel {
         };
     }
     async *_streamResponseChunks(messages, options, runManager) {
-        const messagesMapped = messages.map((message) => ({
-            role: messageToOpenAIRole(message),
-            content: message.content,
-            name: message.name,
-            function_call: message.additional_kwargs
-                .function_call,
-        }));
+        const messagesMapped = convertMessagesToOpenAIParams(messages);
         const params = {
             ...this.invocationParams(options),
             messages: messagesMapped,
@@ -409,6 +424,10 @@ export class ChatOpenAI extends BaseChatModel {
                 prompt: options.promptIndex ?? 0,
                 completion: choice.index ?? 0,
             };
+            if (typeof chunk.content !== "string") {
+                console.log("[WARNING]: Received non-string content from OpenAI. This is currently not supported.");
+                continue;
+            }
             const generationChunk = new ChatGenerationChunk({
                 message: chunk,
                 text: chunk.content,
@@ -433,13 +452,7 @@ export class ChatOpenAI extends BaseChatModel {
     async _generate(messages, options, runManager) {
         const tokenUsage = {};
         const params = this.invocationParams(options);
-        const messagesMapped = messages.map((message) => ({
-            role: messageToOpenAIRole(message),
-            content: message.content,
-            name: message.name,
-            function_call: message.additional_kwargs
-                .function_call,
-        }));
+        const messagesMapped = convertMessagesToOpenAIParams(messages);
         if (params.stream) {
             const stream = this._streamResponseChunks(messages, options, runManager);
             const finalChunks = {};
@@ -458,7 +471,7 @@ export class ChatOpenAI extends BaseChatModel {
             const { functions, function_call } = this.invocationParams(options);
             // OpenAI does not support token usage report under stream mode,
             // fallback to estimation.
-            const promptTokenUsage = await this.getNumTokensFromPrompt(messages, functions, function_call);
+            const promptTokenUsage = await this.getEstimatedTokenCountFromPrompt(messages, functions, function_call);
             const completionTokenUsage = await this.getNumTokensFromGenerations(generations);
             tokenUsage.promptTokens = promptTokenUsage;
             tokenUsage.completionTokens = completionTokenUsage;
@@ -507,11 +520,9 @@ export class ChatOpenAI extends BaseChatModel {
      * Estimate the number of tokens a prompt will use.
      * Modified from: https://github.com/hmarr/openai-chat-tokens/blob/main/src/index.ts
      */
-    async getNumTokensFromPrompt(messages, functions, function_call) {
+    async getEstimatedTokenCountFromPrompt(messages, functions, function_call) {
         // It appears that if functions are present, the first system message is padded with a trailing newline. This
         // was inferred by trying lots of combinations of messages and functions and seeing what the token counts were.
-        // let paddedSystem = false;
-        const openaiMessages = messages.map((m) => messageToOpenAIMessage(m));
         let tokens = (await this.getNumTokensFromMessages(messages)).totalCount;
         // If there are functions, add the function definitions as they count towards token usage
         if (functions && function_call !== "auto") {
@@ -522,7 +533,7 @@ export class ChatOpenAI extends BaseChatModel {
         // If there's a system message _and_ functions are present, subtract four tokens. I assume this is because
         // functions typically add a system message, but reuse the first one if it's already there. This offsets
         // the extra 9 tokens added by the function definitions.
-        if (functions && openaiMessages.find((m) => m.role === "system")) {
+        if (functions && messages.find((m) => m._getType() === "system")) {
             tokens -= 4;
         }
         // If function_call is 'none', add one token.
@@ -541,8 +552,7 @@ export class ChatOpenAI extends BaseChatModel {
      */
     async getNumTokensFromGenerations(generations) {
         const generationUsages = await Promise.all(generations.map(async (generation) => {
-            const openAIMessage = messageToOpenAIMessage(generation.message);
-            if (openAIMessage.function_call) {
+            if (generation.message.additional_kwargs?.function_call) {
                 return (await this.getNumTokensFromMessages([generation.message]))
                     .countPerMessage[0];
             }
@@ -573,19 +583,20 @@ export class ChatOpenAI extends BaseChatModel {
                 : 0;
             let count = textCount + tokensPerMessage + roleCount + nameCount;
             // From: https://github.com/hmarr/openai-chat-tokens/blob/main/src/index.ts messageTokenEstimate
-            const openAIMessage = messageToOpenAIMessage(message);
-            if (openAIMessage.role === "function" ||
-                openAIMessage.role === "tool") {
+            const openAIMessage = message;
+            if (openAIMessage._getType() === "function") {
                 count -= 2;
             }
-            if (openAIMessage.function_call) {
+            if (openAIMessage.additional_kwargs?.function_call) {
                 count += 3;
             }
-            if (openAIMessage.function_call?.name) {
-                count += await this.getNumTokens(openAIMessage.function_call?.name);
+            if (openAIMessage?.additional_kwargs.function_call?.name) {
+                count += await this.getNumTokens(openAIMessage.additional_kwargs.function_call?.name);
             }
-            if (openAIMessage.function_call?.arguments) {
-                count += await this.getNumTokens(openAIMessage.function_call?.arguments);
+            if (openAIMessage.additional_kwargs.function_call?.arguments) {
+                count += await this.getNumTokens(
+                // Remove newlines and spaces
+                JSON.stringify(JSON.parse(openAIMessage.additional_kwargs.function_call?.arguments)));
             }
             totalCount += count;
             return count;

package/dist/document_loaders/web/apify_dataset.cjs

@@ -1,4 +1,5 @@
 "use strict";
+/* eslint-disable @typescript-eslint/no-explicit-any */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ApifyDatasetLoader = void 0;
 const apify_client_1 = require("apify-client");

package/dist/document_loaders/web/apify_dataset.d.ts

@@ -1,22 +1,22 @@
-import { ApifyClient, ApifyClientOptions, ActorCallOptions, TaskCallOptions } from "apify-client";
-import { Document } from "../../document.js";
+import { ActorCallOptions, ApifyClient, ApifyClientOptions, TaskCallOptions } from "apify-client";
 import { BaseDocumentLoader, DocumentLoader } from "../base.js";
+import { Document } from "../../document.js";
 /**
  * A type that represents a function that takes a single object (an Apify
  * dataset item) and converts it to an instance of the Document class.
  */
-export type ApifyDatasetMappingFunction = (item: Record<string | number, unknown>) => Document;
+export type ApifyDatasetMappingFunction<Metadata extends Record<string, any>> = (item: Record<string | number, unknown>) => Document<Metadata>;
 /**
  * A class that extends the BaseDocumentLoader and implements the
  * DocumentLoader interface. It represents a document loader that loads
  * documents from an Apify dataset.
  */
-export declare class ApifyDatasetLoader extends BaseDocumentLoader implements DocumentLoader {
+export declare class ApifyDatasetLoader<Metadata extends Record<string, any>> extends BaseDocumentLoader implements DocumentLoader {
     protected apifyClient: ApifyClient;
     protected datasetId: string;
-    protected datasetMappingFunction: (item: Record<string | number, unknown>) => Document;
+    protected datasetMappingFunction: (item: Record<string | number, unknown>) => Document<Metadata>;
     constructor(datasetId: string, config: {
-        datasetMappingFunction: ApifyDatasetMappingFunction;
+        datasetMappingFunction: ApifyDatasetMappingFunction<Metadata>;
         clientOptions?: ApifyClientOptions;
     });
     private static _getApifyApiToken;
@@ -26,7 +26,7 @@ export declare class ApifyDatasetLoader extends BaseDocumentLoader implements Do
      * instances.
      * @returns An array of Document instances.
      */
-    load(): Promise<Document[]>;
+    load(): Promise<Document<Metadata>[]>;
     /**
      * Create an ApifyDatasetLoader by calling an Actor on the Apify platform and waiting for its results to be ready.
      * @param actorId The ID or name of the Actor on the Apify platform.
@@ -35,11 +35,11 @@ export declare class ApifyDatasetLoader extends BaseDocumentLoader implements Do
      * @param options.datasetMappingFunction A function that takes a single object (an Apify dataset item) and converts it to an instance of the Document class.
      * @returns An instance of `ApifyDatasetLoader` with the results from the Actor run.
      */
-    static fromActorCall(actorId: string, input: Record<string | number, unknown>, config: {
+    static fromActorCall<Metadata extends Record<string, any>>(actorId: string, input: Record<string | number, unknown>, config: {
         callOptions?: ActorCallOptions;
         clientOptions?: ApifyClientOptions;
-        datasetMappingFunction: ApifyDatasetMappingFunction;
-    }): Promise<ApifyDatasetLoader>;
+        datasetMappingFunction: ApifyDatasetMappingFunction<Metadata>;
+    }): Promise<ApifyDatasetLoader<Metadata>>;
     /**
      * Create an ApifyDatasetLoader by calling a saved Actor task on the Apify platform and waiting for its results to be ready.
      * @param taskId The ID or name of the task on the Apify platform.
@@ -48,9 +48,9 @@ export declare class ApifyDatasetLoader extends BaseDocumentLoader implements Do
      * @param options.datasetMappingFunction A function that takes a single object (an Apify dataset item) and converts it to an instance of the Document class.
      * @returns An instance of `ApifyDatasetLoader` with the results from the task's run.
      */
-    static fromActorTaskCall(taskId: string, input: Record<string | number, unknown>, config: {
+    static fromActorTaskCall<Metadata extends Record<string, any>>(taskId: string, input: Record<string | number, unknown>, config: {
         callOptions?: TaskCallOptions;
         clientOptions?: ApifyClientOptions;
-        datasetMappingFunction: ApifyDatasetMappingFunction;
-    }): Promise<ApifyDatasetLoader>;
+        datasetMappingFunction: ApifyDatasetMappingFunction<Metadata>;
+    }): Promise<ApifyDatasetLoader<Metadata>>;
 }

package/dist/document_loaders/web/apify_dataset.js

@@ -1,3 +1,4 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
 import { ApifyClient, } from "apify-client";
 import { BaseDocumentLoader } from "../base.js";
 import { getEnvironmentVariable } from "../../util/env.js";

package/dist/runnables/remote.cjs

@@ -52,6 +52,12 @@ function revive(obj) {
                         name: obj.name,
                     });
                 }
+                if (obj.type === "tool") {
+                    return new index_js_2.ToolMessage({
+                        content: obj.content,
+                        tool_call_id: obj.tool_call_id,
+                    });
+                }
                 if (obj.type === "ai") {
                     return new index_js_2.AIMessage({
                         content: obj.content,
@@ -81,6 +87,12 @@ function revive(obj) {
                         name: obj.name,
                     });
                 }
+                if (obj.type === "tool") {
+                    return new index_js_2.ToolMessageChunk({
+                        content: obj.content,
+                        tool_call_id: obj.tool_call_id,
+                    });
+                }
                 if (obj.type === "ai") {
                     return new index_js_2.AIMessageChunk({
                         content: obj.content,

package/dist/runnables/remote.js

@@ -1,7 +1,7 @@
 import { Runnable } from "../schema/runnable/index.js";
 import { getBytes, getLines, getMessages } from "../util/event-source-parse.js";
 import { Document } from "../document.js";
-import { AIMessage, AIMessageChunk, ChatMessage, ChatMessageChunk, FunctionMessage, FunctionMessageChunk, HumanMessage, HumanMessageChunk, SystemMessage, SystemMessageChunk, } from "../schema/index.js";
+import { AIMessage, AIMessageChunk, ChatMessage, ChatMessageChunk, FunctionMessage, FunctionMessageChunk, HumanMessage, HumanMessageChunk, SystemMessage, SystemMessageChunk, ToolMessage, ToolMessageChunk, } from "../schema/index.js";
 import { StringPromptValue } from "../prompts/base.js";
 import { ChatPromptValue } from "../prompts/chat.js";
 import { IterableReadableStream } from "../util/stream.js";
@@ -49,6 +49,12 @@ function revive(obj) {
                         name: obj.name,
                     });
                 }
+                if (obj.type === "tool") {
+                    return new ToolMessage({
+                        content: obj.content,
+                        tool_call_id: obj.tool_call_id,
+                    });
+                }
                 if (obj.type === "ai") {
                     return new AIMessage({
                         content: obj.content,
@@ -78,6 +84,12 @@ function revive(obj) {
                         name: obj.name,
                     });
                 }
+                if (obj.type === "tool") {
+                    return new ToolMessageChunk({
+                        content: obj.content,
+                        tool_call_id: obj.tool_call_id,
+                    });
+                }
                 if (obj.type === "ai") {
                     return new AIMessageChunk({
                         content: obj.content,

package/dist/schema/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Docstore = exports.BaseEntityStore = exports.BaseFileStore = exports.BaseCache = exports.BaseListChatMessageHistory = exports.BaseChatMessageHistory = exports.BasePromptValue = exports.mapStoredMessageToChatMessage = exports.ChatGenerationChunk = exports.ChatMessageChunk = exports.coerceMessageLikeToMessage = exports.isBaseMessageChunk = exports.isBaseMessage = exports.ChatMessage = exports.FunctionMessageChunk = exports.FunctionMessage = exports.SystemChatMessage = exports.AIChatMessage = exports.HumanChatMessage = exports.BaseChatMessage = exports.SystemMessageChunk = exports.SystemMessage = exports.AIMessageChunk = exports.AIMessage = exports.HumanMessageChunk = exports.HumanMessage = exports.BaseMessageChunk = exports.BaseMessage = exports.GenerationChunk = exports.RUN_KEY = void 0;
+exports.Docstore = exports.BaseEntityStore = exports.BaseFileStore = exports.BaseCache = exports.BaseListChatMessageHistory = exports.BaseChatMessageHistory = exports.BasePromptValue = exports.mapStoredMessageToChatMessage = exports.ChatGenerationChunk = exports.ChatMessageChunk = exports.coerceMessageLikeToMessage = exports.isBaseMessageChunk = exports.isBaseMessage = exports.ChatMessage = exports.ToolMessageChunk = exports.ToolMessage = exports.FunctionMessageChunk = exports.FunctionMessage = exports.SystemChatMessage = exports.AIChatMessage = exports.HumanChatMessage = exports.BaseChatMessage = exports.SystemMessageChunk = exports.SystemMessage = exports.AIMessageChunk = exports.AIMessage = exports.HumanMessageChunk = exports.HumanMessage = exports.BaseMessageChunk = exports.BaseMessage = exports.GenerationChunk = exports.RUN_KEY = void 0;
 const serializable_js_1 = require("../load/serializable.cjs");
 exports.RUN_KEY = "__run";
 /**
@@ -154,6 +154,10 @@ class BaseMessage extends serializable_js_1.Serializable {
     }
 }
 exports.BaseMessage = BaseMessage;
+function isOpenAIToolCallArray(value) {
+    return (Array.isArray(value) &&
+        value.every((v) => typeof v.index === "number"));
+}
 /**
  * Represents a chunk of a message, which can be concatenated with other
  * message chunks. It includes a method `_merge_kwargs_dict()` for merging
@@ -178,6 +182,31 @@ class BaseMessageChunk extends BaseMessage {
                 typeof merged[key] === "object") {
                 merged[key] = this._mergeAdditionalKwargs(merged[key], value);
             }
+            else if (key === "tool_calls" &&
+                isOpenAIToolCallArray(merged[key]) &&
+                isOpenAIToolCallArray(value)) {
+                for (const toolCall of value) {
+                    if (merged[key]?.[toolCall.index] !== undefined) {
+                        merged[key] = merged[key]?.map((value, i) => {
+                            if (i !== toolCall.index) {
+                                return value;
+                            }
+                            return {
+                                ...value,
+                                ...toolCall,
+                                function: {
+                                    name: toolCall.function.name ?? value.function.name,
+                                    arguments: (value.function.arguments ?? "") +
+                                        (toolCall.function.arguments ?? ""),
+                                },
+                            };
+                        });
+                    }
+                    else {
+                        merged[key][toolCall.index] = toolCall;
+                    }
+                }
+            }
             else {
                 throw new Error(`additional_kwargs[${key}] already exists in this message chunk.`);
             }
@@ -340,6 +369,62 @@ class FunctionMessageChunk extends BaseMessageChunk {
     }
 }
 exports.FunctionMessageChunk = FunctionMessageChunk;
+/**
+ * Represents a tool message in a conversation.
+ */
+class ToolMessage extends BaseMessage {
+    static lc_name() {
+        return "ToolMessage";
+    }
+    constructor(fields, tool_call_id, name) {
+        if (typeof fields === "string") {
+            // eslint-disable-next-line no-param-reassign, @typescript-eslint/no-non-null-assertion
+            fields = { content: fields, name, tool_call_id: tool_call_id };
+        }
+        super(fields);
+        Object.defineProperty(this, "tool_call_id", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.tool_call_id = fields.tool_call_id;
+    }
+    _getType() {
+        return "tool";
+    }
+}
+exports.ToolMessage = ToolMessage;
+/**
+ * Represents a chunk of a function message, which can be concatenated
+ * with other function message chunks.
+ */
+class ToolMessageChunk extends BaseMessageChunk {
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "tool_call_id", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.tool_call_id = fields.tool_call_id;
+    }
+    static lc_name() {
+        return "ToolMessageChunk";
+    }
+    _getType() {
+        return "tool";
+    }
+    concat(chunk) {
+        return new ToolMessageChunk({
+            content: mergeContent(this.content, chunk.content),
+            additional_kwargs: ToolMessageChunk._mergeAdditionalKwargs(this.additional_kwargs, chunk.additional_kwargs),
+            tool_call_id: this.tool_call_id,
+        });
+    }
+}
+exports.ToolMessageChunk = ToolMessageChunk;
 /**
  * Represents a chat message in a conversation.
  */
@@ -477,6 +562,7 @@ function mapV1MessageToStoredMessage(message) {
                 content: v1Message.text,
                 role: v1Message.role,
                 name: undefined,
+                tool_call_id: undefined,
             },
         };
     }
@@ -495,6 +581,11 @@ function mapStoredMessageToChatMessage(message) {
                 throw new Error("Name must be defined for function messages");
             }
             return new FunctionMessage(storedMessage.data);
+        case "tool":
+            if (storedMessage.data.tool_call_id === undefined) {
+                throw new Error("Tool call ID must be defined for tool messages");
+            }
+            return new ToolMessage(storedMessage.data);
         case "chat": {
             if (storedMessage.data.role === undefined) {
                 throw new Error("Role must be defined for chat messages");

package/dist/schema/index.d.ts

@@ -53,6 +53,7 @@ export interface StoredMessageData {
     content: string;
     role: string | undefined;
     name: string | undefined;
+    tool_call_id: string | undefined;
     additional_kwargs?: Record<string, any>;
 }
 export interface StoredMessage {
@@ -63,7 +64,7 @@ export interface StoredGeneration {
     text: string;
     message?: StoredMessage;
 }
-export type MessageType = "human" | "ai" | "generic" | "system" | "function";
+export type MessageType = "human" | "ai" | "generic" | "system" | "function" | "tool";
 export type MessageContent = string | {
     type: "text" | "image_url";
     text?: string;
@@ -76,6 +77,7 @@ export interface BaseMessageFields {
     name?: string;
     additional_kwargs?: {
         function_call?: OpenAIClient.Chat.ChatCompletionMessage.FunctionCall;
+        tool_calls?: OpenAIClient.Chat.ChatCompletionMessageToolCall[];
         [key: string]: unknown;
     };
 }
@@ -85,6 +87,9 @@ export interface ChatMessageFieldsWithRole extends BaseMessageFields {
 export interface FunctionMessageFieldsWithName extends BaseMessageFields {
     name: string;
 }
+export interface ToolMessageFieldsWithToolCallId extends BaseMessageFields {
+    tool_call_id: string;
+}
 /**
  * Base class for all types of messages in a conversation. It includes
  * properties like `content`, `name`, and `additional_kwargs`. It also
@@ -112,6 +117,9 @@ export declare abstract class BaseMessage extends Serializable implements BaseMe
     toDict(): StoredMessage;
     toChunk(): BaseMessageChunk;
 }
+export type OpenAIToolCall = OpenAIClient.ChatCompletionMessageToolCall & {
+    index: number;
+};
 /**
  * Represents a chunk of a message, which can be concatenated with other
  * message chunks. It includes a method `_merge_kwargs_dict()` for merging
@@ -211,6 +219,27 @@ export declare class FunctionMessageChunk extends BaseMessageChunk {
     _getType(): MessageType;
     concat(chunk: FunctionMessageChunk): FunctionMessageChunk;
 }
+/**
+ * Represents a tool message in a conversation.
+ */
+export declare class ToolMessage extends BaseMessage {
+    static lc_name(): string;
+    tool_call_id: string;
+    constructor(fields: ToolMessageFieldsWithToolCallId);
+    constructor(fields: string | BaseMessageFields, tool_call_id: string, name?: string);
+    _getType(): MessageType;
+}
+/**
+ * Represents a chunk of a function message, which can be concatenated
+ * with other function message chunks.
+ */
+export declare class ToolMessageChunk extends BaseMessageChunk {
+    tool_call_id: string;
+    constructor(fields: ToolMessageFieldsWithToolCallId);
+    static lc_name(): string;
+    _getType(): MessageType;
+    concat(chunk: ToolMessageChunk): ToolMessageChunk;
+}
 /**
  * Represents a chat message in a conversation.
  */
@@ -252,7 +281,7 @@ export declare class ChatGenerationChunk extends GenerationChunk implements Chat
     constructor(fields: ChatGenerationChunkFields);
     concat(chunk: ChatGenerationChunk): ChatGenerationChunk;
 }
-export declare function mapStoredMessageToChatMessage(message: StoredMessage): ChatMessage | HumanMessage | AIMessage | SystemMessage | FunctionMessage;
+export declare function mapStoredMessageToChatMessage(message: StoredMessage): ChatMessage | HumanMessage | AIMessage | SystemMessage | FunctionMessage | ToolMessage;
 export interface ChatResult {
     generations: ChatGeneration[];
     llmOutput?: Record<string, any>;

package/dist/schema/index.js

@@ -149,6 +149,10 @@ export class BaseMessage extends Serializable {
         }
     }
 }
+function isOpenAIToolCallArray(value) {
+    return (Array.isArray(value) &&
+        value.every((v) => typeof v.index === "number"));
+}
 /**
  * Represents a chunk of a message, which can be concatenated with other
  * message chunks. It includes a method `_merge_kwargs_dict()` for merging
@@ -173,6 +177,31 @@ export class BaseMessageChunk extends BaseMessage {
                 typeof merged[key] === "object") {
                 merged[key] = this._mergeAdditionalKwargs(merged[key], value);
             }
+            else if (key === "tool_calls" &&
+                isOpenAIToolCallArray(merged[key]) &&
+                isOpenAIToolCallArray(value)) {
+                for (const toolCall of value) {
+                    if (merged[key]?.[toolCall.index] !== undefined) {
+                        merged[key] = merged[key]?.map((value, i) => {
+                            if (i !== toolCall.index) {
+                                return value;
+                            }
+                            return {
+                                ...value,
+                                ...toolCall,
+                                function: {
+                                    name: toolCall.function.name ?? value.function.name,
+                                    arguments: (value.function.arguments ?? "") +
+                                        (toolCall.function.arguments ?? ""),
+                                },
+                            };
+                        });
+                    }
+                    else {
+                        merged[key][toolCall.index] = toolCall;
+                    }
+                }
+            }
             else {
                 throw new Error(`additional_kwargs[${key}] already exists in this message chunk.`);
             }
@@ -326,6 +355,60 @@ export class FunctionMessageChunk extends BaseMessageChunk {
         });
     }
 }
+/**
+ * Represents a tool message in a conversation.
+ */
+export class ToolMessage extends BaseMessage {
+    static lc_name() {
+        return "ToolMessage";
+    }
+    constructor(fields, tool_call_id, name) {
+        if (typeof fields === "string") {
+            // eslint-disable-next-line no-param-reassign, @typescript-eslint/no-non-null-assertion
+            fields = { content: fields, name, tool_call_id: tool_call_id };
+        }
+        super(fields);
+        Object.defineProperty(this, "tool_call_id", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.tool_call_id = fields.tool_call_id;
+    }
+    _getType() {
+        return "tool";
+    }
+}
+/**
+ * Represents a chunk of a function message, which can be concatenated
+ * with other function message chunks.
+ */
+export class ToolMessageChunk extends BaseMessageChunk {
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "tool_call_id", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.tool_call_id = fields.tool_call_id;
+    }
+    static lc_name() {
+        return "ToolMessageChunk";
+    }
+    _getType() {
+        return "tool";
+    }
+    concat(chunk) {
+        return new ToolMessageChunk({
+            content: mergeContent(this.content, chunk.content),
+            additional_kwargs: ToolMessageChunk._mergeAdditionalKwargs(this.additional_kwargs, chunk.additional_kwargs),
+            tool_call_id: this.tool_call_id,
+        });
+    }
+}
 /**
  * Represents a chat message in a conversation.
  */
@@ -457,6 +540,7 @@ function mapV1MessageToStoredMessage(message) {
                 content: v1Message.text,
                 role: v1Message.role,
                 name: undefined,
+                tool_call_id: undefined,
             },
         };
     }
@@ -475,6 +559,11 @@ export function mapStoredMessageToChatMessage(message) {
                 throw new Error("Name must be defined for function messages");
             }
             return new FunctionMessage(storedMessage.data);
+        case "tool":
+            if (storedMessage.data.tool_call_id === undefined) {
+                throw new Error("Tool call ID must be defined for tool messages");
+            }
+            return new ToolMessage(storedMessage.data);
         case "chat": {
             if (storedMessage.data.role === undefined) {
                 throw new Error("Role must be defined for chat messages");

package/dist/stores/message/convex.d.ts

@@ -1,11 +1,11 @@
-import { DocumentByName, FieldPaths, FunctionReference, GenericActionCtx, GenericDataModel, NamedTableInfo, TableNamesInDataModel, VectorIndexNames } from "convex/server";
+import { DocumentByName, FieldPaths, FunctionReference, GenericActionCtx, GenericDataModel, NamedTableInfo, TableNamesInDataModel, IndexNames } from "convex/server";
 import { BaseMessage, BaseListChatMessageHistory } from "../../schema/index.js";
 /**
  * Type that defines the config required to initialize the
  * ConvexChatMessageHistory class. At minimum it needs a sessionId
  * and an ActionCtx.
  */
-export type ConvexChatMessageHistoryInput<DataModel extends GenericDataModel, TableName extends TableNamesInDataModel<DataModel> = "messages", IndexName extends VectorIndexNames<NamedTableInfo<DataModel, TableName>> = "bySessionId", SessionIdFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "sessionId", MessageTextFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "message", InsertMutation extends FunctionReference<"mutation", "internal", {
+export type ConvexChatMessageHistoryInput<DataModel extends GenericDataModel, TableName extends TableNamesInDataModel<DataModel> = "messages", IndexName extends IndexNames<NamedTableInfo<DataModel, TableName>> = "bySessionId", SessionIdFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "sessionId", MessageTextFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "message", InsertMutation extends FunctionReference<"mutation", "internal", {
     table: string;
     document: object;
 }> = any, LookupQuery extends FunctionReference<"query", "internal", {
@@ -50,7 +50,7 @@ export type ConvexChatMessageHistoryInput<DataModel extends GenericDataModel, Ta
      */
     readonly deleteMany?: DeleteManyMutation;
 };
-export declare class ConvexChatMessageHistory<DataModel extends GenericDataModel, SessionIdFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "sessionId", TableName extends TableNamesInDataModel<DataModel> = "messages", IndexName extends VectorIndexNames<NamedTableInfo<DataModel, TableName>> = "bySessionId", MessageTextFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "message", InsertMutation extends FunctionReference<"mutation", "internal", {
+export declare class ConvexChatMessageHistory<DataModel extends GenericDataModel, SessionIdFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "sessionId", TableName extends TableNamesInDataModel<DataModel> = "messages", IndexName extends IndexNames<NamedTableInfo<DataModel, TableName>> = "bySessionId", MessageTextFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "message", InsertMutation extends FunctionReference<"mutation", "internal", {
     table: string;
     document: object;
 }> = any, LookupQuery extends FunctionReference<"query", "internal", {

package/dist/tools/convert_to_openai.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.formatToOpenAIFunction = void 0;
+exports.formatToOpenAITool = exports.formatToOpenAIFunction = void 0;
 const zod_to_json_schema_1 = require("zod-to-json-schema");
 /**
  * Formats a `StructuredTool` instance into a format that is compatible
@@ -16,3 +16,14 @@ function formatToOpenAIFunction(tool) {
     };
 }
 exports.formatToOpenAIFunction = formatToOpenAIFunction;
+function formatToOpenAITool(tool) {
+    return {
+        type: "function",
+        function: {
+            name: tool.name,
+            description: tool.description,
+            parameters: (0, zod_to_json_schema_1.zodToJsonSchema)(tool.schema),
+        },
+    };
+}
+exports.formatToOpenAITool = formatToOpenAITool;

package/dist/tools/convert_to_openai.d.ts

@@ -7,3 +7,4 @@ import { StructuredTool } from "./base.js";
  * schema, which is then used as the parameters for the OpenAI function.
  */
 export declare function formatToOpenAIFunction(tool: StructuredTool): OpenAIClient.Chat.ChatCompletionCreateParams.Function;
+export declare function formatToOpenAITool(tool: StructuredTool): OpenAIClient.Chat.ChatCompletionTool;

package/dist/tools/convert_to_openai.js

@@ -12,3 +12,13 @@ export function formatToOpenAIFunction(tool) {
         parameters: zodToJsonSchema(tool.schema),
     };
 }
+export function formatToOpenAITool(tool) {
+    return {
+        type: "function",
+        function: {
+            name: tool.name,
+            description: tool.description,
+            parameters: zodToJsonSchema(tool.schema),
+        },
+    };
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langchain",
-  "version": "0.0.182-rc.1",
+  "version": "0.0.182",
   "description": "Typescript bindings for langchain",
   "type": "module",
   "engines": {
@@ -796,10 +796,11 @@
     "url": "git@github.com:langchain-ai/langchainjs.git"
   },
   "scripts": {
-    "build": "yarn clean && yarn build:esm && yarn build:cjs && node scripts/create-entrypoints.js && node scripts/check-tree-shaking.js",
+    "build": "yarn clean && yarn build:esm && yarn build:cjs && yarn build:scripts",
     "build:esm": "NODE_OPTIONS=--max-old-space-size=4096 tsc --outDir dist/ && rimraf dist/tests dist/**/tests",
     "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": "NODE_OPTIONS=--max-old-space-size=4096 eslint src && dpdm --exit-code circular:1 --no-warning --no-tree src/*.ts src/**/*.ts",
     "lint:fix": "yarn lint --fix",
     "precommit": "lint-staged",
@@ -1360,7 +1361,7 @@
     "langchainhub": "~0.0.6",
     "langsmith": "~0.0.48",
     "ml-distance": "^4.0.0",
-    "openai": "~4.16.0",
+    "openai": "^4.16.1",
     "openapi-types": "^12.1.3",
     "p-queue": "^6.6.2",
     "p-retry": "4",