empathy-framework 5.2.1__py3-none-any.whl → 5.4.0__py3-none-any.whl

empathy_os/meta_workflows/session_context.py

@@ -1,397 +0,0 @@
-"""Session Context for Meta-Workflows
-
-Tracks user choices and preferences across workflow executions using short-term memory.
-Enables intelligent defaults and personalized recommendations based on recent history.
-
-Features:
-- Record form choices per template
-- Suggest defaults based on recent choices
-- Session isolation per user
-- TTL-based expiration (1 hour default)
-- Graceful fallback when memory unavailable
-
-Usage:
-    from empathy_os.meta_workflows.session_context import SessionContext
-    from empathy_os.memory.unified import UnifiedMemory
-
-    memory = UnifiedMemory(user_id="user@example.com")
-    session = SessionContext(memory=memory)
-
-    # Record choice
-    session.record_choice("python_package_publish", "has_tests", "Yes")
-
-    # Get suggested defaults
-    defaults = session.suggest_defaults("python_package_publish")
-
-Copyright 2025 Smart AI Memory, LLC
-Licensed under Fair Source 0.9
-"""
-
-import logging
-import uuid
-from datetime import datetime
-from typing import TYPE_CHECKING, Any
-
-if TYPE_CHECKING:
-    from empathy_os.memory.unified import UnifiedMemory
-    from empathy_os.meta_workflows.models import FormSchema
-
-logger = logging.getLogger(__name__)
-
-
-class SessionContext:
-    """Track session-level patterns and user preferences.
-
-    Uses short-term memory to record form choices and suggest intelligent
-    defaults for subsequent workflow executions.
-
-    Attributes:
-        memory: UnifiedMemory instance for storage
-        session_id: Unique session identifier
-        user_id: User identifier (from memory)
-        default_ttl: Time-to-live for session data (seconds)
-    """
-
-    def __init__(
-        self,
-        memory: "UnifiedMemory | None" = None,
-        session_id: str | None = None,
-        default_ttl: int = 3600,
-    ):
-        """Initialize session context.
-
-        Args:
-            memory: UnifiedMemory instance (optional, graceful fallback if None)
-            session_id: Optional session ID (generates new if None)
-            default_ttl: TTL for session data in seconds (default: 1 hour)
-        """
-        self.memory = memory
-        self.session_id = session_id or str(uuid.uuid4())
-        self.user_id = memory.user_id if memory else "anonymous"
-        self.default_ttl = default_ttl
-
-        logger.info(
-            f"SessionContext initialized: session_id={self.session_id}, "
-            f"user_id={self.user_id}, memory={'enabled' if memory else 'disabled'}"
-        )
-
-    def record_choice(
-        self,
-        template_id: str,
-        question_id: str,
-        choice: Any,
-        ttl: int | None = None,
-    ) -> bool:
-        """Record a form choice in short-term memory.
-
-        Args:
-            template_id: Template identifier
-            question_id: Question identifier
-            choice: User's choice (any JSON-serializable type)
-            ttl: Optional TTL override (uses default_ttl if None)
-
-        Returns:
-            True if recorded successfully, False otherwise
-        """
-        if not self.memory:
-            logger.debug("Memory not available, cannot record choice")
-            return False
-
-        try:
-            key = self._make_choice_key(template_id, question_id)
-            value = {
-                "choice": choice,
-                "timestamp": datetime.utcnow().isoformat(),
-                "template_id": template_id,
-                "question_id": question_id,
-            }
-
-            self.memory.stash(key, value, ttl_seconds=ttl or self.default_ttl)
-
-            logger.debug(
-                f"Recorded choice: template={template_id}, question={question_id}, choice={choice}"
-            )
-            return True
-
-        except Exception as e:
-            logger.error(f"Failed to record choice: {e}")
-            return False
-
-    def get_recent_choice(
-        self,
-        template_id: str,
-        question_id: str,
-    ) -> Any | None:
-        """Get the most recent choice for a specific question.
-
-        Args:
-            template_id: Template identifier
-            question_id: Question identifier
-
-        Returns:
-            Most recent choice, or None if not found
-        """
-        if not self.memory:
-            return None
-
-        try:
-            key = self._make_choice_key(template_id, question_id)
-            value = self.memory.retrieve(key)
-
-            if value and isinstance(value, dict):
-                return value.get("choice")
-
-            return None
-
-        except Exception as e:
-            logger.debug(f"Failed to get recent choice: {e}")
-            return None
-
-    def get_recent_choices(
-        self,
-        template_id: str,
-    ) -> dict[str, Any]:
-        """Get all recent choices for a template.
-
-        Args:
-            template_id: Template identifier
-
-        Returns:
-            Dict mapping question_id -> choice
-        """
-        if not self.memory:
-            return {}
-
-        try:
-            # Get all keys for this template and session
-            pattern = f"session:{self.session_id}:form:{template_id}:*"
-
-            # Note: This requires pattern matching support in short-term memory
-            # For now, we return empty dict (could be enhanced later)
-            logger.debug(f"Pattern matching not yet implemented: {pattern}")
-            return {}
-
-        except Exception as e:
-            logger.error(f"Failed to get recent choices: {e}")
-            return {}
-
-    def suggest_defaults(
-        self,
-        template_id: str,
-        form_schema: "FormSchema | None" = None,
-    ) -> dict[str, Any]:
-        """Suggest default values based on recent choices.
-
-        Args:
-            template_id: Template identifier
-            form_schema: Optional form schema for validation
-
-        Returns:
-            Dict mapping question_id -> suggested_default
-        """
-        if not self.memory:
-            return {}
-
-        try:
-            # Get recent choices for this template
-            recent_choices = self.get_recent_choices(template_id)
-
-            # If form_schema provided, validate suggestions
-            if form_schema:
-                validated_suggestions = {}
-                for question in form_schema.questions:
-                    if question.id in recent_choices:
-                        # Validate that suggestion is valid for question type
-                        choice = recent_choices[question.id]
-                        if self._validate_choice(choice, question):
-                            validated_suggestions[question.id] = choice
-
-                return validated_suggestions
-
-            return recent_choices
-
-        except Exception as e:
-            logger.error(f"Failed to suggest defaults: {e}")
-            return {}
-
-    def record_execution(
-        self,
-        template_id: str,
-        run_id: str,
-        success: bool,
-        cost: float,
-        duration: float,
-        ttl: int | None = None,
-    ) -> bool:
-        """Record workflow execution metadata.
-
-        Args:
-            template_id: Template identifier
-            run_id: Execution run ID
-            success: Whether execution succeeded
-            cost: Total cost in USD
-            duration: Duration in seconds
-            ttl: Optional TTL override
-
-        Returns:
-            True if recorded successfully
-        """
-        if not self.memory:
-            return False
-
-        try:
-            key = f"session:{self.session_id}:execution:{run_id}"
-            value = {
-                "template_id": template_id,
-                "run_id": run_id,
-                "success": success,
-                "cost": cost,
-                "duration": duration,
-                "timestamp": datetime.utcnow().isoformat(),
-            }
-
-            self.memory.stash(key, value, ttl_seconds=ttl or self.default_ttl)
-
-            logger.debug(f"Recorded execution: run_id={run_id}, success={success}")
-            return True
-
-        except Exception as e:
-            logger.error(f"Failed to record execution: {e}")
-            return False
-
-    def get_session_stats(self) -> dict[str, Any]:
-        """Get statistics for current session.
-
-        Returns:
-            Dict with session statistics (executions, success rate, etc.)
-        """
-        if not self.memory:
-            return {"session_id": self.session_id, "memory_enabled": False}
-
-        try:
-            stats = {
-                "session_id": self.session_id,
-                "user_id": self.user_id,
-                "memory_enabled": True,
-                "executions": 0,
-                "successful_executions": 0,
-                "total_cost": 0.0,
-                "total_duration": 0.0,
-            }
-
-            # Note: Would need to query all execution records
-            # For now, return basic stats
-            return stats
-
-        except Exception as e:
-            logger.error(f"Failed to get session stats: {e}")
-            return {"session_id": self.session_id, "error": str(e)}
-
-    def clear_session(self) -> bool:
-        """Clear all session data from memory.
-
-        Returns:
-            True if cleared successfully
-        """
-        if not self.memory:
-            return False
-
-        try:
-            # Note: Would need to delete all keys matching session pattern
-            # For now, just log
-            logger.info(f"Session clear requested: {self.session_id}")
-            # Implementation would require pattern delete support
-            return True
-
-        except Exception as e:
-            logger.error(f"Failed to clear session: {e}")
-            return False
-
-    # =========================================================================
-    # HELPER METHODS
-    # =========================================================================
-
-    def _make_choice_key(self, template_id: str, question_id: str) -> str:
-        """Create Redis key for a form choice.
-
-        Args:
-            template_id: Template identifier
-            question_id: Question identifier
-
-        Returns:
-            Redis key string
-        """
-        return f"session:{self.session_id}:form:{template_id}:{question_id}"
-
-    def _validate_choice(self, choice: Any, question: Any) -> bool:
-        """Validate that a choice is valid for a question.
-
-        Args:
-            choice: User's choice
-            question: FormQuestion object
-
-        Returns:
-            True if valid, False otherwise
-        """
-        try:
-            # Basic validation - could be enhanced
-            if hasattr(question, "options") and question.options:
-                # Check if choice is in options (only if options are defined)
-                if isinstance(choice, list):
-                    # Multi-select - all choices must be in options
-                    return all(c in question.options for c in choice)
-                else:
-                    # Single-select - choice must be in options
-                    return choice in question.options
-
-            # No specific validation - assume valid
-            return True
-
-        except Exception:
-            return False
-
-
-# =========================================================================
-# CONVENIENCE FUNCTIONS
-# =========================================================================
-
-
-def create_session_context(
-    memory: "UnifiedMemory | None" = None,
-    session_id: str | None = None,
-) -> SessionContext:
-    """Create a new session context.
-
-    Convenience function for creating SessionContext instances.
-
-    Args:
-        memory: UnifiedMemory instance (optional)
-        session_id: Optional session ID
-
-    Returns:
-        SessionContext instance
-    """
-    return SessionContext(memory=memory, session_id=session_id)
-
-
-def get_session_defaults(
-    template_id: str,
-    form_schema: "FormSchema | None" = None,
-    memory: "UnifiedMemory | None" = None,
-    session_id: str | None = None,
-) -> dict[str, Any]:
-    """Get suggested defaults for a template.
-
-    Convenience function for getting defaults without creating SessionContext.
-
-    Args:
-        template_id: Template identifier
-        form_schema: Optional form schema
-        memory: Optional UnifiedMemory instance
-        session_id: Optional session ID
-
-    Returns:
-        Dict of suggested defaults
-    """
-    session = SessionContext(memory=memory, session_id=session_id)
-    return session.suggest_defaults(template_id, form_schema)

