Adjust shared state import

Author: moonbox3

python/packages/core/tests/workflow/test_edge.py

@@ -10,7 +10,6 @@
     Executor,
     InProcRunnerContext,
     Message,
-    SharedState,
     WorkflowContext,
     handler,
 )
@@ -24,6 +23,7 @@
     SwitchCaseEdgeGroupDefault,
 )
 from agent_framework._workflows._edge_runner import create_edge_runner
+from agent_framework._workflows._shared_state import SharedState
 from agent_framework.observability import EdgeGroupDeliveryStatus
 
 # Add for test

python/packages/core/tests/workflow/test_workflow_states.py

@@ -8,7 +8,6 @@
     ExecutorFailedEvent,
     InProcRunnerContext,
     RequestInfoEvent,
-    SharedState,
     Workflow,
     WorkflowBuilder,
     WorkflowContext,
@@ -20,6 +19,7 @@
     WorkflowStatusEvent,
     handler,
 )
+from agent_framework._workflows._shared_state import SharedState
 
 
 class FailingExecutor(Executor):

python/packages/declarative/agent_framework_declarative/_workflows/_declarative_base.py

@@ -32,9 +32,9 @@
 
 from agent_framework._workflows import (
     Executor,
-    SharedState,
     WorkflowContext,
 )
+from agent_framework._workflows._shared_state import SharedState
 from powerfx import Engine
 
 if sys.version_info >= (3, 11):