Method load_local

finagent/memory/interface.py:163–220

(
        self,
        memory_path: str = None,
    )

        return res

    def load_local(
        self,
        memory_path: str = None,
    ) -> None:

        if memory_path is None:
            memory_path = self.memory_path

        """Load the memory from the local file."""
        for symbol in self.symbols:

            try:
                path = os.path.join(memory_path, symbol, "market_intelligence")
                os.makedirs(path, exist_ok=True)
                vecstore = FAISS(memory_path=path, embedding_dim=self.embedding_dim)
                vecstore.load_local(memory_path=path, embedding_dim=self.embedding_dim)

                # print length of index
                print(f"symbols: {symbol}, memory_path: {path}, vecstore length: {vecstore.index.ntotal}")

                self.market_intelligence_memorys[symbol].load_local(
                    memory_path=path,
                    vectorstore=vecstore,
                )
            except Exception as e:
                print(f"Failed to load market_intelligence_memorys: {e}")

            try:
                path = os.path.join(memory_path, symbol, "low_level_reflection")
                os.makedirs(path, exist_ok=True)
                vecstore = FAISS(memory_path=path, embedding_dim=self.embedding_dim)
                vecstore.load_local(memory_path=path, embedding_dim=self.embedding_dim)

                # print length of index
                print(f"symbols: {symbol}, memory_path: {path}, vecstore length: {vecstore.index.ntotal}")

                self.low_level_reflection_memorys[symbol].load_local(
                    memory_path=path,
                    vectorstore=vecstore,
                )
            except Exception as e:
                print(f"Failed to load low_level_reflection_memorys: {e}")

            try:
                path = os.path.join(memory_path, symbol, "high_level_reflection")
                os.makedirs(path, exist_ok=True)
                vecstore = FAISS(memory_path=path, embedding_dim=self.embedding_dim)
                vecstore.load_local(memory_path=path, embedding_dim=self.embedding_dim)

                # print length of index
                print(f"symbols: {symbol}, memory_path: {path}, vecstore length: {vecstore.index.ntotal}")

                self.high_level_reflection_memorys[symbol].load_local(
                    memory_path=path,
                    vectorstore=vecstore,
                )
            except Exception as e:
                print(f"Failed to load high_level_reflection_memorys: {e}")