demo seed change

services/demo_session/app/api/internal.py

@@ -9,7 +9,7 @@ import structlog
 
 from app.core import get_db, settings
 from app.core.redis_wrapper import get_redis, DemoRedisWrapper
-from app.services.data_cloner import DemoDataCloner
+from app.services.cleanup_service import DemoCleanupService
 
 logger = structlog.get_logger()
 router = APIRouter()
@@ -41,24 +41,31 @@ async def cleanup_demo_session_internal(
 
         if not all([tenant_id, session_id]):
             raise HTTPException(
-                status_code=400, 
+                status_code=400,
                 detail="Missing required parameters: tenant_id, session_id"
             )
 
         logger.info(
-            "Internal cleanup requested", 
+            "Internal cleanup requested",
             tenant_id=tenant_id,
             session_id=session_id
         )
 
-        data_cloner = DemoDataCloner(db, redis)
-        
+        cleanup_service = DemoCleanupService(db, redis)
+
+        # Validate required fields
+        if not tenant_id or not session_id:
+            raise ValueError("tenant_id and session_id are required")
+
         # Delete session data for this tenant
-        await data_cloner.delete_session_data(
-            str(tenant_id),
-            session_id
+        await cleanup_service._delete_tenant_data(
+            tenant_id=str(tenant_id),
+            session_id=str(session_id)
         )
 
+        # Delete Redis data
+        await redis.delete_session_data(str(session_id))
+
         logger.info(
             "Internal cleanup completed",
             tenant_id=tenant_id,
@@ -73,7 +80,7 @@ async def cleanup_demo_session_internal(
 
     except Exception as e:
         logger.error(
-            "Internal cleanup failed", 
+            "Internal cleanup failed",
             error=str(e),
             tenant_id=cleanup_request.get('tenant_id'),
             session_id=cleanup_request.get('session_id'),