Update websocket_storage.py

websocket_storage.py

@@ -236,15 +236,10 @@ class WebSocketGPUStorage:
             # Store state in memory
             operation = {
                 'operation': 'state',
-                'type': 'write',
-                'key': state_key,
-                'data': state_data,
-                'timestamp': time.time(),
-                'metadata': {
-                    'component': component,
-                    'state_id': state_id,
-                    'storage_type': 'memory'
-                }
+                'type': 'save',
+                'component': component,
+                'state_id': state_id,
+                'data': state_data
             }
             
             response = self._send_operation(operation)