From 9c5220ce75538ff2e179f209624507d001a593fc Mon Sep 17 00:00:00 2001
From: Logan Markewich <logan.markewich@live.com>
Date: Sun, 23 Jul 2023 19:28:14 -0600
Subject: [PATCH] fix tests
---
packages/core/src/tests/CallbackManager.test.ts | 11 +++++------
1 file changed, 5 insertions(+), 6 deletions(-)
diff --git a/packages/core/src/tests/CallbackManager.test.ts b/packages/core/src/tests/CallbackManager.test.ts
index cb82242af..ff04057da 100644
--- a/packages/core/src/tests/CallbackManager.test.ts
+++ b/packages/core/src/tests/CallbackManager.test.ts
@@ -67,8 +67,7 @@ describe("CallbackManager: onLLMStream and onRetrieve", () => {
test("For VectorStoreIndex w/ a SimpleResponseBuilder", async () => {
const vectorStoreIndex = await VectorStoreIndex.fromDocuments(
[document],
- undefined,
- serviceContext
+ { serviceContext }
);
const queryEngine = vectorStoreIndex.asQueryEngine();
const query = "What is the author's name?";
@@ -138,10 +137,10 @@ describe("CallbackManager: onLLMStream and onRetrieve", () => {
});
test("For ListIndex w/ a ListIndexRetriever", async () => {
- const listIndex = await ListIndex.fromDocuments({
- documents: [document],
- serviceContext,
- });
+ const listIndex = await ListIndex.fromDocuments(
+ [document],
+ { serviceContext },
+ );
const responseBuilder = new SimpleResponseBuilder(serviceContext);
const responseSynthesizer = new ResponseSynthesizer({
serviceContext: serviceContext,
--
GitLab