gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'main' into validate-content-before-enqueue

Browse Source
This commit is contained in:
yangdx
2025-03-03 01:34:55 +08:00
parent aa5888042e 2d6d0be128
commit e854313438
1 changed files with 3 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
lightrag/lightrag.py
View File

@@ -1013,7 +1013,7 @@ class LightRAG:
# Insert entities into knowledge graph # Insert entities into knowledge graph
all_entities_data: list[dict[str, str]] = [] all_entities_data: list[dict[str, str]] = []
for entity_data in custom_kg.get("entities", []): for entity_data in custom_kg.get("entities", []):
entity_name = f'"{entity_data["entity_name"].upper()}"' entity_name = entity_data["entity_name"]
entity_type = entity_data.get("entity_type", "UNKNOWN") entity_type = entity_data.get("entity_type", "UNKNOWN")
description = entity_data.get("description", "No description provided") description = entity_data.get("description", "No description provided")
# source_id = entity_data["source_id"] # source_id = entity_data["source_id"]
@@ -1043,8 +1043,8 @@ class LightRAG:
# Insert relationships into knowledge graph # Insert relationships into knowledge graph
all_relationships_data: list[dict[str, str]] = [] all_relationships_data: list[dict[str, str]] = []
for relationship_data in custom_kg.get("relationships", []): for relationship_data in custom_kg.get("relationships", []):
src_id = f'"{relationship_data["src_id"].upper()}"' src_id = relationship_data["src_id"]
tgt_id = f'"{relationship_data["tgt_id"].upper()}"' tgt_id = relationship_data["tgt_id"]
description = relationship_data["description"] description = relationship_data["description"]
keywords = relationship_data["keywords"] keywords = relationship_data["keywords"]
weight = relationship_data.get("weight", 1.0) weight = relationship_data.get("weight", 1.0)
@@ -1346,8 +1346,6 @@ class LightRAG:
return loop.run_until_complete(self.adelete_by_entity(entity_name)) return loop.run_until_complete(self.adelete_by_entity(entity_name))
async def adelete_by_entity(self, entity_name: str) -> None: async def adelete_by_entity(self, entity_name: str) -> None:
entity_name = f'"{entity_name.upper()}"'
try: try:
await self.entities_vdb.delete_entity(entity_name) await self.entities_vdb.delete_entity(entity_name)
await self.relationships_vdb.delete_entity_relation(entity_name) await self.relationships_vdb.delete_entity_relation(entity_name)
@@ -1642,7 +1640,6 @@ class LightRAG:
- graph_data: Complete node data from the graph database - graph_data: Complete node data from the graph database
- vector_data: (optional) Data from the vector database - vector_data: (optional) Data from the vector database
""" """
entity_name = f'"{entity_name.upper()}"'
# Get information from the graph # Get information from the graph
node_data = await self.chunk_entity_relation_graph.get_node(entity_name) node_data = await self.chunk_entity_relation_graph.get_node(entity_name)
@@ -1680,8 +1677,6 @@ class LightRAG:
- graph_data: Complete edge data from the graph database - graph_data: Complete edge data from the graph database
- vector_data: (optional) Data from the vector database - vector_data: (optional) Data from the vector database
""" """
src_entity = f'"{src_entity.upper()}"'
tgt_entity = f'"{tgt_entity.upper()}"'
# Get information from the graph # Get information from the graph
edge_data = await self.chunk_entity_relation_graph.get_edge( edge_data = await self.chunk_entity_relation_graph.get_edge(