diff --git a/packages/core/src/tests/CallbackManager.test.ts b/packages/core/src/tests/CallbackManager.test.ts index cb82242af038d0a12413f68ed1caa7f0c45dadc4..ff04057da5de48b5f658840178d98a48e4a77be4 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,