mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor similarity calculation and sorting in HypaProcesser class

Browse Source
This commit is contained in:
kwaroran
2024-04-24 20:30:14 +09:00
parent acf72727b6
commit d2288d326b
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/ts/process/memory/hypamemory.ts
View File

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