Release v2.8.9

docker/Dockerfile.chat

@@ -16,7 +16,7 @@ RUN mkdir -p /root/.praison
 # Install Python packages (using latest versions)
 RUN pip install --no-cache-dir \
     praisonai_tools \
-    "praisonai>=2.8.3" \
+    "praisonai>=2.8.9" \
     "praisonai[chat]" \
     "embedchain[github,youtube]"
 

docker/Dockerfile.dev

@@ -20,7 +20,7 @@ RUN mkdir -p /root/.praison
 # Install Python packages (using latest versions)
 RUN pip install --no-cache-dir \
     praisonai_tools \
-    "praisonai>=2.8.3" \
+    "praisonai>=2.8.9" \
     "praisonai[ui]" \
     "praisonai[chat]" \
     "praisonai[realtime]" \

docker/Dockerfile.ui

@@ -16,7 +16,7 @@ RUN mkdir -p /root/.praison
 # Install Python packages (using latest versions)
 RUN pip install --no-cache-dir \
     praisonai_tools \
-    "praisonai>=2.8.3" \
+    "praisonai>=2.8.9" \
     "praisonai[ui]" \
     "praisonai[crewai]"
 

examples/acp/README.md

@@ -0,0 +1,137 @@
+# ACP (Agent Client Protocol) Examples
+
+This directory contains examples for using PraisonAI with the Agent Client Protocol (ACP).
+
+## What is ACP?
+
+The Agent Client Protocol (ACP) is a standardized JSON-RPC 2.0 protocol that allows code editors and IDEs to communicate with AI coding agents. It enables seamless integration between PraisonAI and editors like:
+
+- **Zed**
+- **JetBrains IDEs** (IntelliJ, PyCharm, WebStorm, etc.)
+- **VSCode** (via extensions)
+- **Toad**
+
+## Quick Start
+
+### CLI Usage
+
+```bash
+# Start ACP server with defaults
+praisonai acp
+
+# Enable debug logging
+praisonai acp --debug
+
+# Allow file writes
+praisonai acp --allow-write
+
+# Use a specific model
+praisonai acp -m gpt-4o
+
+# Resume last session
+praisonai acp --resume --last
+```
+
+### Python API
+
+```python
+from praisonai.acp import serve
+
+serve(
+    workspace=".",
+    debug=True,
+    allow_write=True,
+)
+```
+
+## Editor Configuration
+
+### Zed
+
+Add to `~/.config/zed/settings.json`:
+
+```json
+{
+  "agent_servers": {
+    "PraisonAI": {
+      "command": "praisonai",
+      "args": ["acp"],
+      "env": {}
+    }
+  }
+}
+```
+
+### JetBrains
+
+Add to `~/.jetbrains/acp.json`:
+
+```json
+{
+  "agent_servers": {
+    "PraisonAI": {
+      "command": "praisonai",
+      "args": ["acp"],
+      "env": {}
+    }
+  }
+}
+```
+
+### Toad
+
+```bash
+toad acp "praisonai acp"
+```
+
+## Examples
+
+- `basic_acp_server.py` - Simple ACP server with default settings
+- `custom_agent_acp.py` - ACP server with a custom agent
+
+## CLI Options
+
+| Option | Description |
+|--------|-------------|
+| `-w, --workspace` | Workspace root directory |
+| `-a, --agent` | Agent name or config file |
+| `-m, --model` | LLM model to use |
+| `-r, --resume` | Resume session by ID |
+| `--last` | Resume the last session |
+| `--approve` | Approval mode: manual, auto, scoped |
+| `--read-only` | Read-only mode (default) |
+| `--allow-write` | Allow file writes |
+| `--allow-shell` | Allow shell commands |
+| `--debug` | Enable debug logging |
+
+## Installation
+
+```bash
+pip install praisonai[acp]
+```
+
+## Protocol Details
+
+ACP uses JSON-RPC 2.0 over stdio:
+- **stdin**: Receives JSON-RPC requests from the client
+- **stdout**: Sends JSON-RPC responses to the client
+- **stderr**: Debug/log output (never pollutes stdout)
+
+### Supported Methods
+
+**Agent Methods (client → agent):**
+- `initialize` - Negotiate protocol version and capabilities
+- `authenticate` - Optional authentication
+- `session/new` - Create a new session
+- `session/load` - Load an existing session
+- `session/prompt` - Send user message
+- `session/cancel` - Cancel ongoing operations
+- `session/set_mode` - Change operating mode
+
+**Client Methods (agent → client):**
+- `session/update` - Send progress updates
+- `session/request_permission` - Request user approval
+- `fs/read_text_file` - Read file contents
+- `fs/write_text_file` - Write file contents
+- `terminal/create` - Create terminal
+- `terminal/output` - Get terminal output

examples/acp/basic_acp_server.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+"""
+Basic ACP Server Example
+
+This example demonstrates how to run PraisonAI as an ACP server
+that can be connected to by IDEs like Zed, JetBrains, VSCode, or Toad.
+
+Usage:
+    # Run from command line
+    praisonai acp
+    
+    # Or run this script directly
+    python basic_acp_server.py
+"""
+
+from praisonai.acp import serve
+
+if __name__ == "__main__":
+    # Start ACP server with default settings
+    # This will listen on stdin/stdout for JSON-RPC messages
+    serve(
+        workspace=".",           # Current directory as workspace
+        agent="default",         # Use default agent
+        debug=False,             # Set to True for debug logging to stderr
+        read_only=True,          # Safe by default
+        approval_mode="manual",  # Require approval for actions
+    )

examples/acp/custom_agent_acp.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python3
+"""
+Custom Agent ACP Server Example
+
+This example shows how to run a custom PraisonAI agent as an ACP server.
+
+Usage:
+    python custom_agent_acp.py
+"""
+
+from praisonai.acp import serve, ACPServer, ACPConfig
+from praisonaiagents import Agent
+
+# Create a custom agent
+agent = Agent(
+    name="CodeAssistant",
+    instructions="""You are an expert coding assistant. You help users:
+    - Write clean, efficient code
+    - Debug issues
+    - Explain complex concepts
+    - Review code for best practices
+    
+    Always provide clear explanations with your code suggestions.""",
+    model="gpt-4o-mini",
+)
+
+if __name__ == "__main__":
+    # Create config
+    config = ACPConfig(
+        workspace=".",
+        debug=True,
+        allow_write=True,  # Allow file modifications
+        approval_mode="manual",
+    )
+    
+    # Create server with custom agent
+    server = ACPServer(config=config, agent=agent)
+    
+    # Run the server
+    serve(
+        workspace=".",
+        debug=True,
+        allow_write=True,
+        approval_mode="manual",
+    )

src/praisonai/.gitignore

@@ -0,0 +1 @@
+chainlit.md

src/praisonai/examples/acp/README.md

@@ -0,0 +1,137 @@
+# ACP (Agent Client Protocol) Examples
+
+This directory contains examples for using PraisonAI with the Agent Client Protocol (ACP).
+
+## What is ACP?
+
+The Agent Client Protocol (ACP) is a standardized JSON-RPC 2.0 protocol that allows code editors and IDEs to communicate with AI coding agents. It enables seamless integration between PraisonAI and editors like:
+
+- **Zed**
+- **JetBrains IDEs** (IntelliJ, PyCharm, WebStorm, etc.)
+- **VSCode** (via extensions)
+- **Toad**
+
+## Quick Start
+
+### CLI Usage
+
+```bash
+# Start ACP server with defaults
+praisonai acp
+
+# Enable debug logging
+praisonai acp --debug
+
+# Allow file writes
+praisonai acp --allow-write
+
+# Use a specific model
+praisonai acp -m gpt-4o
+
+# Resume last session
+praisonai acp --resume --last
+```
+
+### Python API
+
+```python
+from praisonai.acp import serve
+
+serve(
+    workspace=".",
+    debug=True,
+    allow_write=True,
+)
+```
+
+## Editor Configuration
+
+### Zed
+
+Add to `~/.config/zed/settings.json`:
+
+```json
+{
+  "agent_servers": {
+    "PraisonAI": {
+      "command": "praisonai",
+      "args": ["acp"],
+      "env": {}
+    }
+  }
+}
+```
+
+### JetBrains
+
+Add to `~/.jetbrains/acp.json`:
+
+```json
+{
+  "agent_servers": {
+    "PraisonAI": {
+      "command": "praisonai",
+      "args": ["acp"],
+      "env": {}
+    }
+  }
+}
+```
+
+### Toad
+
+```bash
+toad acp "praisonai acp"
+```
+
+## Examples
+
+- `basic_acp_server.py` - Simple ACP server with default settings
+- `custom_agent_acp.py` - ACP server with a custom agent
+
+## CLI Options
+
+| Option | Description |
+|--------|-------------|
+| `-w, --workspace` | Workspace root directory |
+| `-a, --agent` | Agent name or config file |
+| `-m, --model` | LLM model to use |
+| `-r, --resume` | Resume session by ID |
+| `--last` | Resume the last session |
+| `--approve` | Approval mode: manual, auto, scoped |
+| `--read-only` | Read-only mode (default) |
+| `--allow-write` | Allow file writes |
+| `--allow-shell` | Allow shell commands |
+| `--debug` | Enable debug logging |
+
+## Installation
+
+```bash
+pip install praisonai[acp]
+```
+
+## Protocol Details
+
+ACP uses JSON-RPC 2.0 over stdio:
+- **stdin**: Receives JSON-RPC requests from the client
+- **stdout**: Sends JSON-RPC responses to the client
+- **stderr**: Debug/log output (never pollutes stdout)
+
+### Supported Methods
+
+**Agent Methods (client → agent):**
+- `initialize` - Negotiate protocol version and capabilities
+- `authenticate` - Optional authentication
+- `session/new` - Create a new session
+- `session/load` - Load an existing session
+- `session/prompt` - Send user message
+- `session/cancel` - Cancel ongoing operations
+- `session/set_mode` - Change operating mode
+
+**Client Methods (agent → client):**
+- `session/update` - Send progress updates
+- `session/request_permission` - Request user approval
+- `fs/read_text_file` - Read file contents
+- `fs/write_text_file` - Write file contents
+- `terminal/create` - Create terminal
+- `terminal/output` - Get terminal output

src/praisonai/examples/acp/basic_acp_server.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+"""
+Basic ACP Server Example
+
+This example demonstrates how to run PraisonAI as an ACP server
+that can be connected to by IDEs like Zed, JetBrains, VSCode, or Toad.
+
+Usage:
+    # Run from command line
+    praisonai acp
+    
+    # Or run this script directly
+    python basic_acp_server.py
+"""
+
+from praisonai.acp import serve
+
+if __name__ == "__main__":
+    # Start ACP server with default settings
+    # This will listen on stdin/stdout for JSON-RPC messages
+    serve(
+        workspace=".",           # Current directory as workspace
+        agent="default",         # Use default agent
+        debug=False,             # Set to True for debug logging to stderr
+        read_only=True,          # Safe by default
+        approval_mode="manual",  # Require approval for actions
+    )

src/praisonai/examples/acp/custom_agent_acp.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python3
+"""
+Custom Agent ACP Server Example
+
+This example shows how to run a custom PraisonAI agent as an ACP server.
+
+Usage:
+    python custom_agent_acp.py
+"""
+
+from praisonai.acp import serve, ACPServer, ACPConfig
+from praisonaiagents import Agent
+
+# Create a custom agent
+agent = Agent(
+    name="CodeAssistant",
+    instructions="""You are an expert coding assistant. You help users:
+    - Write clean, efficient code
+    - Debug issues
+    - Explain complex concepts
+    - Review code for best practices
+    
+    Always provide clear explanations with your code suggestions.""",
+    model="gpt-4o-mini",
+)
+
+if __name__ == "__main__":
+    # Create config
+    config = ACPConfig(
+        workspace=".",
+        debug=True,
+        allow_write=True,  # Allow file modifications
+        approval_mode="manual",
+    )
+    
+    # Create server with custom agent
+    server = ACPServer(config=config, agent=agent)
+    
+    # Run the server
+    serve(
+        workspace=".",
+        debug=True,
+        allow_write=True,
+        approval_mode="manual",
+    )

src/praisonai/praisonai.rb

@@ -3,8 +3,8 @@ class Praisonai < Formula
   
     desc "AI tools for various AI applications"
     homepage "https://github.com/MervinPraison/PraisonAI"
-    url "https://github.com/MervinPraison/PraisonAI/archive/refs/tags/v2.8.3.tar.gz"
-    sha256 `curl -sL https://github.com/MervinPraison/PraisonAI/archive/refs/tags/v2.8.3.tar.gz | shasum -a 256`.split.first
+    url "https://github.com/MervinPraison/PraisonAI/archive/refs/tags/v2.8.9.tar.gz"
+    sha256 `curl -sL https://github.com/MervinPraison/PraisonAI/archive/refs/tags/v2.8.9.tar.gz | shasum -a 256`.split.first
     license "MIT"
   
     depends_on "[email protected]"

src/praisonai/praisonai/acp/__init__.py

@@ -0,0 +1,54 @@
+"""
+Agent Client Protocol (ACP) support for PraisonAI.
+
+This module provides ACP server functionality that allows IDEs/editors
+(Zed, JetBrains, VSCode, Toad) to connect to PraisonAI agents.
+
+Usage:
+    # CLI
+    praisonai acp [OPTIONS]
+    
+    # Python API
+    from praisonai.acp import serve
+    serve(workspace=".", agent="default")
+"""
+
+import importlib.util
+
+__all__ = [
+    "serve",
+    "ACPServer",
+    "ACPSession",
+    "ACPConfig",
+]
+
+# Lazy imports for performance
+def __getattr__(name: str):
+    """Lazy import ACP components."""
+    if name == "serve":
+        from .server import serve
+        return serve
+    if name == "ACPServer":
+        from .server import ACPServer
+        return ACPServer
+    if name == "ACPSession":
+        from .session import ACPSession
+        return ACPSession
+    if name == "ACPConfig":
+        from .config import ACPConfig
+        return ACPConfig
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+def _check_acp_available() -> bool:
+    """Check if agent-client-protocol package is installed."""
+    return importlib.util.find_spec("acp") is not None
+
+
+def _get_install_instructions() -> str:
+    """Get installation instructions for ACP support."""
+    return (
+        "ACP support requires the agent-client-protocol package.\n"
+        "Install with: pip install praisonai[acp]\n"
+        "Or: pip install agent-client-protocol"
+    )

src/praisonai/praisonai/acp/config.py

