PraisonAI 4.6.161__tar.gz → 4.6.162__tar.gz

{praisonai-4.6.161 → praisonai-4.6.162}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PraisonAI
-Version: 4.6.161
+Version: 4.6.162
 Summary: PraisonAI is an AI Agents Framework with Self Reflection. PraisonAI application combines PraisonAI Agents, AutoGen, and CrewAI into a low-code solution for building and managing multi-agent LLM systems, focusing on simplicity, customisation, and efficient human-agent collaboration.
 Author: Mervin Praison
 License: MIT
@@ -11,14 +11,14 @@ Description-Content-Type: text/markdown
 Requires-Dist: rich>=13.7
 Requires-Dist: markdown>=3.5
 Requires-Dist: pyparsing>=3.0.0
-Requires-Dist: praisonaiagents>=1.6.165
-Requires-Dist: praisonai-code>=0.0.61
-Requires-Dist: praisonai-bot>=0.0.46
-Requires-Dist: praisonai-train>=0.0.16
-Requires-Dist: praisonai-browser>=0.0.12
-Requires-Dist: praisonai-mcp>=0.0.12
-Requires-Dist: praisonai-sandbox>=0.0.10
-Requires-Dist: praisonai-deploy>=0.0.5
+Requires-Dist: praisonaiagents>=1.6.166
+Requires-Dist: praisonai-code>=0.0.62
+Requires-Dist: praisonai-bot>=0.0.47
+Requires-Dist: praisonai-train>=0.0.17
+Requires-Dist: praisonai-browser>=0.0.13
+Requires-Dist: praisonai-mcp>=0.0.13
+Requires-Dist: praisonai-sandbox>=0.0.11
+Requires-Dist: praisonai-deploy>=0.0.6
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: litellm<2,>=1.83.14
 Requires-Dist: PyYAML>=6.0

{praisonai-4.6.161 → praisonai-4.6.162}/PraisonAI.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PraisonAI
-Version: 4.6.161
+Version: 4.6.162
 Summary: PraisonAI is an AI Agents Framework with Self Reflection. PraisonAI application combines PraisonAI Agents, AutoGen, and CrewAI into a low-code solution for building and managing multi-agent LLM systems, focusing on simplicity, customisation, and efficient human-agent collaboration.
 Author: Mervin Praison
 License: MIT
@@ -11,14 +11,14 @@ Description-Content-Type: text/markdown
 Requires-Dist: rich>=13.7
 Requires-Dist: markdown>=3.5
 Requires-Dist: pyparsing>=3.0.0
-Requires-Dist: praisonaiagents>=1.6.165
-Requires-Dist: praisonai-code>=0.0.61
-Requires-Dist: praisonai-bot>=0.0.46
-Requires-Dist: praisonai-train>=0.0.16
-Requires-Dist: praisonai-browser>=0.0.12
-Requires-Dist: praisonai-mcp>=0.0.12
-Requires-Dist: praisonai-sandbox>=0.0.10
-Requires-Dist: praisonai-deploy>=0.0.5
+Requires-Dist: praisonaiagents>=1.6.166
+Requires-Dist: praisonai-code>=0.0.62
+Requires-Dist: praisonai-bot>=0.0.47
+Requires-Dist: praisonai-train>=0.0.17
+Requires-Dist: praisonai-browser>=0.0.13
+Requires-Dist: praisonai-mcp>=0.0.13
+Requires-Dist: praisonai-sandbox>=0.0.11
+Requires-Dist: praisonai-deploy>=0.0.6
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: litellm<2,>=1.83.14
 Requires-Dist: PyYAML>=6.0

{praisonai-4.6.161 → praisonai-4.6.162}/PraisonAI.egg-info/requires.txt

@@ -1,14 +1,14 @@
 rich>=13.7
 markdown>=3.5
 pyparsing>=3.0.0
