From 8b7a94f8c54f4e1f74c6bcd62d6d06a249d64c8c Mon Sep 17 00:00:00 2001
From: Yi Ding <yi.s.ding@gmail.com>
Date: Mon, 17 Jul 2023 21:51:14 -0700
Subject: [PATCH] fix
---
packages/core/src/tests/utility/mockOpenAI.ts | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/packages/core/src/tests/utility/mockOpenAI.ts b/packages/core/src/tests/utility/mockOpenAI.ts
index ccbadcf71..c80617862 100644
--- a/packages/core/src/tests/utility/mockOpenAI.ts
+++ b/packages/core/src/tests/utility/mockOpenAI.ts
@@ -62,12 +62,12 @@ export function mockLlmGeneration({
}
export function mockEmbeddingModel(embedModel: OpenAIEmbedding) {
- jest.spyOn(embedModel, "aGetTextEmbedding").mockImplementation(async (x) => {
+ jest.spyOn(embedModel, "getTextEmbedding").mockImplementation(async (x) => {
return new Promise((resolve) => {
resolve([1, 0, 0, 0, 0, 0]);
});
});
- jest.spyOn(embedModel, "aGetQueryEmbedding").mockImplementation(async (x) => {
+ jest.spyOn(embedModel, "getQueryEmbedding").mockImplementation(async (x) => {
return new Promise((resolve) => {
resolve([0, 1, 0, 0, 0, 0]);
});
--
GitLab