Remove redandent code. (#5121)

### What problem does this PR solve?

#5107

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2025-02-19 15:46:52 +08:00 committed by GitHub
parent 3ced290eb5
commit ef95f08c48
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -8,7 +8,6 @@ Reference:
import logging import logging
import json import json
import re import re
import traceback
from typing import Callable from typing import Callable
from dataclasses import dataclass from dataclasses import dataclass
import networkx as nx 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.extractor import Extractor
from graphrag.general.leiden import add_community_info2graph from graphrag.general.leiden import add_community_info2graph
from rag.llm.chat_model import Base as CompletionLLM 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 rag.utils import num_tokens_from_string
from timeit import default_timer as timer from timeit import default_timer as timer
@ -36,7 +35,6 @@ class CommunityReportsExtractor(Extractor):
_extraction_prompt: str _extraction_prompt: str
_output_formatter_prompt: str _output_formatter_prompt: str
_on_error: ErrorHandlerFn
_max_report_length: int _max_report_length: int
def __init__( def __init__(
@ -107,9 +105,8 @@ class CommunityReportsExtractor(Extractor):
continue continue
response["weight"] = weight response["weight"] = weight
response["entities"] = ents response["entities"] = ents
except Exception as e: except Exception:
logging.exception("CommunityReportsExtractor got exception") logging.exception("CommunityReportsExtractor got exception")
self._on_error(e, traceback.format_exc(), None)
continue continue
add_community_info2graph(graph, ents, response["title"]) add_community_info2graph(graph, ents, response["title"])