-praisonaiagents>=1.6.165
-praisonai-code>=0.0.61
-praisonai-bot>=0.0.46
-praisonai-train>=0.0.16
-praisonai-browser>=0.0.12
-praisonai-mcp>=0.0.12
-praisonai-sandbox>=0.0.10
-praisonai-deploy>=0.0.5
+praisonaiagents>=1.6.166
+praisonai-code>=0.0.62
+praisonai-bot>=0.0.47
+praisonai-train>=0.0.17
+praisonai-browser>=0.0.13
+praisonai-mcp>=0.0.13
+praisonai-sandbox>=0.0.11
+praisonai-deploy>=0.0.6
 python-dotenv>=0.19.0
 litellm<2,>=1.83.14
 PyYAML>=6.0

{praisonai-4.6.161 → praisonai-4.6.162}/praisonai/api/agent_invoke.py

@@ -142,32 +142,70 @@ else:
 
 
 # Agent Registry
-_agent_registry: Dict[str, Any] = {}
+import threading
 
 
-def register_agent(agent_id: str, agent: Any) -> None:
+class AgentRegistry:
+    """Thread-safe registry of agents scoped to an embedding/tenant.
+
+    Encapsulates the previously module-global dict so multiple ``AgentOS`` /
+    ``praisonai serve`` gateways sharing one interpreter can each hold an
+    isolated registry, and so concurrent unregisters are race-free.
+    """
+
+    def __init__(self) -> None:
+        self._agents: Dict[str, Any] = {}
+        self._lock = threading.RLock()
+
+    def register(self, agent_id: str, agent: Any) -> None:
+        with self._lock:
+            self._agents[agent_id] = agent
+        logger.debug(f"Registered agent: {agent_id}")
+
+    def unregister(self, agent_id: str) -> bool:
+        # Atomic under the lock: test membership and pop together so a concurrent
+        # unregister can't race between the check and the delete, while still
+        # reporting removal correctly even when the stored value is falsy/None.
+        _MISSING = object()
+        with self._lock:
+            removed = self._agents.pop(agent_id, _MISSING) is not _MISSING
+        if removed:
+            logger.debug(f"Unregistered agent: {agent_id}")
+        return removed
+
+    def get(self, agent_id: str) -> Optional[Any]:
+        with self._lock:
+            return self._agents.get(agent_id)
+
+    def list(self) -> list:
+        with self._lock:
+            return list(self._agents.keys())
+
+
+_default_registry = AgentRegistry()
+
+
+def register_agent(agent_id: str, agent: Any,
+                   *, registry: Optional[AgentRegistry] = None) -> None:
     """Register an agent for invocation via API."""
-    _agent_registry[agent_id] = agent
-    logger.debug(f"Registered agent: {agent_id}")
+    (registry or _default_registry).register(agent_id, agent)
 
 
-def unregister_agent(agent_id: str) -> bool:
+def unregister_agent(agent_id: str,
+                     *, registry: Optional[AgentRegistry] = None) -> bool:
     """Unregister an agent."""
-    if agent_id in _agent_registry:
-        del _agent_registry[agent_id]
-        logger.debug(f"Unregistered agent: {agent_id}")
-        return True
-    return False
+    return (registry or _default_registry).unregister(agent_id)
 
 
-def get_agent(agent_id: str) -> Optional[Any]:
+def get_agent(agent_id: str,
+              *, registry: Optional[AgentRegistry] = None) -> Optional[Any]:
     """Get a registered agent by ID."""
-    return _agent_registry.get(agent_id)
+    return (registry or _default_registry).get(agent_id)
 
 
-def list_registered_agents() -> list:
+def list_registered_agents(*, registry: Optional[AgentRegistry] = None) -> list:
     """List all registered agent IDs."""
-    return list(_agent_registry.keys())
+    return (registry or _default_registry).list()
 
 
 def _is_real_agent(agent: Any) -> bool:

{praisonai-4.6.161 → praisonai-4.6.162}/praisonai/cli/features/eval.py

