get_similarity.py

from src.pgdb.knowledge.similarity import VectorStore_FAISS


class GetSimilarity:
    def __init__(self, _question, _faiss_db: VectorStore_FAISS):
        self.question = _question
        self.faiss_db = _faiss_db
        self.similarity_doc = self.faiss_db.join_document(self.faiss_db.get_text_similarity(self.question))

    def get_similarity_doc(self):
        return self.similarity_doc