添加GraphRAG后端
This commit is contained in:
37
backend/run_import.py
Normal file
37
backend/run_import.py
Normal 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()
|
||||
Reference in New Issue
Block a user