Typefix for findSimilar

axonzeta · Nov 11, 2024 · 30f7266 · 30f7266
1 parent ba51f48
commit 30f7266
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/extension/contentscript-main/AIEmbedding/AIEmbeddingUtil.ts b/src/extension/contentscript-main/AIEmbedding/AIEmbeddingUtil.ts
@@ -57,7 +57,7 @@ export function calculateCosineSimilarity (vectorA: AIEmbeddingVector, vectorB:
  * @param target: the target vector to compare against
  * @returns a list of embedding ids sorted by similarity
  */
-export function findSimilar (embeddings: AIEmbeddingVectorListItem, target: AIEmbeddingVector): AIEmbeddingVectorListItem[] {
+export function findSimilar (embeddings: AIEmbeddingVectorListItem, target: AIEmbeddingVector): any[] {
   const similarities = new Map<any, number>()
   for (const { id, vector } of embeddings) {
     similarities.set(id, calculateCosineSimilarity(vector, target))