@@ -476,84 +476,3 @@ def add_eval_parser_subcommands(subparsers) -> None:
     perf_parser.add_argument('--output', '-o', help='Output file')
     perf_parser.add_argument('--verbose', '-v', action='store_true', help='Verbose output')
     perf_parser.add_argument('--quiet', '-q', action='store_true', help='Suppress JSON output')
-
-
-def add_eval_parser(subparsers) -> None:
-    """
-    Add eval subcommand parser.
-    
-    Args:
-        subparsers: Argument parser subparsers
-    """
-    eval_parser = subparsers.add_parser(
-        'eval',
-        help='Run agent evaluations'
-    )
-    
-    eval_subparsers = eval_parser.add_subparsers(dest='eval_type')
-    
-    accuracy_parser = eval_subparsers.add_parser(
-        'accuracy',
-        help='Run accuracy evaluation'
-    )
-    accuracy_parser.add_argument('--agent', '-a', help='Agent config file (optional if --prompt used)')
-    accuracy_parser.add_argument('--prompt', '-p', help='Direct prompt (alternative to --agent)')
-    accuracy_parser.add_argument('--llm', help='LLM model for agent (when using --prompt)')
-    accuracy_parser.add_argument('--input', '-i', help='Input text (defaults to --prompt if not provided)')
-    accuracy_parser.add_argument('--expected', '-e', required=True, help='Expected output')
-    accuracy_parser.add_argument('--iterations', '-n', type=int, default=1, help='Number of iterations')
-    accuracy_parser.add_argument('--model', '-m', help='Judge model')
-    accuracy_parser.add_argument('--output', '-o', help='Output file')
-    accuracy_parser.add_argument('--verbose', '-v', action='store_true', help='Verbose output')
-    accuracy_parser.add_argument('--quiet', '-q', action='store_true', help='Suppress JSON output')
-    
-    perf_parser = eval_subparsers.add_parser(
-        'performance',
-        help='Run performance evaluation'
-    )
-    perf_parser.add_argument('--agent', '-a', default='agents.yaml', help='Agent config file')
-    perf_parser.add_argument('--input', '-i', default='Hello', help='Input text')
-    perf_parser.add_argument('--iterations', '-n', type=int, default=10, help='Number of iterations')
-    perf_parser.add_argument('--warmup', '-w', type=int, default=2, help='Warmup runs')
-    perf_parser.add_argument('--memory', action='store_true', default=True, help='Track memory')
-    perf_parser.add_argument('--output', '-o', help='Output file')
-    perf_parser.add_argument('--verbose', '-v', action='store_true', help='Verbose output')
-    perf_parser.add_argument('--quiet', '-q', action='store_true', help='Suppress JSON output')
-    
-    rel_parser = eval_subparsers.add_parser(
-        'reliability',
-        help='Run reliability evaluation'
-    )
-    rel_parser.add_argument('--agent', '-a', default='agents.yaml', help='Agent config file')
-    rel_parser.add_argument('--input', '-i', required=True, help='Input text')
-    rel_parser.add_argument('--expected-tools', '-t', required=True, help='Expected tools (comma-separated)')
-    rel_parser.add_argument('--forbidden-tools', '-f', help='Forbidden tools (comma-separated)')
-    rel_parser.add_argument('--output', '-o', help='Output file')
-    rel_parser.add_argument('--verbose', '-v', action='store_true', help='Verbose output')
-    rel_parser.add_argument('--quiet', '-q', action='store_true', help='Suppress JSON output')
-    
-    criteria_parser = eval_subparsers.add_parser(
-        'criteria',
-        help='Run criteria-based evaluation'
-    )
-    criteria_parser.add_argument('--agent', '-a', default='agents.yaml', help='Agent config file')
-    criteria_parser.add_argument('--input', '-i', required=True, help='Input text')
-    criteria_parser.add_argument('--criteria', '-c', required=True, help='Evaluation criteria')
-    criteria_parser.add_argument('--scoring', '-s', choices=['numeric', 'binary'], default='numeric', help='Scoring type')
-    criteria_parser.add_argument('--threshold', type=float, default=7.0, help='Pass threshold')
-    criteria_parser.add_argument('--iterations', '-n', type=int, default=1, help='Number of iterations')
-    criteria_parser.add_argument('--model', '-m', help='Judge model')
-    criteria_parser.add_argument('--output', '-o', help='Output file')
-    criteria_parser.add_argument('--verbose', '-v', action='store_true', help='Verbose output')
-    criteria_parser.add_argument('--quiet', '-q', action='store_true', help='Suppress JSON output')
-    
-    batch_parser = eval_subparsers.add_parser(
-        'batch',
-        help='Run batch evaluation from test file'
-    )
-    batch_parser.add_argument('--agent', '-a', default='agents.yaml', help='Agent config file')
-    batch_parser.add_argument('--test-file', '-t', required=True, help='JSON test file')
-    batch_parser.add_argument('--batch-type', '-b', choices=['accuracy', 'criteria'], default='accuracy', help='Evaluation type')
-    batch_parser.add_argument('--output', '-o', help='Output file')
-    batch_parser.add_argument('--verbose', '-v', action='store_true', help='Verbose output')
-    batch_parser.add_argument('--quiet', '-q', action='store_true', help='Suppress JSON output')

