添加GraphRAG后端

This commit is contained in:
龙澳
2026-03-31 17:18:30 +08:00
parent 7b1b92d825
commit d2cf97387b
9 changed files with 601 additions and 0 deletions

37
backend/run_import.py Normal file
View File

@@ -0,0 +1,37 @@
#!/usr/bin/env python3
"""
独立命令行导入脚本。
用法:
python run_import.py # 增量导入MERGE不删除现有数据
python run_import.py --clear # 清空图谱后全量重新导入
"""
import sys
from dotenv import load_dotenv
from graph_query import get_driver
from graph_builder import build_graph
load_dotenv()
def main():
clear = "--clear" in sys.argv
print("Connecting to Neo4j...")
driver = get_driver()
driver.verify_connectivity()
print("Connected.\n")
build_graph(driver, clear=clear)
print("\nGraph stats:")
from graph_query import get_graph_stats
for k, v in get_graph_stats().items():
print(f" {k}: {v}")
driver.close()
if __name__ == "__main__":
main()