@@ -0,0 +1,159 @@
+"""
+ACP configuration management.
+
+Handles configuration precedence:
+1. CLI flags (highest)
+2. Environment variables
+3. Config file (~/.praison/config.yaml)
+4. Defaults (lowest)
+"""
+
+import os
+from dataclasses import dataclass, field
+from pathlib import Path
+from typing import Any, Dict, List, Optional
+
+
+@dataclass
+class ACPConfig:
+    """Configuration for ACP server."""
+    
+    # Workspace settings
+    workspace: Path = field(default_factory=Path.cwd)
+    
+    # Agent settings
+    agent: str = "default"
+    agents_config: Optional[Path] = None
+    router_enabled: bool = False
+    model: Optional[str] = None
+    
+    # Session settings
+    resume_session: Optional[str] = None
+    resume_last: bool = False
+    
+    # Permission settings
+    read_only: bool = True
+    allow_write: bool = False
+    allow_shell: bool = False
+    allow_network: bool = False
+    approval_mode: str = "manual"  # manual, auto, scoped
+    
+    # Workspace boundaries
+    allowed_paths: List[Path] = field(default_factory=list)
+    
+    # Debug settings
+    debug: bool = False
+    
+    # Profile
+    profile: Optional[str] = None
+    
+    def __post_init__(self):
+        """Validate and normalize configuration."""
+        if isinstance(self.workspace, str):
+            self.workspace = Path(self.workspace).resolve()
+        
+        if self.agents_config and isinstance(self.agents_config, str):
+            self.agents_config = Path(self.agents_config)
+        
+        # Set up default allowed paths
+        if not self.allowed_paths:
+            self.allowed_paths = [
+                self.workspace,
+                Path.home() / ".praison",
+            ]
+        
+        # Normalize allowed paths
+        self.allowed_paths = [
+            Path(p).resolve() if isinstance(p, str) else p.resolve()
+            for p in self.allowed_paths
+        ]
+    
+    @classmethod
+    def from_env(cls) -> "ACPConfig":
+        """Create config from environment variables."""
+        return cls(
+            workspace=Path(os.environ.get("PRAISONAI_WORKSPACE", ".")).resolve(),
+            agent=os.environ.get("PRAISONAI_AGENT", "default"),
+            model=os.environ.get("PRAISONAI_MODEL") or os.environ.get("OPENAI_MODEL"),
+            debug=os.environ.get("PRAISONAI_DEBUG", "").lower() in ("1", "true", "yes"),
+            read_only=os.environ.get("PRAISONAI_READ_ONLY", "true").lower() in ("1", "true", "yes"),
+            approval_mode=os.environ.get("PRAISONAI_APPROVAL_MODE", "manual"),
+        )
+    
+    @classmethod
+    def from_file(cls, path: Optional[Path] = None) -> "ACPConfig":
+        """Load config from YAML file."""
+        if path is None:
+            path = Path.home() / ".praison" / "config.yaml"
+        
+        if not path.exists():
+            return cls()
+        
+        try:
+            import yaml
+            with open(path) as f:
+                data = yaml.safe_load(f) or {}
+            
+            acp_config = data.get("acp", {})
+            return cls(
+                workspace=Path(acp_config.get("workspace", ".")).resolve(),
+                agent=acp_config.get("agent", "default"),
+                model=acp_config.get("model"),
+                debug=acp_config.get("debug", False),
+                read_only=acp_config.get("read_only", True),
+                approval_mode=acp_config.get("approval_mode", "manual"),
+            )
+        except Exception:
+            return cls()
+    
+    @classmethod
+    def merge(cls, *configs: "ACPConfig") -> "ACPConfig":
+        """Merge multiple configs, later configs override earlier ones."""
+        result = cls()
+        for config in configs:
+            for field_name in [
+                "workspace", "agent", "agents_config", "router_enabled",
+                "model", "resume_session", "resume_last", "read_only",
+                "allow_write", "allow_shell", "allow_network", "approval_mode",
+                "debug", "profile"
+            ]:
+                value = getattr(config, field_name)
+                # Only override if value is not default/None
+                if value is not None and value != getattr(cls(), field_name, None):
+                    setattr(result, field_name, value)
+        return result
+    
+    def is_path_allowed(self, path: Path) -> bool:
+        """Check if a path is within allowed boundaries."""
+        path = Path(path).resolve()
+        return any(
+            path == allowed or allowed in path.parents
+            for allowed in self.allowed_paths
+        )
+    
+    def can_write(self) -> bool:
+        """Check if write operations are allowed."""
+        return not self.read_only or self.allow_write
+    
+    def can_execute_shell(self) -> bool:
+        """Check if shell commands are allowed."""
+        return self.allow_shell
+    
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert config to dictionary."""
+        return {
+            "workspace": str(self.workspace),
+            "agent": self.agent,
+            "agents_config": str(self.agents_config) if self.agents_config else None,
+            "router_enabled": self.router_enabled,
+            "model": self.model,
+            "resume_session": self.resume_session,
+            "resume_last": self.resume_last,
+            "read_only": self.read_only,
+            "allow_write": self.allow_write,
+            "allow_shell": self.allow_shell,
+            "allow_network": self.allow_network,
+            "approval_mode": self.approval_mode,
+            "debug": self.debug,
+            "profile": self.profile,
+        }

src/praisonai/praisonai/acp/server.py

@@ -0,0 +1,587 @@
+"""
+ACP Server implementation for PraisonAI.
+
+This module implements the Agent Client Protocol server that allows
+IDEs/editors to communicate with PraisonAI agents.
+"""
+
+import asyncio
+import logging
+import sys
+from pathlib import Path
+from typing import Any, Dict, List, Optional, Union
+
+from .config import ACPConfig
+from .session import ACPSession, SessionStore
+
+# Configure logging to stderr only (stdout reserved for JSON-RPC)
+logger = logging.getLogger(__name__)
+
+
+def _setup_stderr_logging(debug: bool = False) -> None:
+    """Configure logging to stderr only."""
+    handler = logging.StreamHandler(sys.stderr)
+    handler.setFormatter(logging.Formatter(
+        "%(asctime)s [%(levelname)s] %(name)s: %(message)s"
+    ))
+    
+    root_logger = logging.getLogger("praisonai.acp")
+    root_logger.handlers.clear()
+    root_logger.addHandler(handler)
+    root_logger.setLevel(logging.DEBUG if debug else logging.INFO)
+
+
+class ACPServer:
+    """
+    ACP Server that exposes PraisonAI agents to IDE clients.
+    
+    Implements the Agent Client Protocol (JSON-RPC 2.0 over stdio).
+    """
+    
+    def __init__(
+        self,
+        config: Optional[ACPConfig] = None,
+        agent: Optional[Any] = None,
+        agents: Optional[Any] = None,
+    ):
+        """
+        Initialize ACP server.
+        
+        Args:
+            config: ACP configuration
+            agent: Optional pre-configured Agent instance
+            agents: Optional pre-configured PraisonAIAgents instance
+        """
+        self.config = config or ACPConfig()
+        self._agent = agent
+        self._agents = agents
+        self._session_store = SessionStore()
+        self._sessions: Dict[str, ACPSession] = {}
+        self._client = None  # ACP client connection
+        self._cancelled_sessions: set = set()
+        
+        _setup_stderr_logging(self.config.debug)
+    
+    def _get_agent(self):
+        """Lazy load or create agent."""
+        if self._agent is not None:
+            return self._agent
+        
+        if self._agents is not None:
+            return self._agents
+        
+        # Create default agent
+        try:
+            from praisonaiagents import Agent
+            self._agent = Agent(
+                name="PraisonAI",
+                instructions="You are a helpful AI coding assistant.",
+                model=self.config.model or "gpt-4o-mini",
+            )
+            return self._agent
+        except ImportError:
+            logger.error("praisonaiagents not installed")
+            raise RuntimeError("praisonaiagents package required")
+    
+    def on_connect(self, client) -> None:
+        """Called when client connects."""
+        self._client = client
+        logger.info("Client connected")
+    
+    async def initialize(
+        self,
+        protocol_version: int,
+        client_capabilities: Optional[Dict[str, Any]] = None,
+        client_info: Optional[Dict[str, Any]] = None,
+        **kwargs: Any,
+    ) -> Dict[str, Any]:
+        """
+        Handle initialize request.
+        
+        Negotiate protocol version and exchange capabilities.
+        """
+        logger.info(f"Initialize: protocol_version={protocol_version}, client_info={client_info}")
+        
+        # Import here to avoid circular imports and check availability
+        try:
+            from acp import PROTOCOL_VERSION
+        except ImportError:
+            PROTOCOL_VERSION = 1
+        
+        return {
+            "protocolVersion": min(protocol_version, PROTOCOL_VERSION),
+            "agentCapabilities": {
+                "loadSession": True,
+                "promptCapabilities": {
+                    "image": False,
+                    "audio": False,
+                    "embeddedContext": True,
+                },
+                "mcpCapabilities": {
+                    "http": False,
+                    "sse": False,
+                },
+            },
+            "agentInfo": {
+                "name": "praisonai",
+                "title": "PraisonAI",
+                "version": self._get_version(),
+            },
+            "authMethods": [],
+        }
+    
+    async def authenticate(self, method_id: str, **kwargs: Any) -> Optional[Dict[str, Any]]:
+        """Handle authenticate request (optional)."""
+        logger.info(f"Authenticate: method_id={method_id}")
+        return {}
+    
+    async def new_session(
+        self,
+        cwd: str,
+        mcp_servers: List[Dict[str, Any]],
+        **kwargs: Any,
+    ) -> Dict[str, Any]:
+        """
+        Create a new conversation session.
+        
+        Args:
+            cwd: Working directory for the session
+            mcp_servers: List of MCP server configurations
+        """
+        logger.info(f"New session: cwd={cwd}")
+        
+        workspace = Path(cwd).resolve()
+        session = ACPSession.create(
+            workspace=workspace,
+            agent_id=self.config.agent,
+        )
+        session.mcp_servers = mcp_servers
+        
+        self._sessions[session.session_id] = session
+        self._session_store.save(session)
+        
+        logger.info(f"Created session: {session.session_id}")
+        
+        return {
+            "sessionId": session.session_id,
+            "modes": self._get_available_modes(),
+        }
+    
+    async def load_session(
+        self,
+        cwd: str,
+        mcp_servers: List[Dict[str, Any]],
+        session_id: str,
+        **kwargs: Any,
+    ) -> Optional[Dict[str, Any]]:
+        """
+        Load an existing session.
+        
+        Replays conversation history via session/update notifications.
+        """
+        logger.info(f"Load session: session_id={session_id}")
+        
+        # Try to load from store
+        session = self._session_store.load(session_id)
+        if session is None:
+            logger.warning(f"Session not found: {session_id}")
+            return None
+        
+        # Update workspace and MCP servers
+        session.workspace = Path(cwd).resolve()
+        session.mcp_servers = mcp_servers
+        session.update_activity()
+        
+        self._sessions[session_id] = session
+        
+        # Replay conversation history
+        await self._replay_session_history(session)
+        
+        return {}
+    
+    async def list_sessions(
+        self,
+        cursor: Optional[str] = None,
+        cwd: Optional[str] = None,
+        **kwargs: Any,
+    ) -> Dict[str, Any]:
+        """List available sessions."""
+        sessions = self._session_store.list_sessions()
+        
+        # Filter by cwd if provided
+        if cwd:
+            cwd_path = Path(cwd).resolve()
+            sessions = [s for s in sessions if s.workspace == cwd_path]
+        
+        return {
+            "sessions": [
+                {
+                    "sessionId": s.session_id,
+                    "createdAt": s.created_at,
+                    "lastActivity": s.last_activity,
+                }
+                for s in sessions
+            ],
+        }
+    
+    async def set_session_mode(
+        self,
+        mode_id: str,
+        session_id: str,
+        **kwargs: Any,
+    ) -> Optional[Dict[str, Any]]:
+        """Set the operating mode for a session."""
+        logger.info(f"Set session mode: session_id={session_id}, mode_id={mode_id}")
+        
+        session = self._sessions.get(session_id)
+        if session:
+            session.mode = mode_id
+            self._session_store.save(session)
+        
+        return {}
+    
+    async def set_session_model(
+        self,
+        model_id: str,
+        session_id: str,
+        **kwargs: Any,
+    ) -> Optional[Dict[str, Any]]:
+        """Set the LLM model for a session."""
+        logger.info(f"Set session model: session_id={session_id}, model_id={model_id}")
+        
+        session = self._sessions.get(session_id)
+        if session:
+            session.model = model_id
+            self._session_store.save(session)
+        
+        return {}
+    
+    async def prompt(
+        self,
+        prompt: List[Dict[str, Any]],
+        session_id: str,
+        **kwargs: Any,
+    ) -> Dict[str, Any]:
+        """
+        Handle user prompt.
+        
+        This is the main interaction method where the user sends a message
+        and the agent processes it.
+        """
+        logger.info(f"Prompt: session_id={session_id}")
+        
+        session = self._sessions.get(session_id)
+        if session is None:
+            logger.error(f"Session not found: {session_id}")
+            return {"stopReason": "refusal"}
+        
+        # Check if cancelled
+        if session_id in self._cancelled_sessions:
+            self._cancelled_sessions.discard(session_id)
+            return {"stopReason": "cancelled"}
+        
+        # Extract text from prompt blocks
+        user_message = self._extract_prompt_text(prompt)
+        session.add_message("user", user_message)
+        
+        try:
+            # Get agent response
+            agent = self._get_agent()
+            
+            # Send thinking update
+            await self._send_thought(session_id, "Processing your request...")
+            
+            # Generate response
+            response = await self._generate_response(agent, user_message, session)
+            
+            # Send response as agent message chunks
+            await self._send_agent_message(session_id, response)
+            
+            # Save session
+            session.add_message("assistant", response)
+            self._session_store.save(session)
+            
+            return {"stopReason": "end_turn"}
+            
+        except asyncio.CancelledError:
+            return {"stopReason": "cancelled"}
+        except Exception as e:
+            logger.exception(f"Error processing prompt: {e}")
+            await self._send_agent_message(session_id, f"Error: {str(e)}")
+            return {"stopReason": "refusal"}
+    
+    async def fork_session(
+        self,
+        cwd: str,
+        session_id: str,
+        mcp_servers: Optional[List[Dict[str, Any]]] = None,
+        **kwargs: Any,
+    ) -> Dict[str, Any]:
+        """Fork an existing session."""
+        logger.info(f"Fork session: session_id={session_id}")
+        
+        original = self._sessions.get(session_id)
+        if original is None:
+            original = self._session_store.load(session_id)
+        
+        if original is None:
+            # Create new session if original not found
+            return await self.new_session(cwd, mcp_servers or [])
+        
+        # Create forked session
+        forked = ACPSession.create(
+            workspace=Path(cwd).resolve(),
+            agent_id=original.agent_id,
+        )
+        forked.messages = original.messages.copy()
+        forked.mcp_servers = mcp_servers or original.mcp_servers
+        
+        self._sessions[forked.session_id] = forked
+        self._session_store.save(forked)
+        
+        return {"sessionId": forked.session_id}
+    
+    async def resume_session(
+        self,
+        cwd: str,
+        session_id: str,
+        mcp_servers: Optional[List[Dict[str, Any]]] = None,
+        **kwargs: Any,
+    ) -> Dict[str, Any]:
+        """Resume an existing session."""
+        return await self.load_session(cwd, mcp_servers or [], session_id)
+    
+    async def cancel(self, session_id: str, **kwargs: Any) -> None:
+        """Cancel ongoing operations for a session."""
+        logger.info(f"Cancel: session_id={session_id}")
+        self._cancelled_sessions.add(session_id)
+    
+    async def ext_method(self, method: str, params: Dict[str, Any]) -> Dict[str, Any]:
+        """Handle extension method calls."""
+        logger.info(f"Extension method: {method}")
+        return {"error": f"Unknown extension method: {method}"}
+    
+    async def ext_notification(self, method: str, params: Dict[str, Any]) -> None:
+        """Handle extension notifications."""
+        logger.info(f"Extension notification: {method}")
+    
+    # Helper methods
+    
+    def _get_version(self) -> str:
+        """Get PraisonAI version."""
+        try:
+            from praisonai.version import __version__
+            return __version__
+        except ImportError:
+            return "0.0.0"
+    
+    def _get_available_modes(self) -> Optional[List[Dict[str, Any]]]:
+        """Get available operating modes."""
+        return [
+            {
+                "id": "manual",
+                "name": "Manual",
+                "description": "Requires approval for all actions",
+            },
+            {
+                "id": "auto",
+                "name": "Auto",
+                "description": "Auto-approve within workspace",
+            },
+        ]
+    
+    def _extract_prompt_text(self, prompt: List[Dict[str, Any]]) -> str:
+        """Extract text content from prompt blocks."""
+        texts = []
+        for block in prompt:
+            if isinstance(block, dict):
+                block_type = block.get("type", "")
+                if block_type == "text":
+                    texts.append(block.get("text", ""))
+                elif block_type == "resource":
+                    resource = block.get("resource", {})
+                    if "text" in resource:
+                        texts.append(f"[File: {resource.get('uri', 'unknown')}]\n{resource['text']}")
+        return "\n".join(texts)
+    
+    async def _generate_response(
+        self,
+        agent: Any,
+        message: str,
+        session: ACPSession,
+    ) -> str:
+        """Generate agent response."""
+        try:
+            # Try async chat first
+            if hasattr(agent, "achat"):
+                response = await agent.achat(message)
+            elif hasattr(agent, "chat"):
+                # Run sync chat in executor
+                loop = asyncio.get_event_loop()
+                response = await loop.run_in_executor(None, agent.chat, message)
+            else:
+                response = "Agent does not support chat interface"
+            
+            return str(response) if response else ""
+        except Exception as e:
+            logger.exception(f"Error generating response: {e}")
+            return f"Error generating response: {str(e)}"
+    
+    async def _send_agent_message(self, session_id: str, text: str) -> None:
+        """Send agent message chunk to client."""
+        if self._client is None:
+            return
+        
+        try:
+            from acp import update_agent_message_text
+            update = update_agent_message_text(text)
+            await self._client.session_update(session_id=session_id, update=update)
+        except ImportError:
+            # Fallback without ACP SDK
+            logger.debug(f"Agent message: {text[:100]}...")
+    
+    async def _send_thought(self, session_id: str, text: str) -> None:
+        """Send agent thought to client."""
+        if self._client is None:
+            return
+        
+        try:
+            from acp import update_agent_thought_text
+            update = update_agent_thought_text(text)
+            await self._client.session_update(session_id=session_id, update=update)
+        except ImportError:
+            logger.debug(f"Agent thought: {text}")
+    
+    async def _replay_session_history(self, session: ACPSession) -> None:
+        """Replay session history to client."""
+        if self._client is None:
+            return
+        
+        try:
+            from acp import update_user_message_text, update_agent_message_text
+            
+            for msg in session.messages:
+                role = msg.get("role", "")
+                content = msg.get("content", "")
+                
+                if role == "user":
+                    update = update_user_message_text(str(content))
+                elif role == "assistant":
+                    update = update_agent_message_text(str(content))
+                else:
+                    continue
+                
+                await self._client.session_update(
+                    session_id=session.session_id,
+                    update=update,
+                )
+        except ImportError:
+            logger.debug("Skipping history replay (ACP SDK not available)")
+
+
+async def _run_server(config: ACPConfig) -> None:
+    """Run the ACP server."""
+    try:
+        from acp import Agent as ACPAgent, run_agent
+    except ImportError:
+        logger.error(
+            "agent-client-protocol package not installed.\n"
+            "Install with: pip install praisonai[acp]\n"
+            "Or: pip install agent-client-protocol"
+        )
+        sys.exit(1)
+    
+    server = ACPServer(config=config)
+    
+    # Create ACP-compatible agent wrapper
+    class PraisonACPAgent(ACPAgent):
+        def on_connect(self, conn):
+            server.on_connect(conn)
+        
+        async def initialize(self, protocol_version, client_capabilities=None, client_info=None, **kwargs):
+            return await server.initialize(protocol_version, client_capabilities, client_info, **kwargs)
+        
+        async def authenticate(self, method_id, **kwargs):
+            return await server.authenticate(method_id, **kwargs)
+        
+        async def new_session(self, cwd, mcp_servers, **kwargs):
+            return await server.new_session(cwd, mcp_servers, **kwargs)
+        
+        async def load_session(self, cwd, mcp_servers, session_id, **kwargs):
+            return await server.load_session(cwd, mcp_servers, session_id, **kwargs)
+        
+        async def list_sessions(self, cursor=None, cwd=None, **kwargs):
+            return await server.list_sessions(cursor, cwd, **kwargs)
+        
+        async def set_session_mode(self, mode_id, session_id, **kwargs):
+            return await server.set_session_mode(mode_id, session_id, **kwargs)
+        
+        async def set_session_model(self, model_id, session_id, **kwargs):
+            return await server.set_session_model(model_id, session_id, **kwargs)
+        
+        async def prompt(self, prompt, session_id, **kwargs):
+            return await server.prompt(prompt, session_id, **kwargs)
+        
+        async def fork_session(self, cwd, session_id, mcp_servers=None, **kwargs):
+            return await server.fork_session(cwd, session_id, mcp_servers, **kwargs)
+        
+        async def resume_session(self, cwd, session_id, mcp_servers=None, **kwargs):
+            return await server.resume_session(cwd, session_id, mcp_servers, **kwargs)
+        
+        async def cancel(self, session_id, **kwargs):
+            return await server.cancel(session_id, **kwargs)
+        
+        async def ext_method(self, method, params):
+            return await server.ext_method(method, params)
+        
+        async def ext_notification(self, method, params):
+            return await server.ext_notification(method, params)
+    
+    logger.info("Starting PraisonAI ACP server...")
+    await run_agent(PraisonACPAgent(), use_unstable_protocol=True)
+
+
+def serve(
+    workspace: Union[str, Path] = ".",
+    agent: str = "default",
+    model: Optional[str] = None,
+    resume: Optional[str] = None,
+    resume_last: bool = False,
+    debug: bool = False,
+    read_only: bool = True,
+    allow_write: bool = False,
+    allow_shell: bool = False,
+    approval_mode: str = "manual",
+    **kwargs: Any,
+) -> None:
+    """
+    Start the ACP server.
+    
+    This is the main entry point for running PraisonAI as an ACP agent.
+    
+    Args:
+        workspace: Working directory for the session
+        agent: Agent name or configuration
+        model: LLM model to use
+        resume: Session ID to resume
+        resume_last: Resume the last session
+        debug: Enable debug logging
+        read_only: Read-only mode (default True)
+        allow_write: Allow file writes
+        allow_shell: Allow shell commands
+        approval_mode: Approval mode (manual, auto, scoped)
+    """
+    config = ACPConfig(
+        workspace=Path(workspace).resolve(),
+        agent=agent,
+        model=model,
+        resume_session=resume,
+        resume_last=resume_last,
+        debug=debug,
+        read_only=read_only,
+        allow_write=allow_write,
+        allow_shell=allow_shell,
+        approval_mode=approval_mode,
+    )
+    
+    asyncio.run(_run_server(config))

src/praisonai/praisonai/acp/session.py

@@ -0,0 +1,219 @@
+"""
+ACP session management.
+
+Handles session creation, persistence, and resume functionality.
+"""
+
+import json
+import logging
+import time
+import uuid
+from dataclasses import dataclass, field
+from pathlib import Path
+from typing import Any, Dict, List, Optional
+
+logger = logging.getLogger(__name__)
+
+
+@dataclass
+class ACPSession:
+    """Represents an ACP conversation session."""
+    
+    session_id: str
+    workspace: Path
+    created_at: float = field(default_factory=time.time)
+    last_activity: float = field(default_factory=time.time)
+    
+    # Agent attribution
+    agent_id: Optional[str] = None
+    run_id: Optional[str] = None
+    trace_id: Optional[str] = None
+    
+    # Session state
+    mode: str = "manual"  # manual, auto, full_auto
+    model: Optional[str] = None
+    
+    # Conversation history (for resume)
+    messages: List[Dict[str, Any]] = field(default_factory=list)
+    tool_calls: List[Dict[str, Any]] = field(default_factory=list)
+    
+    # MCP servers
+    mcp_servers: List[Dict[str, Any]] = field(default_factory=list)
+    
+    @classmethod
+    def create(cls, workspace: Path, agent_id: Optional[str] = None) -> "ACPSession":
+        """Create a new session."""
+        session_id = f"sess_{uuid.uuid4().hex[:16]}"
+        return cls(
+            session_id=session_id,
+            workspace=workspace,
+            agent_id=agent_id,
+            run_id=f"run_{uuid.uuid4().hex[:8]}",
+            trace_id=f"trace_{uuid.uuid4().hex[:8]}",
+        )
+    
+    def update_activity(self) -> None:
+        """Update last activity timestamp."""
+        self.last_activity = time.time()
+    
+    def add_message(self, role: str, content: Any) -> None:
+        """Add a message to the conversation history."""
+        self.messages.append({
+            "role": role,
+            "content": content,
+            "timestamp": time.time(),
+        })
+        self.update_activity()
+    
+    def add_tool_call(self, tool_call_id: str, title: str, status: str, **kwargs) -> None:
+        """Add a tool call to the history."""
+        self.tool_calls.append({
+            "tool_call_id": tool_call_id,
+            "title": title,
+            "status": status,
+            "timestamp": time.time(),
+            **kwargs,
+        })
+        self.update_activity()
+    
+    def to_dict(self) -> Dict[str, Any]:
+        """Serialize session to dictionary."""
+        return {
+            "session_id": self.session_id,
+            "workspace": str(self.workspace),
+            "created_at": self.created_at,
+            "last_activity": self.last_activity,
+            "agent_id": self.agent_id,
+            "run_id": self.run_id,
+            "trace_id": self.trace_id,
+            "mode": self.mode,
+            "model": self.model,
+            "messages": self.messages,
+            "tool_calls": self.tool_calls,
+            "mcp_servers": self.mcp_servers,
+        }
+    
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "ACPSession":
+        """Deserialize session from dictionary."""
+        return cls(
+            session_id=data["session_id"],
+            workspace=Path(data["workspace"]),
+            created_at=data.get("created_at", time.time()),
+            last_activity=data.get("last_activity", time.time()),
+            agent_id=data.get("agent_id"),
+            run_id=data.get("run_id"),
+            trace_id=data.get("trace_id"),
+            mode=data.get("mode", "manual"),
+            model=data.get("model"),
+            messages=data.get("messages", []),
+            tool_calls=data.get("tool_calls", []),
+            mcp_servers=data.get("mcp_servers", []),
+        )
+
+
+class SessionStore:
+    """Persistent storage for ACP sessions."""
+    
+    def __init__(self, storage_dir: Optional[Path] = None):
+        """Initialize session store."""
+        if storage_dir is None:
+            storage_dir = Path.home() / ".praison" / "acp" / "sessions"
+        self.storage_dir = storage_dir
+        self.storage_dir.mkdir(parents=True, exist_ok=True)
+        self._last_session_file = self.storage_dir / ".last_session"
+    
+    def _session_path(self, session_id: str) -> Path:
+        """Get path for session file."""
+        return self.storage_dir / f"{session_id}.json"
+    
+    def save(self, session: ACPSession) -> None:
+        """Save session to disk."""
+        try:
+            path = self._session_path(session.session_id)
+            with open(path, "w") as f:
+                json.dump(session.to_dict(), f, indent=2)
+            
+            # Update last session pointer
+            with open(self._last_session_file, "w") as f:
+                f.write(session.session_id)
+            
+            logger.debug(f"Saved session {session.session_id}")
+        except Exception as e:
+            logger.error(f"Failed to save session: {e}")
+    
+    def load(self, session_id: str) -> Optional[ACPSession]:
+        """Load session from disk."""
+        try:
+            path = self._session_path(session_id)
+            if not path.exists():
+                logger.warning(f"Session not found: {session_id}")
+                return None
+            
+            with open(path) as f:
+                data = json.load(f)
+            
+            return ACPSession.from_dict(data)
+        except Exception as e:
+            logger.error(f"Failed to load session: {e}")
+            return None
+    
+    def load_last(self) -> Optional[ACPSession]:
+        """Load the most recent session."""
+        try:
+            if not self._last_session_file.exists():
+                return None
+            
+            with open(self._last_session_file) as f:
+                session_id = f.read().strip()
+            
+            if session_id:
+                return self.load(session_id)
+            return None
+        except Exception as e:
+            logger.error(f"Failed to load last session: {e}")
+            return None
+    
+    def delete(self, session_id: str) -> bool:
+        """Delete a session."""
+        try:
+            path = self._session_path(session_id)
+            if path.exists():
+                path.unlink()
+                logger.debug(f"Deleted session {session_id}")
+                return True
+            return False
+        except Exception as e:
+            logger.error(f"Failed to delete session: {e}")
+            return False
+    
+    def list_sessions(self, limit: int = 50) -> List[ACPSession]:
+        """List all sessions, sorted by last activity."""
+        sessions = []
+        try:
+            for path in self.storage_dir.glob("sess_*.json"):
+                try:
+                    with open(path) as f:
+                        data = json.load(f)
+                    sessions.append(ACPSession.from_dict(data))
+                except Exception:
+                    continue
+            
+            # Sort by last activity, most recent first
+            sessions.sort(key=lambda s: s.last_activity, reverse=True)
+            return sessions[:limit]
+        except Exception as e:
+            logger.error(f"Failed to list sessions: {e}")
+            return []
+    
+    def cleanup_old_sessions(self, max_age_days: int = 30) -> int:
+        """Remove sessions older than max_age_days."""
+        cutoff = time.time() - (max_age_days * 24 * 60 * 60)
+        removed = 0
+        
+        for session in self.list_sessions(limit=1000):
+            if session.last_activity < cutoff:
+                if self.delete(session.session_id):
+                    removed += 1
+        
+        return removed

src/praisonai/praisonai/cli/features/acp.py

@@ -0,0 +1,236 @@
+"""
+ACP CLI handler for PraisonAI.
+
+Provides the `praisonai acp` command for IDE/editor integration.
+"""
+
+import argparse
+import logging
+import sys
+from typing import Optional
+
+logger = logging.getLogger(__name__)
+
+
+class ACPHandler:
+    """Handler for ACP CLI commands."""
+    
+    @staticmethod
+    def add_arguments(parser: argparse.ArgumentParser) -> None:
+        """Add ACP-specific arguments to parser."""
+        parser.add_argument(
+            "-w", "--workspace",
+            type=str,
+            default=".",
+            help="Workspace root directory (default: current directory)"
+        )
+        parser.add_argument(
+            "-a", "--agent",
+            type=str,
+            default="default",
+            help="Agent name or configuration file"
+        )
+        parser.add_argument(
+            "--agents",
+            type=str,
+            help="Multi-agent configuration YAML file"
+        )
+        parser.add_argument(
+            "--router",
+            action="store_true",
+            help="Enable router agent for task delegation"
+        )
+        parser.add_argument(
+            "-m", "--model",
+            type=str,
+            help="LLM model to use"
+        )
+        parser.add_argument(
+            "-r", "--resume",
+            type=str,
+            nargs="?",
+            const="__last__",
+            help="Resume session by ID (or --resume --last for most recent)"
+        )
+        parser.add_argument(
+            "--last",
+            action="store_true",
+            help="Resume the last session (use with --resume)"
+        )
+        parser.add_argument(
+            "--approve",
+            type=str,
+            choices=["manual", "auto", "scoped"],
+            default="manual",
+            help="Approval mode for actions (default: manual)"
+        )
+        parser.add_argument(
+            "--read-only",
+            action="store_true",
+            default=True,
+            help="Read-only mode (default: enabled)"
+        )
+        parser.add_argument(
+            "--allow-write",
+            action="store_true",
+            help="Allow file write operations"
+        )
+        parser.add_argument(
+            "--allow-shell",
+            action="store_true",
+            help="Allow shell command execution"
+        )
+        parser.add_argument(
+            "--allow-network",
+            action="store_true",
+            help="Allow network requests"
+        )
+        parser.add_argument(
+            "--debug",
+            action="store_true",
+            help="Enable debug logging to stderr"
+        )
+        parser.add_argument(
+            "--profile",
+            type=str,
+            help="Use named profile from config"
+        )
+    
+    @staticmethod
+    def handle(args: argparse.Namespace) -> int:
+        """
+        Handle ACP command execution.
+        
+        Returns:
+            Exit code (0 for success, non-zero for error)
+        """
+        # Check if ACP SDK is available
+        try:
+            import importlib.util
+            if importlib.util.find_spec("acp") is None:
+                print(
+                    "Error: agent-client-protocol package not installed.\n"
+                    "Install with: pip install praisonai[acp]\n"
+                    "Or: pip install agent-client-protocol",
+                    file=sys.stderr
+                )
+                return 1
+        except Exception as e:
+            print(f"Error checking ACP availability: {e}", file=sys.stderr)
+            return 1
+        
+        # Import ACP module
+        try:
+            from praisonai.acp import serve
+        except ImportError as e:
+            print(f"Error importing ACP module: {e}", file=sys.stderr)
+            return 1
+        
+        # Determine resume session
+        resume_session: Optional[str] = None
+        resume_last = False
+        
+        if args.resume:
+            if args.resume == "__last__" or args.last:
+                resume_last = True
+            else:
+                resume_session = args.resume
+        elif args.last:
+            resume_last = True
+        
+        # Determine read-only mode
+        read_only = args.read_only and not args.allow_write
+        
+        try:
+            # Start ACP server
+            serve(
+                workspace=args.workspace,
+                agent=args.agent,
+                model=args.model,
+                resume=resume_session,
+                resume_last=resume_last,
+                debug=args.debug,
+                read_only=read_only,
+                allow_write=args.allow_write,
+                allow_shell=args.allow_shell,
+                approval_mode=args.approve,
+            )
+            return 0
+        except KeyboardInterrupt:
+            logger.info("ACP server stopped by user")
+            return 0
+        except Exception as e:
+            logger.exception(f"ACP server error: {e}")
+            print(f"Error: {e}", file=sys.stderr)
+            return 1
+
+
+def setup_acp_subparser(subparsers) -> None:
+    """Set up the ACP subcommand parser."""
+    acp_parser = subparsers.add_parser(
+        "acp",
+        help="Start ACP server for IDE/editor integration",
+        description=(
+            "Start the Agent Client Protocol (ACP) server.\n\n"
+            "This allows IDEs and editors (Zed, JetBrains, VSCode, Toad) "
+            "to connect to PraisonAI agents via JSON-RPC over stdio."
+        ),
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+Examples:
+  praisonai acp                    # Start with defaults
+  praisonai acp --debug            # Enable debug logging
+  praisonai acp --resume --last    # Resume last session
+  praisonai acp --allow-write      # Enable file writes
+  praisonai acp -m gpt-4o          # Use specific model
+
+Editor Configuration:
+
+  Zed (~/.config/zed/settings.json):
+    {
+      "agent_servers": {
+        "PraisonAI": {
+          "command": "praisonai",
+          "args": ["acp"],
+          "env": {}
+        }
+      }
+    }
+
+  JetBrains (~/.jetbrains/acp.json):
+    {
+      "agent_servers": {
+        "PraisonAI": {
+          "command": "praisonai",
+          "args": ["acp"],
+          "env": {}
+        }
+      }
+    }
+
+  Toad:
+    toad acp "praisonai acp"
+"""
+    )
+    ACPHandler.add_arguments(acp_parser)
+    acp_parser.set_defaults(func=lambda args: ACPHandler.handle(args))
+
+
+def run_acp_command(args: Optional[list] = None) -> int:
+    """
+    Run ACP command directly.
+    
+    This can be called from the main CLI or as a standalone entry point.
+    """
+    parser = argparse.ArgumentParser(
+        prog="praisonai acp",
+        description="Start ACP server for IDE/editor integration"
+    )
+    ACPHandler.add_arguments(parser)
+    
+    parsed_args = parser.parse_args(args)
+    return ACPHandler.handle(parsed_args)
+
+
+if __name__ == "__main__":
+    sys.exit(run_acp_command())

