@langchain/openai 0.1.2 → 0.2.0

package/dist/chat_models.cjs

@@ -47,7 +47,7 @@ function messageToOpenAIRole(message) {
     }
 }
 exports.messageToOpenAIRole = messageToOpenAIRole;
-function openAIResponseToChatMessage(message) {
+function openAIResponseToChatMessage(message, messageId) {
     const rawToolCalls = message.tool_calls;
     switch (message.role) {
         case "assistant": {
@@ -70,6 +70,7 @@ function openAIResponseToChatMessage(message) {
                     function_call: message.function_call,
                     tool_calls: rawToolCalls,
                 },
+                id: messageId,
             });
         }
         default:
@@ -78,7 +79,7 @@ function openAIResponseToChatMessage(message) {
 }
 function _convertDeltaToMessageChunk(
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-delta, defaultRole) {
+delta, messageId, defaultRole) {
     const role = delta.role ?? defaultRole;
     const content = delta.content ?? "";
     let additional_kwargs;
@@ -114,6 +115,7 @@ delta, defaultRole) {
             content,
             tool_call_chunks: toolCallChunks,
             additional_kwargs,
+            id: messageId,
         });
     }
     else if (role === "system") {
@@ -150,9 +152,11 @@ function convertMessagesToOpenAIParams(messages) {
         }
         if (message.additional_kwargs.function_call != null) {
             completionParam.function_call = message.additional_kwargs.function_call;
+            completionParam.content = null;
         }
         if ((0, messages_1.isAIMessage)(message) && !!message.tool_calls?.length) {
             completionParam.tool_calls = message.tool_calls.map(openai_tools_1.convertLangChainToolCallToOpenAI);
+            completionParam.content = null;
         }
         else {
             if (message.additional_kwargs.tool_calls != null) {
@@ -168,14 +172,13 @@ function convertMessagesToOpenAIParams(messages) {
 /**
  * Wrapper around OpenAI large language models that use the Chat endpoint.
  *
- * To use you should have the `openai` package installed, with the
- * `OPENAI_API_KEY` environment variable set.
+ * To use you should have the `OPENAI_API_KEY` environment variable set.
  *
- * To use with Azure you should have the `openai` package installed, with the
+ * To use with Azure you should have the:
  * `AZURE_OPENAI_API_KEY`,
  * `AZURE_OPENAI_API_INSTANCE_NAME`,
  * `AZURE_OPENAI_API_DEPLOYMENT_NAME`
- * and `AZURE_OPENAI_API_VERSION` environment variable set.
+ * and `AZURE_OPENAI_API_VERSION` environment variables set.
  * `AZURE_OPENAI_BASE_PATH` is optional and will override `AZURE_OPENAI_API_INSTANCE_NAME` if you need to use a custom endpoint.
  *
  * @remarks
@@ -329,6 +332,12 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
             writable: true,
             value: false
         });
+        Object.defineProperty(this, "streamUsage", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
         Object.defineProperty(this, "maxTokens", {
             enumerable: true,
             configurable: true,
@@ -457,6 +466,7 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
         this.stopSequences = this?.stop;
         this.user = fields?.user;
         this.streaming = fields?.streaming ?? false;
+        this.streamUsage = fields?.streamUsage ?? this.streamUsage;
         if (this.azureOpenAIApiKey || this.azureADTokenProvider) {
             if (!this.azureOpenAIApiInstanceName && !this.azureOpenAIBasePath) {
                 throw new Error("Azure OpenAI API instance name not found");
@@ -507,6 +517,13 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
             return (tools !== undefined &&
                 tools.every((tool) => Array.isArray(tool.lc_namespace)));
         }
+        let streamOptionsConfig = {};
+        if (options?.stream_options !== undefined) {
+            streamOptionsConfig = { stream_options: options.stream_options };
+        }
+        else if (this.streamUsage && this.streaming) {
+            streamOptionsConfig = { stream_options: { include_usage: true } };
+        }
         const params = {
             model: this.model,
             temperature: this.temperature,
@@ -520,6 +537,7 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
             logit_bias: this.logitBias,
             stop: options?.stop ?? this.stopSequences,
             user: this.user,
+            // if include_usage is set or streamUsage then stream must be set to true.
             stream: this.streaming,
             functions: options?.functions,
             function_call: options?.function_call,
@@ -529,9 +547,8 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
             tool_choice: options?.tool_choice,
             response_format: options?.response_format,
             seed: options?.seed,
-            ...(options?.stream_options !== undefined
-                ? { stream_options: options.stream_options }
-                : {}),
+            ...streamOptionsConfig,
+            parallel_tool_calls: options?.parallel_tool_calls,
             ...this.modelKwargs,
         };
         return params;
@@ -566,7 +583,7 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
             if (!delta) {
                 continue;
             }
-            const chunk = _convertDeltaToMessageChunk(delta, defaultRole);
+            const chunk = _convertDeltaToMessageChunk(delta, data.id, defaultRole);
             defaultRole = delta.role ?? defaultRole;
             const newTokenIndices = {
                 prompt: options.promptIndex ?? 0,
@@ -677,7 +694,7 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
                 const text = part.message?.content ?? "";
                 const generation = {
                     text,
-                    message: openAIResponseToChatMessage(part.message ?? { role: "assistant" }),
+                    message: openAIResponseToChatMessage(part.message ?? { role: "assistant" }, data.id),
                 };
                 generation.generationInfo = {
                     ...(part.finish_reason ? { finish_reason: part.finish_reason } : {}),

package/dist/chat_models.d.ts

@@ -27,21 +27,34 @@ export interface ChatOpenAICallOptions extends OpenAICallOptions, BaseFunctionCa
         type: "json_object";
     };
     seed?: number;
+    /**
+     * Additional options to pass to streamed completions.
+     * If provided takes precedence over "streamUsage" set at initialization time.
+     */
     stream_options?: {
+        /**
+         * Whether or not to include token usage in the stream.
+         * If set to `true`, this will include an additional
+         * chunk at the end of the stream with the token usage.
+         */
         include_usage: boolean;
     };
+    /**
+     * Whether or not to restrict the ability to
+     * call multiple tools in one response.
+     */
+    parallel_tool_calls?: boolean;
 }
 /**
  * Wrapper around OpenAI large language models that use the Chat endpoint.
  *
- * To use you should have the `openai` package installed, with the
- * `OPENAI_API_KEY` environment variable set.
+ * To use you should have the `OPENAI_API_KEY` environment variable set.
  *
- * To use with Azure you should have the `openai` package installed, with the
+ * To use with Azure you should have the:
  * `AZURE_OPENAI_API_KEY`,
  * `AZURE_OPENAI_API_INSTANCE_NAME`,
  * `AZURE_OPENAI_API_DEPLOYMENT_NAME`
- * and `AZURE_OPENAI_API_VERSION` environment variable set.
+ * and `AZURE_OPENAI_API_VERSION` environment variables set.
  * `AZURE_OPENAI_BASE_PATH` is optional and will override `AZURE_OPENAI_API_INSTANCE_NAME` if you need to use a custom endpoint.
  *
  * @remarks
@@ -87,6 +100,7 @@ export declare class ChatOpenAI<CallOptions extends ChatOpenAICallOptions = Chat
     user?: string;
     timeout?: number;
     streaming: boolean;
+    streamUsage: boolean;
     maxTokens?: number;
     logprobs?: boolean;
     topLogprobs?: number;

package/dist/chat_models.js

@@ -43,7 +43,7 @@ export function messageToOpenAIRole(message) {
             throw new Error(`Unknown message type: ${type}`);
     }
 }
-function openAIResponseToChatMessage(message) {
+function openAIResponseToChatMessage(message, messageId) {
     const rawToolCalls = message.tool_calls;
     switch (message.role) {
         case "assistant": {
@@ -66,6 +66,7 @@ function openAIResponseToChatMessage(message) {
                     function_call: message.function_call,
                     tool_calls: rawToolCalls,
                 },
+                id: messageId,
             });
         }
         default:
@@ -74,7 +75,7 @@ function openAIResponseToChatMessage(message) {
 }
 function _convertDeltaToMessageChunk(
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-delta, defaultRole) {
+delta, messageId, defaultRole) {
     const role = delta.role ?? defaultRole;
     const content = delta.content ?? "";
     let additional_kwargs;
@@ -110,6 +111,7 @@ delta, defaultRole) {
             content,
             tool_call_chunks: toolCallChunks,
             additional_kwargs,
+            id: messageId,
         });
     }
     else if (role === "system") {
@@ -146,9 +148,11 @@ function convertMessagesToOpenAIParams(messages) {
         }
         if (message.additional_kwargs.function_call != null) {
             completionParam.function_call = message.additional_kwargs.function_call;
+            completionParam.content = null;
         }
         if (isAIMessage(message) && !!message.tool_calls?.length) {
             completionParam.tool_calls = message.tool_calls.map(convertLangChainToolCallToOpenAI);
+            completionParam.content = null;
         }
         else {
             if (message.additional_kwargs.tool_calls != null) {
@@ -164,14 +168,13 @@ function convertMessagesToOpenAIParams(messages) {
 /**
  * Wrapper around OpenAI large language models that use the Chat endpoint.
  *
- * To use you should have the `openai` package installed, with the
- * `OPENAI_API_KEY` environment variable set.
+ * To use you should have the `OPENAI_API_KEY` environment variable set.
  *
- * To use with Azure you should have the `openai` package installed, with the
+ * To use with Azure you should have the:
  * `AZURE_OPENAI_API_KEY`,
  * `AZURE_OPENAI_API_INSTANCE_NAME`,
  * `AZURE_OPENAI_API_DEPLOYMENT_NAME`
- * and `AZURE_OPENAI_API_VERSION` environment variable set.
+ * and `AZURE_OPENAI_API_VERSION` environment variables set.
  * `AZURE_OPENAI_BASE_PATH` is optional and will override `AZURE_OPENAI_API_INSTANCE_NAME` if you need to use a custom endpoint.
  *
  * @remarks
@@ -325,6 +328,12 @@ export class ChatOpenAI extends BaseChatModel {
             writable: true,
             value: false
         });
+        Object.defineProperty(this, "streamUsage", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
         Object.defineProperty(this, "maxTokens", {
             enumerable: true,
             configurable: true,
@@ -453,6 +462,7 @@ export class ChatOpenAI extends BaseChatModel {
         this.stopSequences = this?.stop;
         this.user = fields?.user;
         this.streaming = fields?.streaming ?? false;
+        this.streamUsage = fields?.streamUsage ?? this.streamUsage;
         if (this.azureOpenAIApiKey || this.azureADTokenProvider) {
             if (!this.azureOpenAIApiInstanceName && !this.azureOpenAIBasePath) {
                 throw new Error("Azure OpenAI API instance name not found");
@@ -503,6 +513,13 @@ export class ChatOpenAI extends BaseChatModel {
             return (tools !== undefined &&
                 tools.every((tool) => Array.isArray(tool.lc_namespace)));
         }
+        let streamOptionsConfig = {};
+        if (options?.stream_options !== undefined) {
+            streamOptionsConfig = { stream_options: options.stream_options };
+        }
+        else if (this.streamUsage && this.streaming) {
+            streamOptionsConfig = { stream_options: { include_usage: true } };
+        }
         const params = {
             model: this.model,
             temperature: this.temperature,
@@ -516,6 +533,7 @@ export class ChatOpenAI extends BaseChatModel {
             logit_bias: this.logitBias,
             stop: options?.stop ?? this.stopSequences,
             user: this.user,
+            // if include_usage is set or streamUsage then stream must be set to true.
             stream: this.streaming,
             functions: options?.functions,
             function_call: options?.function_call,
@@ -525,9 +543,8 @@ export class ChatOpenAI extends BaseChatModel {
             tool_choice: options?.tool_choice,
             response_format: options?.response_format,
             seed: options?.seed,
-            ...(options?.stream_options !== undefined
-                ? { stream_options: options.stream_options }
-                : {}),
+            ...streamOptionsConfig,
+            parallel_tool_calls: options?.parallel_tool_calls,
             ...this.modelKwargs,
         };
         return params;
@@ -562,7 +579,7 @@ export class ChatOpenAI extends BaseChatModel {
             if (!delta) {
                 continue;
             }
-            const chunk = _convertDeltaToMessageChunk(delta, defaultRole);
+            const chunk = _convertDeltaToMessageChunk(delta, data.id, defaultRole);
             defaultRole = delta.role ?? defaultRole;
             const newTokenIndices = {
                 prompt: options.promptIndex ?? 0,
@@ -673,7 +690,7 @@ export class ChatOpenAI extends BaseChatModel {
                 const text = part.message?.content ?? "";
                 const generation = {
                     text,
-                    message: openAIResponseToChatMessage(part.message ?? { role: "assistant" }),
+                    message: openAIResponseToChatMessage(part.message ?? { role: "assistant" }, data.id),
                 };
                 generation.generationInfo = {
                     ...(part.finish_reason ? { finish_reason: part.finish_reason } : {}),

package/dist/legacy.cjs

@@ -227,7 +227,9 @@ class OpenAIChat extends llms_1.LLM {
             value: void 0
         });
         this.openAIApiKey =
-            fields?.openAIApiKey ?? (0, env_1.getEnvironmentVariable)("OPENAI_API_KEY");
+            fields?.apiKey ??
+                fields?.openAIApiKey ??
+                (0, env_1.getEnvironmentVariable)("OPENAI_API_KEY");
         this.azureOpenAIApiKey =
             fields?.azureOpenAIApiKey ??
                 (0, env_1.getEnvironmentVariable)("AZURE_OPENAI_API_KEY");

package/dist/legacy.js

@@ -224,7 +224,9 @@ export class OpenAIChat extends LLM {
             value: void 0
         });
         this.openAIApiKey =
-            fields?.openAIApiKey ?? getEnvironmentVariable("OPENAI_API_KEY");
+            fields?.apiKey ??
+                fields?.openAIApiKey ??
+                getEnvironmentVariable("OPENAI_API_KEY");
         this.azureOpenAIApiKey =
             fields?.azureOpenAIApiKey ??
                 getEnvironmentVariable("AZURE_OPENAI_API_KEY");

package/dist/types.d.ts

@@ -24,6 +24,11 @@ export declare interface OpenAIBaseInput {
     user?: string;
     /** Whether to stream the results or not. Enabling disables tokenUsage reporting */
     streaming: boolean;
+    /**
+     * Whether or not to include token usage data in streamed chunks.
+     * @default true
+     */
+    streamUsage?: boolean;
     /**
      * Model name to use
      * Alias for `model`

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/openai",
-  "version": "0.1.2",
+  "version": "0.2.0",
   "description": "OpenAI integrations for LangChain.js",
   "type": "module",
   "engines": {
@@ -35,9 +35,9 @@
   "author": "LangChain",
   "license": "MIT",
   "dependencies": {
-    "@langchain/core": ">=0.2.5 <0.3.0",
+    "@langchain/core": ">=0.2.8 <0.3.0",
     "js-tiktoken": "^1.0.12",
-    "openai": "^4.41.1",
+    "openai": "^4.49.1",
     "zod": "^3.22.4",
     "zod-to-json-schema": "^3.22.3"
   },
@@ -45,7 +45,7 @@
     "@azure/identity": "^4.2.0",
     "@jest/globals": "^29.5.0",
     "@langchain/scripts": "~0.0.14",
-    "@langchain/standard-tests": "workspace:*",
+    "@langchain/standard-tests": "0.0.0",
     "@swc/core": "^1.3.90",
     "@swc/jest": "^0.2.29",
     "dpdm": "^3.12.0",
@@ -61,6 +61,7 @@
     "prettier": "^2.8.3",
     "release-it": "^15.10.1",
     "rimraf": "^5.0.1",
+    "ts-jest": "^29.1.0",
     "typescript": "~5.1.6"
   },
   "publishConfig": {