修改GraphRAG后端,另外两本小说入库

This commit is contained in:
2026-04-01 15:55:53 +08:00
parent 403b7dfacf
commit bfcf840013
4 changed files with 170 additions and 70 deletions

View File

@@ -5,25 +5,29 @@
用法:
python run_import.py # 增量导入MERGE不删除现有数据
python run_import.py --clear # 清空图谱后全量重新导入
python run_import.py ldj # 仅导入鹿鼎记
python run_import.py dtslz tlbb --clear # 清空后导入指定小说
"""
import sys
from dotenv import load_dotenv
from graph_query import get_driver
from graph_builder import build_graph
from graph_builder import build_all_graphs, SUPPORTED_NOVELS
load_dotenv()
def main():
clear = "--clear" in sys.argv
novels = [arg for arg in sys.argv[1:] if not arg.startswith("--")]
selected = novels or list(SUPPORTED_NOVELS)
print("Connecting to Neo4j...")
driver = get_driver()
driver.verify_connectivity()
print("Connected.\n")
build_graph(driver, clear=clear)
build_all_graphs(driver, novels=selected, clear=clear)
print("\nGraph stats:")
from graph_query import get_graph_stats