{praisonai-4.6.161 → praisonai-4.6.162}/praisonai/cli/features/recipe_optimizer.py

@@ -12,140 +12,20 @@ DRY: Reuses ContextEffectivenessJudge from replay module.
 import os
 import re
 import logging
-import base64
 from pathlib import Path
 from typing import Optional, List, Tuple, Any, Dict
 
 # DRY: Import SDK knowledge from shared modules
 from .sdk_knowledge import get_sdk_knowledge_prompt
 
-logger = logging.getLogger(__name__)
-
-
-def _read_image_for_optimizer(image_path: str, model: str = "gpt-4o-mini") -> str:
-    """
-    Read and describe an image for optimizer context.
-    
-    Uses vision LLM to describe the image content so the optimizer can
-    validate if agent outputs match the actual image.
-    
-    Args:
-        image_path: Local file path or URL to the image
-        model: Vision-capable model to use
-        
-    Returns:
-        Description of the image content
-    """
-    try:
-        import litellm
-        
-        # Prepare image content
-        if image_path.startswith(('http://', 'https://')):
-            # URL - pass directly
-            image_content = {
-                "type": "image_url",
-                "image_url": {"url": image_path}
-            }
-        elif os.path.exists(image_path):
-            # Local file - encode as base64
-            with open(image_path, 'rb') as f:
-                image_data = base64.b64encode(f.read()).decode('utf-8')
-            
-            # Detect mime type
-            ext = os.path.splitext(image_path)[1].lower()
-            mime_types = {
-                '.jpg': 'image/jpeg', '.jpeg': 'image/jpeg',
-                '.png': 'image/png', '.gif': 'image/gif',
-                '.webp': 'image/webp', '.bmp': 'image/bmp'
-            }
-            mime_type = mime_types.get(ext, 'image/jpeg')
-            
-            image_content = {
-                "type": "image_url",
-                "image_url": {"url": f"data:{mime_type};base64,{image_data}"}
-            }
-        else:
-            return f"[Image not found: {image_path}]"
-        
-        # Use vision model to describe the image
-        response = litellm.completion(
-            model=model,
-            messages=[{
-                "role": "user",
-                "content": [
-                    {"type": "text", "text": "Describe this image in detail. Include: main subject, colors, text visible, key elements, and overall composition. Be factual and specific."},
-                    image_content
-                ]
-            }],
-            max_tokens=500,
-        )
-        
-        description = response.choices[0].message.content or ""
-        return f"[IMAGE DESCRIPTION: {description}]"
-        
-    except Exception as e:
-        logger.warning(f"Failed to read image for optimizer: {e}")
-        return f"[Failed to read image: {str(e)[:100]}]"
+# DRY: Reuse image reading and URL crawling from the canonical judge module
+# instead of maintaining verbatim copies here.
+from praisonai.replay.judge import (
+    _read_image_for_judge as _read_image_for_optimizer,
+    _crawl_url_for_judge as _crawl_url_for_optimizer,
+)
 
