From e7206a7deaeec7414846133849a26e7b6d18bf28 Mon Sep 17 00:00:00 2001
From: Sourabh Desai <sourabhdesai@gmail.com>
Date: Tue, 18 Jul 2023 02:20:49 +0000
Subject: [PATCH] simplify test
---
packages/core/src/tests/CallbackManager.test.ts | 17 +++++++++++++++--
1 file changed, 15 insertions(+), 2 deletions(-)
diff --git a/packages/core/src/tests/CallbackManager.test.ts b/packages/core/src/tests/CallbackManager.test.ts
index 5e453b712..772b9d85e 100644
--- a/packages/core/src/tests/CallbackManager.test.ts
+++ b/packages/core/src/tests/CallbackManager.test.ts
@@ -8,7 +8,11 @@ import {
RetrievalCallbackResponse,
StreamCallbackResponse,
} from "../callbacks/CallbackManager";
-import { ListIndex } from "../index/list";
+import { ListIndex, ListRetrieverMode } from "../index/list";
+import {
+ ResponseSynthesizer,
+ SimpleResponseBuilder,
+} from "../ResponseSynthesizer";
import { mockEmbeddingModel, mockLlmGeneration } from "./utility/mockOpenAI";
// Mock the OpenAI getOpenAISession function during testing
@@ -138,10 +142,19 @@ describe("CallbackManager: onLLMStream and onRetrieve", () => {
documents: [document],
serviceContext,
});
- const queryEngine = listIndex.asQueryEngine();
+ const responseBuilder = new SimpleResponseBuilder(serviceContext);
+ const responseSynthesizer = new ResponseSynthesizer({
+ serviceContext: serviceContext,
+ responseBuilder,
+ });
+ const queryEngine = listIndex.asQueryEngine(
+ ListRetrieverMode.DEFAULT,
+ responseSynthesizer
+ );
const query = "What is the author's name?";
const response = await queryEngine.aquery(query);
expect(response.toString()).toBe("MOCK_TOKEN_1-MOCK_TOKEN_2");
+ console.log(JSON.stringify(streamCallbackData, null, 2));
expect(streamCallbackData).toEqual([
{
event: {
--
GitLab