refactor: bulk AST replacement and fault-tolerant graph handlers

server/src/handlers/graph.rs

@@ -218,19 +218,25 @@ impl McpTool for UpdateEntitiesHandler {
 
         state.modify_graph(|g| {
             for update in req.updates {
-                if !g.entities.contains_key(&update.name) {
-                    not_found.push(update.name.clone());
-                    continue;
+                let mut target_name = update.name.clone();
+                if !g.entities.contains_key(&target_name) {
+                    let lower_target = target_name.to_lowercase();
+                    if let Some(matched_key) = g.entities.keys().find(|k| k.to_lowercase() == lower_target).cloned() {
+                        target_name = matched_key;
+                    } else {
+                        not_found.push(update.name.clone());
+                        continue;
+                    }
                 }
 
                 if let Some(new_name) = &update.new_name {
-                    if update.name != *new_name && g.entities.contains_key(new_name) {
+                    if target_name != *new_name && g.entities.contains_key(new_name) {
                         conflict_names.push(new_name.clone());
                         continue;
                     }
                 }
 
-                if let Some(mut entity) = g.entities.remove(&update.name) {
+                if let Some(mut entity) = g.entities.remove(&target_name) {
                     let mut renamed = false;
                     let old_name = entity.name.clone();
 
@@ -269,18 +275,6 @@ impl McpTool for UpdateEntitiesHandler {
             }
         });
 
-        if !not_found.is_empty() {
-            return Err(crate::error::AppError::Internal(format!(
-                "Error: Entities not found: {}", not_found.join(", ")
-            )));
-        }
-
-        if !conflict_names.is_empty() {
-            return Err(crate::error::AppError::Internal(format!(
-                "Error: Cannot rename to existing entity names: {}", conflict_names.join(", ")
-            )));
-        }
-
         let idx = state.get_search_index().await;
         for old_name in deleted_names {
             drop(idx.delete_document(&old_name));
@@ -291,7 +285,16 @@ impl McpTool for UpdateEntitiesHandler {
         }
 
         let names: Vec<String> = updated_entities.iter().map(|e| e.name.clone()).collect();
-        Ok(format!("Successfully updated {} entity/entities: {}", names.len(), names.join(", ")))
+        let mut msg = format!("Successfully updated {} entity/entities: {}", names.len(), names.join(", "));
+        
+        if !not_found.is_empty() {
+            msg.push_str(&format!("\nNote: {} entities were not found and skipped: {}", not_found.len(), not_found.join(", ")));
+        }
+        if !conflict_names.is_empty() {
+            msg.push_str(&format!("\nNote: {} entity renames were skipped due to name conflicts: {}", conflict_names.len(), conflict_names.join(", ")));
+        }
+        
+        Ok(msg)
     }
 }
 
@@ -459,37 +462,43 @@ impl McpTool for DeleteEntitiesHandler {
 
     async fn execute(&self, args: Value, state: Arc<MemoryState>) -> crate::error::Result<String> {
         let req: DeleteEntitiesTool = serde_json::from_value(args).map_err(|e| crate::error::AppError::BadRequest(format!("Schema validation failed. Your JSON arguments do not match the expected tool schema: {}", e)))?;
-        let to_delete: std::collections::HashSet<_> = req.entity_names.into_iter().collect();
-        let mut missing = Vec::new();
-        state.read_graph(|g| {
-            for name in &to_delete {
-                if !g.entities.contains_key(name) {
-                    missing.push(name.clone());
-                }
-            }
-        });
-
-        if !missing.is_empty() {
-            return Err(crate::error::AppError::Internal(format!(
-                "Error: Entities not found: {}. Please use the search_nodes or read_graph tools to verify the exact entity names.",
-                missing.join(", ")
-            )));
-        }
+        
+        let mut actual_deletes = Vec::new();
+        let mut not_found = Vec::new();
 
         state.modify_graph(|master| {
-            for name in &to_delete {
+            for target in req.entity_names {
+                if master.entities.contains_key(&target) {
+                    actual_deletes.push(target);
+                } else {
+                    let lower_target = target.to_lowercase();
+                    if let Some(matched_key) = master.entities.keys().find(|k| k.to_lowercase() == lower_target).cloned() {
+                        actual_deletes.push(matched_key);
+                    } else {
+                        not_found.push(target);
+                    }
+                }
+            }
+
+            for name in &actual_deletes {
                 master.entities.remove(name);
             }
+            let delete_set: std::collections::HashSet<_> = actual_deletes.iter().cloned().collect();
             master
                 .relations
-                .retain(|r| !to_delete.contains(&r.from) && !to_delete.contains(&r.to));
+                .retain(|r| !delete_set.contains(&r.from) && !delete_set.contains(&r.to));
         });
 
         let idx = state.get_search_index().await;
-        for name in to_delete {
-            drop(idx.delete_document(&name));
+        for name in &actual_deletes {
+            drop(idx.delete_document(name));
         }
-        Ok("Entities deleted".to_string())
+        
+        let mut msg = format!("Successfully deleted {} entities.", actual_deletes.len());
+        if !not_found.is_empty() {
+            msg.push_str(&format!(" Note: {} entities were not found and skipped: {}", not_found.len(), not_found.join(", ")));
+        }
+        Ok(msg)
     }
 }
 
@@ -581,7 +590,12 @@ impl McpTool for DeleteRelationsHandler {
             let initial_len = master.relations.len();
             master.relations.retain(|r| {
                 let should_delete = req.relations.iter().any(|target| {
-                    target.from == r.from && target.to == r.to && target.relation_type == r.relation_type
+                    if target.from == r.from && target.to == r.to && target.relation_type == r.relation_type {
+                        return true;
+                    }
+                    target.from.eq_ignore_ascii_case(&r.from) 
+                        && target.to.eq_ignore_ascii_case(&r.to) 
+                        && target.relation_type.eq_ignore_ascii_case(&r.relation_type)
                 });
                 !should_delete
             });
@@ -589,13 +603,11 @@ impl McpTool for DeleteRelationsHandler {
         });
         
         let missing_count = requested_count.saturating_sub(deleted_count);
+        let mut msg = format!("Successfully deleted {} relations.", deleted_count);
         if missing_count > 0 {
-            return Err(crate::error::AppError::Internal(format!(
-                "Error: {} relation(s) not found in graph. Please verify exact relation properties (from, to, relation_type) using read_graph or get_subgraph.",
-                missing_count
-            )));
+            msg.push_str(&format!(" Note: {} relations were not found and skipped.", missing_count));
         }
-        Ok("Relations deleted".to_string())
+        Ok(msg)
     }
 }
 
@@ -1601,7 +1613,7 @@ mod tests {
             .await
             .map_err(|e| crate::error::AppError::Internal(e.to_string()))
             .unwrap();
-        assert_eq!(res4, "Entities deleted");
+        assert_eq!(res4, "Successfully deleted 1 entities.");
 
         let res5 = read_graph
             .execute(json!({"namespace": "global"}), state.clone())
@@ -1774,7 +1786,7 @@ mod tests {
             )
             .await
             .unwrap();
-        assert_eq!(del_rel_res, "Relations deleted");
+        assert_eq!(del_rel_res, "Successfully deleted 1 relations.");
 
         let bcast_handler = AgentSignalsHandler;
         let bcast_res = bcast_handler