-
-def _crawl_url_for_optimizer(url: str) -> str:
-    """
-    Crawl a URL to get content for optimizer context.
-    
-    Uses the web_crawl tool to extract content so the optimizer can
-    validate if agent outputs match the actual URL content.
-    
-    Args:
-        url: URL to crawl
-        
-    Returns:
-        Extracted content from the URL (truncated)
-    """
-    try:
-        # Try to use the web_crawl tool
-        from praisonaiagents.tools import web_crawl
-        
-        result = web_crawl(url)
-        
-        if isinstance(result, dict):
-            if result.get('error'):
-                return f"[URL crawl error: {result.get('error')}]"
-            content = result.get('content', '')
-            title = result.get('title', '')
-            provider = result.get('provider', 'unknown')
-            
-            # Truncate content for optimizer context
-            if len(content) > 3000:
-                content = content[:3000] + "... [truncated]"
-            
-            return f"[URL CONTENT (via {provider})]\nTitle: {title}\n\n{content}"
-        
-        return f"[URL content: {str(result)[:3000]}]"
-        
-    except ImportError:
-        # Fallback to basic HTTP fetch
-        try:
-            import urllib.request
-            
-            with urllib.request.urlopen(url, timeout=30) as response:
-                content = response.read().decode('utf-8', errors='ignore')
-            
-            # Basic HTML to text
-            content = re.sub(r'<script[^>]*>.*?</script>', '', content, flags=re.DOTALL | re.IGNORECASE)
-            content = re.sub(r'<style[^>]*>.*?</style>', '', content, flags=re.DOTALL | re.IGNORECASE)
-            content = re.sub(r'<[^>]+>', ' ', content)
-            content = re.sub(r'\s+', ' ', content).strip()
-            
-            if len(content) > 3000:
-                content = content[:3000] + "... [truncated]"
-            
-            return f"[URL CONTENT (via urllib)]\n{content}"
-            
-        except Exception as e:
-            return f"[Failed to crawl URL: {str(e)[:100]}]"
-    except Exception as e:
-        logger.warning(f"Failed to crawl URL for optimizer: {e}")
-        return f"[Failed to crawl URL: {str(e)[:100]}]"
+logger = logging.getLogger(__name__)
 
 
 # Common issue patterns and their fixes - comprehensive patterns

{praisonai-4.6.161 → praisonai-4.6.162}/praisonai/cli/features/sandbox_cli.py

@@ -227,69 +227,3 @@ def handle_sandbox_command(args) -> None:
     else:
         print(f"Unknown sandbox command: {subcommand}")
         print("Available commands: run, shell, status")
-
-
-def add_sandbox_parser(subparsers) -> None:
-    """Add sandbox subparser to CLI."""
-    sandbox_parser = subparsers.add_parser(
-        "sandbox",
-        help="Run code in a sandbox",
-    )
-    
-    sandbox_subparsers = sandbox_parser.add_subparsers(
-        dest="sandbox_command",
-        help="Sandbox commands",
-    )
-    
-    run_parser = sandbox_subparsers.add_parser(
-        "run",
-        help="Run code in sandbox",
-    )
-    run_parser.add_argument(
-        "--code", "-c",
-        help="Code to execute",
-    )
-    run_parser.add_argument(
-        "--file", "-f",
-        help="File to execute",
-    )
-    run_parser.add_argument(
-        "--type", "-t",
-        choices=["subprocess", "docker", "daytona"],
-        default="subprocess",
-        help="Sandbox type (default: subprocess)",
-    )
-    run_parser.add_argument(
-        "--image",
-        default="python:3.11-slim",
-        help="Docker image (default: python:3.11-slim)",
-    )
-    run_parser.add_argument(
-        "--timeout",
-        type=int,
-        default=60,
-        help="Timeout in seconds (default: 60)",
-    )
-    
-    shell_parser = sandbox_subparsers.add_parser(
-        "shell",
-        help="Start interactive sandbox shell",
-    )
-    shell_parser.add_argument(
-        "--type", "-t",
-        choices=["subprocess", "docker", "daytona"],
-        default="subprocess",
-        help="Sandbox type (default: subprocess)",
-    )
-    shell_parser.add_argument(
-        "--image",
-        default="python:3.11-slim",
-        help="Docker image (default: python:3.11-slim)",
-    )
-    
-    sandbox_subparsers.add_parser(
-        "status",
-        help="Check sandbox availability",
-    )
-    
-    sandbox_parser.set_defaults(func=handle_sandbox_command)

