[feat] emotion minilm processer

This commit is contained in:
kwaroran
2023-08-15 17:04:45 +09:00
parent 22ea6b0890
commit 3f9e08bfb0
5 changed files with 74 additions and 14 deletions

View File

@@ -122,12 +122,15 @@ export class HypaProcesser{
async similaritySearch(query: string) {
const results = await this.similaritySearchVectorWithScore((await this.getEmbeds(query))[0],);
console.log(results)
return results.map((result) => result[0]);
}
async similaritySearchVectorWithScore(
async similaritySearchScored(query: string) {
const results = await this.similaritySearchVectorWithScore((await this.getEmbeds(query))[0],);
return results
}
private async similaritySearchVectorWithScore(
query: number[],
): Promise<[string, number][]> {
const memoryVectors = this.vectors