Refactor similarity calculation and sorting in HypaProcesser class

This commit is contained in:
kwaroran
2024-04-24 20:30:14 +09:00
parent acf72727b6
commit d2288d326b

View File

@@ -128,11 +128,11 @@ export class HypaProcesser{
): Promise<[string, number][]> {
const memoryVectors = this.vectors
const searches = memoryVectors
.map((vector, index) => ({
similarity: similarity(query, vector.embedding),
index,
}))
.sort((a, b) => (a.similarity > b.similarity ? -1 : 0))
.map((vector, index) => ({
similarity: similarity(query, vector.embedding),
index,
}))
.sort((a, b) => (a.similarity > b.similarity ? -1 : 0))
const result: [string, number][] = searches.map((search) => [
memoryVectors[search.index].content,