[BREAKING] Python: Refactor SharedState to State with sync methods and superstep caching (#3667)

* Refactor SharedState to State with sync methods and superstep caching

* Fixes

* Address PR feedback

* Remove dead links

* Fix lab test import

python/packages/devui/frontend/src/components/features/workflow/checkpoint-info-modal.tsx

@@ -86,8 +86,8 @@ export function CheckpointInfoModal({
     (cp) => cp.checkpoint_id === selectedCheckpointId
   );
 
-  const executorIds = fullCheckpoint?.shared_state?._executor_state
-    ? Object.keys(fullCheckpoint.shared_state._executor_state)
+  const executorIds = fullCheckpoint?.state?._executor_state
+    ? Object.keys(fullCheckpoint.state._executor_state)
     : [];
   const messageExecutors = fullCheckpoint?.messages
     ? Object.keys(fullCheckpoint.messages)
@@ -345,14 +345,14 @@ export function CheckpointInfoModal({
                       </div>
                     )}
 
-                  {/* Shared State */}
+                  {/* Workflow State */}
                   <div>
-                    <div className="text-sm font-medium mb-3">Shared State</div>
-                    {fullCheckpoint?.shared_state && Object.keys(fullCheckpoint.shared_state).filter(
+                    <div className="text-sm font-medium mb-3">Workflow State</div>
+                    {fullCheckpoint?.state && Object.keys(fullCheckpoint.state).filter(
                       (k) => k !== "_executor_state"
                     ).length > 0 ? (
                       <div className="flex flex-wrap gap-2">
-                        {Object.keys(fullCheckpoint.shared_state)
+                        {Object.keys(fullCheckpoint.state)
                           .filter((k) => k !== "_executor_state")
                           .map((key) => (
                             <Badge key={key} variant="secondary" className="font-mono text-xs">

python/packages/devui/frontend/src/types/index.ts

@@ -290,7 +290,7 @@ export interface FullCheckpoint {
   workflow_id: string;
   timestamp: string;
   messages: Record<string, unknown[]>;
-  shared_state: Record<string, unknown>;
+  state: Record<string, unknown>;
   pending_request_info_events: Record<string, PendingRequestInfoEvent>;
   iteration_count: number;
   metadata: Record<string, unknown>;

python/packages/devui/tests/test_checkpoints.py

@@ -106,7 +106,7 @@ class TestCheckpointConversationManager:
         from agent_framework._workflows._checkpoint import WorkflowCheckpoint
 
         checkpoint = WorkflowCheckpoint(
-            checkpoint_id=str(uuid.uuid4()), workflow_id=test_workflow.id, messages={}, shared_state={"test": "data"}
+            checkpoint_id=str(uuid.uuid4()), workflow_id=test_workflow.id, messages={}, state={"test": "data"}
         )
 
         # Get checkpoint storage for this conversation and save
@@ -144,7 +144,7 @@ class TestCheckpointConversationManager:
             checkpoint_id=str(uuid.uuid4()),
             workflow_id=test_workflow.id,
             messages={},
-            shared_state={"conversation": "A"},
+            state={"conversation": "A"},
         )
         storage_a = checkpoint_manager.get_checkpoint_storage(conv_a)
         await storage_a.save_checkpoint(checkpoint_a)
@@ -181,7 +181,7 @@ class TestCheckpointConversationManager:
                 checkpoint_id=str(uuid.uuid4()),
                 workflow_id=test_workflow.id,
                 messages={},
-                shared_state={"iteration": i},
+                state={"iteration": i},
             )
             saved_id = await storage.save_checkpoint(checkpoint)
             checkpoint_ids.append(saved_id)
@@ -217,7 +217,7 @@ class TestCheckpointConversationManager:
                 checkpoint_id=f"checkpoint_{i}",
                 workflow_id=test_workflow.id,
                 messages={},
-                shared_state={"iteration": i},
+                state={"iteration": i},
             )
             saved_id = await storage.save_checkpoint(checkpoint)
             checkpoint_ids.append(saved_id)
@@ -259,7 +259,7 @@ class TestCheckpointConversationManager:
             checkpoint_id=str(uuid.uuid4()),
             workflow_id=test_workflow.id,
             messages={},
-            shared_state={"test_key": "test_value"},
+            state={"test_key": "test_value"},
         )
 
         # Save to this session
@@ -272,7 +272,7 @@ class TestCheckpointConversationManager:
         assert loaded_checkpoint is not None
         assert loaded_checkpoint.checkpoint_id == original_checkpoint.checkpoint_id
         assert loaded_checkpoint.workflow_id == original_checkpoint.workflow_id
-        assert loaded_checkpoint.shared_state == {"test_key": "test_value"}
+        assert loaded_checkpoint.state == {"test_key": "test_value"}
 
 
 class TestCheckpointStorage:
@@ -298,7 +298,7 @@ class TestCheckpointStorage:
         from agent_framework._workflows._checkpoint import WorkflowCheckpoint
 
         checkpoint = WorkflowCheckpoint(
-            checkpoint_id=str(uuid.uuid4()), workflow_id=test_workflow.id, messages={}, shared_state={"test": "data"}
+            checkpoint_id=str(uuid.uuid4()), workflow_id=test_workflow.id, messages={}, state={"test": "data"}
         )
 
         # Test save_checkpoint
@@ -348,7 +348,7 @@ class TestIntegration:
         from agent_framework._workflows._checkpoint import WorkflowCheckpoint
 
         checkpoint = WorkflowCheckpoint(
-            checkpoint_id=str(uuid.uuid4()), workflow_id=test_workflow.id, messages={}, shared_state={"injected": True}
+            checkpoint_id=str(uuid.uuid4()), workflow_id=test_workflow.id, messages={}, state={"injected": True}
         )
         await checkpoint_storage.save_checkpoint(checkpoint)
 
@@ -381,7 +381,7 @@ class TestIntegration:
             checkpoint_id=str(uuid.uuid4()),
             workflow_id=test_workflow.id,
             messages={},
-            shared_state={"ready_to_resume": True},
+            state={"ready_to_resume": True},
         )
         checkpoint_id = await checkpoint_storage.save_checkpoint(checkpoint)
 
@@ -389,7 +389,7 @@ class TestIntegration:
         loaded = await checkpoint_storage.load_checkpoint(checkpoint_id)
         assert loaded is not None
         assert loaded.checkpoint_id == checkpoint_id
-        assert loaded.shared_state == {"ready_to_resume": True}
+        assert loaded.state == {"ready_to_resume": True}
 
         # Verify checkpoint is accessible via storage (for UI to list checkpoints)
         checkpoints = await checkpoint_storage.list_checkpoints()

python/packages/devui/tests/test_server.py

@@ -384,7 +384,7 @@ async def test_checkpoint_api_endpoints(test_entities_dir):
     checkpoint = WorkflowCheckpoint(
         checkpoint_id="test_checkpoint_1",
         workflow_id="test_workflow",
-        shared_state={"key": "value"},
+        state={"key": "value"},
         iteration_count=1,
     )
     await storage.save_checkpoint(checkpoint)