{praisonai-4.6.161 → praisonai-4.6.162}/praisonai/cli/interactive/async_tui.py

@@ -230,6 +230,8 @@ class AsyncTUI:
         self._runtime = None  # InteractiveRuntime for ACP/LSP
         self._runtime_started = False
         self._registry = None  # Unified command registry (lazy)
+        self._checkpoints = None  # SessionCheckpointManager (lazy)
+        self._checkpoints_ready = False
         self._prev_permission_mode = None  # Approval mode to restore when leaving PLAN
         # Sync PLAN indicator/toggle with a backend launched via --approval plan.
         self._sync_plan_mode_from_backend()
@@ -262,6 +264,123 @@ class AsyncTUI:
         except Exception:
             pass
     
+    def _get_checkpoints(self):
+        """Lazily build the coherent turn-checkpoint manager for this session.
+
+        Binds ``SessionCheckpointManager`` to a view over the TUI's in-memory
+        conversation history so ``/undo`` and ``/revert`` roll back **files and
+        chat history together** on the default path — the same proven engine the
+        legacy REPL uses. Enabled by default for interactive runs
+        (``default=True``); opt out with ``checkpoints.auto: false`` or
+        ``PRAISONAI_CHECKPOINTS=off``. Degrades to ``None`` on any failure so a
+        checkpoint problem never breaks the TUI.
+        """
+        if self._checkpoints_ready:
+            return self._checkpoints
+        self._checkpoints_ready = True
+        try:
+            from praisonai_code.cli.interactive.repl import (
+                _InMemoryConversationStore,
+            )
+            from praisonai.cli.features.session_checkpoints import (
+                SessionCheckpointManager,
+            )
+
+            config = None
+            try:
+                from praisonai.cli.configuration.resolver import resolve_config
+
+                config = resolve_config().extra
+            except Exception:
+                config = None
+
+            store = _InMemoryConversationStore(
+                self._conversation_history,
+                agent_provider=lambda: self._agent,
+            )
+            mgr = SessionCheckpointManager.from_config(
+                workspace_dir=(
+                    os.environ.get("PRAISONAI_WORKSPACE")
+                    or self.config.workspace
+                    or os.getcwd()
+                ),
+                config=config,
+                verbose=self.config.debug,
+                session_store=store,
+                session_id=self.session_id or "tui",
+                default=True,
+            )
+            if mgr.enabled:
+                mgr.checkpoint_turn("session start")
+                self._checkpoints = mgr
+        except Exception as exc:  # pragma: no cover - defensive
+            logger.debug("Session checkpoints unavailable: %s", exc)
+            self._checkpoints = None
+        return self._checkpoints
+
+    def _checkpoint_turn(self, label: str) -> None:
+        """Record a pre-turn checkpoint so the revert timeline exists."""
+        ckpt = self._get_checkpoints()
+        if ckpt is not None:
+            try:
+                ckpt.checkpoint_turn(label)
+            except Exception:  # pragma: no cover - never break a turn
+                pass
+
+    def _handle_undo(self) -> None:
+        """/undo — roll files + conversation back to the previous turn."""
+        ckpt = self._get_checkpoints()
+        if ckpt is None or not getattr(ckpt, "enabled", False) or not ckpt.turns:
+            self.messages.append(ChatMessage(role="system", content="Nothing to undo."))
+            return
+        ckpt.preview(1)
+        restored = ckpt.revert(1)
+        if restored:
+            self.messages.append(ChatMessage(
+                role="system",
+                content=f"Undone last turn — files + conversation reverted to {restored.short_id}.",
+            ))
+        else:
+            self.messages.append(ChatMessage(role="system", content="No checkpoint to undo."))
+
+    def _handle_revert(self, args: str) -> None:
+        """/revert [n] — roll files + conversation back n turns (default 1)."""
+        ckpt = self._get_checkpoints()
+        if ckpt is None or not getattr(ckpt, "enabled", False):
+            self.messages.append(ChatMessage(
+                role="system",
+                content=(
+                    "Checkpointing is disabled. Enable with checkpoints.auto: true "
+                    "or PRAISONAI_CHECKPOINTS=on."
+                ),
+            ))
+            return
+        if not ckpt.turns:
+            self.messages.append(ChatMessage(role="system", content="No checkpoints to revert to."))
+            return
+        n = 1
+        token = str(args).strip().lower()
+        if token and token != "last":
+            try:
+                n = int(token)
+            except ValueError:
+                self.messages.append(ChatMessage(role="system", content="Usage: /revert [n|last]"))
+                return
+        if n < 1 or n > len(ckpt.turns):
+            self.messages.append(ChatMessage(
+                role="system", content=f"Can only revert 1..{len(ckpt.turns)} turn(s).",
+            ))
+            return
+        ckpt.preview(n)
+        restored = ckpt.revert(n)
+        if restored:
+            self.messages.append(ChatMessage(
+                role="system",
+                content=f"Reverted {n} turn(s) — files + conversation restored to {restored.short_id}.",
+            ))
+        else:
+            self.messages.append(ChatMessage(role="system", content="Failed to revert."))
+
     def _process_file_mentions(self, prompt: str) -> str:
         """Process @file mentions and include file contents."""
         import os
