refactor: Rename context to runtime_data with enhanced organization (#537)

Refactors the automation context module to runtime_data with better
separation of concerns between state management and session storage.

Changes:
- Renamed: context/ → runtime_data/
- Renamed: structural_context.py → state.py
- Renamed: test_structural_context.py → test_runtime_data.py
- Added: runtime_data/session.py (new session storage module)

New Structure:
- runtime_data/state.py: Workflow execution state (WorkflowContext, StepResult, StepStatus)
- runtime_data/session.py: Session storage (SessionData, SessionStorage)
- runtime_data/__init__.py: Unified exports for both state and session

Features:
- SessionData: Session-level data container with metadata
- SessionStorage: In-memory session manager with CRUD operations
- get_storage(): Global session storage accessor
- Maintains backward compatibility (same exports, new location)

Updated all imports across codebase:
- 9 Python files updated to use runtime_data
- All tests passing (141/141)

🤖 Generated with Claude Code

Co-authored-by: Claude <[email protected]>

Author: plusplusoneplusplus

plugins/automation/__init__.py

@@ -12,7 +12,7 @@
 
 from .agents import ExploreAgent, ExploreAgentConfig, explore_codebase
 from .tools import AgentTool
-from .context import WorkflowContext, StepResult, StepStatus
+from .runtime_data import WorkflowContext, StepResult, StepStatus
 
 __all__ = [
     "ExploreAgent",

plugins/automation/context/__init__.py

@@ -0,0 +1,24 @@
+"""
+Runtime Data Module
+
+Manages runtime data for automation workflows, including:
+- State: Structural context (WorkflowContext, StepResult, StepStatus)
+- Session: Session-level storage and persistence
+"""
+
+# State management (structural context)
+from .state import StepResult, StepStatus, WorkflowContext
+
+# Session storage
+from .session import SessionData, SessionStorage, get_storage
+
+__all__ = [
+    # State
+    "StepResult",
+    "StepStatus",
+    "WorkflowContext",
+    # Session
+    "SessionData",
+    "SessionStorage",
+    "get_storage",
+]

plugins/automation/runtime_data/__init__.py

@@ -0,0 +1,156 @@
+"""
+Session Storage Module
+
+Manages session-level storage and persistence for automation workflows.
+"""
+
+from typing import Any, Dict, Optional
+from datetime import datetime
+from dataclasses import dataclass, field
+
+
+@dataclass
+class SessionData:
+    """Session storage for workflow execution context."""
+
+    session_id: str
+    created_at: datetime = field(default_factory=datetime.utcnow)
+    updated_at: datetime = field(default_factory=datetime.utcnow)
+    data: Dict[str, Any] = field(default_factory=dict)
+    metadata: Dict[str, Any] = field(default_factory=dict)
+
+    def get(self, key: str, default: Any = None) -> Any:
+        """
+        Get value from session data.
+
+        Args:
+            key: Key to retrieve
+            default: Default value if key not found
+
+        Returns:
+            Value from session data or default
+        """
+        return self.data.get(key, default)
+
+    def set(self, key: str, value: Any) -> None:
+        """
+        Set value in session data.
+
+        Args:
+            key: Key to set
+            value: Value to store
+        """
+        self.data[key] = value
+        self.updated_at = datetime.utcnow()
+
+    def delete(self, key: str) -> None:
+        """
+        Delete value from session data.
+
+        Args:
+            key: Key to delete
+        """
+        if key in self.data:
+            del self.data[key]
+            self.updated_at = datetime.utcnow()
+
+    def clear(self) -> None:
+        """Clear all session data."""
+        self.data.clear()
+        self.updated_at = datetime.utcnow()
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert to dictionary."""
+        return {
+            "session_id": self.session_id,
+            "created_at": self.created_at.isoformat(),
+            "updated_at": self.updated_at.isoformat(),
+            "data": self.data,
+            "metadata": self.metadata,
+        }
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "SessionData":
+        """Create from dictionary."""
+        return cls(
+            session_id=data["session_id"],
+            created_at=datetime.fromisoformat(data["created_at"]),
+            updated_at=datetime.fromisoformat(data["updated_at"]),
+            data=data.get("data", {}),
+            metadata=data.get("metadata", {}),
+        )
+
+
+class SessionStorage:
+    """In-memory session storage manager."""
+
+    def __init__(self):
+        """Initialize session storage."""
+        self._sessions: Dict[str, SessionData] = {}
+
+    def create_session(self, session_id: str, metadata: Optional[Dict[str, Any]] = None) -> SessionData:
+        """
+        Create a new session.
+
+        Args:
+            session_id: Unique session identifier
+            metadata: Optional metadata for the session
+
+        Returns:
+            Created SessionData instance
+        """
+        session = SessionData(
+            session_id=session_id,
+            metadata=metadata or {}
+        )
+        self._sessions[session_id] = session
+        return session
+
+    def get_session(self, session_id: str) -> Optional[SessionData]:
+        """
+        Get session by ID.
+
+        Args:
+            session_id: Session identifier
+
+        Returns:
+            SessionData if found, None otherwise
+        """
+        return self._sessions.get(session_id)
+
+    def delete_session(self, session_id: str) -> bool:
+        """
+        Delete session by ID.
+
+        Args:
+            session_id: Session identifier
+
+        Returns:
+            True if deleted, False if not found
+        """
+        if session_id in self._sessions:
+            del self._sessions[session_id]
+            return True
+        return False
+
+    def list_sessions(self) -> list[str]:
+        """
+        List all session IDs.
+
+        Returns:
+            List of session IDs
+        """
+        return list(self._sessions.keys())
+
+    def clear_all(self) -> None:
+        """Clear all sessions."""
+        self._sessions.clear()
+
+
+# Global session storage instance
+_global_storage = SessionStorage()
+
+
+def get_storage() -> SessionStorage:
+    """Get global session storage instance."""
+    return _global_storage

plugins/automation/runtime_data/session.py

@@ -5,7 +5,7 @@
 import pytest
 from datetime import datetime
 
-from plugins.automation.context import (
+from plugins.automation.runtime_data import (
     WorkflowContext,
     StepResult,
     StepStatus,

…automation/context/structural_context.py plugins/automation/runtime_data/state.pyplugins/automation/context/structural_context.py renamed to plugins/automation/runtime_data/state.py plugins/automation/context/structural_context.py renamed to plugins/automation/runtime_data/state.py

@@ -11,7 +11,7 @@
     WorkflowContext,
 )
 from plugins.automation.workflows.engine import WorkflowStatus
-from plugins.automation.context import StepStatus
+from plugins.automation.runtime_data import StepStatus
 
 
 class TestWorkflowEngine:

…omation/tests/test_structural_context.py …ns/automation/tests/test_runtime_data.pyplugins/automation/tests/test_structural_context.py renamed to plugins/automation/tests/test_runtime_data.py plugins/automation/tests/test_structural_context.py renamed to plugins/automation/tests/test_runtime_data.py

@@ -11,7 +11,7 @@
 - MCP tool interface for workflow execution
 """
 
-from ..context import WorkflowContext, StepResult
+from ..runtime_data import WorkflowContext, StepResult
 from .definition import WorkflowDefinition
 from .engine import WorkflowEngine, WorkflowExecutionResult
 

plugins/automation/tests/test_workflow_engine.py

@@ -11,7 +11,7 @@
 from typing import Any, Dict, List, Optional
 from dataclasses import dataclass, field
 
-from ..context import WorkflowContext, StepResult, StepStatus
+from ..runtime_data import WorkflowContext, StepResult, StepStatus
 from .definition import WorkflowDefinition, StepDefinition
 from .steps import BaseStep, AgentStep, TransformStep
 

plugins/automation/workflows/__init__.py

@@ -8,7 +8,7 @@
 from typing import Any, Dict
 
 from .base import BaseStep
-from ...context import WorkflowContext, StepResult, StepStatus
+from ...runtime_data import WorkflowContext, StepResult, StepStatus
 from ..definition import StepDefinition
 from ...agents import ExploreAgent, ExploreAgentConfig
 from utils.agent import CLIType