From 5198240cf97324b808392941f23dbb4611362efa Mon Sep 17 00:00:00 2001
From: zrguo <zrguo.bupt@qq.com>
Date: Sun, 2 Mar 2025 14:23:06 +0800
Subject: [PATCH 1/2] remove " and upper()

---
 lightrag/lightrag.py | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index 15338874..ea6bdb88 100644
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -880,7 +880,7 @@ class LightRAG:
             # Insert entities into knowledge graph
             all_entities_data: list[dict[str, str]] = []
             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")
                 description = entity_data.get("description", "No description provided")
                 # source_id = entity_data["source_id"]
@@ -910,8 +910,8 @@ class LightRAG:
             # Insert relationships into knowledge graph
             all_relationships_data: list[dict[str, str]] = []
             for relationship_data in custom_kg.get("relationships", []):
-                src_id = f'"{relationship_data["src_id"].upper()}"'
-                tgt_id = f'"{relationship_data["tgt_id"].upper()}"'
+                src_id = relationship_data["src_id"]
+                tgt_id = relationship_data["tgt_id"]
                 description = relationship_data["description"]
                 keywords = relationship_data["keywords"]
                 weight = relationship_data.get("weight", 1.0)
@@ -1213,7 +1213,6 @@ class LightRAG:
         return loop.run_until_complete(self.adelete_by_entity(entity_name))
 
     async def adelete_by_entity(self, entity_name: str) -> None:
-        entity_name = f'"{entity_name.upper()}"'
 
         try:
             await self.entities_vdb.delete_entity(entity_name)
@@ -1509,7 +1508,6 @@ class LightRAG:
                 - graph_data: Complete node data from the graph database
                 - vector_data: (optional) Data from the vector database
         """
-        entity_name = f'"{entity_name.upper()}"'
 
         # Get information from the graph
         node_data = await self.chunk_entity_relation_graph.get_node(entity_name)
@@ -1547,8 +1545,6 @@ class LightRAG:
                 - graph_data: Complete edge data from the graph 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
         edge_data = await self.chunk_entity_relation_graph.get_edge(

From 265fc2c87f85cafaf1761649d8a2dca2dceac7f3 Mon Sep 17 00:00:00 2001
From: zrguo <zrguo.bupt@qq.com>
Date: Sun, 2 Mar 2025 14:24:54 +0800
Subject: [PATCH 2/2] fix linting

---
 lightrag/lightrag.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index ea6bdb88..805da1a2 100644
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -1213,7 +1213,6 @@ class LightRAG:
         return loop.run_until_complete(self.adelete_by_entity(entity_name))
 
     async def adelete_by_entity(self, entity_name: str) -> None:
-
         try:
             await self.entities_vdb.delete_entity(entity_name)
             await self.relationships_vdb.delete_entity_relation(entity_name)