马一丁

Align graph API lookup path with GraphStorage output dir

@@ -292,7 +292,16 @@ class GraphStorage: @@ -292,7 +292,16 @@ class GraphStorage:
292 """图谱存储管理器""" 292 """图谱存储管理器"""
293 293
294 FILENAME = "graphrag.json" 294 FILENAME = "graphrag.json"
295 - DEFAULT_CHAPTERS_DIR = Path("chapters") 295 +
  296 + @property
  297 + def chapters_dir(self) -> Path:
  298 + """获取章节目录路径(与 ChapterStorage 保持一致)"""
  299 + try:
  300 + from ..utils.config import settings
  301 + return Path(settings.CHAPTER_OUTPUT_DIR)
  302 + except ImportError:
  303 + # 回退到默认值
  304 + return Path("final_reports/chapters")
296 305
297 def save(self, graph: Graph, task_id: str, run_dir: Path) -> Path: 306 def save(self, graph: Graph, task_id: str, run_dir: Path) -> Path:
298 """ 307 """
@@ -363,8 +372,8 @@ class GraphStorage: @@ -363,8 +372,8 @@ class GraphStorage:
363 Returns: 372 Returns:
364 图谱文件路径,未找到返回 None 373 图谱文件路径,未找到返回 None
365 """ 374 """
366 - # 在默认目录中搜索  
367 - chapters_dir = self.DEFAULT_CHAPTERS_DIR 375 + # 在章节目录中搜索(与 ChapterStorage 保持一致)
  376 + chapters_dir = self.chapters_dir
368 if not chapters_dir.exists(): 377 if not chapters_dir.exists():
369 return None 378 return None
370 379
@@ -388,7 +397,7 @@ class GraphStorage: @@ -388,7 +397,7 @@ class GraphStorage:
388 Returns: 397 Returns:
389 最新图谱文件路径,未找到返回 None 398 最新图谱文件路径,未找到返回 None
390 """ 399 """
391 - chapters_dir = self.DEFAULT_CHAPTERS_DIR 400 + chapters_dir = self.chapters_dir
392 if not chapters_dir.exists(): 401 if not chapters_dir.exists():
393 return None 402 return None
394 403
@@ -416,7 +425,7 @@ class GraphStorage: @@ -416,7 +425,7 @@ class GraphStorage:
416 Returns: 425 Returns:
417 图谱信息列表,包含路径、报告ID、创建时间等 426 图谱信息列表,包含路径、报告ID、创建时间等
418 """ 427 """
419 - chapters_dir = self.DEFAULT_CHAPTERS_DIR 428 + chapters_dir = self.chapters_dir
420 if not chapters_dir.exists(): 429 if not chapters_dir.exists():
421 return [] 430 return []
422 431