@@ -645,6 +764,8 @@ class AsyncTUI:
         "multiline": "Toggle multiline input mode",
         "files": "List workspace files for @ mentions",
         "queue": "Show pending prompts in queue",
+        "undo": "Undo the last turn (files + conversation)",
+        "revert": "Revert the last N turns (files + conversation)",
     }
 
     def _get_registry(self):
@@ -708,6 +829,8 @@ class AsyncTUI:
   /multiline       Toggle multiline input mode
   /files           List workspace files for @ mentions
   /queue           Show pending prompts in queue
+  /undo            Undo the last turn (files + conversation)
+  /revert [n]      Revert the last N turns (files + conversation)
 
 Keyboard Shortcuts:
   PageUp/Down      Scroll output
@@ -1101,6 +1224,14 @@ Example: /handoff code "refactor the auth module" """
                         ))
             return True
         
+        elif cmd == "undo":
+            self._handle_undo()
+            return True
+        
+        elif cmd == "revert":
+            self._handle_revert(args)
+            return True
+        
         else:
             # Not a built-in: consult the unified command registry so custom
             # .praisonai/commands/*.md commands are first-class /name commands.
@@ -1363,6 +1494,13 @@ Example: /handoff code "refactor the auth module" """
                 else self._process_file_mentions(prompt)
             )
             
+            # Checkpoint files + conversation at the turn boundary before the
+            # agent may mutate the workspace, so /undo and /revert can roll both
+            # back together. Read-only turns (reviews) never mutate files, so
+            # skip them to keep the revert timeline meaningful.
+            if not read_only:
+                self._checkpoint_turn(prompt[:60])
+            
             # Add to conversation history
             self._conversation_history.append({"role": "user", "content": prompt})