Support graph=True in embeddings config, closes #848

This commit is contained in:
davidmezzetti
2025-01-09 13:11:34 -05:00
parent 1352040add
commit c7e0fe3f99
2 changed files with 14 additions and 0 deletions
+10
View File
@@ -146,6 +146,16 @@ class TestGraph(unittest.TestCase):
# Close graph
graph.close()
def testDefault(self):
"""
Test embeddings default graph setting
"""
embeddings = Embeddings(content=True, graph=True)
embeddings.index([(uid, text, None) for uid, text in enumerate(self.data)])
self.assertEqual(embeddings.graph.count(), len(self.data))
def testDelete(self):
"""
Test delete