|
|
@ -12,6 +12,8 @@ KEMU = 'Chemistry' # Chinese,Math,Chemistry
|
|
|
|
|
|
|
|
|
|
|
|
# 组装文件路径
|
|
|
|
# 组装文件路径
|
|
|
|
WORKING_DIR = "./Topic/" + KEMU
|
|
|
|
WORKING_DIR = "./Topic/" + KEMU
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# 文档文件
|
|
|
|
docx_file = 'static/Txt/' + KEMU + '.docx'
|
|
|
|
docx_file = 'static/Txt/' + KEMU + '.docx'
|
|
|
|
|
|
|
|
|
|
|
|
async def main():
|
|
|
|
async def main():
|
|
|
@ -20,31 +22,31 @@ async def main():
|
|
|
|
os.environ["NEO4J_USERNAME"] = NEO4J_USERNAME
|
|
|
|
os.environ["NEO4J_USERNAME"] = NEO4J_USERNAME
|
|
|
|
os.environ["NEO4J_PASSWORD"] = NEO4J_PASSWORD
|
|
|
|
os.environ["NEO4J_PASSWORD"] = NEO4J_PASSWORD
|
|
|
|
|
|
|
|
|
|
|
|
# 清库
|
|
|
|
# 清库 【视情况看看是不是需要注释掉】
|
|
|
|
executor = Neo4jExecutor.create_default()
|
|
|
|
executor = Neo4jExecutor.create_default()
|
|
|
|
executor.graph.run("MATCH (n) DETACH DELETE n")
|
|
|
|
executor.graph.run("MATCH (n) DETACH DELETE n")
|
|
|
|
print("清库成功")
|
|
|
|
print("清库成功")
|
|
|
|
|
|
|
|
|
|
|
|
# 注释掉或删除以下清理代码
|
|
|
|
# 注释掉或删除以下清理代码
|
|
|
|
files_to_delete = [
|
|
|
|
# files_to_delete = [
|
|
|
|
"graph_chunk_entity_relation.graphml",
|
|
|
|
# "graph_chunk_entity_relation.graphml",
|
|
|
|
"kv_store_doc_status.json",
|
|
|
|
# "kv_store_doc_status.json",
|
|
|
|
"kv_store_full_docs.json",
|
|
|
|
# "kv_store_full_docs.json",
|
|
|
|
"kv_store_text_chunks.json",
|
|
|
|
# "kv_store_text_chunks.json",
|
|
|
|
"vdb_chunks.json",
|
|
|
|
# "vdb_chunks.json",
|
|
|
|
"vdb_entities.json",
|
|
|
|
# "vdb_entities.json",
|
|
|
|
"vdb_relationships.json",
|
|
|
|
# "vdb_relationships.json",
|
|
|
|
]
|
|
|
|
# ]
|
|
|
|
|
|
|
|
|
|
|
|
# 获取docx文件的内容
|
|
|
|
# 获取docx文件的内容
|
|
|
|
content = get_docx_content_by_pandoc(docx_file)
|
|
|
|
content = get_docx_content_by_pandoc(docx_file)
|
|
|
|
|
|
|
|
|
|
|
|
# 删除文件
|
|
|
|
# 删除文件
|
|
|
|
for file in files_to_delete:
|
|
|
|
# for file in files_to_delete:
|
|
|
|
file_path = os.path.join(WORKING_DIR, file)
|
|
|
|
# file_path = os.path.join(WORKING_DIR, file)
|
|
|
|
if os.path.exists(file_path):
|
|
|
|
# if os.path.exists(file_path):
|
|
|
|
os.remove(file_path)
|
|
|
|
# os.remove(file_path)
|
|
|
|
print(f"Deleting old file:: {file_path}")
|
|
|
|
# print(f"Deleting old file:: {file_path}")
|
|
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
# 注意:默认设置使用NetworkX,使用Neo4J实现初始化LightRAG
|
|
|
|
# 注意:默认设置使用NetworkX,使用Neo4J实现初始化LightRAG
|
|
|
|