diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index cd472922..9b22a9b8 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -1152,9 +1152,6 @@ class LightRAG: try: chunk_results = await extract_entities( chunk, - knowledge_graph_inst=self.chunk_entity_relation_graph, - entity_vdb=self.entities_vdb, - relationships_vdb=self.relationships_vdb, global_config=asdict(self), pipeline_status=pipeline_status, pipeline_status_lock=pipeline_status_lock, diff --git a/lightrag/operate.py b/lightrag/operate.py index 2e583525..4f828b6b 100644 --- a/lightrag/operate.py +++ b/lightrag/operate.py @@ -616,9 +616,6 @@ async def merge_nodes_and_edges( async def extract_entities( chunks: dict[str, TextChunkSchema], - knowledge_graph_inst: BaseGraphStorage, - entity_vdb: BaseVectorStorage, - relationships_vdb: BaseVectorStorage, global_config: dict[str, str], pipeline_status: dict = None, pipeline_status_lock=None,