@langchain/openai 0.2.10 → 0.2.11

package/dist/chat_models.cjs

@@ -1066,6 +1066,16 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
         };
     }
     async *_streamResponseChunks(messages, options, runManager) {
+        if (this.model.includes("o1-")) {
+            console.warn("[WARNING]: OpenAI o1 models do not yet support token-level streaming. Streaming will yield single chunk.");
+            const result = await this._generate(messages, options, runManager);
+            const messageChunk = (0, messages_1.convertToChunk)(result.generations[0].message);
+            yield new outputs_1.ChatGenerationChunk({
+                message: messageChunk,
+                text: typeof messageChunk.content === "string" ? messageChunk.content : "",
+            });
+            return;
+        }
         const messagesMapped = convertMessagesToOpenAIParams(messages);
         const params = {
             ...this.invocationParams(options, {

package/dist/chat_models.js

@@ -1,5 +1,5 @@
 import { OpenAI as OpenAIClient } from "openai";
-import { AIMessage, AIMessageChunk, ChatMessage, ChatMessageChunk, FunctionMessageChunk, HumanMessageChunk, SystemMessageChunk, ToolMessageChunk, isAIMessage, } from "@langchain/core/messages";
+import { AIMessage, AIMessageChunk, ChatMessage, ChatMessageChunk, FunctionMessageChunk, HumanMessageChunk, SystemMessageChunk, ToolMessageChunk, isAIMessage, convertToChunk, } from "@langchain/core/messages";
 import { ChatGenerationChunk, } from "@langchain/core/outputs";
 import { getEnvironmentVariable } from "@langchain/core/utils/env";
 import { BaseChatModel, } from "@langchain/core/language_models/chat_models";
@@ -1062,6 +1062,16 @@ export class ChatOpenAI extends BaseChatModel {
         };
     }
     async *_streamResponseChunks(messages, options, runManager) {
+        if (this.model.includes("o1-")) {
+            console.warn("[WARNING]: OpenAI o1 models do not yet support token-level streaming. Streaming will yield single chunk.");
+            const result = await this._generate(messages, options, runManager);
+            const messageChunk = convertToChunk(result.generations[0].message);
+            yield new ChatGenerationChunk({
+                message: messageChunk,
+                text: typeof messageChunk.content === "string" ? messageChunk.content : "",
+            });
+            return;
+        }
         const messagesMapped = convertMessagesToOpenAIParams(messages);
         const params = {
             ...this.invocationParams(options, {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/openai",
-  "version": "0.2.10",
+  "version": "0.2.11",
   "description": "OpenAI integrations for LangChain.js",
   "type": "module",
   "engines": {