Showing
1 changed file
with
14 additions
and
5 deletions
| @@ -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 |
-
Please register or login to post a comment