From 5d8d344e1f0fcd07e861f1a42abc224b8465525c Mon Sep 17 00:00:00 2001
From: Abdul Jamjoom <abduljamjoom@gmail.com>
Date: Sun, 9 Jul 2023 16:42:58 -0700
Subject: [PATCH] updated tests
---
packages/core/src/QueryEngine.ts | 2 +-
packages/core/src/tests/CallbackManager.test.ts | 8 ++++++++
2 files changed, 9 insertions(+), 1 deletion(-)
diff --git a/packages/core/src/QueryEngine.ts b/packages/core/src/QueryEngine.ts
index 3c622f0a4..543305cad 100644
--- a/packages/core/src/QueryEngine.ts
+++ b/packages/core/src/QueryEngine.ts
@@ -34,7 +34,7 @@ export class RetrieverQueryEngine implements BaseQueryEngine {
tags: ["final"],
};
const nodes = await this.retriever.aretrieve(query, _parentEvent);
- return this.responseSynthesizer.asynthesize(query, nodes, parentEvent);
+ return this.responseSynthesizer.asynthesize(query, nodes, _parentEvent);
}
}
diff --git a/packages/core/src/tests/CallbackManager.test.ts b/packages/core/src/tests/CallbackManager.test.ts
index dfedc2c39..26bcbee2f 100644
--- a/packages/core/src/tests/CallbackManager.test.ts
+++ b/packages/core/src/tests/CallbackManager.test.ts
@@ -76,6 +76,7 @@ describe("CallbackManager: onLLMStream and onRetrieve", () => {
id: expect.any(String),
parentId: expect.any(String),
type: "llmPredict",
+ tags: ["final"],
},
index: 0,
token: {
@@ -91,6 +92,7 @@ describe("CallbackManager: onLLMStream and onRetrieve", () => {
id: expect.any(String),
parentId: expect.any(String),
type: "llmPredict",
+ tags: ["final"],
},
index: 1,
token: {
@@ -106,6 +108,7 @@ describe("CallbackManager: onLLMStream and onRetrieve", () => {
id: expect.any(String),
parentId: expect.any(String),
type: "llmPredict",
+ tags: ["final"],
},
index: 2,
isDone: true,
@@ -119,6 +122,7 @@ describe("CallbackManager: onLLMStream and onRetrieve", () => {
id: expect.any(String),
parentId: expect.any(String),
type: "retrieve",
+ tags: ["final"],
},
},
]);
@@ -145,6 +149,7 @@ describe("CallbackManager: onLLMStream and onRetrieve", () => {
id: expect.any(String),
parentId: expect.any(String),
type: "llmPredict",
+ tags: ["final"],
},
index: 0,
token: {
@@ -160,6 +165,7 @@ describe("CallbackManager: onLLMStream and onRetrieve", () => {
id: expect.any(String),
parentId: expect.any(String),
type: "llmPredict",
+ tags: ["final"],
},
index: 1,
token: {
@@ -175,6 +181,7 @@ describe("CallbackManager: onLLMStream and onRetrieve", () => {
id: expect.any(String),
parentId: expect.any(String),
type: "llmPredict",
+ tags: ["final"],
},
index: 2,
isDone: true,
@@ -188,6 +195,7 @@ describe("CallbackManager: onLLMStream and onRetrieve", () => {
id: expect.any(String),
parentId: expect.any(String),
type: "retrieve",
+ tags: ["final"],
},
},
]);
--
GitLab