Compare commits

...
7 Commits
Author SHA1 Message Date
Mervin Praison d8d84ee3e3 Release v2.9.0 2025-12-31 15:49:07 +00:00
Mervin Praison 9f2c4d10b5 Release v2.8.9 2025-12-31 14:05:30 +00:00
Mervin Praison 42a86c831f Release v2.8.8 2025-12-31 12:46:10 +00:00
Mervin Praison 48443b11d6 Release v2.8.7 2025-12-31 12:28:44 +00:00
Mervin Praison 71745aa14a Release v2.8.6 2025-12-31 12:27:20 +00:00
Mervin Praison 4f309ec16d Release v2.8.5 2025-12-31 12:00:29 +00:00
Mervin Praison e48afc8993 Release v2.8.4 2025-12-31 11:39:37 +00:00
44 changed files with 7070 additions and 126 deletions

No files matched your search

+71
View File
@@ -0,0 +1,71 @@
"""
Auto-generated API server for PraisonAI agents.
"""
from flask import Flask, request, jsonify
from flask_cors import CORS
from praisonai import PraisonAI
import os
app = Flask(__name__)
# CORS configuration
CORS(app)
# Authentication
AUTH_ENABLED = False
AUTH_TOKEN = None
def check_auth():
"""Check authentication if enabled."""
if not AUTH_ENABLED:
return True
token = request.headers.get('Authorization', '').replace('Bearer ', '')
return token == AUTH_TOKEN
@app.route('/health', methods=['GET'])
def health():
"""Health check endpoint."""
return jsonify({"status": "ok", "service": "praisonai-api"})
@app.route('/chat', methods=['POST'])
def chat():
"""Chat endpoint for agent interaction."""
if not check_auth():
return jsonify({"error": "Unauthorized"}), 401
data = request.get_json()
if not data or 'message' not in data:
return jsonify({"error": "Message required"}), 400
try:
praisonai = PraisonAI(agent_file="agents.yaml")
result = praisonai.run()
return jsonify({
"response": result,
"status": "success"
})
except Exception as e:
return jsonify({
"error": str(e),
"status": "error"
}), 500
@app.route('/agents', methods=['GET'])
def list_agents():
"""List available agents."""
if not check_auth():
return jsonify({"error": "Unauthorized"}), 401
return jsonify({
"agents": ["default"],
"agent_file": "agents.yaml"
})
if __name__ == '__main__':
app.run(
host='0.0.0.0',
port=8080,
debug=False
)
+1 -1
View File
@@ -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.9.0" \
"praisonai[chat]" \
"embedchain[github,youtube]"
+1 -1
View File
@@ -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.9.0" \
"praisonai[ui]" \
"praisonai[chat]" \
"praisonai[realtime]" \
+1 -1
View File
@@ -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.9.0" \
"praisonai[ui]" \
"praisonai[crewai]"
+137
View File
@@ -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
+27
View File
@@ -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
)
+45
View File
@@ -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",
)
+228
View File
@@ -0,0 +1,228 @@
"""
Debug Interactive Mode Example
This example demonstrates how to use the debug CLI commands to test
the Interactive Coding Assistant without entering TUI mode.
Features demonstrated:
- LSP symbol listing
- LSP definition lookup
- ACP action planning
- ACP action execution
- Trace capture and JSON output
"""
import asyncio
import json
import sys
from pathlib import Path
# Add the src directory to path for development
sys.path.insert(0, str(Path(__file__).parent.parent.parent / "src" / "praisonai"))
from praisonai.cli.features.interactive_runtime import create_runtime
from praisonai.cli.features.code_intelligence import CodeIntelligenceRouter
from praisonai.cli.features.action_orchestrator import ActionOrchestrator
async def demo_lsp_queries(workspace: str):
"""Demonstrate LSP-based code queries."""
print("\n" + "=" * 60)
print("LSP CODE QUERIES DEMO")
print("=" * 60)
runtime = create_runtime(
workspace=workspace,
lsp=True,
acp=False,
trace=True
)
try:
await runtime.start()
router = CodeIntelligenceRouter(runtime)
# Query 1: List symbols
print("\n[Query 1] List all symbols in app.py")
result = await router.handle_query("list all functions in app.py", "app.py")
print(f" Success: {result.success}")
print(f" LSP Used: {result.lsp_used}")
print(f" Fallback Used: {result.fallback_used}")
if result.data:
print(f" Symbols found: {len(result.data)}")
for symbol in result.data[:5]:
print(f" - {symbol['name']} ({symbol['kind']}) at line {symbol['line']}")
# Query 2: Find definition
print("\n[Query 2] Find definition of Calculator")
result = await router.handle_query("where is Calculator defined", "app.py")
print(f" Success: {result.success}")
if result.citations:
print(f" Found at: {result.citations[0]}")
# Get trace
trace = runtime.get_trace()
print(f"\n[Trace] {len(trace.entries)} entries captured")
finally:
await runtime.stop()
async def demo_acp_actions(workspace: str):
"""Demonstrate ACP-based code modifications."""
print("\n" + "=" * 60)
print("ACP CODE MODIFICATIONS DEMO")
print("=" * 60)
runtime = create_runtime(
workspace=workspace,
lsp=False,
acp=True,
approval="auto",
trace=True
)
try:
await runtime.start()
print(f"\n ACP Ready: {runtime.acp_ready}")
print(f" Read Only: {runtime.read_only}")
orchestrator = ActionOrchestrator(runtime)
# Action 1: Create a plan
print("\n[Action 1] Create plan for new file")
result = await orchestrator.create_plan("create a new file called demo_output.py")
print(f" Success: {result.success}")
if result.plan:
print(f" Plan ID: {result.plan.id}")
print(f" Steps: {len(result.plan.steps)}")
for step in result.plan.steps:
print(f" - {step.action_type.value}: {step.target}")
# Action 2: Execute with auto-approve
print("\n[Action 2] Execute file creation")
exec_result = await orchestrator.execute(
"create a new file called demo_output.py with a hello function",
auto_approve=True
)
print(f" Success: {exec_result.success}")
print(f" Applied Steps: {exec_result.applied_steps}")
# Verify file was created
output_file = Path(workspace) / "demo_output.py"
if output_file.exists():
print(f" File created: {output_file}")
print(f" Content preview: {output_file.read_text()[:100]}...")
# Get trace
trace = runtime.get_trace()
print(f"\n[Trace] {len(trace.entries)} entries captured")
for entry in trace.entries:
print(f" [{entry.category}] {entry.action}")
finally:
await runtime.stop()
async def demo_full_interactive_turn(workspace: str):
"""Demonstrate full interactive turn simulation."""
print("\n" + "=" * 60)
print("FULL INTERACTIVE TURN SIMULATION")
print("=" * 60)
runtime = create_runtime(
workspace=workspace,
lsp=True,
acp=True,
approval="auto",
trace=True
)
try:
status = await runtime.start()
print("\n Runtime Status:")
print(f" LSP: {'ready' if status['lsp']['ready'] else 'fallback'}")
print(f" ACP: {'ready' if status['acp']['ready'] else 'unavailable'}")
router = CodeIntelligenceRouter(runtime)
orchestrator = ActionOrchestrator(runtime)
# Simulate user prompt: code query
prompt1 = "What functions are in app.py?"
print(f"\n[Turn 1] User: {prompt1}")
# Classify and route
result = await router.handle_query(prompt1, "app.py")
print(f" Intent: {result.intent.value}")
print(f" Response: Found {len(result.data) if result.data else 0} symbols")
# Simulate user prompt: code modification
prompt2 = "Create a new test file"
print(f"\n[Turn 2] User: {prompt2}")
exec_result = await orchestrator.execute(
"create a new file called test_demo.py",
auto_approve=True
)
print(" Action: file_create")
print(f" Result: {'success' if exec_result.success else 'failed'}")
# Export trace as JSON
trace = runtime.get_trace()
trace_json = trace.to_dict()
print("[Trace Export]")
print(json.dumps(trace_json, indent=2, default=str)[:500] + "...")
finally:
await runtime.stop()
def main():
"""Run the debug demo."""
# Create a temporary workspace
workspace = Path(__file__).parent / "_demo_workspace"
workspace.mkdir(exist_ok=True)
# Create a sample app.py
app_file = workspace / "app.py"
app_file.write_text('''"""Sample application."""
def greet(name: str) -> str:
"""Greet someone."""
return f"Hello, {name}!"
def add(a: int, b: int) -> int:
"""Add two numbers."""
return a + b
class Calculator:
"""Simple calculator."""
def __init__(self):
self.value = 0
def add(self, x):
self.value += x
return self.value
if __name__ == "__main__":
print(greet("World"))
''')
print("=" * 60)
print("PRAISONAI DEBUG MODE DEMO")
print("=" * 60)
print(f"Workspace: {workspace}")
# Run demos
asyncio.run(demo_lsp_queries(str(workspace)))
asyncio.run(demo_acp_actions(str(workspace)))
asyncio.run(demo_full_interactive_turn(str(workspace)))
print("\n" + "=" * 60)
print("DEMO COMPLETE")
print("=" * 60)
if __name__ == "__main__":
main()
+1
View File
@@ -0,0 +1 @@
chainlit.md
+137
View File
@@ -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
@@ -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
)
@@ -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",
)
+2 -2
View File
@@ -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.9.0.tar.gz"
sha256 `curl -sL https://github.com/MervinPraison/PraisonAI/archive/refs/tags/v2.9.0.tar.gz | shasum -a 256`.split.first
license "MIT"
depends_on "[email protected]"
+54
View File
@@ -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"
)
+159
View File
@@ -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,
}
+587
View File
@@ -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))
+219
View File
@@ -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
+236
View File
@@ -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())
@@ -0,0 +1,546 @@
"""
Action Orchestrator for PraisonAI.
Orchestrates code modifications through ACP with plan → approval → apply → verify steps.
Enforces read-only mode when ACP is unavailable.
"""
import logging
import time
from dataclasses import dataclass, field
from enum import Enum
from pathlib import Path
from typing import Any, Dict, List, Optional, TYPE_CHECKING
if TYPE_CHECKING:
from .interactive_runtime import InteractiveRuntime
logger = logging.getLogger(__name__)
class ActionType(Enum):
"""Types of actions that can be orchestrated."""
FILE_CREATE = "file_create"
FILE_EDIT = "file_edit"
FILE_DELETE = "file_delete"
FILE_RENAME = "file_rename"
SHELL_COMMAND = "shell_command"
REFACTOR = "refactor"
UNKNOWN = "unknown"
class ActionStatus(Enum):
"""Status of an action."""
PENDING = "pending"
APPROVED = "approved"
REJECTED = "rejected"
APPLIED = "applied"
FAILED = "failed"
VERIFIED = "verified"
@dataclass
class ActionStep:
"""A single step in an action plan."""
id: str
action_type: ActionType
description: str
target: str # File path or command
params: Dict[str, Any] = field(default_factory=dict)
status: ActionStatus = ActionStatus.PENDING
result: Optional[Any] = None
error: Optional[str] = None
@dataclass
class ActionPlan:
"""A plan of actions to execute."""
id: str
prompt: str
steps: List[ActionStep] = field(default_factory=list)
created_at: float = field(default_factory=time.time)
status: ActionStatus = ActionStatus.PENDING
approval_mode: str = "manual"
def to_dict(self) -> Dict[str, Any]:
return {
"id": self.id,
"prompt": self.prompt,
"steps": [
{
"id": s.id,
"action_type": s.action_type.value,
"description": s.description,
"target": s.target,
"params": s.params,
"status": s.status.value,
"result": s.result,
"error": s.error
}
for s in self.steps
],
"created_at": self.created_at,
"status": self.status.value,
"approval_mode": self.approval_mode
}
@dataclass
class ActionResult:
"""Result of action orchestration."""
success: bool
plan: Optional[ActionPlan] = None
applied_steps: int = 0
failed_steps: int = 0
error: Optional[str] = None
read_only_blocked: bool = False
diff_summary: Optional[str] = None
def to_dict(self) -> Dict[str, Any]:
return {
"success": self.success,
"plan": self.plan.to_dict() if self.plan else None,
"applied_steps": self.applied_steps,
"failed_steps": self.failed_steps,
"error": self.error,
"read_only_blocked": self.read_only_blocked,
"diff_summary": self.diff_summary
}
class ActionOrchestrator:
"""
Orchestrates code modifications through ACP.
All edits must go through:
1. Plan creation
2. Approval (manual, auto, or scoped)
3. Application
4. Verification
If ACP is unavailable, runtime is read-only and edits are blocked.
"""
def __init__(self, runtime: "InteractiveRuntime"):
"""Initialize with runtime reference."""
self.runtime = runtime
self._plan_counter = 0
def _generate_plan_id(self) -> str:
"""Generate a unique plan ID."""
self._plan_counter += 1
return f"plan_{int(time.time())}_{self._plan_counter}"
def _generate_step_id(self, plan_id: str, index: int) -> str:
"""Generate a unique step ID."""
return f"{plan_id}_step_{index}"
async def create_plan(self, prompt: str) -> ActionResult:
"""
Create an action plan for a modification request.
Args:
prompt: The user's modification request
Returns:
ActionResult with the plan
"""
if self.runtime.read_only:
return ActionResult(
success=False,
error="Runtime is in read-only mode. ACP is required for modifications.",
read_only_blocked=True
)
plan_id = self._generate_plan_id()
# Always use fallback plan creation (ACP session tracks but doesn't create plans)
# In production, this would integrate with an LLM for intelligent planning
return await self._create_fallback_plan(prompt, plan_id)
async def _create_fallback_plan(self, prompt: str, plan_id: str) -> ActionResult:
"""Create a fallback plan when ACP is not available for planning."""
# This is a simplified fallback - in production, this would use LLM
steps = []
# Analyze prompt for common patterns
prompt_lower = prompt.lower()
if "create" in prompt_lower or "new file" in prompt_lower:
# Extract file path if mentioned
import re
file_match = re.search(r'(?:create|new)\s+(?:a\s+)?(?:file\s+)?(?:called\s+)?([^\s]+\.\w+)', prompt_lower)
target = file_match.group(1) if file_match else "new_file.py"
# Generate default content based on file type
if target.endswith('.py'):
content = f'"""\n{target} - Auto-generated file\n"""\n\ndef main():\n pass\n\n\nif __name__ == "__main__":\n main()\n'
else:
content = f"// {target} - Auto-generated file\n"
steps.append(ActionStep(
id=self._generate_step_id(plan_id, 0),
action_type=ActionType.FILE_CREATE,
description=f"Create file: {target}",
target=target,
params={"content": content},
status=ActionStatus.APPROVED # Auto-approve for fallback
))
elif "edit" in prompt_lower or "modify" in prompt_lower or "change" in prompt_lower:
file_match = re.search(r'(?:edit|modify|change)\s+([^\s]+\.\w+)', prompt_lower)
target = file_match.group(1) if file_match else ""
if target:
steps.append(ActionStep(
id=self._generate_step_id(plan_id, 0),
action_type=ActionType.FILE_EDIT,
description=f"Edit file: {target}",
target=target,
params={"changes": prompt}
))
elif "delete" in prompt_lower or "remove" in prompt_lower:
file_match = re.search(r'(?:delete|remove)\s+([^\s]+\.\w+)', prompt_lower)
target = file_match.group(1) if file_match else ""
if target:
steps.append(ActionStep(
id=self._generate_step_id(plan_id, 0),
action_type=ActionType.FILE_DELETE,
description=f"Delete file: {target}",
target=target
))
elif "rename" in prompt_lower:
steps.append(ActionStep(
id=self._generate_step_id(plan_id, 0),
action_type=ActionType.REFACTOR,
description="Rename operation",
target="",
params={"prompt": prompt}
))
elif "run" in prompt_lower or "execute" in prompt_lower:
cmd_match = re.search(r'(?:run|execute)\s+[`"\']?([^`"\']+)[`"\']?', prompt_lower)
command = cmd_match.group(1) if cmd_match else ""
if command:
steps.append(ActionStep(
id=self._generate_step_id(plan_id, 0),
action_type=ActionType.SHELL_COMMAND,
description=f"Execute: {command}",
target=command
))
if not steps:
return ActionResult(
success=False,
error="Could not determine action from prompt. Please be more specific."
)
plan = ActionPlan(
id=plan_id,
prompt=prompt,
steps=steps,
approval_mode=self.runtime.config.approval_mode
)
return ActionResult(
success=True,
plan=plan
)
def _classify_action_type(self, step_data: Dict[str, Any]) -> ActionType:
"""Classify the action type from step data."""
action = step_data.get("action", "").lower()
if "create" in action:
return ActionType.FILE_CREATE
elif "edit" in action or "modify" in action:
return ActionType.FILE_EDIT
elif "delete" in action or "remove" in action:
return ActionType.FILE_DELETE
elif "rename" in action:
return ActionType.FILE_RENAME
elif "shell" in action or "command" in action or "run" in action:
return ActionType.SHELL_COMMAND
elif "refactor" in action:
return ActionType.REFACTOR
return ActionType.UNKNOWN
async def approve_plan(self, plan: ActionPlan, auto: bool = False) -> bool:
"""
Request approval for a plan.
Args:
plan: The plan to approve
auto: Whether to auto-approve
Returns:
True if approved, False if rejected
"""
if self.runtime.read_only:
logger.warning("Cannot approve plan in read-only mode")
return False
approval_mode = self.runtime.config.approval_mode
if approval_mode == "auto" or auto:
plan.status = ActionStatus.APPROVED
for step in plan.steps:
step.status = ActionStatus.APPROVED
return True
elif approval_mode == "scoped":
# Auto-approve safe actions, require manual for dangerous ones
dangerous_types = {ActionType.FILE_DELETE, ActionType.SHELL_COMMAND}
for step in plan.steps:
if step.action_type in dangerous_types:
step.status = ActionStatus.PENDING # Needs manual approval
else:
step.status = ActionStatus.APPROVED
# Plan is approved if all steps are approved
if all(s.status == ActionStatus.APPROVED for s in plan.steps):
plan.status = ActionStatus.APPROVED
return True
return False # Some steps need manual approval
else: # manual
# In manual mode, we return False to indicate approval is needed
return False
async def apply_plan(self, plan: ActionPlan, force: bool = False) -> ActionResult:
"""
Apply an approved plan.
Args:
plan: The plan to apply
force: Force apply even if not fully approved
Returns:
ActionResult with application status
"""
if self.runtime.read_only:
return ActionResult(
success=False,
plan=plan,
error="Runtime is in read-only mode",
read_only_blocked=True
)
if plan.status != ActionStatus.APPROVED and not force:
return ActionResult(
success=False,
plan=plan,
error="Plan is not approved"
)
applied = 0
failed = 0
diff_parts = []
# Track in ACP session if available (but always do manual application)
if self.runtime.acp_ready:
try:
auto_approve = self.runtime.config.approval_mode == "auto"
await self.runtime.acp_apply_plan(
plan.to_dict(),
auto_approve=auto_approve
)
except Exception as e:
logger.warning(f"ACP tracking failed: {e}")
# Manual application (always execute)
for step in plan.steps:
if step.status not in [ActionStatus.APPROVED, ActionStatus.PENDING]:
continue
try:
result = await self._apply_step(step)
if result:
step.status = ActionStatus.APPLIED
step.result = result
applied += 1
if isinstance(result, dict) and "diff" in result:
diff_parts.append(result["diff"])
else:
step.status = ActionStatus.FAILED
failed += 1
except Exception as e:
step.status = ActionStatus.FAILED
step.error = str(e)
failed += 1
plan.status = ActionStatus.APPLIED if failed == 0 else ActionStatus.FAILED
return ActionResult(
success=failed == 0,
plan=plan,
applied_steps=applied,
failed_steps=failed,
diff_summary="\n".join(diff_parts) if diff_parts else None
)
async def _apply_step(self, step: ActionStep) -> Optional[Dict[str, Any]]:
"""Apply a single step."""
workspace = Path(self.runtime.config.workspace)
if step.action_type == ActionType.FILE_CREATE:
target = workspace / step.target
target.parent.mkdir(parents=True, exist_ok=True)
content = step.params.get("content", "")
target.write_text(content)
return {"created": str(target), "diff": f"+++ {target}\n+ (new file)"}
elif step.action_type == ActionType.FILE_EDIT:
target = workspace / step.target
if not target.exists():
return None
old_content = target.read_text()
new_content = step.params.get("new_content", old_content)
target.write_text(new_content)
return {
"edited": str(target),
"diff": f"--- {target}\n+++ {target}\n(content changed)"
}
elif step.action_type == ActionType.FILE_DELETE:
target = workspace / step.target
if target.exists():
target.unlink()
return {"deleted": str(target), "diff": f"--- {target}\n- (deleted)"}
return None
elif step.action_type == ActionType.FILE_RENAME:
source = workspace / step.params.get("source", "")
dest = workspace / step.params.get("dest", "")
if source.exists():
source.rename(dest)
return {"renamed": f"{source} -> {dest}"}
return None
elif step.action_type == ActionType.SHELL_COMMAND:
import subprocess
result = subprocess.run(
step.target,
shell=True,
capture_output=True,
text=True,
cwd=str(workspace),
timeout=30
)
return {
"command": step.target,
"stdout": result.stdout,
"stderr": result.stderr,
"returncode": result.returncode
}
return None
async def verify_plan(self, plan: ActionPlan) -> ActionResult:
"""
Verify that a plan was applied correctly.
Args:
plan: The plan to verify
Returns:
ActionResult with verification status
"""
if plan.status != ActionStatus.APPLIED:
return ActionResult(
success=False,
plan=plan,
error="Plan was not applied"
)
workspace = Path(self.runtime.config.workspace)
verified = 0
failed = 0
for step in plan.steps:
if step.status != ActionStatus.APPLIED:
continue
try:
if step.action_type == ActionType.FILE_CREATE:
target = workspace / step.target
if target.exists():
step.status = ActionStatus.VERIFIED
verified += 1
else:
step.status = ActionStatus.FAILED
step.error = "File was not created"
failed += 1
elif step.action_type == ActionType.FILE_DELETE:
target = workspace / step.target
if not target.exists():
step.status = ActionStatus.VERIFIED
verified += 1
else:
step.status = ActionStatus.FAILED
step.error = "File was not deleted"
failed += 1
else:
# For other types, assume verified if applied
step.status = ActionStatus.VERIFIED
verified += 1
except Exception as e:
step.status = ActionStatus.FAILED
step.error = str(e)
failed += 1
plan.status = ActionStatus.VERIFIED if failed == 0 else ActionStatus.FAILED
return ActionResult(
success=failed == 0,
plan=plan,
applied_steps=verified,
failed_steps=failed
)
async def execute(self, prompt: str, auto_approve: bool = False) -> ActionResult:
"""
Execute a full plan → approve → apply → verify cycle.
Args:
prompt: The modification request
auto_approve: Whether to auto-approve
Returns:
ActionResult with final status
"""
# Create plan
result = await self.create_plan(prompt)
if not result.success:
return result
plan = result.plan
# Approve
approved = await self.approve_plan(plan, auto=auto_approve)
if not approved and self.runtime.config.approval_mode == "manual":
return ActionResult(
success=False,
plan=plan,
error="Plan requires manual approval"
)
# Apply
result = await self.apply_plan(plan)
if not result.success:
return result
# Verify
return await self.verify_plan(plan)
@@ -0,0 +1,652 @@
"""
Code Intelligence Router for PraisonAI.
Routes code-related queries to LSP when available, with fallback to grep/file search.
"""
import logging
import re
from dataclasses import dataclass
from enum import Enum
from pathlib import Path
from typing import Any, Dict, List, Optional, TYPE_CHECKING
if TYPE_CHECKING:
from .interactive_runtime import InteractiveRuntime
logger = logging.getLogger(__name__)
class CodeIntent(Enum):
"""Classification of code-related intents."""
LIST_SYMBOLS = "list_symbols"
GO_TO_DEFINITION = "go_to_definition"
FIND_REFERENCES = "find_references"
GET_DIAGNOSTICS = "get_diagnostics"
EXPLAIN_CODE = "explain_code"
SEARCH_CODE = "search_code"
UNKNOWN = "unknown"
@dataclass
class CodeQueryResult:
"""Result of a code intelligence query."""
intent: CodeIntent
success: bool
lsp_used: bool
data: Any = None
citations: List[Dict[str, Any]] = None
error: Optional[str] = None
fallback_used: bool = False
def __post_init__(self):
if self.citations is None:
self.citations = []
def to_dict(self) -> Dict[str, Any]:
return {
"intent": self.intent.value,
"success": self.success,
"lsp_used": self.lsp_used,
"data": self.data,
"citations": self.citations,
"error": self.error,
"fallback_used": self.fallback_used
}
class CodeIntelligenceRouter:
"""
Routes code intelligence queries to LSP or fallback mechanisms.
LSP-first approach:
- If LSP is available, use it for semantic queries
- If LSP fails or unavailable, fall back to grep/file search
- Always provide citations with file:line references
"""
# Patterns for intent classification
INTENT_PATTERNS = {
CodeIntent.LIST_SYMBOLS: [
r"list\s+(all\s+)?(functions?|classes?|methods?|symbols?)",
r"what\s+(functions?|classes?|methods?)\s+(are|exist)",
r"show\s+(me\s+)?(all\s+)?(functions?|classes?|symbols?)",
r"find\s+(all\s+)?(functions?|classes?|methods?)",
],
CodeIntent.GO_TO_DEFINITION: [
r"(go\s+to|find|show|where\s+is)\s+(the\s+)?definition",
r"where\s+is\s+(\w+)\s+defined",
r"definition\s+of\s+(\w+)",
r"(\w+)\s+is\s+defined\s+where",
],
CodeIntent.FIND_REFERENCES: [
r"(find|show|list)\s+(all\s+)?references",
r"where\s+is\s+(\w+)\s+used",
r"usages?\s+of\s+(\w+)",
r"who\s+(calls?|uses?)\s+(\w+)",
],
CodeIntent.GET_DIAGNOSTICS: [
r"(show|list|get)\s+(all\s+)?(errors?|warnings?|diagnostics?|problems?)",
r"what('s|\s+is)\s+wrong",
r"any\s+(errors?|issues?|problems?)",
r"check\s+(for\s+)?(errors?|issues?)",
],
CodeIntent.EXPLAIN_CODE: [
r"explain\s+(this|the)\s+code",
r"what\s+does\s+(this|the)\s+code\s+do",
r"how\s+does\s+(\w+)\s+work",
],
CodeIntent.SEARCH_CODE: [
r"search\s+(for\s+)?",
r"find\s+(the\s+)?string",
r"grep\s+",
],
}
def __init__(self, runtime: "InteractiveRuntime"):
"""Initialize with runtime reference."""
self.runtime = runtime
def classify_intent(self, query: str) -> CodeIntent:
"""Classify the intent of a code query."""
query_lower = query.lower()
for intent, patterns in self.INTENT_PATTERNS.items():
for pattern in patterns:
if re.search(pattern, query_lower):
return intent
return CodeIntent.UNKNOWN
async def handle_query(self, query: str, file_path: str = None) -> CodeQueryResult:
"""
Handle a code intelligence query.
Args:
query: The user's query
file_path: Optional file path context
Returns:
CodeQueryResult with data and citations
"""
intent = self.classify_intent(query)
if intent == CodeIntent.LIST_SYMBOLS:
return await self._handle_list_symbols(query, file_path)
elif intent == CodeIntent.GO_TO_DEFINITION:
return await self._handle_go_to_definition(query, file_path)
elif intent == CodeIntent.FIND_REFERENCES:
return await self._handle_find_references(query, file_path)
elif intent == CodeIntent.GET_DIAGNOSTICS:
return await self._handle_get_diagnostics(query, file_path)
elif intent == CodeIntent.SEARCH_CODE:
return await self._handle_search_code(query, file_path)
else:
return CodeQueryResult(
intent=intent,
success=False,
lsp_used=False,
error="Could not determine code query intent"
)
async def _handle_list_symbols(self, query: str, file_path: str = None) -> CodeQueryResult:
"""Handle list symbols query."""
if not file_path:
# Try to find a file from the query
file_path = self._extract_file_from_query(query)
if not file_path:
return CodeQueryResult(
intent=CodeIntent.LIST_SYMBOLS,
success=False,
lsp_used=False,
error="No file specified. Please specify a file path."
)
# Try LSP first
if self.runtime.lsp_ready:
try:
symbols = await self.runtime.lsp_get_symbols(file_path)
if symbols:
citations = [
{"file": file_path, "type": "symbols", "count": len(symbols)}
]
return CodeQueryResult(
intent=CodeIntent.LIST_SYMBOLS,
success=True,
lsp_used=True,
data=symbols,
citations=citations
)
except Exception as e:
logger.warning(f"LSP symbols failed: {e}")
# Fallback to regex-based symbol extraction
return await self._fallback_list_symbols(file_path)
async def _fallback_list_symbols(self, file_path: str) -> CodeQueryResult:
"""Fallback symbol listing using regex."""
try:
path = Path(file_path)
if not path.exists():
path = Path(self.runtime.config.workspace) / file_path
if not path.exists():
return CodeQueryResult(
intent=CodeIntent.LIST_SYMBOLS,
success=False,
lsp_used=False,
fallback_used=True,
error=f"File not found: {file_path}"
)
content = path.read_text()
symbols = []
# Python patterns
if path.suffix == ".py":
# Functions
for match in re.finditer(r'^(async\s+)?def\s+(\w+)\s*\(', content, re.MULTILINE):
line_num = content[:match.start()].count('\n') + 1
symbols.append({
"name": match.group(2),
"kind": "function",
"line": line_num
})
# Classes
for match in re.finditer(r'^class\s+(\w+)\s*[:\(]', content, re.MULTILINE):
line_num = content[:match.start()].count('\n') + 1
symbols.append({
"name": match.group(1),
"kind": "class",
"line": line_num
})
# JavaScript/TypeScript patterns
elif path.suffix in [".js", ".ts", ".jsx", ".tsx"]:
# Functions
for match in re.finditer(r'(?:async\s+)?function\s+(\w+)\s*\(', content):
line_num = content[:match.start()].count('\n') + 1
symbols.append({
"name": match.group(1),
"kind": "function",
"line": line_num
})
# Arrow functions assigned to const/let/var
for match in re.finditer(r'(?:const|let|var)\s+(\w+)\s*=\s*(?:async\s+)?\(', content):
line_num = content[:match.start()].count('\n') + 1
symbols.append({
"name": match.group(1),
"kind": "function",
"line": line_num
})
# Classes
for match in re.finditer(r'class\s+(\w+)\s*(?:extends|implements|{)', content):
line_num = content[:match.start()].count('\n') + 1
symbols.append({
"name": match.group(1),
"kind": "class",
"line": line_num
})
citations = [
{"file": str(path), "type": "symbols", "count": len(symbols)}
]
return CodeQueryResult(
intent=CodeIntent.LIST_SYMBOLS,
success=True,
lsp_used=False,
fallback_used=True,
data=symbols,
citations=citations
)
except Exception as e:
return CodeQueryResult(
intent=CodeIntent.LIST_SYMBOLS,
success=False,
lsp_used=False,
fallback_used=True,
error=str(e)
)
async def _handle_go_to_definition(self, query: str, file_path: str = None) -> CodeQueryResult:
"""Handle go to definition query."""
# Extract symbol name from query
symbol = self._extract_symbol_from_query(query)
if not symbol:
return CodeQueryResult(
intent=CodeIntent.GO_TO_DEFINITION,
success=False,
lsp_used=False,
error="Could not identify symbol to find definition for"
)
if not file_path:
file_path = self._extract_file_from_query(query)
# Try LSP first
if self.runtime.lsp_ready and file_path:
try:
# Need to find the symbol position first
line, col = await self._find_symbol_position(file_path, symbol)
if line is not None:
definitions = await self.runtime.lsp_get_definition(file_path, line, col)
if definitions:
citations = [
{"file": d.get("uri", "").replace("file://", ""),
"line": d.get("range", {}).get("start", {}).get("line", 0) + 1,
"type": "definition"}
for d in definitions
]
return CodeQueryResult(
intent=CodeIntent.GO_TO_DEFINITION,
success=True,
lsp_used=True,
data={"symbol": symbol, "definitions": definitions},
citations=citations
)
except Exception as e:
logger.warning(f"LSP definition failed: {e}")
# Fallback to grep
return await self._fallback_find_definition(symbol)
async def _fallback_find_definition(self, symbol: str) -> CodeQueryResult:
"""Fallback definition finding using grep."""
try:
import subprocess
workspace = self.runtime.config.workspace
# Search for definition patterns
patterns = [
f"def {symbol}\\s*\\(", # Python function
f"class {symbol}\\s*[:\\(]", # Python class
f"function {symbol}\\s*\\(", # JS function
f"const {symbol}\\s*=", # JS const
f"let {symbol}\\s*=", # JS let
]
results = []
for pattern in patterns:
try:
result = subprocess.run(
["grep", "-rn", "-E", pattern, workspace],
capture_output=True,
text=True,
timeout=5
)
if result.stdout:
for line in result.stdout.strip().split('\n'):
if line:
parts = line.split(':', 2)
if len(parts) >= 2:
results.append({
"file": parts[0],
"line": int(parts[1]),
"content": parts[2] if len(parts) > 2 else ""
})
except subprocess.TimeoutExpired:
pass
if results:
citations = [
{"file": r["file"], "line": r["line"], "type": "definition"}
for r in results
]
return CodeQueryResult(
intent=CodeIntent.GO_TO_DEFINITION,
success=True,
lsp_used=False,
fallback_used=True,
data={"symbol": symbol, "definitions": results},
citations=citations
)
return CodeQueryResult(
intent=CodeIntent.GO_TO_DEFINITION,
success=False,
lsp_used=False,
fallback_used=True,
error=f"Definition for '{symbol}' not found"
)
except Exception as e:
return CodeQueryResult(
intent=CodeIntent.GO_TO_DEFINITION,
success=False,
lsp_used=False,
fallback_used=True,
error=str(e)
)
async def _handle_find_references(self, query: str, file_path: str = None) -> CodeQueryResult:
"""Handle find references query."""
symbol = self._extract_symbol_from_query(query)
if not symbol:
return CodeQueryResult(
intent=CodeIntent.FIND_REFERENCES,
success=False,
lsp_used=False,
error="Could not identify symbol to find references for"
)
if not file_path:
file_path = self._extract_file_from_query(query)
# Try LSP first
if self.runtime.lsp_ready and file_path:
try:
line, col = await self._find_symbol_position(file_path, symbol)
if line is not None:
references = await self.runtime.lsp_get_references(file_path, line, col)
if references:
citations = [
{"file": r.get("uri", "").replace("file://", ""),
"line": r.get("range", {}).get("start", {}).get("line", 0) + 1,
"type": "reference"}
for r in references
]
return CodeQueryResult(
intent=CodeIntent.FIND_REFERENCES,
success=True,
lsp_used=True,
data={"symbol": symbol, "references": references},
citations=citations
)
except Exception as e:
logger.warning(f"LSP references failed: {e}")
# Fallback to grep
return await self._fallback_find_references(symbol)
async def _fallback_find_references(self, symbol: str) -> CodeQueryResult:
"""Fallback reference finding using grep."""
try:
import subprocess
workspace = self.runtime.config.workspace
result = subprocess.run(
["grep", "-rn", "-w", symbol, workspace],
capture_output=True,
text=True,
timeout=10
)
results = []
if result.stdout:
for line in result.stdout.strip().split('\n'):
if line:
parts = line.split(':', 2)
if len(parts) >= 2:
results.append({
"file": parts[0],
"line": int(parts[1]),
"content": parts[2] if len(parts) > 2 else ""
})
if results:
citations = [
{"file": r["file"], "line": r["line"], "type": "reference"}
for r in results
]
return CodeQueryResult(
intent=CodeIntent.FIND_REFERENCES,
success=True,
lsp_used=False,
fallback_used=True,
data={"symbol": symbol, "references": results},
citations=citations
)
return CodeQueryResult(
intent=CodeIntent.FIND_REFERENCES,
success=False,
lsp_used=False,
fallback_used=True,
error=f"No references found for '{symbol}'"
)
except Exception as e:
return CodeQueryResult(
intent=CodeIntent.FIND_REFERENCES,
success=False,
lsp_used=False,
fallback_used=True,
error=str(e)
)
async def _handle_get_diagnostics(self, query: str, file_path: str = None) -> CodeQueryResult:
"""Handle get diagnostics query."""
if not file_path:
file_path = self._extract_file_from_query(query)
# Try LSP
if self.runtime.lsp_ready:
try:
diagnostics = await self.runtime.lsp_get_diagnostics(file_path)
citations = []
if diagnostics:
for d in diagnostics:
if isinstance(d, dict):
citations.append({
"file": file_path or "workspace",
"line": d.get("range", {}).get("start", {}).get("line", 0) + 1,
"type": "diagnostic",
"severity": d.get("severity", "unknown")
})
return CodeQueryResult(
intent=CodeIntent.GET_DIAGNOSTICS,
success=True,
lsp_used=True,
data=diagnostics,
citations=citations
)
except Exception as e:
logger.warning(f"LSP diagnostics failed: {e}")
return CodeQueryResult(
intent=CodeIntent.GET_DIAGNOSTICS,
success=False,
lsp_used=False,
error="LSP not available for diagnostics"
)
async def _handle_search_code(self, query: str, file_path: str = None) -> CodeQueryResult:
"""Handle code search query."""
# Extract search term
search_term = self._extract_search_term(query)
if not search_term:
return CodeQueryResult(
intent=CodeIntent.SEARCH_CODE,
success=False,
lsp_used=False,
error="Could not identify search term"
)
try:
import subprocess
workspace = self.runtime.config.workspace
cmd = ["grep", "-rn", search_term]
if file_path:
cmd.append(file_path)
else:
cmd.append(workspace)
result = subprocess.run(
cmd,
capture_output=True,
text=True,
timeout=10
)
results = []
if result.stdout:
for line in result.stdout.strip().split('\n')[:50]: # Limit results
if line:
parts = line.split(':', 2)
if len(parts) >= 2:
results.append({
"file": parts[0],
"line": int(parts[1]),
"content": parts[2] if len(parts) > 2 else ""
})
citations = [
{"file": r["file"], "line": r["line"], "type": "search_result"}
for r in results
]
return CodeQueryResult(
intent=CodeIntent.SEARCH_CODE,
success=True,
lsp_used=False,
data={"search_term": search_term, "results": results},
citations=citations
)
except Exception as e:
return CodeQueryResult(
intent=CodeIntent.SEARCH_CODE,
success=False,
lsp_used=False,
error=str(e)
)
def _extract_file_from_query(self, query: str) -> Optional[str]:
"""Extract file path from query."""
# Look for file patterns
patterns = [
r'(?:in|from|file)\s+["\']?([^\s"\']+\.\w+)["\']?',
r'([^\s]+\.(?:py|js|ts|tsx|jsx|go|rs|java|cpp|c|h))',
]
for pattern in patterns:
match = re.search(pattern, query, re.IGNORECASE)
if match:
return match.group(1)
return None
def _extract_symbol_from_query(self, query: str) -> Optional[str]:
"""Extract symbol name from query."""
# Common patterns for symbol extraction
patterns = [
r'(?:definition\s+of|where\s+is|find|show)\s+["\']?(\w+)["\']?',
r'["\'](\w+)["\']',
r'`(\w+)`',
r'(\w+)\s+(?:is\s+)?(?:defined|used)',
]
for pattern in patterns:
match = re.search(pattern, query, re.IGNORECASE)
if match:
symbol = match.group(1)
# Filter out common words
if symbol.lower() not in ['the', 'a', 'an', 'is', 'are', 'in', 'of', 'to', 'for']:
return symbol
return None
def _extract_search_term(self, query: str) -> Optional[str]:
"""Extract search term from query."""
patterns = [
r'search\s+(?:for\s+)?["\']([^"\']+)["\']',
r'find\s+["\']([^"\']+)["\']',
r'grep\s+["\']?([^\s"\']+)["\']?',
]
for pattern in patterns:
match = re.search(pattern, query, re.IGNORECASE)
if match:
return match.group(1)
return None
async def _find_symbol_position(self, file_path: str, symbol: str) -> tuple:
"""Find the position of a symbol in a file."""
try:
path = Path(file_path)
if not path.exists():
path = Path(self.runtime.config.workspace) / file_path
if not path.exists():
return None, None
content = path.read_text()
lines = content.split('\n')
for i, line in enumerate(lines):
col = line.find(symbol)
if col >= 0:
return i, col
return None, None
except Exception:
return None, None
@@ -0,0 +1,810 @@
"""
Debug CLI handler for PraisonAI.
Provides non-interactive debug commands for testing interactive flows:
- debug interactive: Run single interactive turn
- debug lsp: Direct LSP probes
- debug acp: Direct ACP probes
- debug trace: Trace record/replay/diff
"""
import argparse
import asyncio
import json
import logging
import sys
import time
from typing import Any, Dict, List
logger = logging.getLogger(__name__)
class DebugHandler:
"""Handler for debug CLI commands."""
@staticmethod
def setup_subparser(subparsers) -> None:
"""Set up the debug subcommand parser."""
debug_parser = subparsers.add_parser(
"debug",
help="Debug and test interactive flows",
description="Debug commands for testing interactive coding assistant flows"
)
debug_subparsers = debug_parser.add_subparsers(dest="debug_command", help="Debug subcommands")
# debug interactive
interactive_parser = debug_subparsers.add_parser(
"interactive",
help="Run single interactive turn non-interactively",
description="Execute one interactive turn with full LSP/ACP pipeline"
)
interactive_parser.add_argument(
"-p", "--prompt",
type=str,
required=True,
help="Prompt to execute"
)
interactive_parser.add_argument(
"--lsp",
action="store_true",
default=True,
help="Enable LSP (default: enabled)"
)
interactive_parser.add_argument(
"--no-lsp",
action="store_true",
help="Disable LSP"
)
interactive_parser.add_argument(
"--acp",
action="store_true",
default=True,
help="Enable ACP (default: enabled)"
)
interactive_parser.add_argument(
"--no-acp",
action="store_true",
help="Disable ACP"
)
interactive_parser.add_argument(
"--approval",
type=str,
choices=["manual", "auto", "scoped"],
default="auto",
help="Approval mode (default: auto for debug)"
)
interactive_parser.add_argument(
"--workspace", "-w",
type=str,
default=".",
help="Workspace root directory"
)
interactive_parser.add_argument(
"--json",
action="store_true",
help="Output JSON trace"
)
interactive_parser.add_argument(
"--trace-file",
type=str,
help="Save trace to file"
)
interactive_parser.add_argument(
"--timeout",
type=float,
default=60.0,
help="Timeout in seconds"
)
interactive_parser.add_argument(
"--model",
type=str,
help="LLM model to use"
)
interactive_parser.set_defaults(func=DebugHandler.handle_interactive)
# debug lsp
lsp_parser = debug_subparsers.add_parser(
"lsp",
help="Direct LSP probes",
description="Probe LSP server directly"
)
lsp_subparsers = lsp_parser.add_subparsers(dest="lsp_command", help="LSP subcommands")
# debug lsp status
lsp_status = lsp_subparsers.add_parser("status", help="Show LSP status")
lsp_status.add_argument("--workspace", "-w", type=str, default=".")
lsp_status.add_argument("--json", action="store_true")
lsp_status.set_defaults(func=DebugHandler.handle_lsp_status)
# debug lsp symbols
lsp_symbols = lsp_subparsers.add_parser("symbols", help="List symbols in file")
lsp_symbols.add_argument("file", type=str, help="File path")
lsp_symbols.add_argument("--workspace", "-w", type=str, default=".")
lsp_symbols.add_argument("--json", action="store_true")
lsp_symbols.set_defaults(func=DebugHandler.handle_lsp_symbols)
# debug lsp definition
lsp_def = lsp_subparsers.add_parser("definition", help="Get definition location")
lsp_def.add_argument("location", type=str, help="file:line:col")
lsp_def.add_argument("--workspace", "-w", type=str, default=".")
lsp_def.add_argument("--json", action="store_true")
lsp_def.set_defaults(func=DebugHandler.handle_lsp_definition)
# debug lsp references
lsp_refs = lsp_subparsers.add_parser("references", help="Get references")
lsp_refs.add_argument("location", type=str, help="file:line:col")
lsp_refs.add_argument("--workspace", "-w", type=str, default=".")
lsp_refs.add_argument("--json", action="store_true")
lsp_refs.set_defaults(func=DebugHandler.handle_lsp_references)
# debug lsp diagnostics
lsp_diag = lsp_subparsers.add_parser("diagnostics", help="Get diagnostics")
lsp_diag.add_argument("file", type=str, nargs="?", help="File path (optional)")
lsp_diag.add_argument("--all", action="store_true", help="Get all diagnostics")
lsp_diag.add_argument("--workspace", "-w", type=str, default=".")
lsp_diag.add_argument("--json", action="store_true")
lsp_diag.set_defaults(func=DebugHandler.handle_lsp_diagnostics)
# debug acp
acp_parser = debug_subparsers.add_parser(
"acp",
help="Direct ACP probes",
description="Probe ACP server directly"
)
acp_subparsers = acp_parser.add_subparsers(dest="acp_command", help="ACP subcommands")
# debug acp status
acp_status = acp_subparsers.add_parser("status", help="Show ACP status")
acp_status.add_argument("--workspace", "-w", type=str, default=".")
acp_status.add_argument("--json", action="store_true")
acp_status.set_defaults(func=DebugHandler.handle_acp_status)
# debug acp plan
acp_plan = acp_subparsers.add_parser("plan", help="Create action plan")
acp_plan.add_argument("-p", "--prompt", type=str, required=True)
acp_plan.add_argument("--workspace", "-w", type=str, default=".")
acp_plan.add_argument("--json", action="store_true")
acp_plan.set_defaults(func=DebugHandler.handle_acp_plan)
# debug acp apply
acp_apply = acp_subparsers.add_parser("apply", help="Apply action plan")
acp_apply.add_argument("-p", "--prompt", type=str, required=True)
acp_apply.add_argument("--approval", type=str, choices=["manual", "auto", "scoped"], default="auto")
acp_apply.add_argument("--workspace", "-w", type=str, default=".")
acp_apply.add_argument("--json", action="store_true")
acp_apply.set_defaults(func=DebugHandler.handle_acp_apply)
# debug trace
trace_parser = debug_subparsers.add_parser(
"trace",
help="Trace record/replay/diff",
description="Manage trace recordings"
)
trace_subparsers = trace_parser.add_subparsers(dest="trace_command", help="Trace subcommands")
# debug trace record
trace_record = trace_subparsers.add_parser("record", help="Record session trace")
trace_record.add_argument("-o", "--output", type=str, required=True, help="Output file")
trace_record.set_defaults(func=DebugHandler.handle_trace_record)
# debug trace replay
trace_replay = trace_subparsers.add_parser("replay", help="Replay recorded trace")
trace_replay.add_argument("file", type=str, help="Trace file to replay")
trace_replay.add_argument("--json", action="store_true")
trace_replay.set_defaults(func=DebugHandler.handle_trace_replay)
# debug trace diff
trace_diff = trace_subparsers.add_parser("diff", help="Compare two traces")
trace_diff.add_argument("file1", type=str, help="First trace file")
trace_diff.add_argument("file2", type=str, help="Second trace file")
trace_diff.add_argument("--json", action="store_true")
trace_diff.set_defaults(func=DebugHandler.handle_trace_diff)
@staticmethod
def handle_interactive(args: argparse.Namespace) -> int:
"""Handle debug interactive command."""
return asyncio.run(DebugHandler._run_interactive(args))
@staticmethod
async def _run_interactive(args: argparse.Namespace) -> int:
"""Run interactive turn asynchronously."""
from .interactive_runtime import create_runtime
from .code_intelligence import CodeIntelligenceRouter
from .action_orchestrator import ActionOrchestrator
start_time = time.time()
# Create runtime
runtime = create_runtime(
workspace=args.workspace,
lsp=args.lsp and not args.no_lsp,
acp=args.acp and not args.no_acp,
approval=args.approval,
trace=True,
trace_file=args.trace_file,
json_output=args.json,
timeout=args.timeout,
model=getattr(args, 'model', None)
)
result = {
"prompt": args.prompt,
"workspace": args.workspace,
"start_time": start_time
}
try:
# Start runtime
status = await runtime.start()
result["runtime_status"] = status
if not status.get("started"):
result["error"] = "Failed to start runtime"
DebugHandler._output_result(result, args.json)
return 1
# Classify intent
code_router = CodeIntelligenceRouter(runtime)
action_orchestrator = ActionOrchestrator(runtime)
prompt = args.prompt.lower()
# Determine if this is a code query or action
is_code_query = any(kw in prompt for kw in [
"list", "show", "find", "where", "definition", "reference",
"symbol", "function", "class", "diagnostic", "error"
])
is_action = any(kw in prompt for kw in [
"create", "edit", "modify", "delete", "rename", "add", "remove",
"change", "update", "refactor"
])
if is_code_query:
# Route to code intelligence
query_result = await code_router.handle_query(args.prompt)
result["type"] = "code_query"
result["query_result"] = query_result.to_dict()
result["lsp_used"] = query_result.lsp_used
result["citations"] = query_result.citations
elif is_action:
# Route to action orchestrator
action_result = await action_orchestrator.execute(
args.prompt,
auto_approve=(args.approval == "auto")
)
result["type"] = "action"
result["action_result"] = action_result.to_dict()
result["read_only_blocked"] = action_result.read_only_blocked
else:
# General query - just return status
result["type"] = "general"
result["message"] = "Query processed"
# Get trace
trace = runtime.get_trace()
if trace:
result["trace"] = trace.to_dict()
result["success"] = True
except Exception as e:
result["error"] = str(e)
result["success"] = False
logger.exception("Debug interactive failed")
finally:
await runtime.stop()
result["end_time"] = time.time()
result["duration_ms"] = (result["end_time"] - start_time) * 1000
# Save trace if requested
if args.trace_file and "trace" in result:
with open(args.trace_file, "w") as f:
json.dump(result["trace"], f, indent=2, default=str)
if not args.json:
print(f"Trace saved to: {args.trace_file}")
DebugHandler._output_result(result, args.json)
return 0 if result.get("success") else 1
@staticmethod
def handle_lsp_status(args: argparse.Namespace) -> int:
"""Handle debug lsp status command."""
return asyncio.run(DebugHandler._run_lsp_status(args))
@staticmethod
async def _run_lsp_status(args: argparse.Namespace) -> int:
"""Run LSP status check."""
from .interactive_runtime import create_runtime
runtime = create_runtime(
workspace=args.workspace,
lsp=True,
acp=False
)
try:
await runtime.start()
result = {
"lsp_enabled": True,
"lsp_ready": runtime.lsp_ready,
"lsp_status": runtime._lsp_state.status.value,
"lsp_error": runtime._lsp_state.error,
"workspace": args.workspace
}
DebugHandler._output_result(result, args.json)
return 0 if runtime.lsp_ready else 1
finally:
await runtime.stop()
@staticmethod
def handle_lsp_symbols(args: argparse.Namespace) -> int:
"""Handle debug lsp symbols command."""
return asyncio.run(DebugHandler._run_lsp_symbols(args))
@staticmethod
async def _run_lsp_symbols(args: argparse.Namespace) -> int:
"""Run LSP symbols query."""
from .interactive_runtime import create_runtime
from .code_intelligence import CodeIntelligenceRouter
runtime = create_runtime(
workspace=args.workspace,
lsp=True,
acp=False
)
try:
await runtime.start()
if not runtime.lsp_ready:
# Use fallback
router = CodeIntelligenceRouter(runtime)
result = await router._fallback_list_symbols(args.file)
else:
symbols = await runtime.lsp_get_symbols(args.file)
result = {
"file": args.file,
"symbols": symbols,
"count": len(symbols) if symbols else 0,
"lsp_used": True
}
if hasattr(result, 'to_dict'):
result = result.to_dict()
DebugHandler._output_result(result, args.json)
return 0
finally:
await runtime.stop()
@staticmethod
def handle_lsp_definition(args: argparse.Namespace) -> int:
"""Handle debug lsp definition command."""
return asyncio.run(DebugHandler._run_lsp_definition(args))
@staticmethod
async def _run_lsp_definition(args: argparse.Namespace) -> int:
"""Run LSP definition query."""
from .interactive_runtime import create_runtime
# Parse location
parts = args.location.split(":")
if len(parts) != 3:
print("Error: Location must be in format file:line:col", file=sys.stderr)
return 1
file_path, line, col = parts[0], int(parts[1]), int(parts[2])
runtime = create_runtime(
workspace=args.workspace,
lsp=True,
acp=False
)
try:
await runtime.start()
definitions = await runtime.lsp_get_definition(file_path, line, col)
result = {
"file": file_path,
"line": line,
"col": col,
"definitions": definitions,
"count": len(definitions) if definitions else 0,
"lsp_used": runtime.lsp_ready
}
DebugHandler._output_result(result, args.json)
return 0
finally:
await runtime.stop()
@staticmethod
def handle_lsp_references(args: argparse.Namespace) -> int:
"""Handle debug lsp references command."""
return asyncio.run(DebugHandler._run_lsp_references(args))
@staticmethod
async def _run_lsp_references(args: argparse.Namespace) -> int:
"""Run LSP references query."""
from .interactive_runtime import create_runtime
# Parse location
parts = args.location.split(":")
if len(parts) != 3:
print("Error: Location must be in format file:line:col", file=sys.stderr)
return 1
file_path, line, col = parts[0], int(parts[1]), int(parts[2])
runtime = create_runtime(
workspace=args.workspace,
lsp=True,
acp=False
)
try:
await runtime.start()
references = await runtime.lsp_get_references(file_path, line, col)
result = {
"file": file_path,
"line": line,
"col": col,
"references": references,
"count": len(references) if references else 0,
"lsp_used": runtime.lsp_ready
}
DebugHandler._output_result(result, args.json)
return 0
finally:
await runtime.stop()
@staticmethod
def handle_lsp_diagnostics(args: argparse.Namespace) -> int:
"""Handle debug lsp diagnostics command."""
return asyncio.run(DebugHandler._run_lsp_diagnostics(args))
@staticmethod
async def _run_lsp_diagnostics(args: argparse.Namespace) -> int:
"""Run LSP diagnostics query."""
from .interactive_runtime import create_runtime
runtime = create_runtime(
workspace=args.workspace,
lsp=True,
acp=False
)
try:
await runtime.start()
file_path = args.file if not args.all else None
diagnostics = await runtime.lsp_get_diagnostics(file_path)
result = {
"file": file_path or "all",
"diagnostics": diagnostics,
"count": len(diagnostics) if diagnostics else 0,
"lsp_used": runtime.lsp_ready
}
DebugHandler._output_result(result, args.json)
return 0
finally:
await runtime.stop()
@staticmethod
def handle_acp_status(args: argparse.Namespace) -> int:
"""Handle debug acp status command."""
return asyncio.run(DebugHandler._run_acp_status(args))
@staticmethod
async def _run_acp_status(args: argparse.Namespace) -> int:
"""Run ACP status check."""
from .interactive_runtime import create_runtime
runtime = create_runtime(
workspace=args.workspace,
lsp=False,
acp=True
)
try:
await runtime.start()
result = {
"acp_enabled": True,
"acp_ready": runtime.acp_ready,
"acp_status": runtime._acp_state.status.value,
"acp_error": runtime._acp_state.error,
"read_only": runtime.read_only,
"workspace": args.workspace
}
DebugHandler._output_result(result, args.json)
return 0 if runtime.acp_ready else 1
finally:
await runtime.stop()
@staticmethod
def handle_acp_plan(args: argparse.Namespace) -> int:
"""Handle debug acp plan command."""
return asyncio.run(DebugHandler._run_acp_plan(args))
@staticmethod
async def _run_acp_plan(args: argparse.Namespace) -> int:
"""Run ACP plan creation."""
from .interactive_runtime import create_runtime
from .action_orchestrator import ActionOrchestrator
runtime = create_runtime(
workspace=args.workspace,
lsp=False,
acp=True
)
try:
await runtime.start()
orchestrator = ActionOrchestrator(runtime)
result = await orchestrator.create_plan(args.prompt)
DebugHandler._output_result(result.to_dict(), args.json)
return 0 if result.success else 1
finally:
await runtime.stop()
@staticmethod
def handle_acp_apply(args: argparse.Namespace) -> int:
"""Handle debug acp apply command."""
return asyncio.run(DebugHandler._run_acp_apply(args))
@staticmethod
async def _run_acp_apply(args: argparse.Namespace) -> int:
"""Run ACP plan and apply."""
from .interactive_runtime import create_runtime
from .action_orchestrator import ActionOrchestrator
runtime = create_runtime(
workspace=args.workspace,
lsp=False,
acp=True,
approval=args.approval
)
try:
await runtime.start()
orchestrator = ActionOrchestrator(runtime)
result = await orchestrator.execute(
args.prompt,
auto_approve=(args.approval == "auto")
)
DebugHandler._output_result(result.to_dict(), args.json)
return 0 if result.success else 1
finally:
await runtime.stop()
@staticmethod
def handle_trace_record(args: argparse.Namespace) -> int:
"""Handle debug trace record command."""
print(f"Trace recording will be saved to: {args.output}")
print("Start an interactive session with --trace-file to record.")
return 0
@staticmethod
def handle_trace_replay(args: argparse.Namespace) -> int:
"""Handle debug trace replay command."""
try:
with open(args.file) as f:
trace = json.load(f)
if args.json:
print(json.dumps(trace, indent=2))
else:
print(f"Trace from: {args.file}")
print(f"Start time: {trace.get('start_time')}")
print(f"End time: {trace.get('end_time')}")
print(f"Entries: {len(trace.get('entries', []))}")
print("\nEntries:")
for entry in trace.get("entries", []):
print(f" [{entry.get('category')}] {entry.get('action')}")
if entry.get('duration_ms'):
print(f" Duration: {entry.get('duration_ms'):.2f}ms")
if entry.get('error'):
print(f" Error: {entry.get('error')}")
return 0
except Exception as e:
print(f"Error: {e}", file=sys.stderr)
return 1
@staticmethod
def handle_trace_diff(args: argparse.Namespace) -> int:
"""Handle debug trace diff command."""
try:
with open(args.file1) as f:
trace1 = json.load(f)
with open(args.file2) as f:
trace2 = json.load(f)
diff = {
"file1": args.file1,
"file2": args.file2,
"entries_diff": {
"file1_count": len(trace1.get("entries", [])),
"file2_count": len(trace2.get("entries", []))
},
"duration_diff": {
"file1": trace1.get("end_time", 0) - trace1.get("start_time", 0),
"file2": trace2.get("end_time", 0) - trace2.get("start_time", 0)
}
}
DebugHandler._output_result(diff, args.json)
return 0
except Exception as e:
print(f"Error: {e}", file=sys.stderr)
return 1
@staticmethod
def _output_result(result: Dict[str, Any], as_json: bool) -> None:
"""Output result in appropriate format."""
if as_json:
print(json.dumps(result, indent=2, default=str))
else:
for key, value in result.items():
if isinstance(value, dict):
print(f"{key}:")
for k, v in value.items():
print(f" {k}: {v}")
elif isinstance(value, list):
print(f"{key}: ({len(value)} items)")
for item in value[:5]: # Show first 5
print(f" - {item}")
if len(value) > 5:
print(f" ... and {len(value) - 5} more")
else:
print(f"{key}: {value}")
def setup_debug_subparser(subparsers) -> None:
"""Set up the debug subcommand parser."""
DebugHandler.setup_subparser(subparsers)
def run_debug_command(args: List[str]) -> int:
"""Run debug command from CLI."""
parser = argparse.ArgumentParser(
prog="praisonai debug",
description="Debug and test interactive coding assistant flows"
)
subparsers = parser.add_subparsers(dest="debug_command", help="Debug subcommands")
# debug interactive
interactive_parser = subparsers.add_parser(
"interactive",
help="Run single interactive turn non-interactively",
description="Execute one interactive turn with full LSP/ACP pipeline"
)
interactive_parser.add_argument("-p", "--prompt", type=str, required=True, help="Prompt to execute")
interactive_parser.add_argument("--lsp", action="store_true", default=True, help="Enable LSP (default: enabled)")
interactive_parser.add_argument("--no-lsp", action="store_true", help="Disable LSP")
interactive_parser.add_argument("--acp", action="store_true", default=True, help="Enable ACP (default: enabled)")
interactive_parser.add_argument("--no-acp", action="store_true", help="Disable ACP")
interactive_parser.add_argument("--approval", type=str, choices=["manual", "auto", "scoped"], default="auto", help="Approval mode")
interactive_parser.add_argument("--workspace", "-w", type=str, default=".", help="Workspace root directory")
interactive_parser.add_argument("--json", action="store_true", help="Output JSON trace")
interactive_parser.add_argument("--trace-file", type=str, help="Save trace to file")
interactive_parser.add_argument("--timeout", type=float, default=60.0, help="Timeout in seconds")
interactive_parser.add_argument("--model", type=str, help="LLM model to use")
interactive_parser.set_defaults(func=DebugHandler.handle_interactive)
# debug lsp
lsp_parser = subparsers.add_parser("lsp", help="Direct LSP probes")
lsp_subparsers = lsp_parser.add_subparsers(dest="lsp_command", help="LSP subcommands")
# debug lsp status
lsp_status = lsp_subparsers.add_parser("status", help="Show LSP status")
lsp_status.add_argument("--workspace", "-w", type=str, default=".")
lsp_status.add_argument("--json", action="store_true")
lsp_status.set_defaults(func=DebugHandler.handle_lsp_status)
# debug lsp symbols
lsp_symbols = lsp_subparsers.add_parser("symbols", help="List symbols in file")
lsp_symbols.add_argument("file", type=str, help="File path")
lsp_symbols.add_argument("--workspace", "-w", type=str, default=".")
lsp_symbols.add_argument("--json", action="store_true")
lsp_symbols.set_defaults(func=DebugHandler.handle_lsp_symbols)
# debug lsp definition
lsp_def = lsp_subparsers.add_parser("definition", help="Get definition location")
lsp_def.add_argument("location", type=str, help="file:line:col")
lsp_def.add_argument("--workspace", "-w", type=str, default=".")
lsp_def.add_argument("--json", action="store_true")
lsp_def.set_defaults(func=DebugHandler.handle_lsp_definition)
# debug lsp references
lsp_refs = lsp_subparsers.add_parser("references", help="Get references")
lsp_refs.add_argument("location", type=str, help="file:line:col")
lsp_refs.add_argument("--workspace", "-w", type=str, default=".")
lsp_refs.add_argument("--json", action="store_true")
lsp_refs.set_defaults(func=DebugHandler.handle_lsp_references)
# debug lsp diagnostics
lsp_diag = lsp_subparsers.add_parser("diagnostics", help="Get diagnostics")
lsp_diag.add_argument("file", type=str, nargs="?", help="File path (optional)")
lsp_diag.add_argument("--all", action="store_true", help="Get all diagnostics")
lsp_diag.add_argument("--workspace", "-w", type=str, default=".")
lsp_diag.add_argument("--json", action="store_true")
lsp_diag.set_defaults(func=DebugHandler.handle_lsp_diagnostics)
# debug acp
acp_parser = subparsers.add_parser("acp", help="Direct ACP probes")
acp_subparsers = acp_parser.add_subparsers(dest="acp_command", help="ACP subcommands")
# debug acp status
acp_status = acp_subparsers.add_parser("status", help="Show ACP status")
acp_status.add_argument("--workspace", "-w", type=str, default=".")
acp_status.add_argument("--json", action="store_true")
acp_status.set_defaults(func=DebugHandler.handle_acp_status)
# debug acp plan
acp_plan = acp_subparsers.add_parser("plan", help="Create action plan")
acp_plan.add_argument("-p", "--prompt", type=str, required=True)
acp_plan.add_argument("--workspace", "-w", type=str, default=".")
acp_plan.add_argument("--json", action="store_true")
acp_plan.set_defaults(func=DebugHandler.handle_acp_plan)
# debug acp apply
acp_apply = acp_subparsers.add_parser("apply", help="Apply action plan")
acp_apply.add_argument("-p", "--prompt", type=str, required=True)
acp_apply.add_argument("--approval", type=str, choices=["manual", "auto", "scoped"], default="auto")
acp_apply.add_argument("--workspace", "-w", type=str, default=".")
acp_apply.add_argument("--json", action="store_true")
acp_apply.set_defaults(func=DebugHandler.handle_acp_apply)
# debug trace
trace_parser = subparsers.add_parser("trace", help="Trace record/replay/diff")
trace_subparsers = trace_parser.add_subparsers(dest="trace_command", help="Trace subcommands")
# debug trace record
trace_record = trace_subparsers.add_parser("record", help="Record session trace")
trace_record.add_argument("-o", "--output", type=str, required=True, help="Output file")
trace_record.set_defaults(func=DebugHandler.handle_trace_record)
# debug trace replay
trace_replay = trace_subparsers.add_parser("replay", help="Replay recorded trace")
trace_replay.add_argument("file", type=str, help="Trace file to replay")
trace_replay.add_argument("--json", action="store_true")
trace_replay.set_defaults(func=DebugHandler.handle_trace_replay)
# debug trace diff
trace_diff = trace_subparsers.add_parser("diff", help="Compare two traces")
trace_diff.add_argument("file1", type=str, help="First trace file")
trace_diff.add_argument("file2", type=str, help="Second trace file")
trace_diff.add_argument("--json", action="store_true")
trace_diff.set_defaults(func=DebugHandler.handle_trace_diff)
parsed = parser.parse_args(args)
if hasattr(parsed, 'func'):
return parsed.func(parsed)
else:
parser.print_help()
return 0
@@ -0,0 +1,289 @@
"""
Diagnostics export CLI handler for PraisonAI.
Provides diagnostic export functionality:
- diag export: Bundle logs, traces, config for bug reports
"""
import argparse
import json
import logging
import os
import shutil
import tempfile
import time
import zipfile
from pathlib import Path
from typing import List
logger = logging.getLogger(__name__)
class DiagHandler:
"""Handler for diag CLI commands."""
@staticmethod
def setup_subparser(subparsers) -> None:
"""Set up the diag subcommand parser."""
diag_parser = subparsers.add_parser(
"diag",
help="Diagnostics export",
description="Export diagnostic information for bug reports"
)
diag_subparsers = diag_parser.add_subparsers(dest="diag_command", help="Diag subcommands")
# diag export
export_parser = diag_subparsers.add_parser("export", help="Export diagnostic bundle")
export_parser.add_argument(
"-o", "--output",
type=str,
help="Output path (default: praisonai-diag-{timestamp}.zip)"
)
export_parser.add_argument(
"--include-logs",
action="store_true",
default=True,
help="Include log files (default: true)"
)
export_parser.add_argument(
"--include-config",
action="store_true",
default=True,
help="Include configuration (default: true)"
)
export_parser.add_argument(
"--include-trace",
action="store_true",
default=True,
help="Include recent traces (default: true)"
)
export_parser.add_argument(
"--workspace", "-w",
type=str,
default=".",
help="Workspace root directory"
)
export_parser.add_argument("--json", action="store_true", help="Output JSON")
export_parser.set_defaults(func=DiagHandler.handle_export)
@staticmethod
def handle_export(args: argparse.Namespace) -> int:
"""Handle diag export command."""
timestamp = int(time.time())
output_path = args.output or f"praisonai-diag-{timestamp}.zip"
collected_files = []
with tempfile.TemporaryDirectory() as tmpdir:
tmppath = Path(tmpdir)
# Collect system info
system_info = DiagHandler._collect_system_info()
system_file = tmppath / "system_info.json"
with open(system_file, "w") as f:
json.dump(system_info, f, indent=2, default=str)
collected_files.append("system_info.json")
# Collect config
if args.include_config:
config_files = DiagHandler._collect_config(args.workspace, tmppath)
collected_files.extend(config_files)
# Collect logs
if args.include_logs:
log_files = DiagHandler._collect_logs(tmppath)
collected_files.extend(log_files)
# Collect traces
if args.include_trace:
trace_files = DiagHandler._collect_traces(args.workspace, tmppath)
collected_files.extend(trace_files)
# Create zip
with zipfile.ZipFile(output_path, 'w', zipfile.ZIP_DEFLATED) as zf:
for file in tmppath.rglob("*"):
if file.is_file():
arcname = file.relative_to(tmppath)
zf.write(file, arcname)
result = {
"action": "export",
"output": output_path,
"files_collected": collected_files,
"count": len(collected_files)
}
DiagHandler._output_result(result, args.json)
return 0
@staticmethod
def _collect_system_info() -> dict:
"""Collect system information."""
import platform
import sys
info = {
"timestamp": time.time(),
"platform": {
"system": platform.system(),
"release": platform.release(),
"version": platform.version(),
"machine": platform.machine(),
"python_version": sys.version
},
"praisonai": {}
}
# Get PraisonAI version
try:
from praisonai.version import __version__
info["praisonai"]["version"] = __version__
except ImportError:
info["praisonai"]["version"] = "unknown"
# Get praisonaiagents version
try:
import praisonaiagents
info["praisonai"]["agents_version"] = getattr(praisonaiagents, '__version__', 'unknown')
except ImportError:
info["praisonai"]["agents_version"] = "not installed"
# Check for optional dependencies
info["dependencies"] = {}
for dep in ["acp", "litellm", "openai", "anthropic", "google.generativeai"]:
try:
__import__(dep.split(".")[0])
info["dependencies"][dep] = "installed"
except ImportError:
info["dependencies"][dep] = "not installed"
# Environment (redacted)
env_keys = [
"OPENAI_API_KEY", "ANTHROPIC_API_KEY", "GOOGLE_API_KEY",
"PRAISONAI_APPROVAL_MODE", "LOGLEVEL"
]
info["environment"] = {}
for key in env_keys:
value = os.environ.get(key)
if value:
if "KEY" in key or "SECRET" in key:
info["environment"][key] = f"{value[:4]}...{value[-4:]}" if len(value) > 8 else "***"
else:
info["environment"][key] = value
else:
info["environment"][key] = None
return info
@staticmethod
def _collect_config(workspace: str, tmppath: Path) -> List[str]:
"""Collect configuration files."""
collected = []
workspace_path = Path(workspace)
config_dir = tmppath / "config"
config_dir.mkdir(exist_ok=True)
# Look for config files
config_patterns = [
"agents.yaml",
"workflow.yaml",
".praisonai.yaml",
".env.example", # Not .env to avoid secrets
"pyproject.toml",
]
for pattern in config_patterns:
for file in workspace_path.glob(pattern):
if file.is_file():
dest = config_dir / file.name
shutil.copy(file, dest)
collected.append(f"config/{file.name}")
return collected
@staticmethod
def _collect_logs(tmppath: Path) -> List[str]:
"""Collect log files."""
collected = []
logs_dir = tmppath / "logs"
logs_dir.mkdir(exist_ok=True)
# Look for log files in common locations
log_locations = [
Path.home() / ".praison" / "logs",
Path.home() / ".praisonai" / "logs",
Path("/var/log/praisonai"),
]
for log_dir in log_locations:
if log_dir.exists():
for log_file in log_dir.glob("*.log"):
if log_file.is_file():
dest = logs_dir / log_file.name
# Only copy last 1000 lines
try:
with open(log_file) as f:
lines = f.readlines()[-1000:]
with open(dest, "w") as f:
f.writelines(lines)
collected.append(f"logs/{log_file.name}")
except Exception:
pass
return collected
@staticmethod
def _collect_traces(workspace: str, tmppath: Path) -> List[str]:
"""Collect trace files."""
collected = []
traces_dir = tmppath / "traces"
traces_dir.mkdir(exist_ok=True)
workspace_path = Path(workspace)
# Look for trace files
trace_patterns = ["*_trace.json", "praisonai_trace*.json", "*.trace"]
for pattern in trace_patterns:
for trace_file in workspace_path.glob(pattern):
if trace_file.is_file():
dest = traces_dir / trace_file.name
shutil.copy(trace_file, dest)
collected.append(f"traces/{trace_file.name}")
return collected
@staticmethod
def _output_result(result: dict, as_json: bool) -> None:
"""Output result in appropriate format."""
if as_json:
print(json.dumps(result, indent=2, default=str))
else:
print(f"Diagnostic bundle created: {result['output']}")
print(f"Files collected: {result['count']}")
for f in result['files_collected']:
print(f" - {f}")
def run_diag_command(args: List[str]) -> int:
"""Run diag command from CLI."""
parser = argparse.ArgumentParser(prog="praisonai diag")
subparsers = parser.add_subparsers(dest="diag_command", help="Diag subcommands")
# diag export
export_parser = subparsers.add_parser("export", help="Export diagnostic bundle")
export_parser.add_argument("-o", "--output", type=str)
export_parser.add_argument("--include-logs", action="store_true", default=True)
export_parser.add_argument("--include-config", action="store_true", default=True)
export_parser.add_argument("--include-trace", action="store_true", default=True)
export_parser.add_argument("--workspace", "-w", type=str, default=".")
export_parser.add_argument("--json", action="store_true")
export_parser.set_defaults(func=DiagHandler.handle_export)
parsed = parser.parse_args(args)
if hasattr(parsed, 'func'):
return parsed.func(parsed)
else:
parser.print_help()
return 0
@@ -20,3 +20,5 @@ def register_all_checks():
from . import performance_checks
from . import selftest_checks
from . import serve_checks
from . import lsp_checks
from . import acp_checks
@@ -0,0 +1,240 @@
"""
ACP health checks for PraisonAI Doctor.
"""
import asyncio
import logging
from ..models import CheckCategory, CheckResult, CheckStatus
from ..registry import register_check
logger = logging.getLogger(__name__)
@register_check(
id="acp_module",
title="ACP Module Available",
category=CheckCategory.TOOLS,
description="Check if ACP module is installed"
)
def check_acp_module() -> CheckResult:
"""Check if ACP module is available."""
try:
import importlib.util
spec = importlib.util.find_spec("praisonai.acp")
if spec is not None:
return CheckResult(
status=CheckStatus.PASS,
message="ACP module is available",
details={"module": "praisonai.acp"}
)
else:
return CheckResult(
status=CheckStatus.WARN,
message="ACP module not found",
details={"hint": "ACP module should be part of praisonai package"}
)
except Exception as e:
return CheckResult(
status=CheckStatus.FAIL,
message=f"Error checking ACP module: {e}",
details={"error": str(e)}
)
@register_check(
id="acp_sdk",
title="ACP SDK Available",
category=CheckCategory.TOOLS,
description="Check if agent-client-protocol SDK is installed"
)
def check_acp_sdk() -> CheckResult:
"""Check if ACP SDK is available."""
try:
import importlib.util
spec = importlib.util.find_spec("acp")
if spec is not None:
return CheckResult(
status=CheckStatus.PASS,
message="ACP SDK (agent-client-protocol) is installed",
details={"package": "agent-client-protocol"}
)
else:
return CheckResult(
status=CheckStatus.INFO,
message="ACP SDK not installed (optional)",
details={"hint": "Install with: pip install agent-client-protocol"}
)
except Exception as e:
return CheckResult(
status=CheckStatus.FAIL,
message=f"Error checking ACP SDK: {e}",
details={"error": str(e)}
)
@register_check(
id="acp_server",
title="ACP Server Import",
category=CheckCategory.TOOLS,
description="Check if ACP server can be imported"
)
def check_acp_server() -> CheckResult:
"""Check if ACP server can be imported."""
try:
from praisonai.acp import ACPServer # noqa: F401
return CheckResult(
status=CheckStatus.PASS,
message="ACP server can be imported",
details={"class": "ACPServer"}
)
except ImportError as e:
return CheckResult(
status=CheckStatus.WARN,
message=f"ACP server import failed: {e}",
details={"error": str(e)}
)
except Exception as e:
return CheckResult(
status=CheckStatus.FAIL,
message=f"Error importing ACP server: {e}",
details={"error": str(e)}
)
@register_check(
id="acp_config",
title="ACP Configuration",
category=CheckCategory.CONFIG,
description="Check ACP configuration"
)
def check_acp_config() -> CheckResult:
"""Check ACP configuration."""
import os
config = {
"approval_mode": os.environ.get("PRAISONAI_APPROVAL_MODE", "manual"),
"allow_write": os.environ.get("PRAISONAI_ALLOW_WRITE", "false"),
"allow_shell": os.environ.get("PRAISONAI_ALLOW_SHELL", "false"),
}
return CheckResult(
status=CheckStatus.PASS,
message=f"ACP configured with approval_mode={config['approval_mode']}",
details=config
)
@register_check(
id="acp_runtime",
title="ACP Runtime Check",
category=CheckCategory.TOOLS,
description="Check if ACP runtime can start",
requires_deep=True
)
def check_acp_runtime() -> CheckResult:
"""Check if ACP runtime can start."""
try:
from praisonai.cli.features.interactive_runtime import create_runtime
async def _check():
runtime = create_runtime(
workspace=".",
lsp=False,
acp=True
)
try:
await runtime.start()
ready = runtime.acp_ready
error = runtime._acp_state.error
read_only = runtime.read_only
return ready, error, read_only
finally:
await runtime.stop()
ready, error, read_only = asyncio.run(_check())
if ready:
return CheckResult(
status=CheckStatus.PASS,
message="ACP runtime started successfully",
details={"ready": True, "read_only": read_only}
)
else:
return CheckResult(
status=CheckStatus.WARN,
message=f"ACP runtime failed to start: {error}",
details={"ready": False, "error": error, "read_only": read_only}
)
except Exception as e:
return CheckResult(
status=CheckStatus.FAIL,
message=f"ACP runtime check failed: {e}",
details={"error": str(e)}
)
@register_check(
id="interactive_runtime",
title="Interactive Runtime Check",
category=CheckCategory.TOOLS,
description="Check if full interactive runtime can start",
requires_deep=True
)
def check_interactive_runtime() -> CheckResult:
"""Check if full interactive runtime can start."""
try:
from praisonai.cli.features.interactive_runtime import create_runtime
async def _check():
runtime = create_runtime(
workspace=".",
lsp=True,
acp=True
)
try:
await runtime.start()
status = runtime.get_status()
return status
finally:
await runtime.stop()
status = asyncio.run(_check())
lsp_ready = status.get("lsp", {}).get("ready", False)
acp_ready = status.get("acp", {}).get("ready", False)
if lsp_ready and acp_ready:
return CheckResult(
status=CheckStatus.PASS,
message="Interactive runtime fully operational",
details=status
)
elif lsp_ready or acp_ready:
return CheckResult(
status=CheckStatus.WARN,
message="Interactive runtime partially operational",
details=status
)
else:
return CheckResult(
status=CheckStatus.FAIL,
message="Interactive runtime failed to start",
details=status
)
except Exception as e:
return CheckResult(
status=CheckStatus.FAIL,
message=f"Interactive runtime check failed: {e}",
details={"error": str(e)}
)
def register_acp_checks():
"""Register all ACP checks."""
pass
@@ -0,0 +1,199 @@
"""
LSP health checks for PraisonAI Doctor.
"""
import asyncio
import logging
from ..models import CheckCategory, CheckResult, CheckStatus
from ..registry import register_check
logger = logging.getLogger(__name__)
@register_check(
id="lsp_module",
title="LSP Module Available",
category=CheckCategory.TOOLS,
description="Check if LSP module is installed"
)
def check_lsp_module() -> CheckResult:
"""Check if LSP module is available."""
try:
import importlib.util
spec = importlib.util.find_spec("praisonaiagents.lsp")
if spec is not None:
return CheckResult(
status=CheckStatus.PASS,
message="LSP module is available",
details={"module": "praisonaiagents.lsp"}
)
else:
return CheckResult(
status=CheckStatus.WARN,
message="LSP module not found",
details={"hint": "Install praisonaiagents with LSP support"}
)
except Exception as e:
return CheckResult(
status=CheckStatus.FAIL,
message=f"Error checking LSP module: {e}",
details={"error": str(e)}
)
@register_check(
id="lsp_client",
title="LSP Client Import",
category=CheckCategory.TOOLS,
description="Check if LSP client can be imported"
)
def check_lsp_client() -> CheckResult:
"""Check if LSP client can be imported."""
try:
from praisonaiagents.lsp import LSPClient # noqa: F401
return CheckResult(
status=CheckStatus.PASS,
message="LSP client can be imported",
details={"class": "LSPClient"}
)
except ImportError as e:
return CheckResult(
status=CheckStatus.WARN,
message=f"LSP client import failed: {e}",
details={"error": str(e)}
)
except Exception as e:
return CheckResult(
status=CheckStatus.FAIL,
message=f"Error importing LSP client: {e}",
details={"error": str(e)}
)
@register_check(
id="lsp_server_python",
title="Python LSP Server",
category=CheckCategory.TOOLS,
description="Check if Python language server is available"
)
def check_lsp_server_python() -> CheckResult:
"""Check if Python language server is available."""
import shutil
# Check for common Python language servers
servers = [
("pyright-langserver", "pyright"),
("pylsp", "python-lsp-server"),
("pyls", "python-language-server"),
]
found = []
for cmd, package in servers:
if shutil.which(cmd):
found.append({"command": cmd, "package": package})
if found:
return CheckResult(
status=CheckStatus.PASS,
message=f"Python LSP server available: {found[0]['command']}",
details={"servers": found}
)
else:
return CheckResult(
status=CheckStatus.WARN,
message="No Python LSP server found",
details={
"hint": "Install with: npm install -g pyright OR pip install python-lsp-server",
"checked": [s[0] for s in servers]
}
)
@register_check(
id="lsp_server_typescript",
title="TypeScript LSP Server",
category=CheckCategory.TOOLS,
description="Check if TypeScript language server is available"
)
def check_lsp_server_typescript() -> CheckResult:
"""Check if TypeScript language server is available."""
import shutil
# Check for TypeScript language server
if shutil.which("typescript-language-server"):
return CheckResult(
status=CheckStatus.PASS,
message="TypeScript LSP server available",
details={"command": "typescript-language-server"}
)
elif shutil.which("tsserver"):
return CheckResult(
status=CheckStatus.PASS,
message="TypeScript server available (tsserver)",
details={"command": "tsserver"}
)
else:
return CheckResult(
status=CheckStatus.INFO,
message="TypeScript LSP server not found (optional)",
details={
"hint": "Install with: npm install -g typescript-language-server typescript"
}
)
@register_check(
id="lsp_runtime",
title="LSP Runtime Check",
category=CheckCategory.TOOLS,
description="Check if LSP runtime can start",
requires_deep=True
)
def check_lsp_runtime() -> CheckResult:
"""Check if LSP runtime can start."""
try:
from praisonai.cli.features.interactive_runtime import create_runtime
async def _check():
runtime = create_runtime(
workspace=".",
lsp=True,
acp=False
)
try:
await runtime.start()
ready = runtime.lsp_ready
error = runtime._lsp_state.error
return ready, error
finally:
await runtime.stop()
ready, error = asyncio.run(_check())
if ready:
return CheckResult(
status=CheckStatus.PASS,
message="LSP runtime started successfully",
details={"ready": True}
)
else:
return CheckResult(
status=CheckStatus.WARN,
message=f"LSP runtime failed to start: {error}",
details={"ready": False, "error": error}
)
except Exception as e:
return CheckResult(
status=CheckStatus.FAIL,
message=f"LSP runtime check failed: {e}",
details={"error": str(e)}
)
def register_lsp_checks():
"""Register all LSP checks."""
pass
@@ -0,0 +1,585 @@
"""
Interactive Runtime for PraisonAI.
Provides the core runtime that powers both TUI interactive mode and debug non-interactive mode.
Manages LSP and ACP subsystem lifecycle with graceful degradation.
"""
import asyncio
import logging
import time
from dataclasses import dataclass, field
from enum import Enum
from pathlib import Path
from typing import Any, Dict, List, Optional
logger = logging.getLogger(__name__)
class SubsystemStatus(Enum):
"""Status of a subsystem."""
NOT_STARTED = "not_started"
STARTING = "starting"
READY = "ready"
FAILED = "failed"
STOPPED = "stopped"
@dataclass
class SubsystemState:
"""State of a subsystem."""
status: SubsystemStatus = SubsystemStatus.NOT_STARTED
error: Optional[str] = None
start_time: Optional[float] = None
ready_time: Optional[float] = None
@dataclass
class RuntimeConfig:
"""Configuration for InteractiveRuntime."""
workspace: str = "."
lsp_enabled: bool = True
acp_enabled: bool = True
approval_mode: str = "manual" # manual, auto, scoped
trace_enabled: bool = False
trace_file: Optional[str] = None
json_output: bool = False
timeout: float = 60.0
model: Optional[str] = None
verbose: bool = False
@dataclass
class TraceEntry:
"""A single trace entry."""
timestamp: float
category: str # lsp, acp, tool, llm, file
action: str
params: Dict[str, Any] = field(default_factory=dict)
result: Optional[Any] = None
duration_ms: Optional[float] = None
error: Optional[str] = None
@dataclass
class RuntimeTrace:
"""Trace of a runtime session."""
version: str = "1.0"
start_time: float = field(default_factory=time.time)
end_time: Optional[float] = None
config: Optional[RuntimeConfig] = None
entries: List[TraceEntry] = field(default_factory=list)
def add_entry(self, category: str, action: str, params: Dict[str, Any] = None,
result: Any = None, duration_ms: float = None, error: str = None):
"""Add a trace entry."""
self.entries.append(TraceEntry(
timestamp=time.time(),
category=category,
action=action,
params=params or {},
result=result,
duration_ms=duration_ms,
error=error
))
def to_dict(self) -> Dict[str, Any]:
"""Convert trace to dictionary."""
return {
"version": self.version,
"start_time": self.start_time,
"end_time": self.end_time,
"config": {
"workspace": self.config.workspace if self.config else None,
"lsp_enabled": self.config.lsp_enabled if self.config else None,
"acp_enabled": self.config.acp_enabled if self.config else None,
"approval_mode": self.config.approval_mode if self.config else None,
} if self.config else None,
"entries": [
{
"timestamp": e.timestamp,
"category": e.category,
"action": e.action,
"params": e.params,
"result": e.result,
"duration_ms": e.duration_ms,
"error": e.error
}
for e in self.entries
]
}
class InteractiveRuntime:
"""
Core runtime for interactive coding assistant.
Manages:
- LSP subsystem for code intelligence
- ACP subsystem for action orchestration
- Trace collection for debugging
- Graceful degradation when subsystems fail
"""
def __init__(self, config: RuntimeConfig = None):
"""Initialize the runtime."""
self.config = config or RuntimeConfig()
self._lsp_state = SubsystemState()
self._acp_state = SubsystemState()
self._lsp_client = None
self._acp_session = None
self._trace = RuntimeTrace(config=self.config) if self.config.trace_enabled else None
self._started = False
self._read_only = False # Enforced when ACP fails
@property
def lsp_ready(self) -> bool:
"""Check if LSP is ready."""
return self._lsp_state.status == SubsystemStatus.READY
@property
def acp_ready(self) -> bool:
"""Check if ACP is ready."""
return self._acp_state.status == SubsystemStatus.READY
@property
def read_only(self) -> bool:
"""Check if runtime is in read-only mode."""
return self._read_only or not self.acp_ready
def get_status(self) -> Dict[str, Any]:
"""Get runtime status."""
return {
"started": self._started,
"workspace": self.config.workspace,
"lsp": {
"enabled": self.config.lsp_enabled,
"status": self._lsp_state.status.value,
"ready": self.lsp_ready,
"error": self._lsp_state.error
},
"acp": {
"enabled": self.config.acp_enabled,
"status": self._acp_state.status.value,
"ready": self.acp_ready,
"error": self._acp_state.error
},
"read_only": self.read_only,
"approval_mode": self.config.approval_mode
}
async def start(self) -> Dict[str, Any]:
"""
Start the runtime and all enabled subsystems.
Returns status dict with subsystem states.
"""
if self._started:
return self.get_status()
workspace = Path(self.config.workspace).resolve()
if not workspace.exists():
workspace.mkdir(parents=True, exist_ok=True)
self.config.workspace = str(workspace)
# Start subsystems in parallel
tasks = []
if self.config.lsp_enabled:
tasks.append(self._start_lsp())
if self.config.acp_enabled:
tasks.append(self._start_acp())
if tasks:
await asyncio.gather(*tasks, return_exceptions=True)
# Enforce read-only if ACP failed
if self.config.acp_enabled and not self.acp_ready:
self._read_only = True
logger.warning("ACP unavailable - runtime is READ-ONLY")
self._started = True
if self._trace:
self._trace.add_entry(
category="runtime",
action="start",
params={"workspace": self.config.workspace},
result=self.get_status()
)
return self.get_status()
async def _start_lsp(self):
"""Start LSP subsystem."""
self._lsp_state.status = SubsystemStatus.STARTING
self._lsp_state.start_time = time.time()
try:
# Lazy import LSP client
from praisonaiagents.lsp import LSPClient
# Detect language from workspace
language = self._detect_workspace_language()
self._lsp_client = LSPClient(
language=language,
root_uri=f"file://{self.config.workspace}"
)
await self._lsp_client.start()
# Wait for initialization
if self._lsp_client.is_running:
self._lsp_state.status = SubsystemStatus.READY
self._lsp_state.ready_time = time.time()
logger.info(f"LSP ready for {language}")
else:
raise RuntimeError("LSP client failed to start")
except ImportError as e:
self._lsp_state.status = SubsystemStatus.FAILED
self._lsp_state.error = f"LSP module not available: {e}"
logger.warning(self._lsp_state.error)
except Exception as e:
self._lsp_state.status = SubsystemStatus.FAILED
self._lsp_state.error = str(e)
logger.warning(f"LSP failed to start: {e}")
async def _start_acp(self):
"""Start ACP subsystem (in-process session)."""
self._acp_state.status = SubsystemStatus.STARTING
self._acp_state.start_time = time.time()
try:
# Lazy import ACP
from praisonai.acp.session import ACPSession
from pathlib import Path
# Create a session for the workspace
self._acp_session = ACPSession.create(
workspace=Path(self.config.workspace)
)
self._acp_session.mode = self.config.approval_mode
self._acp_state.status = SubsystemStatus.READY
self._acp_state.ready_time = time.time()
logger.info("ACP session ready")
except ImportError as e:
self._acp_state.status = SubsystemStatus.FAILED
self._acp_state.error = f"ACP module not available: {e}"
logger.warning(self._acp_state.error)
except Exception as e:
self._acp_state.status = SubsystemStatus.FAILED
self._acp_state.error = str(e)
logger.warning(f"ACP failed to start: {e}")
async def stop(self):
"""Stop the runtime and all subsystems."""
if self._lsp_client:
try:
await self._lsp_client.stop()
except Exception as e:
logger.warning(f"Error stopping LSP: {e}")
self._lsp_state.status = SubsystemStatus.STOPPED
if self._acp_session:
# ACPSession is a dataclass, no close method needed
self._acp_session = None
self._acp_state.status = SubsystemStatus.STOPPED
if self._trace:
self._trace.end_time = time.time()
self._trace.add_entry(
category="runtime",
action="stop",
result=self.get_status()
)
self._started = False
def _detect_workspace_language(self) -> str:
"""Detect primary language in workspace."""
workspace = Path(self.config.workspace)
# Count files by extension
extensions = {}
for ext in [".py", ".js", ".ts", ".tsx", ".jsx", ".go", ".rs", ".java"]:
count = len(list(workspace.rglob(f"*{ext}")))
if count > 0:
extensions[ext] = count
if not extensions:
return "python" # Default
# Map extensions to languages
ext_to_lang = {
".py": "python",
".js": "javascript",
".ts": "typescript",
".tsx": "typescript",
".jsx": "javascript",
".go": "go",
".rs": "rust",
".java": "java"
}
# Return language with most files
top_ext = max(extensions, key=extensions.get)
return ext_to_lang.get(top_ext, "python")
# LSP Operations
async def lsp_get_symbols(self, file_path: str) -> List[Dict[str, Any]]:
"""Get symbols in a file via LSP."""
if not self.lsp_ready:
return []
start = time.time()
try:
# Use workspace symbols or document symbols
result = await self._lsp_client.get_document_symbols(file_path)
duration = (time.time() - start) * 1000
if self._trace:
self._trace.add_entry(
category="lsp",
action="get_symbols",
params={"file": file_path},
result={"count": len(result) if result else 0},
duration_ms=duration
)
return result or []
except Exception as e:
if self._trace:
self._trace.add_entry(
category="lsp",
action="get_symbols",
params={"file": file_path},
error=str(e)
)
return []
async def lsp_get_definition(self, file_path: str, line: int, col: int) -> List[Dict[str, Any]]:
"""Get definition location via LSP."""
if not self.lsp_ready:
return []
start = time.time()
try:
result = await self._lsp_client.get_definition(file_path, line, col)
duration = (time.time() - start) * 1000
if self._trace:
self._trace.add_entry(
category="lsp",
action="get_definition",
params={"file": file_path, "line": line, "col": col},
result={"count": len(result) if result else 0},
duration_ms=duration
)
return [loc.__dict__ if hasattr(loc, '__dict__') else loc for loc in (result or [])]
except Exception as e:
if self._trace:
self._trace.add_entry(
category="lsp",
action="get_definition",
params={"file": file_path, "line": line, "col": col},
error=str(e)
)
return []
async def lsp_get_references(self, file_path: str, line: int, col: int) -> List[Dict[str, Any]]:
"""Get references via LSP."""
if not self.lsp_ready:
return []
start = time.time()
try:
result = await self._lsp_client.get_references(file_path, line, col)
duration = (time.time() - start) * 1000
if self._trace:
self._trace.add_entry(
category="lsp",
action="get_references",
params={"file": file_path, "line": line, "col": col},
result={"count": len(result) if result else 0},
duration_ms=duration
)
return [loc.__dict__ if hasattr(loc, '__dict__') else loc for loc in (result or [])]
except Exception as e:
if self._trace:
self._trace.add_entry(
category="lsp",
action="get_references",
params={"file": file_path, "line": line, "col": col},
error=str(e)
)
return []
async def lsp_get_diagnostics(self, file_path: str = None) -> List[Dict[str, Any]]:
"""Get diagnostics via LSP."""
if not self.lsp_ready:
return []
start = time.time()
try:
if file_path:
result = await self._lsp_client.get_diagnostics(file_path)
else:
result = self._lsp_client._diagnostics # All cached diagnostics
duration = (time.time() - start) * 1000
if self._trace:
self._trace.add_entry(
category="lsp",
action="get_diagnostics",
params={"file": file_path},
result={"count": len(result) if result else 0},
duration_ms=duration
)
return [d.__dict__ if hasattr(d, '__dict__') else d for d in (result or [])]
except Exception as e:
if self._trace:
self._trace.add_entry(
category="lsp",
action="get_diagnostics",
params={"file": file_path},
error=str(e)
)
return []
# ACP Operations
async def acp_create_plan(self, prompt: str) -> Dict[str, Any]:
"""Create an action plan via ACP session tracking."""
if not self.acp_ready:
return {"error": "ACP not available", "read_only": True}
start = time.time()
try:
# ACP session tracks the plan but doesn't create it
# Return empty steps - ActionOrchestrator handles actual planning
result = {"steps": [], "prompt": prompt}
duration = (time.time() - start) * 1000
if self._trace:
self._trace.add_entry(
category="acp",
action="create_plan",
params={"prompt": prompt[:100]},
result=result,
duration_ms=duration
)
return result
except Exception as e:
if self._trace:
self._trace.add_entry(
category="acp",
action="create_plan",
params={"prompt": prompt[:100]},
error=str(e)
)
return {"error": str(e)}
async def acp_apply_plan(self, plan: Dict[str, Any], auto_approve: bool = False) -> Dict[str, Any]:
"""Apply an action plan via ACP session tracking."""
if self.read_only:
return {"error": "Runtime is read-only", "applied": False}
start = time.time()
try:
# Track the action in session
if self._acp_session:
self._acp_session.add_tool_call(
tool_call_id=f"plan_{int(time.time())}",
title="apply_plan",
status="applied" if auto_approve else "pending"
)
result = {"applied": True, "auto_approve": auto_approve}
duration = (time.time() - start) * 1000
if self._trace:
self._trace.add_entry(
category="acp",
action="apply_plan",
params={"plan_id": plan.get("id"), "auto_approve": auto_approve},
result=result,
duration_ms=duration
)
return result
except Exception as e:
if self._trace:
self._trace.add_entry(
category="acp",
action="apply_plan",
params={"plan_id": plan.get("id")},
error=str(e)
)
return {"error": str(e), "applied": False}
def get_trace(self) -> Optional[RuntimeTrace]:
"""Get the runtime trace."""
return self._trace
def save_trace(self, path: str = None):
"""Save trace to file."""
if not self._trace:
return
import json
path = path or self.config.trace_file or "praisonai_trace.json"
with open(path, "w") as f:
json.dump(self._trace.to_dict(), f, indent=2, default=str)
logger.info(f"Trace saved to {path}")
def create_runtime(
workspace: str = ".",
lsp: bool = True,
acp: bool = True,
approval: str = "manual",
trace: bool = False,
trace_file: str = None,
json_output: bool = False,
timeout: float = 60.0,
model: str = None,
verbose: bool = False
) -> InteractiveRuntime:
"""
Factory function to create an InteractiveRuntime.
Args:
workspace: Workspace root directory
lsp: Enable LSP subsystem
acp: Enable ACP subsystem
approval: Approval mode (manual, auto, scoped)
trace: Enable trace collection
trace_file: Path to save trace
json_output: Enable JSON output mode
timeout: Operation timeout
model: LLM model to use
verbose: Enable verbose output
Returns:
Configured InteractiveRuntime instance
"""
config = RuntimeConfig(
workspace=workspace,
lsp_enabled=lsp,
acp_enabled=acp,
approval_mode=approval,
trace_enabled=trace,
trace_file=trace_file,
json_output=json_output,
timeout=timeout,
model=model,
verbose=verbose
)
return InteractiveRuntime(config)
@@ -0,0 +1,225 @@
"""
LSP CLI handler for PraisonAI.
Provides LSP service lifecycle commands:
- lsp start: Start LSP server(s)
- lsp stop: Stop LSP server(s)
- lsp status: Show status
- lsp logs: Show recent logs
"""
import argparse
import asyncio
import json
import logging
from typing import List
logger = logging.getLogger(__name__)
class LSPHandler:
"""Handler for LSP CLI commands."""
@staticmethod
def setup_subparser(subparsers) -> None:
"""Set up the lsp subcommand parser."""
lsp_parser = subparsers.add_parser(
"lsp",
help="LSP service lifecycle management",
description="Manage Language Server Protocol servers"
)
lsp_subparsers = lsp_parser.add_subparsers(dest="lsp_command", help="LSP subcommands")
# lsp start
start_parser = lsp_subparsers.add_parser("start", help="Start LSP server(s)")
start_parser.add_argument(
"--language", "-l",
type=str,
help="Language to start (default: auto-detect)"
)
start_parser.add_argument(
"--workspace", "-w",
type=str,
default=".",
help="Workspace root directory"
)
start_parser.add_argument("--json", action="store_true", help="Output JSON")
start_parser.set_defaults(func=LSPHandler.handle_start)
# lsp stop
stop_parser = lsp_subparsers.add_parser("stop", help="Stop LSP server(s)")
stop_parser.add_argument("--json", action="store_true", help="Output JSON")
stop_parser.set_defaults(func=LSPHandler.handle_stop)
# lsp status
status_parser = lsp_subparsers.add_parser("status", help="Show LSP status")
status_parser.add_argument(
"--workspace", "-w",
type=str,
default=".",
help="Workspace root directory"
)
status_parser.add_argument("--json", action="store_true", help="Output JSON")
status_parser.set_defaults(func=LSPHandler.handle_status)
# lsp logs
logs_parser = lsp_subparsers.add_parser("logs", help="Show recent logs")
logs_parser.add_argument(
"--tail", "-n",
type=int,
default=50,
help="Number of lines to show"
)
logs_parser.add_argument("--json", action="store_true", help="Output JSON")
logs_parser.set_defaults(func=LSPHandler.handle_logs)
@staticmethod
def handle_start(args: argparse.Namespace) -> int:
"""Handle lsp start command."""
return asyncio.run(LSPHandler._run_start(args))
@staticmethod
async def _run_start(args: argparse.Namespace) -> int:
"""Start LSP server."""
from .interactive_runtime import create_runtime
runtime = create_runtime(
workspace=args.workspace,
lsp=True,
acp=False
)
try:
await runtime.start()
result = {
"action": "start",
"lsp_ready": runtime.lsp_ready,
"lsp_status": runtime._lsp_state.status.value,
"workspace": args.workspace,
"language": getattr(args, 'language', None) or "auto-detected"
}
if runtime._lsp_state.error:
result["error"] = runtime._lsp_state.error
LSPHandler._output_result(result, args.json)
# Keep running if started successfully
if runtime.lsp_ready:
if not args.json:
print("\nLSP server running. Press Ctrl+C to stop.")
try:
while True:
await asyncio.sleep(1)
except KeyboardInterrupt:
pass
return 0 if runtime.lsp_ready else 1
finally:
await runtime.stop()
@staticmethod
def handle_stop(args: argparse.Namespace) -> int:
"""Handle lsp stop command."""
result = {
"action": "stop",
"message": "LSP servers stopped"
}
LSPHandler._output_result(result, args.json)
return 0
@staticmethod
def handle_status(args: argparse.Namespace) -> int:
"""Handle lsp status command."""
return asyncio.run(LSPHandler._run_status(args))
@staticmethod
async def _run_status(args: argparse.Namespace) -> int:
"""Check LSP status."""
from .interactive_runtime import create_runtime
runtime = create_runtime(
workspace=args.workspace,
lsp=True,
acp=False
)
try:
await runtime.start()
result = {
"lsp_enabled": True,
"lsp_ready": runtime.lsp_ready,
"lsp_status": runtime._lsp_state.status.value,
"workspace": args.workspace
}
if runtime._lsp_state.error:
result["error"] = runtime._lsp_state.error
LSPHandler._output_result(result, args.json)
return 0 if runtime.lsp_ready else 1
finally:
await runtime.stop()
@staticmethod
def handle_logs(args: argparse.Namespace) -> int:
"""Handle lsp logs command."""
result = {
"action": "logs",
"tail": args.tail,
"logs": [],
"message": "LSP logging not yet implemented"
}
LSPHandler._output_result(result, args.json)
return 0
@staticmethod
def _output_result(result: dict, as_json: bool) -> None:
"""Output result in appropriate format."""
if as_json:
print(json.dumps(result, indent=2, default=str))
else:
for key, value in result.items():
print(f"{key}: {value}")
def run_lsp_command(args: List[str]) -> int:
"""Run lsp command from CLI."""
parser = argparse.ArgumentParser(prog="praisonai lsp")
subparsers = parser.add_subparsers(dest="lsp_command", help="LSP subcommands")
# lsp start
start_parser = subparsers.add_parser("start", help="Start LSP server(s)")
start_parser.add_argument("--language", "-l", type=str)
start_parser.add_argument("--workspace", "-w", type=str, default=".")
start_parser.add_argument("--json", action="store_true")
start_parser.set_defaults(func=LSPHandler.handle_start)
# lsp stop
stop_parser = subparsers.add_parser("stop", help="Stop LSP server(s)")
stop_parser.add_argument("--json", action="store_true")
stop_parser.set_defaults(func=LSPHandler.handle_stop)
# lsp status
status_parser = subparsers.add_parser("status", help="Show LSP status")
status_parser.add_argument("--workspace", "-w", type=str, default=".")
status_parser.add_argument("--json", action="store_true")
status_parser.set_defaults(func=LSPHandler.handle_status)
# lsp logs
logs_parser = subparsers.add_parser("logs", help="Show recent logs")
logs_parser.add_argument("--tail", "-n", type=int, default=50)
logs_parser.add_argument("--json", action="store_true")
logs_parser.set_defaults(func=LSPHandler.handle_logs)
parsed = parser.parse_args(args)
if hasattr(parsed, 'func'):
return parsed.func(parsed)
else:
parser.print_help()
return 0
+34 -2
View File
@@ -254,6 +254,11 @@ class PraisonAI:
invocation_cmd = "praisonai"
version_string = f"PraisonAI version {__version__}"
# Handle -p/--prompt flag - treat as direct prompt
if getattr(args, 'prompt_flag', None):
args.direct_prompt = args.prompt_flag
args.command = None
# Handle --interactive flag - start interactive TUI mode
if getattr(args, 'interactive', False):
self._start_interactive_mode(args)
@@ -261,7 +266,7 @@ class PraisonAI:
# Handle --chat-mode flag - run single prompt in interactive style (non-interactive)
if getattr(args, 'chat_mode', False):
prompt = args.command or getattr(args, 'direct_prompt', None)
prompt = args.command or getattr(args, 'direct_prompt', None) or getattr(args, 'prompt_flag', None)
if prompt:
return self._run_chat_mode(prompt, args)
@@ -698,7 +703,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', 'debug', 'lsp', 'diag']
parser = argparse.ArgumentParser(prog="praisonai", description="praisonAI command-line interface")
parser.add_argument("--framework", choices=["crewai", "autogen", "praisonai"], help="Specify the framework")
@@ -818,6 +823,9 @@ class PraisonAI:
parser.add_argument("--interactive", "-i", action="store_true", help="Start interactive terminal mode with slash commands")
parser.add_argument("--chat-mode", "--chat", action="store_true", dest="chat_mode", help="Run single prompt in interactive style (non-interactive, for testing)")
# Direct prompt flag - alternative to positional command
parser.add_argument("-p", "--prompt", type=str, dest="prompt_flag", help="Direct prompt to execute (alternative to positional argument)")
# Autonomy Mode - control AI action approval
parser.add_argument("--autonomy", type=str, choices=["suggest", "auto_edit", "full_auto"], help="Set autonomy mode for AI actions")
@@ -898,6 +906,30 @@ 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 debug command - Debug and test interactive flows
if args.command == 'debug':
from praisonai.cli.features.debug import run_debug_command
exit_code = run_debug_command(unknown_args)
sys.exit(exit_code)
# Handle lsp command - LSP service lifecycle
if args.command == 'lsp':
from praisonai.cli.features.lsp_cli import run_lsp_command
exit_code = run_lsp_command(unknown_args)
sys.exit(exit_code)
# Handle diag command - Diagnostics export
if args.command == 'diag':
from praisonai.cli.features.diag import run_diag_command
exit_code = run_diag_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:
+1 -1
View File
@@ -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.9.0 gunicorn markdown\n")
file.write("EXPOSE 8080\n")
file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
+92 -1
View File
@@ -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',
]
+52 -24
View File
@@ -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})
+32 -17
View File
@@ -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})
+6 -2
View File
@@ -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"""
+1 -1
View File
@@ -1 +1 @@
__version__ = "2.8.3"
__version__ = "2.9.0"
+7 -4
View File
@@ -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"
@@ -0,0 +1 @@
"""Integration tests for ACP module."""
@@ -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}")
@@ -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
+1
View File
@@ -0,0 +1 @@
"""Unit tests for ACP module."""
+166
View File
@@ -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
+244
View File
@@ -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"
@@ -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
@@ -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"])
+89 -69
View File
@@ -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]