From 35a6795559110766c10f168e94ee56c89781d9a5 Mon Sep 17 00:00:00 2001
From: Abdul Jamjoom <abduljamjoom@gmail.com>
Date: Tue, 4 Jul 2023 22:34:49 -0700
Subject: [PATCH] polish
---
packages/core/src/ServiceContext.ts | 2 +-
packages/core/src/callbacks/utility/aHandleOpenAIStream.ts | 5 +++--
2 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/packages/core/src/ServiceContext.ts b/packages/core/src/ServiceContext.ts
index 0803e14f3..8d3eb72d4 100644
--- a/packages/core/src/ServiceContext.ts
+++ b/packages/core/src/ServiceContext.ts
@@ -1,6 +1,6 @@
import { BaseEmbedding, OpenAIEmbedding } from "./Embedding";
import { BaseLLMPredictor, ChatGPTLLMPredictor } from "./LLMPredictor";
-import { BaseLanguageModel, ChatOpenAI } from "./LanguageModel";
+import { ChatOpenAI } from "./LanguageModel";
import { NodeParser, SimpleNodeParser } from "./NodeParser";
import { PromptHelper } from "./PromptHelper";
import { CallbackManager } from "./callbacks/CallbackManager";
diff --git a/packages/core/src/callbacks/utility/aHandleOpenAIStream.ts b/packages/core/src/callbacks/utility/aHandleOpenAIStream.ts
index 023c17ff2..9f01d6345 100644
--- a/packages/core/src/callbacks/utility/aHandleOpenAIStream.ts
+++ b/packages/core/src/callbacks/utility/aHandleOpenAIStream.ts
@@ -31,8 +31,9 @@ export async function aHandleOpenAIStream({
}
/*
- TODO: upgrade to latest version of openai-node and use the native streaming support (https://github.com/openai/openai-node/issues/18#issuecomment-1595805163)
- source: https://github.com/openai/openai-node/issues/18#issuecomment-1372047643
+ sources:
+ - https://github.com/openai/openai-node/issues/18#issuecomment-1372047643
+ - https://github.com/openai/openai-node/issues/18#issuecomment-1595805163
*/
async function* __astreamCompletion(data: string[]) {
yield* __alinesToText(__achunksToLines(data));
--
GitLab