浏览代码

Remove redandent code. (#5121)

### What problem does this PR solve?

#5107

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.17.0
Kevin Hu 8 个月前
父节点
当前提交
ef95f08c48
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 2 次插入5 次删除
  1. 2
    5
      graphrag/general/community_reports_extractor.py

+ 2
- 5
graphrag/general/community_reports_extractor.py 查看文件

@@ -8,7 +8,6 @@ Reference:
import logging
import json
import re
import traceback
from typing import Callable
from dataclasses import dataclass
import networkx as nx
@@ -18,7 +17,7 @@ from graphrag.general.community_report_prompt import COMMUNITY_REPORT_PROMPT
from graphrag.general.extractor import Extractor
from graphrag.general.leiden import add_community_info2graph
from rag.llm.chat_model import Base as CompletionLLM
from graphrag.utils import ErrorHandlerFn, perform_variable_replacements, dict_has_keys_with_types
from graphrag.utils import perform_variable_replacements, dict_has_keys_with_types
from rag.utils import num_tokens_from_string
from timeit import default_timer as timer

@@ -36,7 +35,6 @@ class CommunityReportsExtractor(Extractor):

_extraction_prompt: str
_output_formatter_prompt: str
_on_error: ErrorHandlerFn
_max_report_length: int

def __init__(
@@ -107,9 +105,8 @@ class CommunityReportsExtractor(Extractor):
continue
response["weight"] = weight
response["entities"] = ents
except Exception as e:
except Exception:
logging.exception("CommunityReportsExtractor got exception")
self._on_error(e, traceback.format_exc(), None)
continue

add_community_info2graph(graph, ents, response["title"])

正在加载...
取消
保存