github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-05 02:51:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add document structure into GraphRAG (#2033)

Browse Source 
Co-authored-by: Appointat <kuda.czk@antgroup.com>
Co-authored-by: tpoisonooo <khj.application@aliyun.com>
Co-authored-by: vritser <vritser@163.com>
This commit is contained in:
lipengfei
2024-10-18 22:03:08 +08:00
committed by GitHub
parent 811ce63493
commit 88e3d12bd3
29 changed files with 1909 additions and 935 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/intetration_tests/datasource/test_conn_tugraph.py
View File

@@ -1,4 +1,5 @@
import pytest
from dbgpt.datasource.conn_tugraph import TuGraphConnector
# Set database connection parameters.
@@ -21,10 +22,10 @@ def connector():
def test_get_table_names(connector):
"""Test retrieving table names from the graph database."""
table_names = connector.get_table_names()
vertex_tables, edge_tables = connector.get_table_names()
# Verify the quantity of vertex and edge tables.
assert len(table_names["vertex_tables"]) == 5
assert len(table_names["edge_tables"]) == 8
assert len(vertex_tables) == 5
assert len(edge_tables) == 8
def test_get_columns(connector):