Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update output path logic since GraphRAG has changed the storage config value in the latest release (#374) bump:patch

Browse Source 
Co-authored-by: Chen, Ron Gang <git@git.com>
This commit is contained in:
ronchengang 2024-10-10 12:20:02 +08:00 committed by GitHub
parent 49a083fd9f
commit ad34395d0b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libs/ktem/ktem/index/file/graph/pipelines.py
View File

@ -177,15 +177,8 @@ class GraphRAGRetrieverPipeline(BaseFileIndexRetriever):
root_path, _ = prepare_graph_index_path(graph_id) root_path, _ = prepare_graph_index_path(graph_id)
output_path = root_path / "output" output_path = root_path / "output"
child_paths = sorted(
list(output_path.iterdir()), key=lambda x: x.stem, reverse=True
)
# get the latest child path INPUT_DIR = output_path
assert child_paths, "GraphRAG index output not found"
latest_child_path = Path(child_paths[0]) / "artifacts"
INPUT_DIR = latest_child_path
LANCEDB_URI = str(INPUT_DIR / "lancedb") LANCEDB_URI = str(INPUT_DIR / "lancedb")
COMMUNITY_REPORT_TABLE = "create_final_community_reports" COMMUNITY_REPORT_TABLE = "create_final_community_reports"
ENTITY_TABLE = "create_final_nodes" ENTITY_TABLE = "create_final_nodes"