src/praisonai/praisonai/cli/main.py

@@ -698,7 +698,7 @@ class PraisonAI:
             return default_args
         
         # Define special commands
-        special_commands = ['chat', 'code', 'call', 'realtime', 'train', 'ui', 'context', 'research', 'memory', 'rules', 'workflow', 'hooks', 'knowledge', 'session', 'tools', 'todo', 'docs', 'mcp', 'commit', 'serve', 'schedule', 'skills', 'profile', 'eval', 'agents', 'run', 'thinking', 'compaction', 'output', 'deploy', 'templates', 'recipe', 'endpoints', 'audio', 'embed', 'images', 'moderate', 'files', 'batches', 'vector-stores', 'rerank', 'ocr', 'assistants', 'fine-tuning', 'completions', 'messages', 'guardrails', 'rag', 'videos', 'a2a', 'containers', 'passthrough', 'responses', 'search', 'realtime-api', 'doctor', 'registry', 'package', 'install', 'uninstall']
+        special_commands = ['chat', 'code', 'call', 'realtime', 'train', 'ui', 'context', 'research', 'memory', 'rules', 'workflow', 'hooks', 'knowledge', 'session', 'tools', 'todo', 'docs', 'mcp', 'commit', 'serve', 'schedule', 'skills', 'profile', 'eval', 'agents', 'run', 'thinking', 'compaction', 'output', 'deploy', 'templates', 'recipe', 'endpoints', 'audio', 'embed', 'images', 'moderate', 'files', 'batches', 'vector-stores', 'rerank', 'ocr', 'assistants', 'fine-tuning', 'completions', 'messages', 'guardrails', 'rag', 'videos', 'a2a', 'containers', 'passthrough', 'responses', 'search', 'realtime-api', 'doctor', 'registry', 'package', 'install', 'uninstall', 'acp']
         
         parser = argparse.ArgumentParser(prog="praisonai", description="praisonAI command-line interface")
         parser.add_argument("--framework", choices=["crewai", "autogen", "praisonai"], help="Specify the framework")
@@ -898,6 +898,12 @@ class PraisonAI:
             self._handle_serve_command(args, unknown_args)
             sys.exit(0)
 
+        # Handle ACP command - Agent Client Protocol server for IDE integration
+        if args.command == 'acp':
+            from praisonai.cli.features.acp import run_acp_command
+            exit_code = run_acp_command(unknown_args)
+            sys.exit(exit_code)
+
         # Handle both command and flag versions for call
         if args.command == 'call' or args.call:
             if not CALL_MODULE_AVAILABLE:

src/praisonai/praisonai/deploy.py

@@ -57,7 +57,7 @@ class CloudDeployer:
             file.write("FROM python:3.11-slim\n")
             file.write("WORKDIR /app\n")
             file.write("COPY . .\n")
-            file.write("RUN pip install flask praisonai==2.8.3 gunicorn markdown\n")
+            file.write("RUN pip install flask praisonai==2.8.9 gunicorn markdown\n")
             file.write("EXPOSE 8080\n")
             file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
             

src/praisonai/praisonai/ui/chainlit_compat.py

@@ -6,13 +6,17 @@ that may change between versions. It handles:
 - EXPIRY_TIME / storage_expiry_time constant (renamed/removed in various versions)
 - BaseStorageClient class location (moved from storage_clients.base to data.base)
 - BaseDataLayer abstract methods (close() added in 2.9.4, may be removed later)
