Update websocket_storage.py

websocket_storage.py

@@ -235,7 +235,7 @@ class WebSocketGPUStorage:
             
             # Store state in memory
             operation = {
-                'operation': 'store_state',
+                'operation': 'state',
                 'type': 'write',
                 'key': state_key,
                 'data': state_data,
@@ -318,11 +318,15 @@ class WebSocketGPUStorage:
                 model_hash = self._calculate_model_hash(model_path)
 
             operation = {
-                'operation': 'load_model',
+                'operation': 'vram',
                 'type': 'write',
-                'model_name': model_name,
-                'model_hash': model_hash,
-                'model_data': model_data
+                'block_id': f"model_{model_name}",
+                'data': model_data,
+                'metadata': {
+                    'hash': model_hash,
+                    'model_name': model_name,
+                    'type': 'model'
+                }
             }
 
             response = self._send_operation(operation)