Webui chat with x added

This commit is contained in:
n4ze3m
2024-04-06 00:30:23 +05:30
parent aae2ddecbc
commit dd496b7b98
7 changed files with 401 additions and 28 deletions

View File

@@ -3,6 +3,7 @@ import { VectorStore } from "@langchain/core/vectorstores"
import type { EmbeddingsInterface } from "@langchain/core/embeddings"
import { Document } from "@langchain/core/documents"
import { getVector, insertVector } from "@/db/vector"
import { cp } from "fs"
/**
* Interface representing a vector in memory. It includes the content
@@ -116,8 +117,10 @@ export class PageAssistVectorStore extends VectorStore {
})
return filter(doc)
}
const pgVector = await getVector(`vector:${this.knownledge_id}`)
const filteredMemoryVectors = pgVector.vectors.filter(filterFunction)
const data = await getVector(`vector:${this.knownledge_id}`)
const pgVector = [...data.vectors]
const filteredMemoryVectors = pgVector.filter(filterFunction)
console.log(filteredMemoryVectors)
const searches = filteredMemoryVectors
.map((vector, index) => ({
similarity: this.similarity(query, vector.embedding),
@@ -125,7 +128,7 @@ export class PageAssistVectorStore extends VectorStore {
}))
.sort((a, b) => (a.similarity > b.similarity ? -1 : 0))
.slice(0, k)
console.log(searches)
const result: [Document, number][] = searches.map((search) => [
new Document({
metadata: filteredMemoryVectors[search.index].metadata,
@@ -133,7 +136,7 @@ export class PageAssistVectorStore extends VectorStore {
}),
search.similarity
])
console.log(result)
return result
}