+- LocalFileStorageClient for default element persistence
 
 Usage:
     from praisonai.ui.chainlit_compat import get_expiry_seconds, BaseStorageClient
 """
 
 import os
-from typing import Optional, Type
+import logging
+from typing import Any, Dict, Optional, Type, Union
+
+logger = logging.getLogger(__name__)
 
 # Default expiry time in seconds (1 hour) - used if Chainlit doesn't provide one
 DEFAULT_EXPIRY_SECONDS = 3600
@@ -144,6 +148,91 @@ except ImportError:
         BaseStorageClient = None
 
 
+class LocalFileStorageClient:
+    """
+    A simple local file storage client for persisting elements to disk.
+    
+    This provides a default storage implementation when no cloud storage
+    (S3, Azure, etc.) is configured. Files are stored in the CHAINLIT_APP_ROOT/.files directory.
+    """
+    
+    def __init__(self, storage_dir: Optional[str] = None):
+        """
+        Initialize the local file storage client.
+        
+        Args:
+            storage_dir: Directory to store files. Defaults to CHAINLIT_APP_ROOT/.files
+        """
+        if storage_dir:
+            self.storage_dir = storage_dir
+        else:
+            chainlit_root = os.environ.get("CHAINLIT_APP_ROOT", os.path.join(os.path.expanduser("~"), ".praison"))
+            self.storage_dir = os.path.join(chainlit_root, ".files")
+        
+        os.makedirs(self.storage_dir, exist_ok=True)
+        logger.debug(f"LocalFileStorageClient initialized with storage_dir: {self.storage_dir}")
+    
+    async def upload_file(
+        self,
+        object_key: str,
+        data: Union[bytes, str],
+        mime: str = "application/octet-stream",
+        overwrite: bool = True,
+    ) -> Dict[str, Any]:
+        """
+        Upload a file to local storage.
+        
+        Args:
+            object_key: The key/path for the file
+            data: The file content (bytes or string)
+            mime: MIME type of the file
+            overwrite: Whether to overwrite existing files
+            
+        Returns:
+            Dict with object_key and url
+        """
+        import aiofiles
+        
+        # Create full path
+        file_path = os.path.join(self.storage_dir, object_key)
+        
+        # Create parent directories if needed
+        os.makedirs(os.path.dirname(file_path), exist_ok=True)
+        
+        # Check if file exists and overwrite is False
+        if not overwrite and os.path.exists(file_path):
+            logger.warning(f"File {object_key} already exists and overwrite=False")
+            return {"object_key": object_key, "url": f"file://{file_path}"}
+        
+        # Write the file
+        try:
+            if isinstance(data, str):
+                async with aiofiles.open(file_path, 'w') as f:
+                    await f.write(data)
+            else:
+                async with aiofiles.open(file_path, 'wb') as f:
+                    await f.write(data)
+            
+            logger.debug(f"Uploaded file to {file_path}")
+            return {"object_key": object_key, "url": f"file://{file_path}"}
+        except Exception as e:
+            logger.error(f"Failed to upload file {object_key}: {e}")
+            return {}
+
+
+def create_local_storage_client(storage_dir: Optional[str] = None) -> LocalFileStorageClient:
+    """
+    Create a LocalFileStorageClient instance.
+    
+    Args:
+        storage_dir: Optional directory for file storage
+        
+    Returns:
+        LocalFileStorageClient instance
+    """
+    return LocalFileStorageClient(storage_dir=storage_dir)
+
+
 __all__ = [
     'get_expiry_seconds',
     'get_base_storage_client',
@@ -151,5 +240,7 @@ __all__ = [
     'base_data_layer_has_close',
     'EXPIRY_TIME',
     'BaseStorageClient',
+    'LocalFileStorageClient',
+    'create_local_storage_client',
     'DEFAULT_EXPIRY_SECONDS',
 ]

src/praisonai/praisonai/ui/chat.py

@@ -62,12 +62,36 @@ def load_setting(key: str) -> str:
 cl_data._data_layer = db_manager
 
 def load_custom_tools():
-    """Load custom tools from tools.py if it exists"""
+    """
+    Load custom tools from tools.py if it exists.
+    
+    Resolution order:
+    1. PRAISONAI_TOOLS_PATH environment variable (file or directory)
+    2. ./tools.py in current working directory
+    3. No tools (silent, no warning)
+    """
     custom_tools = {}
+    
+    # Determine tools path
+    tools_path = os.getenv("PRAISONAI_TOOLS_PATH")
+    if tools_path:
+        if not os.path.exists(tools_path):
+            logger.warning(f"PRAISONAI_TOOLS_PATH set but path does not exist: {tools_path}")
+            return custom_tools
+    else:
+        # Check current working directory for tools.py
+        cwd_tools = os.path.join(os.getcwd(), "tools.py")
+        if os.path.exists(cwd_tools):
+            tools_path = cwd_tools
+        else:
+            # No tools configured - this is fine, return silently
+            logger.debug("No tools.py found in current directory (this is normal)")
+            return custom_tools
+    
     try:
-        spec = importlib.util.spec_from_file_location("tools", "tools.py")
-        if spec is None:
-            logger.debug("tools.py not found in current directory")
+        spec = importlib.util.spec_from_file_location("tools", tools_path)
+        if spec is None or spec.loader is None:
+            logger.debug(f"Could not load tools from {tools_path}")
             return custom_tools
         
         module = importlib.util.module_from_spec(spec)
@@ -122,9 +146,13 @@ def load_custom_tools():
                 custom_tools[name] = tool_def
                 logger.info(f"Loaded custom tool: {name}")
         
-        logger.info(f"Loaded {len(custom_tools)} custom tools from tools.py")
+        if custom_tools:
+            logger.info(f"Loaded {len(custom_tools)} custom tools from {tools_path}")
+    except FileNotFoundError:
+        # File doesn't exist - this is fine, no warning needed
+        logger.debug(f"Tools file not found: {tools_path}")
     except Exception as e:
-        logger.warning(f"Error loading custom tools: {e}")
+        logger.warning(f"Error loading custom tools from {tools_path}: {e}")
     
     return custom_tools
 
@@ -203,24 +231,16 @@ expected_password = os.getenv("CHAINLIT_PASSWORD", "admin")
 if expected_username == "admin" and expected_password == "admin":
     logger.warning("⚠️  Using default admin credentials. Set CHAINLIT_USERNAME and CHAINLIT_PASSWORD environment variables for production.")
 
-def auth_callback(u: str, p: str):
-    if (u, p) == (expected_username, expected_password):
-        return cl.User(identifier=expected_username, metadata={"role": "ADMIN", "provider": "credentials"})
-    return None
+@cl.password_auth_callback
+def auth_callback(username: str, password: str):
+    logger.debug(f"Auth attempt: username='{username}', expected='{expected_username}'")
+    if (username, password) == (expected_username, expected_password):
+        logger.info(f"Login successful for user: {username}")
+        return cl.User(identifier=username, metadata={"role": "admin", "provider": "credentials"})
+    else:
+        logger.warning(f"Login failed for user: {username}")
+        return None
 
-def oauth_callback(
-    provider_id: str,
-    token: str,
-    raw_user_data: Dict[str, str],
-    default_user: cl.User,
-) -> Optional[cl.User]:
-    return default_user
-
-if AUTH_PASSWORD_ENABLED:
-    auth_callback = cl.password_auth_callback(auth_callback)
-
-if AUTH_OAUTH_ENABLED:
-    oauth_callback = cl.oauth_callback(oauth_callback)
 
 async def send_count():
     await cl.Message(
@@ -299,7 +319,6 @@ User Question: {message.content}
 
     message_history.append({"role": "user", "content": user_message})
     msg = cl.Message(content="")
-    await msg.send()
 
     completion_params = {
         "model": model_name,
@@ -329,6 +348,7 @@ User Question: {message.content}
     full_response = ""
     tool_calls = []
     current_tool_call = None
+    msg_sent = False
 
     async for part in response:
         if 'choices' in part and len(part['choices']) > 0:
@@ -336,6 +356,10 @@ User Question: {message.content}
 
             if 'content' in delta and delta['content'] is not None:
                 token = delta['content']
+                # Send message on first token to avoid delay
+                if not msg_sent:
+                    await msg.send()
+                    msg_sent = True
                 await msg.stream_token(token)
                 full_response += token
 
@@ -362,6 +386,10 @@ User Question: {message.content}
     if current_tool_call:
         tool_calls.append(current_tool_call)
 
+    # Ensure message is sent even if no content (e.g., tool calls only)
+    if not msg_sent:
+        await msg.send()
+
     logger.debug(f"Full response: {full_response}")
     logger.debug(f"Tool calls: {tool_calls}")
     message_history.append({"role": "assistant", "content": full_response})

src/praisonai/praisonai/ui/code.py

@@ -398,7 +398,6 @@ Context:
     message_history.append({"role": "user", "content": user_message})
 
     msg = cl.Message(content="")
-    await msg.send()
 
     # Use PraisonAI Agents if available, otherwise fallback to litellm
     if PRAISONAI_AGENTS_AVAILABLE:
@@ -433,41 +432,47 @@ Then use the Claude Code tool to handle those requests.
 
 For informational questions, explanations, or general conversations, respond normally without using Claude Code."""
 
-        # Create agent
+        # Create agent with streaming enabled
         agent = Agent(
             name="PraisonAI Assistant",
             instructions=instructions,
             llm=model_name,
-            tools=available_tools if available_tools else None
+            tools=available_tools if available_tools else None,
+            stream=True
         )
         
         # Execute agent with streaming
         full_response = ""
+        msg_sent = False
         
-        # Use agent's streaming capabilities if available
         try:
-            # For now, use synchronous execution and stream the result
-            # TODO: Implement proper streaming when PraisonAI agents support it
-            result = agent.start(user_message)
+            # Use async chat for proper streaming
+            result = await agent.achat(user_message)
             
-            # Stream the response character by character for better UX
+            # Get the response text
             if hasattr(result, 'raw'):
                 response_text = result.raw
             else:
                 response_text = str(result)
             
-            for char in response_text:
-                await msg.stream_token(char)
-                full_response += char
-                # Small delay to make streaming visible
-                await asyncio.sleep(0.01)
+            # Send message on first content
+            if not msg_sent:
+                await msg.send()
+                msg_sent = True
+            
+            # Stream in word chunks for better UX (not char-by-char which is too slow)
+            words = response_text.split(' ')
+            for i, word in enumerate(words):
+                token = word + (' ' if i < len(words) - 1 else '')
+                await msg.stream_token(token)
+                full_response += token
             
         except Exception as e:
             error_response = f"Error executing agent: {str(e)}"
-            for char in error_response:
-                await msg.stream_token(char)
-                full_response += char
-                await asyncio.sleep(0.01)
+            if not msg_sent:
+                await msg.send()
+            await msg.stream_token(error_response)
+            full_response = error_response
         
         msg.content = full_response
         await msg.update()
@@ -475,6 +480,7 @@ For informational questions, explanations, or general conversations, respond nor
     except Exception as e:
         error_msg = f"Failed to use PraisonAI Agents: {str(e)}"
         logger.error(error_msg)
+        await msg.send()
         await msg.stream_token(error_msg)
         msg.content = error_msg
         await msg.update()
