kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix graph init bugs

Browse Source
This commit is contained in:
Wenjie Zhang 2024-11-21 23:50:52 +08:00
parent d5c8e4f279
commit 8c26722aa9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/core/graphbase.py
View File

@ -159,7 +159,7 @@ class GraphDatabase:
embed_info = EMBED_MODEL_INFO[self.config.embed_model] embed_info = EMBED_MODEL_INFO[self.config.embed_model]
with self.driver.session() as session: with self.driver.session() as session:
session.execute_write(_create_graph, triples) 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): for i, entry in enumerate(triples):
logger.info(f"Adding entity {i+1}/{len(triples)}") logger.info(f"Adding entity {i+1}/{len(triples)}")
embedding_h = self.get_embedding(entry['h']) embedding_h = self.get_embedding(entry['h'])