Method load_local

finagent/memory/faiss.py:167–193

Load FAISS index and index_to_key from disk. Args: embedding_provider: Embeddings to use when generating queries memory_path: folder path to load index and index_to_key from. name: name of the vectorstore. Returns: The FAISS vectorsto

(
        self,
        embedding_dim: int,
        memory_path: str,
    )

        return key_and_score

    def load_local(
        self,
        embedding_dim: int,
        memory_path: str,
    ):
        """Load FAISS index and index_to_key from disk.

        Args:
            embedding_provider: Embeddings to use when generating queries
            memory_path: folder path to load index and index_to_key from.
            name: name of the vectorstore.

        Returns:
            The FAISS vectorstore class.
        """
        # load index separately since it is not picklable
        faiss = dependable_faiss_import()
        index = faiss.read_index(os.path.join(memory_path, "index.faiss"))

        # load index_to_key
        with open(os.path.join(memory_path, "index2key.pkl"), "rb") as f:
            index_to_key = pickle.load(f)

        self.index = index
        self.index_to_key = index_to_key
        self.memory_path = memory_path
        self.embedding_dim = embedding_dim

    def save_local(self, memory_path = None) -> None: