From 8c26722aa95332dfb942c4840f7ac48410d31d25 Mon Sep 17 00:00:00 2001 From: Wenjie Zhang Date: Thu, 21 Nov 2024 23:50:52 +0800 Subject: [PATCH] fix graph init bugs --- src/core/graphbase.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/core/graphbase.py b/src/core/graphbase.py index ef4f7af1..b33e3fac 100644 --- a/src/core/graphbase.py +++ b/src/core/graphbase.py @@ -159,7 +159,7 @@ class GraphDatabase: embed_info = EMBED_MODEL_INFO[self.config.embed_model] with self.driver.session() as session: session.execute_write(_create_graph, triples) - session.execute_write(_create_vector_index, embed_info.dimension) + session.execute_write(_create_vector_index, embed_info.get('dimension')) for i, entry in enumerate(triples): logger.info(f"Adding entity {i+1}/{len(triples)}") embedding_h = self.get_embedding(entry['h'])