From c4cb37786b2095d978f14485038f9798a77d70e5 Mon Sep 17 00:00:00 2001
From: Marcus Schiesser <mail@marcusschiesser.de>
Date: Tue, 30 Jan 2024 11:28:14 +0700
Subject: [PATCH] fix: use Array not Float32Array (#482)
---
packages/core/src/embeddings/HuggingFaceEmbedding.ts | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/packages/core/src/embeddings/HuggingFaceEmbedding.ts b/packages/core/src/embeddings/HuggingFaceEmbedding.ts
index 13ee9139e..f27eb07ad 100644
--- a/packages/core/src/embeddings/HuggingFaceEmbedding.ts
+++ b/packages/core/src/embeddings/HuggingFaceEmbedding.ts
@@ -39,7 +39,7 @@ export class HuggingFaceEmbedding extends BaseEmbedding {
async getTextEmbedding(text: string): Promise<number[]> {
const extractor = await this.getExtractor();
const output = await extractor(text, { pooling: "mean", normalize: true });
- return output.data;
+ return Array.from(output.data);
}
async getQueryEmbedding(query: string): Promise<number[]> {
--
GitLab