fix bug of merging entities (#546)

pingcap · Dec 25, 2024 · 2a0a8c3 · 2a0a8c3
1 parent 1604ada
commit 2a0a8c3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/backend/app/rag/knowledge_graph/graph_store/tidb_graph_store.py b/backend/app/rag/knowledge_graph/graph_store/tidb_graph_store.py
@@ -359,6 +359,8 @@ def get_or_create_entity(self, entity: Entity, commit: bool = True) -> SQLModel:
                     db_obj.meta_vec = get_entity_metadata_embedding(
                         db_obj.meta, self._embed_model
                     )
+
+                    self._session.add(db_obj)
                     if commit:
                         self._session.commit()
                         self._session.refresh(db_obj)