chore: lint files

langchain-ai · Jun 26, 2024 · ddcc08b · ddcc08b
1 parent 611114c
commit ddcc08b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/langchain/src/experimental/chrome_ai/chat_models.ts b/langchain/src/experimental/chrome_ai/chat_models.ts
@@ -6,6 +6,7 @@ import type { BaseLanguageModelCallOptions } from "@langchain/core/language_mode
 import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
 import { BaseMessage, AIMessageChunk } from "@langchain/core/messages";
 import { ChatGenerationChunk } from "@langchain/core/outputs";
+import { IterableReadableStream } from "@langchain/core/utils/stream";
 
 export interface AI {
   canCreateTextSession(): Promise<AIModelAvailability>;
@@ -138,7 +139,7 @@ export class ChatChromeAI extends SimpleChatModel<ChromeAICallOptions> {
     const textPrompt = formatPrompt(messages);
 
     const stream = this.session.promptStreaming(textPrompt);
-    const iterableStream = stream.pipeThrough(new TextDecoderStream()).getIterator();
+    const iterableStream = IterableReadableStream.fromReadableStream(stream);
 
     let previousContent = "";
     for await (const chunk of iterableStream) {