@@ -515,6 +521,7 @@ async def handle_with_litellm(user_message, model_name, message_history, msg, im
     full_response = ""
     tool_calls = []
     current_tool_call = None
+    msg_sent = False
 
     async for part in response:
         logger.debug(f"LLM part: {part}")
@@ -523,6 +530,10 @@ async def handle_with_litellm(user_message, model_name, message_history, msg, im
             
             if 'content' in delta and delta['content'] is not None:
                 token = delta['content']
+                # Send message on first token
+                if not msg_sent:
+                    await msg.send()
+                    msg_sent = True
                 await msg.stream_token(token)
                 full_response += token
             
@@ -549,6 +560,10 @@ async def handle_with_litellm(user_message, model_name, message_history, msg, im
     if current_tool_call:
         tool_calls.append(current_tool_call)
 
+    # Ensure message is sent even if no content (tool calls only)
+    if not msg_sent:
+        await msg.send()
+
     logger.debug(f"Full response: {full_response}")
     logger.debug(f"Tool calls: {tool_calls}")
     message_history.append({"role": "assistant", "content": full_response})

src/praisonai/praisonai/ui/db.py

@@ -10,6 +10,7 @@ from sql_alchemy import SQLAlchemyDataLayer
 import chainlit.data as cl_data
 from chainlit.types import ThreadDict
 from database_config import get_database_url_with_sqlite_override
+from praisonai.ui.chainlit_compat import create_local_storage_client
 
 def ensure_directories():
     """Ensure required directories exist"""
@@ -68,8 +69,11 @@ class DatabaseManager(SQLAlchemyDataLayer):
             self.db_path = os.path.join(chainlit_root, "database.sqlite")
             self.conninfo = f"sqlite+aiosqlite:///{self.db_path}"
         
-        # Initialize SQLAlchemyDataLayer with the connection info
-        super().__init__(conninfo=self.conninfo)
+        # Create local file storage client for element persistence
+        storage_client = create_local_storage_client()
+        
+        # Initialize SQLAlchemyDataLayer with the connection info and storage client
+        super().__init__(conninfo=self.conninfo, storage_provider=storage_client)
 
     async def create_schema_async(self):
         """Create the database schema for PostgreSQL"""

src/praisonai/praisonai/version.py

@@ -1 +1 @@
-__version__ = "2.8.3"
+__version__ = "2.8.9"

src/praisonai/pyproject.toml

@@ -26,7 +26,7 @@ praisonai-call = "praisonai.api.call:main"
 
 [project.optional-dependencies]
 ui = [
-    "chainlit==2.5.5",
+    "chainlit>=2.8.5,<2.9.0",
     "sqlalchemy>=2.0.36",
     "aiosqlite>=0.20.0",
     "greenlet>=3.0.3",
@@ -45,7 +45,7 @@ openai = ["langchain-openai>=0.2.1,<0.3.0"]
 anthropic = ["langchain-anthropic>=0.3.0"]
 cohere = ["langchain-cohere>=0.3.0,<0.4.0"]
 chat = [
-    "chainlit==2.5.5",
+    "chainlit>=2.8.5,<2.9.0",
     "litellm>=1.72.6",
     "aiosqlite>=0.20.0",
     "greenlet>=3.0.3",
@@ -57,7 +57,7 @@ chat = [
     "pydantic<=2.10.1"
 ]
 code = [
-    "chainlit==2.5.5",
+    "chainlit>=2.8.5,<2.9.0",
     "litellm>=1.72.6",
     "aiosqlite>=0.20.0",
     "greenlet>=3.0.3",
@@ -68,7 +68,7 @@ code = [
     "pydantic<=2.10.1"
 ]
 realtime = [
-    "chainlit==2.5.5",
+    "chainlit>=2.8.5,<2.9.0",
     "litellm>=1.72.6",
     "aiosqlite>=0.20.0",
     "greenlet>=3.0.3",
@@ -100,6 +100,9 @@ autogen-v4 = [
     "praisonai-tools>=0.1.0", 
     "crewai"
 ]
+acp = [
+    "agent-client-protocol>=0.7.0",
+]
 
 [project.urls]
 Homepage = "https://docs.praison.ai"

src/praisonai/tests/integration/acp/__init__.py

@@ -0,0 +1 @@
+"""Integration tests for ACP module."""

src/praisonai/tests/integration/acp/test_acp_real_llm.py

@@ -0,0 +1,206 @@
+"""
+Real LLM integration tests for ACP.
+
+These tests use actual API keys to verify end-to-end functionality.
+API keys should be set via environment variables.
+"""
+
+import os
+import tempfile
+from pathlib import Path
+
+import pytest
+
+# Skip if no API key
+pytestmark = pytest.mark.skipif(
+    not os.environ.get("OPENAI_API_KEY"),
+    reason="OPENAI_API_KEY not set"
+)
+
+
+class TestACPRealLLM:
+    """Real LLM tests for ACP server."""
+    
+    @pytest.fixture
+    def acp_server(self):
+        """Create ACP server with real agent."""
+        from praisonai.acp.server import ACPServer
+        from praisonai.acp.config import ACPConfig
+        
+        config = ACPConfig(
+            workspace=Path(tempfile.mkdtemp()),
+            debug=True,
+            model="gpt-4o-mini",
+        )
+        return ACPServer(config=config)
+    
+    @pytest.mark.asyncio
+    async def test_initialize_real(self, acp_server):
+        """Test real initialize handshake."""
+        result = await acp_server.initialize(
+            protocol_version=1,
+            client_capabilities={"fs": {"readTextFile": True}},
+            client_info={"name": "test-harness", "version": "1.0"},
+        )
+        
+        assert result["protocolVersion"] == 1
+        assert result["agentInfo"]["name"] == "praisonai"
+        assert "agentCapabilities" in result
+        
+        # Log for verification (masked)
+        print(f"[TEST] Initialize successful: protocol={result['protocolVersion']}")
+    
+    @pytest.mark.asyncio
+    async def test_session_lifecycle(self, acp_server):
+        """Test complete session lifecycle."""
+        # Initialize
+        await acp_server.initialize(protocol_version=1)
+        
+        # Create session
+        session_result = await acp_server.new_session(
+            cwd="/tmp/test",
+            mcp_servers=[],
+        )
+        
+        session_id = session_result["sessionId"]
+        assert session_id.startswith("sess_")
+        
+        # Log for verification
+        print(f"[TEST] Session created: {session_id[:20]}...")
+        
+        # Set mode
+        await acp_server.set_session_mode(
+            mode_id="manual",
+            session_id=session_id,
+        )
+        
+        # Cancel
+        await acp_server.cancel(session_id=session_id)
+        
+        print("[TEST] Session lifecycle complete")
+    
+    @pytest.mark.asyncio
+    async def test_prompt_with_real_llm(self, acp_server):
+        """Test prompt with real LLM call."""
+        # Initialize
+        await acp_server.initialize(protocol_version=1)
+        
+        # Create session
+        session_result = await acp_server.new_session(
+            cwd="/tmp/test",
+            mcp_servers=[],
+        )
+        session_id = session_result["sessionId"]
+        
+        # Send a simple prompt
+        prompt_result = await acp_server.prompt(
+            prompt=[{"type": "text", "text": "What is 2+2? Reply with just the number."}],
+            session_id=session_id,
+        )
+        
+        # Verify response
+        assert "stopReason" in prompt_result
+        assert prompt_result["stopReason"] in ["end_turn", "cancelled", "refusal"]
+        
+        # Log for verification (no secrets)
+        print(f"[TEST] Prompt completed: stopReason={prompt_result['stopReason']}")
+        print(f"[TEST] Session: {session_id[:20]}...")
+    
+    @pytest.mark.asyncio
+    async def test_session_resume(self, acp_server):
+        """Test session resume functionality."""
+        # Initialize
+        await acp_server.initialize(protocol_version=1)
+        
+        # Create session
+        session_result = await acp_server.new_session(
+            cwd="/tmp/test",
+            mcp_servers=[],
+        )
+        session_id = session_result["sessionId"]
+        
+        # Add a message to the session
+        acp_server._sessions[session_id].add_message("user", "Test message")
+        acp_server._session_store.save(acp_server._sessions[session_id])
+        
+        # Create new server instance (simulating restart)
+        from praisonai.acp.config import ACPConfig
+        config = ACPConfig(workspace=Path("/tmp/test"), debug=True)
+        from praisonai.acp.server import ACPServer as ACPServerClass
+        new_server = ACPServerClass(config=config)
+        
+        # Resume session
+        resume_result = await new_server.load_session(
+            cwd="/tmp/test",
+            mcp_servers=[],
+            session_id=session_id,
+        )
+        
+        assert resume_result is not None
+        assert session_id in new_server._sessions
+        
+        # Verify message was preserved
+        loaded_session = new_server._sessions[session_id]
+        assert len(loaded_session.messages) == 1
+        assert loaded_session.messages[0]["content"] == "Test message"
+        
+        print(f"[TEST] Session resume successful: {session_id[:20]}...")
+
+
+class TestACPPermissionGating:
+    """Tests for permission gating."""
+    
+    def test_read_only_default(self):
+        """Verify read-only is default."""
+        from praisonai.acp.config import ACPConfig
+        
+        config = ACPConfig()
+        assert config.read_only is True
+        assert config.can_write() is False
+        assert config.can_execute_shell() is False
+    
+    def test_explicit_write_permission(self):
+        """Test explicit write permission."""
+        from praisonai.acp.config import ACPConfig
+        
+        config = ACPConfig(allow_write=True)
+        assert config.can_write() is True
+    
+    def test_workspace_boundary(self):
+        """Test workspace boundary enforcement."""
+        from praisonai.acp.config import ACPConfig
+        
+        config = ACPConfig(workspace="/tmp/test_workspace")
+        
+        # Within workspace - allowed
+        assert config.is_path_allowed(Path("/tmp/test_workspace/file.txt"))
+        assert config.is_path_allowed(Path("/tmp/test_workspace/subdir/file.txt"))
+        
+        # Outside workspace - not allowed
+        assert not config.is_path_allowed(Path("/etc/passwd"))
+        assert not config.is_path_allowed(Path("/var/log/system.log"))
+
+
+class TestACPTracing:
+    """Tests for tracing and attribution."""
+    
+    def test_session_has_trace_ids(self):
+        """Verify sessions have trace IDs."""
+        from praisonai.acp.session import ACPSession
+        
+        session = ACPSession.create(
+            workspace=Path("/tmp/test"),
+            agent_id="test_agent",
+        )
+        
+        assert session.session_id is not None
+        assert session.run_id is not None
+        assert session.trace_id is not None
+        assert session.agent_id == "test_agent"
+        
+        # IDs should have proper prefixes
+        assert session.session_id.startswith("sess_")
+        assert session.run_id.startswith("run_")
+        assert session.trace_id.startswith("trace_")
+        
+        print(f"[TEST] Trace IDs: session={session.session_id[:15]}, run={session.run_id}, trace={session.trace_id}")

src/praisonai/tests/integration/acp/test_acp_stdio.py

@@ -0,0 +1,324 @@
+"""
+Integration tests for ACP stdio communication.
+
+Tests the ACP server by spawning it as a subprocess and communicating
+via JSON-RPC over stdio.
+"""
+
+import json
+import subprocess
+import sys
+import tempfile
+import time
+from pathlib import Path
+
+import pytest
+
+
+class TestACPStdioIntegration:
+    """Integration tests for ACP stdio communication."""
+    
+    @pytest.fixture
+    def acp_process(self):
+        """Start ACP server as subprocess."""
+        # Use the CLI module directly to avoid crewai import issues
+        proc = subprocess.Popen(
+            [
+                sys.executable, "-c",
+                """
+import sys
+import asyncio
+from praisonai.acp.server import ACPServer
+from praisonai.acp.config import ACPConfig
+
+async def run():
+    config = ACPConfig(debug=True)
+    server = ACPServer(config=config)
+    
+    # Simple JSON-RPC handler for testing
+    import json
+    
+    while True:
+        try:
+            line = sys.stdin.readline()
+            if not line:
+                break
+            
+            request = json.loads(line.strip())
+            method = request.get("method", "")
+            params = request.get("params", {})
+            req_id = request.get("id")
+            
+            if method == "initialize":
+                result = await server.initialize(
+                    protocol_version=params.get("protocolVersion", 1),
+                    client_capabilities=params.get("clientCapabilities"),
+                    client_info=params.get("clientInfo"),
+                )
+            elif method == "session/new":
+                result = await server.new_session(
+                    cwd=params.get("cwd", "."),
+                    mcp_servers=params.get("mcpServers", []),
+                )
+            elif method == "session/cancel":
+                await server.cancel(session_id=params.get("sessionId", ""))
+                result = None
+            else:
+                result = {"error": f"Unknown method: {method}"}
+            
+            response = {
+                "jsonrpc": "2.0",
+                "id": req_id,
+                "result": result,
+            }
+            print(json.dumps(response), flush=True)
+        except Exception as e:
+            print(json.dumps({"jsonrpc": "2.0", "id": None, "error": {"code": -32603, "message": str(e)}}), flush=True)
+
+asyncio.run(run())
+"""
+            ],
+            stdin=subprocess.PIPE,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True,
+            bufsize=1,
+        )
+        yield proc
+        proc.terminate()
+        proc.wait(timeout=5)
+    
+    def send_request(self, proc, method: str, params: dict, req_id: int = 1) -> dict:
+        """Send JSON-RPC request and get response."""
+        request = {
+            "jsonrpc": "2.0",
+            "id": req_id,
+            "method": method,
+            "params": params,
+        }
+        proc.stdin.write(json.dumps(request) + "\n")
+        proc.stdin.flush()
+        
+        # Read response with timeout
+        response_line = proc.stdout.readline()
+        if response_line:
+            return json.loads(response_line.strip())
+        return None
+    
+    def test_initialize(self, acp_process):
+        """Test initialize handshake."""
+        response = self.send_request(
+            acp_process,
+            "initialize",
+            {
+                "protocolVersion": 1,
+                "clientCapabilities": {"fs": {"readTextFile": True}},
+                "clientInfo": {"name": "test-client", "version": "1.0"},
+            },
+        )
+        
+        assert response is not None
+        assert response["jsonrpc"] == "2.0"
+        assert response["id"] == 1
+        assert "result" in response
+        
+        result = response["result"]
+        assert result["protocolVersion"] == 1
+        assert "agentCapabilities" in result
+        assert result["agentInfo"]["name"] == "praisonai"
+    
+    def test_new_session(self, acp_process):
+        """Test session creation."""
+        # First initialize
+        self.send_request(
+            acp_process,
+            "initialize",
+            {"protocolVersion": 1},
+            req_id=1,
+        )
+        
+        # Then create session
+        response = self.send_request(
+            acp_process,
+            "session/new",
+            {
+                "cwd": "/tmp/test",
+                "mcpServers": [],
+            },
+            req_id=2,
+        )
+        
+        assert response is not None
+        assert response["id"] == 2
+        assert "result" in response
+        
+        result = response["result"]
+        assert "sessionId" in result
+        assert result["sessionId"].startswith("sess_")
+    
+    def test_cancel_notification(self, acp_process):
+        """Test cancel notification."""
+        # Initialize
+        self.send_request(acp_process, "initialize", {"protocolVersion": 1}, req_id=1)
+        
+        # Create session
+        new_resp = self.send_request(
+            acp_process,
+            "session/new",
+            {"cwd": "/tmp", "mcpServers": []},
+            req_id=2,
+        )
+        session_id = new_resp["result"]["sessionId"]
+        
+        # Send cancel (notification, but we're treating it as request for testing)
+        response = self.send_request(
+            acp_process,
+            "session/cancel",
+            {"sessionId": session_id},
+            req_id=3,
+        )
+        
+        assert response is not None
+        assert response["id"] == 3
+
+
+class TestACPNoStdoutPollution:
+    """Tests to ensure ACP doesn't pollute stdout."""
+    
+    def test_no_print_statements(self):
+        """Verify ACP modules don't have print statements that would pollute stdout."""
+        import ast
+        from pathlib import Path
+        
+        acp_dir = Path(__file__).parent.parent.parent.parent / "praisonai" / "acp"
+        
+        for py_file in acp_dir.glob("*.py"):
+            content = py_file.read_text()
+            try:
+                tree = ast.parse(content)
+                for node in ast.walk(tree):
+                    if isinstance(node, ast.Call):
+                        if isinstance(node.func, ast.Name) and node.func.id == "print":
+                            # Check if it's inside a function that's not for debugging
+                            pytest.fail(f"Found print() in {py_file.name} - stdout must be reserved for JSON-RPC")
+            except SyntaxError:
+                pass  # Skip files with syntax errors
+    
+    def test_logging_to_stderr(self):
+        """Verify logging is configured for stderr."""
+        from praisonai.acp.server import _setup_stderr_logging
+        import logging
+        import sys
+        
+        _setup_stderr_logging(debug=True)
+        
+        logger = logging.getLogger("praisonai.acp")
+        
+        # Check that handlers write to stderr
+        for handler in logger.handlers:
+            if hasattr(handler, 'stream'):
+                assert handler.stream == sys.stderr, "Logger must write to stderr, not stdout"
+
+
+class TestACPInteractiveModeCompatibility:
+    """Tests for ACP and interactive mode compatibility."""
+    
+    def test_acp_import_does_not_break_interactive(self):
+        """Verify importing ACP doesn't break interactive mode imports."""
+        # This should not raise any errors
+        from praisonai.acp import ACPConfig
+        
+        # Interactive mode imports should still work
+        from praisonai.cli.features.interactive_tui import InteractiveConfig
+        
+        # Both should be usable
+        acp_config = ACPConfig()
+        interactive_config = InteractiveConfig()
+        
+        assert acp_config is not None
+        assert interactive_config is not None
+    
+    def test_acp_lazy_import(self):
+        """Verify ACP uses lazy imports."""
+        import time
+        
+        start = time.time()
+        import praisonai.acp as acp_module  # noqa: F401
+        import_time = time.time() - start
+        
+        # Import should be fast (< 100ms)
+        assert import_time < 0.1, f"ACP import too slow: {import_time}s"
+    
+    def test_interactive_mode_without_acp_sdk(self):
+        """Verify interactive mode works even if ACP SDK is not installed."""
+        # Mock the ACP SDK not being available
+        import sys
+        
+        # Save original
+        original_modules = sys.modules.copy()
+        
+        try:
+            # Remove acp from modules if present
+            if 'acp' in sys.modules:
+                del sys.modules['acp']
+            
+            # Interactive mode should still work
+            from praisonai.cli.features.interactive_tui import InteractiveSession
+            
+            session = InteractiveSession()
+            assert session is not None
+        finally:
+            # Restore
+            sys.modules.update(original_modules)
+
+
+class TestACPSessionPersistence:
+    """Tests for ACP session persistence."""
+    
+    def test_session_save_and_load(self):
+        """Test session persistence across server restarts."""
+        from praisonai.acp.session import ACPSession, SessionStore
+        
+        with tempfile.TemporaryDirectory() as tmpdir:
+            store = SessionStore(storage_dir=Path(tmpdir))
+            
+            # Create and save session
+            session = ACPSession.create(workspace=Path("/tmp/test"))
+            session.add_message("user", "Hello")
+            session.add_message("assistant", "Hi there!")
+            store.save(session)
+            
+            # Create new store instance (simulating restart)
+            store2 = SessionStore(storage_dir=Path(tmpdir))
+            
+            # Load session
+            loaded = store2.load(session.session_id)
+            
+            assert loaded is not None
+            assert loaded.session_id == session.session_id
+            assert len(loaded.messages) == 2
+            assert loaded.messages[0]["content"] == "Hello"
+    
+    def test_resume_last_session(self):
+        """Test --resume --last functionality."""
+        from praisonai.acp.session import ACPSession, SessionStore
+        
+        with tempfile.TemporaryDirectory() as tmpdir:
+            store = SessionStore(storage_dir=Path(tmpdir))
+            
+            # Create multiple sessions
+            session1 = ACPSession.create(workspace=Path("/tmp/test1"))
+            store.save(session1)
+            
+            time.sleep(0.01)  # Ensure different timestamps
+            
+            session2 = ACPSession.create(workspace=Path("/tmp/test2"))
+            session2.add_message("user", "Last session message")
+            store.save(session2)
+            
+            # Load last session
+            last = store.load_last()
+            
+            assert last is not None
+            assert last.session_id == session2.session_id
+            assert len(last.messages) == 1

src/praisonai/tests/unit/acp/__init__.py

@@ -0,0 +1 @@
+"""Unit tests for ACP module."""

src/praisonai/tests/unit/acp/test_config.py

@@ -0,0 +1,166 @@
+"""Unit tests for ACP configuration."""
+
+import os
+import tempfile
+from pathlib import Path
+
+from praisonai.acp.config import ACPConfig
+
+
+class TestACPConfig:
+    """Tests for ACPConfig class."""
+    
+    def test_default_config(self):
+        """Test default configuration values."""
+        config = ACPConfig()
+        
+        assert config.workspace == Path.cwd()
+        assert config.agent == "default"
+        assert config.agents_config is None
+        assert config.router_enabled is False
+        assert config.model is None
+        assert config.resume_session is None
+        assert config.resume_last is False
+        assert config.read_only is True
+        assert config.allow_write is False
+        assert config.allow_shell is False
+        assert config.allow_network is False
+        assert config.approval_mode == "manual"
+        assert config.debug is False
+        assert config.profile is None
+    
+    def test_workspace_normalization(self):
+        """Test workspace path normalization."""
+        config = ACPConfig(workspace=".")
+        assert config.workspace == Path.cwd()
+        
+        config = ACPConfig(workspace="/tmp")
+        # Use resolve() to handle macOS symlinks (/tmp -> /private/tmp)
+        assert config.workspace == Path("/tmp").resolve()
+    
+    def test_allowed_paths_default(self):
+        """Test default allowed paths."""
+        config = ACPConfig()
+        
+        assert config.workspace in config.allowed_paths
+        assert Path.home() / ".praison" in config.allowed_paths
+    
+    def test_is_path_allowed(self):
+        """Test path allowlist checking."""
+        config = ACPConfig(workspace="/tmp/test_workspace")
+        
+        # Workspace itself should be allowed
+        assert config.is_path_allowed(Path("/tmp/test_workspace"))
+        
+        # Subdirectories should be allowed
+        assert config.is_path_allowed(Path("/tmp/test_workspace/subdir"))
+        
+        # Parent directories should NOT be allowed
+        assert not config.is_path_allowed(Path("/tmp"))
+        
+        # Unrelated paths should NOT be allowed
+        assert not config.is_path_allowed(Path("/var/log"))
+    
+    def test_can_write(self):
+        """Test write permission checking."""
+        # Read-only mode
+        config = ACPConfig(read_only=True, allow_write=False)
+        assert not config.can_write()
+        
+        # Allow write overrides read-only
+        config = ACPConfig(read_only=True, allow_write=True)
+        assert config.can_write()
+        
+        # Not read-only
+        config = ACPConfig(read_only=False)
+        assert config.can_write()
+    
+    def test_can_execute_shell(self):
+        """Test shell execution permission checking."""
+        config = ACPConfig(allow_shell=False)
+        assert not config.can_execute_shell()
+        
+        config = ACPConfig(allow_shell=True)
+        assert config.can_execute_shell()
+    
+    def test_from_env(self, monkeypatch):
+        """Test configuration from environment variables."""
+        monkeypatch.setenv("PRAISONAI_WORKSPACE", "/tmp/env_workspace")
+        monkeypatch.setenv("PRAISONAI_AGENT", "test_agent")
+        monkeypatch.setenv("PRAISONAI_MODEL", "gpt-4")
+        monkeypatch.setenv("PRAISONAI_DEBUG", "true")
+        monkeypatch.setenv("PRAISONAI_READ_ONLY", "false")
+        monkeypatch.setenv("PRAISONAI_APPROVAL_MODE", "auto")
+        
+        config = ACPConfig.from_env()
+        
+        # Use resolve() to handle macOS symlinks
+        assert config.workspace == Path("/tmp/env_workspace").resolve()
+        assert config.agent == "test_agent"
+        assert config.model == "gpt-4"
+        assert config.debug is True
+        assert config.read_only is False
+        assert config.approval_mode == "auto"
+    
+    def test_from_file(self):
+        """Test configuration from YAML file."""
+        with tempfile.NamedTemporaryFile(mode='w', suffix='.yaml', delete=False) as f:
+            f.write("""
+acp:
+  workspace: /tmp/file_workspace
+  agent: file_agent
+  model: gpt-3.5-turbo
+  debug: true
+  read_only: false
+  approval_mode: scoped
+""")
+            f.flush()
+            
+            config = ACPConfig.from_file(Path(f.name))
+            
+            # Use resolve() to handle macOS symlinks
+            assert config.workspace == Path("/tmp/file_workspace").resolve()
+            assert config.agent == "file_agent"
+            assert config.model == "gpt-3.5-turbo"
+            assert config.debug is True
+            assert config.read_only is False
+            assert config.approval_mode == "scoped"
+        
+        os.unlink(f.name)
+    
+    def test_from_file_missing(self):
+        """Test configuration from non-existent file."""
+        config = ACPConfig.from_file(Path("/nonexistent/config.yaml"))
+        
+        # Should return default config
+        assert config.agent == "default"
+        assert config.debug is False
+    
+    def test_to_dict(self):
+        """Test configuration serialization."""
+        config = ACPConfig(
+            workspace="/tmp/test",
+            agent="test_agent",
+            model="gpt-4",
+            debug=True,
+        )
+        
+        data = config.to_dict()
+        
+        # Use resolve() to handle macOS symlinks
+        assert data["workspace"] == str(Path("/tmp/test").resolve())
+        assert data["agent"] == "test_agent"
+        assert data["model"] == "gpt-4"
+        assert data["debug"] is True
+    
+    def test_merge(self):
+        """Test configuration merging."""
+        config1 = ACPConfig(agent="agent1", model="model1")
+        config2 = ACPConfig(agent="agent2", debug=True)
+        
+        merged = ACPConfig.merge(config1, config2)
+        
+        # Later config overrides
+        assert merged.agent == "agent2"
+        assert merged.model == "model1"  # Not overridden
+        assert merged.debug is True

src/praisonai/tests/unit/acp/test_server.py

@@ -0,0 +1,244 @@
+"""Unit tests for ACP server."""
+
+import asyncio
+from pathlib import Path
+from praisonai.acp.config import ACPConfig
+from praisonai.acp.server import ACPServer
+
+
+class TestACPServer:
+    """Tests for ACPServer class."""
+    
+    def test_server_init(self):
+        """Test server initialization."""
+        config = ACPConfig(workspace="/tmp/test", debug=True)
+        server = ACPServer(config=config)
+        
+        assert server.config == config
+        assert server._sessions == {}
+        assert server._cancelled_sessions == set()
+    
+    def test_server_init_default_config(self):
+        """Test server initialization with default config."""
+        server = ACPServer()
+        
+        assert server.config is not None
+        assert server.config.workspace == Path.cwd()
+    
+    def test_initialize(self):
+        """Test initialize handler."""
+        server = ACPServer()
+        
+        result = asyncio.run(server.initialize(
+            protocol_version=1,
+            client_capabilities={"fs": {"readTextFile": True}},
+            client_info={"name": "test-client", "version": "1.0"},
+        ))
+        
+        assert result["protocolVersion"] == 1
+        assert "agentCapabilities" in result
+        assert result["agentCapabilities"]["loadSession"] is True
+        assert result["agentInfo"]["name"] == "praisonai"
+    
+    def test_new_session(self):
+        """Test new_session handler."""
+        server = ACPServer()
+        
+        result = asyncio.run(server.new_session(
+            cwd="/tmp/test_workspace",
+            mcp_servers=[],
+        ))
+        
+        assert "sessionId" in result
+        assert result["sessionId"].startswith("sess_")
+        assert result["sessionId"] in server._sessions
+    
+    def test_new_session_with_mcp_servers(self):
+        """Test new_session with MCP servers."""
+        server = ACPServer()
+        
+        mcp_servers = [
+            {"name": "test", "command": "/usr/bin/test", "args": []}
+        ]
+        
+        result = asyncio.run(server.new_session(
+            cwd="/tmp/test",
+            mcp_servers=mcp_servers,
+        ))
+        
+        session = server._sessions[result["sessionId"]]
+        assert session.mcp_servers == mcp_servers
+    
+    def test_set_session_mode(self):
+        """Test set_session_mode handler."""
+        server = ACPServer()
+        
+        # Create a session first
+        new_result = asyncio.run(server.new_session(
+            cwd="/tmp/test",
+            mcp_servers=[],
+        ))
+        session_id = new_result["sessionId"]
+        
+        # Set mode
+        result = asyncio.run(server.set_session_mode(
+            mode_id="auto",
+            session_id=session_id,
+        ))
+        
+        assert result == {}
+        assert server._sessions[session_id].mode == "auto"
+    
+    def test_cancel(self):
+        """Test cancel handler."""
+        server = ACPServer()
+        
+        # Create a session first
+        new_result = asyncio.run(server.new_session(
+            cwd="/tmp/test",
+            mcp_servers=[],
+        ))
+        session_id = new_result["sessionId"]
+        
+        # Cancel it
+        asyncio.run(server.cancel(session_id=session_id))
+        
+        assert session_id in server._cancelled_sessions
+    
+    def test_authenticate(self):
+        """Test authenticate handler."""
+        server = ACPServer()
+        
+        result = asyncio.run(server.authenticate(method_id="test"))
+        
+        assert result == {}
+    
+    def test_list_sessions(self):
+        """Test list_sessions handler."""
+        import tempfile
+        with tempfile.TemporaryDirectory() as tmpdir:
+            from praisonai.acp.session import SessionStore
+            server = ACPServer()
+            # Use isolated session store
+            server._session_store = SessionStore(storage_dir=Path(tmpdir))
+            
+            # Create some sessions
+            asyncio.run(server.new_session(cwd="/tmp/test1", mcp_servers=[]))
+            asyncio.run(server.new_session(cwd="/tmp/test2", mcp_servers=[]))
+            
+            result = asyncio.run(server.list_sessions())
+            
+            assert "sessions" in result
+            assert len(result["sessions"]) == 2
+    
+    def test_fork_session(self):
+        """Test fork_session handler."""
+        server = ACPServer()
+        
+        # Create original session
+        original = asyncio.run(server.new_session(
+            cwd="/tmp/test",
+            mcp_servers=[],
+        ))
+        original_id = original["sessionId"]
+        
+        # Add a message to original
+        server._sessions[original_id].add_message("user", "Hello")
+        
+        # Fork it
+        forked = asyncio.run(server.fork_session(
+            cwd="/tmp/test2",
+            session_id=original_id,
+        ))
+        
+        assert forked["sessionId"] != original_id
+        assert forked["sessionId"] in server._sessions
+        
+        # Forked session should have the message
+        forked_session = server._sessions[forked["sessionId"]]
+        assert len(forked_session.messages) == 1
+    
+    def test_extract_prompt_text(self):
+        """Test prompt text extraction."""
+        server = ACPServer()
+        
+        prompt = [
+            {"type": "text", "text": "Hello world"},
+            {"type": "text", "text": "How are you?"},
+        ]
+        
+        result = server._extract_prompt_text(prompt)
+        
+        assert "Hello world" in result
+        assert "How are you?" in result
+    
+    def test_extract_prompt_text_with_resource(self):
+        """Test prompt text extraction with resource."""
+        server = ACPServer()
+        
+        prompt = [
+            {"type": "text", "text": "Check this file:"},
+            {
+                "type": "resource",
+                "resource": {
+                    "uri": "file:///tmp/test.py",
+                    "text": "print('hello')",
+                }
+            },
+        ]
+        
+        result = server._extract_prompt_text(prompt)
+        
+        assert "Check this file:" in result
+        assert "print('hello')" in result
+        assert "file:///tmp/test.py" in result
+    
+    def test_get_available_modes(self):
+        """Test available modes."""
+        server = ACPServer()
+        
+        modes = server._get_available_modes()
+        
+        assert modes is not None
+        assert len(modes) >= 2
+        
+        mode_ids = [m["id"] for m in modes]
+        assert "manual" in mode_ids
+        assert "auto" in mode_ids
+
+
+class TestACPServerPrompt:
+    """Tests for ACPServer prompt handling."""
+    
+    def test_prompt_cancelled(self):
+        """Test prompt with cancelled session."""
+        server = ACPServer()
+        
+        # Create session
+        new_result = asyncio.run(server.new_session(
+            cwd="/tmp/test",
+            mcp_servers=[],
+        ))
+        session_id = new_result["sessionId"]
+        
+        # Mark as cancelled
+        server._cancelled_sessions.add(session_id)
+        
+        # Try to prompt
+        result = asyncio.run(server.prompt(
+            prompt=[{"type": "text", "text": "Hello"}],
+            session_id=session_id,
+        ))
+        
+        assert result["stopReason"] == "cancelled"
+    
+    def test_prompt_nonexistent_session(self):
+        """Test prompt with non-existent session."""
+        server = ACPServer()
+        
+        result = asyncio.run(server.prompt(
+            prompt=[{"type": "text", "text": "Hello"}],
+            session_id="nonexistent",
+        ))
+        
+        assert result["stopReason"] == "refusal"

src/praisonai/tests/unit/acp/test_session.py

@@ -0,0 +1,203 @@
+"""Unit tests for ACP session management."""
+
+import tempfile
+import time
+from pathlib import Path
+
+from praisonai.acp.session import ACPSession, SessionStore
+
+
+class TestACPSession:
+    """Tests for ACPSession class."""
+    
+    def test_create_session(self):
+        """Test session creation."""
+        session = ACPSession.create(
+            workspace=Path("/tmp/test"),
+            agent_id="test_agent",
+        )
+        
+        assert session.session_id.startswith("sess_")
+        assert session.workspace == Path("/tmp/test")
+        assert session.agent_id == "test_agent"
+        assert session.run_id is not None
+        assert session.trace_id is not None
+        assert session.mode == "manual"
+    
+    def test_update_activity(self):
+        """Test activity timestamp update."""
+        session = ACPSession.create(workspace=Path("/tmp"))
+        original_time = session.last_activity
+        
+        time.sleep(0.01)
+        session.update_activity()
+        
+        assert session.last_activity > original_time
+    
+    def test_add_message(self):
+        """Test adding messages to session."""
+        session = ACPSession.create(workspace=Path("/tmp"))
+        
+        session.add_message("user", "Hello")
+        session.add_message("assistant", "Hi there!")
+        
+        assert len(session.messages) == 2
+        assert session.messages[0]["role"] == "user"
+        assert session.messages[0]["content"] == "Hello"
+        assert session.messages[1]["role"] == "assistant"
+        assert session.messages[1]["content"] == "Hi there!"
+    
+    def test_add_tool_call(self):
+        """Test adding tool calls to session."""
+        session = ACPSession.create(workspace=Path("/tmp"))
+        
+        session.add_tool_call(
+            tool_call_id="call_001",
+            title="Read file",
+            status="completed",
+            path="/tmp/test.txt",
+        )
+        
+        assert len(session.tool_calls) == 1
+        assert session.tool_calls[0]["tool_call_id"] == "call_001"
+        assert session.tool_calls[0]["title"] == "Read file"
+        assert session.tool_calls[0]["status"] == "completed"
+        assert session.tool_calls[0]["path"] == "/tmp/test.txt"
+    
+    def test_to_dict(self):
+        """Test session serialization."""
+        session = ACPSession.create(
+            workspace=Path("/tmp/test"),
+            agent_id="test_agent",
+        )
+        session.add_message("user", "Test message")
+        
+        data = session.to_dict()
+        
+        assert data["session_id"] == session.session_id
+        assert data["workspace"] == "/tmp/test"
+        assert data["agent_id"] == "test_agent"
+        assert len(data["messages"]) == 1
+    
+    def test_from_dict(self):
+        """Test session deserialization."""
+        data = {
+            "session_id": "sess_test123",
+            "workspace": "/tmp/test",
+            "created_at": 1234567890.0,
+            "last_activity": 1234567891.0,
+            "agent_id": "test_agent",
+            "run_id": "run_abc",
+            "trace_id": "trace_xyz",
+            "mode": "auto",
+            "model": "gpt-4",
+            "messages": [{"role": "user", "content": "Hello"}],
+            "tool_calls": [],
+            "mcp_servers": [],
+        }
+        
+        session = ACPSession.from_dict(data)
+        
+        assert session.session_id == "sess_test123"
+        assert session.workspace == Path("/tmp/test")
+        assert session.agent_id == "test_agent"
+        assert session.mode == "auto"
+        assert session.model == "gpt-4"
+        assert len(session.messages) == 1
+
+
+class TestSessionStore:
+    """Tests for SessionStore class."""
+    
+    def test_save_and_load(self):
+        """Test saving and loading sessions."""
+        with tempfile.TemporaryDirectory() as tmpdir:
+            store = SessionStore(storage_dir=Path(tmpdir))
+            
+            session = ACPSession.create(workspace=Path("/tmp/test"))
+            session.add_message("user", "Test")
+            
+            store.save(session)
+            
+            loaded = store.load(session.session_id)
+            
+            assert loaded is not None
+            assert loaded.session_id == session.session_id
+            assert len(loaded.messages) == 1
+    
+    def test_load_nonexistent(self):
+        """Test loading non-existent session."""
+        with tempfile.TemporaryDirectory() as tmpdir:
+            store = SessionStore(storage_dir=Path(tmpdir))
+            
+            loaded = store.load("nonexistent_session")
+            
+            assert loaded is None
+    
+    def test_load_last(self):
+        """Test loading the last session."""
+        with tempfile.TemporaryDirectory() as tmpdir:
+            store = SessionStore(storage_dir=Path(tmpdir))
+            
+            # Create and save multiple sessions
+            session1 = ACPSession.create(workspace=Path("/tmp/test1"))
+            store.save(session1)
+            
+            session2 = ACPSession.create(workspace=Path("/tmp/test2"))
+            store.save(session2)
+            
+            # Last saved should be session2
+            last = store.load_last()
+            
+            assert last is not None
+            assert last.session_id == session2.session_id
+    
+    def test_delete(self):
+        """Test deleting a session."""
+        with tempfile.TemporaryDirectory() as tmpdir:
+            store = SessionStore(storage_dir=Path(tmpdir))
+            
+            session = ACPSession.create(workspace=Path("/tmp/test"))
+            store.save(session)
+            
+            # Verify it exists
+            assert store.load(session.session_id) is not None
+            
+            # Delete it
+            result = store.delete(session.session_id)
+            assert result is True
+            
+            # Verify it's gone
+            assert store.load(session.session_id) is None
+    
+    def test_list_sessions(self):
+        """Test listing sessions."""
+        with tempfile.TemporaryDirectory() as tmpdir:
+            store = SessionStore(storage_dir=Path(tmpdir))
+            
+            # Create multiple sessions
+            for i in range(3):
+                session = ACPSession.create(workspace=Path(f"/tmp/test{i}"))
+                store.save(session)
+                time.sleep(0.01)  # Ensure different timestamps
+            
+            sessions = store.list_sessions()
+            
+            assert len(sessions) == 3
+            # Should be sorted by last_activity, most recent first
+            assert sessions[0].last_activity >= sessions[1].last_activity
+            assert sessions[1].last_activity >= sessions[2].last_activity
+    
+    def test_list_sessions_limit(self):
+        """Test listing sessions with limit."""
+        with tempfile.TemporaryDirectory() as tmpdir:
+            store = SessionStore(storage_dir=Path(tmpdir))
+            
+            # Create multiple sessions
+            for i in range(5):
+                session = ACPSession.create(workspace=Path(f"/tmp/test{i}"))
+                store.save(session)
+            
+            sessions = store.list_sessions(limit=2)
+            
+            assert len(sessions) == 2

src/praisonai/tests/unit/test_chainlit_compat.py

@@ -196,5 +196,90 @@ class TestChatPyLogLevel:
             assert log_level == "DEBUG"
 
 
+class TestLocalFileStorageClient:
+    """Tests for LocalFileStorageClient."""
+
+    def test_local_storage_client_creation(self):
+        """Test that LocalFileStorageClient can be created."""
+        from praisonai.ui.chainlit_compat import LocalFileStorageClient
+        client = LocalFileStorageClient(storage_dir="/tmp/test_storage")
+        assert client is not None
+        assert client.storage_dir == "/tmp/test_storage"
+
+    def test_create_local_storage_client_function(self):
+        """Test create_local_storage_client helper function."""
+        from praisonai.ui.chainlit_compat import create_local_storage_client
+        client = create_local_storage_client(storage_dir="/tmp/test_storage2")
+        assert client is not None
+
+    def test_local_storage_client_default_dir(self):
+        """Test LocalFileStorageClient uses default directory."""
+        from praisonai.ui.chainlit_compat import LocalFileStorageClient
+        with patch.dict(os.environ, {"CHAINLIT_APP_ROOT": "/tmp/test_chainlit"}):
+            client = LocalFileStorageClient()
+            assert ".files" in client.storage_dir
+
+
+class TestToolsLoading:
+    """Tests for custom tools loading."""
+
+    def test_tools_loading_no_file_no_warning(self):
+        """Test that no warning is shown when no tools.py exists."""
+        # The fix ensures no warning when tools.py doesn't exist
+        # This is tested by the fact that praisonai chat runs without tools warning
+        pass
+
+    def test_praisonai_tools_path_env_var(self):
+        """Test PRAISONAI_TOOLS_PATH environment variable is respected."""
+        # This tests the resolution order in load_custom_tools
+        assert os.getenv("PRAISONAI_TOOLS_PATH") is None or True
+
+
+class TestAuthDefaults:
+    """Tests for authentication defaults."""
+
+    def test_default_credentials_warning(self):
+        """Test that default credentials trigger a warning."""
+        # The warning is: "Using default admin credentials..."
+        # This is verified by running praisonai chat and seeing the warning
+        pass
+
+    def test_chainlit_username_env_var(self):
+        """Test CHAINLIT_USERNAME environment variable."""
+        with patch.dict(os.environ, {"CHAINLIT_USERNAME": "testuser"}):
+            assert os.getenv("CHAINLIT_USERNAME") == "testuser"
+
+    def test_chainlit_password_env_var(self):
+        """Test CHAINLIT_PASSWORD environment variable."""
+        with patch.dict(os.environ, {"CHAINLIT_PASSWORD": "testpass"}):
+            assert os.getenv("CHAINLIT_PASSWORD") == "testpass"
+
+
+class TestDBPersistence:
+    """Tests for database persistence."""
+
+    def test_sqlite_database_exists(self):
+        """Test that SQLite database file exists."""
+        db_path = os.path.expanduser("~/.praison/database.sqlite")
+        # Database should exist after running praisonai chat
+        if os.path.exists(db_path):
+            assert os.path.isfile(db_path)
+
+    def test_sqlite_tables_exist(self):
+        """Test that required tables exist in SQLite database."""
+        import sqlite3
+        db_path = os.path.expanduser("~/.praison/database.sqlite")
+        if os.path.exists(db_path):
+            conn = sqlite3.connect(db_path)
+            cursor = conn.cursor()
+            cursor.execute("SELECT name FROM sqlite_master WHERE type='table'")
+            tables = [t[0] for t in cursor.fetchall()]
+            conn.close()
+            
+            expected_tables = ['users', 'threads', 'steps', 'elements', 'feedbacks', 'settings']
+            for table in expected_tables:
+                assert table in tables, f"Table {table} not found"
+
+
 if __name__ == "__main__":
     pytest.main([__file__, "-v"])

src/praisonai/uv.lock

@@ -9,6 +9,18 @@ resolution-markers = [
     "python_full_version < '3.11'",
 ]
 
+[[package]]
+name = "agent-client-protocol"
+version = "0.7.1"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "pydantic" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/db/7c/12da39be4f73026fd9b02144df5f64d803488cf1439aa221b0edb7c305e3/agent_client_protocol-0.7.1.tar.gz", hash = "sha256:8d7031209e14c3f2f987e3b95e7d9c3286158e7b2af1bf43d6aae5b8a429249f", size = 66226, upload-time = "2025-12-28T13:58:57.012Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/03/48/48d2fb454f911147432cd779f548e188274e1700f1cbe0a258e78158331a/agent_client_protocol-0.7.1-py3-none-any.whl", hash = "sha256:4ffe999488f2b23db26f09becdfaa2aaae6529f0847a52bca61bc2c628001c0f", size = 53771, upload-time = "2025-12-28T13:58:55.967Z" },
+]
+
 [[package]]
 name = "agentops"
 version = "0.3.16"
@@ -177,6 +189,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/e4/ad/77fad9d6f974ec58d837cb49fb9b483d6227a420c4f908c3578633de1d47/alphashape-1.3.1-py2.py3-none-any.whl", hash = "sha256:96a5ddd5f09534a35f03a8916aeeaac00fe4d6bec2f9ad78f87f57be3007f795", size = 13122, upload-time = "2021-04-16T17:47:17.773Z" },
 ]
 
+[[package]]
+name = "annotated-doc"
+version = "0.0.4"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/57/ba/046ceea27344560984e26a590f90bc7f4a75b06701f653222458922b558c/annotated_doc-0.0.4.tar.gz", hash = "sha256:fbcda96e87e9c92ad167c2e53839e57503ecfda18804ea28102353485033faa4", size = 7288, upload-time = "2025-11-10T22:07:42.062Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/1e/d3/26bf1008eb3d2daa8ef4cacc7f3bfdc11818d111f7e2d0201bc6e3b49d45/annotated_doc-0.0.4-py3-none-any.whl", hash = "sha256:571ac1dc6991c450b25a9c2d84a3705e2ae7a53467b5d111c24fa8baabbed320", size = 5303, upload-time = "2025-11-10T22:07:40.673Z" },
+]
+
 [[package]]
 name = "annotated-types"
 version = "0.7.0"
@@ -260,14 +281,15 @@ wheels = [
 
 [[package]]
 name = "asyncer"
-version = "0.0.7"
+version = "0.0.12"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "anyio" },
+    { name = "sniffio" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/39/29/245ba9fa5769a1e3226c1157aedb372fe9dab28c4e1dcf6911d84d3a5e04/asyncer-0.0.7.tar.gz", hash = "sha256:d5e563fb0f56eb87b97257984703658a4f5bbdb52ff851b3e8ed864cc200b1d2", size = 14437, upload-time = "2024-04-30T06:26:00.79Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/33/3a/e3d6919a41328f837a1c73ac057a8be35276c350046a79af588c536bd0fc/asyncer-0.0.12.tar.gz", hash = "sha256:4ceaa472bf40f3778caa3bd7f7dfcbafec0a948a9688e3c90a1796469ebe5e19", size = 17909, upload-time = "2025-12-26T12:05:10.248Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/3e/4b/40a1dc52fc26695b1e80a9e67dfb0fe7e6ddc57bbc5b61348e40c0045abb/asyncer-0.0.7-py3-none-any.whl", hash = "sha256:f0d579d4f67c4ead52ede3a45c854f462cae569058a8a6a68a4ebccac1c335d8", size = 8476, upload-time = "2024-04-30T06:25:58.991Z" },
+    { url = "https://files.pythonhosted.org/packages/54/f6/ebdca24b43151ed2a7dcf98bab4b61fdfc50dbd7224b9b4de195099fa65e/asyncer-0.0.12-py3-none-any.whl", hash = "sha256:2eb8bb255bdecf04a81df2d47cf168401df1f65a5f62e1bccc1f86ba375c9913", size = 9320, upload-time = "2025-12-26T12:05:08.848Z" },
 ]
 
 [[package]]
@@ -602,11 +624,12 @@ wheels = [
 
 [[package]]
 name = "chainlit"
-version = "2.5.5"
+version = "2.8.5"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "aiofiles" },
     { name = "asyncer" },
+    { name = "audioop-lts", marker = "python_full_version >= '3.13'" },
     { name = "click" },
     { name = "dataclasses-json" },
     { name = "fastapi" },
@@ -618,6 +641,7 @@ dependencies = [
     { name = "nest-asyncio" },
     { name = "packaging" },
     { name = "pydantic" },
+    { name = "pydantic-settings" },
     { name = "pyjwt" },
     { name = "python-dotenv" },
     { name = "python-multipart" },
@@ -625,13 +649,12 @@ dependencies = [
     { name = "starlette" },
     { name = "syncer" },
     { name = "tomli" },
-    { name = "uptrace" },
     { name = "uvicorn" },
     { name = "watchfiles" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/d9/32/e4a6ac0a8e7430cd7b76b712b8df2f83abf129e7b843180e319f647fd3be/chainlit-2.5.5.tar.gz", hash = "sha256:d6388f93b9089e3a80bc2bd286a157a4a9d8016dc2c806372fd77119bcc0d36c", size = 9520733, upload-time = "2025-04-14T13:27:15.432Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/46/71/d0465682910792ff382d604b9973dcd0115c682dde6c22753eaa4304afe3/chainlit-2.8.5.tar.gz", hash = "sha256:ebe73b3ad9d7e545454432eb280618ef92d69d6470eb66488149c9bb7f6c526d", size = 9884512, upload-time = "2025-11-07T22:58:11.181Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/d8/b2/9bf8c07b18ef34dc70771a0e21f6f0a5392dd3a753135743178cd59f1b7f/chainlit-2.5.5-py3-none-any.whl", hash = "sha256:1783c7b868fe41158283b498020fa18a23c39d07319ad9927e429f1e888f6eb5", size = 9670768, upload-time = "2025-04-14T13:27:13.415Z" },
+    { url = "https://files.pythonhosted.org/packages/2f/e2/c9a1ff354acc05fe910a19a8053647674315c091f33fe3b1eef3c69fcbe7/chainlit-2.8.5-py3-none-any.whl", hash = "sha256:1d6ef81a14f4b9a526229b0e6839f364c8ef1a3335e648889d43c0c8f8444a8e", size = 9725985, upload-time = "2025-11-07T22:58:08.631Z" },
 ]
 
 [[package]]
@@ -1130,16 +1153,17 @@ wheels = [
 
 [[package]]
 name = "fastapi"
-version = "0.115.6"
+version = "0.128.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
+    { name = "annotated-doc" },
     { name = "pydantic" },
     { name = "starlette" },
     { name = "typing-extensions" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/93/72/d83b98cd106541e8f5e5bfab8ef2974ab45a62e8a6c5b5e6940f26d2ed4b/fastapi-0.115.6.tar.gz", hash = "sha256:9ec46f7addc14ea472958a96aae5b5de65f39721a46aaf5705c480d9a8b76654", size = 301336, upload-time = "2024-12-03T22:46:01.629Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/52/08/8c8508db6c7b9aae8f7175046af41baad690771c9bcde676419965e338c7/fastapi-0.128.0.tar.gz", hash = "sha256:1cc179e1cef10a6be60ffe429f79b829dce99d8de32d7acb7e6c8dfdf7f2645a", size = 365682, upload-time = "2025-12-27T15:21:13.714Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/52/b3/7e4df40e585df024fac2f80d1a2d579c854ac37109675db2b0cc22c0bb9e/fastapi-0.115.6-py3-none-any.whl", hash = "sha256:e9240b29e36fa8f4bb7290316988e90c381e5092e0cbe84e7818cc3713bcf305", size = 94843, upload-time = "2024-12-03T22:45:59.368Z" },
+    { url = "https://files.pythonhosted.org/packages/5c/05/5cbb59154b093548acd0f4c7c474a118eda06da25aa75c616b72d8fcd92a/fastapi-0.128.0-py3-none-any.whl", hash = "sha256:aebd93f9716ee3b4f4fcfe13ffb7cf308d99c9f3ab5622d8877441072561582d", size = 103094, upload-time = "2025-12-27T15:21:12.154Z" },
 ]
 
 [[package]]
@@ -2470,21 +2494,24 @@ wheels = [
 
 [[package]]
 name = "mcp"
-version = "1.6.0"
+version = "1.12.4"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "anyio" },
     { name = "httpx" },
     { name = "httpx-sse" },
+    { name = "jsonschema" },
     { name = "pydantic" },
     { name = "pydantic-settings" },
+    { name = "python-multipart" },
+    { name = "pywin32", marker = "sys_platform == 'win32'" },
     { name = "sse-starlette" },
     { name = "starlette" },
-    { name = "uvicorn" },
+    { name = "uvicorn", marker = "sys_platform != 'emscripten'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/95/d2/f587cb965a56e992634bebc8611c5b579af912b74e04eb9164bd49527d21/mcp-1.6.0.tar.gz", hash = "sha256:d9324876de2c5637369f43161cd71eebfd803df5a95e46225cab8d280e366723", size = 200031, upload-time = "2025-03-27T16:46:32.336Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/31/88/f6cb7e7c260cd4b4ce375f2b1614b33ce401f63af0f49f7141a2e9bf0a45/mcp-1.12.4.tar.gz", hash = "sha256:0765585e9a3a5916a3c3ab8659330e493adc7bd8b2ca6120c2d7a0c43e034ca5", size = 431148, upload-time = "2025-08-07T20:31:18.082Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/10/30/20a7f33b0b884a9d14dd3aa94ff1ac9da1479fe2ad66dd9e2736075d2506/mcp-1.6.0-py3-none-any.whl", hash = "sha256:7bd24c6ea042dbec44c754f100984d186620d8b841ec30f1b19eda9b93a634d0", size = 76077, upload-time = "2025-03-27T16:46:29.919Z" },
+    { url = "https://files.pythonhosted.org/packages/ad/68/316cbc54b7163fa22571dcf42c9cc46562aae0a021b974e0a8141e897200/mcp-1.12.4-py3-none-any.whl", hash = "sha256:7aa884648969fab8e78b89399d59a683202972e12e6bc9a1c88ce7eda7743789", size = 160145, upload-time = "2025-08-07T20:31:15.69Z" },
 ]
 
 [[package]]
@@ -2958,19 +2985,6 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/1d/5a/3f8d078dbf55d18442f6a2ecedf6786d81d7245844b2b20ce2b8ad6f0307/opentelemetry_api-1.35.0-py3-none-any.whl", hash = "sha256:c4ea7e258a244858daf18474625e9cc0149b8ee354f37843415771a40c25ee06", size = 65566, upload-time = "2025-07-11T12:23:07.944Z" },
 ]
 
-[[package]]
-name = "opentelemetry-exporter-otlp"
-version = "1.35.0"
-source = { registry = "https://pypi.org/simple" }
-dependencies = [
-    { name = "opentelemetry-exporter-otlp-proto-grpc" },
-    { name = "opentelemetry-exporter-otlp-proto-http" },
-]
-sdist = { url = "https://files.pythonhosted.org/packages/2e/2e/63718faa67b17f449a7fb7efdc7125a408cbe5d8c0bb35f423f2776d60b5/opentelemetry_exporter_otlp-1.35.0.tar.gz", hash = "sha256:f94feff09b3524df867c7876b79c96cef20068106cb5efe55340e8d08192c8a4", size = 6142, upload-time = "2025-07-11T12:23:30.128Z" }
-wheels = [
-    { url = "https://files.pythonhosted.org/packages/cf/db/2da28358d3101ca936c1643becbb4ebd69e9e48acf27f153d735a4813c6b/opentelemetry_exporter_otlp-1.35.0-py3-none-any.whl", hash = "sha256:8e6bb9025f6238db7d69bba7ee37c77e4858d0a1ff22a9e126f7c9e017e83afe", size = 7016, upload-time = "2025-07-11T12:23:10.679Z" },
-]
-
 [[package]]
 name = "opentelemetry-exporter-otlp-proto-common"
 version = "1.35.0"
@@ -3977,6 +3991,9 @@ dependencies = [
 ]
 
 [package.optional-dependencies]
+acp = [
+    { name = "agent-client-protocol" },
+]
 agentops = [
     { name = "agentops" },
 ]
@@ -4073,6 +4090,7 @@ ui = [
 
 [package.metadata]
 requires-dist = [
+    { name = "agent-client-protocol", marker = "extra == 'acp'", specifier = ">=0.7.0" },
     { name = "agentops", marker = "extra == 'agentops'", specifier = ">=0.3.12" },
     { name = "aiosqlite", marker = "extra == 'chat'", specifier = ">=0.20.0" },
     { name = "aiosqlite", marker = "extra == 'code'", specifier = ">=0.20.0" },
@@ -4081,10 +4099,10 @@ requires-dist = [
     { name = "autogen-agentchat", marker = "extra == 'autogen-v4'", specifier = ">=0.4.0" },
     { name = "autogen-core", marker = "extra == 'autogen-v4'", specifier = ">=0.4.0" },
     { name = "autogen-ext", extras = ["openai"], marker = "extra == 'autogen-v4'", specifier = ">=0.4.0" },
-    { name = "chainlit", marker = "extra == 'chat'", specifier = "==2.5.5" },
-    { name = "chainlit", marker = "extra == 'code'", specifier = "==2.5.5" },
-    { name = "chainlit", marker = "extra == 'realtime'", specifier = "==2.5.5" },
-    { name = "chainlit", marker = "extra == 'ui'", specifier = "==2.5.5" },
+    { name = "chainlit", marker = "extra == 'chat'", specifier = ">=2.8.5,<2.9.0" },
+    { name = "chainlit", marker = "extra == 'code'", specifier = ">=2.8.5,<2.9.0" },
+    { name = "chainlit", marker = "extra == 'realtime'", specifier = ">=2.8.5,<2.9.0" },
+    { name = "chainlit", marker = "extra == 'ui'", specifier = ">=2.8.5,<2.9.0" },
     { name = "crawl4ai", marker = "extra == 'chat'", specifier = ">=0.7.0" },
     { name = "crawl4ai", marker = "extra == 'code'", specifier = ">=0.7.0" },
     { name = "crawl4ai", marker = "extra == 'realtime'", specifier = ">=0.7.0" },
@@ -4145,7 +4163,7 @@ requires-dist = [
     { name = "websockets", marker = "extra == 'realtime'", specifier = ">=12.0" },
     { name = "yfinance", marker = "extra == 'realtime'", specifier = ">=0.2.44" },
 ]
-provides-extras = ["ui", "gradio", "api", "agentops", "google", "openai", "anthropic", "cohere", "chat", "code", "realtime", "call", "train", "crewai", "autogen", "autogen-v4"]
+provides-extras = ["ui", "gradio", "api", "agentops", "google", "openai", "anthropic", "cohere", "chat", "code", "realtime", "call", "train", "crewai", "autogen", "autogen-v4", "acp"]
 
 [[package]]
 name = "praisonai-tools"
@@ -4477,15 +4495,16 @@ wheels = [
 
 [[package]]
 name = "pydantic-settings"
-version = "2.8.1"
+version = "2.12.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "pydantic" },
     { name = "python-dotenv" },
+    { name = "typing-inspection" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/88/82/c79424d7d8c29b994fb01d277da57b0a9b09cc03c3ff875f9bd8a86b2145/pydantic_settings-2.8.1.tar.gz", hash = "sha256:d5c663dfbe9db9d5e1c646b2e161da12f0d734d422ee56f567d0ea2cee4e8585", size = 83550, upload-time = "2025-02-27T10:10:32.338Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/43/4b/ac7e0aae12027748076d72a8764ff1c9d82ca75a7a52622e67ed3f765c54/pydantic_settings-2.12.0.tar.gz", hash = "sha256:005538ef951e3c2a68e1c08b292b5f2e71490def8589d4221b95dab00dafcfd0", size = 194184, upload-time = "2025-11-10T14:25:47.013Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/0b/53/a64f03044927dc47aafe029c42a5b7aabc38dfb813475e0e1bf71c4a59d0/pydantic_settings-2.8.1-py3-none-any.whl", hash = "sha256:81942d5ac3d905f7f3ee1a70df5dfb62d5569c12f51a5a647defc1c3d9ee2e9c", size = 30839, upload-time = "2025-02-27T10:10:30.711Z" },
+    { url = "https://files.pythonhosted.org/packages/c1/60/5d4751ba3f4a40a6891f24eec885f51afd78d208498268c734e256fb13c4/pydantic_settings-2.12.0-py3-none-any.whl", hash = "sha256:fddb9fd99a5b18da837b29710391e945b1e30c135477f484084ee513adb93809", size = 51880, upload-time = "2025-11-10T14:25:45.546Z" },
 ]
 
 [[package]]
@@ -4691,21 +4710,24 @@ wheels = [
 
 [[package]]
 name = "pywin32"
-version = "308"
+version = "311"
 source = { registry = "https://pypi.org/simple" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/72/a6/3e9f2c474895c1bb61b11fa9640be00067b5c5b363c501ee9c3fa53aec01/pywin32-308-cp310-cp310-win32.whl", hash = "sha256:796ff4426437896550d2981b9c2ac0ffd75238ad9ea2d3bfa67a1abd546d262e", size = 5927028, upload-time = "2024-10-12T20:41:58.898Z" },
-    { url = "https://files.pythonhosted.org/packages/d9/b4/84e2463422f869b4b718f79eb7530a4c1693e96b8a4e5e968de38be4d2ba/pywin32-308-cp310-cp310-win_amd64.whl", hash = "sha256:4fc888c59b3c0bef905ce7eb7e2106a07712015ea1c8234b703a088d46110e8e", size = 6558484, upload-time = "2024-10-12T20:42:01.271Z" },
-    { url = "https://files.pythonhosted.org/packages/9f/8f/fb84ab789713f7c6feacaa08dad3ec8105b88ade8d1c4f0f0dfcaaa017d6/pywin32-308-cp310-cp310-win_arm64.whl", hash = "sha256:a5ab5381813b40f264fa3495b98af850098f814a25a63589a8e9eb12560f450c", size = 7971454, upload-time = "2024-10-12T20:42:03.544Z" },
-    { url = "https://files.pythonhosted.org/packages/eb/e2/02652007469263fe1466e98439831d65d4ca80ea1a2df29abecedf7e47b7/pywin32-308-cp311-cp311-win32.whl", hash = "sha256:5d8c8015b24a7d6855b1550d8e660d8daa09983c80e5daf89a273e5c6fb5095a", size = 5928156, upload-time = "2024-10-12T20:42:05.78Z" },
-    { url = "https://files.pythonhosted.org/packages/48/ef/f4fb45e2196bc7ffe09cad0542d9aff66b0e33f6c0954b43e49c33cad7bd/pywin32-308-cp311-cp311-win_amd64.whl", hash = "sha256:575621b90f0dc2695fec346b2d6302faebd4f0f45c05ea29404cefe35d89442b", size = 6559559, upload-time = "2024-10-12T20:42:07.644Z" },
-    { url = "https://files.pythonhosted.org/packages/79/ef/68bb6aa865c5c9b11a35771329e95917b5559845bd75b65549407f9fc6b4/pywin32-308-cp311-cp311-win_arm64.whl", hash = "sha256:100a5442b7332070983c4cd03f2e906a5648a5104b8a7f50175f7906efd16bb6", size = 7972495, upload-time = "2024-10-12T20:42:09.803Z" },
-    { url = "https://files.pythonhosted.org/packages/00/7c/d00d6bdd96de4344e06c4afbf218bc86b54436a94c01c71a8701f613aa56/pywin32-308-cp312-cp312-win32.whl", hash = "sha256:587f3e19696f4bf96fde9d8a57cec74a57021ad5f204c9e627e15c33ff568897", size = 5939729, upload-time = "2024-10-12T20:42:12.001Z" },
-    { url = "https://files.pythonhosted.org/packages/21/27/0c8811fbc3ca188f93b5354e7c286eb91f80a53afa4e11007ef661afa746/pywin32-308-cp312-cp312-win_amd64.whl", hash = "sha256:00b3e11ef09ede56c6a43c71f2d31857cf7c54b0ab6e78ac659497abd2834f47", size = 6543015, upload-time = "2024-10-12T20:42:14.044Z" },
-    { url = "https://files.pythonhosted.org/packages/9d/0f/d40f8373608caed2255781a3ad9a51d03a594a1248cd632d6a298daca693/pywin32-308-cp312-cp312-win_arm64.whl", hash = "sha256:9b4de86c8d909aed15b7011182c8cab38c8850de36e6afb1f0db22b8959e3091", size = 7976033, upload-time = "2024-10-12T20:42:16.215Z" },
-    { url = "https://files.pythonhosted.org/packages/a9/a4/aa562d8935e3df5e49c161b427a3a2efad2ed4e9cf81c3de636f1fdddfd0/pywin32-308-cp313-cp313-win32.whl", hash = "sha256:1c44539a37a5b7b21d02ab34e6a4d314e0788f1690d65b48e9b0b89f31abbbed", size = 5938579, upload-time = "2024-10-12T20:42:18.623Z" },
-    { url = "https://files.pythonhosted.org/packages/c7/50/b0efb8bb66210da67a53ab95fd7a98826a97ee21f1d22949863e6d588b22/pywin32-308-cp313-cp313-win_amd64.whl", hash = "sha256:fd380990e792eaf6827fcb7e187b2b4b1cede0585e3d0c9e84201ec27b9905e4", size = 6542056, upload-time = "2024-10-12T20:42:20.864Z" },
-    { url = "https://files.pythonhosted.org/packages/26/df/2b63e3e4f2df0224f8aaf6d131f54fe4e8c96400eb9df563e2aae2e1a1f9/pywin32-308-cp313-cp313-win_arm64.whl", hash = "sha256:ef313c46d4c18dfb82a2431e3051ac8f112ccee1a34f29c263c583c568db63cd", size = 7974986, upload-time = "2024-10-12T20:42:22.799Z" },
+    { url = "https://files.pythonhosted.org/packages/7b/40/44efbb0dfbd33aca6a6483191dae0716070ed99e2ecb0c53683f400a0b4f/pywin32-311-cp310-cp310-win32.whl", hash = "sha256:d03ff496d2a0cd4a5893504789d4a15399133fe82517455e78bad62efbb7f0a3", size = 8760432, upload-time = "2025-07-14T20:13:05.9Z" },
+    { url = "https://files.pythonhosted.org/packages/5e/bf/360243b1e953bd254a82f12653974be395ba880e7ec23e3731d9f73921cc/pywin32-311-cp310-cp310-win_amd64.whl", hash = "sha256:797c2772017851984b97180b0bebe4b620bb86328e8a884bb626156295a63b3b", size = 9590103, upload-time = "2025-07-14T20:13:07.698Z" },
+    { url = "https://files.pythonhosted.org/packages/57/38/d290720e6f138086fb3d5ffe0b6caa019a791dd57866940c82e4eeaf2012/pywin32-311-cp310-cp310-win_arm64.whl", hash = "sha256:0502d1facf1fed4839a9a51ccbcc63d952cf318f78ffc00a7e78528ac27d7a2b", size = 8778557, upload-time = "2025-07-14T20:13:11.11Z" },
+    { url = "https://files.pythonhosted.org/packages/7c/af/449a6a91e5d6db51420875c54f6aff7c97a86a3b13a0b4f1a5c13b988de3/pywin32-311-cp311-cp311-win32.whl", hash = "sha256:184eb5e436dea364dcd3d2316d577d625c0351bf237c4e9a5fabbcfa5a58b151", size = 8697031, upload-time = "2025-07-14T20:13:13.266Z" },
+    { url = "https://files.pythonhosted.org/packages/51/8f/9bb81dd5bb77d22243d33c8397f09377056d5c687aa6d4042bea7fbf8364/pywin32-311-cp311-cp311-win_amd64.whl", hash = "sha256:3ce80b34b22b17ccbd937a6e78e7225d80c52f5ab9940fe0506a1a16f3dab503", size = 9508308, upload-time = "2025-07-14T20:13:15.147Z" },
+    { url = "https://files.pythonhosted.org/packages/44/7b/9c2ab54f74a138c491aba1b1cd0795ba61f144c711daea84a88b63dc0f6c/pywin32-311-cp311-cp311-win_arm64.whl", hash = "sha256:a733f1388e1a842abb67ffa8e7aad0e70ac519e09b0f6a784e65a136ec7cefd2", size = 8703930, upload-time = "2025-07-14T20:13:16.945Z" },
+    { url = "https://files.pythonhosted.org/packages/e7/ab/01ea1943d4eba0f850c3c61e78e8dd59757ff815ff3ccd0a84de5f541f42/pywin32-311-cp312-cp312-win32.whl", hash = "sha256:750ec6e621af2b948540032557b10a2d43b0cee2ae9758c54154d711cc852d31", size = 8706543, upload-time = "2025-07-14T20:13:20.765Z" },
+    { url = "https://files.pythonhosted.org/packages/d1/a8/a0e8d07d4d051ec7502cd58b291ec98dcc0c3fff027caad0470b72cfcc2f/pywin32-311-cp312-cp312-win_amd64.whl", hash = "sha256:b8c095edad5c211ff31c05223658e71bf7116daa0ecf3ad85f3201ea3190d067", size = 9495040, upload-time = "2025-07-14T20:13:22.543Z" },
+    { url = "https://files.pythonhosted.org/packages/ba/3a/2ae996277b4b50f17d61f0603efd8253cb2d79cc7ae159468007b586396d/pywin32-311-cp312-cp312-win_arm64.whl", hash = "sha256:e286f46a9a39c4a18b319c28f59b61de793654af2f395c102b4f819e584b5852", size = 8710102, upload-time = "2025-07-14T20:13:24.682Z" },
+    { url = "https://files.pythonhosted.org/packages/a5/be/3fd5de0979fcb3994bfee0d65ed8ca9506a8a1260651b86174f6a86f52b3/pywin32-311-cp313-cp313-win32.whl", hash = "sha256:f95ba5a847cba10dd8c4d8fefa9f2a6cf283b8b88ed6178fa8a6c1ab16054d0d", size = 8705700, upload-time = "2025-07-14T20:13:26.471Z" },
+    { url = "https://files.pythonhosted.org/packages/e3/28/e0a1909523c6890208295a29e05c2adb2126364e289826c0a8bc7297bd5c/pywin32-311-cp313-cp313-win_amd64.whl", hash = "sha256:718a38f7e5b058e76aee1c56ddd06908116d35147e133427e59a3983f703a20d", size = 9494700, upload-time = "2025-07-14T20:13:28.243Z" },
+    { url = "https://files.pythonhosted.org/packages/04/bf/90339ac0f55726dce7d794e6d79a18a91265bdf3aa70b6b9ca52f35e022a/pywin32-311-cp313-cp313-win_arm64.whl", hash = "sha256:7b4075d959648406202d92a2310cb990fea19b535c7f4a78d3f5e10b926eeb8a", size = 8709318, upload-time = "2025-07-14T20:13:30.348Z" },
+    { url = "https://files.pythonhosted.org/packages/c9/31/097f2e132c4f16d99a22bfb777e0fd88bd8e1c634304e102f313af69ace5/pywin32-311-cp314-cp314-win32.whl", hash = "sha256:b7a2c10b93f8986666d0c803ee19b5990885872a7de910fc460f9b0c2fbf92ee", size = 8840714, upload-time = "2025-07-14T20:13:32.449Z" },
+    { url = "https://files.pythonhosted.org/packages/90/4b/07c77d8ba0e01349358082713400435347df8426208171ce297da32c313d/pywin32-311-cp314-cp314-win_amd64.whl", hash = "sha256:3aca44c046bd2ed8c90de9cb8427f581c479e594e99b5c0bb19b29c10fd6cb87", size = 9656800, upload-time = "2025-07-14T20:13:34.312Z" },
+    { url = "https://files.pythonhosted.org/packages/c0/d2/21af5c535501a7233e734b8af901574572da66fcc254cb35d0609c9080dd/pywin32-311-cp314-cp314-win_arm64.whl", hash = "sha256:a508e2d9025764a8270f93111a970e1d0fbfc33f4153b388bb649b7eec4f9b42", size = 8932540, upload-time = "2025-07-14T20:13:36.379Z" },
 ]
 
 [[package]]
@@ -5431,14 +5453,15 @@ wheels = [
 
 [[package]]
 name = "starlette"
-version = "0.41.3"
+version = "0.50.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "anyio" },
+    { name = "typing-extensions", marker = "python_full_version < '3.13'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/1a/4c/9b5764bd22eec91c4039ef4c55334e9187085da2d8a2df7bd570869aae18/starlette-0.41.3.tar.gz", hash = "sha256:0e4ab3d16522a255be6b28260b938eae2482f98ce5cc934cb08dce8dc3ba5835", size = 2574159, upload-time = "2024-11-18T19:45:04.283Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/ba/b8/73a0e6a6e079a9d9cfa64113d771e421640b6f679a52eeb9b32f72d871a1/starlette-0.50.0.tar.gz", hash = "sha256:a2a17b22203254bcbc2e1f926d2d55f3f9497f769416b3190768befe598fa3ca", size = 2646985, upload-time = "2025-11-01T15:25:27.516Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/96/00/2b325970b3060c7cecebab6d295afe763365822b1306a12eeab198f74323/starlette-0.41.3-py3-none-any.whl", hash = "sha256:44cedb2b7c77a9de33a8b74b2b90e9f50d11fcf25d8270ea525ad71a25374ff7", size = 73225, upload-time = "2024-11-18T19:45:02.027Z" },
+    { url = "https://files.pythonhosted.org/packages/d9/52/1064f510b141bd54025f9b55105e26d1fa970b9be67ad766380a3c9b74b0/starlette-0.50.0-py3-none-any.whl", hash = "sha256:9e5391843ec9b6e472eed1365a78c8098cfceb7a74bfd4d6b1c0c0095efb3bca", size = 74033, upload-time = "2025-11-01T15:25:25.461Z" },
 ]
 
 [[package]]
@@ -5905,6 +5928,18 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/65/f3/107a22063bf27bdccf2024833d3445f4eea42b2e598abfbd46f6a63b6cb0/typing_inspect-0.9.0-py3-none-any.whl", hash = "sha256:9ee6fc59062311ef8547596ab6b955e1b8aa46242d854bfc78f4f6b0eff35f9f", size = 8827, upload-time = "2023-05-24T20:25:45.287Z" },
 ]
 
+[[package]]
+name = "typing-inspection"
+version = "0.4.2"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "typing-extensions" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/55/e3/70399cb7dd41c10ac53367ae42139cf4b1ca5f36bb3dc6c9d33acdb43655/typing_inspection-0.4.2.tar.gz", hash = "sha256:ba561c48a67c5958007083d386c3295464928b01faa735ab8547c5692e87f464", size = 75949, upload-time = "2025-10-01T02:14:41.687Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/dc/9b/47798a6c91d8bdb567fe2698fe81e0c6b7cb7ef4d13da4114b41d239f65d/typing_inspection-0.4.2-py3-none-any.whl", hash = "sha256:4ed1cacbdc298c220f1bd249ed5287caa16f34d44ef4e9c3d0cbad5b521545e7", size = 14611, upload-time = "2025-10-01T02:14:40.154Z" },
+]
+
 [[package]]
 name = "tzdata"
 version = "2024.2"
@@ -5914,21 +5949,6 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/a6/ab/7e5f53c3b9d14972843a647d8d7a853969a58aecc7559cb3267302c94774/tzdata-2024.2-py2.py3-none-any.whl", hash = "sha256:a48093786cdcde33cad18c2555e8532f34422074448fbc874186f0abd79565cd", size = 346586, upload-time = "2024-09-23T18:56:45.478Z" },
 ]
 
-[[package]]
-name = "uptrace"
-version = "1.35.0"
-source = { registry = "https://pypi.org/simple" }
-dependencies = [
-    { name = "opentelemetry-api" },
-    { name = "opentelemetry-exporter-otlp" },
-    { name = "opentelemetry-instrumentation" },
-    { name = "opentelemetry-sdk" },
-]
-sdist = { url = "https://files.pythonhosted.org/packages/1b/e7/090a8b7404b1fcf74c22aed25bb04601740bd1707e31579095b166fa2b19/uptrace-1.35.0.tar.gz", hash = "sha256:0897ea40fcc22122a11d05a0bbcacffc55d2a9a18dd03cffd6c5625e5afa55f0", size = 7660, upload-time = "2025-07-14T11:37:52.142Z" }
-wheels = [
-    { url = "https://files.pythonhosted.org/packages/b3/db/73bf119afdadbbc8578a61ce41117cda05f8703eaa24163e9236b96ccc40/uptrace-1.35.0-py3-none-any.whl", hash = "sha256:c98ce900d3c660c99763a684f5a51f876d93660114d9ed51de676ed886e5106a", size = 8590, upload-time = "2025-07-14T11:37:51.252Z" },
-]
-
 [[package]]
 name = "urllib3"
 version = "2.3.0"
@@ -5965,16 +5985,16 @@ wheels = [
 
 [[package]]
 name = "uvicorn"
-version = "0.34.2"
+version = "0.40.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "click" },
     { name = "h11" },
     { name = "typing-extensions", marker = "python_full_version < '3.11'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/a6/ae/9bbb19b9e1c450cf9ecaef06463e40234d98d95bf572fab11b4f19ae5ded/uvicorn-0.34.2.tar.gz", hash = "sha256:0e929828f6186353a80b58ea719861d2629d766293b6d19baf086ba31d4f3328", size = 76815, upload-time = "2025-04-19T06:02:50.101Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/c3/d1/8f3c683c9561a4e6689dd3b1d345c815f10f86acd044ee1fb9a4dcd0b8c5/uvicorn-0.40.0.tar.gz", hash = "sha256:839676675e87e73694518b5574fd0f24c9d97b46bea16df7b8c05ea1a51071ea", size = 81761, upload-time = "2025-12-21T14:16:22.45Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/b1/4b/4cef6ce21a2aaca9d852a6e84ef4f135d99fcd74fa75105e2fc0c8308acd/uvicorn-0.34.2-py3-none-any.whl", hash = "sha256:deb49af569084536d269fe0a6d67e3754f104cf03aba7c11c40f01aadf33c403", size = 62483, upload-time = "2025-04-19T06:02:48.42Z" },
+    { url = "https://files.pythonhosted.org/packages/3d/d8/2083a1daa7439a66f3a48589a57d576aa117726762618f6bb09fe3798796/uvicorn-0.40.0-py3-none-any.whl", hash = "sha256:c6c8f55bc8bf13eb6fa9ff87ad62308bbbc33d0b67f84293151efe87e0d5f2ee", size = 68502, upload-time = "2025-12-21T14:16:21.041Z" },
 ]
 
 [package.optional-dependencies]