From 669d6f3469ca9ce852fc95b18f97d1da177b93dc Mon Sep 17 00:00:00 2001
From: Tushar Saxena <019saxenatushar@gmail.com>
Date: Sat, 16 May 2026 17:02:13 +0530
Subject: [PATCH] Fix MCP soul_forget deletion count reporting

Signed-off-by: Tushar Saxena <019saxenatushar@gmail.com>
---
 src/soul_protocol/mcp/server.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/soul_protocol/mcp/server.py b/src/soul_protocol/mcp/server.py
index 8beca86..b4bf4d5 100644
--- a/src/soul_protocol/mcp/server.py
+++ b/src/soul_protocol/mcp/server.py
@@ -1220,10 +1220,14 @@ async def soul_forget(
             "status": "deleted",
             "soul": s.name,
             "query": query,
-            "total_deleted": result.get("total_deleted", 0),
-            "tiers": result.get("tiers", {}),
-        }
-    )
+            "total_deleted": result.get("total", 0),
+            "tiers": {
+                k: len(v)
+                for k, v in result.items()
+                if k != "total" and isinstance(v, list)
+            },
+         }
+     )
 
 
 @mcp.tool