feat(GraphRAG): enhance GraphRAG by graph community summary (#1801)

Co-authored-by: Florian <fanzhidongyzby@163.com>
Co-authored-by: KingSkyLi <15566300566@163.com>
Co-authored-by: aries_ckt <916701291@qq.com>
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
Co-authored-by: yvonneyx <zhuyuxin0627@gmail.com>
This commit is contained in:
M1n9X
2024-08-30 21:59:44 +08:00
committed by GitHub
parent 471689ba20
commit 759f7d99cc
59 changed files with 29316 additions and 411 deletions

View File

@@ -40,3 +40,15 @@ def test_get_indexes(connector):
# Get the index information of the vertex table named 'person'.
indexes = connector.get_indexes("person", "vertex")
assert len(indexes) > 0
def test_run_without_stream(connector):
query = "MATCH (n) RETURN n limit 10"
result = connector.run(query)
assert len(result) == 10
def test_run_with_stream(connector):
query = "MATCH (n) RETURN n limit 10"
result = list(connector.run_stream(query))
assert len(result) == 10