empathy_os/meta_workflows/template_registry.py

@@ -1,229 +0,0 @@
-"""Template registry for meta-workflows.
-
-Handles loading, saving, and discovering meta-workflow templates.
-
-Created: 2026-01-17
-Updated: 2026-01-18 (v4.3.0 - Added built-in templates)
-Purpose: Manage reusable workflow templates
-"""
-
-import importlib.util
-import logging
-from pathlib import Path
-
-from empathy_os.meta_workflows.builtin_templates import (
-    BUILTIN_TEMPLATES,
-    get_builtin_template,
-    list_builtin_templates,
-)
-from empathy_os.meta_workflows.models import MetaWorkflowTemplate
-
-logger = logging.getLogger(__name__)
-
-# Import _validate_file_path from parent config.py (not config package)
-_config_py_path = Path(__file__).parent.parent / "config.py"
-_spec = importlib.util.spec_from_file_location("_config_module", _config_py_path)
-if _spec and _spec.loader:
-    _config_module = importlib.util.module_from_spec(_spec)
-    _spec.loader.exec_module(_config_module)
-    _validate_file_path = _config_module._validate_file_path
-else:
-    # Fallback: simple validation if import fails
-    def _validate_file_path(path: str, allowed_dir: str | None = None) -> Path:
-        if not path:
-            raise ValueError("path must be non-empty")
-        return Path(path).resolve()
-
-
-class TemplateRegistry:
-    """Registry for meta-workflow templates.
-
-    Manages template storage, loading, and discovery.
-    Templates are stored as JSON files in a directory.
-
-    Attributes:
-        storage_dir: Directory where templates are stored
-    """
-
-    def __init__(self, storage_dir: str | None = None):
-        """Initialize template registry.
-
-        Args:
-            storage_dir: Directory for template storage
-                        (default: .empathy/meta_workflows/templates/)
-
-        Raises:
-            ValueError: If storage_dir is invalid
-        """
-        if storage_dir is None:
-            storage_dir = str(Path.home() / ".empathy" / "meta_workflows" / "templates")
-
-        # Validate and create storage directory
-        self.storage_dir = Path(_validate_file_path(storage_dir))
-        self.storage_dir.mkdir(parents=True, exist_ok=True)
-
-        logger.info(f"Template registry initialized at: {self.storage_dir}")
-
-    def load_template(self, template_id: str) -> MetaWorkflowTemplate | None:
-        """Load a template by ID.
-
-        Checks built-in templates first, then user templates.
-
-        Args:
-            template_id: ID of template to load
-
-        Returns:
-            MetaWorkflowTemplate if found, None otherwise
-
-        Raises:
-            ValueError: If template file is invalid or corrupted
-        """
-        # Check built-in templates first
-        builtin = get_builtin_template(template_id)
-        if builtin is not None:
-            logger.info(f"Loaded built-in template: {template_id}")
-            return builtin
-
-        # Fall back to user templates
-        template_path = self.storage_dir / f"{template_id}.json"
-
-        if not template_path.exists():
-            logger.warning(f"Template not found: {template_id}")
-            return None
-
-        try:
-            json_str = template_path.read_text(encoding="utf-8")
-            template = MetaWorkflowTemplate.from_json(json_str)
-            logger.info(f"Loaded template: {template_id}")
-            return template
-
-        except (ValueError, KeyError) as e:
-            logger.error(f"Failed to load template {template_id}: {e}")
-            raise ValueError(f"Invalid template file {template_id}: {e}") from e
-
-    def save_template(self, template: MetaWorkflowTemplate) -> Path:
-        """Save a template to disk.
-
-        Args:
-            template: Template to save
-
-        Returns:
-            Path where template was saved
-
-        Raises:
-            ValueError: If template is invalid or path is unsafe
-            OSError: If write operation fails
-        """
-        template_path = self.storage_dir / f"{template.template_id}.json"
-
-        # Validate path (prevent path traversal)
-        validated_path = _validate_file_path(str(template_path))
-
-        try:
-            json_str = template.to_json()
-            validated_path.write_text(json_str, encoding="utf-8")
-            logger.info(f"Saved template: {template.template_id} → {validated_path}")
-            return validated_path
-
-        except OSError as e:
-            logger.error(f"Failed to save template {template.template_id}: {e}")
-            raise
-
-    def list_templates(self, include_builtin: bool = True) -> list[str]:
-        """List all available template IDs.
-
-        Args:
-            include_builtin: Whether to include built-in templates (default: True)
-
-        Returns:
-            List of template IDs (sorted, built-in templates marked with *)
-        """
-        # User templates
-        template_files = self.storage_dir.glob("*.json")
-        template_ids = {f.stem for f in template_files}
-
-        # Add built-in templates
-        if include_builtin:
-            template_ids.update(list_builtin_templates())
-
-        result = sorted(template_ids)
-        logger.debug(f"Found {len(result)} templates ({len(list_builtin_templates())} built-in)")
-        return result
-
-    def is_builtin(self, template_id: str) -> bool:
-        """Check if a template is built-in.
-
-        Args:
-            template_id: ID of template to check
-
-        Returns:
-            True if template is built-in
-        """
-        return template_id in BUILTIN_TEMPLATES
-
-    def get_template_info(self, template_id: str) -> dict | None:
-        """Get basic info about a template without loading it fully.
-
-        Args:
-            template_id: ID of template
-
-        Returns:
-            Dictionary with basic template info, or None if not found
-        """
-        template = self.load_template(template_id)
-        if template is None:
-            return None
-
-        return {
-            "template_id": template.template_id,
-            "name": template.name,
-            "description": template.description,
-            "version": template.version,
-            "tags": template.tags,
-            "author": template.author,
-            "estimated_cost_range": template.estimated_cost_range,
-            "estimated_duration_minutes": template.estimated_duration_minutes,
-            "question_count": len(template.form_schema.questions),
-            "agent_rule_count": len(template.agent_composition_rules),
-        }
-
-    def delete_template(self, template_id: str) -> bool:
-        """Delete a template.
-
-        Args:
-            template_id: ID of template to delete
-
-        Returns:
-            True if deleted, False if not found
-
-        Raises:
-            OSError: If delete operation fails
-        """
-        template_path = self.storage_dir / f"{template_id}.json"
-
-        if not template_path.exists():
-            logger.warning(f"Template not found for deletion: {template_id}")
-            return False
-
-        try:
-            template_path.unlink()
-            logger.info(f"Deleted template: {template_id}")
-            return True
-
-        except OSError as e:
-            logger.error(f"Failed to delete template {template_id}: {e}")
-            raise
-
-
-# =============================================================================
-# Module-level helpers
-# =============================================================================
-
-
-def get_default_registry() -> TemplateRegistry:
-    """Get default template registry instance.
-
-    Returns:
-        TemplateRegistry using default storage location
-    """
-    return TemplateRegistry()