in skills/text_to_sql/evaluation/vectordb.py [0:0]
def load_db(self):
if os.path.exists(self.db_path):
with open(self.db_path, "rb") as file:
data = pickle.load(file)
self.embeddings, self.metadata, self.query_cache = data['embeddings'], data['metadata'], json.loads(data['query_cache'])
else:
self.embeddings, self.metadata, self.query_cache = [], [], {}