mirror of
https://github.com/MervinPraison/PraisonAI.git
synced 2026-11-09 11:51:37 -10:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
4ffb9d6e3f | ||
|
|
b686e5b00f |
No files matched your search
@@ -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.9.0" \
|
||||
"praisonai>=2.9.2" \
|
||||
"praisonai[chat]" \
|
||||
"embedchain[github,youtube]"
|
||||
|
||||
|
||||
@@ -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.9.0" \
|
||||
"praisonai>=2.9.2" \
|
||||
"praisonai[ui]" \
|
||||
"praisonai[chat]" \
|
||||
"praisonai[realtime]" \
|
||||
|
||||
@@ -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.9.0" \
|
||||
"praisonai>=2.9.2" \
|
||||
"praisonai[ui]" \
|
||||
"praisonai[crewai]"
|
||||
|
||||
|
||||
@@ -0,0 +1,135 @@
|
||||
"""
|
||||
Agent-Centric Tools Example
|
||||
|
||||
This example demonstrates how to use LSP/ACP-powered tools that make
|
||||
the Agent the central orchestrator for file operations.
|
||||
|
||||
How to run:
|
||||
export OPENAI_API_KEY=your_key
|
||||
python examples/agent_tools/agent_centric_example.py
|
||||
|
||||
Prerequisites:
|
||||
pip install praisonai praisonaiagents
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import os
|
||||
import sys
|
||||
import tempfile
|
||||
import shutil
|
||||
|
||||
# Add paths for development
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../../src/praisonai'))
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../../src/praisonai-agents'))
|
||||
|
||||
|
||||
async def main():
|
||||
print("=" * 60)
|
||||
print("AGENT-CENTRIC TOOLS EXAMPLE")
|
||||
print("=" * 60)
|
||||
print()
|
||||
|
||||
# Create temporary workspace
|
||||
workspace = tempfile.mkdtemp(prefix="praisonai_example_")
|
||||
print(f"Workspace: {workspace}")
|
||||
print()
|
||||
|
||||
try:
|
||||
# Import after path setup
|
||||
from praisonai.cli.features import (
|
||||
create_agent_centric_tools,
|
||||
InteractiveRuntime,
|
||||
RuntimeConfig
|
||||
)
|
||||
from praisonaiagents import Agent
|
||||
|
||||
# 1. Create runtime with ACP enabled
|
||||
print("1. Creating InteractiveRuntime with ACP...")
|
||||
config = RuntimeConfig(
|
||||
workspace=workspace,
|
||||
lsp_enabled=False, # Skip LSP for this example
|
||||
acp_enabled=True,
|
||||
approval_mode="auto" # Auto-approve for demo
|
||||
)
|
||||
runtime = InteractiveRuntime(config)
|
||||
status = await runtime.start()
|
||||
|
||||
print(f" ACP ready: {runtime.acp_ready}")
|
||||
print(f" Read-only: {runtime.read_only}")
|
||||
print()
|
||||
|
||||
# 2. Create agent-centric tools
|
||||
print("2. Creating agent-centric tools...")
|
||||
tools = create_agent_centric_tools(runtime)
|
||||
print(f" Tools created: {len(tools)}")
|
||||
for tool in tools:
|
||||
print(f" - {tool.__name__}")
|
||||
print()
|
||||
|
||||
# 3. Create Agent with ACP-powered tools
|
||||
print("3. Creating Agent with ACP-powered tools...")
|
||||
agent = Agent(
|
||||
name="FileAgent",
|
||||
instructions="""You help create and manage files.
|
||||
Use acp_create_file to create files.
|
||||
Use read_file to read files.
|
||||
Use list_files to list directory contents.""",
|
||||
tools=tools,
|
||||
verbose=True
|
||||
)
|
||||
print()
|
||||
|
||||
# 4. Ask agent to create a file
|
||||
print("4. Asking agent to create a Python file...")
|
||||
print("-" * 40)
|
||||
result = agent.start(
|
||||
"Create a Python file called calculator.py with add and subtract functions"
|
||||
)
|
||||
print("-" * 40)
|
||||
print(f"Agent response: {result[:200]}..." if len(str(result)) > 200 else f"Agent response: {result}")
|
||||
print()
|
||||
|
||||
# 5. Verify file was created
|
||||
print("5. Verifying file creation...")
|
||||
calc_path = os.path.join(workspace, "calculator.py")
|
||||
if os.path.exists(calc_path):
|
||||
print(f" ✓ File created at {calc_path}")
|
||||
with open(calc_path) as f:
|
||||
content = f.read()
|
||||
print(f" Content preview:")
|
||||
for line in content.split('\n')[:10]:
|
||||
print(f" {line}")
|
||||
else:
|
||||
# Check what files exist
|
||||
files = os.listdir(workspace)
|
||||
print(f" Files in workspace: {files}")
|
||||
for f in files:
|
||||
fp = os.path.join(workspace, f)
|
||||
if os.path.isfile(fp):
|
||||
print(f" ✓ Found: {f}")
|
||||
print()
|
||||
|
||||
# 6. Ask agent to read the file
|
||||
print("6. Asking agent to read the file...")
|
||||
print("-" * 40)
|
||||
result2 = agent.start("Read the calculator.py file and tell me what functions it has")
|
||||
print("-" * 40)
|
||||
print(f"Agent response: {result2[:300]}..." if len(str(result2)) > 300 else f"Agent response: {result2}")
|
||||
print()
|
||||
|
||||
# Cleanup
|
||||
await runtime.stop()
|
||||
|
||||
finally:
|
||||
# Clean up workspace
|
||||
shutil.rmtree(workspace, ignore_errors=True)
|
||||
print(f"Cleaned up workspace: {workspace}")
|
||||
|
||||
print()
|
||||
print("=" * 60)
|
||||
print("EXAMPLE COMPLETE")
|
||||
print("=" * 60)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
@@ -0,0 +1,130 @@
|
||||
"""
|
||||
LSP Code Intelligence Example
|
||||
|
||||
This example demonstrates how to use LSP-powered tools through an Agent
|
||||
for semantic code analysis (listing symbols, finding definitions, etc.).
|
||||
|
||||
How to run:
|
||||
export OPENAI_API_KEY=your_key
|
||||
python examples/lsp/lsp_code_analysis_example.py
|
||||
|
||||
Prerequisites:
|
||||
pip install praisonai praisonaiagents
|
||||
pip install python-lsp-server # Optional, for full LSP support
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import os
|
||||
import sys
|
||||
|
||||
# Add paths for development
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../../src/praisonai'))
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../../src/praisonai-agents'))
|
||||
|
||||
|
||||
async def main():
|
||||
print("=" * 60)
|
||||
print("LSP CODE INTELLIGENCE EXAMPLE")
|
||||
print("=" * 60)
|
||||
print()
|
||||
|
||||
# Import after path setup
|
||||
from praisonai.cli.features import (
|
||||
create_agent_centric_tools,
|
||||
InteractiveRuntime,
|
||||
RuntimeConfig
|
||||
)
|
||||
from praisonaiagents import Agent
|
||||
|
||||
# Use the praisonaiagents source as workspace for analysis
|
||||
workspace = os.path.join(
|
||||
os.path.dirname(__file__),
|
||||
'../../src/praisonai-agents/praisonaiagents'
|
||||
)
|
||||
workspace = os.path.abspath(workspace)
|
||||
|
||||
print(f"Workspace: {workspace}")
|
||||
print()
|
||||
|
||||
# 1. Create runtime with LSP enabled
|
||||
print("1. Creating InteractiveRuntime with LSP...")
|
||||
config = RuntimeConfig(
|
||||
workspace=workspace,
|
||||
lsp_enabled=True, # Enable LSP for code intelligence
|
||||
acp_enabled=False, # Disable ACP for this example
|
||||
approval_mode="auto"
|
||||
)
|
||||
runtime = InteractiveRuntime(config)
|
||||
status = await runtime.start()
|
||||
|
||||
print(f" LSP enabled: {status['lsp']['enabled']}")
|
||||
print(f" LSP ready: {runtime.lsp_ready}")
|
||||
print(f" LSP status: {status['lsp']['status']}")
|
||||
if not runtime.lsp_ready:
|
||||
print(" Note: LSP not available, will use regex fallback")
|
||||
print()
|
||||
|
||||
# 2. Create agent-centric tools
|
||||
print("2. Creating agent-centric tools...")
|
||||
tools = create_agent_centric_tools(runtime)
|
||||
|
||||
# Filter to show only LSP tools
|
||||
lsp_tools = [t for t in tools if t.__name__.startswith('lsp_')]
|
||||
print(f" LSP tools available:")
|
||||
for tool in lsp_tools:
|
||||
print(f" - {tool.__name__}")
|
||||
print()
|
||||
|
||||
# 3. Create Agent with LSP-powered tools
|
||||
print("3. Creating Agent with LSP tools...")
|
||||
agent = Agent(
|
||||
name="CodeAnalyzer",
|
||||
instructions="""You are a code analysis assistant.
|
||||
Use lsp_list_symbols to list functions and classes in files.
|
||||
Use lsp_find_definition to find where symbols are defined.
|
||||
Use lsp_find_references to find where symbols are used.
|
||||
Provide clear, organized summaries of code structure.""",
|
||||
tools=tools,
|
||||
verbose=True
|
||||
)
|
||||
print()
|
||||
|
||||
# 4. Test 1: List symbols in a file
|
||||
print("4. TEST 1: List symbols in agent/agent.py")
|
||||
print("-" * 50)
|
||||
result1 = agent.start("List all the main classes and public methods in agent/agent.py. Give me a brief summary.")
|
||||
print("-" * 50)
|
||||
print(f"Summary: {str(result1)[:300]}...")
|
||||
print()
|
||||
|
||||
# 5. Test 2: Find definition of a symbol
|
||||
print("5. TEST 2: Find where 'Agent' class is defined")
|
||||
print("-" * 50)
|
||||
result2 = agent.start("Where is the Agent class defined? Use lsp_find_definition to find it.")
|
||||
print("-" * 50)
|
||||
print(f"Result: {str(result2)[:300]}...")
|
||||
print()
|
||||
|
||||
# 6. Test 3: Analyze main.py
|
||||
print("6. TEST 3: Analyze main.py structure")
|
||||
print("-" * 50)
|
||||
result3 = agent.start("List all the functions in main.py and briefly describe what the file does.")
|
||||
print("-" * 50)
|
||||
print(f"Analysis: {str(result3)[:300]}...")
|
||||
print()
|
||||
|
||||
# Cleanup
|
||||
await runtime.stop()
|
||||
|
||||
print("=" * 60)
|
||||
print("EXAMPLE COMPLETE")
|
||||
print("=" * 60)
|
||||
print()
|
||||
print("Key takeaways:")
|
||||
print("- LSP tools work through the Agent (agentic LSP)")
|
||||
print("- Falls back to regex if LSP server not available")
|
||||
print("- Returns structured data with citations")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
@@ -0,0 +1,133 @@
|
||||
"""
|
||||
PraisonAI Output UX Examples
|
||||
|
||||
This example demonstrates the various output modes and configurations
|
||||
available in PraisonAI agents.
|
||||
|
||||
Features demonstrated:
|
||||
- verbose=True: Full output with agent info, task, and response panels
|
||||
- verbose=False: Clean output with just the result
|
||||
- Tool call display with consolidated output
|
||||
- Multi-agent output with clear attribution
|
||||
"""
|
||||
|
||||
import sys
|
||||
sys.path.insert(0, '/Users/praison/praisonai-package/src/praisonai-agents')
|
||||
from praisonaiagents import Agent, Task, PraisonAIAgents
|
||||
|
||||
|
||||
def get_weather(city: str) -> str:
|
||||
"""Get the current weather for a city.
|
||||
|
||||
Args:
|
||||
city: The name of the city to get weather for
|
||||
|
||||
Returns:
|
||||
Weather information string
|
||||
"""
|
||||
return f"The weather in {city} is sunny and 72°F"
|
||||
|
||||
|
||||
def example_verbose_true():
|
||||
"""Example with verbose=True (default) - shows full output panels."""
|
||||
print("\n" + "=" * 60)
|
||||
print("EXAMPLE 1: verbose=True (Full Output)")
|
||||
print("=" * 60)
|
||||
|
||||
agent = Agent(
|
||||
name="Assistant",
|
||||
instructions="You are a helpful assistant. Be concise.",
|
||||
verbose=True # Default - shows Agent Info, Task, and Response panels
|
||||
)
|
||||
result = agent.start("What is the capital of France? Answer in one sentence.")
|
||||
print(f"\nReturned value: {result}")
|
||||
|
||||
|
||||
def example_verbose_false():
|
||||
"""Example with verbose=False - clean output, just the result."""
|
||||
print("\n" + "=" * 60)
|
||||
print("EXAMPLE 2: verbose=False (Clean Output)")
|
||||
print("=" * 60)
|
||||
|
||||
agent = Agent(
|
||||
name="Assistant",
|
||||
instructions="You are a helpful assistant. Be concise.",
|
||||
verbose=False # No panels, just returns the result
|
||||
)
|
||||
result = agent.start("What is 2+2? Answer in one word.")
|
||||
print(f"Result: {result}")
|
||||
|
||||
|
||||
def example_with_tools():
|
||||
"""Example showing tool call output."""
|
||||
print("\n" + "=" * 60)
|
||||
print("EXAMPLE 3: Agent with Tools")
|
||||
print("=" * 60)
|
||||
|
||||
agent = Agent(
|
||||
name="Weather Bot",
|
||||
instructions="You help users check the weather. Use the get_weather tool.",
|
||||
tools=[get_weather],
|
||||
verbose=True
|
||||
)
|
||||
result = agent.start("What's the weather in Tokyo?")
|
||||
print(f"\nReturned value: {result}")
|
||||
|
||||
|
||||
def example_multi_agent():
|
||||
"""Example showing multi-agent output with clear attribution."""
|
||||
print("\n" + "=" * 60)
|
||||
print("EXAMPLE 4: Multi-Agent Output")
|
||||
print("=" * 60)
|
||||
|
||||
researcher = Agent(
|
||||
name="Researcher",
|
||||
role="Research Analyst",
|
||||
instructions="You research topics and provide factual information.",
|
||||
verbose=True
|
||||
)
|
||||
|
||||
writer = Agent(
|
||||
name="Writer",
|
||||
role="Content Writer",
|
||||
instructions="You write clear summaries based on research.",
|
||||
verbose=True
|
||||
)
|
||||
|
||||
task1 = Task(
|
||||
description="Research what AI is in one sentence",
|
||||
expected_output="A brief factual description",
|
||||
agent=researcher
|
||||
)
|
||||
|
||||
task2 = Task(
|
||||
description="Write a one-sentence summary about AI",
|
||||
expected_output="A clear summary",
|
||||
agent=writer
|
||||
)
|
||||
|
||||
agents = PraisonAIAgents(
|
||||
agents=[researcher, writer],
|
||||
tasks=[task1, task2],
|
||||
process="sequential",
|
||||
verbose=1
|
||||
)
|
||||
|
||||
result = agents.start()
|
||||
print(f"\nFinal result: {result}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
print("=" * 60)
|
||||
print("PRAISONAI OUTPUT UX EXAMPLES")
|
||||
print("=" * 60)
|
||||
|
||||
# Run examples
|
||||
example_verbose_true()
|
||||
example_verbose_false()
|
||||
example_with_tools()
|
||||
example_multi_agent()
|
||||
|
||||
print("\n" + "=" * 60)
|
||||
print("ALL EXAMPLES COMPLETE")
|
||||
print("=" * 60)
|
||||
@@ -0,0 +1,30 @@
|
||||
"""
|
||||
Basic TUI Usage Example for PraisonAI.
|
||||
|
||||
This example demonstrates how to launch the interactive TUI.
|
||||
"""
|
||||
|
||||
import os
|
||||
|
||||
# Ensure API key is set
|
||||
if not os.environ.get("OPENAI_API_KEY"):
|
||||
print("Please set OPENAI_API_KEY environment variable")
|
||||
exit(1)
|
||||
|
||||
|
||||
def main():
|
||||
"""Launch the TUI."""
|
||||
try:
|
||||
from praisonai.cli.features.tui import run_tui
|
||||
|
||||
# Launch with default settings
|
||||
run_tui()
|
||||
|
||||
except ImportError:
|
||||
print("Textual is required for TUI. Install with:")
|
||||
print(" pip install praisonai[tui]")
|
||||
exit(1)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -0,0 +1,134 @@
|
||||
"""
|
||||
Headless TUI Simulation Example for PraisonAI.
|
||||
|
||||
This example demonstrates how to run TUI simulations without
|
||||
launching the interactive UI - useful for testing and CI.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
from praisonai.cli.features.tui import (
|
||||
TuiOrchestrator,
|
||||
SimulationRunner,
|
||||
MockProvider,
|
||||
)
|
||||
from praisonai.cli.features.tui.orchestrator import OutputMode
|
||||
from praisonai.cli.features.queue import QueueConfig
|
||||
|
||||
|
||||
async def run_simulation():
|
||||
"""Run a headless simulation."""
|
||||
|
||||
# Create orchestrator with mock provider (no real API calls)
|
||||
config = QueueConfig(enable_persistence=False)
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.PRETTY,
|
||||
debug=True,
|
||||
)
|
||||
|
||||
# Create simulation runner
|
||||
runner = SimulationRunner(orchestrator, assert_mode=True)
|
||||
|
||||
# Define simulation script
|
||||
script = {
|
||||
"session_id": "sim-example",
|
||||
"model": "gpt-4o-mini",
|
||||
"steps": [
|
||||
# Navigate to different screens
|
||||
{"action": "navigate", "args": {"screen": "main"}},
|
||||
{"action": "focus", "args": {"widget": "composer"}},
|
||||
|
||||
# Change model
|
||||
{
|
||||
"action": "model",
|
||||
"args": {"model": "gpt-4"},
|
||||
"expected": {"model": "gpt-4"}
|
||||
},
|
||||
|
||||
# Navigate to queue screen
|
||||
{"action": "navigate", "args": {"screen": "queue"}},
|
||||
|
||||
# Take a snapshot
|
||||
{"action": "snapshot"},
|
||||
|
||||
# Navigate back
|
||||
{"action": "navigate", "args": {"screen": "main"}},
|
||||
|
||||
# Small delay
|
||||
{"action": "sleep", "args": {"seconds": 0.5}},
|
||||
]
|
||||
}
|
||||
|
||||
# Run the simulation
|
||||
print("Running headless TUI simulation...")
|
||||
print("=" * 50)
|
||||
|
||||
success = await runner.run_script(script)
|
||||
|
||||
# Print results
|
||||
print("=" * 50)
|
||||
summary = runner.get_summary()
|
||||
|
||||
print(f"\nSimulation {'PASSED' if success else 'FAILED'}")
|
||||
print(f" Assertions passed: {summary['assertions_passed']}")
|
||||
print(f" Assertions failed: {summary['assertions_failed']}")
|
||||
|
||||
if summary['errors']:
|
||||
print("\nErrors:")
|
||||
for error in summary['errors']:
|
||||
print(f" - {error}")
|
||||
|
||||
return success
|
||||
|
||||
|
||||
async def run_with_events():
|
||||
"""Run simulation and capture all events."""
|
||||
|
||||
config = QueueConfig(enable_persistence=False)
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
|
||||
# Capture events
|
||||
events = []
|
||||
orchestrator.add_event_callback(lambda e: events.append(e))
|
||||
|
||||
await orchestrator.start(session_id="event-capture")
|
||||
|
||||
# Perform some actions
|
||||
orchestrator.set_model("gpt-4")
|
||||
orchestrator.navigate_screen("queue")
|
||||
orchestrator.set_focus("queue-panel")
|
||||
orchestrator.navigate_screen("main")
|
||||
|
||||
await orchestrator.stop()
|
||||
|
||||
# Print captured events
|
||||
print("\nCaptured Events:")
|
||||
print("-" * 40)
|
||||
for event in events:
|
||||
print(f" {event.event_type.value}")
|
||||
|
||||
return events
|
||||
|
||||
|
||||
def main():
|
||||
"""Main entry point."""
|
||||
print("PraisonAI TUI Headless Simulation Example")
|
||||
print("=" * 50)
|
||||
|
||||
# Run simulation
|
||||
success = asyncio.run(run_simulation())
|
||||
|
||||
# Run event capture
|
||||
print("\n" + "=" * 50)
|
||||
events = asyncio.run(run_with_events())
|
||||
|
||||
print(f"\nTotal events captured: {len(events)}")
|
||||
|
||||
return 0 if success else 1
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
exit(main())
|
||||
@@ -0,0 +1,122 @@
|
||||
"""
|
||||
Queue Management Example for PraisonAI.
|
||||
|
||||
This example demonstrates how to use the queue system programmatically.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
from praisonai.cli.features.queue import (
|
||||
QueueManager,
|
||||
QueueConfig,
|
||||
QueuedRun,
|
||||
RunState,
|
||||
RunPriority,
|
||||
)
|
||||
|
||||
|
||||
async def main():
|
||||
"""Demonstrate queue management."""
|
||||
|
||||
print("PraisonAI Queue Management Example")
|
||||
print("=" * 50)
|
||||
|
||||
# Create queue configuration
|
||||
config = QueueConfig(
|
||||
max_concurrent_global=2,
|
||||
max_concurrent_per_agent=1,
|
||||
max_queue_size=10,
|
||||
enable_persistence=False, # Use in-memory for demo
|
||||
)
|
||||
|
||||
# Track outputs
|
||||
outputs = {}
|
||||
|
||||
async def on_output(run_id: str, chunk: str):
|
||||
if run_id not in outputs:
|
||||
outputs[run_id] = ""
|
||||
outputs[run_id] += chunk
|
||||
print(f" [{run_id[:8]}] {chunk}", end="", flush=True)
|
||||
|
||||
async def on_complete(run_id: str, run: QueuedRun):
|
||||
print(f"\n [{run_id[:8]}] Completed: {run.state.value}")
|
||||
|
||||
async def on_error(run_id: str, error: Exception):
|
||||
print(f"\n [{run_id[:8]}] Error: {error}")
|
||||
|
||||
# Create queue manager
|
||||
manager = QueueManager(
|
||||
config=config,
|
||||
on_output=on_output,
|
||||
on_complete=on_complete,
|
||||
on_error=on_error,
|
||||
)
|
||||
|
||||
# Start the manager
|
||||
await manager.start(recover=False)
|
||||
print("Queue manager started")
|
||||
|
||||
# Submit some runs with different priorities
|
||||
print("\nSubmitting runs...")
|
||||
|
||||
run1 = await manager.submit(
|
||||
input_content="Low priority task",
|
||||
agent_name="Agent1",
|
||||
priority=RunPriority.LOW,
|
||||
)
|
||||
print(f" Submitted run1: {run1} (LOW priority)")
|
||||
|
||||
run2 = await manager.submit(
|
||||
input_content="High priority task",
|
||||
agent_name="Agent2",
|
||||
priority=RunPriority.HIGH,
|
||||
)
|
||||
print(f" Submitted run2: {run2} (HIGH priority)")
|
||||
|
||||
run3 = await manager.submit(
|
||||
input_content="Normal priority task",
|
||||
agent_name="Agent1",
|
||||
priority=RunPriority.NORMAL,
|
||||
)
|
||||
print(f" Submitted run3: {run3} (NORMAL priority)")
|
||||
|
||||
# Check queue status
|
||||
print(f"\nQueue status:")
|
||||
print(f" Queued: {manager.queued_count}")
|
||||
print(f" Running: {manager.running_count}")
|
||||
|
||||
# List runs
|
||||
runs = manager.list_runs()
|
||||
print(f"\nAll runs ({len(runs)}):")
|
||||
for run in runs:
|
||||
print(f" {run.run_id[:8]}: {run.agent_name} - {run.state.value} ({run.priority.name})")
|
||||
|
||||
# Wait a bit for processing
|
||||
print("\nWaiting for processing...")
|
||||
await asyncio.sleep(2)
|
||||
|
||||
# Cancel a run
|
||||
if runs:
|
||||
cancel_id = runs[-1].run_id
|
||||
print(f"\nCancelling run: {cancel_id[:8]}")
|
||||
await manager.cancel(cancel_id)
|
||||
|
||||
# Get stats
|
||||
stats = manager.get_stats()
|
||||
print(f"\nQueue statistics:")
|
||||
print(f" Total runs: {stats.total_runs}")
|
||||
print(f" Queued: {stats.queued_count}")
|
||||
print(f" Running: {stats.running_count}")
|
||||
print(f" Succeeded: {stats.succeeded_count}")
|
||||
print(f" Failed: {stats.failed_count}")
|
||||
print(f" Cancelled: {stats.cancelled_count}")
|
||||
|
||||
# Stop the manager
|
||||
await manager.stop()
|
||||
print("\nQueue manager stopped")
|
||||
|
||||
print("\n" + "=" * 50)
|
||||
print("Example complete!")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
@@ -150,7 +150,10 @@ def _clean_display_content(content: str, max_length: int = 20000) -> str:
|
||||
return content.strip()
|
||||
|
||||
def display_interaction(message, response, markdown=True, generation_time=None, console=None, agent_name=None, agent_role=None, agent_tools=None, task_name=None, task_description=None, task_id=None):
|
||||
"""Synchronous version of display_interaction."""
|
||||
"""Synchronous version of display_interaction.
|
||||
|
||||
Displays the task/message and response in clean panels with response time in title.
|
||||
"""
|
||||
if console is None:
|
||||
console = Console()
|
||||
|
||||
@@ -179,16 +182,18 @@ def display_interaction(message, response, markdown=True, generation_time=None,
|
||||
task_description=task_description,
|
||||
task_id=task_id
|
||||
)
|
||||
# Rest of the display logic...
|
||||
|
||||
# Build response title with time (Agno-style: "Response (1.2s)")
|
||||
response_title = "Response"
|
||||
if generation_time:
|
||||
console.print(Text(f"Response generated in {generation_time:.1f}s", style="dim"))
|
||||
response_title = f"Response ({generation_time:.1f}s)"
|
||||
|
||||
if markdown:
|
||||
console.print(Panel.fit(Markdown(message), title="Task", border_style="cyan"))
|
||||
console.print(Panel.fit(Markdown(response), title="Response", border_style="cyan"))
|
||||
console.print(Panel.fit(Markdown(response), title=response_title, border_style="cyan"))
|
||||
else:
|
||||
console.print(Panel.fit(Text(message, style="bold green"), title="Task", border_style="cyan"))
|
||||
console.print(Panel.fit(Text(response, style="bold blue"), title="Response", border_style="cyan"))
|
||||
console.print(Panel.fit(Text(response, style="bold blue"), title=response_title, border_style="cyan"))
|
||||
|
||||
def display_self_reflection(message: str, console=None):
|
||||
if not message or not message.strip():
|
||||
@@ -226,7 +231,16 @@ def display_instruction(message: str, console=None, agent_name: str = None, agen
|
||||
if logging.getLogger().getEffectiveLevel() == logging.DEBUG:
|
||||
console.print(Panel.fit(Text(message, style="bold blue"), title="Instruction", border_style="cyan"))
|
||||
|
||||
def display_tool_call(message: str, console=None):
|
||||
def display_tool_call(message: str, console=None, tool_name: str = None, tool_input: dict = None, tool_output: str = None):
|
||||
"""Display tool call information in a consolidated panel.
|
||||
|
||||
Args:
|
||||
message: The tool call message (legacy format)
|
||||
console: Rich console for output
|
||||
tool_name: Name of the tool being called
|
||||
tool_input: Input arguments to the tool
|
||||
tool_output: Output from the tool (if available)
|
||||
"""
|
||||
logging.debug(f"display_tool_call called with message: {repr(message)}")
|
||||
if not message or not message.strip():
|
||||
logging.debug("Empty message in display_tool_call, returning early")
|
||||
@@ -235,11 +249,21 @@ def display_tool_call(message: str, console=None):
|
||||
logging.debug(f"Cleaned message in display_tool_call: {repr(message)}")
|
||||
|
||||
# Execute synchronous callbacks (always, even when console is None)
|
||||
execute_sync_callback('tool_call', message=message)
|
||||
execute_sync_callback('tool_call', message=message, tool_name=tool_name, tool_input=tool_input, tool_output=tool_output)
|
||||
|
||||
# Only print panel if console is provided (verbose mode)
|
||||
if console is not None:
|
||||
console.print(Panel.fit(Text(message, style="bold cyan"), title="Tool Call", border_style="green"))
|
||||
# Build consolidated tool call display (Agno-style)
|
||||
if tool_name and tool_input is not None:
|
||||
# Format: tool_name(arg1=val1, arg2=val2)
|
||||
args_str = ", ".join(f"{k}={v}" for k, v in tool_input.items()) if tool_input else ""
|
||||
tool_display = f"• {tool_name}({args_str})"
|
||||
if tool_output:
|
||||
tool_display += f"\n → {tool_output}"
|
||||
console.print(Panel.fit(Text(tool_display, style="bold cyan"), title="Tool Call", border_style="green"))
|
||||
else:
|
||||
# Legacy format
|
||||
console.print(Panel.fit(Text(message, style="bold cyan"), title="Tool Call", border_style="green"))
|
||||
|
||||
def display_error(message: str, console=None):
|
||||
if not message or not message.strip():
|
||||
|
||||
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
|
||||
|
||||
[project]
|
||||
name = "praisonaiagents"
|
||||
version = "0.5.2"
|
||||
version = "0.5.3"
|
||||
description = "Praison AI agents for completing complex tasks with Self Reflection Agents"
|
||||
readme = "README.md"
|
||||
requires-python = ">=3.10"
|
||||
|
||||
Generated
+1
-1
@@ -2422,7 +2422,7 @@ wheels = [
|
||||
|
||||
[[package]]
|
||||
name = "praisonaiagents"
|
||||
version = "0.5.2"
|
||||
version = "0.5.3"
|
||||
source = { editable = "." }
|
||||
dependencies = [
|
||||
{ name = "aiohttp" },
|
||||
|
||||
@@ -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.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
|
||||
url "https://github.com/MervinPraison/PraisonAI/archive/refs/tags/v2.9.2.tar.gz"
|
||||
sha256 `curl -sL https://github.com/MervinPraison/PraisonAI/archive/refs/tags/v2.9.2.tar.gz | shasum -a 256`.split.first
|
||||
license "MIT"
|
||||
|
||||
depends_on "[email protected]"
|
||||
|
||||
@@ -1,10 +1,15 @@
|
||||
# __main__.py
|
||||
"""
|
||||
PraisonAI CLI Entry Point.
|
||||
|
||||
from .cli import PraisonAI
|
||||
Supports both new Typer-based CLI and legacy argparse CLI.
|
||||
"""
|
||||
|
||||
def main():
|
||||
praison_ai = PraisonAI()
|
||||
praison_ai.main()
|
||||
"""Main entry point with legacy support."""
|
||||
from .cli.legacy import main_with_legacy_support
|
||||
main_with_legacy_support()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -0,0 +1,211 @@
|
||||
"""
|
||||
PraisonAI CLI Typer Application.
|
||||
|
||||
Main Typer app that registers all command groups and handles global options.
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
from enum import Enum
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from .output.console import OutputController, OutputMode, set_output_controller
|
||||
from .state.identifiers import create_context
|
||||
|
||||
|
||||
class OutputFormat(str, Enum):
|
||||
"""Output format options."""
|
||||
text = "text"
|
||||
json = "json"
|
||||
stream_json = "stream-json"
|
||||
|
||||
|
||||
# Create main Typer app
|
||||
app = typer.Typer(
|
||||
name="praisonai",
|
||||
help="PraisonAI - AI Agents Framework CLI",
|
||||
add_completion=False, # We handle completion manually
|
||||
no_args_is_help=False, # Allow running without args for legacy compatibility
|
||||
rich_markup_mode="rich",
|
||||
)
|
||||
|
||||
|
||||
# Global state for options
|
||||
class GlobalState:
|
||||
"""Global state for CLI options."""
|
||||
output_format: OutputFormat = OutputFormat.text
|
||||
no_color: bool = False
|
||||
quiet: bool = False
|
||||
verbose: bool = False
|
||||
screen_reader: bool = False
|
||||
output_controller: Optional[OutputController] = None
|
||||
|
||||
|
||||
state = GlobalState()
|
||||
|
||||
|
||||
def version_callback(value: bool):
|
||||
"""Show version and exit."""
|
||||
if value:
|
||||
from praisonai.version import __version__
|
||||
typer.echo(f"PraisonAI version {__version__}")
|
||||
raise typer.Exit()
|
||||
|
||||
|
||||
@app.callback(invoke_without_command=True)
|
||||
def main_callback(
|
||||
ctx: typer.Context,
|
||||
version: bool = typer.Option(
|
||||
False,
|
||||
"--version",
|
||||
"-V",
|
||||
callback=version_callback,
|
||||
is_eager=True,
|
||||
help="Show version and exit",
|
||||
),
|
||||
output_format: OutputFormat = typer.Option(
|
||||
OutputFormat.text,
|
||||
"--output-format",
|
||||
"-o",
|
||||
help="Output format",
|
||||
envvar="PRAISONAI_OUTPUT_FORMAT",
|
||||
),
|
||||
json_output: bool = typer.Option(
|
||||
False,
|
||||
"--json",
|
||||
help="Output in JSON format (alias for --output-format json)",
|
||||
),
|
||||
no_color: bool = typer.Option(
|
||||
False,
|
||||
"--no-color",
|
||||
help="Disable colored output",
|
||||
envvar="NO_COLOR",
|
||||
),
|
||||
quiet: bool = typer.Option(
|
||||
False,
|
||||
"--quiet",
|
||||
"-q",
|
||||
help="Minimal output",
|
||||
),
|
||||
verbose: bool = typer.Option(
|
||||
False,
|
||||
"--verbose",
|
||||
"-v",
|
||||
help="Verbose output with debug details",
|
||||
),
|
||||
screen_reader: bool = typer.Option(
|
||||
False,
|
||||
"--screen-reader",
|
||||
help="Screen reader friendly output (no spinners/panels)",
|
||||
),
|
||||
):
|
||||
"""
|
||||
PraisonAI - AI Agents Framework CLI.
|
||||
|
||||
Run agents, manage configuration, and more.
|
||||
"""
|
||||
# Store global options
|
||||
state.output_format = output_format
|
||||
state.no_color = no_color
|
||||
state.quiet = quiet
|
||||
state.verbose = verbose
|
||||
state.screen_reader = screen_reader
|
||||
|
||||
# Handle --json alias
|
||||
if json_output:
|
||||
state.output_format = OutputFormat.json
|
||||
|
||||
# Determine output mode
|
||||
if state.quiet:
|
||||
mode = OutputMode.QUIET
|
||||
elif state.verbose:
|
||||
mode = OutputMode.VERBOSE
|
||||
elif state.screen_reader:
|
||||
mode = OutputMode.SCREEN_READER
|
||||
elif state.output_format == OutputFormat.json:
|
||||
mode = OutputMode.JSON
|
||||
elif state.output_format == OutputFormat.stream_json:
|
||||
mode = OutputMode.STREAM_JSON
|
||||
else:
|
||||
mode = OutputMode.TEXT
|
||||
|
||||
# Create run context
|
||||
context = create_context()
|
||||
|
||||
# Create and set output controller
|
||||
state.output_controller = OutputController(
|
||||
mode=mode,
|
||||
no_color=state.no_color,
|
||||
run_id=context.run_id,
|
||||
trace_id=context.trace_id,
|
||||
)
|
||||
set_output_controller(state.output_controller)
|
||||
|
||||
# If no command provided, show help or handle legacy behavior
|
||||
if ctx.invoked_subcommand is None:
|
||||
# Check if there are remaining args (legacy behavior)
|
||||
# This is handled by the legacy adapter in main.py
|
||||
pass
|
||||
|
||||
|
||||
def get_output_controller() -> OutputController:
|
||||
"""Get the current output controller."""
|
||||
if state.output_controller is None:
|
||||
state.output_controller = OutputController()
|
||||
return state.output_controller
|
||||
|
||||
|
||||
# Import and register command groups
|
||||
def register_commands():
|
||||
"""Register all command groups."""
|
||||
# Import command modules
|
||||
from .commands.config import app as config_app
|
||||
from .commands.traces import app as traces_app
|
||||
from .commands.environment import app as env_app
|
||||
from .commands.session import app as session_app
|
||||
from .commands.completion import app as completion_app
|
||||
from .commands.version import app as version_app
|
||||
from .commands.debug import app as debug_app
|
||||
from .commands.lsp import app as lsp_app
|
||||
from .commands.diag import app as diag_app
|
||||
from .commands.doctor import app as doctor_app
|
||||
from .commands.acp import app as acp_app
|
||||
from .commands.mcp import app as mcp_app
|
||||
from .commands.serve import app as serve_app
|
||||
from .commands.schedule import app as schedule_app
|
||||
from .commands.run import app as run_app
|
||||
|
||||
# Import TUI and queue commands
|
||||
from .features.tui.debug import create_debug_app as create_tui_debug_app
|
||||
from .features.tui.cli import create_queue_app
|
||||
|
||||
# Register sub-apps
|
||||
app.add_typer(config_app, name="config", help="Configuration management")
|
||||
app.add_typer(traces_app, name="traces", help="Trace collection management")
|
||||
app.add_typer(env_app, name="env", help="Environment and diagnostics")
|
||||
app.add_typer(session_app, name="session", help="Session management")
|
||||
app.add_typer(completion_app, name="completion", help="Shell completion scripts")
|
||||
app.add_typer(version_app, name="version", help="Version information")
|
||||
app.add_typer(debug_app, name="debug", help="Debug and test interactive flows")
|
||||
app.add_typer(lsp_app, name="lsp", help="LSP service lifecycle")
|
||||
app.add_typer(diag_app, name="diag", help="Diagnostics export")
|
||||
app.add_typer(doctor_app, name="doctor", help="Health checks and diagnostics")
|
||||
app.add_typer(acp_app, name="acp", help="Agent Client Protocol server")
|
||||
app.add_typer(mcp_app, name="mcp", help="MCP server management")
|
||||
app.add_typer(serve_app, name="serve", help="API server management")
|
||||
app.add_typer(schedule_app, name="schedule", help="Scheduler management")
|
||||
app.add_typer(run_app, name="run", help="Run agents")
|
||||
|
||||
# Register TUI and queue commands
|
||||
tui_app = create_tui_debug_app()
|
||||
queue_app = create_queue_app()
|
||||
if tui_app:
|
||||
app.add_typer(tui_app, name="tui", help="Interactive TUI and simulation")
|
||||
if queue_app:
|
||||
app.add_typer(queue_app, name="queue", help="Queue management")
|
||||
|
||||
|
||||
# Register commands on import
|
||||
register_commands()
|
||||
@@ -0,0 +1,75 @@
|
||||
"""
|
||||
PraisonAI CLI Commands Module.
|
||||
|
||||
This module contains all Typer command groups and subcommands.
|
||||
Each file represents a command group (e.g., config, traces, session).
|
||||
"""
|
||||
|
||||
# Lazy imports to avoid loading all commands at startup
|
||||
__all__ = [
|
||||
'run_app',
|
||||
'config_app',
|
||||
'traces_app',
|
||||
'env_app',
|
||||
'session_app',
|
||||
'schedule_app',
|
||||
'serve_app',
|
||||
'completion_app',
|
||||
'version_app',
|
||||
'debug_app',
|
||||
'lsp_app',
|
||||
'diag_app',
|
||||
'doctor_app',
|
||||
'acp_app',
|
||||
'mcp_app',
|
||||
]
|
||||
|
||||
|
||||
def __getattr__(name: str):
|
||||
"""Lazy load command apps."""
|
||||
if name == 'run_app':
|
||||
from .run import app as run_app
|
||||
return run_app
|
||||
elif name == 'config_app':
|
||||
from .config import app as config_app
|
||||
return config_app
|
||||
elif name == 'traces_app':
|
||||
from .traces import app as traces_app
|
||||
return traces_app
|
||||
elif name == 'env_app':
|
||||
from .environment import app as env_app
|
||||
return env_app
|
||||
elif name == 'session_app':
|
||||
from .session import app as session_app
|
||||
return session_app
|
||||
elif name == 'schedule_app':
|
||||
from .schedule import app as schedule_app
|
||||
return schedule_app
|
||||
elif name == 'serve_app':
|
||||
from .serve import app as serve_app
|
||||
return serve_app
|
||||
elif name == 'completion_app':
|
||||
from .completion import app as completion_app
|
||||
return completion_app
|
||||
elif name == 'version_app':
|
||||
from .version import app as version_app
|
||||
return version_app
|
||||
elif name == 'debug_app':
|
||||
from .debug import app as debug_app
|
||||
return debug_app
|
||||
elif name == 'lsp_app':
|
||||
from .lsp import app as lsp_app
|
||||
return lsp_app
|
||||
elif name == 'diag_app':
|
||||
from .diag import app as diag_app
|
||||
return diag_app
|
||||
elif name == 'doctor_app':
|
||||
from .doctor import app as doctor_app
|
||||
return doctor_app
|
||||
elif name == 'acp_app':
|
||||
from .acp import app as acp_app
|
||||
return acp_app
|
||||
elif name == 'mcp_app':
|
||||
from .mcp import app as mcp_app
|
||||
return mcp_app
|
||||
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|
||||
@@ -0,0 +1,70 @@
|
||||
"""
|
||||
ACP command group for PraisonAI CLI.
|
||||
|
||||
Wraps existing ACP functionality from features/acp.py.
|
||||
Provides Agent Client Protocol server for IDE integration.
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
|
||||
app = typer.Typer(help="Agent Client Protocol server")
|
||||
|
||||
|
||||
@app.callback(invoke_without_command=True)
|
||||
def acp_main(
|
||||
ctx: typer.Context,
|
||||
workspace: str = typer.Option(".", "--workspace", "-w", help="Workspace root directory"),
|
||||
agent: str = typer.Option("default", "--agent", "-a", help="Agent name or configuration file"),
|
||||
agents_file: Optional[str] = typer.Option(None, "--agents", help="Multi-agent configuration YAML file"),
|
||||
router: bool = typer.Option(False, "--router", help="Enable router agent for task delegation"),
|
||||
model: Optional[str] = typer.Option(None, "--model", "-m", help="LLM model to use"),
|
||||
resume: Optional[str] = typer.Option(None, "--resume", "-r", help="Resume session by ID"),
|
||||
last: bool = typer.Option(False, "--last", help="Resume the last session"),
|
||||
approve: str = typer.Option("manual", "--approve", help="Approval mode: manual, auto, scoped"),
|
||||
read_only: bool = typer.Option(True, "--read-only/--allow-write", help="Read-only mode"),
|
||||
allow_shell: bool = typer.Option(False, "--allow-shell", help="Allow shell command execution"),
|
||||
allow_network: bool = typer.Option(False, "--allow-network", help="Allow network requests"),
|
||||
debug: bool = typer.Option(False, "--debug", help="Enable debug logging"),
|
||||
profile: Optional[str] = typer.Option(None, "--profile", help="Use named profile from config"),
|
||||
):
|
||||
"""Start ACP server for IDE integration."""
|
||||
# Build args for existing handler
|
||||
args = [
|
||||
"--workspace", workspace,
|
||||
"--agent", agent,
|
||||
"--approve", approve,
|
||||
]
|
||||
|
||||
if agents_file:
|
||||
args.extend(["--agents", agents_file])
|
||||
if router:
|
||||
args.append("--router")
|
||||
if model:
|
||||
args.extend(["--model", model])
|
||||
if resume:
|
||||
args.extend(["--resume", resume])
|
||||
if last:
|
||||
args.append("--last")
|
||||
if not read_only:
|
||||
args.append("--allow-write")
|
||||
if allow_shell:
|
||||
args.append("--allow-shell")
|
||||
if allow_network:
|
||||
args.append("--allow-network")
|
||||
if debug:
|
||||
args.append("--debug")
|
||||
if profile:
|
||||
args.extend(["--profile", profile])
|
||||
|
||||
try:
|
||||
from ..features.acp import run_acp_command
|
||||
exit_code = run_acp_command(args)
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output = get_output_controller()
|
||||
output.print_error(f"ACP module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
@@ -0,0 +1,333 @@
|
||||
"""
|
||||
Completion command group for PraisonAI CLI.
|
||||
|
||||
Provides shell completion script generation:
|
||||
- completion bash: Generate bash completions
|
||||
- completion zsh: Generate zsh completions
|
||||
- completion fish: Generate fish completions
|
||||
"""
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
|
||||
app = typer.Typer(help="Shell completion scripts")
|
||||
|
||||
|
||||
BASH_COMPLETION = '''
|
||||
# PraisonAI bash completion
|
||||
_praisonai_completion() {
|
||||
local cur prev opts
|
||||
COMPREPLY=()
|
||||
cur="${COMP_WORDS[COMP_CWORD]}"
|
||||
prev="${COMP_WORDS[COMP_CWORD-1]}"
|
||||
|
||||
# Top-level commands
|
||||
opts="run config traces env session schedule serve completion version debug lsp diag doctor acp mcp chat code call realtime train ui context research memory rules workflow hooks knowledge tools todo docs commit skills profile eval templates recipe endpoints agents thinking compaction output deploy registry package install uninstall"
|
||||
|
||||
# Subcommands for specific commands
|
||||
case "${prev}" in
|
||||
config)
|
||||
COMPREPLY=( $(compgen -W "list get set reset path" -- ${cur}) )
|
||||
return 0
|
||||
;;
|
||||
traces)
|
||||
COMPREPLY=( $(compgen -W "enable disable status list" -- ${cur}) )
|
||||
return 0
|
||||
;;
|
||||
env)
|
||||
COMPREPLY=( $(compgen -W "view check doctor" -- ${cur}) )
|
||||
return 0
|
||||
;;
|
||||
session)
|
||||
COMPREPLY=( $(compgen -W "list resume delete export show" -- ${cur}) )
|
||||
return 0
|
||||
;;
|
||||
schedule)
|
||||
COMPREPLY=( $(compgen -W "start stop list logs restart delete describe save stop-all stats" -- ${cur}) )
|
||||
return 0
|
||||
;;
|
||||
serve)
|
||||
COMPREPLY=( $(compgen -W "start stop status" -- ${cur}) )
|
||||
return 0
|
||||
;;
|
||||
completion)
|
||||
COMPREPLY=( $(compgen -W "bash zsh fish" -- ${cur}) )
|
||||
return 0
|
||||
;;
|
||||
debug)
|
||||
COMPREPLY=( $(compgen -W "interactive lsp acp trace" -- ${cur}) )
|
||||
return 0
|
||||
;;
|
||||
lsp)
|
||||
COMPREPLY=( $(compgen -W "start stop status logs" -- ${cur}) )
|
||||
return 0
|
||||
;;
|
||||
diag)
|
||||
COMPREPLY=( $(compgen -W "export" -- ${cur}) )
|
||||
return 0
|
||||
;;
|
||||
doctor)
|
||||
COMPREPLY=( $(compgen -W "env config tools db mcp obs skills memory permissions network performance ci selftest" -- ${cur}) )
|
||||
return 0
|
||||
;;
|
||||
mcp)
|
||||
COMPREPLY=( $(compgen -W "list add remove test" -- ${cur}) )
|
||||
return 0
|
||||
;;
|
||||
*)
|
||||
;;
|
||||
esac
|
||||
|
||||
# Global options
|
||||
if [[ ${cur} == -* ]] ; then
|
||||
COMPREPLY=( $(compgen -W "--help --version --output-format --json --no-color --quiet --verbose --screen-reader" -- ${cur}) )
|
||||
return 0
|
||||
fi
|
||||
|
||||
COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
|
||||
return 0
|
||||
}
|
||||
|
||||
complete -F _praisonai_completion praisonai
|
||||
'''
|
||||
|
||||
ZSH_COMPLETION = '''
|
||||
#compdef praisonai
|
||||
|
||||
_praisonai() {
|
||||
local -a commands
|
||||
local -a subcommands
|
||||
|
||||
commands=(
|
||||
'run:Run agents'
|
||||
'config:Configuration management'
|
||||
'traces:Trace collection management'
|
||||
'env:Environment and diagnostics'
|
||||
'session:Session management'
|
||||
'schedule:Scheduler management'
|
||||
'serve:API server management'
|
||||
'completion:Shell completion scripts'
|
||||
'version:Version information'
|
||||
'debug:Debug and test interactive flows'
|
||||
'lsp:LSP service lifecycle'
|
||||
'diag:Diagnostics export'
|
||||
'doctor:Health checks and diagnostics'
|
||||
'acp:Agent Client Protocol server'
|
||||
'mcp:MCP server management'
|
||||
'chat:Start chat UI'
|
||||
'code:Start code UI'
|
||||
'call:Start call server'
|
||||
'realtime:Start realtime interface'
|
||||
'train:Training commands'
|
||||
'ui:Start UI'
|
||||
'context:Context engineering'
|
||||
'research:Deep research'
|
||||
'memory:Memory management'
|
||||
'rules:Rules management'
|
||||
'workflow:Workflow management'
|
||||
'hooks:Hooks management'
|
||||
'knowledge:Knowledge management'
|
||||
'tools:Tools management'
|
||||
'todo:Todo management'
|
||||
'docs:Documentation'
|
||||
'commit:Git commit'
|
||||
'skills:Skills management'
|
||||
'profile:Profile management'
|
||||
'eval:Evaluation'
|
||||
'templates:Templates'
|
||||
'recipe:Recipe management'
|
||||
'endpoints:Endpoints'
|
||||
'agents:Agents management'
|
||||
'deploy:Deployment'
|
||||
'registry:Registry'
|
||||
'package:Package management'
|
||||
'install:Install packages'
|
||||
'uninstall:Uninstall packages'
|
||||
)
|
||||
|
||||
_arguments -C \\
|
||||
'--help[Show help]' \\
|
||||
'--version[Show version]' \\
|
||||
'--output-format[Output format]:format:(text json stream-json)' \\
|
||||
'--json[JSON output]' \\
|
||||
'--no-color[Disable colors]' \\
|
||||
'--quiet[Minimal output]' \\
|
||||
'--verbose[Verbose output]' \\
|
||||
'--screen-reader[Screen reader mode]' \\
|
||||
'1: :->command' \\
|
||||
'*::arg:->args'
|
||||
|
||||
case $state in
|
||||
command)
|
||||
_describe 'command' commands
|
||||
;;
|
||||
args)
|
||||
case $words[1] in
|
||||
config)
|
||||
subcommands=('list:List config' 'get:Get value' 'set:Set value' 'reset:Reset config' 'path:Show path')
|
||||
_describe 'subcommand' subcommands
|
||||
;;
|
||||
traces)
|
||||
subcommands=('enable:Enable traces' 'disable:Disable traces' 'status:Show status' 'list:List traces')
|
||||
_describe 'subcommand' subcommands
|
||||
;;
|
||||
env)
|
||||
subcommands=('view:View env vars' 'check:Check API keys' 'doctor:Run diagnostics')
|
||||
_describe 'subcommand' subcommands
|
||||
;;
|
||||
session)
|
||||
subcommands=('list:List sessions' 'resume:Resume session' 'delete:Delete session' 'export:Export session' 'show:Show session')
|
||||
_describe 'subcommand' subcommands
|
||||
;;
|
||||
schedule)
|
||||
subcommands=('start:Start scheduler' 'stop:Stop scheduler' 'list:List jobs' 'logs:View logs' 'restart:Restart' 'delete:Delete job' 'describe:Describe job' 'save:Save config' 'stop-all:Stop all' 'stats:Show stats')
|
||||
_describe 'subcommand' subcommands
|
||||
;;
|
||||
serve)
|
||||
subcommands=('start:Start server' 'stop:Stop server' 'status:Show status')
|
||||
_describe 'subcommand' subcommands
|
||||
;;
|
||||
completion)
|
||||
subcommands=('bash:Bash completion' 'zsh:Zsh completion' 'fish:Fish completion')
|
||||
_describe 'subcommand' subcommands
|
||||
;;
|
||||
debug)
|
||||
subcommands=('interactive:Interactive debug' 'lsp:LSP debug' 'acp:ACP debug' 'trace:Trace debug')
|
||||
_describe 'subcommand' subcommands
|
||||
;;
|
||||
lsp)
|
||||
subcommands=('start:Start LSP' 'stop:Stop LSP' 'status:LSP status' 'logs:LSP logs')
|
||||
_describe 'subcommand' subcommands
|
||||
;;
|
||||
diag)
|
||||
subcommands=('export:Export diagnostics')
|
||||
_describe 'subcommand' subcommands
|
||||
;;
|
||||
doctor)
|
||||
subcommands=('env:Check env' 'config:Check config' 'tools:Check tools' 'db:Check DB' 'mcp:Check MCP' 'obs:Check observability' 'skills:Check skills' 'memory:Check memory' 'permissions:Check permissions' 'network:Check network' 'performance:Check performance' 'ci:CI checks' 'selftest:Self test')
|
||||
_describe 'subcommand' subcommands
|
||||
;;
|
||||
mcp)
|
||||
subcommands=('list:List servers' 'add:Add server' 'remove:Remove server' 'test:Test server')
|
||||
_describe 'subcommand' subcommands
|
||||
;;
|
||||
esac
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
_praisonai "$@"
|
||||
'''
|
||||
|
||||
FISH_COMPLETION = '''
|
||||
# PraisonAI fish completion
|
||||
|
||||
# Top-level commands
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "run" -d "Run agents"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "config" -d "Configuration management"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "traces" -d "Trace collection management"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "env" -d "Environment and diagnostics"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "session" -d "Session management"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "schedule" -d "Scheduler management"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "serve" -d "API server management"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "completion" -d "Shell completion scripts"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "version" -d "Version information"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "debug" -d "Debug and test interactive flows"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "lsp" -d "LSP service lifecycle"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "diag" -d "Diagnostics export"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "doctor" -d "Health checks and diagnostics"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "acp" -d "Agent Client Protocol server"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "mcp" -d "MCP server management"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "chat" -d "Start chat UI"
|
||||
complete -c praisonai -f -n "__fish_use_subcommand" -a "code" -d "Start code UI"
|
||||
|
||||
# Global options
|
||||
complete -c praisonai -l help -d "Show help"
|
||||
complete -c praisonai -l version -d "Show version"
|
||||
complete -c praisonai -l output-format -x -a "text json stream-json" -d "Output format"
|
||||
complete -c praisonai -l json -d "JSON output"
|
||||
complete -c praisonai -l no-color -d "Disable colors"
|
||||
complete -c praisonai -l quiet -s q -d "Minimal output"
|
||||
complete -c praisonai -l verbose -s v -d "Verbose output"
|
||||
complete -c praisonai -l screen-reader -d "Screen reader mode"
|
||||
|
||||
# config subcommands
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from config" -a "list" -d "List config"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from config" -a "get" -d "Get value"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from config" -a "set" -d "Set value"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from config" -a "reset" -d "Reset config"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from config" -a "path" -d "Show path"
|
||||
|
||||
# traces subcommands
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from traces" -a "enable" -d "Enable traces"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from traces" -a "disable" -d "Disable traces"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from traces" -a "status" -d "Show status"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from traces" -a "list" -d "List traces"
|
||||
|
||||
# env subcommands
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from env" -a "view" -d "View env vars"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from env" -a "check" -d "Check API keys"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from env" -a "doctor" -d "Run diagnostics"
|
||||
|
||||
# session subcommands
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from session" -a "list" -d "List sessions"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from session" -a "resume" -d "Resume session"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from session" -a "delete" -d "Delete session"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from session" -a "export" -d "Export session"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from session" -a "show" -d "Show session"
|
||||
|
||||
# completion subcommands
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from completion" -a "bash" -d "Bash completion"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from completion" -a "zsh" -d "Zsh completion"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from completion" -a "fish" -d "Fish completion"
|
||||
|
||||
# debug subcommands
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from debug" -a "interactive" -d "Interactive debug"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from debug" -a "lsp" -d "LSP debug"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from debug" -a "acp" -d "ACP debug"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from debug" -a "trace" -d "Trace debug"
|
||||
|
||||
# doctor subcommands
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from doctor" -a "env" -d "Check env"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from doctor" -a "config" -d "Check config"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from doctor" -a "tools" -d "Check tools"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from doctor" -a "db" -d "Check DB"
|
||||
complete -c praisonai -f -n "__fish_seen_subcommand_from doctor" -a "mcp" -d "Check MCP"
|
||||
'''
|
||||
|
||||
|
||||
@app.command("bash")
|
||||
def completion_bash():
|
||||
"""Generate bash completion script."""
|
||||
print(BASH_COMPLETION.strip())
|
||||
|
||||
|
||||
@app.command("zsh")
|
||||
def completion_zsh():
|
||||
"""Generate zsh completion script."""
|
||||
print(ZSH_COMPLETION.strip())
|
||||
|
||||
|
||||
@app.command("fish")
|
||||
def completion_fish():
|
||||
"""Generate fish completion script."""
|
||||
print(FISH_COMPLETION.strip())
|
||||
|
||||
|
||||
@app.callback(invoke_without_command=True)
|
||||
def completion_callback(ctx: typer.Context):
|
||||
"""Show completion installation instructions."""
|
||||
if ctx.invoked_subcommand is None:
|
||||
output = get_output_controller()
|
||||
output.print_panel(
|
||||
"Generate shell completion scripts.\n\n"
|
||||
"Usage:\n"
|
||||
" # Bash\n"
|
||||
" praisonai completion bash >> ~/.bashrc\n\n"
|
||||
" # Zsh\n"
|
||||
" praisonai completion zsh > ~/.zsh/completions/_praisonai\n\n"
|
||||
" # Fish\n"
|
||||
" praisonai completion fish > ~/.config/fish/completions/praisonai.fish",
|
||||
title="Shell Completions"
|
||||
)
|
||||
@@ -0,0 +1,166 @@
|
||||
"""
|
||||
Config command group for PraisonAI CLI.
|
||||
|
||||
Provides configuration management commands:
|
||||
- config list: Show all configuration
|
||||
- config get: Get a specific value
|
||||
- config set: Set a configuration value
|
||||
- config reset: Reset to defaults
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..configuration.loader import get_config_loader
|
||||
from ..output.console import get_output_controller
|
||||
|
||||
app = typer.Typer(help="Configuration management")
|
||||
|
||||
|
||||
@app.command("list")
|
||||
def config_list(
|
||||
scope: str = typer.Option(
|
||||
"all",
|
||||
"--scope",
|
||||
"-s",
|
||||
help="Scope to list: all, user, project",
|
||||
),
|
||||
):
|
||||
"""List all configuration values."""
|
||||
output = get_output_controller()
|
||||
loader = get_config_loader()
|
||||
|
||||
config = loader.list_all()
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json(config)
|
||||
return
|
||||
|
||||
output.print_panel("Configuration", title="PraisonAI Config")
|
||||
|
||||
def print_dict(d, prefix=""):
|
||||
for key, value in d.items():
|
||||
full_key = f"{prefix}.{key}" if prefix else key
|
||||
if isinstance(value, dict):
|
||||
print_dict(value, full_key)
|
||||
else:
|
||||
output.print(f" {full_key} = {value}")
|
||||
|
||||
print_dict(config)
|
||||
|
||||
|
||||
@app.command("get")
|
||||
def config_get(
|
||||
key: str = typer.Argument(..., help="Configuration key (dotted notation, e.g., output.format)"),
|
||||
):
|
||||
"""Get a configuration value."""
|
||||
output = get_output_controller()
|
||||
loader = get_config_loader()
|
||||
|
||||
value = loader.get(key)
|
||||
|
||||
if value is None:
|
||||
output.print_error(f"Key not found: {key}")
|
||||
raise typer.Exit(1)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({"key": key, "value": value})
|
||||
else:
|
||||
output.print(f"{key} = {value}")
|
||||
|
||||
|
||||
@app.command("set")
|
||||
def config_set(
|
||||
key: str = typer.Argument(..., help="Configuration key (dotted notation)"),
|
||||
value: str = typer.Argument(..., help="Value to set"),
|
||||
scope: str = typer.Option(
|
||||
"user",
|
||||
"--scope",
|
||||
"-s",
|
||||
help="Scope: user or project",
|
||||
),
|
||||
):
|
||||
"""Set a configuration value."""
|
||||
output = get_output_controller()
|
||||
loader = get_config_loader()
|
||||
|
||||
# Parse value
|
||||
if value.lower() == "true":
|
||||
parsed_value = True
|
||||
elif value.lower() == "false":
|
||||
parsed_value = False
|
||||
else:
|
||||
try:
|
||||
parsed_value = int(value)
|
||||
except ValueError:
|
||||
try:
|
||||
parsed_value = float(value)
|
||||
except ValueError:
|
||||
parsed_value = value
|
||||
|
||||
loader.set(key, parsed_value, scope=scope)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({"key": key, "value": parsed_value, "scope": scope})
|
||||
else:
|
||||
output.print_success(f"Set {key} = {parsed_value} ({scope})")
|
||||
|
||||
|
||||
@app.command("reset")
|
||||
def config_reset(
|
||||
scope: str = typer.Option(
|
||||
"user",
|
||||
"--scope",
|
||||
"-s",
|
||||
help="Scope to reset: user or project",
|
||||
),
|
||||
confirm: bool = typer.Option(
|
||||
False,
|
||||
"--yes",
|
||||
"-y",
|
||||
help="Skip confirmation",
|
||||
),
|
||||
):
|
||||
"""Reset configuration to defaults."""
|
||||
output = get_output_controller()
|
||||
|
||||
if not confirm:
|
||||
confirmed = typer.confirm(f"Reset {scope} configuration to defaults?")
|
||||
if not confirmed:
|
||||
output.print_info("Cancelled")
|
||||
raise typer.Exit(0)
|
||||
|
||||
loader = get_config_loader()
|
||||
loader.reset(scope=scope)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({"reset": True, "scope": scope})
|
||||
else:
|
||||
output.print_success(f"Reset {scope} configuration to defaults")
|
||||
|
||||
|
||||
@app.command("path")
|
||||
def config_path(
|
||||
scope: str = typer.Option(
|
||||
"user",
|
||||
"--scope",
|
||||
"-s",
|
||||
help="Scope: user or project",
|
||||
),
|
||||
):
|
||||
"""Show configuration file path."""
|
||||
output = get_output_controller()
|
||||
|
||||
from ..configuration.paths import get_user_config_path, get_project_config_path
|
||||
|
||||
if scope == "project":
|
||||
path = get_project_config_path()
|
||||
else:
|
||||
path = get_user_config_path()
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({"path": str(path), "exists": path.exists()})
|
||||
else:
|
||||
exists = "✓" if path.exists() else "✗"
|
||||
output.print(f"{path} [{exists}]")
|
||||
@@ -0,0 +1,142 @@
|
||||
"""
|
||||
Debug command group for PraisonAI CLI.
|
||||
|
||||
Wraps existing debug functionality from features/debug.py.
|
||||
Provides debug commands for testing interactive flows.
|
||||
"""
|
||||
|
||||
import sys
|
||||
from typing import List, Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
|
||||
app = typer.Typer(help="Debug and test interactive flows")
|
||||
|
||||
|
||||
@app.command("interactive")
|
||||
def debug_interactive(
|
||||
prompt: str = typer.Option(..., "--prompt", "-p", help="Prompt to execute"),
|
||||
workspace: str = typer.Option(".", "--workspace", "-w", help="Workspace root directory"),
|
||||
lsp: bool = typer.Option(True, "--lsp/--no-lsp", help="Enable/disable LSP"),
|
||||
acp: bool = typer.Option(True, "--acp/--no-acp", help="Enable/disable ACP"),
|
||||
approval: str = typer.Option("auto", "--approval", help="Approval mode: manual, auto, scoped"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON trace"),
|
||||
trace_file: Optional[str] = typer.Option(None, "--trace-file", help="Save trace to file"),
|
||||
timeout: float = typer.Option(60.0, "--timeout", help="Timeout in seconds"),
|
||||
model: Optional[str] = typer.Option(None, "--model", "-m", help="LLM model to use"),
|
||||
):
|
||||
"""Run single interactive turn non-interactively."""
|
||||
output = get_output_controller()
|
||||
|
||||
# Build args for existing handler
|
||||
args = [
|
||||
"interactive",
|
||||
"--prompt", prompt,
|
||||
"--workspace", workspace,
|
||||
"--approval", approval,
|
||||
"--timeout", str(timeout),
|
||||
]
|
||||
|
||||
if not lsp:
|
||||
args.append("--no-lsp")
|
||||
if not acp:
|
||||
args.append("--no-acp")
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
if trace_file:
|
||||
args.extend(["--trace-file", trace_file])
|
||||
if model:
|
||||
args.extend(["--model", model])
|
||||
|
||||
# Delegate to existing handler
|
||||
try:
|
||||
from ..features.debug import run_debug_command
|
||||
exit_code = run_debug_command(args)
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output.print_error(f"Debug module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
|
||||
|
||||
@app.command("lsp")
|
||||
def debug_lsp(
|
||||
workspace: str = typer.Option(".", "--workspace", "-w", help="Workspace root directory"),
|
||||
language: Optional[str] = typer.Option(None, "--language", "-l", help="Language to probe"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Direct LSP probes."""
|
||||
args = ["lsp", "--workspace", workspace]
|
||||
if language:
|
||||
args.extend(["--language", language])
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
|
||||
try:
|
||||
from ..features.debug import run_debug_command
|
||||
exit_code = run_debug_command(args)
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output = get_output_controller()
|
||||
output.print_error(f"Debug module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
|
||||
|
||||
@app.command("acp")
|
||||
def debug_acp(
|
||||
workspace: str = typer.Option(".", "--workspace", "-w", help="Workspace root directory"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Direct ACP probes."""
|
||||
args = ["acp", "--workspace", workspace]
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
|
||||
try:
|
||||
from ..features.debug import run_debug_command
|
||||
exit_code = run_debug_command(args)
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output = get_output_controller()
|
||||
output.print_error(f"Debug module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
|
||||
|
||||
@app.command("trace")
|
||||
def debug_trace(
|
||||
action: str = typer.Argument("list", help="Action: record, replay, diff, list"),
|
||||
trace_id: Optional[str] = typer.Option(None, "--trace-id", "-t", help="Trace ID"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Trace record/replay/diff."""
|
||||
args = ["trace", action]
|
||||
if trace_id:
|
||||
args.extend(["--trace-id", trace_id])
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
|
||||
try:
|
||||
from ..features.debug import run_debug_command
|
||||
exit_code = run_debug_command(args)
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output = get_output_controller()
|
||||
output.print_error(f"Debug module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
|
||||
|
||||
@app.callback(invoke_without_command=True)
|
||||
def debug_callback(ctx: typer.Context):
|
||||
"""Show debug help."""
|
||||
if ctx.invoked_subcommand is None:
|
||||
output = get_output_controller()
|
||||
output.print_panel(
|
||||
"Debug commands for testing interactive coding assistant flows.\n\n"
|
||||
"Commands:\n"
|
||||
" interactive Run single interactive turn\n"
|
||||
" lsp Direct LSP probes\n"
|
||||
" acp Direct ACP probes\n"
|
||||
" trace Trace record/replay/diff",
|
||||
title="Debug Commands"
|
||||
)
|
||||
@@ -0,0 +1,55 @@
|
||||
"""
|
||||
Diag command group for PraisonAI CLI.
|
||||
|
||||
Wraps existing diagnostics functionality from features/diag.py.
|
||||
Provides diagnostic export for bug reports.
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
|
||||
app = typer.Typer(help="Diagnostics export")
|
||||
|
||||
|
||||
@app.command("export")
|
||||
def diag_export(
|
||||
output_path: Optional[str] = typer.Option(None, "--output", "-o", help="Output path"),
|
||||
workspace: str = typer.Option(".", "--workspace", "-w", help="Workspace root directory"),
|
||||
include_logs: bool = typer.Option(True, "--include-logs/--no-logs", help="Include log files"),
|
||||
include_config: bool = typer.Option(True, "--include-config/--no-config", help="Include configuration"),
|
||||
include_trace: bool = typer.Option(True, "--include-trace/--no-trace", help="Include recent traces"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Export diagnostic bundle for bug reports."""
|
||||
args = ["export", "--workspace", workspace]
|
||||
|
||||
if output_path:
|
||||
args.extend(["--output", output_path])
|
||||
if not include_logs:
|
||||
args.append("--no-logs")
|
||||
if not include_config:
|
||||
args.append("--no-config")
|
||||
if not include_trace:
|
||||
args.append("--no-trace")
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
|
||||
try:
|
||||
from ..features.diag import run_diag_command
|
||||
exit_code = run_diag_command(args)
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output = get_output_controller()
|
||||
output.print_error(f"Diag module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
|
||||
|
||||
@app.callback(invoke_without_command=True)
|
||||
def diag_callback(ctx: typer.Context):
|
||||
"""Show diag help or run export."""
|
||||
if ctx.invoked_subcommand is None:
|
||||
# Default to export
|
||||
diag_export()
|
||||
@@ -0,0 +1,166 @@
|
||||
"""
|
||||
Doctor command group for PraisonAI CLI.
|
||||
|
||||
Wraps existing doctor functionality from features/doctor/.
|
||||
Provides health checks and diagnostics.
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
|
||||
app = typer.Typer(help="Health checks and diagnostics")
|
||||
|
||||
|
||||
def _run_doctor(args: list) -> int:
|
||||
"""Run doctor command with args."""
|
||||
try:
|
||||
from ..features.doctor.handler import DoctorHandler
|
||||
handler = DoctorHandler()
|
||||
# DoctorHandler uses execute(action, action_args) pattern
|
||||
action = args[0] if args else None
|
||||
action_args = args[1:] if len(args) > 1 else []
|
||||
return handler.execute(action, action_args)
|
||||
except ImportError as e:
|
||||
output = get_output_controller()
|
||||
output.print_error(f"Doctor module not available: {e}")
|
||||
return 4
|
||||
except Exception as e:
|
||||
output = get_output_controller()
|
||||
output.print_error(f"Doctor error: {e}")
|
||||
return 1
|
||||
|
||||
|
||||
@app.command("env")
|
||||
def doctor_env(
|
||||
deep: bool = typer.Option(False, "--deep", help="Enable deeper probes"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
timeout: float = typer.Option(10.0, "--timeout", help="Per-check timeout"),
|
||||
):
|
||||
"""Check environment variables and API keys."""
|
||||
args = ["env"]
|
||||
if deep:
|
||||
args.append("--deep")
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
args.extend(["--timeout", str(timeout)])
|
||||
raise typer.Exit(_run_doctor(args))
|
||||
|
||||
|
||||
@app.command("config")
|
||||
def doctor_config(
|
||||
deep: bool = typer.Option(False, "--deep", help="Enable deeper probes"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Check configuration files."""
|
||||
args = ["config"]
|
||||
if deep:
|
||||
args.append("--deep")
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
raise typer.Exit(_run_doctor(args))
|
||||
|
||||
|
||||
@app.command("tools")
|
||||
def doctor_tools(
|
||||
deep: bool = typer.Option(False, "--deep", help="Enable deeper probes"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Check tools availability."""
|
||||
args = ["tools"]
|
||||
if deep:
|
||||
args.append("--deep")
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
raise typer.Exit(_run_doctor(args))
|
||||
|
||||
|
||||
@app.command("db")
|
||||
def doctor_db(
|
||||
deep: bool = typer.Option(False, "--deep", help="Enable deeper probes"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Check database connections."""
|
||||
args = ["db"]
|
||||
if deep:
|
||||
args.append("--deep")
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
raise typer.Exit(_run_doctor(args))
|
||||
|
||||
|
||||
@app.command("mcp")
|
||||
def doctor_mcp(
|
||||
deep: bool = typer.Option(False, "--deep", help="Enable deeper probes"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Check MCP servers."""
|
||||
args = ["mcp"]
|
||||
if deep:
|
||||
args.append("--deep")
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
raise typer.Exit(_run_doctor(args))
|
||||
|
||||
|
||||
@app.command("network")
|
||||
def doctor_network(
|
||||
deep: bool = typer.Option(False, "--deep", help="Enable deeper probes"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Check network connectivity."""
|
||||
args = ["network"]
|
||||
if deep:
|
||||
args.append("--deep")
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
raise typer.Exit(_run_doctor(args))
|
||||
|
||||
|
||||
@app.command("performance")
|
||||
def doctor_performance(
|
||||
deep: bool = typer.Option(False, "--deep", help="Enable deeper probes"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Check performance metrics."""
|
||||
args = ["performance"]
|
||||
if deep:
|
||||
args.append("--deep")
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
raise typer.Exit(_run_doctor(args))
|
||||
|
||||
|
||||
@app.command("selftest")
|
||||
def doctor_selftest(
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Run self-test."""
|
||||
args = ["selftest"]
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
raise typer.Exit(_run_doctor(args))
|
||||
|
||||
|
||||
@app.callback(invoke_without_command=True)
|
||||
def doctor_callback(
|
||||
ctx: typer.Context,
|
||||
deep: bool = typer.Option(False, "--deep", help="Enable deeper probes"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
strict: bool = typer.Option(False, "--strict", help="Treat warnings as failures"),
|
||||
quiet: bool = typer.Option(False, "--quiet", "-q", help="Minimal output"),
|
||||
):
|
||||
"""Run all fast health checks."""
|
||||
if ctx.invoked_subcommand is None:
|
||||
args = []
|
||||
if deep:
|
||||
args.append("--deep")
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
if strict:
|
||||
args.append("--strict")
|
||||
if quiet:
|
||||
args.append("--quiet")
|
||||
raise typer.Exit(_run_doctor(args))
|
||||
@@ -0,0 +1,179 @@
|
||||
"""
|
||||
Environment command group for PraisonAI CLI.
|
||||
|
||||
Provides environment management and diagnostics:
|
||||
- env view: Show environment variables
|
||||
- env check: Validate API keys
|
||||
- env doctor: Run diagnostics (alias for doctor command)
|
||||
"""
|
||||
|
||||
import os
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
|
||||
app = typer.Typer(help="Environment and diagnostics")
|
||||
|
||||
|
||||
# Known API key patterns
|
||||
API_KEY_PATTERNS = {
|
||||
"OPENAI_API_KEY": {"prefix": "sk-", "min_length": 20},
|
||||
"ANTHROPIC_API_KEY": {"prefix": "sk-ant-", "min_length": 20},
|
||||
"GOOGLE_API_KEY": {"prefix": "AI", "min_length": 20},
|
||||
"GEMINI_API_KEY": {"prefix": "AI", "min_length": 20},
|
||||
"TAVILY_API_KEY": {"prefix": "tvly-", "min_length": 20},
|
||||
"GROQ_API_KEY": {"prefix": "gsk_", "min_length": 20},
|
||||
}
|
||||
|
||||
|
||||
def _redact_key(value: str, show_chars: int = 4) -> str:
|
||||
"""Redact an API key for display."""
|
||||
if len(value) <= show_chars * 2:
|
||||
return "*" * len(value)
|
||||
return value[:show_chars] + "*" * (len(value) - show_chars * 2) + value[-show_chars:]
|
||||
|
||||
|
||||
def _validate_key(name: str, value: str) -> tuple[bool, str]:
|
||||
"""Validate an API key format."""
|
||||
if name not in API_KEY_PATTERNS:
|
||||
return True, "Unknown key type"
|
||||
|
||||
pattern = API_KEY_PATTERNS[name]
|
||||
|
||||
if len(value) < pattern["min_length"]:
|
||||
return False, f"Too short (expected >= {pattern['min_length']} chars)"
|
||||
|
||||
if pattern.get("prefix") and not value.startswith(pattern["prefix"]):
|
||||
return False, f"Invalid prefix (expected {pattern['prefix']}...)"
|
||||
|
||||
return True, "Valid format"
|
||||
|
||||
|
||||
@app.command("view")
|
||||
def env_view(
|
||||
show_values: bool = typer.Option(
|
||||
False,
|
||||
"--show-values",
|
||||
help="Show actual values (redacted by default)",
|
||||
),
|
||||
filter_prefix: Optional[str] = typer.Option(
|
||||
None,
|
||||
"--filter",
|
||||
"-f",
|
||||
help="Filter by prefix (e.g., OPENAI, PRAISONAI)",
|
||||
),
|
||||
):
|
||||
"""Show relevant environment variables."""
|
||||
output = get_output_controller()
|
||||
|
||||
# Relevant prefixes
|
||||
prefixes = ["OPENAI", "ANTHROPIC", "GOOGLE", "GEMINI", "TAVILY", "GROQ",
|
||||
"PRAISONAI", "MODEL", "LLM", "OLLAMA", "AZURE"]
|
||||
|
||||
if filter_prefix:
|
||||
prefixes = [filter_prefix.upper()]
|
||||
|
||||
env_vars = {}
|
||||
for key, value in os.environ.items():
|
||||
for prefix in prefixes:
|
||||
if key.upper().startswith(prefix):
|
||||
if show_values:
|
||||
if "KEY" in key.upper() or "SECRET" in key.upper() or "TOKEN" in key.upper():
|
||||
env_vars[key] = _redact_key(value)
|
||||
else:
|
||||
env_vars[key] = value
|
||||
else:
|
||||
env_vars[key] = "***" if value else "(empty)"
|
||||
break
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({"environment": env_vars})
|
||||
return
|
||||
|
||||
if not env_vars:
|
||||
output.print_info("No relevant environment variables found")
|
||||
return
|
||||
|
||||
headers = ["Variable", "Value"]
|
||||
rows = [[k, v] for k, v in sorted(env_vars.items())]
|
||||
output.print_table(headers, rows, title="Environment Variables")
|
||||
|
||||
|
||||
@app.command("check")
|
||||
def env_check():
|
||||
"""Validate API keys and configuration."""
|
||||
output = get_output_controller()
|
||||
|
||||
results = []
|
||||
all_valid = True
|
||||
|
||||
for key_name, pattern in API_KEY_PATTERNS.items():
|
||||
value = os.environ.get(key_name)
|
||||
|
||||
if value:
|
||||
valid, message = _validate_key(key_name, value)
|
||||
results.append({
|
||||
"key": key_name,
|
||||
"present": True,
|
||||
"valid": valid,
|
||||
"message": message,
|
||||
})
|
||||
if not valid:
|
||||
all_valid = False
|
||||
else:
|
||||
results.append({
|
||||
"key": key_name,
|
||||
"present": False,
|
||||
"valid": False,
|
||||
"message": "Not set",
|
||||
})
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({"checks": results, "all_valid": all_valid})
|
||||
return
|
||||
|
||||
headers = ["Key", "Status", "Message"]
|
||||
rows = []
|
||||
for r in results:
|
||||
if r["present"]:
|
||||
status = "✅" if r["valid"] else "⚠️"
|
||||
else:
|
||||
status = "❌"
|
||||
rows.append([r["key"], status, r["message"]])
|
||||
|
||||
output.print_table(headers, rows, title="API Key Validation")
|
||||
|
||||
if not all_valid:
|
||||
output.print_warning("Some API keys are missing or invalid")
|
||||
raise typer.Exit(3) # Exit code 3 = missing config/env
|
||||
|
||||
|
||||
@app.command("doctor")
|
||||
def env_doctor(
|
||||
deep: bool = typer.Option(
|
||||
False,
|
||||
"--deep",
|
||||
help="Run deep checks (network, DB connections)",
|
||||
),
|
||||
):
|
||||
"""Run environment diagnostics (alias for 'praisonai doctor env')."""
|
||||
output = get_output_controller()
|
||||
|
||||
# Delegate to doctor command
|
||||
try:
|
||||
from ..features.doctor.handler import DoctorHandler
|
||||
handler = DoctorHandler()
|
||||
|
||||
# Run env checks
|
||||
if output.is_json_mode:
|
||||
# Run and get JSON output
|
||||
result = handler.handle(["env", "--json"] if not deep else ["env", "--json", "--deep"])
|
||||
else:
|
||||
result = handler.handle(["env"] if not deep else ["env", "--deep"])
|
||||
|
||||
raise typer.Exit(result)
|
||||
except ImportError:
|
||||
# Fallback to basic env check
|
||||
env_check()
|
||||
@@ -0,0 +1,112 @@
|
||||
"""
|
||||
LSP command group for PraisonAI CLI.
|
||||
|
||||
Wraps existing LSP functionality from features/lsp_cli.py.
|
||||
Provides LSP service lifecycle management.
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
|
||||
app = typer.Typer(help="LSP service lifecycle management")
|
||||
|
||||
|
||||
@app.command("start")
|
||||
def lsp_start(
|
||||
workspace: str = typer.Option(".", "--workspace", "-w", help="Workspace root directory"),
|
||||
language: Optional[str] = typer.Option(None, "--language", "-l", help="Language to start"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Start LSP server(s)."""
|
||||
args = ["start", "--workspace", workspace]
|
||||
if language:
|
||||
args.extend(["--language", language])
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
|
||||
try:
|
||||
from ..features.lsp_cli import run_lsp_command
|
||||
exit_code = run_lsp_command(args)
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output = get_output_controller()
|
||||
output.print_error(f"LSP module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
|
||||
|
||||
@app.command("stop")
|
||||
def lsp_stop(
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Stop LSP server(s)."""
|
||||
args = ["stop"]
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
|
||||
try:
|
||||
from ..features.lsp_cli import run_lsp_command
|
||||
exit_code = run_lsp_command(args)
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output = get_output_controller()
|
||||
output.print_error(f"LSP module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
|
||||
|
||||
@app.command("status")
|
||||
def lsp_status(
|
||||
workspace: str = typer.Option(".", "--workspace", "-w", help="Workspace root directory"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Show LSP status."""
|
||||
args = ["status", "--workspace", workspace]
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
|
||||
try:
|
||||
from ..features.lsp_cli import run_lsp_command
|
||||
exit_code = run_lsp_command(args)
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output = get_output_controller()
|
||||
output.print_error(f"LSP module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
|
||||
|
||||
@app.command("logs")
|
||||
def lsp_logs(
|
||||
tail: int = typer.Option(50, "--tail", "-n", help="Number of lines to show"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Show recent LSP logs."""
|
||||
args = ["logs", "--tail", str(tail)]
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
|
||||
try:
|
||||
from ..features.lsp_cli import run_lsp_command
|
||||
exit_code = run_lsp_command(args)
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output = get_output_controller()
|
||||
output.print_error(f"LSP module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
|
||||
|
||||
@app.callback(invoke_without_command=True)
|
||||
def lsp_callback(ctx: typer.Context):
|
||||
"""Show LSP help."""
|
||||
if ctx.invoked_subcommand is None:
|
||||
output = get_output_controller()
|
||||
output.print_panel(
|
||||
"LSP service lifecycle management.\n\n"
|
||||
"Commands:\n"
|
||||
" start Start LSP server(s)\n"
|
||||
" stop Stop LSP server(s)\n"
|
||||
" status Show LSP status\n"
|
||||
" logs Show recent logs",
|
||||
title="LSP Commands"
|
||||
)
|
||||
@@ -0,0 +1,210 @@
|
||||
"""
|
||||
MCP command group for PraisonAI CLI.
|
||||
|
||||
Provides MCP (Model Context Protocol) server management.
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
from ..configuration.loader import get_config_loader
|
||||
|
||||
app = typer.Typer(help="MCP server management")
|
||||
|
||||
|
||||
@app.command("list")
|
||||
def mcp_list(
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""List configured MCP servers."""
|
||||
output = get_output_controller()
|
||||
loader = get_config_loader()
|
||||
|
||||
config = loader.load()
|
||||
servers = config.mcp.servers
|
||||
|
||||
if output.is_json_mode or json_output:
|
||||
output.print_json({
|
||||
"servers": {
|
||||
name: {
|
||||
"command": s.command,
|
||||
"args": s.args,
|
||||
"enabled": s.enabled,
|
||||
}
|
||||
for name, s in servers.items()
|
||||
}
|
||||
})
|
||||
return
|
||||
|
||||
if not servers:
|
||||
output.print_info("No MCP servers configured")
|
||||
output.print("\nAdd a server with: praisonai mcp add <name> <command>")
|
||||
return
|
||||
|
||||
headers = ["Name", "Command", "Enabled"]
|
||||
rows = []
|
||||
for name, server in servers.items():
|
||||
cmd = f"{server.command} {' '.join(server.args)}"
|
||||
if len(cmd) > 50:
|
||||
cmd = cmd[:47] + "..."
|
||||
rows.append([name, cmd, "✓" if server.enabled else "✗"])
|
||||
|
||||
output.print_table(headers, rows, title="MCP Servers")
|
||||
|
||||
|
||||
@app.command("add")
|
||||
def mcp_add(
|
||||
name: str = typer.Argument(..., help="Server name"),
|
||||
command: str = typer.Argument(..., help="Command to run"),
|
||||
args: Optional[str] = typer.Option(None, "--args", "-a", help="Command arguments (space-separated)"),
|
||||
env: Optional[str] = typer.Option(None, "--env", "-e", help="Environment variables (KEY=VALUE,...)"),
|
||||
):
|
||||
"""Add an MCP server."""
|
||||
output = get_output_controller()
|
||||
loader = get_config_loader()
|
||||
|
||||
# Parse args
|
||||
arg_list = args.split() if args else []
|
||||
|
||||
# Parse env
|
||||
env_dict = {}
|
||||
if env:
|
||||
for pair in env.split(","):
|
||||
if "=" in pair:
|
||||
key, value = pair.split("=", 1)
|
||||
env_dict[key.strip()] = value.strip()
|
||||
|
||||
# Build server config
|
||||
server_config = {
|
||||
"command": command,
|
||||
"args": arg_list,
|
||||
"env": env_dict,
|
||||
"enabled": True,
|
||||
}
|
||||
|
||||
# Save to config
|
||||
loader.set(f"mcp.servers.{name}", server_config)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({"added": name, "config": server_config})
|
||||
else:
|
||||
output.print_success(f"Added MCP server: {name}")
|
||||
|
||||
|
||||
@app.command("remove")
|
||||
def mcp_remove(
|
||||
name: str = typer.Argument(..., help="Server name to remove"),
|
||||
confirm: bool = typer.Option(False, "--yes", "-y", help="Skip confirmation"),
|
||||
):
|
||||
"""Remove an MCP server."""
|
||||
output = get_output_controller()
|
||||
loader = get_config_loader()
|
||||
|
||||
config = loader.load()
|
||||
if name not in config.mcp.servers:
|
||||
output.print_error(f"Server not found: {name}")
|
||||
raise typer.Exit(1)
|
||||
|
||||
if not confirm:
|
||||
confirmed = typer.confirm(f"Remove MCP server '{name}'?")
|
||||
if not confirmed:
|
||||
output.print_info("Cancelled")
|
||||
raise typer.Exit(0)
|
||||
|
||||
# Remove from config by setting to None (will be cleaned up)
|
||||
# For now, we need to reload and modify the raw config
|
||||
from ..configuration.paths import get_user_config_path
|
||||
from ..configuration.loader import _load_toml, _save_toml
|
||||
|
||||
config_path = get_user_config_path()
|
||||
if config_path.exists():
|
||||
raw_config = _load_toml(config_path)
|
||||
if "mcp" in raw_config and "servers" in raw_config["mcp"]:
|
||||
if name in raw_config["mcp"]["servers"]:
|
||||
del raw_config["mcp"]["servers"][name]
|
||||
_save_toml(config_path, raw_config)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({"removed": name})
|
||||
else:
|
||||
output.print_success(f"Removed MCP server: {name}")
|
||||
|
||||
|
||||
@app.command("test")
|
||||
def mcp_test(
|
||||
name: str = typer.Argument(..., help="Server name to test"),
|
||||
timeout: float = typer.Option(10.0, "--timeout", "-t", help="Timeout in seconds"),
|
||||
):
|
||||
"""Test an MCP server connection."""
|
||||
output = get_output_controller()
|
||||
loader = get_config_loader()
|
||||
|
||||
config = loader.load()
|
||||
if name not in config.mcp.servers:
|
||||
output.print_error(f"Server not found: {name}")
|
||||
raise typer.Exit(1)
|
||||
|
||||
server = config.mcp.servers[name]
|
||||
|
||||
output.print_info(f"Testing MCP server: {name}")
|
||||
output.print(f" Command: {server.command} {' '.join(server.args)}")
|
||||
|
||||
# Try to start the server and check if it responds
|
||||
import subprocess
|
||||
import time
|
||||
|
||||
try:
|
||||
cmd = [server.command] + server.args
|
||||
env = dict(**server.env) if server.env else None
|
||||
|
||||
# Start process
|
||||
proc = subprocess.Popen(
|
||||
cmd,
|
||||
stdin=subprocess.PIPE,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
env=env,
|
||||
)
|
||||
|
||||
# Wait briefly and check if it's running
|
||||
time.sleep(0.5)
|
||||
|
||||
if proc.poll() is None:
|
||||
# Process is running
|
||||
proc.terminate()
|
||||
proc.wait(timeout=2)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({"name": name, "status": "ok", "message": "Server started successfully"})
|
||||
else:
|
||||
output.print_success(f"Server '{name}' started successfully")
|
||||
else:
|
||||
# Process exited
|
||||
stderr = proc.stderr.read().decode() if proc.stderr else ""
|
||||
if output.is_json_mode:
|
||||
output.print_json({"name": name, "status": "error", "message": stderr or "Server exited immediately"})
|
||||
else:
|
||||
output.print_error(f"Server exited immediately: {stderr}")
|
||||
raise typer.Exit(1)
|
||||
|
||||
except FileNotFoundError:
|
||||
if output.is_json_mode:
|
||||
output.print_json({"name": name, "status": "error", "message": f"Command not found: {server.command}"})
|
||||
else:
|
||||
output.print_error(f"Command not found: {server.command}")
|
||||
raise typer.Exit(1)
|
||||
except Exception as e:
|
||||
if output.is_json_mode:
|
||||
output.print_json({"name": name, "status": "error", "message": str(e)})
|
||||
else:
|
||||
output.print_error(f"Test failed: {e}")
|
||||
raise typer.Exit(1)
|
||||
|
||||
|
||||
@app.callback(invoke_without_command=True)
|
||||
def mcp_callback(ctx: typer.Context):
|
||||
"""Show MCP help or list servers."""
|
||||
if ctx.invoked_subcommand is None:
|
||||
mcp_list(json_output=False)
|
||||
@@ -0,0 +1,228 @@
|
||||
"""
|
||||
Run command group for PraisonAI CLI.
|
||||
|
||||
Provides agent execution commands.
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
from ..state.identifiers import get_current_context
|
||||
|
||||
app = typer.Typer(help="Run agents")
|
||||
|
||||
|
||||
@app.callback(invoke_without_command=True)
|
||||
def run_main(
|
||||
ctx: typer.Context,
|
||||
target: Optional[str] = typer.Argument(None, help="Agent file or prompt"),
|
||||
model: Optional[str] = typer.Option(None, "--model", "-m", help="LLM model to use"),
|
||||
framework: Optional[str] = typer.Option(None, "--framework", "-f", help="Framework: praisonai, crewai, autogen"),
|
||||
interactive: bool = typer.Option(False, "--interactive", "-i", help="Interactive mode"),
|
||||
verbose: bool = typer.Option(False, "--verbose", "-v", help="Verbose output"),
|
||||
stream: bool = typer.Option(True, "--stream/--no-stream", help="Stream output"),
|
||||
trace: bool = typer.Option(False, "--trace", help="Enable tracing"),
|
||||
memory: bool = typer.Option(False, "--memory", help="Enable memory"),
|
||||
tools: Optional[str] = typer.Option(None, "--tools", "-t", help="Tools file path"),
|
||||
max_tokens: int = typer.Option(16000, "--max-tokens", help="Maximum output tokens"),
|
||||
):
|
||||
"""
|
||||
Run agents from a file or prompt.
|
||||
|
||||
Examples:
|
||||
praisonai run agents.yaml
|
||||
praisonai run "What is the weather?"
|
||||
praisonai run agents.yaml --interactive
|
||||
"""
|
||||
output = get_output_controller()
|
||||
context = get_current_context()
|
||||
|
||||
if not target:
|
||||
output.print_panel(
|
||||
"Run agents from a file or prompt.\n\n"
|
||||
"Usage:\n"
|
||||
" praisonai run agents.yaml\n"
|
||||
" praisonai run \"What is the weather?\"\n"
|
||||
" praisonai run agents.yaml --interactive\n\n"
|
||||
"Options:\n"
|
||||
" --model, -m LLM model to use\n"
|
||||
" --framework, -f Framework (praisonai, crewai, autogen)\n"
|
||||
" --interactive, -i Interactive mode\n"
|
||||
" --verbose, -v Verbose output\n"
|
||||
" --trace Enable tracing\n"
|
||||
" --memory Enable memory",
|
||||
title="Run Command"
|
||||
)
|
||||
return
|
||||
|
||||
# Emit start event
|
||||
output.emit_start(
|
||||
message=f"Starting run: {target[:50]}..." if len(target) > 50 else f"Starting run: {target}",
|
||||
data={
|
||||
"target": target,
|
||||
"model": model,
|
||||
"framework": framework,
|
||||
}
|
||||
)
|
||||
|
||||
# Check if target is a file or prompt
|
||||
import os
|
||||
is_file = os.path.exists(target) and (target.endswith('.yaml') or target.endswith('.yml'))
|
||||
|
||||
if is_file:
|
||||
# Run from file
|
||||
_run_from_file(
|
||||
target,
|
||||
model=model,
|
||||
framework=framework,
|
||||
interactive=interactive,
|
||||
verbose=verbose,
|
||||
stream=stream,
|
||||
trace=trace,
|
||||
memory=memory,
|
||||
tools=tools,
|
||||
max_tokens=max_tokens,
|
||||
)
|
||||
else:
|
||||
# Run as prompt
|
||||
_run_prompt(
|
||||
target,
|
||||
model=model,
|
||||
verbose=verbose,
|
||||
stream=stream,
|
||||
trace=trace,
|
||||
memory=memory,
|
||||
tools=tools,
|
||||
max_tokens=max_tokens,
|
||||
)
|
||||
|
||||
|
||||
def _run_from_file(
|
||||
file_path: str,
|
||||
model: Optional[str] = None,
|
||||
framework: Optional[str] = None,
|
||||
interactive: bool = False,
|
||||
verbose: bool = False,
|
||||
stream: bool = True,
|
||||
trace: bool = False,
|
||||
memory: bool = False,
|
||||
tools: Optional[str] = None,
|
||||
max_tokens: int = 16000,
|
||||
):
|
||||
"""Run agents from a YAML file."""
|
||||
output = get_output_controller()
|
||||
|
||||
try:
|
||||
# Use existing PraisonAI class
|
||||
from praisonai.cli.main import PraisonAI
|
||||
|
||||
praison = PraisonAI(
|
||||
agent_file=file_path,
|
||||
framework=framework or "praisonai",
|
||||
)
|
||||
|
||||
# Set model if provided
|
||||
if model:
|
||||
praison.config_list[0]['model'] = model
|
||||
|
||||
# Run
|
||||
result = praison.run()
|
||||
|
||||
output.emit_result(
|
||||
message="Run completed",
|
||||
data={"result": str(result) if result else None}
|
||||
)
|
||||
|
||||
if result:
|
||||
if not output.is_json_mode:
|
||||
output.print_success("Run completed")
|
||||
|
||||
except Exception as e:
|
||||
output.emit_error(message=str(e))
|
||||
output.print_error(str(e))
|
||||
raise typer.Exit(1)
|
||||
|
||||
|
||||
def _run_prompt(
|
||||
prompt: str,
|
||||
model: Optional[str] = None,
|
||||
verbose: bool = False,
|
||||
stream: bool = True,
|
||||
trace: bool = False,
|
||||
memory: bool = False,
|
||||
tools: Optional[str] = None,
|
||||
max_tokens: int = 16000,
|
||||
):
|
||||
"""Run a direct prompt."""
|
||||
output = get_output_controller()
|
||||
|
||||
try:
|
||||
# Use existing handle_direct_prompt
|
||||
from praisonai.cli.main import PraisonAI
|
||||
|
||||
praison = PraisonAI()
|
||||
|
||||
if model:
|
||||
praison.config_list[0]['model'] = model
|
||||
|
||||
# Create args-like object for handle_direct_prompt
|
||||
class Args:
|
||||
pass
|
||||
|
||||
args = Args()
|
||||
args.llm = model
|
||||
args.verbose = verbose
|
||||
args.memory = memory
|
||||
args.tools = tools
|
||||
args.max_tokens = max_tokens
|
||||
args.web_search = False
|
||||
args.web_fetch = False
|
||||
args.prompt_caching = False
|
||||
args.planning = False
|
||||
args.planning_tools = None
|
||||
args.planning_reasoning = False
|
||||
args.auto_approve_plan = False
|
||||
args.final_agent = None
|
||||
args.user_id = None
|
||||
args.auto_save = None
|
||||
args.history = None
|
||||
args.include_rules = None
|
||||
args.workflow = None
|
||||
args.workflow_var = None
|
||||
args.claude_memory = False
|
||||
args.guardrail = None
|
||||
args.metrics = False
|
||||
args.image = None
|
||||
args.image_generate = False
|
||||
args.telemetry = False
|
||||
args.mcp = None
|
||||
args.mcp_env = None
|
||||
args.fast_context = None
|
||||
args.handoff = None
|
||||
args.auto_memory = False
|
||||
args.todo = False
|
||||
args.router = False
|
||||
args.router_provider = None
|
||||
args.query_rewrite = False
|
||||
args.rewrite_tools = None
|
||||
args.expand_prompt = False
|
||||
args.expand_tools = None
|
||||
args.no_tools = False
|
||||
|
||||
praison.args = args
|
||||
result = praison.handle_direct_prompt(prompt)
|
||||
|
||||
output.emit_result(
|
||||
message="Prompt completed",
|
||||
data={"result": str(result) if result else None}
|
||||
)
|
||||
|
||||
if result and not output.is_json_mode:
|
||||
print(result)
|
||||
|
||||
except Exception as e:
|
||||
output.emit_error(message=str(e))
|
||||
output.print_error(str(e))
|
||||
raise typer.Exit(1)
|
||||
@@ -0,0 +1,150 @@
|
||||
"""
|
||||
Schedule command group for PraisonAI CLI.
|
||||
|
||||
Provides scheduler management.
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
|
||||
app = typer.Typer(help="Scheduler management")
|
||||
|
||||
|
||||
def _run_schedule(args: list) -> int:
|
||||
"""Run schedule command with args."""
|
||||
try:
|
||||
from ..features.agent_scheduler import AgentSchedulerHandler
|
||||
|
||||
# Parse subcommand
|
||||
if args and args[0] in ['start', 'list', 'stop', 'logs', 'restart', 'delete', 'describe', 'save', 'stop-all', 'stats']:
|
||||
subcommand = args[0]
|
||||
remaining = args[1:] if len(args) > 1 else []
|
||||
|
||||
# Create minimal args namespace
|
||||
class Args:
|
||||
pass
|
||||
|
||||
fake_args = Args()
|
||||
return AgentSchedulerHandler.handle_daemon_command(subcommand, fake_args, remaining)
|
||||
|
||||
return 0
|
||||
except ImportError as e:
|
||||
output = get_output_controller()
|
||||
output.print_error(f"Scheduler module not available: {e}")
|
||||
return 4
|
||||
|
||||
|
||||
@app.command("start")
|
||||
def schedule_start(
|
||||
agents_file: str = typer.Argument("agents.yaml", help="Agents YAML file"),
|
||||
interval: Optional[str] = typer.Option(None, "--interval", "-i", help="Schedule interval (e.g., 'hourly', '*/30m')"),
|
||||
daemon: bool = typer.Option(True, "--daemon/--no-daemon", help="Run as daemon"),
|
||||
name: Optional[str] = typer.Option(None, "--name", "-n", help="Job name"),
|
||||
):
|
||||
"""Start scheduled agent execution."""
|
||||
args = ["start", agents_file]
|
||||
if interval:
|
||||
args.extend(["--interval", interval])
|
||||
if not daemon:
|
||||
args.append("--no-daemon")
|
||||
if name:
|
||||
args.extend(["--name", name])
|
||||
|
||||
raise typer.Exit(_run_schedule(args))
|
||||
|
||||
|
||||
@app.command("stop")
|
||||
def schedule_stop(
|
||||
job_id: Optional[str] = typer.Argument(None, help="Job ID to stop (or 'all')"),
|
||||
):
|
||||
"""Stop scheduled job(s)."""
|
||||
if job_id == "all":
|
||||
raise typer.Exit(_run_schedule(["stop-all"]))
|
||||
elif job_id:
|
||||
raise typer.Exit(_run_schedule(["stop", job_id]))
|
||||
else:
|
||||
raise typer.Exit(_run_schedule(["stop"]))
|
||||
|
||||
|
||||
@app.command("list")
|
||||
def schedule_list(
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""List scheduled jobs."""
|
||||
args = ["list"]
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
raise typer.Exit(_run_schedule(args))
|
||||
|
||||
|
||||
@app.command("logs")
|
||||
def schedule_logs(
|
||||
job_id: Optional[str] = typer.Argument(None, help="Job ID"),
|
||||
tail: int = typer.Option(50, "--tail", "-n", help="Number of lines"),
|
||||
follow: bool = typer.Option(False, "--follow", "-f", help="Follow log output"),
|
||||
):
|
||||
"""View scheduler logs."""
|
||||
args = ["logs"]
|
||||
if job_id:
|
||||
args.append(job_id)
|
||||
args.extend(["--tail", str(tail)])
|
||||
if follow:
|
||||
args.append("--follow")
|
||||
raise typer.Exit(_run_schedule(args))
|
||||
|
||||
|
||||
@app.command("restart")
|
||||
def schedule_restart(
|
||||
job_id: str = typer.Argument(..., help="Job ID to restart"),
|
||||
):
|
||||
"""Restart a scheduled job."""
|
||||
raise typer.Exit(_run_schedule(["restart", job_id]))
|
||||
|
||||
|
||||
@app.command("delete")
|
||||
def schedule_delete(
|
||||
job_id: str = typer.Argument(..., help="Job ID to delete"),
|
||||
confirm: bool = typer.Option(False, "--yes", "-y", help="Skip confirmation"),
|
||||
):
|
||||
"""Delete a scheduled job."""
|
||||
if not confirm:
|
||||
confirmed = typer.confirm(f"Delete job {job_id}?")
|
||||
if not confirmed:
|
||||
output = get_output_controller()
|
||||
output.print_info("Cancelled")
|
||||
raise typer.Exit(0)
|
||||
|
||||
raise typer.Exit(_run_schedule(["delete", job_id]))
|
||||
|
||||
|
||||
@app.command("describe")
|
||||
def schedule_describe(
|
||||
job_id: str = typer.Argument(..., help="Job ID"),
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Show job details."""
|
||||
args = ["describe", job_id]
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
raise typer.Exit(_run_schedule(args))
|
||||
|
||||
|
||||
@app.command("stats")
|
||||
def schedule_stats(
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Show scheduler statistics."""
|
||||
args = ["stats"]
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
raise typer.Exit(_run_schedule(args))
|
||||
|
||||
|
||||
@app.callback(invoke_without_command=True)
|
||||
def schedule_callback(ctx: typer.Context):
|
||||
"""Show schedule help or list jobs."""
|
||||
if ctx.invoked_subcommand is None:
|
||||
schedule_list(json_output=False)
|
||||
@@ -0,0 +1,97 @@
|
||||
"""
|
||||
Serve command group for PraisonAI CLI.
|
||||
|
||||
Provides API server management.
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
|
||||
app = typer.Typer(help="API server management")
|
||||
|
||||
|
||||
@app.command("start")
|
||||
def serve_start(
|
||||
host: str = typer.Option("0.0.0.0", "--host", "-h", help="Host to bind to"),
|
||||
port: int = typer.Option(8000, "--port", "-p", help="Port to bind to"),
|
||||
agents_file: Optional[str] = typer.Option(None, "--agents", "-a", help="Agents YAML file"),
|
||||
reload: bool = typer.Option(False, "--reload", help="Enable auto-reload"),
|
||||
workers: int = typer.Option(1, "--workers", "-w", help="Number of workers"),
|
||||
):
|
||||
"""Start API server."""
|
||||
output = get_output_controller()
|
||||
|
||||
# Build args for existing handler
|
||||
args = [
|
||||
"start",
|
||||
"--host", host,
|
||||
"--port", str(port),
|
||||
"--workers", str(workers),
|
||||
]
|
||||
|
||||
if agents_file:
|
||||
args.extend(["--agents", agents_file])
|
||||
if reload:
|
||||
args.append("--reload")
|
||||
|
||||
try:
|
||||
from ..features.serve import handle_serve_command
|
||||
exit_code = handle_serve_command(args)
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output.print_error(f"Serve module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
|
||||
|
||||
@app.command("stop")
|
||||
def serve_stop():
|
||||
"""Stop API server."""
|
||||
output = get_output_controller()
|
||||
|
||||
try:
|
||||
from ..features.serve import handle_serve_command
|
||||
exit_code = handle_serve_command(["stop"])
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output.print_error(f"Serve module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
|
||||
|
||||
@app.command("status")
|
||||
def serve_status(
|
||||
json_output: bool = typer.Option(False, "--json", help="Output JSON"),
|
||||
):
|
||||
"""Show server status."""
|
||||
output = get_output_controller()
|
||||
|
||||
args = ["status"]
|
||||
if json_output:
|
||||
args.append("--json")
|
||||
|
||||
try:
|
||||
from ..features.serve import handle_serve_command
|
||||
exit_code = handle_serve_command(args)
|
||||
raise typer.Exit(exit_code)
|
||||
except ImportError as e:
|
||||
output.print_error(f"Serve module not available: {e}")
|
||||
raise typer.Exit(4)
|
||||
|
||||
|
||||
@app.callback(invoke_without_command=True)
|
||||
def serve_callback(ctx: typer.Context):
|
||||
"""Show serve help or start server."""
|
||||
if ctx.invoked_subcommand is None:
|
||||
output = get_output_controller()
|
||||
output.print_panel(
|
||||
"API server management.\n\n"
|
||||
"Commands:\n"
|
||||
" start Start API server\n"
|
||||
" stop Stop API server\n"
|
||||
" status Show server status\n\n"
|
||||
"Quick start:\n"
|
||||
" praisonai serve start --agents agents.yaml",
|
||||
title="Serve Commands"
|
||||
)
|
||||
@@ -0,0 +1,212 @@
|
||||
"""
|
||||
Session command group for PraisonAI CLI.
|
||||
|
||||
Provides session management:
|
||||
- session list: List all sessions
|
||||
- session resume: Resume a session
|
||||
- session delete: Delete a session
|
||||
- session export: Export a session
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
from ..state.sessions import get_session_manager
|
||||
|
||||
app = typer.Typer(help="Session management")
|
||||
|
||||
|
||||
@app.command("list")
|
||||
def session_list(
|
||||
limit: int = typer.Option(
|
||||
20,
|
||||
"--limit",
|
||||
"-n",
|
||||
help="Maximum number of sessions to show",
|
||||
),
|
||||
):
|
||||
"""List all sessions."""
|
||||
output = get_output_controller()
|
||||
manager = get_session_manager()
|
||||
|
||||
sessions = manager.list(limit=limit)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({
|
||||
"sessions": [s.to_dict() for s in sessions]
|
||||
})
|
||||
return
|
||||
|
||||
if not sessions:
|
||||
output.print_info("No sessions found")
|
||||
return
|
||||
|
||||
headers = ["ID", "Name", "Status", "Events", "Updated"]
|
||||
rows = []
|
||||
for s in sessions:
|
||||
rows.append([
|
||||
s.session_id[:20] + "..." if len(s.session_id) > 20 else s.session_id,
|
||||
s.name or "-",
|
||||
s.status,
|
||||
str(s.event_count),
|
||||
s.updated_at.strftime("%Y-%m-%d %H:%M"),
|
||||
])
|
||||
|
||||
output.print_table(headers, rows, title="Sessions")
|
||||
|
||||
|
||||
@app.command("resume")
|
||||
def session_resume(
|
||||
session_id: str = typer.Argument(..., help="Session ID to resume"),
|
||||
):
|
||||
"""Resume a session."""
|
||||
output = get_output_controller()
|
||||
manager = get_session_manager()
|
||||
|
||||
session = manager.get(session_id)
|
||||
|
||||
if not session:
|
||||
output.print_error(
|
||||
f"Session not found: {session_id}",
|
||||
remediation="Use 'praisonai session list' to see available sessions"
|
||||
)
|
||||
raise typer.Exit(1)
|
||||
|
||||
# Load session events
|
||||
events = manager.get_events(session_id)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({
|
||||
"session": session.to_dict(),
|
||||
"events": events,
|
||||
"message": "Session loaded. True resume not supported - showing transcript.",
|
||||
})
|
||||
return
|
||||
|
||||
output.print_panel(
|
||||
f"Session: {session.name or session.session_id}\n"
|
||||
f"Run ID: {session.run_id}\n"
|
||||
f"Trace ID: {session.trace_id}\n"
|
||||
f"Events: {session.event_count}\n"
|
||||
f"Status: {session.status}",
|
||||
title="Session Loaded"
|
||||
)
|
||||
|
||||
output.print_warning(
|
||||
"True session resume is not yet supported. "
|
||||
"Showing session transcript instead."
|
||||
)
|
||||
|
||||
# Show recent events
|
||||
if events:
|
||||
output.print("\n--- Recent Events ---\n")
|
||||
for event in events[-10:]:
|
||||
event_type = event.get("event", "unknown")
|
||||
message = event.get("message", "")
|
||||
output.print(f"[{event_type}] {message}")
|
||||
|
||||
|
||||
@app.command("delete")
|
||||
def session_delete(
|
||||
session_id: str = typer.Argument(..., help="Session ID to delete"),
|
||||
confirm: bool = typer.Option(
|
||||
False,
|
||||
"--yes",
|
||||
"-y",
|
||||
help="Skip confirmation",
|
||||
),
|
||||
):
|
||||
"""Delete a session."""
|
||||
output = get_output_controller()
|
||||
manager = get_session_manager()
|
||||
|
||||
session = manager.get(session_id)
|
||||
|
||||
if not session:
|
||||
output.print_error(f"Session not found: {session_id}")
|
||||
raise typer.Exit(1)
|
||||
|
||||
if not confirm:
|
||||
confirmed = typer.confirm(f"Delete session {session_id}?")
|
||||
if not confirmed:
|
||||
output.print_info("Cancelled")
|
||||
raise typer.Exit(0)
|
||||
|
||||
deleted = manager.delete(session_id)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({"deleted": deleted, "session_id": session_id})
|
||||
else:
|
||||
if deleted:
|
||||
output.print_success(f"Deleted session: {session_id}")
|
||||
else:
|
||||
output.print_error(f"Failed to delete session: {session_id}")
|
||||
raise typer.Exit(1)
|
||||
|
||||
|
||||
@app.command("export")
|
||||
def session_export(
|
||||
session_id: str = typer.Argument(..., help="Session ID to export"),
|
||||
format: str = typer.Option(
|
||||
"md",
|
||||
"--format",
|
||||
"-f",
|
||||
help="Export format: md or json",
|
||||
),
|
||||
output_file: Optional[str] = typer.Option(
|
||||
None,
|
||||
"--output",
|
||||
"-o",
|
||||
help="Output file path",
|
||||
),
|
||||
):
|
||||
"""Export a session."""
|
||||
output = get_output_controller()
|
||||
manager = get_session_manager()
|
||||
|
||||
content = manager.export(session_id, format=format)
|
||||
|
||||
if content is None:
|
||||
output.print_error(f"Session not found: {session_id}")
|
||||
raise typer.Exit(1)
|
||||
|
||||
if output_file:
|
||||
with open(output_file, "w", encoding="utf-8") as f:
|
||||
f.write(content)
|
||||
output.print_success(f"Exported to: {output_file}")
|
||||
else:
|
||||
print(content)
|
||||
|
||||
|
||||
@app.command("show")
|
||||
def session_show(
|
||||
session_id: str = typer.Argument(..., help="Session ID to show"),
|
||||
):
|
||||
"""Show session details."""
|
||||
output = get_output_controller()
|
||||
manager = get_session_manager()
|
||||
|
||||
session = manager.get(session_id)
|
||||
|
||||
if not session:
|
||||
output.print_error(f"Session not found: {session_id}")
|
||||
raise typer.Exit(1)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json(session.to_dict())
|
||||
return
|
||||
|
||||
output.print_panel(
|
||||
f"Session ID: {session.session_id}\n"
|
||||
f"Name: {session.name or '-'}\n"
|
||||
f"Run ID: {session.run_id}\n"
|
||||
f"Trace ID: {session.trace_id}\n"
|
||||
f"Created: {session.created_at.isoformat()}\n"
|
||||
f"Updated: {session.updated_at.isoformat()}\n"
|
||||
f"Status: {session.status}\n"
|
||||
f"Events: {session.event_count}\n"
|
||||
f"Workspace: {session.workspace or '-'}",
|
||||
title="Session Details"
|
||||
)
|
||||
@@ -0,0 +1,145 @@
|
||||
"""
|
||||
Traces command group for PraisonAI CLI.
|
||||
|
||||
Provides trace collection management:
|
||||
- traces enable: Enable trace collection
|
||||
- traces disable: Disable trace collection
|
||||
- traces status: Show current status
|
||||
- traces list: List recent traces
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
|
||||
from ..configuration.loader import get_config_loader
|
||||
from ..output.console import get_output_controller
|
||||
|
||||
app = typer.Typer(help="Trace collection management")
|
||||
|
||||
|
||||
@app.command("enable")
|
||||
def traces_enable(
|
||||
endpoint: Optional[str] = typer.Option(
|
||||
None,
|
||||
"--endpoint",
|
||||
"-e",
|
||||
help="Trace endpoint URL",
|
||||
),
|
||||
sample_rate: float = typer.Option(
|
||||
1.0,
|
||||
"--sample-rate",
|
||||
"-r",
|
||||
help="Sample rate (0.0 to 1.0)",
|
||||
),
|
||||
):
|
||||
"""Enable trace collection."""
|
||||
output = get_output_controller()
|
||||
loader = get_config_loader()
|
||||
|
||||
loader.set("traces.enabled", True)
|
||||
if endpoint:
|
||||
loader.set("traces.endpoint", endpoint)
|
||||
if sample_rate != 1.0:
|
||||
loader.set("traces.sample_rate", sample_rate)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({
|
||||
"traces_enabled": True,
|
||||
"endpoint": endpoint,
|
||||
"sample_rate": sample_rate,
|
||||
})
|
||||
else:
|
||||
output.print_success("Trace collection enabled")
|
||||
if endpoint:
|
||||
output.print(f" Endpoint: {endpoint}")
|
||||
output.print(f" Sample rate: {sample_rate}")
|
||||
|
||||
|
||||
@app.command("disable")
|
||||
def traces_disable():
|
||||
"""Disable trace collection."""
|
||||
output = get_output_controller()
|
||||
loader = get_config_loader()
|
||||
|
||||
loader.set("traces.enabled", False)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({"traces_enabled": False})
|
||||
else:
|
||||
output.print_success("Trace collection disabled")
|
||||
|
||||
|
||||
@app.command("status")
|
||||
def traces_status():
|
||||
"""Show trace collection status."""
|
||||
output = get_output_controller()
|
||||
loader = get_config_loader()
|
||||
|
||||
enabled = loader.get("traces.enabled", False)
|
||||
endpoint = loader.get("traces.endpoint")
|
||||
sample_rate = loader.get("traces.sample_rate", 1.0)
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({
|
||||
"enabled": enabled,
|
||||
"endpoint": endpoint,
|
||||
"sample_rate": sample_rate,
|
||||
})
|
||||
else:
|
||||
status = "✅ Enabled" if enabled else "❌ Disabled"
|
||||
output.print_panel(
|
||||
f"Status: {status}\n"
|
||||
f"Endpoint: {endpoint or 'Not configured'}\n"
|
||||
f"Sample Rate: {sample_rate}",
|
||||
title="Trace Collection Status"
|
||||
)
|
||||
|
||||
|
||||
@app.command("list")
|
||||
def traces_list(
|
||||
limit: int = typer.Option(
|
||||
20,
|
||||
"--limit",
|
||||
"-n",
|
||||
help="Maximum number of traces to show",
|
||||
),
|
||||
):
|
||||
"""List recent traces."""
|
||||
output = get_output_controller()
|
||||
|
||||
from ..configuration.paths import get_traces_dir
|
||||
|
||||
traces_dir = get_traces_dir()
|
||||
|
||||
if not traces_dir.exists():
|
||||
if output.is_json_mode:
|
||||
output.print_json({"traces": []})
|
||||
else:
|
||||
output.print_info("No traces found")
|
||||
return
|
||||
|
||||
traces = []
|
||||
for trace_file in sorted(traces_dir.iterdir(), reverse=True)[:limit]:
|
||||
if trace_file.is_file() and trace_file.suffix == ".json":
|
||||
traces.append({
|
||||
"name": trace_file.stem,
|
||||
"size": trace_file.stat().st_size,
|
||||
"modified": trace_file.stat().st_mtime,
|
||||
})
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({"traces": traces})
|
||||
else:
|
||||
if not traces:
|
||||
output.print_info("No traces found")
|
||||
return
|
||||
|
||||
headers = ["Name", "Size", "Modified"]
|
||||
rows = []
|
||||
for t in traces:
|
||||
from datetime import datetime
|
||||
modified = datetime.fromtimestamp(t["modified"]).strftime("%Y-%m-%d %H:%M")
|
||||
rows.append([t["name"], f"{t['size']} bytes", modified])
|
||||
|
||||
output.print_table(headers, rows, title="Recent Traces")
|
||||
@@ -0,0 +1,101 @@
|
||||
"""
|
||||
Version command group for PraisonAI CLI.
|
||||
|
||||
Provides version information:
|
||||
- version show: Show version
|
||||
- version check: Check for updates
|
||||
"""
|
||||
|
||||
import typer
|
||||
|
||||
from ..output.console import get_output_controller
|
||||
|
||||
app = typer.Typer(help="Version information")
|
||||
|
||||
|
||||
@app.command("show")
|
||||
def version_show():
|
||||
"""Show version information."""
|
||||
output = get_output_controller()
|
||||
|
||||
from praisonai.version import __version__
|
||||
|
||||
# Try to get additional version info
|
||||
versions = {
|
||||
"praisonai": __version__,
|
||||
}
|
||||
|
||||
# Check praisonaiagents version
|
||||
try:
|
||||
import praisonaiagents
|
||||
versions["praisonaiagents"] = getattr(praisonaiagents, "__version__", "unknown")
|
||||
except ImportError:
|
||||
versions["praisonaiagents"] = "not installed"
|
||||
|
||||
# Check Python version
|
||||
import sys
|
||||
versions["python"] = f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}"
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json(versions)
|
||||
return
|
||||
|
||||
output.print_panel(
|
||||
f"PraisonAI: {versions['praisonai']}\n"
|
||||
f"PraisonAI Agents: {versions['praisonaiagents']}\n"
|
||||
f"Python: {versions['python']}",
|
||||
title="Version Information"
|
||||
)
|
||||
|
||||
|
||||
@app.command("check")
|
||||
def version_check():
|
||||
"""Check for updates."""
|
||||
output = get_output_controller()
|
||||
|
||||
from praisonai.version import __version__
|
||||
|
||||
output.print_info("Checking for updates...")
|
||||
|
||||
try:
|
||||
import urllib.request
|
||||
import json
|
||||
|
||||
url = "https://pypi.org/pypi/praisonai/json"
|
||||
with urllib.request.urlopen(url, timeout=5) as response:
|
||||
data = json.loads(response.read().decode())
|
||||
latest = data["info"]["version"]
|
||||
|
||||
if output.is_json_mode:
|
||||
output.print_json({
|
||||
"current": __version__,
|
||||
"latest": latest,
|
||||
"update_available": latest != __version__,
|
||||
})
|
||||
return
|
||||
|
||||
if latest != __version__:
|
||||
output.print_warning(
|
||||
f"Update available: {__version__} → {latest}\n"
|
||||
f"Run: pip install --upgrade praisonai"
|
||||
)
|
||||
else:
|
||||
output.print_success(f"You are using the latest version ({__version__})")
|
||||
|
||||
except Exception as e:
|
||||
if output.is_json_mode:
|
||||
output.print_json({
|
||||
"current": __version__,
|
||||
"latest": None,
|
||||
"error": str(e),
|
||||
})
|
||||
else:
|
||||
output.print_error(f"Failed to check for updates: {e}")
|
||||
output.print(f"Current version: {__version__}")
|
||||
|
||||
|
||||
@app.callback(invoke_without_command=True)
|
||||
def version_callback(ctx: typer.Context):
|
||||
"""Show version (default behavior)."""
|
||||
if ctx.invoked_subcommand is None:
|
||||
version_show()
|
||||
@@ -0,0 +1,18 @@
|
||||
"""
|
||||
PraisonAI CLI Configuration Module.
|
||||
|
||||
Provides configuration management with TOML support and precedence handling.
|
||||
"""
|
||||
|
||||
from .loader import ConfigLoader, get_config
|
||||
from .schema import ConfigSchema
|
||||
from .paths import get_config_paths, get_user_config_path, get_project_config_path
|
||||
|
||||
__all__ = [
|
||||
'ConfigLoader',
|
||||
'ConfigSchema',
|
||||
'get_config',
|
||||
'get_config_paths',
|
||||
'get_user_config_path',
|
||||
'get_project_config_path',
|
||||
]
|
||||
@@ -0,0 +1,353 @@
|
||||
"""
|
||||
Configuration loader for PraisonAI CLI.
|
||||
|
||||
Handles loading, merging, and saving configuration with precedence.
|
||||
"""
|
||||
|
||||
import os
|
||||
import threading
|
||||
from pathlib import Path
|
||||
from typing import Any, Dict, Optional
|
||||
|
||||
from .paths import (
|
||||
get_config_paths,
|
||||
get_user_config_path,
|
||||
get_project_config_path,
|
||||
get_env_prefix,
|
||||
ensure_config_dirs,
|
||||
)
|
||||
from .schema import ConfigSchema, DEFAULT_CONFIG
|
||||
|
||||
|
||||
# Thread-safe config cache
|
||||
_config_lock = threading.Lock()
|
||||
_config_cache: Optional[ConfigSchema] = None
|
||||
|
||||
|
||||
def _load_toml(path: Path) -> Dict[str, Any]:
|
||||
"""Load a TOML file."""
|
||||
try:
|
||||
# Try tomllib (Python 3.11+) first
|
||||
import tomllib
|
||||
with open(path, "rb") as f:
|
||||
return tomllib.load(f)
|
||||
except ImportError:
|
||||
pass
|
||||
|
||||
try:
|
||||
# Fall back to tomli
|
||||
import tomli
|
||||
with open(path, "rb") as f:
|
||||
return tomli.load(f)
|
||||
except ImportError:
|
||||
pass
|
||||
|
||||
# Last resort: basic TOML parsing
|
||||
return _basic_toml_parse(path)
|
||||
|
||||
|
||||
def _basic_toml_parse(path: Path) -> Dict[str, Any]:
|
||||
"""Basic TOML parser for simple configs."""
|
||||
result: Dict[str, Any] = {}
|
||||
current_section: Optional[str] = None
|
||||
|
||||
with open(path, "r", encoding="utf-8") as f:
|
||||
for line in f:
|
||||
line = line.strip()
|
||||
if not line or line.startswith("#"):
|
||||
continue
|
||||
|
||||
if line.startswith("[") and line.endswith("]"):
|
||||
current_section = line[1:-1]
|
||||
if current_section not in result:
|
||||
result[current_section] = {}
|
||||
elif "=" in line:
|
||||
key, value = line.split("=", 1)
|
||||
key = key.strip()
|
||||
value = value.strip()
|
||||
|
||||
# Parse value
|
||||
if value.lower() == "true":
|
||||
parsed_value = True
|
||||
elif value.lower() == "false":
|
||||
parsed_value = False
|
||||
elif value.startswith('"') and value.endswith('"'):
|
||||
parsed_value = value[1:-1]
|
||||
elif value.startswith("'") and value.endswith("'"):
|
||||
parsed_value = value[1:-1]
|
||||
else:
|
||||
try:
|
||||
parsed_value = int(value)
|
||||
except ValueError:
|
||||
try:
|
||||
parsed_value = float(value)
|
||||
except ValueError:
|
||||
parsed_value = value
|
||||
|
||||
if current_section:
|
||||
result[current_section][key] = parsed_value
|
||||
else:
|
||||
result[key] = parsed_value
|
||||
|
||||
return result
|
||||
|
||||
|
||||
def _save_toml(path: Path, data: Dict[str, Any]) -> None:
|
||||
"""Save data to a TOML file."""
|
||||
try:
|
||||
import tomli_w
|
||||
with open(path, "wb") as f:
|
||||
tomli_w.dump(data, f)
|
||||
return
|
||||
except ImportError:
|
||||
pass
|
||||
|
||||
# Fall back to basic TOML writing
|
||||
_basic_toml_write(path, data)
|
||||
|
||||
|
||||
def _basic_toml_write(path: Path, data: Dict[str, Any]) -> None:
|
||||
"""Basic TOML writer."""
|
||||
lines = []
|
||||
|
||||
# Write top-level keys first
|
||||
for key, value in data.items():
|
||||
if not isinstance(value, dict):
|
||||
lines.append(f"{key} = {_format_toml_value(value)}")
|
||||
|
||||
# Write sections
|
||||
for key, value in data.items():
|
||||
if isinstance(value, dict):
|
||||
lines.append(f"\n[{key}]")
|
||||
for sub_key, sub_value in value.items():
|
||||
if not isinstance(sub_value, dict):
|
||||
lines.append(f"{sub_key} = {_format_toml_value(sub_value)}")
|
||||
else:
|
||||
# Nested section
|
||||
lines.append(f"\n[{key}.{sub_key}]")
|
||||
for nested_key, nested_value in sub_value.items():
|
||||
lines.append(f"{nested_key} = {_format_toml_value(nested_value)}")
|
||||
|
||||
path.parent.mkdir(parents=True, exist_ok=True)
|
||||
with open(path, "w", encoding="utf-8") as f:
|
||||
f.write("\n".join(lines) + "\n")
|
||||
|
||||
|
||||
def _format_toml_value(value: Any) -> str:
|
||||
"""Format a value for TOML."""
|
||||
if isinstance(value, bool):
|
||||
return "true" if value else "false"
|
||||
elif isinstance(value, str):
|
||||
return f'"{value}"'
|
||||
elif isinstance(value, (int, float)):
|
||||
return str(value)
|
||||
elif isinstance(value, list):
|
||||
items = ", ".join(_format_toml_value(v) for v in value)
|
||||
return f"[{items}]"
|
||||
else:
|
||||
return f'"{value}"'
|
||||
|
||||
|
||||
def _merge_dicts(base: Dict[str, Any], override: Dict[str, Any]) -> Dict[str, Any]:
|
||||
"""Deep merge two dictionaries."""
|
||||
result = base.copy()
|
||||
for key, value in override.items():
|
||||
if key in result and isinstance(result[key], dict) and isinstance(value, dict):
|
||||
result[key] = _merge_dicts(result[key], value)
|
||||
else:
|
||||
result[key] = value
|
||||
return result
|
||||
|
||||
|
||||
def _get_env_overrides() -> Dict[str, Any]:
|
||||
"""Get configuration overrides from environment variables."""
|
||||
prefix = get_env_prefix()
|
||||
overrides: Dict[str, Any] = {}
|
||||
|
||||
for key, value in os.environ.items():
|
||||
if not key.startswith(prefix):
|
||||
continue
|
||||
|
||||
# Convert PRAISONAI_OUTPUT_FORMAT to output.format
|
||||
config_key = key[len(prefix):].lower()
|
||||
parts = config_key.split("_")
|
||||
|
||||
# Parse value
|
||||
if value.lower() == "true":
|
||||
parsed_value = True
|
||||
elif value.lower() == "false":
|
||||
parsed_value = False
|
||||
else:
|
||||
try:
|
||||
parsed_value = int(value)
|
||||
except ValueError:
|
||||
try:
|
||||
parsed_value = float(value)
|
||||
except ValueError:
|
||||
parsed_value = value
|
||||
|
||||
# Build nested dict
|
||||
if len(parts) >= 2:
|
||||
section = parts[0]
|
||||
sub_key = "_".join(parts[1:])
|
||||
if section not in overrides:
|
||||
overrides[section] = {}
|
||||
overrides[section][sub_key] = parsed_value
|
||||
else:
|
||||
overrides[config_key] = parsed_value
|
||||
|
||||
return overrides
|
||||
|
||||
|
||||
def _get_dotted_value(data: Dict[str, Any], key: str) -> Any:
|
||||
"""Get a value using dotted key notation."""
|
||||
parts = key.split(".")
|
||||
current = data
|
||||
for part in parts:
|
||||
if isinstance(current, dict) and part in current:
|
||||
current = current[part]
|
||||
else:
|
||||
return None
|
||||
return current
|
||||
|
||||
|
||||
def _set_dotted_value(data: Dict[str, Any], key: str, value: Any) -> None:
|
||||
"""Set a value using dotted key notation."""
|
||||
parts = key.split(".")
|
||||
current = data
|
||||
for part in parts[:-1]:
|
||||
if part not in current:
|
||||
current[part] = {}
|
||||
current = current[part]
|
||||
current[parts[-1]] = value
|
||||
|
||||
|
||||
class ConfigLoader:
|
||||
"""
|
||||
Configuration loader with precedence handling.
|
||||
|
||||
Precedence (highest to lowest):
|
||||
1. CLI flags (not handled here)
|
||||
2. Environment variables (PRAISONAI_*)
|
||||
3. Project config (.praison/config.toml)
|
||||
4. User config (~/.praison/config.toml)
|
||||
5. Defaults
|
||||
"""
|
||||
|
||||
def __init__(self, project_root: Optional[Path] = None):
|
||||
self.project_root = project_root or Path.cwd()
|
||||
self._config: Optional[ConfigSchema] = None
|
||||
self._raw_config: Dict[str, Any] = {}
|
||||
|
||||
def load(self, force_reload: bool = False) -> ConfigSchema:
|
||||
"""Load configuration with precedence."""
|
||||
if self._config is not None and not force_reload:
|
||||
return self._config
|
||||
|
||||
# Start with defaults
|
||||
merged = DEFAULT_CONFIG.to_dict()
|
||||
|
||||
# Load config files in reverse precedence order
|
||||
config_paths = get_config_paths(self.project_root)
|
||||
for path in reversed(config_paths):
|
||||
try:
|
||||
file_config = _load_toml(path)
|
||||
merged = _merge_dicts(merged, file_config)
|
||||
except Exception:
|
||||
pass # Skip invalid files
|
||||
|
||||
# Apply environment overrides
|
||||
env_overrides = _get_env_overrides()
|
||||
merged = _merge_dicts(merged, env_overrides)
|
||||
|
||||
self._raw_config = merged
|
||||
self._config = ConfigSchema.from_dict(merged)
|
||||
return self._config
|
||||
|
||||
def get(self, key: str, default: Any = None) -> Any:
|
||||
"""Get a configuration value using dotted key notation."""
|
||||
self.load()
|
||||
value = _get_dotted_value(self._raw_config, key)
|
||||
return value if value is not None else default
|
||||
|
||||
def set(self, key: str, value: Any, scope: str = "user") -> None:
|
||||
"""
|
||||
Set a configuration value.
|
||||
|
||||
Args:
|
||||
key: Dotted key notation (e.g., "output.format")
|
||||
value: Value to set
|
||||
scope: "user" or "project"
|
||||
"""
|
||||
if scope == "project":
|
||||
config_path = get_project_config_path(self.project_root)
|
||||
else:
|
||||
config_path = get_user_config_path()
|
||||
|
||||
# Load existing config or create empty
|
||||
if config_path.exists():
|
||||
try:
|
||||
existing = _load_toml(config_path)
|
||||
except Exception:
|
||||
existing = {}
|
||||
else:
|
||||
existing = {}
|
||||
|
||||
# Set the value
|
||||
_set_dotted_value(existing, key, value)
|
||||
|
||||
# Ensure directory exists
|
||||
ensure_config_dirs()
|
||||
if scope == "project":
|
||||
config_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Save
|
||||
_save_toml(config_path, existing)
|
||||
|
||||
# Reload
|
||||
self.load(force_reload=True)
|
||||
|
||||
def reset(self, scope: str = "user") -> None:
|
||||
"""Reset configuration to defaults."""
|
||||
if scope == "project":
|
||||
config_path = get_project_config_path(self.project_root)
|
||||
else:
|
||||
config_path = get_user_config_path()
|
||||
|
||||
if config_path.exists():
|
||||
config_path.unlink()
|
||||
|
||||
self.load(force_reload=True)
|
||||
|
||||
def list_all(self) -> Dict[str, Any]:
|
||||
"""List all configuration values."""
|
||||
self.load()
|
||||
return self._raw_config.copy()
|
||||
|
||||
|
||||
# Global config loader
|
||||
_global_loader: Optional[ConfigLoader] = None
|
||||
|
||||
|
||||
def get_config(project_root: Optional[Path] = None) -> ConfigSchema:
|
||||
"""Get the global configuration."""
|
||||
global _global_loader, _config_cache
|
||||
|
||||
with _config_lock:
|
||||
if _global_loader is None or project_root is not None:
|
||||
_global_loader = ConfigLoader(project_root)
|
||||
|
||||
if _config_cache is None:
|
||||
_config_cache = _global_loader.load()
|
||||
|
||||
return _config_cache
|
||||
|
||||
|
||||
def get_config_loader(project_root: Optional[Path] = None) -> ConfigLoader:
|
||||
"""Get the global config loader."""
|
||||
global _global_loader
|
||||
|
||||
with _config_lock:
|
||||
if _global_loader is None or project_root is not None:
|
||||
_global_loader = ConfigLoader(project_root)
|
||||
return _global_loader
|
||||
@@ -0,0 +1,114 @@
|
||||
"""
|
||||
Configuration paths for PraisonAI CLI.
|
||||
|
||||
Defines standard locations for configuration files.
|
||||
"""
|
||||
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import List, Optional
|
||||
|
||||
|
||||
def get_user_config_dir() -> Path:
|
||||
"""Get user configuration directory (~/.praison/)."""
|
||||
return Path.home() / ".praison"
|
||||
|
||||
|
||||
def get_user_config_path() -> Path:
|
||||
"""Get user configuration file path (~/.praison/config.toml)."""
|
||||
return get_user_config_dir() / "config.toml"
|
||||
|
||||
|
||||
def get_project_config_dir(project_root: Optional[Path] = None) -> Path:
|
||||
"""Get project configuration directory (.praison/)."""
|
||||
root = project_root or Path.cwd()
|
||||
return root / ".praison"
|
||||
|
||||
|
||||
def get_project_config_path(project_root: Optional[Path] = None) -> Path:
|
||||
"""Get project configuration file path (.praison/config.toml)."""
|
||||
return get_project_config_dir(project_root) / "config.toml"
|
||||
|
||||
|
||||
def get_sessions_dir() -> Path:
|
||||
"""Get sessions directory (~/.praison/sessions/)."""
|
||||
return get_user_config_dir() / "sessions"
|
||||
|
||||
|
||||
def get_traces_dir() -> Path:
|
||||
"""Get traces directory (~/.praison/traces/)."""
|
||||
return get_user_config_dir() / "traces"
|
||||
|
||||
|
||||
def get_logs_dir() -> Path:
|
||||
"""Get logs directory (~/.praison/logs/)."""
|
||||
return get_user_config_dir() / "logs"
|
||||
|
||||
|
||||
def get_cache_dir() -> Path:
|
||||
"""Get cache directory (~/.praison/cache/)."""
|
||||
return get_user_config_dir() / "cache"
|
||||
|
||||
|
||||
def get_config_paths(project_root: Optional[Path] = None) -> List[Path]:
|
||||
"""
|
||||
Get all configuration file paths in precedence order (highest first).
|
||||
|
||||
Precedence:
|
||||
1. Project config: .praison/config.toml
|
||||
2. User config: ~/.praison/config.toml
|
||||
|
||||
Returns:
|
||||
List of paths in precedence order
|
||||
"""
|
||||
paths = []
|
||||
|
||||
# Project config (highest precedence)
|
||||
project_config = get_project_config_path(project_root)
|
||||
if project_config.exists():
|
||||
paths.append(project_config)
|
||||
|
||||
# User config
|
||||
user_config = get_user_config_path()
|
||||
if user_config.exists():
|
||||
paths.append(user_config)
|
||||
|
||||
return paths
|
||||
|
||||
|
||||
def ensure_config_dirs() -> None:
|
||||
"""Ensure all configuration directories exist."""
|
||||
dirs = [
|
||||
get_user_config_dir(),
|
||||
get_sessions_dir(),
|
||||
get_traces_dir(),
|
||||
get_logs_dir(),
|
||||
get_cache_dir(),
|
||||
]
|
||||
for d in dirs:
|
||||
d.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
|
||||
def get_env_prefix() -> str:
|
||||
"""Get environment variable prefix."""
|
||||
return "PRAISONAI_"
|
||||
|
||||
|
||||
def env_to_config_key(env_var: str) -> Optional[str]:
|
||||
"""
|
||||
Convert environment variable name to config key.
|
||||
|
||||
Example: PRAISONAI_OUTPUT_FORMAT -> output.format
|
||||
"""
|
||||
prefix = get_env_prefix()
|
||||
if not env_var.startswith(prefix):
|
||||
return None
|
||||
|
||||
key = env_var[len(prefix):].lower()
|
||||
# Convert underscores to dots for nested keys
|
||||
# Single underscore = dot, double underscore = single underscore
|
||||
parts = key.split("__")
|
||||
result_parts = []
|
||||
for part in parts:
|
||||
result_parts.append(part.replace("_", "."))
|
||||
return "_".join(result_parts)
|
||||
@@ -0,0 +1,164 @@
|
||||
"""
|
||||
Configuration schema for PraisonAI CLI.
|
||||
|
||||
Defines the structure and defaults for configuration.
|
||||
"""
|
||||
|
||||
from dataclasses import dataclass, field
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
|
||||
@dataclass
|
||||
class OutputConfig:
|
||||
"""Output configuration."""
|
||||
format: str = "text" # text, json, stream-json
|
||||
color: bool = True
|
||||
verbose: bool = False
|
||||
quiet: bool = False
|
||||
screen_reader: bool = False
|
||||
|
||||
|
||||
@dataclass
|
||||
class TracesConfig:
|
||||
"""Traces configuration."""
|
||||
enabled: bool = False
|
||||
endpoint: Optional[str] = None
|
||||
sample_rate: float = 1.0
|
||||
|
||||
|
||||
@dataclass
|
||||
class MCPServerConfig:
|
||||
"""MCP server configuration."""
|
||||
command: str = ""
|
||||
args: List[str] = field(default_factory=list)
|
||||
env: Dict[str, str] = field(default_factory=dict)
|
||||
enabled: bool = True
|
||||
|
||||
|
||||
@dataclass
|
||||
class MCPConfig:
|
||||
"""MCP configuration."""
|
||||
servers: Dict[str, MCPServerConfig] = field(default_factory=dict)
|
||||
|
||||
|
||||
@dataclass
|
||||
class ModelConfig:
|
||||
"""Model configuration."""
|
||||
default: str = "gpt-4o-mini"
|
||||
temperature: float = 0.7
|
||||
max_tokens: int = 16000
|
||||
|
||||
|
||||
@dataclass
|
||||
class SessionConfig:
|
||||
"""Session configuration."""
|
||||
auto_save: bool = False
|
||||
history_limit: int = 10
|
||||
|
||||
|
||||
@dataclass
|
||||
class ConfigSchema:
|
||||
"""
|
||||
Complete configuration schema.
|
||||
|
||||
Represents all configurable options for PraisonAI CLI.
|
||||
"""
|
||||
output: OutputConfig = field(default_factory=OutputConfig)
|
||||
traces: TracesConfig = field(default_factory=TracesConfig)
|
||||
mcp: MCPConfig = field(default_factory=MCPConfig)
|
||||
model: ModelConfig = field(default_factory=ModelConfig)
|
||||
session: SessionConfig = field(default_factory=SessionConfig)
|
||||
|
||||
# Additional settings stored as dict for flexibility
|
||||
extra: Dict[str, Any] = field(default_factory=dict)
|
||||
|
||||
def to_dict(self) -> Dict[str, Any]:
|
||||
"""Convert to dictionary."""
|
||||
return {
|
||||
"output": {
|
||||
"format": self.output.format,
|
||||
"color": self.output.color,
|
||||
"verbose": self.output.verbose,
|
||||
"quiet": self.output.quiet,
|
||||
"screen_reader": self.output.screen_reader,
|
||||
},
|
||||
"traces": {
|
||||
"enabled": self.traces.enabled,
|
||||
"endpoint": self.traces.endpoint,
|
||||
"sample_rate": self.traces.sample_rate,
|
||||
},
|
||||
"mcp": {
|
||||
"servers": {
|
||||
name: {
|
||||
"command": server.command,
|
||||
"args": server.args,
|
||||
"env": server.env,
|
||||
"enabled": server.enabled,
|
||||
}
|
||||
for name, server in self.mcp.servers.items()
|
||||
}
|
||||
},
|
||||
"model": {
|
||||
"default": self.model.default,
|
||||
"temperature": self.model.temperature,
|
||||
"max_tokens": self.model.max_tokens,
|
||||
},
|
||||
"session": {
|
||||
"auto_save": self.session.auto_save,
|
||||
"history_limit": self.session.history_limit,
|
||||
},
|
||||
**self.extra,
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def from_dict(cls, data: Dict[str, Any]) -> "ConfigSchema":
|
||||
"""Create from dictionary."""
|
||||
output_data = data.get("output", {})
|
||||
traces_data = data.get("traces", {})
|
||||
mcp_data = data.get("mcp", {})
|
||||
model_data = data.get("model", {})
|
||||
session_data = data.get("session", {})
|
||||
|
||||
# Extract known keys
|
||||
known_keys = {"output", "traces", "mcp", "model", "session"}
|
||||
extra = {k: v for k, v in data.items() if k not in known_keys}
|
||||
|
||||
# Parse MCP servers
|
||||
mcp_servers = {}
|
||||
for name, server_data in mcp_data.get("servers", {}).items():
|
||||
mcp_servers[name] = MCPServerConfig(
|
||||
command=server_data.get("command", ""),
|
||||
args=server_data.get("args", []),
|
||||
env=server_data.get("env", {}),
|
||||
enabled=server_data.get("enabled", True),
|
||||
)
|
||||
|
||||
return cls(
|
||||
output=OutputConfig(
|
||||
format=output_data.get("format", "text"),
|
||||
color=output_data.get("color", True),
|
||||
verbose=output_data.get("verbose", False),
|
||||
quiet=output_data.get("quiet", False),
|
||||
screen_reader=output_data.get("screen_reader", False),
|
||||
),
|
||||
traces=TracesConfig(
|
||||
enabled=traces_data.get("enabled", False),
|
||||
endpoint=traces_data.get("endpoint"),
|
||||
sample_rate=traces_data.get("sample_rate", 1.0),
|
||||
),
|
||||
mcp=MCPConfig(servers=mcp_servers),
|
||||
model=ModelConfig(
|
||||
default=model_data.get("default", "gpt-4o-mini"),
|
||||
temperature=model_data.get("temperature", 0.7),
|
||||
max_tokens=model_data.get("max_tokens", 16000),
|
||||
),
|
||||
session=SessionConfig(
|
||||
auto_save=session_data.get("auto_save", False),
|
||||
history_limit=session_data.get("history_limit", 10),
|
||||
),
|
||||
extra=extra,
|
||||
)
|
||||
|
||||
|
||||
# Default configuration
|
||||
DEFAULT_CONFIG = ConfigSchema()
|
||||
@@ -96,6 +96,12 @@ __all__ = [
|
||||
'PerformanceHandler',
|
||||
# Lite agent (BYO-LLM)
|
||||
'LiteHandler',
|
||||
# Agent-centric tools (LSP/ACP powered)
|
||||
'create_agent_centric_tools',
|
||||
'InteractiveRuntime',
|
||||
'RuntimeConfig',
|
||||
'CodeIntelligenceRouter',
|
||||
'ActionOrchestrator',
|
||||
]
|
||||
|
||||
def __getattr__(name):
|
||||
@@ -225,4 +231,20 @@ def __getattr__(name):
|
||||
elif name == 'LiteHandler':
|
||||
from .lite import LiteHandler
|
||||
return LiteHandler
|
||||
# Agent-centric tools (LSP/ACP powered)
|
||||
elif name == 'create_agent_centric_tools':
|
||||
from .agent_tools import create_agent_centric_tools
|
||||
return create_agent_centric_tools
|
||||
elif name == 'InteractiveRuntime':
|
||||
from .interactive_runtime import InteractiveRuntime
|
||||
return InteractiveRuntime
|
||||
elif name == 'RuntimeConfig':
|
||||
from .interactive_runtime import RuntimeConfig
|
||||
return RuntimeConfig
|
||||
elif name == 'CodeIntelligenceRouter':
|
||||
from .code_intelligence import CodeIntelligenceRouter
|
||||
return CodeIntelligenceRouter
|
||||
elif name == 'ActionOrchestrator':
|
||||
from .action_orchestrator import ActionOrchestrator
|
||||
return ActionOrchestrator
|
||||
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|
||||
@@ -0,0 +1,474 @@
|
||||
"""
|
||||
Agent-Centric Tools for PraisonAI Interactive Mode.
|
||||
|
||||
These tools route file operations and code intelligence through LSP/ACP,
|
||||
making the Agent the central orchestrator for all actions.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import json
|
||||
import logging
|
||||
from typing import Callable, Dict, List, TYPE_CHECKING
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from .interactive_runtime import InteractiveRuntime
|
||||
from .code_intelligence import CodeIntelligenceRouter
|
||||
from .action_orchestrator import ActionOrchestrator
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
def create_agent_centric_tools(
|
||||
runtime: "InteractiveRuntime",
|
||||
router: "CodeIntelligenceRouter" = None,
|
||||
orchestrator: "ActionOrchestrator" = None
|
||||
) -> List[Callable]:
|
||||
"""
|
||||
Create tools that route through LSP/ACP for agent-centric architecture.
|
||||
|
||||
Args:
|
||||
runtime: The InteractiveRuntime instance
|
||||
router: Optional CodeIntelligenceRouter (created if not provided)
|
||||
orchestrator: Optional ActionOrchestrator (created if not provided)
|
||||
|
||||
Returns:
|
||||
List of tool functions for the Agent
|
||||
"""
|
||||
from .code_intelligence import CodeIntelligenceRouter
|
||||
from .action_orchestrator import ActionOrchestrator
|
||||
|
||||
if router is None:
|
||||
router = CodeIntelligenceRouter(runtime)
|
||||
if orchestrator is None:
|
||||
orchestrator = ActionOrchestrator(runtime)
|
||||
|
||||
# Helper to run async functions synchronously
|
||||
def run_async(coro):
|
||||
"""Run async coroutine synchronously."""
|
||||
try:
|
||||
loop = asyncio.get_event_loop()
|
||||
if loop.is_running():
|
||||
# Create a new loop in a thread
|
||||
import concurrent.futures
|
||||
with concurrent.futures.ThreadPoolExecutor() as executor:
|
||||
future = executor.submit(asyncio.run, coro)
|
||||
return future.result(timeout=60)
|
||||
else:
|
||||
return loop.run_until_complete(coro)
|
||||
except RuntimeError:
|
||||
return asyncio.run(coro)
|
||||
|
||||
# =========================================================================
|
||||
# ACP-Powered File Tools
|
||||
# =========================================================================
|
||||
|
||||
def acp_create_file(filepath: str, content: str) -> str:
|
||||
"""
|
||||
Create a file through ACP with plan/approve/apply/verify flow.
|
||||
|
||||
This tool routes file creation through the ActionOrchestrator,
|
||||
ensuring proper tracking, approval, and verification.
|
||||
|
||||
Args:
|
||||
filepath: Path to the file to create (relative to workspace)
|
||||
content: Content to write to the file
|
||||
|
||||
Returns:
|
||||
JSON string with result including plan status and verification
|
||||
"""
|
||||
async def _create():
|
||||
# Build a detailed prompt for the orchestrator
|
||||
prompt = f"create file {filepath}"
|
||||
|
||||
# Create plan
|
||||
result = await orchestrator.create_plan(prompt)
|
||||
if not result.success:
|
||||
return json.dumps({
|
||||
"success": False,
|
||||
"error": result.error,
|
||||
"read_only": result.read_only_blocked
|
||||
})
|
||||
|
||||
plan = result.plan
|
||||
|
||||
# Update the plan step with actual content
|
||||
if plan.steps:
|
||||
plan.steps[0].params["content"] = content
|
||||
|
||||
# Approve based on runtime config
|
||||
auto_approve = runtime.config.approval_mode == "auto"
|
||||
approved = await orchestrator.approve_plan(plan, auto=auto_approve)
|
||||
|
||||
if not approved and runtime.config.approval_mode == "manual":
|
||||
return json.dumps({
|
||||
"success": False,
|
||||
"error": "Manual approval required",
|
||||
"plan": plan.to_dict(),
|
||||
"requires_approval": True
|
||||
})
|
||||
|
||||
# Apply the plan
|
||||
result = await orchestrator.apply_plan(plan, force=auto_approve)
|
||||
if not result.success:
|
||||
return json.dumps({
|
||||
"success": False,
|
||||
"error": result.error,
|
||||
"plan": plan.to_dict()
|
||||
})
|
||||
|
||||
# Verify
|
||||
result = await orchestrator.verify_plan(plan)
|
||||
|
||||
return json.dumps({
|
||||
"success": result.success,
|
||||
"file_created": filepath,
|
||||
"plan_id": plan.id,
|
||||
"status": plan.status.value,
|
||||
"verified": result.success
|
||||
})
|
||||
|
||||
return run_async(_create())
|
||||
|
||||
def acp_edit_file(filepath: str, new_content: str) -> str:
|
||||
"""
|
||||
Edit a file through ACP with plan/approve/apply/verify flow.
|
||||
|
||||
Args:
|
||||
filepath: Path to the file to edit (relative to workspace)
|
||||
new_content: New content for the file
|
||||
|
||||
Returns:
|
||||
JSON string with result including plan status
|
||||
"""
|
||||
async def _edit():
|
||||
prompt = f"edit file {filepath}"
|
||||
|
||||
result = await orchestrator.create_plan(prompt)
|
||||
if not result.success:
|
||||
return json.dumps({
|
||||
"success": False,
|
||||
"error": result.error,
|
||||
"read_only": result.read_only_blocked
|
||||
})
|
||||
|
||||
plan = result.plan
|
||||
|
||||
# Update step with new content
|
||||
if plan.steps:
|
||||
plan.steps[0].params["new_content"] = new_content
|
||||
|
||||
auto_approve = runtime.config.approval_mode == "auto"
|
||||
approved = await orchestrator.approve_plan(plan, auto=auto_approve)
|
||||
|
||||
if not approved and runtime.config.approval_mode == "manual":
|
||||
return json.dumps({
|
||||
"success": False,
|
||||
"error": "Manual approval required",
|
||||
"requires_approval": True
|
||||
})
|
||||
|
||||
result = await orchestrator.apply_plan(plan, force=auto_approve)
|
||||
|
||||
return json.dumps({
|
||||
"success": result.success,
|
||||
"file_edited": filepath,
|
||||
"plan_id": plan.id,
|
||||
"error": result.error
|
||||
})
|
||||
|
||||
return run_async(_edit())
|
||||
|
||||
def acp_delete_file(filepath: str) -> str:
|
||||
"""
|
||||
Delete a file through ACP with plan/approve/apply/verify flow.
|
||||
|
||||
Args:
|
||||
filepath: Path to the file to delete
|
||||
|
||||
Returns:
|
||||
JSON string with result
|
||||
"""
|
||||
async def _delete():
|
||||
prompt = f"delete file {filepath}"
|
||||
|
||||
result = await orchestrator.create_plan(prompt)
|
||||
if not result.success:
|
||||
return json.dumps({
|
||||
"success": False,
|
||||
"error": result.error
|
||||
})
|
||||
|
||||
plan = result.plan
|
||||
|
||||
# Delete requires explicit approval even in auto mode
|
||||
auto_approve = runtime.config.approval_mode == "auto"
|
||||
approved = await orchestrator.approve_plan(plan, auto=auto_approve)
|
||||
|
||||
if not approved:
|
||||
return json.dumps({
|
||||
"success": False,
|
||||
"error": "Delete requires approval",
|
||||
"requires_approval": True
|
||||
})
|
||||
|
||||
result = await orchestrator.apply_plan(plan)
|
||||
result = await orchestrator.verify_plan(plan)
|
||||
|
||||
return json.dumps({
|
||||
"success": result.success,
|
||||
"file_deleted": filepath,
|
||||
"verified": result.success
|
||||
})
|
||||
|
||||
return run_async(_delete())
|
||||
|
||||
def acp_execute_command(command: str, cwd: str = None) -> str:
|
||||
"""
|
||||
Execute a shell command through ACP with tracking.
|
||||
|
||||
Args:
|
||||
command: The command to execute
|
||||
cwd: Working directory (optional)
|
||||
|
||||
Returns:
|
||||
JSON string with command output
|
||||
"""
|
||||
async def _execute():
|
||||
prompt = f"run command: {command}"
|
||||
|
||||
result = await orchestrator.create_plan(prompt)
|
||||
if not result.success:
|
||||
return json.dumps({
|
||||
"success": False,
|
||||
"error": result.error
|
||||
})
|
||||
|
||||
plan = result.plan
|
||||
|
||||
# Commands require approval
|
||||
auto_approve = runtime.config.approval_mode == "auto"
|
||||
approved = await orchestrator.approve_plan(plan, auto=auto_approve)
|
||||
|
||||
if not approved:
|
||||
return json.dumps({
|
||||
"success": False,
|
||||
"error": "Command requires approval",
|
||||
"requires_approval": True
|
||||
})
|
||||
|
||||
result = await orchestrator.apply_plan(plan)
|
||||
|
||||
# Extract command result
|
||||
if plan.steps and plan.steps[0].result:
|
||||
cmd_result = plan.steps[0].result
|
||||
return json.dumps({
|
||||
"success": result.success,
|
||||
"command": command,
|
||||
"stdout": cmd_result.get("stdout", ""),
|
||||
"stderr": cmd_result.get("stderr", ""),
|
||||
"returncode": cmd_result.get("returncode", -1)
|
||||
})
|
||||
|
||||
return json.dumps({
|
||||
"success": result.success,
|
||||
"error": result.error
|
||||
})
|
||||
|
||||
return run_async(_execute())
|
||||
|
||||
# =========================================================================
|
||||
# LSP-Powered Code Intelligence Tools
|
||||
# =========================================================================
|
||||
|
||||
def lsp_list_symbols(file_path: str) -> str:
|
||||
"""
|
||||
List all symbols (functions, classes, methods) in a file using LSP.
|
||||
|
||||
Falls back to regex-based extraction if LSP is unavailable.
|
||||
|
||||
Args:
|
||||
file_path: Path to the file to analyze
|
||||
|
||||
Returns:
|
||||
JSON string with list of symbols and their locations
|
||||
"""
|
||||
async def _list():
|
||||
result = await router.handle_query(
|
||||
f"list all functions and classes in {file_path}",
|
||||
file_path=file_path
|
||||
)
|
||||
return json.dumps(result.to_dict())
|
||||
|
||||
return run_async(_list())
|
||||
|
||||
def lsp_find_definition(symbol: str, file_path: str = None) -> str:
|
||||
"""
|
||||
Find where a symbol is defined using LSP.
|
||||
|
||||
Args:
|
||||
symbol: The symbol name to find
|
||||
file_path: Optional file path for context
|
||||
|
||||
Returns:
|
||||
JSON string with definition location(s)
|
||||
"""
|
||||
async def _find():
|
||||
query = f"where is {symbol} defined"
|
||||
if file_path:
|
||||
query += f" in {file_path}"
|
||||
|
||||
result = await router.handle_query(query, file_path=file_path)
|
||||
return json.dumps(result.to_dict())
|
||||
|
||||
return run_async(_find())
|
||||
|
||||
def lsp_find_references(symbol: str, file_path: str = None) -> str:
|
||||
"""
|
||||
Find all references to a symbol using LSP.
|
||||
|
||||
Args:
|
||||
symbol: The symbol name to find references for
|
||||
file_path: Optional file path for context
|
||||
|
||||
Returns:
|
||||
JSON string with reference locations
|
||||
"""
|
||||
async def _find():
|
||||
query = f"find all references to {symbol}"
|
||||
if file_path:
|
||||
query += f" in {file_path}"
|
||||
|
||||
result = await router.handle_query(query, file_path=file_path)
|
||||
return json.dumps(result.to_dict())
|
||||
|
||||
return run_async(_find())
|
||||
|
||||
def lsp_get_diagnostics(file_path: str = None) -> str:
|
||||
"""
|
||||
Get diagnostics (errors, warnings) for a file using LSP.
|
||||
|
||||
Args:
|
||||
file_path: Path to the file (optional, gets all if not specified)
|
||||
|
||||
Returns:
|
||||
JSON string with diagnostic information
|
||||
"""
|
||||
async def _get():
|
||||
query = "show all errors and warnings"
|
||||
if file_path:
|
||||
query += f" in {file_path}"
|
||||
|
||||
result = await router.handle_query(query, file_path=file_path)
|
||||
return json.dumps(result.to_dict())
|
||||
|
||||
return run_async(_get())
|
||||
|
||||
# =========================================================================
|
||||
# Basic File Tools (for read-only operations)
|
||||
# =========================================================================
|
||||
|
||||
def read_file(filepath: str) -> str:
|
||||
"""
|
||||
Read content from a file.
|
||||
|
||||
This is a read-only operation that doesn't require ACP.
|
||||
|
||||
Args:
|
||||
filepath: Path to the file to read
|
||||
|
||||
Returns:
|
||||
File content or error message
|
||||
"""
|
||||
from pathlib import Path
|
||||
|
||||
try:
|
||||
path = Path(filepath)
|
||||
if not path.is_absolute():
|
||||
path = Path(runtime.config.workspace) / filepath
|
||||
|
||||
if not path.exists():
|
||||
return json.dumps({"error": f"File not found: {filepath}"})
|
||||
|
||||
content = path.read_text()
|
||||
|
||||
# Track in trace if enabled
|
||||
if runtime._trace:
|
||||
runtime._trace.add_entry(
|
||||
category="file",
|
||||
action="read",
|
||||
params={"file": str(path)},
|
||||
result={"size": len(content)}
|
||||
)
|
||||
|
||||
return content
|
||||
|
||||
except Exception as e:
|
||||
return json.dumps({"error": str(e)})
|
||||
|
||||
def list_files(directory: str = ".", pattern: str = "*") -> str:
|
||||
"""
|
||||
List files in a directory.
|
||||
|
||||
Args:
|
||||
directory: Directory to list (relative to workspace)
|
||||
pattern: Glob pattern to filter files
|
||||
|
||||
Returns:
|
||||
JSON string with list of files
|
||||
"""
|
||||
from pathlib import Path
|
||||
|
||||
try:
|
||||
path = Path(directory)
|
||||
if not path.is_absolute():
|
||||
path = Path(runtime.config.workspace) / directory
|
||||
|
||||
if not path.exists():
|
||||
return json.dumps({"error": f"Directory not found: {directory}"})
|
||||
|
||||
files = []
|
||||
for f in path.glob(pattern):
|
||||
files.append({
|
||||
"name": f.name,
|
||||
"path": str(f.relative_to(runtime.config.workspace)),
|
||||
"is_dir": f.is_dir(),
|
||||
"size": f.stat().st_size if f.is_file() else None
|
||||
})
|
||||
|
||||
return json.dumps({"files": files, "count": len(files)})
|
||||
|
||||
except Exception as e:
|
||||
return json.dumps({"error": str(e)})
|
||||
|
||||
# Return all tools
|
||||
return [
|
||||
# ACP-powered file tools
|
||||
acp_create_file,
|
||||
acp_edit_file,
|
||||
acp_delete_file,
|
||||
acp_execute_command,
|
||||
# LSP-powered code intelligence
|
||||
lsp_list_symbols,
|
||||
lsp_find_definition,
|
||||
lsp_find_references,
|
||||
lsp_get_diagnostics,
|
||||
# Basic read-only tools
|
||||
read_file,
|
||||
list_files,
|
||||
]
|
||||
|
||||
|
||||
def get_tool_descriptions() -> Dict[str, str]:
|
||||
"""Get descriptions of all agent-centric tools."""
|
||||
return {
|
||||
"acp_create_file": "Create a file through ACP with plan/approve/apply/verify",
|
||||
"acp_edit_file": "Edit a file through ACP with tracking",
|
||||
"acp_delete_file": "Delete a file through ACP (requires approval)",
|
||||
"acp_execute_command": "Execute a shell command through ACP",
|
||||
"lsp_list_symbols": "List symbols in a file using LSP",
|
||||
"lsp_find_definition": "Find where a symbol is defined",
|
||||
"lsp_find_references": "Find all references to a symbol",
|
||||
"lsp_get_diagnostics": "Get errors and warnings for a file",
|
||||
"read_file": "Read content from a file (read-only)",
|
||||
"list_files": "List files in a directory",
|
||||
}
|
||||
@@ -0,0 +1,73 @@
|
||||
"""
|
||||
Queue System for PraisonAI TUI.
|
||||
|
||||
Provides multi-run, multi-agent queueing with priorities, cancel/retry,
|
||||
streaming output, and persistence.
|
||||
|
||||
Lazy-loaded to avoid import overhead when not used.
|
||||
"""
|
||||
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from .models import QueuedRun, RunState, RunPriority, QueueConfig
|
||||
from .scheduler import QueueScheduler
|
||||
from .worker import WorkerPool
|
||||
from .persistence import QueuePersistence
|
||||
|
||||
_lazy_cache = {}
|
||||
|
||||
|
||||
def __getattr__(name: str):
|
||||
"""Lazy load queue components."""
|
||||
global _lazy_cache
|
||||
|
||||
if name in _lazy_cache:
|
||||
return _lazy_cache[name]
|
||||
|
||||
if name == "QueuedRun":
|
||||
from .models import QueuedRun
|
||||
_lazy_cache[name] = QueuedRun
|
||||
return QueuedRun
|
||||
elif name == "RunState":
|
||||
from .models import RunState
|
||||
_lazy_cache[name] = RunState
|
||||
return RunState
|
||||
elif name == "RunPriority":
|
||||
from .models import RunPriority
|
||||
_lazy_cache[name] = RunPriority
|
||||
return RunPriority
|
||||
elif name == "QueueConfig":
|
||||
from .models import QueueConfig
|
||||
_lazy_cache[name] = QueueConfig
|
||||
return QueueConfig
|
||||
elif name == "QueueScheduler":
|
||||
from .scheduler import QueueScheduler
|
||||
_lazy_cache[name] = QueueScheduler
|
||||
return QueueScheduler
|
||||
elif name == "WorkerPool":
|
||||
from .worker import WorkerPool
|
||||
_lazy_cache[name] = WorkerPool
|
||||
return WorkerPool
|
||||
elif name == "QueuePersistence":
|
||||
from .persistence import QueuePersistence
|
||||
_lazy_cache[name] = QueuePersistence
|
||||
return QueuePersistence
|
||||
elif name == "QueueManager":
|
||||
from .manager import QueueManager
|
||||
_lazy_cache[name] = QueueManager
|
||||
return QueueManager
|
||||
|
||||
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|
||||
|
||||
|
||||
__all__ = [
|
||||
"QueuedRun",
|
||||
"RunState",
|
||||
"RunPriority",
|
||||
"QueueConfig",
|
||||
"QueueScheduler",
|
||||
"WorkerPool",
|
||||
"QueuePersistence",
|
||||
"QueueManager",
|
||||
]
|
||||
@@ -0,0 +1,395 @@
|
||||
"""
|
||||
Queue Manager for PraisonAI.
|
||||
|
||||
High-level interface that coordinates scheduler, workers, and persistence.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import logging
|
||||
import os
|
||||
import time
|
||||
from typing import Any, Callable, Coroutine, Dict, List, Optional
|
||||
|
||||
from .models import QueuedRun, RunState, RunPriority, QueueConfig, QueueStats, QueueEvent
|
||||
from .scheduler import QueueScheduler, QueueFullError
|
||||
from .worker import WorkerPool
|
||||
from .persistence import QueuePersistence
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class QueueManager:
|
||||
"""
|
||||
High-level queue manager that coordinates all queue components.
|
||||
|
||||
Provides a unified interface for:
|
||||
- Submitting and managing runs
|
||||
- Starting/stopping workers
|
||||
- Persistence and crash recovery
|
||||
- Event handling
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
config: Optional[QueueConfig] = None,
|
||||
on_output: Optional[Callable[[str, str], Coroutine[Any, Any, None]]] = None,
|
||||
on_complete: Optional[Callable[[str, QueuedRun], Coroutine[Any, Any, None]]] = None,
|
||||
on_error: Optional[Callable[[str, Exception], Coroutine[Any, Any, None]]] = None,
|
||||
on_event: Optional[Callable[[QueueEvent], Coroutine[Any, Any, None]]] = None,
|
||||
):
|
||||
"""
|
||||
Initialize queue manager.
|
||||
|
||||
Args:
|
||||
config: Queue configuration.
|
||||
on_output: Callback for streaming output.
|
||||
on_complete: Callback when run completes.
|
||||
on_error: Callback when run fails.
|
||||
on_event: Callback for queue events.
|
||||
"""
|
||||
self.config = config or QueueConfig()
|
||||
|
||||
# Initialize components
|
||||
self.scheduler = QueueScheduler(self.config)
|
||||
self.persistence = QueuePersistence(self.config.db_path) if self.config.enable_persistence else None
|
||||
|
||||
# Store callbacks
|
||||
self._on_output = on_output
|
||||
self._on_complete = on_complete
|
||||
self._on_error = on_error
|
||||
self._on_event = on_event
|
||||
|
||||
# Worker pool (created on start)
|
||||
self.workers: Optional[WorkerPool] = None
|
||||
|
||||
# Autosave task
|
||||
self._autosave_task: Optional[asyncio.Task] = None
|
||||
|
||||
# Running state
|
||||
self._running = False
|
||||
|
||||
# Session tracking
|
||||
self._current_session_id: Optional[str] = None
|
||||
|
||||
async def start(self, recover: bool = True) -> None:
|
||||
"""
|
||||
Start the queue manager.
|
||||
|
||||
Args:
|
||||
recover: If True, recover pending runs from persistence.
|
||||
"""
|
||||
if self._running:
|
||||
return
|
||||
|
||||
logger.info("Starting queue manager")
|
||||
|
||||
# Initialize persistence
|
||||
if self.persistence:
|
||||
self.persistence.initialize()
|
||||
|
||||
if recover:
|
||||
await self._recover_runs()
|
||||
|
||||
# Create and start worker pool
|
||||
self.workers = WorkerPool(
|
||||
scheduler=self.scheduler,
|
||||
on_output=self._on_output,
|
||||
on_complete=self._wrap_complete_callback(),
|
||||
on_error=self._wrap_error_callback(),
|
||||
on_event=self._on_event,
|
||||
max_workers=self.config.max_concurrent_global,
|
||||
poll_interval=self.config.worker_poll_interval,
|
||||
stream_buffer_size=self.config.stream_buffer_size,
|
||||
)
|
||||
|
||||
await self.workers.start()
|
||||
|
||||
# Start autosave
|
||||
if self.persistence and self.config.autosave_interval_seconds > 0:
|
||||
self._autosave_task = asyncio.create_task(
|
||||
self._autosave_loop(),
|
||||
name="queue-autosave"
|
||||
)
|
||||
|
||||
self._running = True
|
||||
logger.info("Queue manager started")
|
||||
|
||||
async def stop(self, timeout: float = 10.0) -> None:
|
||||
"""
|
||||
Stop the queue manager gracefully.
|
||||
|
||||
Args:
|
||||
timeout: Seconds to wait for workers to finish.
|
||||
"""
|
||||
if not self._running:
|
||||
return
|
||||
|
||||
logger.info("Stopping queue manager")
|
||||
|
||||
# Stop autosave
|
||||
if self._autosave_task:
|
||||
self._autosave_task.cancel()
|
||||
try:
|
||||
await self._autosave_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
self._autosave_task = None
|
||||
|
||||
# Stop workers
|
||||
if self.workers:
|
||||
await self.workers.stop(timeout=timeout)
|
||||
self.workers = None
|
||||
|
||||
# Final save
|
||||
if self.persistence:
|
||||
await self._save_all_runs()
|
||||
self.persistence.close()
|
||||
|
||||
self._running = False
|
||||
logger.info("Queue manager stopped")
|
||||
|
||||
async def submit(
|
||||
self,
|
||||
input_content: str,
|
||||
agent_name: str = "Assistant",
|
||||
priority: RunPriority = RunPriority.NORMAL,
|
||||
session_id: Optional[str] = None,
|
||||
workspace: Optional[str] = None,
|
||||
config: Optional[Dict[str, Any]] = None,
|
||||
run_id: Optional[str] = None,
|
||||
) -> str:
|
||||
"""
|
||||
Submit a new run to the queue.
|
||||
|
||||
Args:
|
||||
input_content: The input/prompt for the agent.
|
||||
agent_name: Name of the agent to use.
|
||||
priority: Run priority.
|
||||
session_id: Session ID for grouping.
|
||||
workspace: Workspace path.
|
||||
config: Additional configuration.
|
||||
run_id: Optional specific run ID.
|
||||
|
||||
Returns:
|
||||
The run ID.
|
||||
"""
|
||||
run = QueuedRun(
|
||||
agent_name=agent_name,
|
||||
input_content=input_content,
|
||||
priority=priority,
|
||||
session_id=session_id or self._current_session_id,
|
||||
workspace=workspace or os.getcwd(),
|
||||
config=config or {},
|
||||
max_retries=self.config.default_max_retries,
|
||||
)
|
||||
|
||||
if run_id:
|
||||
run.run_id = run_id
|
||||
|
||||
# Submit to scheduler
|
||||
result_id = await self.scheduler.submit(run)
|
||||
|
||||
# Persist
|
||||
if self.persistence:
|
||||
self.persistence.save_run(run)
|
||||
|
||||
return result_id
|
||||
|
||||
async def cancel(self, run_id: str) -> bool:
|
||||
"""Cancel a run."""
|
||||
result = await self.scheduler.cancel(run_id)
|
||||
|
||||
if result and self.persistence:
|
||||
run = self.scheduler.get_run(run_id)
|
||||
if run:
|
||||
self.persistence.save_run(run)
|
||||
|
||||
return result
|
||||
|
||||
async def retry(self, run_id: str) -> Optional[str]:
|
||||
"""Retry a failed run."""
|
||||
new_id = await self.scheduler.retry(run_id)
|
||||
|
||||
if new_id and self.persistence:
|
||||
run = self.scheduler.get_run(new_id)
|
||||
if run:
|
||||
self.persistence.save_run(run)
|
||||
|
||||
return new_id
|
||||
|
||||
async def pause(self, run_id: str) -> bool:
|
||||
"""Pause a running run."""
|
||||
return await self.scheduler.pause(run_id)
|
||||
|
||||
async def resume(self, run_id: str) -> bool:
|
||||
"""Resume a paused run."""
|
||||
return await self.scheduler.resume(run_id)
|
||||
|
||||
def get_run(self, run_id: str) -> Optional[QueuedRun]:
|
||||
"""Get a run by ID."""
|
||||
return self.scheduler.get_run(run_id)
|
||||
|
||||
def get_queued(self) -> List[QueuedRun]:
|
||||
"""Get all queued runs."""
|
||||
return self.scheduler.get_queued()
|
||||
|
||||
def get_running(self) -> List[QueuedRun]:
|
||||
"""Get all running runs."""
|
||||
return self.scheduler.get_running()
|
||||
|
||||
def list_runs(
|
||||
self,
|
||||
state: Optional[RunState] = None,
|
||||
session_id: Optional[str] = None,
|
||||
limit: int = 100,
|
||||
) -> List[QueuedRun]:
|
||||
"""
|
||||
List runs with optional filters.
|
||||
|
||||
For active runs, uses scheduler. For historical, uses persistence.
|
||||
"""
|
||||
if state and state.is_terminal() and self.persistence:
|
||||
return self.persistence.list_runs(
|
||||
state=state,
|
||||
session_id=session_id,
|
||||
limit=limit,
|
||||
)
|
||||
|
||||
# Get from scheduler
|
||||
runs = self.scheduler.get_all()
|
||||
|
||||
if state:
|
||||
runs = [r for r in runs if r.state == state]
|
||||
if session_id:
|
||||
runs = [r for r in runs if r.session_id == session_id]
|
||||
|
||||
return runs[:limit]
|
||||
|
||||
async def clear_queue(self) -> int:
|
||||
"""Clear all queued runs."""
|
||||
count = await self.scheduler.clear_queue()
|
||||
return count
|
||||
|
||||
def get_stats(self) -> QueueStats:
|
||||
"""Get queue statistics."""
|
||||
if self.persistence:
|
||||
return self.persistence.get_stats(self._current_session_id)
|
||||
|
||||
# Calculate from scheduler
|
||||
queued = self.scheduler.get_queued()
|
||||
running = self.scheduler.get_running()
|
||||
|
||||
return QueueStats(
|
||||
queued_count=len(queued),
|
||||
running_count=len(running),
|
||||
)
|
||||
|
||||
# Session management
|
||||
|
||||
def set_session(self, session_id: str) -> None:
|
||||
"""Set the current session ID."""
|
||||
self._current_session_id = session_id
|
||||
|
||||
def save_session_state(self, state: Dict[str, Any]) -> None:
|
||||
"""Save session state."""
|
||||
if self.persistence and self._current_session_id:
|
||||
self.persistence.save_session(
|
||||
self._current_session_id,
|
||||
state=state,
|
||||
)
|
||||
|
||||
def load_session_state(self, session_id: str) -> Optional[Dict[str, Any]]:
|
||||
"""Load session state."""
|
||||
if self.persistence:
|
||||
return self.persistence.load_session(session_id)
|
||||
return None
|
||||
|
||||
def list_sessions(self, limit: int = 50) -> List[Dict[str, Any]]:
|
||||
"""List recent sessions."""
|
||||
if self.persistence:
|
||||
return self.persistence.list_sessions(limit)
|
||||
return []
|
||||
|
||||
# Internal methods
|
||||
|
||||
async def _recover_runs(self) -> None:
|
||||
"""Recover pending runs from persistence."""
|
||||
if not self.persistence:
|
||||
return
|
||||
|
||||
# Mark interrupted runs as failed
|
||||
interrupted = self.persistence.mark_interrupted_as_failed()
|
||||
if interrupted:
|
||||
logger.warning(f"Marked {interrupted} interrupted runs as failed")
|
||||
|
||||
# Load pending runs
|
||||
pending = self.persistence.load_pending_runs()
|
||||
if pending:
|
||||
logger.info(f"Recovering {len(pending)} pending runs")
|
||||
self.scheduler.load_runs(pending)
|
||||
|
||||
for run in pending:
|
||||
self.persistence.mark_recovered(run.run_id)
|
||||
|
||||
async def _save_all_runs(self) -> None:
|
||||
"""Save all active runs to persistence."""
|
||||
if not self.persistence:
|
||||
return
|
||||
|
||||
for run in self.scheduler.get_all():
|
||||
self.persistence.save_run(run)
|
||||
|
||||
async def _autosave_loop(self) -> None:
|
||||
"""Periodically save runs to persistence."""
|
||||
while True:
|
||||
try:
|
||||
await asyncio.sleep(self.config.autosave_interval_seconds)
|
||||
await self._save_all_runs()
|
||||
logger.debug("Autosaved queue state")
|
||||
except asyncio.CancelledError:
|
||||
break
|
||||
except Exception as e:
|
||||
logger.error(f"Autosave error: {e}")
|
||||
|
||||
def _wrap_complete_callback(self):
|
||||
"""Wrap complete callback to include persistence."""
|
||||
async def wrapper(run_id: str, run: QueuedRun):
|
||||
# Persist
|
||||
if self.persistence:
|
||||
self.persistence.save_run(run)
|
||||
|
||||
# Call user callback
|
||||
if self._on_complete:
|
||||
await self._on_complete(run_id, run)
|
||||
|
||||
return wrapper
|
||||
|
||||
def _wrap_error_callback(self):
|
||||
"""Wrap error callback to include persistence."""
|
||||
async def wrapper(run_id: str, error: Exception):
|
||||
# Persist
|
||||
if self.persistence:
|
||||
run = self.scheduler.get_run(run_id)
|
||||
if run:
|
||||
self.persistence.save_run(run)
|
||||
|
||||
# Call user callback
|
||||
if self._on_error:
|
||||
await self._on_error(run_id, error)
|
||||
|
||||
return wrapper
|
||||
|
||||
@property
|
||||
def is_running(self) -> bool:
|
||||
"""Whether the queue manager is running."""
|
||||
return self._running
|
||||
|
||||
@property
|
||||
def queued_count(self) -> int:
|
||||
"""Number of queued runs."""
|
||||
return self.scheduler.queued_count
|
||||
|
||||
@property
|
||||
def running_count(self) -> int:
|
||||
"""Number of running runs."""
|
||||
return self.scheduler.running_count
|
||||
@@ -0,0 +1,286 @@
|
||||
"""
|
||||
Data models for the PraisonAI Queue System.
|
||||
|
||||
Defines QueuedRun, RunState, RunPriority, and QueueConfig.
|
||||
"""
|
||||
|
||||
from dataclasses import dataclass, field
|
||||
from enum import Enum, IntEnum
|
||||
from typing import Any, Dict, List, Optional
|
||||
import time
|
||||
import uuid
|
||||
|
||||
|
||||
class RunState(str, Enum):
|
||||
"""State of a queued run."""
|
||||
QUEUED = "queued"
|
||||
RUNNING = "running"
|
||||
PAUSED = "paused"
|
||||
SUCCEEDED = "succeeded"
|
||||
FAILED = "failed"
|
||||
CANCELLED = "cancelled"
|
||||
|
||||
def is_terminal(self) -> bool:
|
||||
"""Check if this is a terminal state."""
|
||||
return self in (RunState.SUCCEEDED, RunState.FAILED, RunState.CANCELLED)
|
||||
|
||||
def is_active(self) -> bool:
|
||||
"""Check if this is an active (non-terminal) state."""
|
||||
return not self.is_terminal()
|
||||
|
||||
|
||||
class RunPriority(IntEnum):
|
||||
"""Priority levels for queued runs. Higher value = higher priority."""
|
||||
LOW = 0
|
||||
NORMAL = 1
|
||||
HIGH = 2
|
||||
URGENT = 3
|
||||
|
||||
@classmethod
|
||||
def from_string(cls, s: str) -> "RunPriority":
|
||||
"""Parse priority from string."""
|
||||
mapping = {
|
||||
"low": cls.LOW,
|
||||
"normal": cls.NORMAL,
|
||||
"high": cls.HIGH,
|
||||
"urgent": cls.URGENT,
|
||||
}
|
||||
return mapping.get(s.lower(), cls.NORMAL)
|
||||
|
||||
|
||||
@dataclass
|
||||
class QueuedRun:
|
||||
"""A single queued agent run."""
|
||||
|
||||
# Identity
|
||||
run_id: str = field(default_factory=lambda: str(uuid.uuid4())[:8])
|
||||
agent_name: str = ""
|
||||
input_content: str = ""
|
||||
|
||||
# State
|
||||
state: RunState = RunState.QUEUED
|
||||
priority: RunPriority = RunPriority.NORMAL
|
||||
|
||||
# Attribution / tracing
|
||||
session_id: Optional[str] = None
|
||||
trace_id: Optional[str] = None
|
||||
workspace: Optional[str] = None
|
||||
user_id: Optional[str] = None
|
||||
|
||||
# Timing
|
||||
created_at: float = field(default_factory=time.time)
|
||||
started_at: Optional[float] = None
|
||||
ended_at: Optional[float] = None
|
||||
|
||||
# Results
|
||||
output_content: Optional[str] = None
|
||||
error: Optional[str] = None
|
||||
metrics: Dict[str, Any] = field(default_factory=dict)
|
||||
|
||||
# Retry tracking
|
||||
retry_count: int = 0
|
||||
max_retries: int = 3
|
||||
parent_run_id: Optional[str] = None # Link to original run if this is a retry
|
||||
|
||||
# Configuration
|
||||
config: Dict[str, Any] = field(default_factory=dict)
|
||||
|
||||
# Streaming state
|
||||
output_chunks: List[str] = field(default_factory=list)
|
||||
|
||||
def __post_init__(self):
|
||||
"""Ensure proper types after initialization."""
|
||||
if isinstance(self.state, str):
|
||||
self.state = RunState(self.state)
|
||||
if isinstance(self.priority, int) and not isinstance(self.priority, RunPriority):
|
||||
self.priority = RunPriority(self.priority)
|
||||
|
||||
@property
|
||||
def duration_seconds(self) -> Optional[float]:
|
||||
"""Get run duration in seconds."""
|
||||
if self.started_at is None:
|
||||
return None
|
||||
end = self.ended_at or time.time()
|
||||
return end - self.started_at
|
||||
|
||||
@property
|
||||
def wait_seconds(self) -> float:
|
||||
"""Get time spent waiting in queue."""
|
||||
start = self.started_at or time.time()
|
||||
return start - self.created_at
|
||||
|
||||
def can_retry(self) -> bool:
|
||||
"""Check if this run can be retried."""
|
||||
return (
|
||||
self.state == RunState.FAILED and
|
||||
self.retry_count < self.max_retries
|
||||
)
|
||||
|
||||
def to_dict(self) -> Dict[str, Any]:
|
||||
"""Convert to dictionary for persistence."""
|
||||
return {
|
||||
"run_id": self.run_id,
|
||||
"agent_name": self.agent_name,
|
||||
"input_content": self.input_content,
|
||||
"state": self.state.value,
|
||||
"priority": int(self.priority),
|
||||
"session_id": self.session_id,
|
||||
"trace_id": self.trace_id,
|
||||
"workspace": self.workspace,
|
||||
"user_id": self.user_id,
|
||||
"created_at": self.created_at,
|
||||
"started_at": self.started_at,
|
||||
"ended_at": self.ended_at,
|
||||
"output_content": self.output_content,
|
||||
"error": self.error,
|
||||
"metrics": self.metrics,
|
||||
"retry_count": self.retry_count,
|
||||
"max_retries": self.max_retries,
|
||||
"parent_run_id": self.parent_run_id,
|
||||
"config": self.config,
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def from_dict(cls, data: Dict[str, Any]) -> "QueuedRun":
|
||||
"""Create from dictionary."""
|
||||
# Handle state conversion
|
||||
state = data.get("state", "queued")
|
||||
if isinstance(state, str):
|
||||
state = RunState(state)
|
||||
|
||||
# Handle priority conversion
|
||||
priority = data.get("priority", 1)
|
||||
if isinstance(priority, int):
|
||||
priority = RunPriority(priority)
|
||||
|
||||
return cls(
|
||||
run_id=data.get("run_id", str(uuid.uuid4())[:8]),
|
||||
agent_name=data.get("agent_name", ""),
|
||||
input_content=data.get("input_content", ""),
|
||||
state=state,
|
||||
priority=priority,
|
||||
session_id=data.get("session_id"),
|
||||
trace_id=data.get("trace_id"),
|
||||
workspace=data.get("workspace"),
|
||||
user_id=data.get("user_id"),
|
||||
created_at=data.get("created_at", time.time()),
|
||||
started_at=data.get("started_at"),
|
||||
ended_at=data.get("ended_at"),
|
||||
output_content=data.get("output_content"),
|
||||
error=data.get("error"),
|
||||
metrics=data.get("metrics", {}),
|
||||
retry_count=data.get("retry_count", 0),
|
||||
max_retries=data.get("max_retries", 3),
|
||||
parent_run_id=data.get("parent_run_id"),
|
||||
config=data.get("config", {}),
|
||||
)
|
||||
|
||||
|
||||
@dataclass
|
||||
class QueueConfig:
|
||||
"""Configuration for the queue system."""
|
||||
|
||||
# Concurrency limits
|
||||
max_concurrent_global: int = 4
|
||||
max_concurrent_per_agent: int = 2
|
||||
max_concurrent_per_workspace: int = 4
|
||||
|
||||
# Queue limits
|
||||
max_queue_size: int = 100
|
||||
|
||||
# Defaults
|
||||
default_priority: RunPriority = RunPriority.NORMAL
|
||||
default_max_retries: int = 3
|
||||
|
||||
# Persistence
|
||||
enable_persistence: bool = True
|
||||
db_path: str = ".praison/queue.db"
|
||||
|
||||
# Autosave
|
||||
autosave_interval_seconds: float = 30.0
|
||||
|
||||
# Backpressure
|
||||
stream_buffer_size: int = 1000
|
||||
drop_strategy: str = "oldest" # "oldest" or "newest"
|
||||
|
||||
# Timeouts
|
||||
run_timeout_seconds: Optional[float] = None # None = no timeout
|
||||
worker_poll_interval: float = 0.1
|
||||
|
||||
def to_dict(self) -> Dict[str, Any]:
|
||||
"""Convert to dictionary."""
|
||||
return {
|
||||
"max_concurrent_global": self.max_concurrent_global,
|
||||
"max_concurrent_per_agent": self.max_concurrent_per_agent,
|
||||
"max_concurrent_per_workspace": self.max_concurrent_per_workspace,
|
||||
"max_queue_size": self.max_queue_size,
|
||||
"default_priority": int(self.default_priority),
|
||||
"default_max_retries": self.default_max_retries,
|
||||
"enable_persistence": self.enable_persistence,
|
||||
"db_path": self.db_path,
|
||||
"autosave_interval_seconds": self.autosave_interval_seconds,
|
||||
"stream_buffer_size": self.stream_buffer_size,
|
||||
"drop_strategy": self.drop_strategy,
|
||||
"run_timeout_seconds": self.run_timeout_seconds,
|
||||
"worker_poll_interval": self.worker_poll_interval,
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def from_dict(cls, data: Dict[str, Any]) -> "QueueConfig":
|
||||
"""Create from dictionary."""
|
||||
priority = data.get("default_priority", 1)
|
||||
if isinstance(priority, int):
|
||||
priority = RunPriority(priority)
|
||||
|
||||
return cls(
|
||||
max_concurrent_global=data.get("max_concurrent_global", 4),
|
||||
max_concurrent_per_agent=data.get("max_concurrent_per_agent", 2),
|
||||
max_concurrent_per_workspace=data.get("max_concurrent_per_workspace", 4),
|
||||
max_queue_size=data.get("max_queue_size", 100),
|
||||
default_priority=priority,
|
||||
default_max_retries=data.get("default_max_retries", 3),
|
||||
enable_persistence=data.get("enable_persistence", True),
|
||||
db_path=data.get("db_path", ".praison/queue.db"),
|
||||
autosave_interval_seconds=data.get("autosave_interval_seconds", 30.0),
|
||||
stream_buffer_size=data.get("stream_buffer_size", 1000),
|
||||
drop_strategy=data.get("drop_strategy", "oldest"),
|
||||
run_timeout_seconds=data.get("run_timeout_seconds"),
|
||||
worker_poll_interval=data.get("worker_poll_interval", 0.1),
|
||||
)
|
||||
|
||||
|
||||
@dataclass
|
||||
class QueueStats:
|
||||
"""Statistics about the queue."""
|
||||
queued_count: int = 0
|
||||
running_count: int = 0
|
||||
succeeded_count: int = 0
|
||||
failed_count: int = 0
|
||||
cancelled_count: int = 0
|
||||
total_runs: int = 0
|
||||
avg_wait_seconds: float = 0.0
|
||||
avg_duration_seconds: float = 0.0
|
||||
|
||||
@property
|
||||
def active_count(self) -> int:
|
||||
"""Count of active (non-terminal) runs."""
|
||||
return self.queued_count + self.running_count
|
||||
|
||||
|
||||
@dataclass
|
||||
class StreamChunk:
|
||||
"""A chunk of streaming output."""
|
||||
run_id: str
|
||||
content: str
|
||||
timestamp: float = field(default_factory=time.time)
|
||||
chunk_index: int = 0
|
||||
is_final: bool = False
|
||||
|
||||
|
||||
@dataclass
|
||||
class QueueEvent:
|
||||
"""An event from the queue system."""
|
||||
event_type: str # "run_started", "run_completed", "run_failed", "output_chunk", "state_changed"
|
||||
run_id: str
|
||||
timestamp: float = field(default_factory=time.time)
|
||||
data: Dict[str, Any] = field(default_factory=dict)
|
||||
@@ -0,0 +1,564 @@
|
||||
"""
|
||||
SQLite persistence layer for the PraisonAI Queue System.
|
||||
|
||||
Provides crash recovery, session persistence, and run history.
|
||||
"""
|
||||
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import sqlite3
|
||||
import threading
|
||||
import time
|
||||
from contextlib import contextmanager
|
||||
from pathlib import Path
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
from .models import QueuedRun, RunState, RunPriority, QueueStats
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
SCHEMA_VERSION = "1.0"
|
||||
|
||||
SCHEMA_SQL = """
|
||||
-- Schema version tracking
|
||||
CREATE TABLE IF NOT EXISTS schema_version (
|
||||
version TEXT PRIMARY KEY,
|
||||
applied_at REAL NOT NULL
|
||||
);
|
||||
|
||||
-- Queue runs
|
||||
CREATE TABLE IF NOT EXISTS runs (
|
||||
run_id TEXT PRIMARY KEY,
|
||||
agent_name TEXT NOT NULL,
|
||||
input_content TEXT,
|
||||
output_content TEXT,
|
||||
state TEXT NOT NULL DEFAULT 'queued',
|
||||
priority INTEGER NOT NULL DEFAULT 1,
|
||||
session_id TEXT,
|
||||
trace_id TEXT,
|
||||
workspace TEXT,
|
||||
user_id TEXT,
|
||||
created_at REAL NOT NULL,
|
||||
started_at REAL,
|
||||
ended_at REAL,
|
||||
error TEXT,
|
||||
retry_count INTEGER DEFAULT 0,
|
||||
max_retries INTEGER DEFAULT 3,
|
||||
parent_run_id TEXT,
|
||||
config TEXT,
|
||||
metrics TEXT,
|
||||
recovered INTEGER DEFAULT 0
|
||||
);
|
||||
|
||||
CREATE INDEX IF NOT EXISTS idx_runs_state ON runs(state);
|
||||
CREATE INDEX IF NOT EXISTS idx_runs_session ON runs(session_id);
|
||||
CREATE INDEX IF NOT EXISTS idx_runs_priority_created ON runs(priority DESC, created_at ASC);
|
||||
CREATE INDEX IF NOT EXISTS idx_runs_workspace ON runs(workspace);
|
||||
|
||||
-- Messages (chat history per run)
|
||||
CREATE TABLE IF NOT EXISTS messages (
|
||||
message_id TEXT PRIMARY KEY,
|
||||
run_id TEXT REFERENCES runs(run_id),
|
||||
role TEXT NOT NULL,
|
||||
content TEXT NOT NULL,
|
||||
timestamp REAL NOT NULL,
|
||||
metadata TEXT
|
||||
);
|
||||
|
||||
CREATE INDEX IF NOT EXISTS idx_messages_run ON messages(run_id);
|
||||
|
||||
-- Tool calls
|
||||
CREATE TABLE IF NOT EXISTS tool_calls (
|
||||
call_id TEXT PRIMARY KEY,
|
||||
run_id TEXT REFERENCES runs(run_id),
|
||||
tool_name TEXT NOT NULL,
|
||||
args TEXT,
|
||||
result TEXT,
|
||||
started_at REAL NOT NULL,
|
||||
ended_at REAL,
|
||||
error TEXT
|
||||
);
|
||||
|
||||
CREATE INDEX IF NOT EXISTS idx_tool_calls_run ON tool_calls(run_id);
|
||||
|
||||
-- Sessions
|
||||
CREATE TABLE IF NOT EXISTS sessions (
|
||||
session_id TEXT PRIMARY KEY,
|
||||
user_id TEXT,
|
||||
created_at REAL NOT NULL,
|
||||
updated_at REAL NOT NULL,
|
||||
state TEXT,
|
||||
config TEXT
|
||||
);
|
||||
|
||||
CREATE INDEX IF NOT EXISTS idx_sessions_user ON sessions(user_id);
|
||||
"""
|
||||
|
||||
|
||||
class QueuePersistence:
|
||||
"""SQLite-backed persistence for the queue system."""
|
||||
|
||||
def __init__(self, db_path: str = ".praison/queue.db"):
|
||||
"""
|
||||
Initialize persistence layer.
|
||||
|
||||
Args:
|
||||
db_path: Path to SQLite database file.
|
||||
"""
|
||||
self.db_path = db_path
|
||||
self._conn: Optional[sqlite3.Connection] = None
|
||||
self._lock = threading.Lock()
|
||||
self._initialized = False
|
||||
|
||||
def _ensure_dir(self) -> None:
|
||||
"""Ensure database directory exists."""
|
||||
db_dir = os.path.dirname(self.db_path)
|
||||
if db_dir:
|
||||
os.makedirs(db_dir, exist_ok=True)
|
||||
|
||||
def _get_connection(self) -> sqlite3.Connection:
|
||||
"""Get or create database connection."""
|
||||
if self._conn is None:
|
||||
self._ensure_dir()
|
||||
self._conn = sqlite3.connect(
|
||||
self.db_path,
|
||||
check_same_thread=False,
|
||||
timeout=30.0
|
||||
)
|
||||
self._conn.row_factory = sqlite3.Row
|
||||
# Enable WAL mode for better concurrency
|
||||
self._conn.execute("PRAGMA journal_mode=WAL")
|
||||
self._conn.execute("PRAGMA synchronous=NORMAL")
|
||||
return self._conn
|
||||
|
||||
@contextmanager
|
||||
def _transaction(self):
|
||||
"""Context manager for database transactions."""
|
||||
conn = self._get_connection()
|
||||
with self._lock:
|
||||
try:
|
||||
yield conn
|
||||
conn.commit()
|
||||
except Exception:
|
||||
conn.rollback()
|
||||
raise
|
||||
|
||||
def initialize(self) -> None:
|
||||
"""Initialize database schema."""
|
||||
if self._initialized:
|
||||
return
|
||||
|
||||
with self._transaction() as conn:
|
||||
conn.executescript(SCHEMA_SQL)
|
||||
|
||||
# Check/set schema version
|
||||
cursor = conn.execute(
|
||||
"SELECT version FROM schema_version ORDER BY applied_at DESC LIMIT 1"
|
||||
)
|
||||
row = cursor.fetchone()
|
||||
|
||||
if row is None:
|
||||
conn.execute(
|
||||
"INSERT INTO schema_version (version, applied_at) VALUES (?, ?)",
|
||||
(SCHEMA_VERSION, time.time())
|
||||
)
|
||||
elif row["version"] != SCHEMA_VERSION:
|
||||
# Future: handle migrations
|
||||
logger.warning(
|
||||
f"Schema version mismatch: {row['version']} vs {SCHEMA_VERSION}"
|
||||
)
|
||||
|
||||
self._initialized = True
|
||||
|
||||
def close(self) -> None:
|
||||
"""Close database connection."""
|
||||
if self._conn is not None:
|
||||
self._conn.close()
|
||||
self._conn = None
|
||||
self._initialized = False
|
||||
|
||||
# Run operations
|
||||
|
||||
def save_run(self, run: QueuedRun) -> None:
|
||||
"""Save or update a run."""
|
||||
self.initialize()
|
||||
|
||||
with self._transaction() as conn:
|
||||
conn.execute("""
|
||||
INSERT OR REPLACE INTO runs (
|
||||
run_id, agent_name, input_content, output_content,
|
||||
state, priority, session_id, trace_id, workspace, user_id,
|
||||
created_at, started_at, ended_at, error,
|
||||
retry_count, max_retries, parent_run_id,
|
||||
config, metrics
|
||||
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
""", (
|
||||
run.run_id,
|
||||
run.agent_name,
|
||||
run.input_content,
|
||||
run.output_content,
|
||||
run.state.value,
|
||||
int(run.priority),
|
||||
run.session_id,
|
||||
run.trace_id,
|
||||
run.workspace,
|
||||
run.user_id,
|
||||
run.created_at,
|
||||
run.started_at,
|
||||
run.ended_at,
|
||||
run.error,
|
||||
run.retry_count,
|
||||
run.max_retries,
|
||||
run.parent_run_id,
|
||||
json.dumps(run.config) if run.config else None,
|
||||
json.dumps(run.metrics) if run.metrics else None,
|
||||
))
|
||||
|
||||
def load_run(self, run_id: str) -> Optional[QueuedRun]:
|
||||
"""Load a run by ID."""
|
||||
self.initialize()
|
||||
|
||||
with self._transaction() as conn:
|
||||
cursor = conn.execute(
|
||||
"SELECT * FROM runs WHERE run_id = ?",
|
||||
(run_id,)
|
||||
)
|
||||
row = cursor.fetchone()
|
||||
|
||||
if row is None:
|
||||
return None
|
||||
|
||||
return self._row_to_run(row)
|
||||
|
||||
def list_runs(
|
||||
self,
|
||||
state: Optional[RunState] = None,
|
||||
session_id: Optional[str] = None,
|
||||
workspace: Optional[str] = None,
|
||||
limit: int = 100,
|
||||
offset: int = 0,
|
||||
) -> List[QueuedRun]:
|
||||
"""List runs with optional filters."""
|
||||
self.initialize()
|
||||
|
||||
query = "SELECT * FROM runs WHERE 1=1"
|
||||
params: List[Any] = []
|
||||
|
||||
if state is not None:
|
||||
query += " AND state = ?"
|
||||
params.append(state.value)
|
||||
|
||||
if session_id is not None:
|
||||
query += " AND session_id = ?"
|
||||
params.append(session_id)
|
||||
|
||||
if workspace is not None:
|
||||
query += " AND workspace = ?"
|
||||
params.append(workspace)
|
||||
|
||||
query += " ORDER BY priority DESC, created_at ASC LIMIT ? OFFSET ?"
|
||||
params.extend([limit, offset])
|
||||
|
||||
with self._transaction() as conn:
|
||||
cursor = conn.execute(query, params)
|
||||
return [self._row_to_run(row) for row in cursor.fetchall()]
|
||||
|
||||
def delete_run(self, run_id: str) -> bool:
|
||||
"""Delete a run and its related data."""
|
||||
self.initialize()
|
||||
|
||||
with self._transaction() as conn:
|
||||
# Delete related data first
|
||||
conn.execute("DELETE FROM messages WHERE run_id = ?", (run_id,))
|
||||
conn.execute("DELETE FROM tool_calls WHERE run_id = ?", (run_id,))
|
||||
|
||||
cursor = conn.execute("DELETE FROM runs WHERE run_id = ?", (run_id,))
|
||||
return cursor.rowcount > 0
|
||||
|
||||
def update_run_state(
|
||||
self,
|
||||
run_id: str,
|
||||
state: RunState,
|
||||
error: Optional[str] = None,
|
||||
output: Optional[str] = None,
|
||||
) -> bool:
|
||||
"""Update run state."""
|
||||
self.initialize()
|
||||
|
||||
updates = ["state = ?"]
|
||||
params: List[Any] = [state.value]
|
||||
|
||||
if state == RunState.RUNNING:
|
||||
updates.append("started_at = ?")
|
||||
params.append(time.time())
|
||||
elif state.is_terminal():
|
||||
updates.append("ended_at = ?")
|
||||
params.append(time.time())
|
||||
|
||||
if error is not None:
|
||||
updates.append("error = ?")
|
||||
params.append(error)
|
||||
|
||||
if output is not None:
|
||||
updates.append("output_content = ?")
|
||||
params.append(output)
|
||||
|
||||
params.append(run_id)
|
||||
|
||||
with self._transaction() as conn:
|
||||
cursor = conn.execute(
|
||||
f"UPDATE runs SET {', '.join(updates)} WHERE run_id = ?",
|
||||
params
|
||||
)
|
||||
return cursor.rowcount > 0
|
||||
|
||||
# Crash recovery
|
||||
|
||||
def load_pending_runs(self) -> List[QueuedRun]:
|
||||
"""Load runs that were QUEUED or RUNNING at crash."""
|
||||
self.initialize()
|
||||
|
||||
with self._transaction() as conn:
|
||||
cursor = conn.execute("""
|
||||
SELECT * FROM runs
|
||||
WHERE state IN ('queued', 'running')
|
||||
ORDER BY priority DESC, created_at ASC
|
||||
""")
|
||||
return [self._row_to_run(row) for row in cursor.fetchall()]
|
||||
|
||||
def mark_recovered(self, run_id: str) -> None:
|
||||
"""Mark a run as recovered after restart."""
|
||||
self.initialize()
|
||||
|
||||
with self._transaction() as conn:
|
||||
conn.execute(
|
||||
"UPDATE runs SET recovered = 1 WHERE run_id = ?",
|
||||
(run_id,)
|
||||
)
|
||||
|
||||
def mark_interrupted_as_failed(self) -> int:
|
||||
"""Mark all RUNNING runs as FAILED (for crash recovery)."""
|
||||
self.initialize()
|
||||
|
||||
with self._transaction() as conn:
|
||||
cursor = conn.execute("""
|
||||
UPDATE runs
|
||||
SET state = 'failed',
|
||||
error = 'Interrupted by crash/restart',
|
||||
ended_at = ?
|
||||
WHERE state = 'running'
|
||||
""", (time.time(),))
|
||||
return cursor.rowcount
|
||||
|
||||
# Statistics
|
||||
|
||||
def get_stats(self, session_id: Optional[str] = None) -> QueueStats:
|
||||
"""Get queue statistics."""
|
||||
self.initialize()
|
||||
|
||||
with self._transaction() as conn:
|
||||
where = ""
|
||||
params: List[Any] = []
|
||||
if session_id:
|
||||
where = "WHERE session_id = ?"
|
||||
params = [session_id]
|
||||
|
||||
# Count by state
|
||||
cursor = conn.execute(f"""
|
||||
SELECT state, COUNT(*) as count
|
||||
FROM runs {where}
|
||||
GROUP BY state
|
||||
""", params)
|
||||
|
||||
counts = {row["state"]: row["count"] for row in cursor.fetchall()}
|
||||
|
||||
# Average wait time
|
||||
cursor = conn.execute(f"""
|
||||
SELECT AVG(started_at - created_at) as avg_wait
|
||||
FROM runs
|
||||
{where + ' AND' if where else 'WHERE'} started_at IS NOT NULL
|
||||
""", params)
|
||||
row = cursor.fetchone()
|
||||
avg_wait = row["avg_wait"] or 0.0
|
||||
|
||||
# Average duration
|
||||
cursor = conn.execute(f"""
|
||||
SELECT AVG(ended_at - started_at) as avg_duration
|
||||
FROM runs
|
||||
{where + ' AND' if where else 'WHERE'} ended_at IS NOT NULL AND started_at IS NOT NULL
|
||||
""", params)
|
||||
row = cursor.fetchone()
|
||||
avg_duration = row["avg_duration"] or 0.0
|
||||
|
||||
return QueueStats(
|
||||
queued_count=counts.get("queued", 0),
|
||||
running_count=counts.get("running", 0),
|
||||
succeeded_count=counts.get("succeeded", 0),
|
||||
failed_count=counts.get("failed", 0),
|
||||
cancelled_count=counts.get("cancelled", 0),
|
||||
total_runs=sum(counts.values()),
|
||||
avg_wait_seconds=avg_wait,
|
||||
avg_duration_seconds=avg_duration,
|
||||
)
|
||||
|
||||
# Session operations
|
||||
|
||||
def save_session(
|
||||
self,
|
||||
session_id: str,
|
||||
user_id: Optional[str] = None,
|
||||
state: Optional[Dict[str, Any]] = None,
|
||||
config: Optional[Dict[str, Any]] = None,
|
||||
) -> None:
|
||||
"""Save or update a session."""
|
||||
self.initialize()
|
||||
|
||||
now = time.time()
|
||||
|
||||
with self._transaction() as conn:
|
||||
# Check if exists
|
||||
cursor = conn.execute(
|
||||
"SELECT session_id FROM sessions WHERE session_id = ?",
|
||||
(session_id,)
|
||||
)
|
||||
exists = cursor.fetchone() is not None
|
||||
|
||||
if exists:
|
||||
conn.execute("""
|
||||
UPDATE sessions SET
|
||||
user_id = COALESCE(?, user_id),
|
||||
updated_at = ?,
|
||||
state = COALESCE(?, state),
|
||||
config = COALESCE(?, config)
|
||||
WHERE session_id = ?
|
||||
""", (
|
||||
user_id,
|
||||
now,
|
||||
json.dumps(state) if state else None,
|
||||
json.dumps(config) if config else None,
|
||||
session_id,
|
||||
))
|
||||
else:
|
||||
conn.execute("""
|
||||
INSERT INTO sessions (session_id, user_id, created_at, updated_at, state, config)
|
||||
VALUES (?, ?, ?, ?, ?, ?)
|
||||
""", (
|
||||
session_id,
|
||||
user_id,
|
||||
now,
|
||||
now,
|
||||
json.dumps(state) if state else None,
|
||||
json.dumps(config) if config else None,
|
||||
))
|
||||
|
||||
def load_session(self, session_id: str) -> Optional[Dict[str, Any]]:
|
||||
"""Load a session."""
|
||||
self.initialize()
|
||||
|
||||
with self._transaction() as conn:
|
||||
cursor = conn.execute(
|
||||
"SELECT * FROM sessions WHERE session_id = ?",
|
||||
(session_id,)
|
||||
)
|
||||
row = cursor.fetchone()
|
||||
|
||||
if row is None:
|
||||
return None
|
||||
|
||||
return {
|
||||
"session_id": row["session_id"],
|
||||
"user_id": row["user_id"],
|
||||
"created_at": row["created_at"],
|
||||
"updated_at": row["updated_at"],
|
||||
"state": json.loads(row["state"]) if row["state"] else None,
|
||||
"config": json.loads(row["config"]) if row["config"] else None,
|
||||
}
|
||||
|
||||
def list_sessions(self, limit: int = 50) -> List[Dict[str, Any]]:
|
||||
"""List recent sessions."""
|
||||
self.initialize()
|
||||
|
||||
with self._transaction() as conn:
|
||||
cursor = conn.execute("""
|
||||
SELECT * FROM sessions
|
||||
ORDER BY updated_at DESC
|
||||
LIMIT ?
|
||||
""", (limit,))
|
||||
|
||||
return [
|
||||
{
|
||||
"session_id": row["session_id"],
|
||||
"user_id": row["user_id"],
|
||||
"created_at": row["created_at"],
|
||||
"updated_at": row["updated_at"],
|
||||
"state": json.loads(row["state"]) if row["state"] else None,
|
||||
"config": json.loads(row["config"]) if row["config"] else None,
|
||||
}
|
||||
for row in cursor.fetchall()
|
||||
]
|
||||
|
||||
# Helper methods
|
||||
|
||||
def _row_to_run(self, row: sqlite3.Row) -> QueuedRun:
|
||||
"""Convert database row to QueuedRun."""
|
||||
return QueuedRun(
|
||||
run_id=row["run_id"],
|
||||
agent_name=row["agent_name"],
|
||||
input_content=row["input_content"] or "",
|
||||
output_content=row["output_content"],
|
||||
state=RunState(row["state"]),
|
||||
priority=RunPriority(row["priority"]),
|
||||
session_id=row["session_id"],
|
||||
trace_id=row["trace_id"],
|
||||
workspace=row["workspace"],
|
||||
user_id=row["user_id"],
|
||||
created_at=row["created_at"],
|
||||
started_at=row["started_at"],
|
||||
ended_at=row["ended_at"],
|
||||
error=row["error"],
|
||||
retry_count=row["retry_count"] or 0,
|
||||
max_retries=row["max_retries"] or 3,
|
||||
parent_run_id=row["parent_run_id"],
|
||||
config=json.loads(row["config"]) if row["config"] else {},
|
||||
metrics=json.loads(row["metrics"]) if row["metrics"] else {},
|
||||
)
|
||||
|
||||
# Cleanup
|
||||
|
||||
def cleanup_old_runs(self, days: int = 30) -> int:
|
||||
"""Delete runs older than specified days."""
|
||||
self.initialize()
|
||||
|
||||
cutoff = time.time() - (days * 24 * 60 * 60)
|
||||
|
||||
with self._transaction() as conn:
|
||||
# Get run IDs to delete
|
||||
cursor = conn.execute("""
|
||||
SELECT run_id FROM runs
|
||||
WHERE created_at < ? AND state IN ('succeeded', 'failed', 'cancelled')
|
||||
""", (cutoff,))
|
||||
run_ids = [row["run_id"] for row in cursor.fetchall()]
|
||||
|
||||
if not run_ids:
|
||||
return 0
|
||||
|
||||
placeholders = ",".join("?" * len(run_ids))
|
||||
|
||||
# Delete related data
|
||||
conn.execute(
|
||||
f"DELETE FROM messages WHERE run_id IN ({placeholders})",
|
||||
run_ids
|
||||
)
|
||||
conn.execute(
|
||||
f"DELETE FROM tool_calls WHERE run_id IN ({placeholders})",
|
||||
run_ids
|
||||
)
|
||||
|
||||
# Delete runs
|
||||
cursor = conn.execute(
|
||||
f"DELETE FROM runs WHERE run_id IN ({placeholders})",
|
||||
run_ids
|
||||
)
|
||||
|
||||
return cursor.rowcount
|
||||
@@ -0,0 +1,484 @@
|
||||
"""
|
||||
Queue Scheduler for PraisonAI.
|
||||
|
||||
Priority-based FIFO scheduler with concurrency limits.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import logging
|
||||
import time
|
||||
import uuid
|
||||
from collections import defaultdict
|
||||
from typing import Any, Callable, Dict, List, Optional, Set
|
||||
|
||||
from .models import QueuedRun, RunState, RunPriority, QueueConfig, QueueEvent
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class QueueFullError(Exception):
|
||||
"""Raised when queue is at capacity."""
|
||||
pass
|
||||
|
||||
|
||||
class RunNotFoundError(Exception):
|
||||
"""Raised when a run is not found."""
|
||||
pass
|
||||
|
||||
|
||||
class QueueScheduler:
|
||||
"""
|
||||
Priority-based FIFO scheduler with concurrency limits.
|
||||
|
||||
Runs are organized by priority (URGENT > HIGH > NORMAL > LOW),
|
||||
with FIFO ordering within each priority level.
|
||||
"""
|
||||
|
||||
def __init__(self, config: Optional[QueueConfig] = None):
|
||||
"""
|
||||
Initialize scheduler.
|
||||
|
||||
Args:
|
||||
config: Queue configuration. Uses defaults if not provided.
|
||||
"""
|
||||
self.config = config or QueueConfig()
|
||||
|
||||
# Priority queues: priority -> list of runs (FIFO order)
|
||||
self._queues: Dict[RunPriority, List[QueuedRun]] = {
|
||||
p: [] for p in RunPriority
|
||||
}
|
||||
|
||||
# Running runs: run_id -> run
|
||||
self._running: Dict[str, QueuedRun] = {}
|
||||
|
||||
# All runs by ID for quick lookup
|
||||
self._all_runs: Dict[str, QueuedRun] = {}
|
||||
|
||||
# Lock for thread safety
|
||||
self._lock = asyncio.Lock()
|
||||
|
||||
# Event callbacks
|
||||
self._event_callbacks: List[Callable[[QueueEvent], None]] = []
|
||||
|
||||
# Cancellation tokens
|
||||
self._cancel_tokens: Set[str] = set()
|
||||
|
||||
def add_event_callback(self, callback: Callable[[QueueEvent], None]) -> None:
|
||||
"""Add a callback for queue events."""
|
||||
self._event_callbacks.append(callback)
|
||||
|
||||
def remove_event_callback(self, callback: Callable[[QueueEvent], None]) -> None:
|
||||
"""Remove an event callback."""
|
||||
if callback in self._event_callbacks:
|
||||
self._event_callbacks.remove(callback)
|
||||
|
||||
def _emit_event(self, event: QueueEvent) -> None:
|
||||
"""Emit an event to all callbacks."""
|
||||
for callback in self._event_callbacks:
|
||||
try:
|
||||
callback(event)
|
||||
except Exception as e:
|
||||
logger.error(f"Error in event callback: {e}")
|
||||
|
||||
async def submit(
|
||||
self,
|
||||
run: QueuedRun,
|
||||
check_duplicate: bool = True,
|
||||
) -> str:
|
||||
"""
|
||||
Submit a run to the queue.
|
||||
|
||||
Args:
|
||||
run: The run to submit.
|
||||
check_duplicate: If True, reject duplicate run_ids.
|
||||
|
||||
Returns:
|
||||
The run_id.
|
||||
|
||||
Raises:
|
||||
QueueFullError: If queue is at capacity.
|
||||
ValueError: If run_id already exists (when check_duplicate=True).
|
||||
"""
|
||||
async with self._lock:
|
||||
# Check for duplicates
|
||||
if check_duplicate and run.run_id in self._all_runs:
|
||||
raise ValueError(f"Run {run.run_id} already exists")
|
||||
|
||||
# Check queue capacity
|
||||
total_queued = sum(len(q) for q in self._queues.values())
|
||||
if total_queued >= self.config.max_queue_size:
|
||||
raise QueueFullError(
|
||||
f"Queue is full ({total_queued}/{self.config.max_queue_size})"
|
||||
)
|
||||
|
||||
# Ensure state is QUEUED
|
||||
run.state = RunState.QUEUED
|
||||
|
||||
# Add to appropriate priority queue
|
||||
self._queues[run.priority].append(run)
|
||||
self._all_runs[run.run_id] = run
|
||||
|
||||
logger.debug(f"Submitted run {run.run_id} with priority {run.priority.name}")
|
||||
|
||||
self._emit_event(QueueEvent(
|
||||
event_type="run_submitted",
|
||||
run_id=run.run_id,
|
||||
data={"priority": run.priority.name, "agent": run.agent_name}
|
||||
))
|
||||
|
||||
return run.run_id
|
||||
|
||||
async def next(self) -> Optional[QueuedRun]:
|
||||
"""
|
||||
Get the next run to execute, respecting priority and concurrency limits.
|
||||
|
||||
Returns:
|
||||
The next run, or None if no run is available.
|
||||
"""
|
||||
async with self._lock:
|
||||
if not self._can_start_new():
|
||||
return None
|
||||
|
||||
# Check priorities in order (URGENT first = highest value)
|
||||
for priority in reversed(list(RunPriority)):
|
||||
queue = self._queues[priority]
|
||||
|
||||
for i, run in enumerate(queue):
|
||||
if self._can_run(run):
|
||||
# Remove from queue
|
||||
queue.pop(i)
|
||||
|
||||
# Update state
|
||||
run.state = RunState.RUNNING
|
||||
run.started_at = time.time()
|
||||
|
||||
# Add to running
|
||||
self._running[run.run_id] = run
|
||||
|
||||
logger.debug(f"Starting run {run.run_id}")
|
||||
|
||||
self._emit_event(QueueEvent(
|
||||
event_type="run_started",
|
||||
run_id=run.run_id,
|
||||
data={"agent": run.agent_name}
|
||||
))
|
||||
|
||||
return run
|
||||
|
||||
return None
|
||||
|
||||
async def complete(
|
||||
self,
|
||||
run_id: str,
|
||||
output: Optional[str] = None,
|
||||
metrics: Optional[Dict[str, Any]] = None,
|
||||
) -> Optional[QueuedRun]:
|
||||
"""
|
||||
Mark a run as completed successfully.
|
||||
|
||||
Args:
|
||||
run_id: The run ID.
|
||||
output: The output content.
|
||||
metrics: Optional metrics.
|
||||
|
||||
Returns:
|
||||
The completed run, or None if not found.
|
||||
"""
|
||||
async with self._lock:
|
||||
run = self._running.pop(run_id, None)
|
||||
if run is None:
|
||||
logger.warning(f"Run {run_id} not found in running")
|
||||
return None
|
||||
|
||||
run.state = RunState.SUCCEEDED
|
||||
run.ended_at = time.time()
|
||||
run.output_content = output
|
||||
if metrics:
|
||||
run.metrics.update(metrics)
|
||||
|
||||
logger.debug(f"Completed run {run_id}")
|
||||
|
||||
self._emit_event(QueueEvent(
|
||||
event_type="run_completed",
|
||||
run_id=run_id,
|
||||
data={"duration": run.duration_seconds}
|
||||
))
|
||||
|
||||
return run
|
||||
|
||||
async def fail(
|
||||
self,
|
||||
run_id: str,
|
||||
error: str,
|
||||
metrics: Optional[Dict[str, Any]] = None,
|
||||
) -> Optional[QueuedRun]:
|
||||
"""
|
||||
Mark a run as failed.
|
||||
|
||||
Args:
|
||||
run_id: The run ID.
|
||||
error: The error message.
|
||||
metrics: Optional metrics.
|
||||
|
||||
Returns:
|
||||
The failed run, or None if not found.
|
||||
"""
|
||||
async with self._lock:
|
||||
run = self._running.pop(run_id, None)
|
||||
if run is None:
|
||||
logger.warning(f"Run {run_id} not found in running")
|
||||
return None
|
||||
|
||||
run.state = RunState.FAILED
|
||||
run.ended_at = time.time()
|
||||
run.error = error
|
||||
if metrics:
|
||||
run.metrics.update(metrics)
|
||||
|
||||
logger.debug(f"Failed run {run_id}: {error}")
|
||||
|
||||
self._emit_event(QueueEvent(
|
||||
event_type="run_failed",
|
||||
run_id=run_id,
|
||||
data={"error": error}
|
||||
))
|
||||
|
||||
return run
|
||||
|
||||
async def cancel(self, run_id: str) -> bool:
|
||||
"""
|
||||
Cancel a queued or running run.
|
||||
|
||||
Args:
|
||||
run_id: The run ID to cancel.
|
||||
|
||||
Returns:
|
||||
True if cancelled, False if not found.
|
||||
"""
|
||||
async with self._lock:
|
||||
# Check if running
|
||||
if run_id in self._running:
|
||||
run = self._running.pop(run_id)
|
||||
run.state = RunState.CANCELLED
|
||||
run.ended_at = time.time()
|
||||
self._cancel_tokens.add(run_id)
|
||||
|
||||
logger.debug(f"Cancelled running run {run_id}")
|
||||
|
||||
self._emit_event(QueueEvent(
|
||||
event_type="run_cancelled",
|
||||
run_id=run_id,
|
||||
data={"was_running": True}
|
||||
))
|
||||
|
||||
return True
|
||||
|
||||
# Check if queued
|
||||
for priority in RunPriority:
|
||||
queue = self._queues[priority]
|
||||
for i, run in enumerate(queue):
|
||||
if run.run_id == run_id:
|
||||
queue.pop(i)
|
||||
run.state = RunState.CANCELLED
|
||||
run.ended_at = time.time()
|
||||
|
||||
logger.debug(f"Cancelled queued run {run_id}")
|
||||
|
||||
self._emit_event(QueueEvent(
|
||||
event_type="run_cancelled",
|
||||
run_id=run_id,
|
||||
data={"was_running": False}
|
||||
))
|
||||
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
def is_cancelled(self, run_id: str) -> bool:
|
||||
"""Check if a run has been cancelled."""
|
||||
return run_id in self._cancel_tokens
|
||||
|
||||
def clear_cancel_token(self, run_id: str) -> None:
|
||||
"""Clear a cancellation token."""
|
||||
self._cancel_tokens.discard(run_id)
|
||||
|
||||
async def retry(self, run_id: str) -> Optional[str]:
|
||||
"""
|
||||
Retry a failed run.
|
||||
|
||||
Creates a new run with incremented retry count and link to parent.
|
||||
|
||||
Args:
|
||||
run_id: The run ID to retry.
|
||||
|
||||
Returns:
|
||||
The new run_id, or None if retry not allowed.
|
||||
"""
|
||||
async with self._lock:
|
||||
# Find the original run
|
||||
original = self._all_runs.get(run_id)
|
||||
if original is None:
|
||||
logger.warning(f"Run {run_id} not found for retry")
|
||||
return None
|
||||
|
||||
if not original.can_retry():
|
||||
logger.warning(
|
||||
f"Run {run_id} cannot be retried "
|
||||
f"(state={original.state}, retries={original.retry_count}/{original.max_retries})"
|
||||
)
|
||||
return None
|
||||
|
||||
# Create new run
|
||||
new_run = QueuedRun(
|
||||
run_id=str(uuid.uuid4())[:8],
|
||||
agent_name=original.agent_name,
|
||||
input_content=original.input_content,
|
||||
state=RunState.QUEUED,
|
||||
priority=original.priority,
|
||||
session_id=original.session_id,
|
||||
trace_id=original.trace_id,
|
||||
workspace=original.workspace,
|
||||
user_id=original.user_id,
|
||||
retry_count=original.retry_count + 1,
|
||||
max_retries=original.max_retries,
|
||||
parent_run_id=original.run_id,
|
||||
config=original.config.copy(),
|
||||
)
|
||||
|
||||
# Submit outside lock to avoid deadlock
|
||||
await self.submit(new_run, check_duplicate=True)
|
||||
|
||||
logger.debug(f"Retrying run {run_id} as {new_run.run_id}")
|
||||
|
||||
self._emit_event(QueueEvent(
|
||||
event_type="run_retried",
|
||||
run_id=new_run.run_id,
|
||||
data={"parent_run_id": run_id, "retry_count": new_run.retry_count}
|
||||
))
|
||||
|
||||
return new_run.run_id
|
||||
|
||||
async def pause(self, run_id: str) -> bool:
|
||||
"""
|
||||
Pause a running run.
|
||||
|
||||
Note: The worker must check for pause state and handle accordingly.
|
||||
"""
|
||||
async with self._lock:
|
||||
if run_id not in self._running:
|
||||
return False
|
||||
|
||||
run = self._running[run_id]
|
||||
run.state = RunState.PAUSED
|
||||
|
||||
self._emit_event(QueueEvent(
|
||||
event_type="run_paused",
|
||||
run_id=run_id,
|
||||
))
|
||||
|
||||
return True
|
||||
|
||||
async def resume(self, run_id: str) -> bool:
|
||||
"""Resume a paused run."""
|
||||
async with self._lock:
|
||||
if run_id not in self._running:
|
||||
return False
|
||||
|
||||
run = self._running[run_id]
|
||||
if run.state != RunState.PAUSED:
|
||||
return False
|
||||
|
||||
run.state = RunState.RUNNING
|
||||
|
||||
self._emit_event(QueueEvent(
|
||||
event_type="run_resumed",
|
||||
run_id=run_id,
|
||||
))
|
||||
|
||||
return True
|
||||
|
||||
def get_run(self, run_id: str) -> Optional[QueuedRun]:
|
||||
"""Get a run by ID."""
|
||||
return self._all_runs.get(run_id)
|
||||
|
||||
def get_queued(self) -> List[QueuedRun]:
|
||||
"""Get all queued runs in priority order."""
|
||||
result = []
|
||||
for priority in reversed(list(RunPriority)):
|
||||
result.extend(self._queues[priority])
|
||||
return result
|
||||
|
||||
def get_running(self) -> List[QueuedRun]:
|
||||
"""Get all running runs."""
|
||||
return list(self._running.values())
|
||||
|
||||
def get_all(self) -> List[QueuedRun]:
|
||||
"""Get all runs."""
|
||||
return list(self._all_runs.values())
|
||||
|
||||
@property
|
||||
def queued_count(self) -> int:
|
||||
"""Number of queued runs."""
|
||||
return sum(len(q) for q in self._queues.values())
|
||||
|
||||
@property
|
||||
def running_count(self) -> int:
|
||||
"""Number of running runs."""
|
||||
return len(self._running)
|
||||
|
||||
def _can_start_new(self) -> bool:
|
||||
"""Check if we can start a new run (global limit)."""
|
||||
return len(self._running) < self.config.max_concurrent_global
|
||||
|
||||
def _can_run(self, run: QueuedRun) -> bool:
|
||||
"""Check if a specific run can start (per-agent and per-workspace limits)."""
|
||||
# Count running by agent
|
||||
agent_count = sum(
|
||||
1 for r in self._running.values()
|
||||
if r.agent_name == run.agent_name
|
||||
)
|
||||
if agent_count >= self.config.max_concurrent_per_agent:
|
||||
return False
|
||||
|
||||
# Count running by workspace
|
||||
if run.workspace:
|
||||
workspace_count = sum(
|
||||
1 for r in self._running.values()
|
||||
if r.workspace == run.workspace
|
||||
)
|
||||
if workspace_count >= self.config.max_concurrent_per_workspace:
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
async def clear_queue(self) -> int:
|
||||
"""Clear all queued (not running) runs."""
|
||||
async with self._lock:
|
||||
count = 0
|
||||
for priority in RunPriority:
|
||||
count += len(self._queues[priority])
|
||||
for run in self._queues[priority]:
|
||||
run.state = RunState.CANCELLED
|
||||
del self._all_runs[run.run_id]
|
||||
self._queues[priority] = []
|
||||
|
||||
return count
|
||||
|
||||
def load_runs(self, runs: List[QueuedRun]) -> None:
|
||||
"""
|
||||
Load runs (e.g., from persistence after restart).
|
||||
|
||||
Queued runs are added to queues, running runs are re-queued.
|
||||
"""
|
||||
for run in runs:
|
||||
if run.state == RunState.RUNNING:
|
||||
# Re-queue running runs (they were interrupted)
|
||||
run.state = RunState.QUEUED
|
||||
run.started_at = None
|
||||
|
||||
if run.state == RunState.QUEUED:
|
||||
self._queues[run.priority].append(run)
|
||||
|
||||
self._all_runs[run.run_id] = run
|
||||
|
||||
logger.info(f"Loaded {len(runs)} runs from persistence")
|
||||
@@ -0,0 +1,372 @@
|
||||
"""
|
||||
Worker Pool for PraisonAI Queue System.
|
||||
|
||||
Manages async workers that execute queued runs.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import logging
|
||||
import time
|
||||
from collections import deque
|
||||
from typing import Any, Callable, Coroutine, Dict, List, Optional, Set
|
||||
|
||||
from .models import QueuedRun, RunState, StreamChunk, QueueEvent
|
||||
from .scheduler import QueueScheduler
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class StreamBuffer:
|
||||
"""Bounded buffer with backpressure for streaming output."""
|
||||
|
||||
def __init__(self, max_size: int = 1000, drop_strategy: str = "oldest"):
|
||||
"""
|
||||
Initialize stream buffer.
|
||||
|
||||
Args:
|
||||
max_size: Maximum number of chunks to buffer.
|
||||
drop_strategy: "oldest" to drop oldest, "newest" to reject new.
|
||||
"""
|
||||
self._buffer: deque = deque(maxlen=max_size)
|
||||
self._lock = asyncio.Lock()
|
||||
self._drop_strategy = drop_strategy
|
||||
self._dropped_count = 0
|
||||
self._max_size = max_size
|
||||
|
||||
async def push(self, chunk: StreamChunk) -> bool:
|
||||
"""
|
||||
Push a chunk to the buffer.
|
||||
|
||||
Returns:
|
||||
True if added, False if dropped.
|
||||
"""
|
||||
async with self._lock:
|
||||
if len(self._buffer) >= self._max_size:
|
||||
if self._drop_strategy == "oldest":
|
||||
self._buffer.popleft()
|
||||
self._dropped_count += 1
|
||||
elif self._drop_strategy == "newest":
|
||||
self._dropped_count += 1
|
||||
return False
|
||||
self._buffer.append(chunk)
|
||||
return True
|
||||
|
||||
async def drain(self, batch_size: int = 50) -> List[StreamChunk]:
|
||||
"""Drain up to batch_size chunks."""
|
||||
async with self._lock:
|
||||
result = []
|
||||
for _ in range(min(batch_size, len(self._buffer))):
|
||||
result.append(self._buffer.popleft())
|
||||
return result
|
||||
|
||||
async def drain_all(self) -> List[StreamChunk]:
|
||||
"""Drain all chunks."""
|
||||
async with self._lock:
|
||||
result = list(self._buffer)
|
||||
self._buffer.clear()
|
||||
return result
|
||||
|
||||
@property
|
||||
def size(self) -> int:
|
||||
"""Current buffer size."""
|
||||
return len(self._buffer)
|
||||
|
||||
@property
|
||||
def dropped_count(self) -> int:
|
||||
"""Number of dropped chunks."""
|
||||
return self._dropped_count
|
||||
|
||||
|
||||
class WorkerPool:
|
||||
"""
|
||||
Manages worker tasks for executing queued runs.
|
||||
|
||||
Workers poll the scheduler for runs and execute them with streaming output.
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
scheduler: QueueScheduler,
|
||||
on_output: Optional[Callable[[str, str], Coroutine[Any, Any, None]]] = None,
|
||||
on_complete: Optional[Callable[[str, QueuedRun], Coroutine[Any, Any, None]]] = None,
|
||||
on_error: Optional[Callable[[str, Exception], Coroutine[Any, Any, None]]] = None,
|
||||
on_event: Optional[Callable[[QueueEvent], Coroutine[Any, Any, None]]] = None,
|
||||
max_workers: int = 4,
|
||||
poll_interval: float = 0.1,
|
||||
stream_buffer_size: int = 1000,
|
||||
):
|
||||
"""
|
||||
Initialize worker pool.
|
||||
|
||||
Args:
|
||||
scheduler: The queue scheduler.
|
||||
on_output: Callback for streaming output (run_id, chunk).
|
||||
on_complete: Callback when run completes (run_id, run).
|
||||
on_error: Callback when run fails (run_id, exception).
|
||||
on_event: Callback for queue events.
|
||||
max_workers: Maximum number of concurrent workers.
|
||||
poll_interval: Seconds between scheduler polls.
|
||||
stream_buffer_size: Size of stream buffer per run.
|
||||
"""
|
||||
self.scheduler = scheduler
|
||||
self.on_output = on_output
|
||||
self.on_complete = on_complete
|
||||
self.on_error = on_error
|
||||
self.on_event = on_event
|
||||
self.max_workers = max_workers
|
||||
self.poll_interval = poll_interval
|
||||
self.stream_buffer_size = stream_buffer_size
|
||||
|
||||
self._workers: List[asyncio.Task] = []
|
||||
self._running = False
|
||||
self._stream_buffers: Dict[str, StreamBuffer] = {}
|
||||
self._active_runs: Set[str] = set()
|
||||
|
||||
async def start(self) -> None:
|
||||
"""Start the worker pool."""
|
||||
if self._running:
|
||||
return
|
||||
|
||||
self._running = True
|
||||
|
||||
for i in range(self.max_workers):
|
||||
task = asyncio.create_task(
|
||||
self._worker_loop(i),
|
||||
name=f"queue-worker-{i}"
|
||||
)
|
||||
self._workers.append(task)
|
||||
|
||||
logger.info(f"Started {self.max_workers} queue workers")
|
||||
|
||||
async def stop(self, timeout: float = 5.0) -> None:
|
||||
"""
|
||||
Stop all workers gracefully.
|
||||
|
||||
Args:
|
||||
timeout: Seconds to wait for workers to finish.
|
||||
"""
|
||||
self._running = False
|
||||
|
||||
if not self._workers:
|
||||
return
|
||||
|
||||
# Wait for workers to finish current work
|
||||
try:
|
||||
await asyncio.wait_for(
|
||||
asyncio.gather(*self._workers, return_exceptions=True),
|
||||
timeout=timeout
|
||||
)
|
||||
except asyncio.TimeoutError:
|
||||
# Cancel remaining workers
|
||||
for task in self._workers:
|
||||
if not task.done():
|
||||
task.cancel()
|
||||
|
||||
await asyncio.gather(*self._workers, return_exceptions=True)
|
||||
|
||||
self._workers.clear()
|
||||
logger.info("Stopped queue workers")
|
||||
|
||||
async def _worker_loop(self, worker_id: int) -> None:
|
||||
"""Main worker loop."""
|
||||
logger.debug(f"Worker {worker_id} started")
|
||||
|
||||
while self._running:
|
||||
try:
|
||||
# Get next run from scheduler
|
||||
run = await self.scheduler.next()
|
||||
|
||||
if run is None:
|
||||
# No work available, wait and retry
|
||||
await asyncio.sleep(self.poll_interval)
|
||||
continue
|
||||
|
||||
# Execute the run
|
||||
self._active_runs.add(run.run_id)
|
||||
try:
|
||||
await self._execute_run(run, worker_id)
|
||||
finally:
|
||||
self._active_runs.discard(run.run_id)
|
||||
|
||||
except asyncio.CancelledError:
|
||||
logger.debug(f"Worker {worker_id} cancelled")
|
||||
break
|
||||
except Exception as e:
|
||||
logger.error(f"Worker {worker_id} error: {e}", exc_info=True)
|
||||
await asyncio.sleep(self.poll_interval)
|
||||
|
||||
logger.debug(f"Worker {worker_id} stopped")
|
||||
|
||||
async def _execute_run(self, run: QueuedRun, worker_id: int) -> None:
|
||||
"""Execute a single run with streaming output."""
|
||||
logger.debug(f"Worker {worker_id} executing run {run.run_id}")
|
||||
|
||||
# Create stream buffer for this run
|
||||
buffer = StreamBuffer(max_size=self.stream_buffer_size)
|
||||
self._stream_buffers[run.run_id] = buffer
|
||||
|
||||
output_chunks: List[str] = []
|
||||
|
||||
try:
|
||||
# Import agent lazily to avoid circular imports
|
||||
from praisonaiagents import Agent
|
||||
|
||||
# Get agent configuration
|
||||
agent_config = run.config.get("agent_config", {})
|
||||
agent_name = run.agent_name or agent_config.get("name", "Assistant")
|
||||
|
||||
# Create agent
|
||||
agent = Agent(
|
||||
name=agent_name,
|
||||
instructions=agent_config.get("instructions", "You are a helpful assistant."),
|
||||
model=agent_config.get("model"),
|
||||
tools=agent_config.get("tools", []),
|
||||
verbose=agent_config.get("verbose", False),
|
||||
)
|
||||
|
||||
# Check for cancellation before starting
|
||||
if self.scheduler.is_cancelled(run.run_id):
|
||||
logger.debug(f"Run {run.run_id} was cancelled before execution")
|
||||
return
|
||||
|
||||
# Execute with streaming if available
|
||||
chunk_index = 0
|
||||
|
||||
# Try streaming first
|
||||
try:
|
||||
async for chunk in self._stream_agent(agent, run.input_content):
|
||||
# Check for cancellation
|
||||
if self.scheduler.is_cancelled(run.run_id):
|
||||
logger.debug(f"Run {run.run_id} cancelled during execution")
|
||||
self.scheduler.clear_cancel_token(run.run_id)
|
||||
return
|
||||
|
||||
# Check for pause
|
||||
while run.state == RunState.PAUSED:
|
||||
await asyncio.sleep(0.1)
|
||||
if self.scheduler.is_cancelled(run.run_id):
|
||||
return
|
||||
|
||||
output_chunks.append(chunk)
|
||||
|
||||
# Buffer the chunk
|
||||
stream_chunk = StreamChunk(
|
||||
run_id=run.run_id,
|
||||
content=chunk,
|
||||
chunk_index=chunk_index,
|
||||
)
|
||||
await buffer.push(stream_chunk)
|
||||
chunk_index += 1
|
||||
|
||||
# Emit output callback
|
||||
if self.on_output:
|
||||
try:
|
||||
await self.on_output(run.run_id, chunk)
|
||||
except Exception as e:
|
||||
logger.error(f"Error in output callback: {e}")
|
||||
|
||||
except Exception as stream_error:
|
||||
# Fallback to non-streaming
|
||||
logger.debug(f"Streaming failed, falling back to sync: {stream_error}")
|
||||
result = agent.chat(run.input_content)
|
||||
output_chunks = [result]
|
||||
|
||||
if self.on_output:
|
||||
try:
|
||||
await self.on_output(run.run_id, result)
|
||||
except Exception as e:
|
||||
logger.error(f"Error in output callback: {e}")
|
||||
|
||||
# Combine output
|
||||
full_output = "".join(output_chunks)
|
||||
|
||||
# Mark final chunk
|
||||
final_chunk = StreamChunk(
|
||||
run_id=run.run_id,
|
||||
content="",
|
||||
chunk_index=chunk_index,
|
||||
is_final=True,
|
||||
)
|
||||
await buffer.push(final_chunk)
|
||||
|
||||
# Complete the run
|
||||
completed_run = await self.scheduler.complete(
|
||||
run.run_id,
|
||||
output=full_output,
|
||||
metrics={
|
||||
"chunks": chunk_index,
|
||||
"output_length": len(full_output),
|
||||
}
|
||||
)
|
||||
|
||||
if completed_run and self.on_complete:
|
||||
try:
|
||||
await self.on_complete(run.run_id, completed_run)
|
||||
except Exception as e:
|
||||
logger.error(f"Error in complete callback: {e}")
|
||||
|
||||
except asyncio.CancelledError:
|
||||
# Worker was cancelled
|
||||
await self.scheduler.cancel(run.run_id)
|
||||
raise
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Run {run.run_id} failed: {e}", exc_info=True)
|
||||
|
||||
# Mark as failed
|
||||
failed_run = await self.scheduler.fail(run.run_id, str(e))
|
||||
|
||||
if self.on_error:
|
||||
try:
|
||||
await self.on_error(run.run_id, e)
|
||||
except Exception as callback_error:
|
||||
logger.error(f"Error in error callback: {callback_error}")
|
||||
|
||||
finally:
|
||||
# Cleanup buffer
|
||||
self._stream_buffers.pop(run.run_id, None)
|
||||
|
||||
async def _stream_agent(self, agent, input_content: str):
|
||||
"""
|
||||
Stream output from agent.
|
||||
|
||||
Yields chunks of output text.
|
||||
"""
|
||||
# Try to use agent's async streaming if available
|
||||
if hasattr(agent, 'astream'):
|
||||
async for chunk in agent.astream(input_content):
|
||||
yield chunk
|
||||
elif hasattr(agent, 'stream'):
|
||||
# Wrap sync stream in async
|
||||
for chunk in agent.stream(input_content):
|
||||
yield chunk
|
||||
await asyncio.sleep(0) # Yield control
|
||||
else:
|
||||
# Fallback: get full response and yield in chunks
|
||||
result = agent.chat(input_content)
|
||||
|
||||
# Simulate streaming by yielding words
|
||||
words = result.split()
|
||||
chunk_size = 3 # Words per chunk
|
||||
|
||||
for i in range(0, len(words), chunk_size):
|
||||
chunk_words = words[i:i + chunk_size]
|
||||
chunk = " ".join(chunk_words)
|
||||
if i + chunk_size < len(words):
|
||||
chunk += " "
|
||||
yield chunk
|
||||
await asyncio.sleep(0.05) # Small delay for streaming effect
|
||||
|
||||
def get_stream_buffer(self, run_id: str) -> Optional[StreamBuffer]:
|
||||
"""Get the stream buffer for a run."""
|
||||
return self._stream_buffers.get(run_id)
|
||||
|
||||
@property
|
||||
def active_run_count(self) -> int:
|
||||
"""Number of currently executing runs."""
|
||||
return len(self._active_runs)
|
||||
|
||||
@property
|
||||
def is_running(self) -> bool:
|
||||
"""Whether the worker pool is running."""
|
||||
return self._running
|
||||
@@ -0,0 +1,83 @@
|
||||
"""
|
||||
TUI (Terminal User Interface) for PraisonAI.
|
||||
|
||||
Provides an app-like interactive experience with:
|
||||
- Event-loop driven async UI
|
||||
- Multi-pane layout (chat, tools, queue, status)
|
||||
- Streaming output
|
||||
- Queue management
|
||||
- Session persistence
|
||||
- Headless simulation mode for CI/testing
|
||||
|
||||
Requires the [tui] extra: pip install praisonai[tui]
|
||||
"""
|
||||
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from .app import TUIApp
|
||||
from .events import TUIEvent, TUIEventType
|
||||
from .orchestrator import TuiOrchestrator, UIStateModel, SimulationRunner
|
||||
from .mock_provider import MockProvider, MockProviderConfig
|
||||
|
||||
_lazy_cache = {}
|
||||
|
||||
|
||||
def __getattr__(name: str):
|
||||
"""Lazy load TUI components."""
|
||||
global _lazy_cache
|
||||
|
||||
if name in _lazy_cache:
|
||||
return _lazy_cache[name]
|
||||
|
||||
if name == "TUIApp":
|
||||
from .app import TUIApp
|
||||
_lazy_cache[name] = TUIApp
|
||||
return TUIApp
|
||||
elif name == "TUIEvent":
|
||||
from .events import TUIEvent
|
||||
_lazy_cache[name] = TUIEvent
|
||||
return TUIEvent
|
||||
elif name == "TUIEventType":
|
||||
from .events import TUIEventType
|
||||
_lazy_cache[name] = TUIEventType
|
||||
return TUIEventType
|
||||
elif name == "run_tui":
|
||||
from .app import run_tui
|
||||
_lazy_cache[name] = run_tui
|
||||
return run_tui
|
||||
elif name == "TuiOrchestrator":
|
||||
from .orchestrator import TuiOrchestrator
|
||||
_lazy_cache[name] = TuiOrchestrator
|
||||
return TuiOrchestrator
|
||||
elif name == "UIStateModel":
|
||||
from .orchestrator import UIStateModel
|
||||
_lazy_cache[name] = UIStateModel
|
||||
return UIStateModel
|
||||
elif name == "SimulationRunner":
|
||||
from .orchestrator import SimulationRunner
|
||||
_lazy_cache[name] = SimulationRunner
|
||||
return SimulationRunner
|
||||
elif name == "MockProvider":
|
||||
from .mock_provider import MockProvider
|
||||
_lazy_cache[name] = MockProvider
|
||||
return MockProvider
|
||||
elif name == "MockProviderConfig":
|
||||
from .mock_provider import MockProviderConfig
|
||||
_lazy_cache[name] = MockProviderConfig
|
||||
return MockProviderConfig
|
||||
|
||||
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|
||||
|
||||
|
||||
__all__ = [
|
||||
"TUIApp",
|
||||
"TUIEvent",
|
||||
"TUIEventType",
|
||||
"run_tui",
|
||||
"TuiOrchestrator",
|
||||
"UIStateModel",
|
||||
"SimulationRunner",
|
||||
"MockProvider",
|
||||
"MockProviderConfig",
|
||||
]
|
||||
@@ -0,0 +1,529 @@
|
||||
"""
|
||||
Main TUI Application for PraisonAI.
|
||||
|
||||
Event-loop driven async TUI with multi-pane layout.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import logging
|
||||
import os
|
||||
import uuid
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
try:
|
||||
from textual.app import App, ComposeResult
|
||||
from textual.binding import Binding
|
||||
from textual.screen import Screen
|
||||
from textual import events
|
||||
TEXTUAL_AVAILABLE = True
|
||||
except ImportError:
|
||||
TEXTUAL_AVAILABLE = False
|
||||
App = object
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
if TEXTUAL_AVAILABLE:
|
||||
from .screens.main import MainScreen
|
||||
from .screens.queue import QueueScreen
|
||||
from .screens.settings import SettingsScreen
|
||||
from .screens.session import SessionScreen
|
||||
from .widgets.chat import ChatMessage
|
||||
from .widgets.status import StatusInfo
|
||||
from .widgets.queue_panel import QueueItem
|
||||
from ..queue import QueueManager, QueueConfig, QueuedRun, RunPriority
|
||||
|
||||
class TUIApp(App):
|
||||
"""
|
||||
PraisonAI Terminal User Interface Application.
|
||||
|
||||
Features:
|
||||
- Event-loop driven async UI
|
||||
- Multi-pane layout (chat, tools, queue, status)
|
||||
- Streaming output
|
||||
- Queue management
|
||||
- Session persistence
|
||||
"""
|
||||
|
||||
TITLE = "PraisonAI"
|
||||
SUB_TITLE = "AI Agent Terminal"
|
||||
|
||||
CSS = """
|
||||
Screen {
|
||||
background: $surface;
|
||||
}
|
||||
"""
|
||||
|
||||
BINDINGS = [
|
||||
Binding("ctrl+q", "quit", "Quit", show=True, priority=True),
|
||||
Binding("ctrl+c", "cancel", "Cancel", show=True),
|
||||
]
|
||||
|
||||
SCREENS = {
|
||||
"main": MainScreen,
|
||||
"queue": QueueScreen,
|
||||
"settings": SettingsScreen,
|
||||
"sessions": SessionScreen,
|
||||
}
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
workspace: Optional[str] = None,
|
||||
session_id: Optional[str] = None,
|
||||
model: Optional[str] = None,
|
||||
agent_config: Optional[Dict[str, Any]] = None,
|
||||
queue_config: Optional[QueueConfig] = None,
|
||||
):
|
||||
super().__init__()
|
||||
|
||||
self.workspace = workspace or os.getcwd()
|
||||
self.session_id = session_id or str(uuid.uuid4())[:8]
|
||||
self.model = model or os.environ.get("OPENAI_MODEL", "gpt-4o-mini")
|
||||
self.agent_config = agent_config or {}
|
||||
|
||||
# Queue manager
|
||||
self.queue_config = queue_config or QueueConfig()
|
||||
self.queue_manager: Optional[QueueManager] = None
|
||||
|
||||
# State
|
||||
self._current_run_id: Optional[str] = None
|
||||
self._streaming_content: str = ""
|
||||
self._total_tokens: int = 0
|
||||
self._total_cost: float = 0.0
|
||||
|
||||
# Slash command handlers
|
||||
self._command_handlers: Dict[str, callable] = {}
|
||||
self._setup_commands()
|
||||
|
||||
def _setup_commands(self) -> None:
|
||||
"""Setup slash command handlers."""
|
||||
self._command_handlers = {
|
||||
"help": self._cmd_help,
|
||||
"clear": self._cmd_clear,
|
||||
"model": self._cmd_model,
|
||||
"queue": self._cmd_queue,
|
||||
"cancel": self._cmd_cancel,
|
||||
"retry": self._cmd_retry,
|
||||
"settings": self._cmd_settings,
|
||||
"sessions": self._cmd_sessions,
|
||||
"cost": self._cmd_cost,
|
||||
"exit": self._cmd_exit,
|
||||
"quit": self._cmd_exit,
|
||||
}
|
||||
|
||||
async def on_mount(self) -> None:
|
||||
"""Handle app mount."""
|
||||
# Initialize queue manager
|
||||
self.queue_manager = QueueManager(
|
||||
config=self.queue_config,
|
||||
on_output=self._handle_output,
|
||||
on_complete=self._handle_complete,
|
||||
on_error=self._handle_error,
|
||||
)
|
||||
|
||||
await self.queue_manager.start(recover=True)
|
||||
self.queue_manager.set_session(self.session_id)
|
||||
|
||||
# Push main screen
|
||||
await self.push_screen("main")
|
||||
|
||||
# Update status
|
||||
self._update_status()
|
||||
|
||||
async def on_unmount(self) -> None:
|
||||
"""Handle app unmount."""
|
||||
if self.queue_manager:
|
||||
await self.queue_manager.stop()
|
||||
|
||||
def compose(self) -> ComposeResult:
|
||||
"""Compose the app."""
|
||||
yield from ()
|
||||
|
||||
# Event handlers
|
||||
|
||||
async def on_main_screen_message_submitted(
|
||||
self, event: MainScreen.MessageSubmitted
|
||||
) -> None:
|
||||
"""Handle message submission from main screen."""
|
||||
content = event.content
|
||||
|
||||
# Add user message to chat
|
||||
main_screen = self.screen
|
||||
if isinstance(main_screen, MainScreen):
|
||||
await main_screen.add_user_message(content)
|
||||
main_screen.set_processing(True)
|
||||
|
||||
# Submit to queue
|
||||
try:
|
||||
run_id = await self.queue_manager.submit(
|
||||
input_content=content,
|
||||
agent_name=self.agent_config.get("name", "Assistant"),
|
||||
config={
|
||||
"agent_config": {
|
||||
"name": self.agent_config.get("name", "Assistant"),
|
||||
"instructions": self.agent_config.get(
|
||||
"instructions",
|
||||
"You are a helpful AI assistant."
|
||||
),
|
||||
"model": self.model,
|
||||
"tools": self.agent_config.get("tools", []),
|
||||
"verbose": False,
|
||||
}
|
||||
}
|
||||
)
|
||||
|
||||
self._current_run_id = run_id
|
||||
self._streaming_content = ""
|
||||
|
||||
# Add streaming placeholder
|
||||
if isinstance(main_screen, MainScreen):
|
||||
await main_screen.add_assistant_message(
|
||||
content="",
|
||||
run_id=run_id,
|
||||
agent_name=self.agent_config.get("name", "Assistant"),
|
||||
streaming=True,
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to submit: {e}")
|
||||
if isinstance(main_screen, MainScreen):
|
||||
main_screen.set_processing(False)
|
||||
await main_screen.add_assistant_message(
|
||||
content=f"Error: {e}",
|
||||
agent_name="System",
|
||||
)
|
||||
|
||||
self._update_status()
|
||||
|
||||
async def on_main_screen_command_executed(
|
||||
self, event: MainScreen.CommandExecuted
|
||||
) -> None:
|
||||
"""Handle command execution."""
|
||||
command = event.command.lower()
|
||||
args = event.args
|
||||
|
||||
handler = self._command_handlers.get(command)
|
||||
if handler:
|
||||
await handler(args)
|
||||
else:
|
||||
# Unknown command
|
||||
main_screen = self.screen
|
||||
if isinstance(main_screen, MainScreen):
|
||||
await main_screen.add_assistant_message(
|
||||
content=f"Unknown command: /{command}\nType /help for available commands.",
|
||||
agent_name="System",
|
||||
)
|
||||
|
||||
async def on_main_screen_cancel_requested(
|
||||
self, event: MainScreen.CancelRequested
|
||||
) -> None:
|
||||
"""Handle cancel request."""
|
||||
if self._current_run_id:
|
||||
await self.queue_manager.cancel(self._current_run_id)
|
||||
self._current_run_id = None
|
||||
|
||||
main_screen = self.screen
|
||||
if isinstance(main_screen, MainScreen):
|
||||
main_screen.set_processing(False)
|
||||
await main_screen.add_assistant_message(
|
||||
content="Operation cancelled.",
|
||||
agent_name="System",
|
||||
)
|
||||
|
||||
self._update_status()
|
||||
|
||||
async def on_queue_screen_run_cancelled(
|
||||
self, event: QueueScreen.RunCancelled
|
||||
) -> None:
|
||||
"""Handle run cancellation from queue screen."""
|
||||
await self.queue_manager.cancel(event.run_id)
|
||||
self._update_queue_screen()
|
||||
|
||||
async def on_queue_screen_run_retried(
|
||||
self, event: QueueScreen.RunRetried
|
||||
) -> None:
|
||||
"""Handle run retry from queue screen."""
|
||||
await self.queue_manager.retry(event.run_id)
|
||||
self._update_queue_screen()
|
||||
|
||||
async def on_queue_screen_queue_cleared(
|
||||
self, event: QueueScreen.QueueCleared
|
||||
) -> None:
|
||||
"""Handle queue clear."""
|
||||
await self.queue_manager.clear_queue()
|
||||
self._update_queue_screen()
|
||||
|
||||
async def on_settings_screen_settings_saved(
|
||||
self, event: SettingsScreen.SettingsSaved
|
||||
) -> None:
|
||||
"""Handle settings save."""
|
||||
settings = event.settings
|
||||
|
||||
if "model" in settings:
|
||||
self.model = settings["model"]
|
||||
|
||||
if "max_concurrent" in settings:
|
||||
self.queue_config.max_concurrent_global = settings["max_concurrent"]
|
||||
|
||||
if "autosave_interval" in settings:
|
||||
self.queue_config.autosave_interval_seconds = settings["autosave_interval"]
|
||||
|
||||
self._update_status()
|
||||
|
||||
async def on_session_screen_session_selected(
|
||||
self, event: SessionScreen.SessionSelected
|
||||
) -> None:
|
||||
"""Handle session selection."""
|
||||
self.session_id = event.session_id
|
||||
self.queue_manager.set_session(self.session_id)
|
||||
self._update_status()
|
||||
|
||||
# Queue callbacks
|
||||
|
||||
async def _handle_output(self, run_id: str, chunk: str) -> None:
|
||||
"""Handle streaming output."""
|
||||
if run_id != self._current_run_id:
|
||||
return
|
||||
|
||||
self._streaming_content += chunk
|
||||
|
||||
main_screen = self.screen
|
||||
if isinstance(main_screen, MainScreen):
|
||||
await main_screen.update_streaming(run_id, self._streaming_content)
|
||||
|
||||
async def _handle_complete(self, run_id: str, run: QueuedRun) -> None:
|
||||
"""Handle run completion."""
|
||||
main_screen = self.screen
|
||||
if isinstance(main_screen, MainScreen):
|
||||
await main_screen.complete_streaming(
|
||||
run_id,
|
||||
run.output_content or self._streaming_content
|
||||
)
|
||||
main_screen.set_processing(False)
|
||||
|
||||
# Update metrics
|
||||
if run.metrics:
|
||||
self._total_tokens += run.metrics.get("tokens", 0)
|
||||
self._total_cost += run.metrics.get("cost", 0.0)
|
||||
|
||||
if run_id == self._current_run_id:
|
||||
self._current_run_id = None
|
||||
self._streaming_content = ""
|
||||
|
||||
self._update_status()
|
||||
|
||||
async def _handle_error(self, run_id: str, error: Exception) -> None:
|
||||
"""Handle run error."""
|
||||
main_screen = self.screen
|
||||
if isinstance(main_screen, MainScreen):
|
||||
await main_screen.complete_streaming(run_id, f"Error: {error}")
|
||||
main_screen.set_processing(False)
|
||||
|
||||
if run_id == self._current_run_id:
|
||||
self._current_run_id = None
|
||||
self._streaming_content = ""
|
||||
|
||||
self._update_status()
|
||||
|
||||
# Command handlers
|
||||
|
||||
async def _cmd_help(self, args: str) -> None:
|
||||
"""Show help."""
|
||||
help_text = """
|
||||
**Available Commands:**
|
||||
|
||||
- `/help` - Show this help
|
||||
- `/clear` - Clear chat history
|
||||
- `/model [name]` - Show or set model
|
||||
- `/queue` - Show queue screen
|
||||
- `/cancel [run_id]` - Cancel a run
|
||||
- `/retry [run_id]` - Retry a failed run
|
||||
- `/settings` - Show settings
|
||||
- `/sessions` - Browse sessions
|
||||
- `/cost` - Show cost summary
|
||||
- `/exit` or `/quit` - Exit TUI
|
||||
|
||||
**Keyboard Shortcuts:**
|
||||
|
||||
- `Ctrl+Enter` - Send message
|
||||
- `Ctrl+C` - Cancel current operation
|
||||
- `Ctrl+Q` - Quit
|
||||
- `Ctrl+L` - Clear screen
|
||||
- `F1` - Help
|
||||
- `F2` - Toggle queue panel
|
||||
- `F3` - Settings
|
||||
- `F5` - Clear chat
|
||||
- `Tab` - Cycle focus
|
||||
- `Esc` - Cancel/close
|
||||
"""
|
||||
main_screen = self.screen
|
||||
if isinstance(main_screen, MainScreen):
|
||||
await main_screen.add_assistant_message(
|
||||
content=help_text,
|
||||
agent_name="Help",
|
||||
)
|
||||
|
||||
async def _cmd_clear(self, args: str) -> None:
|
||||
"""Clear chat."""
|
||||
main_screen = self.screen
|
||||
if isinstance(main_screen, MainScreen):
|
||||
await main_screen.query_one("#chat-widget").clear()
|
||||
|
||||
async def _cmd_model(self, args: str) -> None:
|
||||
"""Show or set model."""
|
||||
main_screen = self.screen
|
||||
if args:
|
||||
self.model = args.strip()
|
||||
msg = f"Model set to: {self.model}"
|
||||
else:
|
||||
msg = f"Current model: {self.model}"
|
||||
|
||||
if isinstance(main_screen, MainScreen):
|
||||
await main_screen.add_assistant_message(content=msg, agent_name="System")
|
||||
|
||||
self._update_status()
|
||||
|
||||
async def _cmd_queue(self, args: str) -> None:
|
||||
"""Show queue screen."""
|
||||
await self.push_screen("queue")
|
||||
self._update_queue_screen()
|
||||
|
||||
async def _cmd_cancel(self, args: str) -> None:
|
||||
"""Cancel a run."""
|
||||
run_id = args.strip() if args else self._current_run_id
|
||||
if run_id:
|
||||
await self.queue_manager.cancel(run_id)
|
||||
main_screen = self.screen
|
||||
if isinstance(main_screen, MainScreen):
|
||||
await main_screen.add_assistant_message(
|
||||
content=f"Cancelled run: {run_id}",
|
||||
agent_name="System",
|
||||
)
|
||||
self._update_status()
|
||||
|
||||
async def _cmd_retry(self, args: str) -> None:
|
||||
"""Retry a failed run."""
|
||||
run_id = args.strip()
|
||||
if run_id:
|
||||
new_id = await self.queue_manager.retry(run_id)
|
||||
main_screen = self.screen
|
||||
if isinstance(main_screen, MainScreen):
|
||||
if new_id:
|
||||
await main_screen.add_assistant_message(
|
||||
content=f"Retrying run {run_id} as {new_id}",
|
||||
agent_name="System",
|
||||
)
|
||||
else:
|
||||
await main_screen.add_assistant_message(
|
||||
content=f"Cannot retry run: {run_id}",
|
||||
agent_name="System",
|
||||
)
|
||||
self._update_status()
|
||||
|
||||
async def _cmd_settings(self, args: str) -> None:
|
||||
"""Show settings."""
|
||||
await self.push_screen(SettingsScreen(current_settings={
|
||||
"model": self.model,
|
||||
"max_concurrent": self.queue_config.max_concurrent_global,
|
||||
"autosave_interval": int(self.queue_config.autosave_interval_seconds),
|
||||
}))
|
||||
|
||||
async def _cmd_sessions(self, args: str) -> None:
|
||||
"""Show sessions."""
|
||||
await self.push_screen("sessions")
|
||||
|
||||
async def _cmd_cost(self, args: str) -> None:
|
||||
"""Show cost summary."""
|
||||
msg = f"""
|
||||
**Cost Summary:**
|
||||
|
||||
- Total Tokens: {self._total_tokens:,}
|
||||
- Total Cost: ${self._total_cost:.4f}
|
||||
- Session: {self.session_id}
|
||||
"""
|
||||
main_screen = self.screen
|
||||
if isinstance(main_screen, MainScreen):
|
||||
await main_screen.add_assistant_message(content=msg, agent_name="System")
|
||||
|
||||
async def _cmd_exit(self, args: str) -> None:
|
||||
"""Exit TUI."""
|
||||
self.exit()
|
||||
|
||||
# Helper methods
|
||||
|
||||
def _update_status(self) -> None:
|
||||
"""Update status bar."""
|
||||
main_screen = self.screen
|
||||
if isinstance(main_screen, MainScreen):
|
||||
main_screen.update_status(StatusInfo(
|
||||
session_id=self.session_id,
|
||||
model=self.model,
|
||||
total_tokens=self._total_tokens,
|
||||
total_cost=self._total_cost,
|
||||
queued_count=self.queue_manager.queued_count if self.queue_manager else 0,
|
||||
running_count=self.queue_manager.running_count if self.queue_manager else 0,
|
||||
is_processing=self._current_run_id is not None,
|
||||
))
|
||||
|
||||
def _update_queue_screen(self) -> None:
|
||||
"""Update queue screen if visible."""
|
||||
if isinstance(self.screen, QueueScreen):
|
||||
runs = self.queue_manager.list_runs(limit=100)
|
||||
self.screen.update_runs([
|
||||
{
|
||||
"run_id": r.run_id,
|
||||
"agent_name": r.agent_name,
|
||||
"input": r.input_content,
|
||||
"state": r.state.value,
|
||||
"priority": r.priority.name.lower(),
|
||||
"wait_time": f"{r.wait_seconds:.1f}s" if r.wait_seconds else "-",
|
||||
"duration": f"{r.duration_seconds:.1f}s" if r.duration_seconds else "-",
|
||||
}
|
||||
for r in runs
|
||||
])
|
||||
|
||||
def action_quit(self) -> None:
|
||||
"""Quit the application."""
|
||||
self.exit()
|
||||
|
||||
def action_cancel(self) -> None:
|
||||
"""Cancel current operation."""
|
||||
if self._current_run_id and self.queue_manager:
|
||||
asyncio.create_task(self.queue_manager.cancel(self._current_run_id))
|
||||
|
||||
else:
|
||||
class TUIApp:
|
||||
"""Placeholder when Textual is not available."""
|
||||
def __init__(self, *args, **kwargs):
|
||||
raise ImportError(
|
||||
"Textual is required for TUI. Install with: pip install praisonai[tui]"
|
||||
)
|
||||
|
||||
|
||||
def run_tui(
|
||||
workspace: Optional[str] = None,
|
||||
session_id: Optional[str] = None,
|
||||
model: Optional[str] = None,
|
||||
agent_config: Optional[Dict[str, Any]] = None,
|
||||
) -> None:
|
||||
"""
|
||||
Run the PraisonAI TUI.
|
||||
|
||||
Args:
|
||||
workspace: Working directory.
|
||||
session_id: Session ID to resume.
|
||||
model: Default model.
|
||||
agent_config: Agent configuration.
|
||||
"""
|
||||
if not TEXTUAL_AVAILABLE:
|
||||
raise ImportError(
|
||||
"Textual is required for TUI. Install with: pip install praisonai[tui]"
|
||||
)
|
||||
|
||||
app = TUIApp(
|
||||
workspace=workspace,
|
||||
session_id=session_id,
|
||||
model=model,
|
||||
agent_config=agent_config,
|
||||
)
|
||||
app.run()
|
||||
@@ -0,0 +1,556 @@
|
||||
"""
|
||||
CLI Commands for PraisonAI TUI and Queue.
|
||||
|
||||
Provides Typer-based CLI commands for:
|
||||
- praison tui - Launch interactive TUI
|
||||
- praison run - Execute single task
|
||||
- praison queue ls/cancel/retry/clear
|
||||
- praison session ls/resume
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
from typing import Optional
|
||||
|
||||
try:
|
||||
import typer
|
||||
from rich.console import Console
|
||||
from rich.table import Table
|
||||
from rich.panel import Panel
|
||||
TYPER_AVAILABLE = True
|
||||
except ImportError:
|
||||
TYPER_AVAILABLE = False
|
||||
typer = None
|
||||
|
||||
console = Console() if TYPER_AVAILABLE else None
|
||||
|
||||
|
||||
def create_tui_app():
|
||||
"""Create the TUI CLI app."""
|
||||
if not TYPER_AVAILABLE:
|
||||
return None
|
||||
|
||||
app = typer.Typer(
|
||||
name="tui",
|
||||
help="PraisonAI TUI and Queue commands",
|
||||
no_args_is_help=True,
|
||||
)
|
||||
|
||||
@app.command("launch")
|
||||
def tui_launch(
|
||||
workspace: Optional[str] = typer.Option(
|
||||
None, "--workspace", "-w",
|
||||
help="Workspace directory"
|
||||
),
|
||||
session: Optional[str] = typer.Option(
|
||||
None, "--session", "-s",
|
||||
help="Resume session ID"
|
||||
),
|
||||
model: Optional[str] = typer.Option(
|
||||
None, "--model", "-m",
|
||||
help="Default model"
|
||||
),
|
||||
agent: Optional[str] = typer.Option(
|
||||
None, "--agent", "-a",
|
||||
help="Agent config file (YAML)"
|
||||
),
|
||||
):
|
||||
"""Launch the interactive TUI."""
|
||||
try:
|
||||
from .app import run_tui
|
||||
except ImportError:
|
||||
console.print(
|
||||
"[red]Textual is required for TUI. Install with:[/red]\n"
|
||||
" pip install praisonai[tui]"
|
||||
)
|
||||
raise typer.Exit(1)
|
||||
|
||||
# Load agent config if provided
|
||||
agent_config = {}
|
||||
if agent:
|
||||
import yaml
|
||||
with open(agent, "r") as f:
|
||||
agent_config = yaml.safe_load(f)
|
||||
|
||||
run_tui(
|
||||
workspace=workspace,
|
||||
session_id=session,
|
||||
model=model,
|
||||
agent_config=agent_config,
|
||||
)
|
||||
|
||||
return app
|
||||
|
||||
|
||||
def create_queue_app():
|
||||
"""Create the queue CLI app."""
|
||||
if not TYPER_AVAILABLE:
|
||||
return None
|
||||
|
||||
app = typer.Typer(
|
||||
name="queue",
|
||||
help="Queue management commands",
|
||||
no_args_is_help=True,
|
||||
)
|
||||
|
||||
@app.command("ls")
|
||||
def queue_ls(
|
||||
state: Optional[str] = typer.Option(
|
||||
None, "--state", "-s",
|
||||
help="Filter by state (queued, running, succeeded, failed, cancelled)"
|
||||
),
|
||||
session: Optional[str] = typer.Option(
|
||||
None, "--session",
|
||||
help="Filter by session ID"
|
||||
),
|
||||
limit: int = typer.Option(
|
||||
20, "--limit", "-n",
|
||||
help="Maximum number of results"
|
||||
),
|
||||
json_output: bool = typer.Option(
|
||||
False, "--json", "-j",
|
||||
help="Output as JSON"
|
||||
),
|
||||
):
|
||||
"""List queued runs."""
|
||||
from ..queue import QueuePersistence, RunState
|
||||
|
||||
persistence = QueuePersistence()
|
||||
persistence.initialize()
|
||||
|
||||
# Parse state filter
|
||||
state_filter = None
|
||||
if state:
|
||||
try:
|
||||
state_filter = RunState(state.lower())
|
||||
except ValueError:
|
||||
console.print(f"[red]Invalid state: {state}[/red]")
|
||||
raise typer.Exit(1)
|
||||
|
||||
runs = persistence.list_runs(
|
||||
state=state_filter,
|
||||
session_id=session,
|
||||
limit=limit,
|
||||
)
|
||||
|
||||
if json_output:
|
||||
output = [r.to_dict() for r in runs]
|
||||
console.print(json.dumps(output, indent=2, default=str))
|
||||
return
|
||||
|
||||
if not runs:
|
||||
console.print("[dim]No runs found.[/dim]")
|
||||
return
|
||||
|
||||
# Create table
|
||||
table = Table(title="Queue Runs")
|
||||
table.add_column("ID", style="cyan")
|
||||
table.add_column("Agent", style="green")
|
||||
table.add_column("Input", max_width=30)
|
||||
table.add_column("State", style="yellow")
|
||||
table.add_column("Priority")
|
||||
table.add_column("Wait", style="dim")
|
||||
table.add_column("Duration", style="dim")
|
||||
|
||||
for run in runs:
|
||||
state_style = {
|
||||
"queued": "yellow",
|
||||
"running": "green",
|
||||
"succeeded": "cyan",
|
||||
"failed": "red",
|
||||
"cancelled": "dim",
|
||||
}.get(run.state.value, "")
|
||||
|
||||
input_preview = run.input_content[:27] + "..." if len(run.input_content) > 30 else run.input_content
|
||||
|
||||
table.add_row(
|
||||
run.run_id[:8],
|
||||
run.agent_name,
|
||||
input_preview,
|
||||
f"[{state_style}]{run.state.value}[/{state_style}]",
|
||||
run.priority.name.lower(),
|
||||
f"{run.wait_seconds:.1f}s" if run.wait_seconds else "-",
|
||||
f"{run.duration_seconds:.1f}s" if run.duration_seconds else "-",
|
||||
)
|
||||
|
||||
console.print(table)
|
||||
persistence.close()
|
||||
|
||||
@app.command("cancel")
|
||||
def queue_cancel(
|
||||
run_id: str = typer.Argument(..., help="Run ID to cancel"),
|
||||
):
|
||||
"""Cancel a queued or running run."""
|
||||
from ..queue import QueuePersistence, RunState
|
||||
|
||||
persistence = QueuePersistence()
|
||||
persistence.initialize()
|
||||
|
||||
# Find the run
|
||||
run = persistence.load_run(run_id)
|
||||
if not run:
|
||||
# Try partial match
|
||||
runs = persistence.list_runs(limit=100)
|
||||
matches = [r for r in runs if r.run_id.startswith(run_id)]
|
||||
if len(matches) == 1:
|
||||
run = matches[0]
|
||||
elif len(matches) > 1:
|
||||
console.print(f"[yellow]Multiple matches for '{run_id}':[/yellow]")
|
||||
for r in matches:
|
||||
console.print(f" {r.run_id}")
|
||||
raise typer.Exit(1)
|
||||
else:
|
||||
console.print(f"[red]Run not found: {run_id}[/red]")
|
||||
raise typer.Exit(1)
|
||||
|
||||
if run.state.is_terminal():
|
||||
console.print(f"[yellow]Run {run.run_id[:8]} is already {run.state.value}[/yellow]")
|
||||
raise typer.Exit(1)
|
||||
|
||||
# Update state
|
||||
persistence.update_run_state(run.run_id, RunState.CANCELLED)
|
||||
console.print(f"[green]Cancelled run: {run.run_id[:8]}[/green]")
|
||||
persistence.close()
|
||||
|
||||
@app.command("retry")
|
||||
def queue_retry(
|
||||
run_id: str = typer.Argument(..., help="Run ID to retry"),
|
||||
):
|
||||
"""Retry a failed run."""
|
||||
from ..queue import QueuePersistence, QueuedRun, RunState
|
||||
import uuid
|
||||
|
||||
persistence = QueuePersistence()
|
||||
persistence.initialize()
|
||||
|
||||
# Find the run
|
||||
run = persistence.load_run(run_id)
|
||||
if not run:
|
||||
runs = persistence.list_runs(limit=100)
|
||||
matches = [r for r in runs if r.run_id.startswith(run_id)]
|
||||
if len(matches) == 1:
|
||||
run = matches[0]
|
||||
else:
|
||||
console.print(f"[red]Run not found: {run_id}[/red]")
|
||||
raise typer.Exit(1)
|
||||
|
||||
if run.state != RunState.FAILED:
|
||||
console.print(f"[yellow]Can only retry failed runs (current: {run.state.value})[/yellow]")
|
||||
raise typer.Exit(1)
|
||||
|
||||
if run.retry_count >= run.max_retries:
|
||||
console.print(f"[yellow]Max retries reached ({run.retry_count}/{run.max_retries})[/yellow]")
|
||||
raise typer.Exit(1)
|
||||
|
||||
# Create new run
|
||||
new_run = QueuedRun(
|
||||
run_id=str(uuid.uuid4())[:8],
|
||||
agent_name=run.agent_name,
|
||||
input_content=run.input_content,
|
||||
state=RunState.QUEUED,
|
||||
priority=run.priority,
|
||||
session_id=run.session_id,
|
||||
workspace=run.workspace,
|
||||
retry_count=run.retry_count + 1,
|
||||
max_retries=run.max_retries,
|
||||
parent_run_id=run.run_id,
|
||||
config=run.config,
|
||||
)
|
||||
|
||||
persistence.save_run(new_run)
|
||||
console.print(f"[green]Created retry run: {new_run.run_id} (from {run.run_id[:8]})[/green]")
|
||||
persistence.close()
|
||||
|
||||
@app.command("clear")
|
||||
def queue_clear(
|
||||
force: bool = typer.Option(
|
||||
False, "--force", "-f",
|
||||
help="Skip confirmation"
|
||||
),
|
||||
):
|
||||
"""Clear all queued runs."""
|
||||
from ..queue import QueuePersistence, RunState
|
||||
|
||||
if not force:
|
||||
confirm = typer.confirm("Clear all queued runs?")
|
||||
if not confirm:
|
||||
raise typer.Abort()
|
||||
|
||||
persistence = QueuePersistence()
|
||||
persistence.initialize()
|
||||
|
||||
# Get queued runs
|
||||
runs = persistence.list_runs(state=RunState.QUEUED, limit=1000)
|
||||
|
||||
count = 0
|
||||
for run in runs:
|
||||
persistence.update_run_state(run.run_id, RunState.CANCELLED)
|
||||
count += 1
|
||||
|
||||
console.print(f"[green]Cleared {count} queued runs[/green]")
|
||||
persistence.close()
|
||||
|
||||
@app.command("stats")
|
||||
def queue_stats(
|
||||
session: Optional[str] = typer.Option(
|
||||
None, "--session",
|
||||
help="Filter by session ID"
|
||||
),
|
||||
):
|
||||
"""Show queue statistics."""
|
||||
from ..queue import QueuePersistence
|
||||
|
||||
persistence = QueuePersistence()
|
||||
persistence.initialize()
|
||||
|
||||
stats = persistence.get_stats(session)
|
||||
|
||||
console.print(Panel(
|
||||
f"""
|
||||
[bold]Queue Statistics[/bold]
|
||||
|
||||
Queued: {stats.queued_count}
|
||||
Running: {stats.running_count}
|
||||
Succeeded: {stats.succeeded_count}
|
||||
Failed: {stats.failed_count}
|
||||
Cancelled: {stats.cancelled_count}
|
||||
|
||||
Total Runs: {stats.total_runs}
|
||||
Avg Wait: {stats.avg_wait_seconds:.1f}s
|
||||
Avg Duration: {stats.avg_duration_seconds:.1f}s
|
||||
""",
|
||||
title="Queue Stats"
|
||||
))
|
||||
persistence.close()
|
||||
|
||||
return app
|
||||
|
||||
|
||||
def create_session_app():
|
||||
"""Create the session CLI app."""
|
||||
if not TYPER_AVAILABLE:
|
||||
return None
|
||||
|
||||
app = typer.Typer(
|
||||
name="session",
|
||||
help="Session management commands",
|
||||
no_args_is_help=True,
|
||||
)
|
||||
|
||||
@app.command("ls")
|
||||
def session_ls(
|
||||
limit: int = typer.Option(
|
||||
20, "--limit", "-n",
|
||||
help="Maximum number of results"
|
||||
),
|
||||
):
|
||||
"""List recent sessions."""
|
||||
from ..queue import QueuePersistence
|
||||
from datetime import datetime
|
||||
|
||||
persistence = QueuePersistence()
|
||||
persistence.initialize()
|
||||
|
||||
sessions = persistence.list_sessions(limit=limit)
|
||||
|
||||
if not sessions:
|
||||
console.print("[dim]No sessions found.[/dim]")
|
||||
return
|
||||
|
||||
table = Table(title="Sessions")
|
||||
table.add_column("ID", style="cyan")
|
||||
table.add_column("User")
|
||||
table.add_column("Created", style="dim")
|
||||
table.add_column("Updated", style="dim")
|
||||
|
||||
for session in sessions:
|
||||
created = datetime.fromtimestamp(session["created_at"]).strftime("%Y-%m-%d %H:%M")
|
||||
updated = datetime.fromtimestamp(session["updated_at"]).strftime("%Y-%m-%d %H:%M")
|
||||
|
||||
table.add_row(
|
||||
session["session_id"][:8],
|
||||
session.get("user_id", "-"),
|
||||
created,
|
||||
updated,
|
||||
)
|
||||
|
||||
console.print(table)
|
||||
persistence.close()
|
||||
|
||||
@app.command("resume")
|
||||
def session_resume(
|
||||
session_id: str = typer.Argument(..., help="Session ID to resume"),
|
||||
):
|
||||
"""Resume a session in TUI."""
|
||||
try:
|
||||
from .app import run_tui
|
||||
except ImportError:
|
||||
console.print(
|
||||
"[red]Textual is required for TUI. Install with:[/red]\n"
|
||||
" pip install praisonai[tui]"
|
||||
)
|
||||
raise typer.Exit(1)
|
||||
|
||||
run_tui(session_id=session_id)
|
||||
|
||||
return app
|
||||
|
||||
|
||||
def create_run_command():
|
||||
"""Create the run command."""
|
||||
if not TYPER_AVAILABLE:
|
||||
return None
|
||||
|
||||
def run_command(
|
||||
prompt: str = typer.Argument(..., help="Prompt to execute"),
|
||||
agent: Optional[str] = typer.Option(
|
||||
None, "--agent", "-a",
|
||||
help="Agent config file (YAML)"
|
||||
),
|
||||
model: Optional[str] = typer.Option(
|
||||
None, "--model", "-m",
|
||||
help="Model to use"
|
||||
),
|
||||
stream: bool = typer.Option(
|
||||
True, "--stream/--no-stream",
|
||||
help="Stream output"
|
||||
),
|
||||
priority: str = typer.Option(
|
||||
"normal", "--priority", "-p",
|
||||
help="Run priority (low, normal, high, urgent)"
|
||||
),
|
||||
):
|
||||
"""Run a single agent task."""
|
||||
from ..queue import QueueManager, QueueConfig, RunPriority
|
||||
|
||||
# Parse priority
|
||||
try:
|
||||
run_priority = RunPriority.from_string(priority)
|
||||
except (ValueError, KeyError):
|
||||
console.print(f"[red]Invalid priority: {priority}[/red]")
|
||||
raise typer.Exit(1)
|
||||
|
||||
# Load agent config
|
||||
agent_config = {}
|
||||
if agent:
|
||||
import yaml
|
||||
with open(agent, "r") as f:
|
||||
agent_config = yaml.safe_load(f)
|
||||
|
||||
# Set model
|
||||
if model:
|
||||
agent_config["model"] = model
|
||||
|
||||
async def execute():
|
||||
output_chunks = []
|
||||
|
||||
async def on_output(run_id: str, chunk: str):
|
||||
output_chunks.append(chunk)
|
||||
if stream:
|
||||
console.print(chunk, end="")
|
||||
|
||||
async def on_complete(run_id: str, run):
|
||||
if not stream:
|
||||
console.print(run.output_content or "".join(output_chunks))
|
||||
|
||||
async def on_error(run_id: str, error: Exception):
|
||||
console.print(f"\n[red]Error: {error}[/red]")
|
||||
|
||||
config = QueueConfig(enable_persistence=False)
|
||||
manager = QueueManager(
|
||||
config=config,
|
||||
on_output=on_output,
|
||||
on_complete=on_complete,
|
||||
on_error=on_error,
|
||||
)
|
||||
|
||||
await manager.start(recover=False)
|
||||
|
||||
try:
|
||||
run_id = await manager.submit(
|
||||
input_content=prompt,
|
||||
agent_name=agent_config.get("name", "Assistant"),
|
||||
priority=run_priority,
|
||||
config={"agent_config": agent_config},
|
||||
)
|
||||
|
||||
# Wait for completion
|
||||
while True:
|
||||
run = manager.get_run(run_id)
|
||||
if run and run.state.is_terminal():
|
||||
break
|
||||
await asyncio.sleep(0.1)
|
||||
|
||||
if stream:
|
||||
console.print() # Newline after streaming
|
||||
|
||||
finally:
|
||||
await manager.stop()
|
||||
|
||||
asyncio.run(execute())
|
||||
|
||||
return run_command
|
||||
|
||||
|
||||
# Main CLI integration
|
||||
def register_tui_commands(app):
|
||||
"""Register TUI commands with the main CLI app."""
|
||||
if not TYPER_AVAILABLE:
|
||||
return
|
||||
|
||||
tui_app = create_tui_app()
|
||||
queue_app = create_queue_app()
|
||||
session_app = create_session_app()
|
||||
run_cmd = create_run_command()
|
||||
|
||||
if tui_app:
|
||||
app.add_typer(tui_app, name="tui")
|
||||
if queue_app:
|
||||
app.add_typer(queue_app, name="queue")
|
||||
if session_app:
|
||||
app.add_typer(session_app, name="session")
|
||||
if run_cmd:
|
||||
app.command("run")(run_cmd)
|
||||
|
||||
|
||||
# Standalone entry point
|
||||
def main():
|
||||
"""Main entry point for TUI CLI."""
|
||||
if not TYPER_AVAILABLE:
|
||||
print("Typer is required. Install with: pip install typer")
|
||||
sys.exit(1)
|
||||
|
||||
app = typer.Typer(
|
||||
name="praison-tui",
|
||||
help="PraisonAI TUI and Queue CLI",
|
||||
)
|
||||
|
||||
register_tui_commands(app)
|
||||
|
||||
# Add direct tui command
|
||||
@app.callback(invoke_without_command=True)
|
||||
def default(
|
||||
ctx: typer.Context,
|
||||
workspace: Optional[str] = typer.Option(None, "--workspace", "-w"),
|
||||
session: Optional[str] = typer.Option(None, "--session", "-s"),
|
||||
model: Optional[str] = typer.Option(None, "--model", "-m"),
|
||||
):
|
||||
"""Launch TUI if no subcommand given."""
|
||||
if ctx.invoked_subcommand is None:
|
||||
try:
|
||||
from .app import run_tui
|
||||
run_tui(workspace=workspace, session_id=session, model=model)
|
||||
except ImportError:
|
||||
console.print(
|
||||
"[red]Textual is required for TUI. Install with:[/red]\n"
|
||||
" pip install praisonai[tui]"
|
||||
)
|
||||
raise typer.Exit(1)
|
||||
|
||||
app()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -0,0 +1,511 @@
|
||||
"""
|
||||
Debug and Simulation CLI Commands for PraisonAI TUI.
|
||||
|
||||
Provides headless TUI simulation, tracing, and debugging capabilities.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import json
|
||||
import sys
|
||||
import time
|
||||
from pathlib import Path
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
try:
|
||||
import typer
|
||||
from rich.console import Console
|
||||
from rich.table import Table
|
||||
from rich.panel import Panel
|
||||
from rich.live import Live
|
||||
from rich.text import Text
|
||||
TYPER_AVAILABLE = True
|
||||
except ImportError:
|
||||
TYPER_AVAILABLE = False
|
||||
typer = None
|
||||
|
||||
console = Console() if TYPER_AVAILABLE else None
|
||||
|
||||
|
||||
def create_debug_app():
|
||||
"""Create the debug CLI app with simulation commands."""
|
||||
if not TYPER_AVAILABLE:
|
||||
return None
|
||||
|
||||
app = typer.Typer(
|
||||
name="tui",
|
||||
help="TUI and simulation commands",
|
||||
no_args_is_help=True,
|
||||
)
|
||||
|
||||
@app.command("launch")
|
||||
def tui_launch(
|
||||
workspace: Optional[str] = typer.Option(None, "--workspace", "-w"),
|
||||
session: Optional[str] = typer.Option(None, "--session", "-s"),
|
||||
model: Optional[str] = typer.Option(None, "--model", "-m"),
|
||||
debug: bool = typer.Option(False, "--debug", "-d", help="Enable debug overlays"),
|
||||
log_jsonl: Optional[str] = typer.Option(None, "--log-jsonl", help="Write events to JSONL file"),
|
||||
profile: bool = typer.Option(False, "--profile", help="Enable performance profiling"),
|
||||
):
|
||||
"""Launch the interactive TUI."""
|
||||
try:
|
||||
from .app import run_tui
|
||||
except ImportError:
|
||||
console.print("[red]Textual is required for TUI. Install with:[/red]\n pip install praisonai[tui]")
|
||||
raise typer.Exit(1)
|
||||
|
||||
# Set debug environment
|
||||
if debug:
|
||||
import os
|
||||
os.environ["PRAISONAI_TUI_DEBUG"] = "1"
|
||||
|
||||
if log_jsonl:
|
||||
import os
|
||||
os.environ["PRAISONAI_TUI_JSONL"] = log_jsonl
|
||||
|
||||
run_tui(workspace=workspace, session_id=session, model=model)
|
||||
|
||||
@app.command("simulate")
|
||||
def tui_simulate(
|
||||
script: str = typer.Argument(..., help="Path to simulation script (YAML/JSON)"),
|
||||
mock: bool = typer.Option(True, "--mock/--real-llm", help="Use mock provider"),
|
||||
pretty: bool = typer.Option(True, "--pretty/--jsonl", help="Output format"),
|
||||
assert_mode: bool = typer.Option(False, "--assert", help="Validate expected outcomes"),
|
||||
timeout: float = typer.Option(60.0, "--timeout", help="Max execution time"),
|
||||
):
|
||||
"""Run a headless TUI simulation script."""
|
||||
import os
|
||||
|
||||
# Safety gate for real LLM
|
||||
if not mock:
|
||||
if not os.environ.get("PRAISONAI_REAL_LLM"):
|
||||
console.print("[red]Real LLM mode requires PRAISONAI_REAL_LLM=1 environment variable[/red]")
|
||||
raise typer.Exit(1)
|
||||
|
||||
# Load script
|
||||
script_path = Path(script)
|
||||
if not script_path.exists():
|
||||
console.print(f"[red]Script not found: {script}[/red]")
|
||||
raise typer.Exit(1)
|
||||
|
||||
with open(script_path) as f:
|
||||
if script_path.suffix in (".yaml", ".yml"):
|
||||
import yaml
|
||||
script_data = yaml.safe_load(f)
|
||||
else:
|
||||
script_data = json.load(f)
|
||||
|
||||
# Run simulation
|
||||
from .orchestrator import TuiOrchestrator, SimulationRunner, OutputMode
|
||||
from ..queue import QueueConfig
|
||||
|
||||
output_mode = OutputMode.PRETTY if pretty else OutputMode.JSONL
|
||||
|
||||
config = QueueConfig(enable_persistence=not mock)
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=output_mode,
|
||||
debug=True,
|
||||
)
|
||||
|
||||
runner = SimulationRunner(orchestrator, assert_mode=assert_mode)
|
||||
|
||||
async def run():
|
||||
return await asyncio.wait_for(
|
||||
runner.run_script(script_data),
|
||||
timeout=timeout
|
||||
)
|
||||
|
||||
try:
|
||||
success = asyncio.run(run())
|
||||
except asyncio.TimeoutError:
|
||||
console.print(f"[red]Simulation timed out after {timeout}s[/red]")
|
||||
raise typer.Exit(1)
|
||||
|
||||
# Print summary
|
||||
summary = runner.get_summary()
|
||||
if assert_mode:
|
||||
console.print(Panel(
|
||||
f"Passed: {summary['assertions_passed']}\n"
|
||||
f"Failed: {summary['assertions_failed']}\n"
|
||||
f"Errors: {len(summary['errors'])}",
|
||||
title="Simulation Results"
|
||||
))
|
||||
|
||||
if summary['errors']:
|
||||
for error in summary['errors']:
|
||||
console.print(f"[red] ✗ {error}[/red]")
|
||||
|
||||
if not success:
|
||||
raise typer.Exit(1)
|
||||
|
||||
@app.command("snapshot")
|
||||
def tui_snapshot(
|
||||
session: Optional[str] = typer.Option(None, "--session", "-s"),
|
||||
run_id: Optional[str] = typer.Option(None, "--run", "-r"),
|
||||
json_output: bool = typer.Option(False, "--json", "-j"),
|
||||
):
|
||||
"""Print a TUI-like snapshot of current state."""
|
||||
from ..queue import QueuePersistence, RunState
|
||||
|
||||
persistence = QueuePersistence()
|
||||
persistence.initialize()
|
||||
|
||||
# Build snapshot
|
||||
snapshot = {
|
||||
"timestamp": time.time(),
|
||||
"session_id": session,
|
||||
}
|
||||
|
||||
# Get runs
|
||||
runs = persistence.list_runs(session_id=session, limit=100)
|
||||
|
||||
queued = [r for r in runs if r.state == RunState.QUEUED]
|
||||
running = [r for r in runs if r.state == RunState.RUNNING]
|
||||
recent = [r for r in runs if r.state.is_terminal()][-5:]
|
||||
|
||||
snapshot["queued_count"] = len(queued)
|
||||
snapshot["running_count"] = len(running)
|
||||
snapshot["recent_runs"] = [
|
||||
{
|
||||
"run_id": r.run_id[:8],
|
||||
"agent": r.agent_name,
|
||||
"state": r.state.value,
|
||||
"input": r.input_content[:50],
|
||||
}
|
||||
for r in recent
|
||||
]
|
||||
|
||||
# Get specific run if requested
|
||||
if run_id:
|
||||
run = persistence.load_run(run_id)
|
||||
if run:
|
||||
snapshot["run"] = run.to_dict()
|
||||
|
||||
# Get session info
|
||||
if session:
|
||||
session_data = persistence.load_session(session)
|
||||
if session_data:
|
||||
snapshot["session"] = session_data
|
||||
|
||||
persistence.close()
|
||||
|
||||
if json_output:
|
||||
console.print(json.dumps(snapshot, indent=2, default=str))
|
||||
else:
|
||||
# Pretty print
|
||||
console.print("─" * 60)
|
||||
console.print(f"◉ PraisonAI Snapshot │ {time.strftime('%Y-%m-%d %H:%M:%S')}")
|
||||
console.print("─" * 60)
|
||||
|
||||
console.print(f"\n[bold]Queue Status[/bold]")
|
||||
console.print(f" Queued: {snapshot['queued_count']} │ Running: {snapshot['running_count']}")
|
||||
|
||||
if snapshot["recent_runs"]:
|
||||
console.print(f"\n[bold]Recent Runs[/bold]")
|
||||
for r in snapshot["recent_runs"]:
|
||||
state_color = {
|
||||
"succeeded": "green",
|
||||
"failed": "red",
|
||||
"cancelled": "dim",
|
||||
}.get(r["state"], "yellow")
|
||||
console.print(f" [{state_color}]{r['run_id']}[/{state_color}] {r['agent']}: {r['input'][:30]}...")
|
||||
|
||||
if "run" in snapshot:
|
||||
console.print(f"\n[bold]Run Details[/bold]")
|
||||
run = snapshot["run"]
|
||||
console.print(f" ID: {run.get('run_id', 'N/A')}")
|
||||
console.print(f" State: {run.get('state', 'N/A')}")
|
||||
console.print(f" Agent: {run.get('agent_name', 'N/A')}")
|
||||
|
||||
console.print("─" * 60)
|
||||
|
||||
@app.command("trace")
|
||||
def tui_trace(
|
||||
id: str = typer.Argument(..., help="Session or run ID to trace"),
|
||||
follow: bool = typer.Option(False, "--follow", "-f", help="Follow new events"),
|
||||
limit: int = typer.Option(50, "--limit", "-n", help="Max events to show"),
|
||||
):
|
||||
"""Replay events from persistence like a timeline."""
|
||||
from ..queue import QueuePersistence
|
||||
|
||||
persistence = QueuePersistence()
|
||||
persistence.initialize()
|
||||
|
||||
# Try to find as session or run
|
||||
session = persistence.load_session(id)
|
||||
run = persistence.load_run(id)
|
||||
|
||||
if not session and not run:
|
||||
# Try partial match
|
||||
runs = persistence.list_runs(limit=100)
|
||||
matches = [r for r in runs if r.run_id.startswith(id) or r.session_id.startswith(id)]
|
||||
if matches:
|
||||
run = matches[0]
|
||||
|
||||
if not session and not run:
|
||||
console.print(f"[red]No session or run found for: {id}[/red]")
|
||||
persistence.close()
|
||||
raise typer.Exit(1)
|
||||
|
||||
# Get runs for timeline
|
||||
if session:
|
||||
runs = persistence.list_runs(session_id=id, limit=limit)
|
||||
else:
|
||||
runs = [run] if run else []
|
||||
|
||||
# Build timeline
|
||||
events = []
|
||||
for r in runs:
|
||||
events.append({
|
||||
"timestamp": r.created_at,
|
||||
"type": "run_created",
|
||||
"run_id": r.run_id[:8],
|
||||
"agent": r.agent_name,
|
||||
"input": r.input_content[:50],
|
||||
})
|
||||
if r.started_at:
|
||||
events.append({
|
||||
"timestamp": r.started_at,
|
||||
"type": "run_started",
|
||||
"run_id": r.run_id[:8],
|
||||
})
|
||||
if r.ended_at:
|
||||
events.append({
|
||||
"timestamp": r.ended_at,
|
||||
"type": f"run_{r.state.value}",
|
||||
"run_id": r.run_id[:8],
|
||||
})
|
||||
|
||||
# Sort by timestamp
|
||||
events.sort(key=lambda e: e["timestamp"])
|
||||
|
||||
persistence.close()
|
||||
|
||||
# Print timeline
|
||||
console.print(f"\n[bold]Event Timeline for {id[:8]}[/bold]\n")
|
||||
|
||||
for event in events[-limit:]:
|
||||
ts = time.strftime("%H:%M:%S", time.localtime(event["timestamp"]))
|
||||
event_type = event["type"]
|
||||
|
||||
type_colors = {
|
||||
"run_created": "cyan",
|
||||
"run_started": "yellow",
|
||||
"run_succeeded": "green",
|
||||
"run_failed": "red",
|
||||
"run_cancelled": "dim",
|
||||
}
|
||||
color = type_colors.get(event_type, "white")
|
||||
|
||||
line = f"[dim]{ts}[/dim] [{color}]{event_type:15}[/{color}]"
|
||||
if "run_id" in event:
|
||||
line += f" run={event['run_id']}"
|
||||
if "agent" in event:
|
||||
line += f" agent={event['agent']}"
|
||||
if "input" in event:
|
||||
line += f" \"{event['input']}...\""
|
||||
|
||||
console.print(line)
|
||||
|
||||
if follow:
|
||||
console.print("\n[dim]Following... (Ctrl+C to stop)[/dim]")
|
||||
# In a real implementation, this would poll for new events
|
||||
|
||||
return app
|
||||
|
||||
|
||||
def create_queue_watch_command():
|
||||
"""Create the queue watch command."""
|
||||
if not TYPER_AVAILABLE:
|
||||
return None
|
||||
|
||||
def queue_watch(
|
||||
state: Optional[str] = typer.Option(None, "--state", "-s"),
|
||||
agent: Optional[str] = typer.Option(None, "--agent", "-a"),
|
||||
run_id: Optional[str] = typer.Option(None, "--run", "-r"),
|
||||
jsonl: bool = typer.Option(False, "--jsonl", "-j"),
|
||||
interval: float = typer.Option(1.0, "--interval", "-i"),
|
||||
):
|
||||
"""Watch queue events in real time."""
|
||||
from ..queue import QueuePersistence, RunState
|
||||
|
||||
persistence = QueuePersistence()
|
||||
persistence.initialize()
|
||||
|
||||
last_seen = {}
|
||||
|
||||
try:
|
||||
while True:
|
||||
runs = persistence.list_runs(limit=100)
|
||||
|
||||
# Filter
|
||||
if state:
|
||||
try:
|
||||
state_filter = RunState(state.lower())
|
||||
runs = [r for r in runs if r.state == state_filter]
|
||||
except ValueError:
|
||||
pass
|
||||
|
||||
if agent:
|
||||
runs = [r for r in runs if agent.lower() in r.agent_name.lower()]
|
||||
|
||||
if run_id:
|
||||
runs = [r for r in runs if r.run_id.startswith(run_id)]
|
||||
|
||||
# Check for changes
|
||||
for run in runs:
|
||||
key = run.run_id
|
||||
current_state = run.state.value
|
||||
|
||||
if key not in last_seen or last_seen[key] != current_state:
|
||||
last_seen[key] = current_state
|
||||
|
||||
if jsonl:
|
||||
event = {
|
||||
"timestamp": time.time(),
|
||||
"run_id": run.run_id,
|
||||
"agent": run.agent_name,
|
||||
"state": current_state,
|
||||
}
|
||||
print(json.dumps(event))
|
||||
else:
|
||||
ts = time.strftime("%H:%M:%S")
|
||||
state_colors = {
|
||||
"queued": "yellow",
|
||||
"running": "cyan",
|
||||
"succeeded": "green",
|
||||
"failed": "red",
|
||||
"cancelled": "dim",
|
||||
}
|
||||
color = state_colors.get(current_state, "white")
|
||||
console.print(
|
||||
f"[dim]{ts}[/dim] [{color}]{current_state:10}[/{color}] "
|
||||
f"{run.run_id[:8]} {run.agent_name}"
|
||||
)
|
||||
|
||||
time.sleep(interval)
|
||||
|
||||
except KeyboardInterrupt:
|
||||
console.print("\n[dim]Stopped watching[/dim]")
|
||||
finally:
|
||||
persistence.close()
|
||||
|
||||
return queue_watch
|
||||
|
||||
|
||||
def create_doctor_command():
|
||||
"""Create the doctor command for TUI diagnostics."""
|
||||
if not TYPER_AVAILABLE:
|
||||
return None
|
||||
|
||||
def doctor_tui():
|
||||
"""Validate TUI dependencies, DB schema, and wiring."""
|
||||
checks = []
|
||||
|
||||
# Check Textual
|
||||
try:
|
||||
import textual
|
||||
checks.append(("Textual", "✓", f"v{textual.__version__}", "green"))
|
||||
except ImportError:
|
||||
checks.append(("Textual", "✗", "Not installed", "red"))
|
||||
|
||||
# Check Rich
|
||||
try:
|
||||
import rich
|
||||
checks.append(("Rich", "✓", f"v{rich.__version__}", "green"))
|
||||
except ImportError:
|
||||
checks.append(("Rich", "✗", "Not installed", "red"))
|
||||
|
||||
# Check Typer
|
||||
try:
|
||||
import typer
|
||||
checks.append(("Typer", "✓", f"v{typer.__version__}", "green"))
|
||||
except ImportError:
|
||||
checks.append(("Typer", "✗", "Not installed", "red"))
|
||||
|
||||
# Check queue modules
|
||||
try:
|
||||
from ..queue import QueueManager, QueueScheduler, QueuePersistence
|
||||
checks.append(("Queue System", "✓", "All modules loaded", "green"))
|
||||
except ImportError as e:
|
||||
checks.append(("Queue System", "✗", str(e), "red"))
|
||||
|
||||
# Check TUI modules
|
||||
try:
|
||||
from .orchestrator import TuiOrchestrator
|
||||
checks.append(("TUI Orchestrator", "✓", "Loaded", "green"))
|
||||
except ImportError as e:
|
||||
checks.append(("TUI Orchestrator", "✗", str(e), "red"))
|
||||
|
||||
# Check events
|
||||
try:
|
||||
from .events import TUIEvent, TUIEventType
|
||||
event_count = len(TUIEventType)
|
||||
checks.append(("TUI Events", "✓", f"{event_count} event types", "green"))
|
||||
except ImportError as e:
|
||||
checks.append(("TUI Events", "✗", str(e), "red"))
|
||||
|
||||
# Check DB
|
||||
try:
|
||||
from ..queue import QueuePersistence
|
||||
p = QueuePersistence()
|
||||
p.initialize()
|
||||
stats = p.get_stats()
|
||||
p.close()
|
||||
checks.append(("Database", "✓", f"{stats.total_runs} runs in DB", "green"))
|
||||
except Exception as e:
|
||||
checks.append(("Database", "✗", str(e), "red"))
|
||||
|
||||
# Check slash commands
|
||||
try:
|
||||
from ..slash_commands import SlashCommandRegistry
|
||||
registry = SlashCommandRegistry()
|
||||
cmd_count = len(registry.commands)
|
||||
checks.append(("Slash Commands", "✓", f"{cmd_count} commands", "green"))
|
||||
except Exception as e:
|
||||
checks.append(("Slash Commands", "⚠", str(e), "yellow"))
|
||||
|
||||
# Print results
|
||||
console.print("\n[bold]PraisonAI TUI Diagnostics[/bold]\n")
|
||||
|
||||
table = Table(show_header=True)
|
||||
table.add_column("Component")
|
||||
table.add_column("Status")
|
||||
table.add_column("Details")
|
||||
|
||||
all_ok = True
|
||||
for name, status, details, color in checks:
|
||||
table.add_row(name, f"[{color}]{status}[/{color}]", details)
|
||||
if status == "✗":
|
||||
all_ok = False
|
||||
|
||||
console.print(table)
|
||||
|
||||
if all_ok:
|
||||
console.print("\n[green]All checks passed![/green]")
|
||||
else:
|
||||
console.print("\n[red]Some checks failed. Run 'pip install praisonai[tui]' to install missing dependencies.[/red]")
|
||||
raise typer.Exit(1)
|
||||
|
||||
return doctor_tui
|
||||
|
||||
|
||||
def register_debug_commands(app):
|
||||
"""Register debug commands with the main CLI app."""
|
||||
if not TYPER_AVAILABLE:
|
||||
return
|
||||
|
||||
debug_app = create_debug_app()
|
||||
watch_cmd = create_queue_watch_command()
|
||||
doctor_cmd = create_doctor_command()
|
||||
|
||||
if debug_app:
|
||||
app.add_typer(debug_app, name="tui")
|
||||
|
||||
if watch_cmd:
|
||||
# Add to queue subcommand
|
||||
pass # Will be added via queue app
|
||||
|
||||
if doctor_cmd:
|
||||
# Add doctor --tui option
|
||||
pass # Will be added via main app
|
||||
@@ -0,0 +1,99 @@
|
||||
"""
|
||||
Custom events for PraisonAI TUI.
|
||||
|
||||
Defines event types for communication between TUI components.
|
||||
"""
|
||||
|
||||
from dataclasses import dataclass, field
|
||||
from enum import Enum
|
||||
from typing import Any, Dict, Optional
|
||||
import time
|
||||
|
||||
|
||||
class TUIEventType(str, Enum):
|
||||
"""Types of TUI events."""
|
||||
# User input events
|
||||
MESSAGE_SUBMITTED = "message_submitted"
|
||||
COMMAND_EXECUTED = "command_executed"
|
||||
|
||||
# Queue events
|
||||
RUN_QUEUED = "run_queued"
|
||||
RUN_STARTED = "run_started"
|
||||
RUN_COMPLETED = "run_completed"
|
||||
RUN_FAILED = "run_failed"
|
||||
RUN_CANCELLED = "run_cancelled"
|
||||
|
||||
# Output events
|
||||
OUTPUT_CHUNK = "output_chunk"
|
||||
OUTPUT_COMPLETE = "output_complete"
|
||||
|
||||
# Tool events
|
||||
TOOL_CALL_STARTED = "tool_call_started"
|
||||
TOOL_CALL_COMPLETED = "tool_call_completed"
|
||||
TOOL_APPROVAL_REQUIRED = "tool_approval_required"
|
||||
|
||||
# Session events
|
||||
SESSION_STARTED = "session_started"
|
||||
SESSION_SAVED = "session_saved"
|
||||
SESSION_LOADED = "session_loaded"
|
||||
|
||||
# UI events
|
||||
SCREEN_CHANGED = "screen_changed"
|
||||
FOCUS_CHANGED = "focus_changed"
|
||||
ERROR_OCCURRED = "error_occurred"
|
||||
STATUS_UPDATED = "status_updated"
|
||||
|
||||
|
||||
@dataclass
|
||||
class TUIEvent:
|
||||
"""A TUI event."""
|
||||
event_type: TUIEventType
|
||||
timestamp: float = field(default_factory=time.time)
|
||||
data: Dict[str, Any] = field(default_factory=dict)
|
||||
|
||||
# Optional identifiers
|
||||
run_id: Optional[str] = None
|
||||
session_id: Optional[str] = None
|
||||
agent_name: Optional[str] = None
|
||||
|
||||
@classmethod
|
||||
def message_submitted(cls, content: str, **kwargs) -> "TUIEvent":
|
||||
"""Create a message submitted event."""
|
||||
return cls(
|
||||
event_type=TUIEventType.MESSAGE_SUBMITTED,
|
||||
data={"content": content, **kwargs}
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def output_chunk(cls, run_id: str, content: str, **kwargs) -> "TUIEvent":
|
||||
"""Create an output chunk event."""
|
||||
return cls(
|
||||
event_type=TUIEventType.OUTPUT_CHUNK,
|
||||
run_id=run_id,
|
||||
data={"content": content, **kwargs}
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def run_completed(cls, run_id: str, output: str, **kwargs) -> "TUIEvent":
|
||||
"""Create a run completed event."""
|
||||
return cls(
|
||||
event_type=TUIEventType.RUN_COMPLETED,
|
||||
run_id=run_id,
|
||||
data={"output": output, **kwargs}
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def error(cls, message: str, **kwargs) -> "TUIEvent":
|
||||
"""Create an error event."""
|
||||
return cls(
|
||||
event_type=TUIEventType.ERROR_OCCURRED,
|
||||
data={"message": message, **kwargs}
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def status_update(cls, status: str, **kwargs) -> "TUIEvent":
|
||||
"""Create a status update event."""
|
||||
return cls(
|
||||
event_type=TUIEventType.STATUS_UPDATED,
|
||||
data={"status": status, **kwargs}
|
||||
)
|
||||
@@ -0,0 +1,328 @@
|
||||
"""
|
||||
Mock Provider for PraisonAI TUI Testing.
|
||||
|
||||
Provides deterministic, replayable responses for CI and headless testing.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import hashlib
|
||||
import json
|
||||
import random
|
||||
import time
|
||||
from dataclasses import dataclass, field
|
||||
from pathlib import Path
|
||||
from typing import Any, AsyncIterator, Callable, Dict, List, Optional
|
||||
|
||||
|
||||
@dataclass
|
||||
class MockResponse:
|
||||
"""A mock response configuration."""
|
||||
content: str
|
||||
chunks: List[str] = field(default_factory=list)
|
||||
delay_per_chunk: float = 0.05
|
||||
tool_calls: List[Dict[str, Any]] = field(default_factory=list)
|
||||
error: Optional[str] = None
|
||||
tokens: int = 0
|
||||
cost: float = 0.0
|
||||
|
||||
def __post_init__(self):
|
||||
if not self.chunks and self.content:
|
||||
# Split content into word chunks
|
||||
words = self.content.split()
|
||||
self.chunks = []
|
||||
for i in range(0, len(words), 3):
|
||||
chunk = " ".join(words[i:i+3])
|
||||
if i > 0:
|
||||
chunk = " " + chunk
|
||||
self.chunks.append(chunk)
|
||||
|
||||
if not self.tokens:
|
||||
self.tokens = len(self.content.split()) * 2 # Rough estimate
|
||||
|
||||
|
||||
@dataclass
|
||||
class MockProviderConfig:
|
||||
"""Configuration for mock provider."""
|
||||
seed: int = 42
|
||||
default_delay: float = 0.05
|
||||
simulate_errors: bool = False
|
||||
error_rate: float = 0.1
|
||||
record_mode: bool = False
|
||||
playback_file: Optional[str] = None
|
||||
|
||||
# Canned responses by input pattern
|
||||
responses: Dict[str, MockResponse] = field(default_factory=dict)
|
||||
|
||||
# Default response if no pattern matches
|
||||
default_response: Optional[MockResponse] = None
|
||||
|
||||
|
||||
class MockProvider:
|
||||
"""
|
||||
Mock LLM provider for testing.
|
||||
|
||||
Features:
|
||||
- Deterministic responses based on seed
|
||||
- Streaming simulation
|
||||
- Tool call simulation
|
||||
- Error injection
|
||||
- Record/playback mode
|
||||
"""
|
||||
|
||||
def __init__(self, config: Optional[MockProviderConfig] = None):
|
||||
self.config = config or MockProviderConfig()
|
||||
self._rng = random.Random(self.config.seed)
|
||||
self._recordings: List[Dict[str, Any]] = []
|
||||
self._playback_index = 0
|
||||
|
||||
# Load playback file if specified
|
||||
if self.config.playback_file:
|
||||
self._load_playback()
|
||||
|
||||
# Setup default responses
|
||||
self._setup_defaults()
|
||||
|
||||
def _setup_defaults(self) -> None:
|
||||
"""Setup default canned responses."""
|
||||
defaults = {
|
||||
"hello": MockResponse(
|
||||
content="Hello! I'm a mock AI assistant. How can I help you today?",
|
||||
tokens=20,
|
||||
cost=0.0001,
|
||||
),
|
||||
"help": MockResponse(
|
||||
content="I can help you with various tasks. Just ask me anything!",
|
||||
tokens=15,
|
||||
cost=0.00008,
|
||||
),
|
||||
"test": MockResponse(
|
||||
content="This is a test response. Everything is working correctly.",
|
||||
tokens=12,
|
||||
cost=0.00006,
|
||||
),
|
||||
"error": MockResponse(
|
||||
content="",
|
||||
error="Simulated error for testing",
|
||||
),
|
||||
"tool": MockResponse(
|
||||
content="I'll use a tool to help with that.",
|
||||
tool_calls=[{
|
||||
"id": "call_mock_001",
|
||||
"name": "mock_tool",
|
||||
"arguments": {"query": "test"},
|
||||
}],
|
||||
tokens=25,
|
||||
cost=0.00012,
|
||||
),
|
||||
}
|
||||
|
||||
for pattern, response in defaults.items():
|
||||
if pattern not in self.config.responses:
|
||||
self.config.responses[pattern] = response
|
||||
|
||||
if not self.config.default_response:
|
||||
self.config.default_response = MockResponse(
|
||||
content="I understand your request. Here's a mock response for testing purposes. "
|
||||
"This response is generated deterministically based on the input.",
|
||||
tokens=30,
|
||||
cost=0.00015,
|
||||
)
|
||||
|
||||
def _load_playback(self) -> None:
|
||||
"""Load recordings for playback."""
|
||||
if not self.config.playback_file:
|
||||
return
|
||||
|
||||
path = Path(self.config.playback_file)
|
||||
if path.exists():
|
||||
with open(path) as f:
|
||||
for line in f:
|
||||
self._recordings.append(json.loads(line))
|
||||
|
||||
def _save_recording(self, input_content: str, response: MockResponse) -> None:
|
||||
"""Save a recording."""
|
||||
if not self.config.record_mode or not self.config.playback_file:
|
||||
return
|
||||
|
||||
record = {
|
||||
"timestamp": time.time(),
|
||||
"input": input_content,
|
||||
"response": {
|
||||
"content": response.content,
|
||||
"chunks": response.chunks,
|
||||
"tool_calls": response.tool_calls,
|
||||
"tokens": response.tokens,
|
||||
"cost": response.cost,
|
||||
}
|
||||
}
|
||||
|
||||
path = Path(self.config.playback_file)
|
||||
with open(path, "a") as f:
|
||||
f.write(json.dumps(record) + "\n")
|
||||
|
||||
def _get_response(self, input_content: str) -> MockResponse:
|
||||
"""Get response for input."""
|
||||
# Check playback mode
|
||||
if self._recordings and self._playback_index < len(self._recordings):
|
||||
record = self._recordings[self._playback_index]
|
||||
self._playback_index += 1
|
||||
return MockResponse(**record["response"])
|
||||
|
||||
# Check for pattern match
|
||||
input_lower = input_content.lower()
|
||||
for pattern, response in self.config.responses.items():
|
||||
if pattern in input_lower:
|
||||
return response
|
||||
|
||||
# Simulate random error
|
||||
if self.config.simulate_errors and self._rng.random() < self.config.error_rate:
|
||||
return MockResponse(content="", error="Random simulated error")
|
||||
|
||||
# Generate deterministic response based on input hash
|
||||
input_hash = hashlib.md5(input_content.encode()).hexdigest()
|
||||
seed = int(input_hash[:8], 16)
|
||||
rng = random.Random(seed)
|
||||
|
||||
# Generate varied response
|
||||
templates = [
|
||||
"I've processed your request about '{topic}'. Here's what I found...",
|
||||
"Regarding '{topic}', I can provide the following information...",
|
||||
"Thank you for asking about '{topic}'. Let me explain...",
|
||||
"Based on your query about '{topic}', here's my response...",
|
||||
]
|
||||
|
||||
topic = input_content[:30] if len(input_content) > 30 else input_content
|
||||
template = rng.choice(templates)
|
||||
content = template.format(topic=topic)
|
||||
|
||||
# Add some mock content
|
||||
content += " This is a deterministic mock response generated for testing. "
|
||||
content += f"Input hash: {input_hash[:8]}. "
|
||||
content += "The response is reproducible given the same input."
|
||||
|
||||
return MockResponse(
|
||||
content=content,
|
||||
tokens=len(content.split()) * 2,
|
||||
cost=len(content) * 0.000001,
|
||||
)
|
||||
|
||||
async def generate(
|
||||
self,
|
||||
input_content: str,
|
||||
model: str = "mock-model",
|
||||
stream: bool = True,
|
||||
on_chunk: Optional[Callable[[str], None]] = None,
|
||||
) -> Dict[str, Any]:
|
||||
"""
|
||||
Generate a mock response.
|
||||
|
||||
Args:
|
||||
input_content: The input prompt
|
||||
model: Model name (ignored for mock)
|
||||
stream: Whether to stream the response
|
||||
on_chunk: Callback for streaming chunks
|
||||
|
||||
Returns:
|
||||
Response dict with content, tokens, cost
|
||||
"""
|
||||
response = self._get_response(input_content)
|
||||
|
||||
# Record if in record mode
|
||||
if self.config.record_mode:
|
||||
self._save_recording(input_content, response)
|
||||
|
||||
# Check for error
|
||||
if response.error:
|
||||
raise Exception(response.error)
|
||||
|
||||
# Stream chunks
|
||||
if stream and on_chunk:
|
||||
for chunk in response.chunks:
|
||||
await asyncio.sleep(response.delay_per_chunk)
|
||||
on_chunk(chunk)
|
||||
else:
|
||||
# Non-streaming delay
|
||||
await asyncio.sleep(len(response.chunks) * response.delay_per_chunk)
|
||||
|
||||
return {
|
||||
"content": response.content,
|
||||
"tokens": response.tokens,
|
||||
"cost": response.cost,
|
||||
"tool_calls": response.tool_calls,
|
||||
"model": model,
|
||||
}
|
||||
|
||||
async def stream(
|
||||
self,
|
||||
input_content: str,
|
||||
model: str = "mock-model",
|
||||
) -> AsyncIterator[str]:
|
||||
"""
|
||||
Stream a mock response.
|
||||
|
||||
Yields chunks of the response.
|
||||
"""
|
||||
response = self._get_response(input_content)
|
||||
|
||||
if response.error:
|
||||
raise Exception(response.error)
|
||||
|
||||
for chunk in response.chunks:
|
||||
await asyncio.sleep(response.delay_per_chunk)
|
||||
yield chunk
|
||||
|
||||
def reset(self) -> None:
|
||||
"""Reset provider state."""
|
||||
self._rng = random.Random(self.config.seed)
|
||||
self._playback_index = 0
|
||||
|
||||
|
||||
class MockExecutor:
|
||||
"""
|
||||
Mock executor for queue worker testing.
|
||||
|
||||
Simulates agent execution with configurable behavior.
|
||||
"""
|
||||
|
||||
def __init__(self, provider: Optional[MockProvider] = None):
|
||||
self.provider = provider or MockProvider()
|
||||
|
||||
async def execute(
|
||||
self,
|
||||
input_content: str,
|
||||
agent_config: Dict[str, Any],
|
||||
on_output: Optional[Callable[[str], None]] = None,
|
||||
) -> Dict[str, Any]:
|
||||
"""Execute a mock agent run."""
|
||||
model = agent_config.get("model", "mock-model")
|
||||
|
||||
result = await self.provider.generate(
|
||||
input_content=input_content,
|
||||
model=model,
|
||||
stream=True,
|
||||
on_chunk=on_output,
|
||||
)
|
||||
|
||||
return {
|
||||
"output": result["content"],
|
||||
"metrics": {
|
||||
"tokens": result["tokens"],
|
||||
"cost": result["cost"],
|
||||
},
|
||||
"tool_calls": result.get("tool_calls", []),
|
||||
}
|
||||
|
||||
|
||||
# Convenience function for tests
|
||||
def create_mock_provider(
|
||||
seed: int = 42,
|
||||
responses: Optional[Dict[str, str]] = None,
|
||||
) -> MockProvider:
|
||||
"""Create a mock provider with optional custom responses."""
|
||||
config = MockProviderConfig(seed=seed)
|
||||
|
||||
if responses:
|
||||
for pattern, content in responses.items():
|
||||
config.responses[pattern] = MockResponse(content=content)
|
||||
|
||||
return MockProvider(config)
|
||||
@@ -0,0 +1,652 @@
|
||||
"""
|
||||
TUI Orchestrator for PraisonAI.
|
||||
|
||||
Provides unified event handling for both interactive TUI and headless simulation modes.
|
||||
Inspired by gemini-cli's event-driven architecture and codex-cli's state management.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import json
|
||||
import logging
|
||||
import time
|
||||
import uuid
|
||||
from dataclasses import dataclass, field
|
||||
from enum import Enum
|
||||
from typing import Any, Callable, Dict, List, Optional, TextIO
|
||||
import sys
|
||||
|
||||
from .events import TUIEvent, TUIEventType
|
||||
from ..queue import QueueManager, QueueConfig, QueuedRun, RunState
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class OutputMode(str, Enum):
|
||||
"""Output mode for headless simulation."""
|
||||
PRETTY = "pretty"
|
||||
JSONL = "jsonl"
|
||||
SILENT = "silent"
|
||||
|
||||
|
||||
@dataclass
|
||||
class UIStateModel:
|
||||
"""
|
||||
In-memory UI state model.
|
||||
|
||||
Mirrors the state that would be displayed in the TUI,
|
||||
enabling headless simulation and snapshot generation.
|
||||
"""
|
||||
# Session info
|
||||
session_id: str = ""
|
||||
workspace: str = ""
|
||||
|
||||
# Model/config
|
||||
model: str = "gpt-4o-mini"
|
||||
|
||||
# Chat history
|
||||
messages: List[Dict[str, Any]] = field(default_factory=list)
|
||||
max_messages: int = 1000
|
||||
|
||||
# Current streaming state
|
||||
current_run_id: Optional[str] = None
|
||||
streaming_content: str = ""
|
||||
is_processing: bool = False
|
||||
|
||||
# Queue state
|
||||
queued_runs: List[Dict[str, Any]] = field(default_factory=list)
|
||||
running_runs: List[Dict[str, Any]] = field(default_factory=list)
|
||||
|
||||
# Tool calls
|
||||
pending_tool_calls: List[Dict[str, Any]] = field(default_factory=list)
|
||||
recent_tool_calls: List[Dict[str, Any]] = field(default_factory=list)
|
||||
|
||||
# Metrics
|
||||
total_tokens: int = 0
|
||||
total_cost: float = 0.0
|
||||
|
||||
# Focus/screen state (for simulation)
|
||||
current_screen: str = "main"
|
||||
focused_widget: str = "composer"
|
||||
|
||||
# Events log (for trace/replay)
|
||||
events: List[Dict[str, Any]] = field(default_factory=list)
|
||||
max_events: int = 10000
|
||||
|
||||
def add_message(self, role: str, content: str, **kwargs) -> None:
|
||||
"""Add a message to history."""
|
||||
msg = {
|
||||
"role": role,
|
||||
"content": content,
|
||||
"timestamp": time.time(),
|
||||
**kwargs
|
||||
}
|
||||
self.messages.append(msg)
|
||||
if len(self.messages) > self.max_messages:
|
||||
self.messages = self.messages[-self.max_messages:]
|
||||
|
||||
def add_event(self, event: TUIEvent) -> None:
|
||||
"""Add an event to the log."""
|
||||
evt = {
|
||||
"type": event.event_type.value,
|
||||
"timestamp": event.timestamp,
|
||||
"run_id": event.run_id,
|
||||
"session_id": event.session_id,
|
||||
"agent_name": event.agent_name,
|
||||
"data": event.data,
|
||||
}
|
||||
self.events.append(evt)
|
||||
if len(self.events) > self.max_events:
|
||||
self.events = self.events[-self.max_events:]
|
||||
|
||||
def to_snapshot(self) -> Dict[str, Any]:
|
||||
"""Generate a snapshot of current state."""
|
||||
return {
|
||||
"session_id": self.session_id,
|
||||
"model": self.model,
|
||||
"current_screen": self.current_screen,
|
||||
"focused_widget": self.focused_widget,
|
||||
"is_processing": self.is_processing,
|
||||
"current_run_id": self.current_run_id,
|
||||
"streaming_content_length": len(self.streaming_content),
|
||||
"message_count": len(self.messages),
|
||||
"last_messages": self.messages[-5:] if self.messages else [],
|
||||
"queued_count": len(self.queued_runs),
|
||||
"running_count": len(self.running_runs),
|
||||
"pending_tool_calls": len(self.pending_tool_calls),
|
||||
"total_tokens": self.total_tokens,
|
||||
"total_cost": self.total_cost,
|
||||
}
|
||||
|
||||
def render_snapshot_pretty(self) -> str:
|
||||
"""Render a pretty text snapshot like TUI would show."""
|
||||
lines = []
|
||||
|
||||
# Status bar
|
||||
status = f"◉ PraisonAI │ Session: {self.session_id[:8] if self.session_id else 'new'}"
|
||||
status += f" │ Model: {self.model}"
|
||||
if self.total_tokens > 0:
|
||||
status += f" │ Tokens: {self.total_tokens:,}"
|
||||
if self.total_cost > 0:
|
||||
status += f" │ ${self.total_cost:.4f}"
|
||||
if self.is_processing:
|
||||
status += " │ ⟳ Processing..."
|
||||
lines.append("─" * 60)
|
||||
lines.append(status)
|
||||
lines.append("─" * 60)
|
||||
|
||||
# Chat messages (last 5)
|
||||
lines.append("\n[Chat History]")
|
||||
for msg in self.messages[-5:]:
|
||||
role = msg.get("role", "unknown")
|
||||
content = msg.get("content", "")[:100]
|
||||
if len(msg.get("content", "")) > 100:
|
||||
content += "..."
|
||||
lines.append(f" {role.upper()}: {content}")
|
||||
|
||||
# Streaming content
|
||||
if self.streaming_content:
|
||||
lines.append(f"\n[Streaming] ({len(self.streaming_content)} chars)")
|
||||
preview = self.streaming_content[-200:]
|
||||
if len(self.streaming_content) > 200:
|
||||
preview = "..." + preview
|
||||
lines.append(f" {preview}")
|
||||
|
||||
# Queue status
|
||||
lines.append(f"\n[Queue] Queued: {len(self.queued_runs)} │ Running: {len(self.running_runs)}")
|
||||
|
||||
# Tool calls
|
||||
if self.pending_tool_calls:
|
||||
lines.append(f"\n[Pending Approvals] {len(self.pending_tool_calls)}")
|
||||
for tc in self.pending_tool_calls[:3]:
|
||||
lines.append(f" ⚠ {tc.get('tool_name', 'unknown')}")
|
||||
|
||||
# Screen/focus
|
||||
lines.append(f"\n[UI] Screen: {self.current_screen} │ Focus: {self.focused_widget}")
|
||||
lines.append("─" * 60)
|
||||
|
||||
return "\n".join(lines)
|
||||
|
||||
|
||||
class TuiOrchestrator:
|
||||
"""
|
||||
Unified orchestrator for TUI and headless modes.
|
||||
|
||||
Subscribes to the event bus and maintains an in-memory UI state model.
|
||||
Can drive Textual widgets (interactive) or output snapshots (headless).
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
queue_manager: Optional[QueueManager] = None,
|
||||
queue_config: Optional[QueueConfig] = None,
|
||||
output_mode: OutputMode = OutputMode.PRETTY,
|
||||
output_stream: Optional[TextIO] = None,
|
||||
jsonl_path: Optional[str] = None,
|
||||
debug: bool = False,
|
||||
):
|
||||
self.queue_config = queue_config or QueueConfig()
|
||||
self.queue_manager = queue_manager
|
||||
self.output_mode = output_mode
|
||||
self.output_stream = output_stream or sys.stdout
|
||||
self.jsonl_path = jsonl_path
|
||||
self.debug = debug
|
||||
|
||||
# State
|
||||
self.state = UIStateModel()
|
||||
self._event_callbacks: List[Callable[[TUIEvent], None]] = []
|
||||
self._jsonl_file: Optional[TextIO] = None
|
||||
self._running = False
|
||||
|
||||
# Trace ID for this orchestrator session
|
||||
self.trace_id = str(uuid.uuid4())[:8]
|
||||
|
||||
async def start(self, session_id: Optional[str] = None, recover: bool = True) -> None:
|
||||
"""Start the orchestrator."""
|
||||
self.state.session_id = session_id or str(uuid.uuid4())[:8]
|
||||
self._running = True
|
||||
|
||||
# Open JSONL file if specified
|
||||
if self.jsonl_path:
|
||||
self._jsonl_file = open(self.jsonl_path, "a")
|
||||
|
||||
# Initialize queue manager if not provided
|
||||
if not self.queue_manager:
|
||||
self.queue_manager = QueueManager(
|
||||
config=self.queue_config,
|
||||
on_output=self._handle_output,
|
||||
on_complete=self._handle_complete,
|
||||
on_error=self._handle_error,
|
||||
)
|
||||
await self.queue_manager.start(recover=recover)
|
||||
self.queue_manager.set_session(self.state.session_id)
|
||||
|
||||
self._emit_event(TUIEvent(
|
||||
event_type=TUIEventType.SESSION_STARTED,
|
||||
session_id=self.state.session_id,
|
||||
data={"trace_id": self.trace_id}
|
||||
))
|
||||
|
||||
if self.debug:
|
||||
self._log_debug(f"Orchestrator started: session={self.state.session_id}, trace={self.trace_id}")
|
||||
|
||||
async def stop(self) -> None:
|
||||
"""Stop the orchestrator."""
|
||||
self._running = False
|
||||
|
||||
if self.queue_manager:
|
||||
await self.queue_manager.stop()
|
||||
|
||||
if self._jsonl_file:
|
||||
self._jsonl_file.close()
|
||||
self._jsonl_file = None
|
||||
|
||||
if self.debug:
|
||||
self._log_debug("Orchestrator stopped")
|
||||
|
||||
def add_event_callback(self, callback: Callable[[TUIEvent], None]) -> None:
|
||||
"""Add an event callback."""
|
||||
self._event_callbacks.append(callback)
|
||||
|
||||
def _emit_event(self, event: TUIEvent) -> None:
|
||||
"""Emit an event to all listeners."""
|
||||
# Add to state log
|
||||
self.state.add_event(event)
|
||||
|
||||
# Write to JSONL if enabled
|
||||
if self._jsonl_file:
|
||||
self._write_jsonl(event)
|
||||
|
||||
# Call callbacks
|
||||
for callback in self._event_callbacks:
|
||||
try:
|
||||
callback(event)
|
||||
except Exception as e:
|
||||
logger.error(f"Event callback error: {e}")
|
||||
|
||||
# Output based on mode
|
||||
if self.output_mode == OutputMode.JSONL:
|
||||
self._print_jsonl(event)
|
||||
elif self.output_mode == OutputMode.PRETTY and self.debug:
|
||||
self._print_event_pretty(event)
|
||||
|
||||
def _write_jsonl(self, event: TUIEvent) -> None:
|
||||
"""Write event to JSONL file."""
|
||||
if not self._jsonl_file:
|
||||
return
|
||||
|
||||
record = {
|
||||
"timestamp": event.timestamp,
|
||||
"trace_id": self.trace_id,
|
||||
"session_id": event.session_id or self.state.session_id,
|
||||
"event_type": event.event_type.value,
|
||||
"run_id": event.run_id,
|
||||
"agent_name": event.agent_name,
|
||||
"data": event.data,
|
||||
}
|
||||
self._jsonl_file.write(json.dumps(record, default=str) + "\n")
|
||||
self._jsonl_file.flush()
|
||||
|
||||
def _print_jsonl(self, event: TUIEvent) -> None:
|
||||
"""Print event as JSONL to output stream."""
|
||||
record = {
|
||||
"timestamp": event.timestamp,
|
||||
"event_type": event.event_type.value,
|
||||
"run_id": event.run_id,
|
||||
"data": event.data,
|
||||
}
|
||||
print(json.dumps(record, default=str), file=self.output_stream)
|
||||
|
||||
def _print_event_pretty(self, event: TUIEvent) -> None:
|
||||
"""Print event in pretty format."""
|
||||
ts = time.strftime("%H:%M:%S", time.localtime(event.timestamp))
|
||||
print(f"[{ts}] {event.event_type.value}", end="", file=self.output_stream)
|
||||
if event.run_id:
|
||||
print(f" run={event.run_id[:8]}", end="", file=self.output_stream)
|
||||
if event.data:
|
||||
data_preview = str(event.data)[:50]
|
||||
print(f" {data_preview}", end="", file=self.output_stream)
|
||||
print(file=self.output_stream)
|
||||
|
||||
def _log_debug(self, message: str) -> None:
|
||||
"""Log debug message."""
|
||||
if self.debug:
|
||||
ts = time.strftime("%H:%M:%S")
|
||||
print(f"[DEBUG {ts}] {message}", file=self.output_stream)
|
||||
|
||||
# Queue callbacks
|
||||
|
||||
async def _handle_output(self, run_id: str, chunk: str) -> None:
|
||||
"""Handle streaming output."""
|
||||
if run_id == self.state.current_run_id:
|
||||
self.state.streaming_content += chunk
|
||||
|
||||
self._emit_event(TUIEvent.output_chunk(run_id, chunk))
|
||||
|
||||
async def _handle_complete(self, run_id: str, run: QueuedRun) -> None:
|
||||
"""Handle run completion."""
|
||||
if run_id == self.state.current_run_id:
|
||||
# Add assistant message
|
||||
self.state.add_message(
|
||||
"assistant",
|
||||
run.output_content or self.state.streaming_content,
|
||||
run_id=run_id,
|
||||
agent_name=run.agent_name,
|
||||
)
|
||||
|
||||
# Update metrics
|
||||
if run.metrics:
|
||||
self.state.total_tokens += run.metrics.get("tokens", 0)
|
||||
self.state.total_cost += run.metrics.get("cost", 0.0)
|
||||
|
||||
# Clear streaming state
|
||||
self.state.current_run_id = None
|
||||
self.state.streaming_content = ""
|
||||
self.state.is_processing = False
|
||||
|
||||
# Update queue state
|
||||
self._update_queue_state()
|
||||
|
||||
self._emit_event(TUIEvent.run_completed(
|
||||
run_id,
|
||||
run.output_content or "",
|
||||
agent_name=run.agent_name,
|
||||
))
|
||||
|
||||
async def _handle_error(self, run_id: str, error: Exception) -> None:
|
||||
"""Handle run error."""
|
||||
if run_id == self.state.current_run_id:
|
||||
self.state.add_message(
|
||||
"system",
|
||||
f"Error: {error}",
|
||||
run_id=run_id,
|
||||
)
|
||||
self.state.current_run_id = None
|
||||
self.state.streaming_content = ""
|
||||
self.state.is_processing = False
|
||||
|
||||
self._update_queue_state()
|
||||
|
||||
self._emit_event(TUIEvent.error(str(error), run_id=run_id))
|
||||
|
||||
def _update_queue_state(self) -> None:
|
||||
"""Update queue state from manager."""
|
||||
if not self.queue_manager:
|
||||
return
|
||||
|
||||
runs = self.queue_manager.list_runs(limit=100)
|
||||
self.state.queued_runs = [
|
||||
r.to_dict() for r in runs if r.state == RunState.QUEUED
|
||||
]
|
||||
self.state.running_runs = [
|
||||
r.to_dict() for r in runs if r.state == RunState.RUNNING
|
||||
]
|
||||
|
||||
# Public API
|
||||
|
||||
async def submit_message(self, content: str, agent_name: str = "Assistant") -> str:
|
||||
"""Submit a message for processing."""
|
||||
# Add user message
|
||||
self.state.add_message("user", content)
|
||||
self.state.is_processing = True
|
||||
|
||||
self._emit_event(TUIEvent.message_submitted(content))
|
||||
|
||||
# Submit to queue
|
||||
run_id = await self.queue_manager.submit(
|
||||
input_content=content,
|
||||
agent_name=agent_name,
|
||||
config={"agent_config": {"name": agent_name, "model": self.state.model}}
|
||||
)
|
||||
|
||||
self.state.current_run_id = run_id
|
||||
self.state.streaming_content = ""
|
||||
|
||||
# Add placeholder for streaming
|
||||
self.state.add_message(
|
||||
"assistant",
|
||||
"",
|
||||
run_id=run_id,
|
||||
agent_name=agent_name,
|
||||
is_streaming=True,
|
||||
)
|
||||
|
||||
self._update_queue_state()
|
||||
|
||||
return run_id
|
||||
|
||||
async def cancel_run(self, run_id: Optional[str] = None) -> bool:
|
||||
"""Cancel a run."""
|
||||
target_id = run_id or self.state.current_run_id
|
||||
if not target_id:
|
||||
return False
|
||||
|
||||
result = await self.queue_manager.cancel(target_id)
|
||||
|
||||
if target_id == self.state.current_run_id:
|
||||
self.state.current_run_id = None
|
||||
self.state.streaming_content = ""
|
||||
self.state.is_processing = False
|
||||
|
||||
self._update_queue_state()
|
||||
|
||||
self._emit_event(TUIEvent(
|
||||
event_type=TUIEventType.RUN_CANCELLED,
|
||||
run_id=target_id,
|
||||
))
|
||||
|
||||
return result
|
||||
|
||||
async def retry_run(self, run_id: str) -> Optional[str]:
|
||||
"""Retry a failed run."""
|
||||
new_id = await self.queue_manager.retry(run_id)
|
||||
self._update_queue_state()
|
||||
return new_id
|
||||
|
||||
def set_model(self, model: str) -> None:
|
||||
"""Set the current model."""
|
||||
self.state.model = model
|
||||
self._emit_event(TUIEvent.status_update(f"Model set to {model}"))
|
||||
|
||||
def navigate_screen(self, screen: str) -> None:
|
||||
"""Navigate to a screen (for simulation)."""
|
||||
self.state.current_screen = screen
|
||||
self._emit_event(TUIEvent(
|
||||
event_type=TUIEventType.SCREEN_CHANGED,
|
||||
data={"screen": screen}
|
||||
))
|
||||
|
||||
def set_focus(self, widget: str) -> None:
|
||||
"""Set focus to a widget (for simulation)."""
|
||||
self.state.focused_widget = widget
|
||||
self._emit_event(TUIEvent(
|
||||
event_type=TUIEventType.FOCUS_CHANGED,
|
||||
data={"widget": widget}
|
||||
))
|
||||
|
||||
def get_snapshot(self) -> Dict[str, Any]:
|
||||
"""Get current state snapshot."""
|
||||
return self.state.to_snapshot()
|
||||
|
||||
def render_snapshot(self) -> str:
|
||||
"""Render pretty snapshot."""
|
||||
return self.state.render_snapshot_pretty()
|
||||
|
||||
async def wait_for_idle(self, timeout: float = 60.0) -> bool:
|
||||
"""Wait until no runs are processing."""
|
||||
start = time.time()
|
||||
while time.time() - start < timeout:
|
||||
if not self.state.is_processing and not self.state.running_runs:
|
||||
return True
|
||||
await asyncio.sleep(0.1)
|
||||
return False
|
||||
|
||||
async def wait_for_run(self, run_id: str, timeout: float = 60.0) -> bool:
|
||||
"""Wait for a specific run to complete."""
|
||||
start = time.time()
|
||||
while time.time() - start < timeout:
|
||||
run = self.queue_manager.get_run(run_id)
|
||||
if run and run.state.is_terminal():
|
||||
return True
|
||||
await asyncio.sleep(0.1)
|
||||
return False
|
||||
|
||||
|
||||
@dataclass
|
||||
class SimulationStep:
|
||||
"""A step in a simulation script."""
|
||||
action: str # "submit", "cancel", "retry", "navigate", "focus", "wait", "approve", "deny"
|
||||
args: Dict[str, Any] = field(default_factory=dict)
|
||||
expected: Optional[Dict[str, Any]] = None # For assertion mode
|
||||
|
||||
|
||||
class SimulationRunner:
|
||||
"""
|
||||
Runs simulation scripts for headless TUI testing.
|
||||
|
||||
Script format (YAML):
|
||||
```yaml
|
||||
session_id: test-session
|
||||
model: gpt-4o-mini
|
||||
steps:
|
||||
- action: submit
|
||||
args:
|
||||
content: "Hello, world!"
|
||||
- action: wait
|
||||
args:
|
||||
condition: idle
|
||||
timeout: 30
|
||||
- action: navigate
|
||||
args:
|
||||
screen: queue
|
||||
- action: cancel
|
||||
args:
|
||||
run_id: current
|
||||
```
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
orchestrator: TuiOrchestrator,
|
||||
assert_mode: bool = False,
|
||||
):
|
||||
self.orchestrator = orchestrator
|
||||
self.assert_mode = assert_mode
|
||||
self.assertions_passed = 0
|
||||
self.assertions_failed = 0
|
||||
self.errors: List[str] = []
|
||||
|
||||
async def run_script(self, script: Dict[str, Any]) -> bool:
|
||||
"""Run a simulation script."""
|
||||
# Initialize session
|
||||
session_id = script.get("session_id")
|
||||
await self.orchestrator.start(session_id=session_id)
|
||||
|
||||
if "model" in script:
|
||||
self.orchestrator.set_model(script["model"])
|
||||
|
||||
# Run steps
|
||||
steps = script.get("steps", [])
|
||||
for i, step_data in enumerate(steps):
|
||||
step = SimulationStep(
|
||||
action=step_data.get("action", ""),
|
||||
args=step_data.get("args", {}),
|
||||
expected=step_data.get("expected"),
|
||||
)
|
||||
|
||||
try:
|
||||
await self._run_step(step, i)
|
||||
except Exception as e:
|
||||
self.errors.append(f"Step {i} ({step.action}): {e}")
|
||||
if self.assert_mode:
|
||||
break
|
||||
|
||||
await self.orchestrator.stop()
|
||||
|
||||
return len(self.errors) == 0
|
||||
|
||||
async def _run_step(self, step: SimulationStep, index: int) -> None:
|
||||
"""Run a single simulation step."""
|
||||
action = step.action.lower()
|
||||
args = step.args
|
||||
|
||||
if action == "submit":
|
||||
content = args.get("content", "")
|
||||
agent = args.get("agent", "Assistant")
|
||||
await self.orchestrator.submit_message(content, agent)
|
||||
|
||||
elif action == "cancel":
|
||||
run_id = args.get("run_id")
|
||||
if run_id == "current":
|
||||
run_id = None
|
||||
await self.orchestrator.cancel_run(run_id)
|
||||
|
||||
elif action == "retry":
|
||||
run_id = args.get("run_id", "")
|
||||
await self.orchestrator.retry_run(run_id)
|
||||
|
||||
elif action == "navigate":
|
||||
screen = args.get("screen", "main")
|
||||
self.orchestrator.navigate_screen(screen)
|
||||
|
||||
elif action == "focus":
|
||||
widget = args.get("widget", "composer")
|
||||
self.orchestrator.set_focus(widget)
|
||||
|
||||
elif action == "wait":
|
||||
condition = args.get("condition", "idle")
|
||||
timeout = args.get("timeout", 30.0)
|
||||
|
||||
if condition == "idle":
|
||||
success = await self.orchestrator.wait_for_idle(timeout)
|
||||
elif condition == "run":
|
||||
run_id = args.get("run_id", "")
|
||||
success = await self.orchestrator.wait_for_run(run_id, timeout)
|
||||
else:
|
||||
await asyncio.sleep(timeout)
|
||||
success = True
|
||||
|
||||
if not success and self.assert_mode:
|
||||
raise TimeoutError(f"Wait condition '{condition}' timed out")
|
||||
|
||||
elif action == "model":
|
||||
model = args.get("model", "gpt-4o-mini")
|
||||
self.orchestrator.set_model(model)
|
||||
|
||||
elif action == "snapshot":
|
||||
snapshot = self.orchestrator.render_snapshot()
|
||||
print(snapshot)
|
||||
|
||||
elif action == "sleep":
|
||||
duration = args.get("seconds", 1.0)
|
||||
await asyncio.sleep(duration)
|
||||
|
||||
else:
|
||||
raise ValueError(f"Unknown action: {action}")
|
||||
|
||||
# Check assertions
|
||||
if self.assert_mode and step.expected:
|
||||
self._check_assertions(step.expected, index)
|
||||
|
||||
def _check_assertions(self, expected: Dict[str, Any], step_index: int) -> None:
|
||||
"""Check assertions against current state."""
|
||||
snapshot = self.orchestrator.get_snapshot()
|
||||
|
||||
for key, expected_value in expected.items():
|
||||
actual_value = snapshot.get(key)
|
||||
|
||||
if actual_value != expected_value:
|
||||
self.assertions_failed += 1
|
||||
self.errors.append(
|
||||
f"Step {step_index}: Expected {key}={expected_value}, got {actual_value}"
|
||||
)
|
||||
else:
|
||||
self.assertions_passed += 1
|
||||
|
||||
def get_summary(self) -> Dict[str, Any]:
|
||||
"""Get simulation summary."""
|
||||
return {
|
||||
"assertions_passed": self.assertions_passed,
|
||||
"assertions_failed": self.assertions_failed,
|
||||
"errors": self.errors,
|
||||
"success": len(self.errors) == 0,
|
||||
}
|
||||
@@ -0,0 +1,50 @@
|
||||
"""
|
||||
TUI Screens for PraisonAI.
|
||||
|
||||
Screen definitions for different views in the TUI application.
|
||||
"""
|
||||
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from .main import MainScreen
|
||||
from .queue import QueueScreen
|
||||
from .settings import SettingsScreen
|
||||
from .session import SessionScreen
|
||||
|
||||
_lazy_cache = {}
|
||||
|
||||
|
||||
def __getattr__(name: str):
|
||||
"""Lazy load screens."""
|
||||
global _lazy_cache
|
||||
|
||||
if name in _lazy_cache:
|
||||
return _lazy_cache[name]
|
||||
|
||||
if name == "MainScreen":
|
||||
from .main import MainScreen
|
||||
_lazy_cache[name] = MainScreen
|
||||
return MainScreen
|
||||
elif name == "QueueScreen":
|
||||
from .queue import QueueScreen
|
||||
_lazy_cache[name] = QueueScreen
|
||||
return QueueScreen
|
||||
elif name == "SettingsScreen":
|
||||
from .settings import SettingsScreen
|
||||
_lazy_cache[name] = SettingsScreen
|
||||
return SettingsScreen
|
||||
elif name == "SessionScreen":
|
||||
from .session import SessionScreen
|
||||
_lazy_cache[name] = SessionScreen
|
||||
return SessionScreen
|
||||
|
||||
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|
||||
|
||||
|
||||
__all__ = [
|
||||
"MainScreen",
|
||||
"QueueScreen",
|
||||
"SettingsScreen",
|
||||
"SessionScreen",
|
||||
]
|
||||
@@ -0,0 +1,245 @@
|
||||
"""
|
||||
Main Screen for PraisonAI TUI.
|
||||
|
||||
The primary chat interface screen.
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
try:
|
||||
from textual.screen import Screen
|
||||
from textual.containers import Horizontal, Vertical, Container
|
||||
from textual.widgets import Static, Footer, Header
|
||||
from textual.binding import Binding
|
||||
from textual.message import Message
|
||||
TEXTUAL_AVAILABLE = True
|
||||
except ImportError:
|
||||
TEXTUAL_AVAILABLE = False
|
||||
Screen = object
|
||||
Message = object
|
||||
|
||||
|
||||
if TEXTUAL_AVAILABLE:
|
||||
from ..widgets.chat import ChatWidget, ChatMessage
|
||||
from ..widgets.composer import ComposerWidget
|
||||
from ..widgets.status import StatusWidget, StatusInfo
|
||||
from ..widgets.queue_panel import QueuePanelWidget
|
||||
from ..widgets.tool_panel import ToolPanelWidget
|
||||
|
||||
class MainScreen(Screen):
|
||||
"""
|
||||
Main chat screen.
|
||||
|
||||
Layout:
|
||||
- Status bar (top)
|
||||
- Chat history (main area)
|
||||
- Tools panel (right sidebar)
|
||||
- Queue panel (right sidebar)
|
||||
- Composer (bottom)
|
||||
- Footer with keybindings
|
||||
"""
|
||||
|
||||
BINDINGS = [
|
||||
Binding("ctrl+q", "quit", "Quit", show=True),
|
||||
Binding("ctrl+c", "cancel", "Cancel", show=True),
|
||||
Binding("ctrl+l", "clear", "Clear", show=True),
|
||||
Binding("f1", "help", "Help", show=True),
|
||||
Binding("f2", "toggle_queue", "Queue", show=True),
|
||||
Binding("f3", "settings", "Settings", show=True),
|
||||
Binding("f5", "clear_chat", "Clear Chat", show=True),
|
||||
Binding("tab", "focus_next", "Next", show=False),
|
||||
Binding("escape", "cancel_input", "Cancel", show=False),
|
||||
]
|
||||
|
||||
DEFAULT_CSS = """
|
||||
MainScreen {
|
||||
layout: grid;
|
||||
grid-size: 2;
|
||||
grid-columns: 3fr 1fr;
|
||||
grid-rows: 1 1fr auto 1;
|
||||
}
|
||||
|
||||
MainScreen #status-bar {
|
||||
column-span: 2;
|
||||
}
|
||||
|
||||
MainScreen #main-area {
|
||||
height: 100%;
|
||||
}
|
||||
|
||||
MainScreen #sidebar {
|
||||
height: 100%;
|
||||
}
|
||||
|
||||
MainScreen #composer-area {
|
||||
column-span: 2;
|
||||
height: auto;
|
||||
max-height: 10;
|
||||
}
|
||||
|
||||
MainScreen Footer {
|
||||
column-span: 2;
|
||||
}
|
||||
|
||||
MainScreen .sidebar-panel {
|
||||
height: 1fr;
|
||||
}
|
||||
"""
|
||||
|
||||
class MessageSubmitted(Message):
|
||||
"""Event when a message is submitted."""
|
||||
def __init__(self, content: str):
|
||||
self.content = content
|
||||
super().__init__()
|
||||
|
||||
class CommandExecuted(Message):
|
||||
"""Event when a command is executed."""
|
||||
def __init__(self, command: str, args: str):
|
||||
self.command = command
|
||||
self.args = args
|
||||
super().__init__()
|
||||
|
||||
class CancelRequested(Message):
|
||||
"""Event when cancel is requested."""
|
||||
pass
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
show_queue: bool = True,
|
||||
show_tools: bool = True,
|
||||
name: Optional[str] = None,
|
||||
id: Optional[str] = None,
|
||||
classes: Optional[str] = None,
|
||||
):
|
||||
super().__init__(name=name, id=id, classes=classes)
|
||||
self._show_queue = show_queue
|
||||
self._show_tools = show_tools
|
||||
|
||||
def compose(self):
|
||||
"""Compose the screen."""
|
||||
yield StatusWidget(id="status-bar")
|
||||
|
||||
yield ChatWidget(id="chat-widget")
|
||||
|
||||
with Vertical(id="sidebar"):
|
||||
if self._show_tools:
|
||||
yield ToolPanelWidget(id="tool-panel", classes="sidebar-panel")
|
||||
if self._show_queue:
|
||||
yield QueuePanelWidget(id="queue-panel", classes="sidebar-panel")
|
||||
|
||||
yield ComposerWidget(id="composer")
|
||||
|
||||
yield Footer()
|
||||
|
||||
def on_mount(self) -> None:
|
||||
"""Handle mount."""
|
||||
# Focus composer
|
||||
composer = self.query_one("#composer", ComposerWidget)
|
||||
composer.focus_input()
|
||||
|
||||
def on_composer_widget_submitted(self, event: ComposerWidget.Submitted) -> None:
|
||||
"""Handle message submission."""
|
||||
if event.is_command:
|
||||
# Parse command
|
||||
parts = event.content[1:].split(maxsplit=1)
|
||||
command = parts[0] if parts else ""
|
||||
args = parts[1] if len(parts) > 1 else ""
|
||||
self.post_message(self.CommandExecuted(command, args))
|
||||
else:
|
||||
self.post_message(self.MessageSubmitted(event.content))
|
||||
|
||||
def action_quit(self) -> None:
|
||||
"""Quit the application."""
|
||||
self.app.exit()
|
||||
|
||||
def action_cancel(self) -> None:
|
||||
"""Cancel current operation."""
|
||||
self.post_message(self.CancelRequested())
|
||||
|
||||
def action_clear(self) -> None:
|
||||
"""Clear screen."""
|
||||
pass
|
||||
|
||||
def action_help(self) -> None:
|
||||
"""Show help."""
|
||||
self.app.push_screen("help")
|
||||
|
||||
def action_toggle_queue(self) -> None:
|
||||
"""Toggle queue panel visibility."""
|
||||
try:
|
||||
queue_panel = self.query_one("#queue-panel", QueuePanelWidget)
|
||||
queue_panel.display = not queue_panel.display
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
def action_settings(self) -> None:
|
||||
"""Show settings."""
|
||||
self.app.push_screen("settings")
|
||||
|
||||
def action_clear_chat(self) -> None:
|
||||
"""Clear chat history."""
|
||||
chat = self.query_one("#chat-widget", ChatWidget)
|
||||
self.app.call_later(chat.clear)
|
||||
|
||||
def action_cancel_input(self) -> None:
|
||||
"""Cancel current input."""
|
||||
composer = self.query_one("#composer", ComposerWidget)
|
||||
composer.action_cancel()
|
||||
|
||||
# Public methods for updating UI
|
||||
|
||||
async def add_user_message(self, content: str) -> None:
|
||||
"""Add a user message to chat."""
|
||||
chat = self.query_one("#chat-widget", ChatWidget)
|
||||
await chat.add_message(ChatMessage(
|
||||
role="user",
|
||||
content=content,
|
||||
))
|
||||
|
||||
async def add_assistant_message(
|
||||
self,
|
||||
content: str,
|
||||
run_id: Optional[str] = None,
|
||||
agent_name: Optional[str] = None,
|
||||
streaming: bool = False,
|
||||
) -> None:
|
||||
"""Add an assistant message to chat."""
|
||||
chat = self.query_one("#chat-widget", ChatWidget)
|
||||
await chat.add_message(ChatMessage(
|
||||
role="assistant",
|
||||
content=content,
|
||||
run_id=run_id,
|
||||
agent_name=agent_name,
|
||||
is_streaming=streaming,
|
||||
))
|
||||
|
||||
async def update_streaming(self, run_id: str, content: str) -> None:
|
||||
"""Update streaming message."""
|
||||
chat = self.query_one("#chat-widget", ChatWidget)
|
||||
await chat.update_streaming(run_id, content)
|
||||
|
||||
async def complete_streaming(self, run_id: str, content: str) -> None:
|
||||
"""Complete streaming message."""
|
||||
chat = self.query_one("#chat-widget", ChatWidget)
|
||||
await chat.complete_streaming(run_id, content)
|
||||
|
||||
def update_status(self, info: StatusInfo) -> None:
|
||||
"""Update status bar."""
|
||||
status = self.query_one("#status-bar", StatusWidget)
|
||||
status.update_info(info)
|
||||
|
||||
def set_processing(self, processing: bool) -> None:
|
||||
"""Set processing state."""
|
||||
composer = self.query_one("#composer", ComposerWidget)
|
||||
composer.set_processing(processing)
|
||||
|
||||
status = self.query_one("#status-bar", StatusWidget)
|
||||
status.is_processing = processing
|
||||
|
||||
else:
|
||||
class MainScreen:
|
||||
"""Placeholder when Textual is not available."""
|
||||
def __init__(self, *args, **kwargs):
|
||||
raise ImportError(
|
||||
"Textual is required for TUI. Install with: pip install praisonai[tui]"
|
||||
)
|
||||
@@ -0,0 +1,174 @@
|
||||
"""
|
||||
Queue Screen for PraisonAI TUI.
|
||||
|
||||
Queue management interface.
|
||||
"""
|
||||
|
||||
from typing import Optional, List
|
||||
|
||||
try:
|
||||
from textual.screen import Screen
|
||||
from textual.containers import Vertical, Horizontal
|
||||
from textual.widgets import Static, Footer, DataTable, Button
|
||||
from textual.binding import Binding
|
||||
from textual.message import Message
|
||||
from rich.text import Text
|
||||
TEXTUAL_AVAILABLE = True
|
||||
except ImportError:
|
||||
TEXTUAL_AVAILABLE = False
|
||||
Screen = object
|
||||
Message = object
|
||||
|
||||
|
||||
if TEXTUAL_AVAILABLE:
|
||||
class QueueScreen(Screen):
|
||||
"""
|
||||
Queue management screen.
|
||||
|
||||
Displays full queue with management options.
|
||||
"""
|
||||
|
||||
BINDINGS = [
|
||||
Binding("escape", "back", "Back", show=True),
|
||||
Binding("c", "cancel", "Cancel", show=True),
|
||||
Binding("r", "retry", "Retry", show=True),
|
||||
Binding("x", "clear", "Clear Queue", show=True),
|
||||
Binding("enter", "select", "Select", show=False),
|
||||
]
|
||||
|
||||
DEFAULT_CSS = """
|
||||
QueueScreen {
|
||||
background: $surface;
|
||||
}
|
||||
|
||||
QueueScreen #queue-header {
|
||||
height: 3;
|
||||
background: $primary;
|
||||
padding: 1;
|
||||
}
|
||||
|
||||
QueueScreen DataTable {
|
||||
height: 1fr;
|
||||
}
|
||||
|
||||
QueueScreen #queue-actions {
|
||||
height: 3;
|
||||
padding: 1;
|
||||
}
|
||||
"""
|
||||
|
||||
class RunCancelled(Message):
|
||||
def __init__(self, run_id: str):
|
||||
self.run_id = run_id
|
||||
super().__init__()
|
||||
|
||||
class RunRetried(Message):
|
||||
def __init__(self, run_id: str):
|
||||
self.run_id = run_id
|
||||
super().__init__()
|
||||
|
||||
class QueueCleared(Message):
|
||||
pass
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
name: Optional[str] = None,
|
||||
id: Optional[str] = None,
|
||||
classes: Optional[str] = None,
|
||||
):
|
||||
super().__init__(name=name, id=id, classes=classes)
|
||||
self._selected_run_id: Optional[str] = None
|
||||
|
||||
def compose(self):
|
||||
"""Compose the screen."""
|
||||
yield Static("Queue Management", id="queue-header")
|
||||
yield DataTable(id="queue-table")
|
||||
with Horizontal(id="queue-actions"):
|
||||
yield Button("Cancel [C]", id="btn-cancel", variant="error")
|
||||
yield Button("Retry [R]", id="btn-retry", variant="warning")
|
||||
yield Button("Clear [X]", id="btn-clear", variant="default")
|
||||
yield Button("Back [Esc]", id="btn-back", variant="primary")
|
||||
yield Footer()
|
||||
|
||||
def on_mount(self) -> None:
|
||||
"""Handle mount."""
|
||||
table = self.query_one("#queue-table", DataTable)
|
||||
table.add_columns(
|
||||
"ID", "Agent", "Input", "State", "Priority",
|
||||
"Wait Time", "Duration"
|
||||
)
|
||||
table.cursor_type = "row"
|
||||
|
||||
def update_runs(self, runs: List[dict]) -> None:
|
||||
"""Update the runs display."""
|
||||
table = self.query_one("#queue-table", DataTable)
|
||||
table.clear()
|
||||
|
||||
for run in runs:
|
||||
state = run.get("state", "unknown")
|
||||
state_style = {
|
||||
"queued": "yellow",
|
||||
"running": "green",
|
||||
"succeeded": "cyan",
|
||||
"failed": "red",
|
||||
"cancelled": "dim",
|
||||
}.get(state, "")
|
||||
|
||||
state_text = Text(state, style=state_style)
|
||||
|
||||
input_preview = run.get("input", "")[:40]
|
||||
if len(run.get("input", "")) > 40:
|
||||
input_preview += "..."
|
||||
|
||||
table.add_row(
|
||||
run.get("run_id", "")[:8],
|
||||
run.get("agent_name", ""),
|
||||
input_preview,
|
||||
state_text,
|
||||
run.get("priority", "normal"),
|
||||
run.get("wait_time", "-"),
|
||||
run.get("duration", "-"),
|
||||
key=run.get("run_id"),
|
||||
)
|
||||
|
||||
def on_data_table_row_selected(self, event: DataTable.RowSelected) -> None:
|
||||
"""Handle row selection."""
|
||||
if event.row_key:
|
||||
self._selected_run_id = str(event.row_key.value)
|
||||
|
||||
def on_button_pressed(self, event: Button.Pressed) -> None:
|
||||
"""Handle button press."""
|
||||
if event.button.id == "btn-cancel":
|
||||
self.action_cancel()
|
||||
elif event.button.id == "btn-retry":
|
||||
self.action_retry()
|
||||
elif event.button.id == "btn-clear":
|
||||
self.action_clear()
|
||||
elif event.button.id == "btn-back":
|
||||
self.action_back()
|
||||
|
||||
def action_back(self) -> None:
|
||||
"""Go back to main screen."""
|
||||
self.app.pop_screen()
|
||||
|
||||
def action_cancel(self) -> None:
|
||||
"""Cancel selected run."""
|
||||
if self._selected_run_id:
|
||||
self.post_message(self.RunCancelled(self._selected_run_id))
|
||||
|
||||
def action_retry(self) -> None:
|
||||
"""Retry selected run."""
|
||||
if self._selected_run_id:
|
||||
self.post_message(self.RunRetried(self._selected_run_id))
|
||||
|
||||
def action_clear(self) -> None:
|
||||
"""Clear the queue."""
|
||||
self.post_message(self.QueueCleared())
|
||||
|
||||
else:
|
||||
class QueueScreen:
|
||||
"""Placeholder when Textual is not available."""
|
||||
def __init__(self, *args, **kwargs):
|
||||
raise ImportError(
|
||||
"Textual is required for TUI. Install with: pip install praisonai[tui]"
|
||||
)
|
||||
@@ -0,0 +1,124 @@
|
||||
"""
|
||||
Session Screen for PraisonAI TUI.
|
||||
|
||||
Session browser and management.
|
||||
"""
|
||||
|
||||
from typing import Optional, List
|
||||
|
||||
try:
|
||||
from textual.screen import Screen
|
||||
from textual.containers import Vertical
|
||||
from textual.widgets import Static, Footer, DataTable, Button
|
||||
from textual.binding import Binding
|
||||
from textual.message import Message
|
||||
from rich.text import Text
|
||||
TEXTUAL_AVAILABLE = True
|
||||
except ImportError:
|
||||
TEXTUAL_AVAILABLE = False
|
||||
Screen = object
|
||||
Message = object
|
||||
|
||||
|
||||
if TEXTUAL_AVAILABLE:
|
||||
class SessionScreen(Screen):
|
||||
"""
|
||||
Session browser screen.
|
||||
|
||||
Displays recent sessions and allows resuming.
|
||||
"""
|
||||
|
||||
BINDINGS = [
|
||||
Binding("escape", "back", "Back", show=True),
|
||||
Binding("enter", "resume", "Resume", show=True),
|
||||
Binding("d", "delete", "Delete", show=True),
|
||||
]
|
||||
|
||||
DEFAULT_CSS = """
|
||||
SessionScreen {
|
||||
background: $surface;
|
||||
}
|
||||
|
||||
SessionScreen #session-header {
|
||||
height: 3;
|
||||
background: $primary;
|
||||
padding: 1;
|
||||
}
|
||||
|
||||
SessionScreen DataTable {
|
||||
height: 1fr;
|
||||
}
|
||||
"""
|
||||
|
||||
class SessionSelected(Message):
|
||||
def __init__(self, session_id: str):
|
||||
self.session_id = session_id
|
||||
super().__init__()
|
||||
|
||||
class SessionDeleted(Message):
|
||||
def __init__(self, session_id: str):
|
||||
self.session_id = session_id
|
||||
super().__init__()
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
name: Optional[str] = None,
|
||||
id: Optional[str] = None,
|
||||
classes: Optional[str] = None,
|
||||
):
|
||||
super().__init__(name=name, id=id, classes=classes)
|
||||
self._selected_session_id: Optional[str] = None
|
||||
|
||||
def compose(self):
|
||||
"""Compose the screen."""
|
||||
yield Static("Sessions", id="session-header")
|
||||
yield DataTable(id="session-table")
|
||||
yield Footer()
|
||||
|
||||
def on_mount(self) -> None:
|
||||
"""Handle mount."""
|
||||
table = self.query_one("#session-table", DataTable)
|
||||
table.add_columns("ID", "Created", "Updated", "Runs")
|
||||
table.cursor_type = "row"
|
||||
|
||||
def update_sessions(self, sessions: List[dict]) -> None:
|
||||
"""Update the sessions display."""
|
||||
table = self.query_one("#session-table", DataTable)
|
||||
table.clear()
|
||||
|
||||
for session in sessions:
|
||||
table.add_row(
|
||||
session.get("session_id", "")[:8],
|
||||
session.get("created_at", "-"),
|
||||
session.get("updated_at", "-"),
|
||||
str(session.get("run_count", 0)),
|
||||
key=session.get("session_id"),
|
||||
)
|
||||
|
||||
def on_data_table_row_selected(self, event: DataTable.RowSelected) -> None:
|
||||
"""Handle row selection."""
|
||||
if event.row_key:
|
||||
self._selected_session_id = str(event.row_key.value)
|
||||
|
||||
def action_back(self) -> None:
|
||||
"""Go back."""
|
||||
self.app.pop_screen()
|
||||
|
||||
def action_resume(self) -> None:
|
||||
"""Resume selected session."""
|
||||
if self._selected_session_id:
|
||||
self.post_message(self.SessionSelected(self._selected_session_id))
|
||||
self.app.pop_screen()
|
||||
|
||||
def action_delete(self) -> None:
|
||||
"""Delete selected session."""
|
||||
if self._selected_session_id:
|
||||
self.post_message(self.SessionDeleted(self._selected_session_id))
|
||||
|
||||
else:
|
||||
class SessionScreen:
|
||||
"""Placeholder when Textual is not available."""
|
||||
def __init__(self, *args, **kwargs):
|
||||
raise ImportError(
|
||||
"Textual is required for TUI. Install with: pip install praisonai[tui]"
|
||||
)
|
||||
@@ -0,0 +1,148 @@
|
||||
"""
|
||||
Settings Screen for PraisonAI TUI.
|
||||
|
||||
Configuration interface.
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
try:
|
||||
from textual.screen import Screen
|
||||
from textual.containers import Vertical
|
||||
from textual.widgets import Static, Footer, Input, Select, Switch, Button
|
||||
from textual.binding import Binding
|
||||
from textual.message import Message
|
||||
TEXTUAL_AVAILABLE = True
|
||||
except ImportError:
|
||||
TEXTUAL_AVAILABLE = False
|
||||
Screen = object
|
||||
Message = object
|
||||
|
||||
|
||||
if TEXTUAL_AVAILABLE:
|
||||
class SettingsScreen(Screen):
|
||||
"""
|
||||
Settings screen.
|
||||
|
||||
Allows configuration of:
|
||||
- Model selection
|
||||
- Queue settings
|
||||
- Display preferences
|
||||
"""
|
||||
|
||||
BINDINGS = [
|
||||
Binding("escape", "back", "Back", show=True),
|
||||
Binding("s", "save", "Save", show=True),
|
||||
]
|
||||
|
||||
DEFAULT_CSS = """
|
||||
SettingsScreen {
|
||||
background: $surface;
|
||||
}
|
||||
|
||||
SettingsScreen #settings-header {
|
||||
height: 3;
|
||||
background: $primary;
|
||||
padding: 1;
|
||||
}
|
||||
|
||||
SettingsScreen .setting-row {
|
||||
height: 3;
|
||||
padding: 0 1;
|
||||
}
|
||||
|
||||
SettingsScreen .setting-label {
|
||||
width: 20;
|
||||
}
|
||||
|
||||
SettingsScreen Input {
|
||||
width: 40;
|
||||
}
|
||||
"""
|
||||
|
||||
class SettingsSaved(Message):
|
||||
def __init__(self, settings: dict):
|
||||
self.settings = settings
|
||||
super().__init__()
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
current_settings: Optional[dict] = None,
|
||||
name: Optional[str] = None,
|
||||
id: Optional[str] = None,
|
||||
classes: Optional[str] = None,
|
||||
):
|
||||
super().__init__(name=name, id=id, classes=classes)
|
||||
self._settings = current_settings or {}
|
||||
|
||||
def compose(self):
|
||||
"""Compose the screen."""
|
||||
yield Static("Settings", id="settings-header")
|
||||
|
||||
with Vertical(id="settings-form"):
|
||||
# Model setting
|
||||
yield Static("Model:", classes="setting-label")
|
||||
yield Input(
|
||||
value=self._settings.get("model", "gpt-4o-mini"),
|
||||
id="setting-model",
|
||||
placeholder="Model name"
|
||||
)
|
||||
|
||||
# Max concurrent runs
|
||||
yield Static("Max Concurrent:", classes="setting-label")
|
||||
yield Input(
|
||||
value=str(self._settings.get("max_concurrent", 4)),
|
||||
id="setting-max-concurrent",
|
||||
placeholder="4"
|
||||
)
|
||||
|
||||
# Auto-save interval
|
||||
yield Static("Autosave (sec):", classes="setting-label")
|
||||
yield Input(
|
||||
value=str(self._settings.get("autosave_interval", 30)),
|
||||
id="setting-autosave",
|
||||
placeholder="30"
|
||||
)
|
||||
|
||||
# Buttons
|
||||
yield Button("Save", id="btn-save", variant="primary")
|
||||
yield Button("Cancel", id="btn-cancel", variant="default")
|
||||
|
||||
yield Footer()
|
||||
|
||||
def on_button_pressed(self, event: Button.Pressed) -> None:
|
||||
"""Handle button press."""
|
||||
if event.button.id == "btn-save":
|
||||
self.action_save()
|
||||
elif event.button.id == "btn-cancel":
|
||||
self.action_back()
|
||||
|
||||
def action_back(self) -> None:
|
||||
"""Go back."""
|
||||
self.app.pop_screen()
|
||||
|
||||
def action_save(self) -> None:
|
||||
"""Save settings."""
|
||||
try:
|
||||
model_input = self.query_one("#setting-model", Input)
|
||||
max_concurrent_input = self.query_one("#setting-max-concurrent", Input)
|
||||
autosave_input = self.query_one("#setting-autosave", Input)
|
||||
|
||||
settings = {
|
||||
"model": model_input.value,
|
||||
"max_concurrent": int(max_concurrent_input.value or "4"),
|
||||
"autosave_interval": int(autosave_input.value or "30"),
|
||||
}
|
||||
|
||||
self.post_message(self.SettingsSaved(settings))
|
||||
self.app.pop_screen()
|
||||
except ValueError:
|
||||
pass
|
||||
|
||||
else:
|
||||
class SettingsScreen:
|
||||
"""Placeholder when Textual is not available."""
|
||||
def __init__(self, *args, **kwargs):
|
||||
raise ImportError(
|
||||
"Textual is required for TUI. Install with: pip install praisonai[tui]"
|
||||
)
|
||||
@@ -0,0 +1,56 @@
|
||||
"""
|
||||
TUI Widgets for PraisonAI.
|
||||
|
||||
Reusable UI components for the TUI application.
|
||||
"""
|
||||
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from .chat import ChatWidget
|
||||
from .composer import ComposerWidget
|
||||
from .status import StatusWidget
|
||||
from .queue_panel import QueuePanelWidget
|
||||
from .tool_panel import ToolPanelWidget
|
||||
|
||||
_lazy_cache = {}
|
||||
|
||||
|
||||
def __getattr__(name: str):
|
||||
"""Lazy load widgets."""
|
||||
global _lazy_cache
|
||||
|
||||
if name in _lazy_cache:
|
||||
return _lazy_cache[name]
|
||||
|
||||
if name == "ChatWidget":
|
||||
from .chat import ChatWidget
|
||||
_lazy_cache[name] = ChatWidget
|
||||
return ChatWidget
|
||||
elif name == "ComposerWidget":
|
||||
from .composer import ComposerWidget
|
||||
_lazy_cache[name] = ComposerWidget
|
||||
return ComposerWidget
|
||||
elif name == "StatusWidget":
|
||||
from .status import StatusWidget
|
||||
_lazy_cache[name] = StatusWidget
|
||||
return StatusWidget
|
||||
elif name == "QueuePanelWidget":
|
||||
from .queue_panel import QueuePanelWidget
|
||||
_lazy_cache[name] = QueuePanelWidget
|
||||
return QueuePanelWidget
|
||||
elif name == "ToolPanelWidget":
|
||||
from .tool_panel import ToolPanelWidget
|
||||
_lazy_cache[name] = ToolPanelWidget
|
||||
return ToolPanelWidget
|
||||
|
||||
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|
||||
|
||||
|
||||
__all__ = [
|
||||
"ChatWidget",
|
||||
"ComposerWidget",
|
||||
"StatusWidget",
|
||||
"QueuePanelWidget",
|
||||
"ToolPanelWidget",
|
||||
]
|
||||
@@ -0,0 +1,261 @@
|
||||
"""
|
||||
Chat Widget for PraisonAI TUI.
|
||||
|
||||
Displays chat history with streaming support.
|
||||
"""
|
||||
|
||||
from typing import List, Optional
|
||||
from dataclasses import dataclass, field
|
||||
import time
|
||||
|
||||
try:
|
||||
from textual.widget import Widget
|
||||
from textual.widgets import Static, RichLog
|
||||
from textual.containers import ScrollableContainer, Vertical
|
||||
from textual.reactive import reactive
|
||||
from textual.message import Message
|
||||
from rich.markdown import Markdown
|
||||
from rich.panel import Panel
|
||||
from rich.text import Text
|
||||
TEXTUAL_AVAILABLE = True
|
||||
except ImportError:
|
||||
TEXTUAL_AVAILABLE = False
|
||||
Widget = object
|
||||
Message = object
|
||||
|
||||
|
||||
@dataclass
|
||||
class ChatMessage:
|
||||
"""A single chat message."""
|
||||
role: str # "user", "assistant", "system", "tool"
|
||||
content: str
|
||||
timestamp: float = field(default_factory=time.time)
|
||||
run_id: Optional[str] = None
|
||||
agent_name: Optional[str] = None
|
||||
is_streaming: bool = False
|
||||
|
||||
@property
|
||||
def display_role(self) -> str:
|
||||
"""Get display name for role."""
|
||||
if self.role == "user":
|
||||
return "You"
|
||||
elif self.role == "assistant":
|
||||
return self.agent_name or "Assistant"
|
||||
elif self.role == "system":
|
||||
return "System"
|
||||
elif self.role == "tool":
|
||||
return "Tool"
|
||||
return self.role.title()
|
||||
|
||||
|
||||
if TEXTUAL_AVAILABLE:
|
||||
class ChatWidget(ScrollableContainer):
|
||||
"""
|
||||
Widget for displaying chat history.
|
||||
|
||||
Features:
|
||||
- Streaming message updates
|
||||
- Markdown rendering
|
||||
- Auto-scroll to bottom
|
||||
- Message history management
|
||||
"""
|
||||
|
||||
DEFAULT_CSS = """
|
||||
ChatWidget {
|
||||
height: 100%;
|
||||
border: solid $primary;
|
||||
background: $surface;
|
||||
padding: 0 1;
|
||||
}
|
||||
|
||||
ChatWidget .message-user {
|
||||
background: $primary-darken-2;
|
||||
margin: 1 0;
|
||||
padding: 1;
|
||||
}
|
||||
|
||||
ChatWidget .message-assistant {
|
||||
background: $surface-darken-1;
|
||||
margin: 1 0;
|
||||
padding: 1;
|
||||
}
|
||||
|
||||
ChatWidget .message-system {
|
||||
background: $warning-darken-3;
|
||||
margin: 1 0;
|
||||
padding: 1;
|
||||
color: $text-muted;
|
||||
}
|
||||
|
||||
ChatWidget .message-streaming {
|
||||
border: dashed $accent;
|
||||
}
|
||||
"""
|
||||
|
||||
class MessageAdded(Message):
|
||||
"""Event when a message is added."""
|
||||
def __init__(self, message: ChatMessage):
|
||||
self.message = message
|
||||
super().__init__()
|
||||
|
||||
class StreamingUpdate(Message):
|
||||
"""Event when streaming content updates."""
|
||||
def __init__(self, run_id: str, content: str):
|
||||
self.run_id = run_id
|
||||
self.content = content
|
||||
super().__init__()
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
max_messages: int = 1000,
|
||||
name: Optional[str] = None,
|
||||
id: Optional[str] = None,
|
||||
classes: Optional[str] = None,
|
||||
):
|
||||
super().__init__(name=name, id=id, classes=classes)
|
||||
self._messages: List[ChatMessage] = []
|
||||
self._max_messages = max_messages
|
||||
self._streaming_widgets: dict = {}
|
||||
|
||||
def compose(self):
|
||||
"""Compose the widget."""
|
||||
yield Vertical(id="chat-messages")
|
||||
|
||||
async def add_message(self, message: ChatMessage) -> None:
|
||||
"""Add a message to the chat."""
|
||||
self._messages.append(message)
|
||||
|
||||
# Trim old messages if needed
|
||||
if len(self._messages) > self._max_messages:
|
||||
self._messages = self._messages[-self._max_messages:]
|
||||
|
||||
# Create message widget
|
||||
await self._render_message(message)
|
||||
|
||||
# Auto-scroll to bottom
|
||||
self.scroll_end(animate=False)
|
||||
|
||||
self.post_message(self.MessageAdded(message))
|
||||
|
||||
async def _render_message(self, message: ChatMessage) -> None:
|
||||
"""Render a single message."""
|
||||
container = self.query_one("#chat-messages", Vertical)
|
||||
|
||||
# Create role label
|
||||
role_style = {
|
||||
"user": "bold cyan",
|
||||
"assistant": "bold green",
|
||||
"system": "bold yellow",
|
||||
"tool": "bold magenta",
|
||||
}.get(message.role, "bold")
|
||||
|
||||
role_text = Text(f"{message.display_role}:", style=role_style)
|
||||
|
||||
# Create content
|
||||
try:
|
||||
content = Markdown(message.content)
|
||||
except Exception:
|
||||
content = Text(message.content)
|
||||
|
||||
# Create panel
|
||||
css_class = f"message-{message.role}"
|
||||
if message.is_streaming:
|
||||
css_class += " message-streaming"
|
||||
|
||||
widget_id = f"msg-{message.run_id or id(message)}"
|
||||
|
||||
panel = Static(
|
||||
Panel(content, title=str(role_text), border_style=role_style),
|
||||
id=widget_id,
|
||||
classes=css_class,
|
||||
)
|
||||
|
||||
await container.mount(panel)
|
||||
|
||||
if message.is_streaming:
|
||||
self._streaming_widgets[message.run_id] = widget_id
|
||||
|
||||
async def update_streaming(self, run_id: str, content: str) -> None:
|
||||
"""Update a streaming message."""
|
||||
if run_id not in self._streaming_widgets:
|
||||
return
|
||||
|
||||
widget_id = self._streaming_widgets[run_id]
|
||||
|
||||
try:
|
||||
widget = self.query_one(f"#{widget_id}", Static)
|
||||
|
||||
# Find the message
|
||||
for msg in self._messages:
|
||||
if msg.run_id == run_id:
|
||||
msg.content = content
|
||||
break
|
||||
|
||||
# Update content
|
||||
try:
|
||||
rendered = Markdown(content + " ▌")
|
||||
except Exception:
|
||||
rendered = Text(content + " ▌")
|
||||
|
||||
widget.update(Panel(rendered, title="Assistant", border_style="bold green"))
|
||||
|
||||
# Auto-scroll
|
||||
self.scroll_end(animate=False)
|
||||
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
async def complete_streaming(self, run_id: str, final_content: str) -> None:
|
||||
"""Complete a streaming message."""
|
||||
if run_id not in self._streaming_widgets:
|
||||
return
|
||||
|
||||
widget_id = self._streaming_widgets.pop(run_id)
|
||||
|
||||
try:
|
||||
widget = self.query_one(f"#{widget_id}", Static)
|
||||
|
||||
# Update message
|
||||
for msg in self._messages:
|
||||
if msg.run_id == run_id:
|
||||
msg.content = final_content
|
||||
msg.is_streaming = False
|
||||
break
|
||||
|
||||
# Update content without cursor
|
||||
try:
|
||||
rendered = Markdown(final_content)
|
||||
except Exception:
|
||||
rendered = Text(final_content)
|
||||
|
||||
widget.update(Panel(rendered, title="Assistant", border_style="bold green"))
|
||||
widget.remove_class("message-streaming")
|
||||
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
async def clear(self) -> None:
|
||||
"""Clear all messages."""
|
||||
self._messages.clear()
|
||||
self._streaming_widgets.clear()
|
||||
|
||||
container = self.query_one("#chat-messages", Vertical)
|
||||
await container.remove_children()
|
||||
|
||||
@property
|
||||
def messages(self) -> List[ChatMessage]:
|
||||
"""Get all messages."""
|
||||
return self._messages.copy()
|
||||
|
||||
@property
|
||||
def message_count(self) -> int:
|
||||
"""Get message count."""
|
||||
return len(self._messages)
|
||||
|
||||
else:
|
||||
class ChatWidget:
|
||||
"""Placeholder when Textual is not available."""
|
||||
def __init__(self, *args, **kwargs):
|
||||
raise ImportError(
|
||||
"Textual is required for TUI. Install with: pip install praisonai[tui]"
|
||||
)
|
||||
@@ -0,0 +1,224 @@
|
||||
"""
|
||||
Composer Widget for PraisonAI TUI.
|
||||
|
||||
Input area for composing messages with slash command support.
|
||||
"""
|
||||
|
||||
from typing import Callable, List, Optional
|
||||
import asyncio
|
||||
|
||||
try:
|
||||
from textual.widget import Widget
|
||||
from textual.widgets import TextArea, Input
|
||||
from textual.containers import Horizontal, Vertical
|
||||
from textual.message import Message
|
||||
from textual.binding import Binding
|
||||
from textual import events
|
||||
from rich.text import Text
|
||||
TEXTUAL_AVAILABLE = True
|
||||
except ImportError:
|
||||
TEXTUAL_AVAILABLE = False
|
||||
Widget = object
|
||||
Message = object
|
||||
|
||||
|
||||
if TEXTUAL_AVAILABLE:
|
||||
class ComposerWidget(Vertical):
|
||||
"""
|
||||
Input composer for chat messages.
|
||||
|
||||
Features:
|
||||
- Multi-line input
|
||||
- Slash command detection
|
||||
- @ mention support
|
||||
- History navigation
|
||||
- Send on Enter or Ctrl+Enter
|
||||
"""
|
||||
|
||||
BINDINGS = [
|
||||
Binding("ctrl+enter", "submit", "Send", show=True),
|
||||
Binding("escape", "cancel", "Cancel", show=False),
|
||||
Binding("up", "history_prev", "Previous", show=False),
|
||||
Binding("down", "history_next", "Next", show=False),
|
||||
]
|
||||
|
||||
DEFAULT_CSS = """
|
||||
ComposerWidget {
|
||||
height: auto;
|
||||
max-height: 10;
|
||||
border: solid $primary;
|
||||
background: $surface;
|
||||
padding: 0 1;
|
||||
}
|
||||
|
||||
ComposerWidget TextArea {
|
||||
height: auto;
|
||||
min-height: 3;
|
||||
max-height: 8;
|
||||
}
|
||||
|
||||
ComposerWidget .hint {
|
||||
color: $text-muted;
|
||||
text-style: italic;
|
||||
}
|
||||
"""
|
||||
|
||||
class Submitted(Message):
|
||||
"""Event when message is submitted."""
|
||||
def __init__(self, content: str, is_command: bool = False):
|
||||
self.content = content
|
||||
self.is_command = is_command
|
||||
super().__init__()
|
||||
|
||||
class CommandDetected(Message):
|
||||
"""Event when a slash command is detected."""
|
||||
def __init__(self, command: str, args: str):
|
||||
self.command = command
|
||||
self.args = args
|
||||
super().__init__()
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
placeholder: str = "Type your message... (Ctrl+Enter to send, /help for commands)",
|
||||
multiline: bool = True,
|
||||
name: Optional[str] = None,
|
||||
id: Optional[str] = None,
|
||||
classes: Optional[str] = None,
|
||||
):
|
||||
super().__init__(name=name, id=id, classes=classes)
|
||||
self._placeholder = placeholder
|
||||
self._multiline = multiline
|
||||
self._history: List[str] = []
|
||||
self._history_index = -1
|
||||
self._current_input = ""
|
||||
self._is_processing = False
|
||||
|
||||
def compose(self):
|
||||
"""Compose the widget."""
|
||||
yield TextArea(
|
||||
"",
|
||||
id="composer-input",
|
||||
language=None,
|
||||
)
|
||||
|
||||
def on_mount(self) -> None:
|
||||
"""Handle mount event."""
|
||||
text_area = self.query_one("#composer-input", TextArea)
|
||||
text_area.focus()
|
||||
|
||||
async def on_text_area_changed(self, event: TextArea.Changed) -> None:
|
||||
"""Handle text changes."""
|
||||
content = event.text_area.text
|
||||
|
||||
# Detect slash commands
|
||||
if content.startswith("/"):
|
||||
parts = content[1:].split(maxsplit=1)
|
||||
if parts:
|
||||
command = parts[0]
|
||||
args = parts[1] if len(parts) > 1 else ""
|
||||
self.post_message(self.CommandDetected(command, args))
|
||||
|
||||
def on_key(self, event: events.Key) -> None:
|
||||
"""Handle key events."""
|
||||
# Submit on Enter (single line mode) or Ctrl+Enter
|
||||
if event.key == "enter" and not self._multiline:
|
||||
event.prevent_default()
|
||||
self.action_submit()
|
||||
elif event.key == "ctrl+enter":
|
||||
event.prevent_default()
|
||||
self.action_submit()
|
||||
|
||||
def action_submit(self) -> None:
|
||||
"""Submit the current input."""
|
||||
if self._is_processing:
|
||||
return
|
||||
|
||||
text_area = self.query_one("#composer-input", TextArea)
|
||||
content = text_area.text.strip()
|
||||
|
||||
if not content:
|
||||
return
|
||||
|
||||
# Add to history
|
||||
if content and (not self._history or self._history[-1] != content):
|
||||
self._history.append(content)
|
||||
self._history_index = -1
|
||||
|
||||
# Check if it's a command
|
||||
is_command = content.startswith("/")
|
||||
|
||||
# Clear input
|
||||
text_area.clear()
|
||||
|
||||
# Emit event
|
||||
self.post_message(self.Submitted(content, is_command))
|
||||
|
||||
def action_cancel(self) -> None:
|
||||
"""Cancel current input."""
|
||||
text_area = self.query_one("#composer-input", TextArea)
|
||||
text_area.clear()
|
||||
self._history_index = -1
|
||||
|
||||
def action_history_prev(self) -> None:
|
||||
"""Navigate to previous history item."""
|
||||
if not self._history:
|
||||
return
|
||||
|
||||
text_area = self.query_one("#composer-input", TextArea)
|
||||
|
||||
if self._history_index == -1:
|
||||
self._current_input = text_area.text
|
||||
self._history_index = len(self._history) - 1
|
||||
elif self._history_index > 0:
|
||||
self._history_index -= 1
|
||||
|
||||
text_area.clear()
|
||||
text_area.insert(self._history[self._history_index])
|
||||
|
||||
def action_history_next(self) -> None:
|
||||
"""Navigate to next history item."""
|
||||
if self._history_index == -1:
|
||||
return
|
||||
|
||||
text_area = self.query_one("#composer-input", TextArea)
|
||||
|
||||
if self._history_index < len(self._history) - 1:
|
||||
self._history_index += 1
|
||||
text_area.clear()
|
||||
text_area.insert(self._history[self._history_index])
|
||||
else:
|
||||
self._history_index = -1
|
||||
text_area.clear()
|
||||
text_area.insert(self._current_input)
|
||||
|
||||
def set_processing(self, processing: bool) -> None:
|
||||
"""Set processing state."""
|
||||
self._is_processing = processing
|
||||
text_area = self.query_one("#composer-input", TextArea)
|
||||
text_area.disabled = processing
|
||||
|
||||
@property
|
||||
def text(self) -> str:
|
||||
"""Get current text."""
|
||||
text_area = self.query_one("#composer-input", TextArea)
|
||||
return text_area.text
|
||||
|
||||
@text.setter
|
||||
def text(self, value: str) -> None:
|
||||
"""Set current text."""
|
||||
text_area = self.query_one("#composer-input", TextArea)
|
||||
text_area.clear()
|
||||
text_area.insert(value)
|
||||
|
||||
def focus_input(self) -> None:
|
||||
"""Focus the input area."""
|
||||
text_area = self.query_one("#composer-input", TextArea)
|
||||
text_area.focus()
|
||||
|
||||
else:
|
||||
class ComposerWidget:
|
||||
"""Placeholder when Textual is not available."""
|
||||
def __init__(self, *args, **kwargs):
|
||||
raise ImportError(
|
||||
"Textual is required for TUI. Install with: pip install praisonai[tui]"
|
||||
)
|
||||
@@ -0,0 +1,200 @@
|
||||
"""
|
||||
Queue Panel Widget for PraisonAI TUI.
|
||||
|
||||
Displays queue status and allows queue management.
|
||||
"""
|
||||
|
||||
from typing import List, Optional
|
||||
from dataclasses import dataclass
|
||||
|
||||
try:
|
||||
from textual.widget import Widget
|
||||
from textual.widgets import Static, DataTable, Button
|
||||
from textual.containers import Vertical
|
||||
from textual.reactive import reactive
|
||||
from textual.message import Message
|
||||
from rich.text import Text
|
||||
from rich.panel import Panel
|
||||
TEXTUAL_AVAILABLE = True
|
||||
except ImportError:
|
||||
TEXTUAL_AVAILABLE = False
|
||||
Widget = object
|
||||
Message = object
|
||||
|
||||
|
||||
@dataclass
|
||||
class QueueItem:
|
||||
"""A queue item for display."""
|
||||
run_id: str
|
||||
agent_name: str
|
||||
input_preview: str
|
||||
state: str
|
||||
priority: str
|
||||
wait_time: str
|
||||
|
||||
|
||||
if TEXTUAL_AVAILABLE:
|
||||
class QueuePanelWidget(Vertical):
|
||||
"""
|
||||
Queue panel widget.
|
||||
|
||||
Displays:
|
||||
- Queue summary (queued/running counts)
|
||||
- List of queued runs
|
||||
- Cancel/retry buttons
|
||||
"""
|
||||
|
||||
DEFAULT_CSS = """
|
||||
QueuePanelWidget {
|
||||
height: 100%;
|
||||
border: solid $secondary;
|
||||
background: $surface;
|
||||
padding: 0 1;
|
||||
}
|
||||
|
||||
QueuePanelWidget .queue-header {
|
||||
height: 1;
|
||||
background: $secondary;
|
||||
padding: 0 1;
|
||||
}
|
||||
|
||||
QueuePanelWidget DataTable {
|
||||
height: 1fr;
|
||||
}
|
||||
|
||||
QueuePanelWidget .queue-item-queued {
|
||||
color: $warning;
|
||||
}
|
||||
|
||||
QueuePanelWidget .queue-item-running {
|
||||
color: $success;
|
||||
}
|
||||
|
||||
QueuePanelWidget .queue-item-failed {
|
||||
color: $error;
|
||||
}
|
||||
"""
|
||||
|
||||
class RunSelected(Message):
|
||||
"""Event when a run is selected."""
|
||||
def __init__(self, run_id: str):
|
||||
self.run_id = run_id
|
||||
super().__init__()
|
||||
|
||||
class CancelRequested(Message):
|
||||
"""Event when cancel is requested."""
|
||||
def __init__(self, run_id: str):
|
||||
self.run_id = run_id
|
||||
super().__init__()
|
||||
|
||||
class RetryRequested(Message):
|
||||
"""Event when retry is requested."""
|
||||
def __init__(self, run_id: str):
|
||||
self.run_id = run_id
|
||||
super().__init__()
|
||||
|
||||
queued_count: reactive[int] = reactive(0)
|
||||
running_count: reactive[int] = reactive(0)
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
name: Optional[str] = None,
|
||||
id: Optional[str] = None,
|
||||
classes: Optional[str] = None,
|
||||
):
|
||||
super().__init__(name=name, id=id, classes=classes)
|
||||
self._items: List[QueueItem] = []
|
||||
self._selected_run_id: Optional[str] = None
|
||||
|
||||
def compose(self):
|
||||
"""Compose the widget."""
|
||||
yield Static("Queue", id="queue-title", classes="queue-header")
|
||||
yield DataTable(id="queue-table")
|
||||
|
||||
def on_mount(self) -> None:
|
||||
"""Handle mount."""
|
||||
table = self.query_one("#queue-table", DataTable)
|
||||
table.add_columns("ID", "Agent", "Input", "State", "Priority")
|
||||
table.cursor_type = "row"
|
||||
self._update_header()
|
||||
|
||||
def watch_queued_count(self, value: int) -> None:
|
||||
self._update_header()
|
||||
|
||||
def watch_running_count(self, value: int) -> None:
|
||||
self._update_header()
|
||||
|
||||
def _update_header(self) -> None:
|
||||
"""Update the header text."""
|
||||
try:
|
||||
header = self.query_one("#queue-title", Static)
|
||||
text = Text()
|
||||
text.append("Queue ", style="bold")
|
||||
text.append(f"⏳ {self.queued_count} queued", style="yellow")
|
||||
text.append(" │ ", style="dim")
|
||||
text.append(f"▶ {self.running_count} running", style="green")
|
||||
header.update(text)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
def update_items(self, items: List[QueueItem]) -> None:
|
||||
"""Update the queue items."""
|
||||
self._items = items
|
||||
|
||||
table = self.query_one("#queue-table", DataTable)
|
||||
table.clear()
|
||||
|
||||
for item in items:
|
||||
# Style based on state
|
||||
state_style = {
|
||||
"queued": "yellow",
|
||||
"running": "green",
|
||||
"failed": "red",
|
||||
"cancelled": "dim",
|
||||
"succeeded": "cyan",
|
||||
}.get(item.state.lower(), "")
|
||||
|
||||
state_text = Text(item.state, style=state_style)
|
||||
|
||||
# Truncate input preview
|
||||
input_preview = item.input_preview[:30]
|
||||
if len(item.input_preview) > 30:
|
||||
input_preview += "..."
|
||||
|
||||
table.add_row(
|
||||
item.run_id[:8],
|
||||
item.agent_name,
|
||||
input_preview,
|
||||
state_text,
|
||||
item.priority,
|
||||
key=item.run_id,
|
||||
)
|
||||
|
||||
def on_data_table_row_selected(self, event: DataTable.RowSelected) -> None:
|
||||
"""Handle row selection."""
|
||||
if event.row_key:
|
||||
self._selected_run_id = str(event.row_key.value)
|
||||
self.post_message(self.RunSelected(self._selected_run_id))
|
||||
|
||||
def cancel_selected(self) -> None:
|
||||
"""Cancel the selected run."""
|
||||
if self._selected_run_id:
|
||||
self.post_message(self.CancelRequested(self._selected_run_id))
|
||||
|
||||
def retry_selected(self) -> None:
|
||||
"""Retry the selected run."""
|
||||
if self._selected_run_id:
|
||||
self.post_message(self.RetryRequested(self._selected_run_id))
|
||||
|
||||
@property
|
||||
def selected_run_id(self) -> Optional[str]:
|
||||
"""Get the selected run ID."""
|
||||
return self._selected_run_id
|
||||
|
||||
else:
|
||||
class QueuePanelWidget:
|
||||
"""Placeholder when Textual is not available."""
|
||||
def __init__(self, *args, **kwargs):
|
||||
raise ImportError(
|
||||
"Textual is required for TUI. Install with: pip install praisonai[tui]"
|
||||
)
|
||||
@@ -0,0 +1,167 @@
|
||||
"""
|
||||
Status Widget for PraisonAI TUI.
|
||||
|
||||
Displays status bar with session info, model, tokens, and cost.
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
from dataclasses import dataclass
|
||||
|
||||
try:
|
||||
from textual.widget import Widget
|
||||
from textual.widgets import Static
|
||||
from textual.containers import Horizontal
|
||||
from textual.reactive import reactive
|
||||
from rich.text import Text
|
||||
from rich.table import Table
|
||||
TEXTUAL_AVAILABLE = True
|
||||
except ImportError:
|
||||
TEXTUAL_AVAILABLE = False
|
||||
Widget = object
|
||||
|
||||
|
||||
@dataclass
|
||||
class StatusInfo:
|
||||
"""Status information."""
|
||||
session_id: str = ""
|
||||
model: str = "gpt-4o-mini"
|
||||
total_tokens: int = 0
|
||||
total_cost: float = 0.0
|
||||
queued_count: int = 0
|
||||
running_count: int = 0
|
||||
is_processing: bool = False
|
||||
status_message: str = ""
|
||||
|
||||
|
||||
if TEXTUAL_AVAILABLE:
|
||||
class StatusWidget(Static):
|
||||
"""
|
||||
Status bar widget.
|
||||
|
||||
Displays:
|
||||
- Session ID
|
||||
- Current model
|
||||
- Token usage
|
||||
- Cost
|
||||
- Queue status
|
||||
- Processing indicator
|
||||
"""
|
||||
|
||||
DEFAULT_CSS = """
|
||||
StatusWidget {
|
||||
height: 1;
|
||||
background: $primary;
|
||||
color: $text;
|
||||
padding: 0 1;
|
||||
}
|
||||
"""
|
||||
|
||||
session_id: reactive[str] = reactive("")
|
||||
model: reactive[str] = reactive("gpt-4o-mini")
|
||||
total_tokens: reactive[int] = reactive(0)
|
||||
total_cost: reactive[float] = reactive(0.0)
|
||||
queued_count: reactive[int] = reactive(0)
|
||||
running_count: reactive[int] = reactive(0)
|
||||
is_processing: reactive[bool] = reactive(False)
|
||||
status_message: reactive[str] = reactive("")
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
name: Optional[str] = None,
|
||||
id: Optional[str] = None,
|
||||
classes: Optional[str] = None,
|
||||
):
|
||||
super().__init__("", name=name, id=id, classes=classes)
|
||||
|
||||
def on_mount(self) -> None:
|
||||
"""Handle mount."""
|
||||
self._update_display()
|
||||
|
||||
def watch_session_id(self, value: str) -> None:
|
||||
self._update_display()
|
||||
|
||||
def watch_model(self, value: str) -> None:
|
||||
self._update_display()
|
||||
|
||||
def watch_total_tokens(self, value: int) -> None:
|
||||
self._update_display()
|
||||
|
||||
def watch_total_cost(self, value: float) -> None:
|
||||
self._update_display()
|
||||
|
||||
def watch_queued_count(self, value: int) -> None:
|
||||
self._update_display()
|
||||
|
||||
def watch_running_count(self, value: int) -> None:
|
||||
self._update_display()
|
||||
|
||||
def watch_is_processing(self, value: bool) -> None:
|
||||
self._update_display()
|
||||
|
||||
def watch_status_message(self, value: str) -> None:
|
||||
self._update_display()
|
||||
|
||||
def _update_display(self) -> None:
|
||||
"""Update the status display."""
|
||||
parts = []
|
||||
|
||||
# App name
|
||||
parts.append(("◉ PraisonAI", "bold cyan"))
|
||||
|
||||
# Session
|
||||
if self.session_id:
|
||||
parts.append((f"Session: {self.session_id[:8]}", ""))
|
||||
|
||||
# Model
|
||||
parts.append((f"Model: {self.model}", ""))
|
||||
|
||||
# Tokens
|
||||
if self.total_tokens > 0:
|
||||
parts.append((f"Tokens: {self.total_tokens:,}", ""))
|
||||
|
||||
# Cost
|
||||
if self.total_cost > 0:
|
||||
parts.append((f"${self.total_cost:.4f}", "green"))
|
||||
|
||||
# Queue status
|
||||
if self.queued_count > 0 or self.running_count > 0:
|
||||
queue_text = f"⏳ {self.queued_count} queued"
|
||||
if self.running_count > 0:
|
||||
queue_text += f" │ ▶ {self.running_count} running"
|
||||
parts.append((queue_text, "yellow"))
|
||||
|
||||
# Processing indicator
|
||||
if self.is_processing:
|
||||
parts.append(("⟳ Processing...", "bold magenta"))
|
||||
|
||||
# Status message
|
||||
if self.status_message:
|
||||
parts.append((self.status_message, "italic"))
|
||||
|
||||
# Build text
|
||||
text = Text()
|
||||
for i, (content, style) in enumerate(parts):
|
||||
if i > 0:
|
||||
text.append(" │ ", style="dim")
|
||||
text.append(content, style=style)
|
||||
|
||||
self.update(text)
|
||||
|
||||
def update_info(self, info: StatusInfo) -> None:
|
||||
"""Update all status info at once."""
|
||||
self.session_id = info.session_id
|
||||
self.model = info.model
|
||||
self.total_tokens = info.total_tokens
|
||||
self.total_cost = info.total_cost
|
||||
self.queued_count = info.queued_count
|
||||
self.running_count = info.running_count
|
||||
self.is_processing = info.is_processing
|
||||
self.status_message = info.status_message
|
||||
|
||||
else:
|
||||
class StatusWidget:
|
||||
"""Placeholder when Textual is not available."""
|
||||
def __init__(self, *args, **kwargs):
|
||||
raise ImportError(
|
||||
"Textual is required for TUI. Install with: pip install praisonai[tui]"
|
||||
)
|
||||
@@ -0,0 +1,248 @@
|
||||
"""
|
||||
Tool Panel Widget for PraisonAI TUI.
|
||||
|
||||
Displays tool execution status and approval dialogs.
|
||||
"""
|
||||
|
||||
from typing import List, Optional
|
||||
from dataclasses import dataclass, field
|
||||
import time
|
||||
|
||||
try:
|
||||
from textual.widget import Widget
|
||||
from textual.widgets import Static, ListView, ListItem
|
||||
from textual.containers import Vertical
|
||||
from textual.reactive import reactive
|
||||
from textual.message import Message
|
||||
from rich.text import Text
|
||||
from rich.panel import Panel
|
||||
TEXTUAL_AVAILABLE = True
|
||||
except ImportError:
|
||||
TEXTUAL_AVAILABLE = False
|
||||
Widget = object
|
||||
Message = object
|
||||
|
||||
|
||||
@dataclass
|
||||
class ToolCall:
|
||||
"""A tool call for display."""
|
||||
call_id: str
|
||||
tool_name: str
|
||||
args_preview: str
|
||||
status: str # "pending", "running", "completed", "failed", "approved", "rejected"
|
||||
result_preview: Optional[str] = None
|
||||
started_at: float = field(default_factory=time.time)
|
||||
ended_at: Optional[float] = None
|
||||
requires_approval: bool = False
|
||||
|
||||
|
||||
if TEXTUAL_AVAILABLE:
|
||||
class ToolPanelWidget(Vertical):
|
||||
"""
|
||||
Tool panel widget.
|
||||
|
||||
Displays:
|
||||
- Recent tool calls
|
||||
- Tool execution status
|
||||
- Approval requests
|
||||
"""
|
||||
|
||||
DEFAULT_CSS = """
|
||||
ToolPanelWidget {
|
||||
height: 100%;
|
||||
border: solid $accent;
|
||||
background: $surface;
|
||||
padding: 0 1;
|
||||
}
|
||||
|
||||
ToolPanelWidget .tool-header {
|
||||
height: 1;
|
||||
background: $accent;
|
||||
padding: 0 1;
|
||||
}
|
||||
|
||||
ToolPanelWidget .tool-pending {
|
||||
color: $warning;
|
||||
}
|
||||
|
||||
ToolPanelWidget .tool-running {
|
||||
color: $primary;
|
||||
}
|
||||
|
||||
ToolPanelWidget .tool-completed {
|
||||
color: $success;
|
||||
}
|
||||
|
||||
ToolPanelWidget .tool-failed {
|
||||
color: $error;
|
||||
}
|
||||
|
||||
ToolPanelWidget .tool-approval {
|
||||
background: $warning-darken-2;
|
||||
border: solid $warning;
|
||||
padding: 1;
|
||||
}
|
||||
"""
|
||||
|
||||
class ApprovalResponse(Message):
|
||||
"""Event when user responds to approval request."""
|
||||
def __init__(self, call_id: str, approved: bool):
|
||||
self.call_id = call_id
|
||||
self.approved = approved
|
||||
super().__init__()
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
max_items: int = 50,
|
||||
name: Optional[str] = None,
|
||||
id: Optional[str] = None,
|
||||
classes: Optional[str] = None,
|
||||
):
|
||||
super().__init__(name=name, id=id, classes=classes)
|
||||
self._calls: List[ToolCall] = []
|
||||
self._max_items = max_items
|
||||
self._pending_approval: Optional[ToolCall] = None
|
||||
|
||||
def compose(self):
|
||||
"""Compose the widget."""
|
||||
yield Static("Tools", id="tool-title", classes="tool-header")
|
||||
yield Vertical(id="tool-list")
|
||||
|
||||
def on_mount(self) -> None:
|
||||
"""Handle mount."""
|
||||
self._update_display()
|
||||
|
||||
def add_call(self, call: ToolCall) -> None:
|
||||
"""Add a tool call."""
|
||||
self._calls.append(call)
|
||||
|
||||
# Trim old calls
|
||||
if len(self._calls) > self._max_items:
|
||||
self._calls = self._calls[-self._max_items:]
|
||||
|
||||
# Check for approval
|
||||
if call.requires_approval and call.status == "pending":
|
||||
self._pending_approval = call
|
||||
|
||||
self._update_display()
|
||||
|
||||
def update_call(self, call_id: str, status: str, result: Optional[str] = None) -> None:
|
||||
"""Update a tool call status."""
|
||||
for call in self._calls:
|
||||
if call.call_id == call_id:
|
||||
call.status = status
|
||||
if result:
|
||||
call.result_preview = result[:100]
|
||||
if status in ("completed", "failed"):
|
||||
call.ended_at = time.time()
|
||||
break
|
||||
|
||||
# Clear pending approval if this was it
|
||||
if self._pending_approval and self._pending_approval.call_id == call_id:
|
||||
self._pending_approval = None
|
||||
|
||||
self._update_display()
|
||||
|
||||
def _update_display(self) -> None:
|
||||
"""Update the display."""
|
||||
try:
|
||||
container = self.query_one("#tool-list", Vertical)
|
||||
except Exception:
|
||||
return
|
||||
|
||||
# Clear existing
|
||||
container.remove_children()
|
||||
|
||||
# Show pending approval first
|
||||
if self._pending_approval:
|
||||
approval_text = Text()
|
||||
approval_text.append("⚠ Approval Required\n", style="bold yellow")
|
||||
approval_text.append(f"Tool: {self._pending_approval.tool_name}\n")
|
||||
approval_text.append(f"Args: {self._pending_approval.args_preview}\n")
|
||||
approval_text.append("[Y] Approve [N] Reject", style="dim")
|
||||
|
||||
container.mount(Static(
|
||||
Panel(approval_text, border_style="yellow"),
|
||||
classes="tool-approval"
|
||||
))
|
||||
|
||||
# Show recent calls
|
||||
for call in reversed(self._calls[-10:]):
|
||||
if call == self._pending_approval:
|
||||
continue
|
||||
|
||||
# Status icon
|
||||
status_icons = {
|
||||
"pending": "⏳",
|
||||
"running": "⟳",
|
||||
"completed": "✓",
|
||||
"failed": "✗",
|
||||
"approved": "✓",
|
||||
"rejected": "✗",
|
||||
}
|
||||
icon = status_icons.get(call.status, "?")
|
||||
|
||||
# Status style
|
||||
status_styles = {
|
||||
"pending": "yellow",
|
||||
"running": "cyan",
|
||||
"completed": "green",
|
||||
"failed": "red",
|
||||
"approved": "green",
|
||||
"rejected": "red",
|
||||
}
|
||||
style = status_styles.get(call.status, "")
|
||||
|
||||
text = Text()
|
||||
text.append(f"{icon} ", style=style)
|
||||
text.append(call.tool_name, style="bold")
|
||||
|
||||
if call.result_preview:
|
||||
text.append(f" → {call.result_preview[:30]}...", style="dim")
|
||||
|
||||
container.mount(Static(text, classes=f"tool-{call.status}"))
|
||||
|
||||
def handle_approval_key(self, key: str) -> bool:
|
||||
"""
|
||||
Handle approval key press.
|
||||
|
||||
Returns True if handled.
|
||||
"""
|
||||
if not self._pending_approval:
|
||||
return False
|
||||
|
||||
if key.lower() == "y":
|
||||
self.post_message(self.ApprovalResponse(
|
||||
self._pending_approval.call_id,
|
||||
approved=True
|
||||
))
|
||||
self.update_call(self._pending_approval.call_id, "approved")
|
||||
return True
|
||||
elif key.lower() == "n":
|
||||
self.post_message(self.ApprovalResponse(
|
||||
self._pending_approval.call_id,
|
||||
approved=False
|
||||
))
|
||||
self.update_call(self._pending_approval.call_id, "rejected")
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
@property
|
||||
def has_pending_approval(self) -> bool:
|
||||
"""Check if there's a pending approval."""
|
||||
return self._pending_approval is not None
|
||||
|
||||
def clear(self) -> None:
|
||||
"""Clear all tool calls."""
|
||||
self._calls.clear()
|
||||
self._pending_approval = None
|
||||
self._update_display()
|
||||
|
||||
else:
|
||||
class ToolPanelWidget:
|
||||
"""Placeholder when Textual is not available."""
|
||||
def __init__(self, *args, **kwargs):
|
||||
raise ImportError(
|
||||
"Textual is required for TUI. Install with: pip install praisonai[tui]"
|
||||
)
|
||||
@@ -0,0 +1,236 @@
|
||||
"""
|
||||
Legacy compatibility adapter for PraisonAI CLI.
|
||||
|
||||
Provides backward compatibility with the original argparse-based CLI.
|
||||
Maps legacy usage patterns to Typer commands or existing feature handlers.
|
||||
"""
|
||||
|
||||
import sys
|
||||
from typing import List, Optional, Tuple
|
||||
|
||||
|
||||
# Legacy commands that should be routed to existing handlers
|
||||
LEGACY_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', 'persistence',
|
||||
}
|
||||
|
||||
# Typer commands that have been implemented
|
||||
TYPER_COMMANDS = {
|
||||
'config', 'traces', 'env', 'session', 'completion', 'version',
|
||||
'debug', 'lsp', 'diag', 'doctor', 'acp', 'mcp', 'serve', 'schedule', 'run',
|
||||
'tui', 'queue', # TUI and queue management commands
|
||||
}
|
||||
|
||||
|
||||
def is_legacy_invocation(argv: List[str]) -> bool:
|
||||
"""
|
||||
Check if the command line represents a legacy invocation.
|
||||
|
||||
Legacy patterns:
|
||||
- praisonai "prompt" (direct prompt)
|
||||
- praisonai agents.yaml (file path)
|
||||
- praisonai --legacy-flag value (legacy-only flags)
|
||||
- praisonai <legacy_command> (commands not yet in Typer)
|
||||
"""
|
||||
if not argv:
|
||||
return False
|
||||
|
||||
# Global Typer options that should NOT trigger legacy mode
|
||||
typer_global_options = {
|
||||
'--json', '--output-format', '-o', '--no-color', '--quiet', '-q',
|
||||
'--verbose', '-v', '--screen-reader', '--help', '-h', '--version', '-V',
|
||||
}
|
||||
|
||||
# Filter out global options to find the actual command
|
||||
filtered_argv = []
|
||||
skip_next = False
|
||||
for i, arg in enumerate(argv):
|
||||
if skip_next:
|
||||
skip_next = False
|
||||
continue
|
||||
if arg in typer_global_options:
|
||||
# Skip options that take a value
|
||||
if arg in ('--output-format', '-o'):
|
||||
skip_next = True
|
||||
continue
|
||||
filtered_argv.append(arg)
|
||||
|
||||
first_arg = filtered_argv[0] if filtered_argv else ""
|
||||
|
||||
# Check for legacy-only flags (not in Typer)
|
||||
legacy_flags = [
|
||||
'--framework', '--ui', '--auto', '--init', '--deploy', '--schedule',
|
||||
'--provider', '--model', '--llm', '--hf', '--ollama', '--dataset',
|
||||
'--realtime', '--call', '--public', '--merge', '--claudecode',
|
||||
'--file', '--url', '--goal', '--auto-analyze', '--research',
|
||||
'--query-rewrite', '--expand-prompt', '--tools', '--no-tools',
|
||||
'--save', '--web-search', '--web-fetch', '--prompt-caching',
|
||||
'--planning', '--memory', '--user-id', '--auto-save', '--history',
|
||||
'--workflow', '--guardrail', '--metrics', '--image', '--telemetry',
|
||||
'--mcp', '--fast-context', '--handoff', '--auto-memory', '--todo',
|
||||
'--router', '--flow-display', '--n8n', '--serve', '--port', '--host',
|
||||
'--interactive', '--chat-mode', '-p', '--autonomy', '--trust',
|
||||
'--sandbox', '--external-agent', '--compare', '--interval', '--timeout',
|
||||
'--max-cost', '--rpm', '--tpm', '--temperature',
|
||||
]
|
||||
|
||||
for arg in argv:
|
||||
if arg in legacy_flags:
|
||||
return True
|
||||
|
||||
# Check if first arg is a legacy command not in Typer
|
||||
if first_arg in LEGACY_COMMANDS and first_arg not in TYPER_COMMANDS:
|
||||
return True
|
||||
|
||||
# Check if first arg looks like a prompt (not a command, not a flag)
|
||||
if first_arg and not first_arg.startswith('-'):
|
||||
# If it's not a known Typer command, treat as legacy
|
||||
if first_arg not in TYPER_COMMANDS:
|
||||
# Could be a file path or prompt
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
|
||||
def route_to_legacy(argv: List[str]) -> int:
|
||||
"""
|
||||
Route to the legacy argparse CLI.
|
||||
|
||||
Returns:
|
||||
Exit code from legacy CLI
|
||||
"""
|
||||
from praisonai.cli.main import PraisonAI
|
||||
|
||||
# Restore sys.argv for argparse
|
||||
original_argv = sys.argv
|
||||
sys.argv = ['praisonai'] + argv
|
||||
|
||||
try:
|
||||
praison = PraisonAI()
|
||||
result = praison.main()
|
||||
return 0 if result is None else (1 if result is False else 0)
|
||||
except SystemExit as e:
|
||||
return e.code if isinstance(e.code, int) else 0
|
||||
finally:
|
||||
sys.argv = original_argv
|
||||
|
||||
|
||||
def translate_to_typer(argv: List[str]) -> Tuple[List[str], bool]:
|
||||
"""
|
||||
Attempt to translate legacy argv to Typer format.
|
||||
|
||||
Returns:
|
||||
Tuple of (translated_argv, was_translated)
|
||||
"""
|
||||
if not argv:
|
||||
return argv, False
|
||||
|
||||
first_arg = argv[0]
|
||||
|
||||
# Direct mapping for commands that exist in both
|
||||
if first_arg in TYPER_COMMANDS:
|
||||
return argv, False # Already Typer-compatible
|
||||
|
||||
# Translate common patterns
|
||||
translations = {
|
||||
# Legacy flag patterns
|
||||
'--version': ['version', 'show'],
|
||||
'-V': ['version', 'show'],
|
||||
}
|
||||
|
||||
if first_arg in translations:
|
||||
return translations[first_arg] + argv[1:], True
|
||||
|
||||
return argv, False
|
||||
|
||||
|
||||
def main_with_legacy_support():
|
||||
"""
|
||||
Main entry point with legacy support.
|
||||
|
||||
Routes to Typer CLI for new commands, legacy argparse CLI for old commands.
|
||||
"""
|
||||
argv = sys.argv[1:]
|
||||
|
||||
# Handle empty args - show help via Typer
|
||||
if not argv:
|
||||
try:
|
||||
from praisonai.cli.app import app
|
||||
app()
|
||||
return
|
||||
except SystemExit:
|
||||
return
|
||||
|
||||
# Check for version flags first (handle specially)
|
||||
if '--version' in argv or '-V' in argv:
|
||||
from praisonai.version import __version__
|
||||
print(f"PraisonAI version {__version__}")
|
||||
return
|
||||
|
||||
# IMPORTANT: Check for legacy invocation FIRST before trying Typer
|
||||
# This ensures legacy commands like 'chat' and flags like '--framework' work
|
||||
if is_legacy_invocation(argv):
|
||||
exit_code = route_to_legacy(argv)
|
||||
sys.exit(exit_code)
|
||||
|
||||
# Global Typer options that should NOT trigger legacy mode
|
||||
typer_global_options = {
|
||||
'--json', '--output-format', '-o', '--no-color', '--quiet', '-q',
|
||||
'--verbose', '-v', '--screen-reader', '--help', '-h',
|
||||
}
|
||||
|
||||
# Find the actual command (skip global options)
|
||||
actual_command = None
|
||||
skip_next = False
|
||||
for arg in argv:
|
||||
if skip_next:
|
||||
skip_next = False
|
||||
continue
|
||||
if arg in typer_global_options:
|
||||
if arg in ('--output-format', '-o'):
|
||||
skip_next = True
|
||||
continue
|
||||
if not arg.startswith('-'):
|
||||
actual_command = arg
|
||||
break
|
||||
|
||||
# Route to Typer for known Typer commands
|
||||
if actual_command in TYPER_COMMANDS:
|
||||
try:
|
||||
from praisonai.cli.app import app
|
||||
sys.argv = ['praisonai'] + argv
|
||||
app()
|
||||
return
|
||||
except SystemExit as e:
|
||||
sys.exit(e.code if isinstance(e.code, int) else 0)
|
||||
|
||||
# Check for help flags - route to Typer for help
|
||||
if '--help' in argv or '-h' in argv:
|
||||
try:
|
||||
from praisonai.cli.app import app
|
||||
sys.argv = ['praisonai'] + argv
|
||||
app()
|
||||
return
|
||||
except SystemExit:
|
||||
return
|
||||
|
||||
# Default: try Typer, fall back to legacy on error
|
||||
try:
|
||||
from praisonai.cli.app import app
|
||||
sys.argv = ['praisonai'] + argv
|
||||
app()
|
||||
except SystemExit as e:
|
||||
sys.exit(e.code if isinstance(e.code, int) else 0)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main_with_legacy_support()
|
||||
@@ -0,0 +1,31 @@
|
||||
"""
|
||||
PraisonAI CLI State Module.
|
||||
|
||||
Provides state management for run IDs, trace IDs, agent IDs, and sessions.
|
||||
"""
|
||||
|
||||
from .identifiers import (
|
||||
generate_run_id,
|
||||
generate_trace_id,
|
||||
generate_agent_id,
|
||||
RunContext,
|
||||
get_current_context,
|
||||
set_current_context,
|
||||
)
|
||||
from .sessions import (
|
||||
SessionManager,
|
||||
SessionMetadata,
|
||||
get_session_manager,
|
||||
)
|
||||
|
||||
__all__ = [
|
||||
'generate_run_id',
|
||||
'generate_trace_id',
|
||||
'generate_agent_id',
|
||||
'RunContext',
|
||||
'get_current_context',
|
||||
'set_current_context',
|
||||
'SessionManager',
|
||||
'SessionMetadata',
|
||||
'get_session_manager',
|
||||
]
|
||||
@@ -0,0 +1,161 @@
|
||||
"""
|
||||
Identifier generation for PraisonAI CLI.
|
||||
|
||||
Provides stable, unique identifiers for runs, traces, and agents.
|
||||
"""
|
||||
|
||||
import hashlib
|
||||
import os
|
||||
import secrets
|
||||
import threading
|
||||
from dataclasses import dataclass, field
|
||||
from datetime import datetime
|
||||
from typing import Dict, Optional
|
||||
|
||||
|
||||
def generate_run_id() -> str:
|
||||
"""
|
||||
Generate a unique run ID.
|
||||
|
||||
Format: run_YYYYMMDD_HHMMSS_<random>
|
||||
Example: run_20241231_143022_a1b2c3
|
||||
"""
|
||||
timestamp = datetime.utcnow().strftime("%Y%m%d_%H%M%S")
|
||||
random_suffix = secrets.token_hex(3) # 6 characters
|
||||
return f"run_{timestamp}_{random_suffix}"
|
||||
|
||||
|
||||
def generate_trace_id() -> str:
|
||||
"""
|
||||
Generate a unique trace ID.
|
||||
|
||||
Format: trace_<random>
|
||||
Example: trace_a1b2c3d4e5f6
|
||||
"""
|
||||
return f"trace_{secrets.token_hex(6)}"
|
||||
|
||||
|
||||
def generate_agent_id(agent_name: str, run_id: str, index: int = 0) -> str:
|
||||
"""
|
||||
Generate a deterministic agent ID.
|
||||
|
||||
The ID is deterministic based on agent name, run ID, and index,
|
||||
ensuring consistent IDs across the same run.
|
||||
|
||||
Format: agent_<hash>_<index>
|
||||
Example: agent_a1b2c3_001
|
||||
|
||||
Args:
|
||||
agent_name: Name of the agent
|
||||
run_id: Current run ID
|
||||
index: Agent index within the run
|
||||
|
||||
Returns:
|
||||
Deterministic agent ID
|
||||
"""
|
||||
# Create deterministic hash from agent name and run ID
|
||||
hash_input = f"{agent_name}:{run_id}:{index}"
|
||||
hash_value = hashlib.sha256(hash_input.encode()).hexdigest()[:6]
|
||||
return f"agent_{hash_value}_{index:03d}"
|
||||
|
||||
|
||||
@dataclass
|
||||
class RunContext:
|
||||
"""
|
||||
Context for a single run.
|
||||
|
||||
Holds all identifiers and metadata for the current execution.
|
||||
"""
|
||||
run_id: str = field(default_factory=generate_run_id)
|
||||
trace_id: str = field(default_factory=generate_trace_id)
|
||||
start_time: datetime = field(default_factory=datetime.utcnow)
|
||||
|
||||
# Agent tracking
|
||||
agents: Dict[str, str] = field(default_factory=dict) # name -> agent_id
|
||||
_agent_counter: int = field(default=0, repr=False)
|
||||
|
||||
# Metadata
|
||||
config_summary: Dict[str, str] = field(default_factory=dict)
|
||||
workspace: Optional[str] = None
|
||||
|
||||
def get_agent_id(self, agent_name: str) -> str:
|
||||
"""
|
||||
Get or create an agent ID for the given agent name.
|
||||
|
||||
Args:
|
||||
agent_name: Name of the agent
|
||||
|
||||
Returns:
|
||||
Agent ID (creates new one if not exists)
|
||||
"""
|
||||
if agent_name not in self.agents:
|
||||
agent_id = generate_agent_id(agent_name, self.run_id, self._agent_counter)
|
||||
self.agents[agent_name] = agent_id
|
||||
self._agent_counter += 1
|
||||
return self.agents[agent_name]
|
||||
|
||||
def to_dict(self) -> Dict:
|
||||
"""Convert to dictionary for serialization."""
|
||||
return {
|
||||
"run_id": self.run_id,
|
||||
"trace_id": self.trace_id,
|
||||
"start_time": self.start_time.isoformat(),
|
||||
"agents": self.agents.copy(),
|
||||
"config_summary": self.config_summary.copy(),
|
||||
"workspace": self.workspace,
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def from_dict(cls, data: Dict) -> "RunContext":
|
||||
"""Create from dictionary."""
|
||||
ctx = cls(
|
||||
run_id=data.get("run_id", generate_run_id()),
|
||||
trace_id=data.get("trace_id", generate_trace_id()),
|
||||
workspace=data.get("workspace"),
|
||||
)
|
||||
ctx.agents = data.get("agents", {})
|
||||
ctx.config_summary = data.get("config_summary", {})
|
||||
ctx._agent_counter = len(ctx.agents)
|
||||
|
||||
start_time = data.get("start_time")
|
||||
if start_time:
|
||||
try:
|
||||
ctx.start_time = datetime.fromisoformat(start_time)
|
||||
except (ValueError, TypeError):
|
||||
pass
|
||||
|
||||
return ctx
|
||||
|
||||
|
||||
# Thread-local storage for current context
|
||||
_context_local = threading.local()
|
||||
|
||||
|
||||
def get_current_context() -> Optional[RunContext]:
|
||||
"""Get the current run context for this thread."""
|
||||
return getattr(_context_local, "context", None)
|
||||
|
||||
|
||||
def set_current_context(context: Optional[RunContext]) -> None:
|
||||
"""Set the current run context for this thread."""
|
||||
_context_local.context = context
|
||||
|
||||
|
||||
def create_context(workspace: Optional[str] = None) -> RunContext:
|
||||
"""
|
||||
Create a new run context and set it as current.
|
||||
|
||||
Args:
|
||||
workspace: Optional workspace path
|
||||
|
||||
Returns:
|
||||
New RunContext instance
|
||||
"""
|
||||
context = RunContext(workspace=workspace or os.getcwd())
|
||||
set_current_context(context)
|
||||
return context
|
||||
|
||||
|
||||
def clear_context() -> None:
|
||||
"""Clear the current run context."""
|
||||
set_current_context(None)
|
||||
@@ -0,0 +1,313 @@
|
||||
"""
|
||||
Session management for PraisonAI CLI.
|
||||
|
||||
Provides persistence and resume functionality for sessions.
|
||||
"""
|
||||
|
||||
import json
|
||||
import os
|
||||
import shutil
|
||||
from dataclasses import dataclass, field
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
from ..configuration.paths import get_sessions_dir, ensure_config_dirs
|
||||
from .identifiers import RunContext
|
||||
|
||||
|
||||
@dataclass
|
||||
class SessionMetadata:
|
||||
"""Metadata for a session."""
|
||||
session_id: str
|
||||
run_id: str
|
||||
trace_id: str
|
||||
created_at: datetime
|
||||
updated_at: datetime
|
||||
name: Optional[str] = None
|
||||
workspace: Optional[str] = None
|
||||
config_summary: Dict[str, Any] = field(default_factory=dict)
|
||||
event_count: int = 0
|
||||
status: str = "active" # active, completed, error
|
||||
|
||||
def to_dict(self) -> Dict[str, Any]:
|
||||
"""Convert to dictionary."""
|
||||
return {
|
||||
"session_id": self.session_id,
|
||||
"run_id": self.run_id,
|
||||
"trace_id": self.trace_id,
|
||||
"created_at": self.created_at.isoformat(),
|
||||
"updated_at": self.updated_at.isoformat(),
|
||||
"name": self.name,
|
||||
"workspace": self.workspace,
|
||||
"config_summary": self.config_summary,
|
||||
"event_count": self.event_count,
|
||||
"status": self.status,
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def from_dict(cls, data: Dict[str, Any]) -> "SessionMetadata":
|
||||
"""Create from dictionary."""
|
||||
return cls(
|
||||
session_id=data["session_id"],
|
||||
run_id=data["run_id"],
|
||||
trace_id=data["trace_id"],
|
||||
created_at=datetime.fromisoformat(data["created_at"]),
|
||||
updated_at=datetime.fromisoformat(data["updated_at"]),
|
||||
name=data.get("name"),
|
||||
workspace=data.get("workspace"),
|
||||
config_summary=data.get("config_summary", {}),
|
||||
event_count=data.get("event_count", 0),
|
||||
status=data.get("status", "active"),
|
||||
)
|
||||
|
||||
|
||||
class SessionManager:
|
||||
"""
|
||||
Manages session persistence and retrieval.
|
||||
|
||||
Sessions are stored in ~/.praison/sessions/<session_id>/
|
||||
Each session contains:
|
||||
- metadata.json: Session metadata
|
||||
- events.jsonl: Stream of events (NDJSON)
|
||||
"""
|
||||
|
||||
def __init__(self, sessions_dir: Optional[Path] = None):
|
||||
self.sessions_dir = sessions_dir or get_sessions_dir()
|
||||
ensure_config_dirs()
|
||||
|
||||
def _get_session_dir(self, session_id: str) -> Path:
|
||||
"""Get the directory for a session."""
|
||||
return self.sessions_dir / session_id
|
||||
|
||||
def _get_metadata_path(self, session_id: str) -> Path:
|
||||
"""Get the metadata file path for a session."""
|
||||
return self._get_session_dir(session_id) / "metadata.json"
|
||||
|
||||
def _get_events_path(self, session_id: str) -> Path:
|
||||
"""Get the events file path for a session."""
|
||||
return self._get_session_dir(session_id) / "events.jsonl"
|
||||
|
||||
def create(self, context: RunContext, name: Optional[str] = None) -> SessionMetadata:
|
||||
"""
|
||||
Create a new session from a run context.
|
||||
|
||||
Args:
|
||||
context: Run context to create session from
|
||||
name: Optional session name
|
||||
|
||||
Returns:
|
||||
Session metadata
|
||||
"""
|
||||
session_id = context.run_id # Use run_id as session_id
|
||||
session_dir = self._get_session_dir(session_id)
|
||||
session_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
now = datetime.utcnow()
|
||||
metadata = SessionMetadata(
|
||||
session_id=session_id,
|
||||
run_id=context.run_id,
|
||||
trace_id=context.trace_id,
|
||||
created_at=now,
|
||||
updated_at=now,
|
||||
name=name,
|
||||
workspace=context.workspace,
|
||||
config_summary=context.config_summary,
|
||||
)
|
||||
|
||||
# Save metadata
|
||||
self._save_metadata(metadata)
|
||||
|
||||
# Create empty events file
|
||||
self._get_events_path(session_id).touch()
|
||||
|
||||
return metadata
|
||||
|
||||
def _save_metadata(self, metadata: SessionMetadata) -> None:
|
||||
"""Save session metadata."""
|
||||
path = self._get_metadata_path(metadata.session_id)
|
||||
path.parent.mkdir(parents=True, exist_ok=True)
|
||||
with open(path, "w", encoding="utf-8") as f:
|
||||
json.dump(metadata.to_dict(), f, indent=2)
|
||||
|
||||
def _load_metadata(self, session_id: str) -> Optional[SessionMetadata]:
|
||||
"""Load session metadata."""
|
||||
path = self._get_metadata_path(session_id)
|
||||
if not path.exists():
|
||||
return None
|
||||
try:
|
||||
with open(path, "r", encoding="utf-8") as f:
|
||||
data = json.load(f)
|
||||
return SessionMetadata.from_dict(data)
|
||||
except (json.JSONDecodeError, KeyError):
|
||||
return None
|
||||
|
||||
def append_event(self, session_id: str, event: Dict[str, Any]) -> None:
|
||||
"""
|
||||
Append an event to a session.
|
||||
|
||||
Args:
|
||||
session_id: Session ID
|
||||
event: Event data to append
|
||||
"""
|
||||
events_path = self._get_events_path(session_id)
|
||||
if not events_path.parent.exists():
|
||||
return
|
||||
|
||||
with open(events_path, "a", encoding="utf-8") as f:
|
||||
f.write(json.dumps(event, default=str) + "\n")
|
||||
|
||||
# Update metadata
|
||||
metadata = self._load_metadata(session_id)
|
||||
if metadata:
|
||||
metadata.event_count += 1
|
||||
metadata.updated_at = datetime.utcnow()
|
||||
self._save_metadata(metadata)
|
||||
|
||||
def get(self, session_id: str) -> Optional[SessionMetadata]:
|
||||
"""Get session metadata by ID."""
|
||||
return self._load_metadata(session_id)
|
||||
|
||||
def list(self, limit: int = 50) -> List[SessionMetadata]:
|
||||
"""
|
||||
List all sessions, sorted by updated_at descending.
|
||||
|
||||
Args:
|
||||
limit: Maximum number of sessions to return
|
||||
|
||||
Returns:
|
||||
List of session metadata
|
||||
"""
|
||||
sessions = []
|
||||
|
||||
if not self.sessions_dir.exists():
|
||||
return sessions
|
||||
|
||||
for session_dir in self.sessions_dir.iterdir():
|
||||
if session_dir.is_dir():
|
||||
metadata = self._load_metadata(session_dir.name)
|
||||
if metadata:
|
||||
sessions.append(metadata)
|
||||
|
||||
# Sort by updated_at descending
|
||||
sessions.sort(key=lambda s: s.updated_at, reverse=True)
|
||||
|
||||
return sessions[:limit]
|
||||
|
||||
def delete(self, session_id: str) -> bool:
|
||||
"""
|
||||
Delete a session.
|
||||
|
||||
Args:
|
||||
session_id: Session ID to delete
|
||||
|
||||
Returns:
|
||||
True if deleted, False if not found
|
||||
"""
|
||||
session_dir = self._get_session_dir(session_id)
|
||||
if not session_dir.exists():
|
||||
return False
|
||||
|
||||
shutil.rmtree(session_dir)
|
||||
return True
|
||||
|
||||
def get_events(self, session_id: str) -> List[Dict[str, Any]]:
|
||||
"""
|
||||
Get all events for a session.
|
||||
|
||||
Args:
|
||||
session_id: Session ID
|
||||
|
||||
Returns:
|
||||
List of events
|
||||
"""
|
||||
events_path = self._get_events_path(session_id)
|
||||
if not events_path.exists():
|
||||
return []
|
||||
|
||||
events = []
|
||||
with open(events_path, "r", encoding="utf-8") as f:
|
||||
for line in f:
|
||||
line = line.strip()
|
||||
if line:
|
||||
try:
|
||||
events.append(json.loads(line))
|
||||
except json.JSONDecodeError:
|
||||
pass
|
||||
|
||||
return events
|
||||
|
||||
def export(self, session_id: str, format: str = "md") -> Optional[str]:
|
||||
"""
|
||||
Export a session to a string.
|
||||
|
||||
Args:
|
||||
session_id: Session ID
|
||||
format: Export format ("md" or "json")
|
||||
|
||||
Returns:
|
||||
Exported content or None if not found
|
||||
"""
|
||||
metadata = self._load_metadata(session_id)
|
||||
if not metadata:
|
||||
return None
|
||||
|
||||
events = self.get_events(session_id)
|
||||
|
||||
if format == "json":
|
||||
return json.dumps({
|
||||
"metadata": metadata.to_dict(),
|
||||
"events": events,
|
||||
}, indent=2, default=str)
|
||||
|
||||
# Markdown format
|
||||
lines = [
|
||||
f"# Session: {metadata.name or metadata.session_id}",
|
||||
"",
|
||||
f"- **Run ID**: {metadata.run_id}",
|
||||
f"- **Trace ID**: {metadata.trace_id}",
|
||||
f"- **Created**: {metadata.created_at.isoformat()}",
|
||||
f"- **Status**: {metadata.status}",
|
||||
"",
|
||||
"## Events",
|
||||
"",
|
||||
]
|
||||
|
||||
for event in events:
|
||||
event_type = event.get("event", "unknown")
|
||||
timestamp = event.get("timestamp", "")
|
||||
message = event.get("message", "")
|
||||
|
||||
lines.append(f"### {event_type} ({timestamp})")
|
||||
if message:
|
||||
lines.append(f"\n{message}")
|
||||
|
||||
data = event.get("data", {})
|
||||
if data:
|
||||
lines.append("\n```json")
|
||||
lines.append(json.dumps(data, indent=2, default=str))
|
||||
lines.append("```")
|
||||
|
||||
lines.append("")
|
||||
|
||||
return "\n".join(lines)
|
||||
|
||||
def update_status(self, session_id: str, status: str) -> None:
|
||||
"""Update session status."""
|
||||
metadata = self._load_metadata(session_id)
|
||||
if metadata:
|
||||
metadata.status = status
|
||||
metadata.updated_at = datetime.utcnow()
|
||||
self._save_metadata(metadata)
|
||||
|
||||
|
||||
# Global session manager
|
||||
_session_manager: Optional[SessionManager] = None
|
||||
|
||||
|
||||
def get_session_manager() -> SessionManager:
|
||||
"""Get the global session manager."""
|
||||
global _session_manager
|
||||
if _session_manager is None:
|
||||
_session_manager = SessionManager()
|
||||
return _session_manager
|
||||
@@ -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.9.0 gunicorn markdown\n")
|
||||
file.write("RUN pip install flask praisonai==2.9.2 gunicorn markdown\n")
|
||||
file.write("EXPOSE 8080\n")
|
||||
file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
|
||||
|
||||
|
||||
@@ -1 +1 @@
|
||||
__version__ = "2.9.0"
|
||||
__version__ = "2.9.2"
|
||||
@@ -12,11 +12,13 @@ dependencies = [
|
||||
"rich>=13.7",
|
||||
"markdown>=3.5",
|
||||
"pyparsing>=3.0.0",
|
||||
"praisonaiagents>=0.5.2",
|
||||
"praisonaiagents>=0.5.3",
|
||||
"python-dotenv>=0.19.0",
|
||||
"instructor>=1.3.3",
|
||||
"PyYAML>=6.0",
|
||||
"mcp>=1.6.0",
|
||||
"typer>=0.9.0",
|
||||
"textual>=0.47.0",
|
||||
]
|
||||
|
||||
[project.scripts]
|
||||
@@ -103,6 +105,15 @@ autogen-v4 = [
|
||||
acp = [
|
||||
"agent-client-protocol>=0.7.0",
|
||||
]
|
||||
lite = [
|
||||
# Minimal installation without TUI, litellm - just core agent functionality
|
||||
# Use: pip install praisonai[lite] for lightweight deployments
|
||||
]
|
||||
dev = [
|
||||
"textual-dev>=1.0.0",
|
||||
"pytest>=7.0.0",
|
||||
"pytest-asyncio>=0.21.0",
|
||||
]
|
||||
|
||||
[project.urls]
|
||||
Homepage = "https://docs.praison.ai"
|
||||
|
||||
+187
@@ -0,0 +1,187 @@
|
||||
#!/bin/bash
|
||||
# PraisonAI TUI Real-Key Verification Script
|
||||
#
|
||||
# This script verifies the TUI and Queue system works with real API keys.
|
||||
# It includes safety gates and secret masking.
|
||||
#
|
||||
# Usage:
|
||||
# ./scripts/verify_tui_real_keys.sh
|
||||
#
|
||||
# Required environment variables:
|
||||
# OPENAI_API_KEY - OpenAI API key
|
||||
#
|
||||
# Optional:
|
||||
# PRAISONAI_REAL_LLM=1 - Required to run real LLM tests
|
||||
|
||||
set -e
|
||||
|
||||
# Colors
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
|
||||
echo " PraisonAI TUI + Queue System Verification"
|
||||
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
|
||||
|
||||
# Function to mask secrets in output
|
||||
mask_secrets() {
|
||||
sed -E 's/(sk-[a-zA-Z0-9]{20})[a-zA-Z0-9]+/\1***/g' | \
|
||||
sed -E 's/(key_[a-zA-Z0-9]{20})[a-zA-Z0-9]+/\1***/g' | \
|
||||
sed -E 's/(tvly-[a-zA-Z0-9]{20})[a-zA-Z0-9]+/\1***/g' | \
|
||||
sed -E 's/(AIza[a-zA-Z0-9]{20})[a-zA-Z0-9]+/\1***/g'
|
||||
}
|
||||
|
||||
# Check environment variables
|
||||
echo -e "\n${YELLOW}[1/7] Checking environment variables...${NC}"
|
||||
|
||||
check_env() {
|
||||
local var_name=$1
|
||||
local value="${!var_name}"
|
||||
if [ -n "$value" ]; then
|
||||
local masked=$(echo "$value" | head -c 10)
|
||||
echo -e " ${GREEN}✓${NC} $var_name: ${masked}***"
|
||||
return 0
|
||||
else
|
||||
echo -e " ${YELLOW}⚠${NC} $var_name: not set"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
OPENAI_SET=false
|
||||
check_env "OPENAI_API_KEY" && OPENAI_SET=true
|
||||
check_env "ANTHROPIC_API_KEY" || true
|
||||
check_env "GOOGLE_API_KEY" || true
|
||||
check_env "TAVILY_API_KEY" || true
|
||||
|
||||
if [ "$OPENAI_SET" = false ]; then
|
||||
echo -e "\n${RED}Error: OPENAI_API_KEY is required${NC}"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Check Python imports
|
||||
echo -e "\n${YELLOW}[2/7] Checking Python imports...${NC}"
|
||||
|
||||
python3 -c "
|
||||
from praisonai.cli.features.queue import QueuedRun, RunState, RunPriority, QueueConfig
|
||||
from praisonai.cli.features.queue import QueueScheduler, QueuePersistence, QueueManager
|
||||
from praisonai.cli.features.tui import TUIEvent, TUIEventType
|
||||
from praisonai.cli.features.tui import TuiOrchestrator, UIStateModel, SimulationRunner
|
||||
from praisonai.cli.features.tui import MockProvider, MockProviderConfig
|
||||
print(' ✓ All imports successful')
|
||||
" 2>&1 | mask_secrets
|
||||
|
||||
# Run unit tests
|
||||
echo -e "\n${YELLOW}[3/7] Running unit tests...${NC}"
|
||||
|
||||
cd "$(dirname "$0")/.."
|
||||
python3 -m pytest tests/unit/cli/queue/ tests/unit/cli/tui/ -v --tb=short --import-mode=importlib 2>&1 | \
|
||||
tail -20 | mask_secrets
|
||||
|
||||
# Test queue CLI commands
|
||||
echo -e "\n${YELLOW}[4/7] Testing queue CLI commands...${NC}"
|
||||
|
||||
python3 -c "
|
||||
from praisonai.cli.features.tui.cli import create_queue_app
|
||||
app = create_queue_app()
|
||||
print(' ✓ Queue CLI app created')
|
||||
print(' Commands:', [cmd.name for cmd in app.registered_commands])
|
||||
" 2>&1 | mask_secrets
|
||||
|
||||
# Test mock provider
|
||||
echo -e "\n${YELLOW}[5/7] Testing mock provider...${NC}"
|
||||
|
||||
python3 -c "
|
||||
import asyncio
|
||||
from praisonai.cli.features.tui.mock_provider import MockProvider, create_mock_provider
|
||||
|
||||
async def test():
|
||||
provider = create_mock_provider(seed=42)
|
||||
|
||||
# Test streaming
|
||||
chunks = []
|
||||
result = await provider.generate(
|
||||
'hello world',
|
||||
stream=True,
|
||||
on_chunk=lambda c: chunks.append(c)
|
||||
)
|
||||
|
||||
print(' ✓ Mock provider works')
|
||||
print(f' Content length: {len(result[\"content\"])} chars')
|
||||
print(f' Chunks: {len(chunks)}')
|
||||
print(f' Tokens: {result[\"tokens\"]}')
|
||||
print(f' Cost: \${result[\"cost\"]:.6f}')
|
||||
|
||||
asyncio.run(test())
|
||||
" 2>&1 | mask_secrets
|
||||
|
||||
# Test orchestrator
|
||||
echo -e "\n${YELLOW}[6/7] Testing TUI orchestrator...${NC}"
|
||||
|
||||
python3 -c "
|
||||
import asyncio
|
||||
from praisonai.cli.features.tui.orchestrator import TuiOrchestrator, OutputMode
|
||||
from praisonai.cli.features.queue import QueueConfig
|
||||
|
||||
async def test():
|
||||
config = QueueConfig(enable_persistence=False)
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
debug=False,
|
||||
)
|
||||
|
||||
await orchestrator.start(session_id='test-verify')
|
||||
|
||||
# Get snapshot
|
||||
snapshot = orchestrator.get_snapshot()
|
||||
print(' ✓ Orchestrator started')
|
||||
print(f' Session: {snapshot[\"session_id\"]}')
|
||||
print(f' Model: {snapshot[\"model\"]}')
|
||||
|
||||
await orchestrator.stop()
|
||||
print(' ✓ Orchestrator stopped cleanly')
|
||||
|
||||
asyncio.run(test())
|
||||
" 2>&1 | mask_secrets
|
||||
|
||||
# Real LLM test (gated)
|
||||
echo -e "\n${YELLOW}[7/7] Real LLM test...${NC}"
|
||||
|
||||
if [ "$PRAISONAI_REAL_LLM" = "1" ]; then
|
||||
echo " Running real LLM test (PRAISONAI_REAL_LLM=1)"
|
||||
|
||||
python3 -c "
|
||||
import asyncio
|
||||
import os
|
||||
|
||||
# Safety: Use minimal tokens
|
||||
async def test():
|
||||
from praisonaiagents import Agent
|
||||
|
||||
agent = Agent(
|
||||
instructions='Reply with exactly one word.',
|
||||
model='gpt-4o-mini',
|
||||
verbose=False,
|
||||
)
|
||||
|
||||
result = agent.start('Say hello')
|
||||
print(f' ✓ Real LLM response: {result[:50]}...')
|
||||
|
||||
asyncio.run(test())
|
||||
" 2>&1 | mask_secrets
|
||||
else
|
||||
echo -e " ${YELLOW}⚠${NC} Skipped (set PRAISONAI_REAL_LLM=1 to enable)"
|
||||
fi
|
||||
|
||||
# Summary
|
||||
echo -e "\n━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
|
||||
echo -e "${GREEN} Verification Complete!${NC}"
|
||||
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
|
||||
echo ""
|
||||
echo "All checks passed. The TUI + Queue system is working correctly."
|
||||
echo ""
|
||||
echo "To run with real LLM:"
|
||||
echo " PRAISONAI_REAL_LLM=1 ./scripts/verify_tui_real_keys.sh"
|
||||
echo ""
|
||||
@@ -0,0 +1 @@
|
||||
"""Unit tests for PraisonAI CLI."""
|
||||
@@ -0,0 +1,414 @@
|
||||
"""
|
||||
Tests for PraisonAI Typer CLI.
|
||||
|
||||
Tests the new modular Typer-based CLI structure.
|
||||
"""
|
||||
|
||||
import json
|
||||
import os
|
||||
import pytest
|
||||
from pathlib import Path
|
||||
from typer.testing import CliRunner
|
||||
from unittest.mock import patch, MagicMock
|
||||
|
||||
|
||||
# Create CLI runner
|
||||
runner = CliRunner()
|
||||
|
||||
|
||||
class TestTyperApp:
|
||||
"""Tests for the main Typer app."""
|
||||
|
||||
def test_app_import(self):
|
||||
"""Test that the Typer app can be imported."""
|
||||
from praisonai.cli.app import app
|
||||
assert app is not None
|
||||
|
||||
def test_help_output(self):
|
||||
"""Test --help shows expected content."""
|
||||
from praisonai.cli.app import app
|
||||
result = runner.invoke(app, ["--help"])
|
||||
assert result.exit_code == 0
|
||||
assert "PraisonAI" in result.output
|
||||
assert "config" in result.output
|
||||
assert "traces" in result.output
|
||||
assert "session" in result.output
|
||||
|
||||
def test_version_flag(self):
|
||||
"""Test --version flag."""
|
||||
from praisonai.cli.app import app
|
||||
result = runner.invoke(app, ["--version"])
|
||||
assert result.exit_code == 0
|
||||
assert "PraisonAI version" in result.output
|
||||
|
||||
|
||||
class TestConfigCommand:
|
||||
"""Tests for config command group."""
|
||||
|
||||
def test_config_list(self):
|
||||
"""Test config list command."""
|
||||
from praisonai.cli.commands.config import app
|
||||
result = runner.invoke(app, ["list"])
|
||||
assert result.exit_code == 0
|
||||
|
||||
def test_config_get_missing_key(self):
|
||||
"""Test config get with missing key."""
|
||||
from praisonai.cli.commands.config import app
|
||||
result = runner.invoke(app, ["get", "nonexistent.key"])
|
||||
assert result.exit_code == 1
|
||||
assert "not found" in result.output.lower()
|
||||
|
||||
def test_config_path(self):
|
||||
"""Test config path command."""
|
||||
from praisonai.cli.commands.config import app
|
||||
result = runner.invoke(app, ["path"])
|
||||
assert result.exit_code == 0
|
||||
assert "config.toml" in result.output
|
||||
|
||||
|
||||
class TestTracesCommand:
|
||||
"""Tests for traces command group."""
|
||||
|
||||
def test_traces_status(self):
|
||||
"""Test traces status command."""
|
||||
from praisonai.cli.commands.traces import app
|
||||
result = runner.invoke(app, ["status"])
|
||||
assert result.exit_code == 0
|
||||
assert "Trace" in result.output or "trace" in result.output.lower()
|
||||
|
||||
def test_traces_enable(self):
|
||||
"""Test traces enable command."""
|
||||
from praisonai.cli.commands.traces import app
|
||||
result = runner.invoke(app, ["enable"])
|
||||
assert result.exit_code == 0
|
||||
|
||||
|
||||
class TestEnvCommand:
|
||||
"""Tests for env command group."""
|
||||
|
||||
def test_env_view(self):
|
||||
"""Test env view command."""
|
||||
from praisonai.cli.commands.environment import app
|
||||
result = runner.invoke(app, ["view"])
|
||||
# Should succeed even if no env vars found
|
||||
assert result.exit_code == 0
|
||||
|
||||
def test_env_check(self):
|
||||
"""Test env check command."""
|
||||
from praisonai.cli.commands.environment import app
|
||||
result = runner.invoke(app, ["check"])
|
||||
# May exit with 3 if keys are missing, but should not crash
|
||||
assert result.exit_code in [0, 3]
|
||||
|
||||
|
||||
class TestSessionCommand:
|
||||
"""Tests for session command group."""
|
||||
|
||||
def test_session_list(self):
|
||||
"""Test session list command."""
|
||||
from praisonai.cli.commands.session import app
|
||||
result = runner.invoke(app, ["list"])
|
||||
assert result.exit_code == 0
|
||||
|
||||
|
||||
class TestCompletionCommand:
|
||||
"""Tests for completion command group."""
|
||||
|
||||
def test_completion_bash(self):
|
||||
"""Test bash completion generation."""
|
||||
from praisonai.cli.commands.completion import app
|
||||
result = runner.invoke(app, ["bash"])
|
||||
assert result.exit_code == 0
|
||||
assert "praisonai" in result.output
|
||||
assert "complete" in result.output or "COMPREPLY" in result.output
|
||||
|
||||
def test_completion_zsh(self):
|
||||
"""Test zsh completion generation."""
|
||||
from praisonai.cli.commands.completion import app
|
||||
result = runner.invoke(app, ["zsh"])
|
||||
assert result.exit_code == 0
|
||||
assert "praisonai" in result.output
|
||||
assert "compdef" in result.output or "_praisonai" in result.output
|
||||
|
||||
def test_completion_fish(self):
|
||||
"""Test fish completion generation."""
|
||||
from praisonai.cli.commands.completion import app
|
||||
result = runner.invoke(app, ["fish"])
|
||||
assert result.exit_code == 0
|
||||
assert "praisonai" in result.output
|
||||
assert "complete" in result.output
|
||||
|
||||
|
||||
class TestVersionCommand:
|
||||
"""Tests for version command group."""
|
||||
|
||||
def test_version_show(self):
|
||||
"""Test version show command."""
|
||||
from praisonai.cli.commands.version import app
|
||||
result = runner.invoke(app, ["show"])
|
||||
assert result.exit_code == 0
|
||||
assert "PraisonAI" in result.output or "praisonai" in result.output.lower()
|
||||
|
||||
|
||||
class TestMCPCommand:
|
||||
"""Tests for MCP command group."""
|
||||
|
||||
def test_mcp_list(self):
|
||||
"""Test mcp list command."""
|
||||
from praisonai.cli.commands.mcp import app
|
||||
result = runner.invoke(app, ["list"])
|
||||
assert result.exit_code == 0
|
||||
|
||||
|
||||
class TestOutputController:
|
||||
"""Tests for output controller."""
|
||||
|
||||
def test_output_modes(self):
|
||||
"""Test output mode enumeration."""
|
||||
from praisonai.cli.output.console import OutputMode
|
||||
assert OutputMode.TEXT == "text"
|
||||
assert OutputMode.JSON == "json"
|
||||
assert OutputMode.STREAM_JSON == "stream-json"
|
||||
assert OutputMode.SCREEN_READER == "screen-reader"
|
||||
assert OutputMode.QUIET == "quiet"
|
||||
assert OutputMode.VERBOSE == "verbose"
|
||||
|
||||
def test_output_controller_creation(self):
|
||||
"""Test output controller creation."""
|
||||
from praisonai.cli.output.console import OutputController, OutputMode
|
||||
|
||||
controller = OutputController(mode=OutputMode.TEXT)
|
||||
assert controller.mode == OutputMode.TEXT
|
||||
assert not controller.is_json_mode
|
||||
|
||||
controller = OutputController(mode=OutputMode.JSON)
|
||||
assert controller.is_json_mode
|
||||
|
||||
def test_stream_event(self):
|
||||
"""Test stream event creation."""
|
||||
from praisonai.cli.output.console import StreamEvent
|
||||
|
||||
event = StreamEvent(
|
||||
event_type="start",
|
||||
run_id="run_123",
|
||||
trace_id="trace_456",
|
||||
message="Test message"
|
||||
)
|
||||
|
||||
data = event.to_dict()
|
||||
assert data["event"] == "start"
|
||||
assert data["run_id"] == "run_123"
|
||||
assert data["trace_id"] == "trace_456"
|
||||
assert data["message"] == "Test message"
|
||||
assert "timestamp" in data
|
||||
|
||||
|
||||
class TestConfigurationSystem:
|
||||
"""Tests for configuration system."""
|
||||
|
||||
def test_config_schema(self):
|
||||
"""Test config schema creation."""
|
||||
from praisonai.cli.configuration.schema import ConfigSchema, DEFAULT_CONFIG
|
||||
|
||||
assert DEFAULT_CONFIG is not None
|
||||
assert DEFAULT_CONFIG.output.format == "text"
|
||||
assert DEFAULT_CONFIG.traces.enabled == False
|
||||
|
||||
def test_config_to_dict(self):
|
||||
"""Test config to dict conversion."""
|
||||
from praisonai.cli.configuration.schema import ConfigSchema
|
||||
|
||||
config = ConfigSchema()
|
||||
data = config.to_dict()
|
||||
|
||||
assert "output" in data
|
||||
assert "traces" in data
|
||||
assert "mcp" in data
|
||||
assert "model" in data
|
||||
assert "session" in data
|
||||
|
||||
def test_config_from_dict(self):
|
||||
"""Test config from dict creation."""
|
||||
from praisonai.cli.configuration.schema import ConfigSchema
|
||||
|
||||
data = {
|
||||
"output": {"format": "json", "verbose": True},
|
||||
"traces": {"enabled": True},
|
||||
}
|
||||
|
||||
config = ConfigSchema.from_dict(data)
|
||||
assert config.output.format == "json"
|
||||
assert config.output.verbose == True
|
||||
assert config.traces.enabled == True
|
||||
|
||||
def test_config_paths(self):
|
||||
"""Test config paths."""
|
||||
from praisonai.cli.configuration.paths import (
|
||||
get_user_config_path,
|
||||
get_project_config_path,
|
||||
get_sessions_dir,
|
||||
)
|
||||
|
||||
user_path = get_user_config_path()
|
||||
assert user_path.name == "config.toml"
|
||||
assert ".praison" in str(user_path)
|
||||
|
||||
project_path = get_project_config_path()
|
||||
assert project_path.name == "config.toml"
|
||||
|
||||
sessions_dir = get_sessions_dir()
|
||||
assert "sessions" in str(sessions_dir)
|
||||
|
||||
|
||||
class TestStateManagement:
|
||||
"""Tests for state management."""
|
||||
|
||||
def test_run_id_generation(self):
|
||||
"""Test run ID generation."""
|
||||
from praisonai.cli.state.identifiers import generate_run_id
|
||||
|
||||
run_id = generate_run_id()
|
||||
assert run_id.startswith("run_")
|
||||
assert len(run_id) > 10
|
||||
|
||||
def test_trace_id_generation(self):
|
||||
"""Test trace ID generation."""
|
||||
from praisonai.cli.state.identifiers import generate_trace_id
|
||||
|
||||
trace_id = generate_trace_id()
|
||||
assert trace_id.startswith("trace_")
|
||||
|
||||
def test_agent_id_generation(self):
|
||||
"""Test agent ID generation is deterministic."""
|
||||
from praisonai.cli.state.identifiers import generate_agent_id
|
||||
|
||||
agent_id1 = generate_agent_id("test_agent", "run_123", 0)
|
||||
agent_id2 = generate_agent_id("test_agent", "run_123", 0)
|
||||
|
||||
assert agent_id1 == agent_id2
|
||||
assert agent_id1.startswith("agent_")
|
||||
|
||||
def test_run_context(self):
|
||||
"""Test run context creation."""
|
||||
from praisonai.cli.state.identifiers import RunContext
|
||||
|
||||
ctx = RunContext()
|
||||
assert ctx.run_id.startswith("run_")
|
||||
assert ctx.trace_id.startswith("trace_")
|
||||
|
||||
# Test agent ID tracking
|
||||
agent_id = ctx.get_agent_id("test_agent")
|
||||
assert agent_id.startswith("agent_")
|
||||
|
||||
# Same agent should get same ID
|
||||
agent_id2 = ctx.get_agent_id("test_agent")
|
||||
assert agent_id == agent_id2
|
||||
|
||||
def test_context_to_dict(self):
|
||||
"""Test context serialization."""
|
||||
from praisonai.cli.state.identifiers import RunContext
|
||||
|
||||
ctx = RunContext(workspace="/test")
|
||||
ctx.get_agent_id("agent1")
|
||||
|
||||
data = ctx.to_dict()
|
||||
assert "run_id" in data
|
||||
assert "trace_id" in data
|
||||
assert "agents" in data
|
||||
assert data["workspace"] == "/test"
|
||||
|
||||
|
||||
class TestSessionManager:
|
||||
"""Tests for session manager."""
|
||||
|
||||
def test_session_metadata(self):
|
||||
"""Test session metadata creation."""
|
||||
from praisonai.cli.state.sessions import SessionMetadata
|
||||
from datetime import datetime
|
||||
|
||||
now = datetime.utcnow()
|
||||
metadata = SessionMetadata(
|
||||
session_id="test_session",
|
||||
run_id="run_123",
|
||||
trace_id="trace_456",
|
||||
created_at=now,
|
||||
updated_at=now,
|
||||
name="Test Session",
|
||||
)
|
||||
|
||||
assert metadata.session_id == "test_session"
|
||||
assert metadata.name == "Test Session"
|
||||
|
||||
data = metadata.to_dict()
|
||||
assert data["session_id"] == "test_session"
|
||||
assert data["name"] == "Test Session"
|
||||
|
||||
|
||||
class TestLegacyCompatibility:
|
||||
"""Tests for legacy compatibility."""
|
||||
|
||||
def test_is_legacy_invocation(self):
|
||||
"""Test legacy invocation detection."""
|
||||
from praisonai.cli.legacy import is_legacy_invocation
|
||||
|
||||
# Legacy patterns
|
||||
assert is_legacy_invocation(["--framework", "crewai"])
|
||||
assert is_legacy_invocation(["--auto", "test"])
|
||||
assert is_legacy_invocation(["chat"]) # Legacy command
|
||||
|
||||
# Typer patterns
|
||||
assert not is_legacy_invocation(["config", "list"])
|
||||
assert not is_legacy_invocation(["version", "show"])
|
||||
|
||||
def test_typer_commands_set(self):
|
||||
"""Test TYPER_COMMANDS contains expected commands."""
|
||||
from praisonai.cli.legacy import TYPER_COMMANDS
|
||||
|
||||
expected = {'config', 'traces', 'env', 'session', 'completion', 'version',
|
||||
'debug', 'lsp', 'diag', 'doctor', 'acp', 'mcp', 'serve', 'schedule', 'run'}
|
||||
|
||||
for cmd in expected:
|
||||
assert cmd in TYPER_COMMANDS
|
||||
|
||||
|
||||
class TestGlobalOptions:
|
||||
"""Tests for global CLI options."""
|
||||
|
||||
def test_json_output_option(self):
|
||||
"""Test --json option."""
|
||||
from praisonai.cli.app import app
|
||||
result = runner.invoke(app, ["--json", "config", "list"])
|
||||
# Should produce JSON output
|
||||
assert result.exit_code == 0
|
||||
|
||||
def test_quiet_option(self):
|
||||
"""Test --quiet option."""
|
||||
from praisonai.cli.app import app
|
||||
result = runner.invoke(app, ["--quiet", "version", "show"])
|
||||
assert result.exit_code == 0
|
||||
|
||||
def test_no_color_option(self):
|
||||
"""Test --no-color option."""
|
||||
from praisonai.cli.app import app
|
||||
result = runner.invoke(app, ["--no-color", "version", "show"])
|
||||
assert result.exit_code == 0
|
||||
|
||||
|
||||
class TestExitCodes:
|
||||
"""Tests for deterministic exit codes."""
|
||||
|
||||
def test_success_exit_code(self):
|
||||
"""Test success returns 0."""
|
||||
from praisonai.cli.app import app
|
||||
result = runner.invoke(app, ["version", "show"])
|
||||
assert result.exit_code == 0
|
||||
|
||||
def test_usage_error_exit_code(self):
|
||||
"""Test usage error returns 2."""
|
||||
from praisonai.cli.app import app
|
||||
result = runner.invoke(app, ["config", "get"]) # Missing required arg
|
||||
assert result.exit_code == 2
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
pytest.main([__file__, "-v"])
|
||||
Generated
+243
-8
@@ -133,6 +133,19 @@ wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/b8/62/c9fa5bafe03186a0e4699150a7fed9b1e73240996d0d2f0e5f70f3fdf471/aiohttp-3.11.11-cp313-cp313-win_amd64.whl", hash = "sha256:c7a06301c2fb096bdb0bd25fe2011531c1453b9f2c163c8031600ec73af1cc99", size = 436081, upload-time = "2024-12-18T21:20:04.557Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "aiohttp-jinja2"
|
||||
version = "1.6"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
dependencies = [
|
||||
{ name = "aiohttp" },
|
||||
{ name = "jinja2" },
|
||||
]
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/e6/39/da5a94dd89b1af7241fb7fc99ae4e73505b5f898b540b6aba6dc7afe600e/aiohttp-jinja2-1.6.tar.gz", hash = "sha256:a3a7ff5264e5bca52e8ae547bbfd0761b72495230d438d05b6c0915be619b0e2", size = 53057, upload-time = "2023-11-18T15:30:52.559Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/eb/90/65238d4246307195411b87a07d03539049819b022c01bcc773826f600138/aiohttp_jinja2-1.6-py3-none-any.whl", hash = "sha256:0df405ee6ad1b58e5a068a105407dc7dcc1704544c559f1938babde954f945c7", size = 11736, upload-time = "2023-11-18T15:30:50.743Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "aiohttp-retry"
|
||||
version = "2.8.3"
|
||||
@@ -398,6 +411,15 @@ wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/df/73/b6e24bd22e6720ca8ee9a85a0c4a2971af8497d8f3193fa05390cbd46e09/backoff-2.2.1-py3-none-any.whl", hash = "sha256:63579f9a0628e06278f7e47b7d7d5b6ce20dc65c5e96a6f3ca99a6adca0396e8", size = 15148, upload-time = "2022-10-05T19:19:30.546Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "backports-asyncio-runner"
|
||||
version = "1.2.0"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/8e/ff/70dca7d7cb1cbc0edb2c6cc0c38b65cba36cccc491eca64cabd5fe7f8670/backports_asyncio_runner-1.2.0.tar.gz", hash = "sha256:a5aa7b2b7d8f8bfcaa2b57313f70792df84e32a2a746f585213373f900b42162", size = 69893, upload-time = "2025-07-02T02:27:15.685Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/a0/59/76ab57e3fe74484f48a53f8e337171b4a2349e506eabe136d7e01d059086/backports_asyncio_runner-1.2.0-py3-none-any.whl", hash = "sha256:0da0a936a8aeb554eccb426dc55af3ba63bcdc69fa1a600b5bb305413a4477b5", size = 12313, upload-time = "2025-07-02T02:27:14.263Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "bcrypt"
|
||||
version = "4.2.1"
|
||||
@@ -1817,6 +1839,15 @@ wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/59/91/aa6bde563e0085a02a435aa99b49ef75b0a4b062635e606dab23ce18d720/inflection-0.5.1-py2.py3-none-any.whl", hash = "sha256:f38b2b640938a4f35ade69ac3d053042959b62a0f1076a5bbaa1b9526605a8a2", size = 9454, upload-time = "2020-08-22T08:16:27.816Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "iniconfig"
|
||||
version = "2.3.0"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/72/34/14ca021ce8e5dfedc35312d08ba8bf51fdd999c576889fc2c24cb97f4f10/iniconfig-2.3.0.tar.gz", hash = "sha256:c76315c77db068650d49c5b56314774a7804df16fee4402c1f19d6d15d8c4730", size = 20503, upload-time = "2025-10-18T21:55:43.219Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/cb/b1/3846dd7f199d53cb17f49cba7e651e9ce294d8497c8c150530ed11865bb8/iniconfig-2.3.0-py3-none-any.whl", hash = "sha256:f631c04d2c48c52b84d0d0549c99ff3859c98df65b3101406327ecc7d53fbf12", size = 7484, upload-time = "2025-10-18T21:55:41.639Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "instructor"
|
||||
version = "1.3.3"
|
||||
@@ -2283,6 +2314,18 @@ wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/d9/28/3be76b591a2e14a031b681b8283acf1dec2ad521f6f1701b7957df68c466/lightgbm-4.5.0-py3-none-win_amd64.whl", hash = "sha256:7ccb73ee9fb74fbbf89ad24c57a6edad505aa8f2165d02b999a082dbbbb0ee57", size = 1444319, upload-time = "2024-07-26T14:39:10.001Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "linkify-it-py"
|
||||
version = "2.0.3"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
dependencies = [
|
||||
{ name = "uc-micro-py" },
|
||||
]
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/2a/ae/bb56c6828e4797ba5a4821eec7c43b8bf40f69cda4d4f5f8c8a2810ec96a/linkify-it-py-2.0.3.tar.gz", hash = "sha256:68cda27e162e9215c17d786649d1da0021a451bdc436ef9e0fa0ba5234b9b048", size = 27946, upload-time = "2024-02-04T14:48:04.179Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/04/1e/b832de447dee8b582cac175871d2f6c3d5077cc56d5575cadba1fd1cccfa/linkify_it_py-2.0.3-py3-none-any.whl", hash = "sha256:6bcbc417b0ac14323382aef5c5192c0075bf8a9d6b41820a2b66371eac6b6d79", size = 19820, upload-time = "2024-02-04T14:48:02.496Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "lit"
|
||||
version = "18.1.8"
|
||||
@@ -2430,6 +2473,14 @@ wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/42/d7/1ec15b46af6af88f19b8e5ffea08fa375d433c998b8a7639e76935c14f1f/markdown_it_py-3.0.0-py3-none-any.whl", hash = "sha256:355216845c60bd96232cd8d8c40e8f9765cc86f46880e43a8fd22dc1a1a8cab1", size = 87528, upload-time = "2023-06-03T06:41:11.019Z" },
|
||||
]
|
||||
|
||||
[package.optional-dependencies]
|
||||
linkify = [
|
||||
{ name = "linkify-it-py" },
|
||||
]
|
||||
plugins = [
|
||||
{ name = "mdit-py-plugins" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "markupsafe"
|
||||
version = "2.1.5"
|
||||
@@ -2514,6 +2565,18 @@ wheels = [
|
||||
{ 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]]
|
||||
name = "mdit-py-plugins"
|
||||
version = "0.5.0"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
dependencies = [
|
||||
{ name = "markdown-it-py" },
|
||||
]
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/b2/fd/a756d36c0bfba5f6e39a1cdbdbfdd448dc02692467d83816dff4592a1ebc/mdit_py_plugins-0.5.0.tar.gz", hash = "sha256:f4918cb50119f50446560513a8e311d574ff6aaed72606ddae6d35716fe809c6", size = 44655, upload-time = "2025-08-11T07:25:49.083Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/fb/86/dd6e5db36df29e76c7a7699123569a4a18c1623ce68d826ed96c62643cae/mdit_py_plugins-0.5.0-py3-none-any.whl", hash = "sha256:07a08422fc1936a5d26d146759e9155ea466e842f5ab2f7d2266dd084c8dab1f", size = 57205, upload-time = "2025-08-11T07:25:47.597Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "mdurl"
|
||||
version = "0.1.2"
|
||||
@@ -2613,6 +2676,67 @@ wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/43/e3/7d92a15f894aa0c9c4b49b8ee9ac9850d6e63b03c9c32c0367a13ae62209/mpmath-1.3.0-py3-none-any.whl", hash = "sha256:a0b2b9fe80bbcd81a6647ff13108738cfb482d481d826cc0e02f5b35e5c88d2c", size = 536198, upload-time = "2023-03-07T16:47:09.197Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "msgpack"
|
||||
version = "1.1.2"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/4d/f2/bfb55a6236ed8725a96b0aa3acbd0ec17588e6a2c3b62a93eb513ed8783f/msgpack-1.1.2.tar.gz", hash = "sha256:3b60763c1373dd60f398488069bcdc703cd08a711477b5d480eecc9f9626f47e", size = 173581, upload-time = "2025-10-08T09:15:56.596Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/f5/a2/3b68a9e769db68668b25c6108444a35f9bd163bb848c0650d516761a59c0/msgpack-1.1.2-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:0051fffef5a37ca2cd16978ae4f0aef92f164df86823871b5162812bebecd8e2", size = 81318, upload-time = "2025-10-08T09:14:38.722Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/5b/e1/2b720cc341325c00be44e1ed59e7cfeae2678329fbf5aa68f5bda57fe728/msgpack-1.1.2-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:a605409040f2da88676e9c9e5853b3449ba8011973616189ea5ee55ddbc5bc87", size = 83786, upload-time = "2025-10-08T09:14:40.082Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/71/e5/c2241de64bfceac456b140737812a2ab310b10538a7b34a1d393b748e095/msgpack-1.1.2-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:8b696e83c9f1532b4af884045ba7f3aa741a63b2bc22617293a2c6a7c645f251", size = 398240, upload-time = "2025-10-08T09:14:41.151Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/b7/09/2a06956383c0fdebaef5aa9246e2356776f12ea6f2a44bd1368abf0e46c4/msgpack-1.1.2-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:365c0bbe981a27d8932da71af63ef86acc59ed5c01ad929e09a0b88c6294e28a", size = 406070, upload-time = "2025-10-08T09:14:42.821Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/0e/74/2957703f0e1ef20637d6aead4fbb314330c26f39aa046b348c7edcf6ca6b/msgpack-1.1.2-cp310-cp310-musllinux_1_2_aarch64.whl", hash = "sha256:41d1a5d875680166d3ac5c38573896453bbbea7092936d2e107214daf43b1d4f", size = 393403, upload-time = "2025-10-08T09:14:44.38Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/a5/09/3bfc12aa90f77b37322fc33e7a8a7c29ba7c8edeadfa27664451801b9860/msgpack-1.1.2-cp310-cp310-musllinux_1_2_x86_64.whl", hash = "sha256:354e81bcdebaab427c3df4281187edc765d5d76bfb3a7c125af9da7a27e8458f", size = 398947, upload-time = "2025-10-08T09:14:45.56Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/4b/4f/05fcebd3b4977cb3d840f7ef6b77c51f8582086de5e642f3fefee35c86fc/msgpack-1.1.2-cp310-cp310-win32.whl", hash = "sha256:e64c8d2f5e5d5fda7b842f55dec6133260ea8f53c4257d64494c534f306bf7a9", size = 64769, upload-time = "2025-10-08T09:14:47.334Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/d0/3e/b4547e3a34210956382eed1c85935fff7e0f9b98be3106b3745d7dec9c5e/msgpack-1.1.2-cp310-cp310-win_amd64.whl", hash = "sha256:db6192777d943bdaaafb6ba66d44bf65aa0e9c5616fa1d2da9bb08828c6b39aa", size = 71293, upload-time = "2025-10-08T09:14:48.665Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/2c/97/560d11202bcd537abca693fd85d81cebe2107ba17301de42b01ac1677b69/msgpack-1.1.2-cp311-cp311-macosx_10_9_x86_64.whl", hash = "sha256:2e86a607e558d22985d856948c12a3fa7b42efad264dca8a3ebbcfa2735d786c", size = 82271, upload-time = "2025-10-08T09:14:49.967Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/83/04/28a41024ccbd67467380b6fb440ae916c1e4f25e2cd4c63abe6835ac566e/msgpack-1.1.2-cp311-cp311-macosx_11_0_arm64.whl", hash = "sha256:283ae72fc89da59aa004ba147e8fc2f766647b1251500182fac0350d8af299c0", size = 84914, upload-time = "2025-10-08T09:14:50.958Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/71/46/b817349db6886d79e57a966346cf0902a426375aadc1e8e7a86a75e22f19/msgpack-1.1.2-cp311-cp311-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:61c8aa3bd513d87c72ed0b37b53dd5c5a0f58f2ff9f26e1555d3bd7948fb7296", size = 416962, upload-time = "2025-10-08T09:14:51.997Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/da/e0/6cc2e852837cd6086fe7d8406af4294e66827a60a4cf60b86575a4a65ca8/msgpack-1.1.2-cp311-cp311-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:454e29e186285d2ebe65be34629fa0e8605202c60fbc7c4c650ccd41870896ef", size = 426183, upload-time = "2025-10-08T09:14:53.477Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/25/98/6a19f030b3d2ea906696cedd1eb251708e50a5891d0978b012cb6107234c/msgpack-1.1.2-cp311-cp311-musllinux_1_2_aarch64.whl", hash = "sha256:7bc8813f88417599564fafa59fd6f95be417179f76b40325b500b3c98409757c", size = 411454, upload-time = "2025-10-08T09:14:54.648Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/b7/cd/9098fcb6adb32187a70b7ecaabf6339da50553351558f37600e53a4a2a23/msgpack-1.1.2-cp311-cp311-musllinux_1_2_x86_64.whl", hash = "sha256:bafca952dc13907bdfdedfc6a5f579bf4f292bdd506fadb38389afa3ac5b208e", size = 422341, upload-time = "2025-10-08T09:14:56.328Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/e6/ae/270cecbcf36c1dc85ec086b33a51a4d7d08fc4f404bdbc15b582255d05ff/msgpack-1.1.2-cp311-cp311-win32.whl", hash = "sha256:602b6740e95ffc55bfb078172d279de3773d7b7db1f703b2f1323566b878b90e", size = 64747, upload-time = "2025-10-08T09:14:57.882Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/2a/79/309d0e637f6f37e83c711f547308b91af02b72d2326ddd860b966080ef29/msgpack-1.1.2-cp311-cp311-win_amd64.whl", hash = "sha256:d198d275222dc54244bf3327eb8cbe00307d220241d9cec4d306d49a44e85f68", size = 71633, upload-time = "2025-10-08T09:14:59.177Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/73/4d/7c4e2b3d9b1106cd0aa6cb56cc57c6267f59fa8bfab7d91df5adc802c847/msgpack-1.1.2-cp311-cp311-win_arm64.whl", hash = "sha256:86f8136dfa5c116365a8a651a7d7484b65b13339731dd6faebb9a0242151c406", size = 64755, upload-time = "2025-10-08T09:15:00.48Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/ad/bd/8b0d01c756203fbab65d265859749860682ccd2a59594609aeec3a144efa/msgpack-1.1.2-cp312-cp312-macosx_10_13_x86_64.whl", hash = "sha256:70a0dff9d1f8da25179ffcf880e10cf1aad55fdb63cd59c9a49a1b82290062aa", size = 81939, upload-time = "2025-10-08T09:15:01.472Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/34/68/ba4f155f793a74c1483d4bdef136e1023f7bcba557f0db4ef3db3c665cf1/msgpack-1.1.2-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:446abdd8b94b55c800ac34b102dffd2f6aa0ce643c55dfc017ad89347db3dbdb", size = 85064, upload-time = "2025-10-08T09:15:03.764Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/f2/60/a064b0345fc36c4c3d2c743c82d9100c40388d77f0b48b2f04d6041dbec1/msgpack-1.1.2-cp312-cp312-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:c63eea553c69ab05b6747901b97d620bb2a690633c77f23feb0c6a947a8a7b8f", size = 417131, upload-time = "2025-10-08T09:15:05.136Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/65/92/a5100f7185a800a5d29f8d14041f61475b9de465ffcc0f3b9fba606e4505/msgpack-1.1.2-cp312-cp312-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:372839311ccf6bdaf39b00b61288e0557916c3729529b301c52c2d88842add42", size = 427556, upload-time = "2025-10-08T09:15:06.837Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/f5/87/ffe21d1bf7d9991354ad93949286f643b2bb6ddbeab66373922b44c3b8cc/msgpack-1.1.2-cp312-cp312-musllinux_1_2_aarch64.whl", hash = "sha256:2929af52106ca73fcb28576218476ffbb531a036c2adbcf54a3664de124303e9", size = 404920, upload-time = "2025-10-08T09:15:08.179Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/ff/41/8543ed2b8604f7c0d89ce066f42007faac1eaa7d79a81555f206a5cdb889/msgpack-1.1.2-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:be52a8fc79e45b0364210eef5234a7cf8d330836d0a64dfbb878efa903d84620", size = 415013, upload-time = "2025-10-08T09:15:09.83Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/41/0d/2ddfaa8b7e1cee6c490d46cb0a39742b19e2481600a7a0e96537e9c22f43/msgpack-1.1.2-cp312-cp312-win32.whl", hash = "sha256:1fff3d825d7859ac888b0fbda39a42d59193543920eda9d9bea44d958a878029", size = 65096, upload-time = "2025-10-08T09:15:11.11Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/8c/ec/d431eb7941fb55a31dd6ca3404d41fbb52d99172df2e7707754488390910/msgpack-1.1.2-cp312-cp312-win_amd64.whl", hash = "sha256:1de460f0403172cff81169a30b9a92b260cb809c4cb7e2fc79ae8d0510c78b6b", size = 72708, upload-time = "2025-10-08T09:15:12.554Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/c5/31/5b1a1f70eb0e87d1678e9624908f86317787b536060641d6798e3cf70ace/msgpack-1.1.2-cp312-cp312-win_arm64.whl", hash = "sha256:be5980f3ee0e6bd44f3a9e9dea01054f175b50c3e6cdb692bc9424c0bbb8bf69", size = 64119, upload-time = "2025-10-08T09:15:13.589Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/6b/31/b46518ecc604d7edf3a4f94cb3bf021fc62aa301f0cb849936968164ef23/msgpack-1.1.2-cp313-cp313-macosx_10_13_x86_64.whl", hash = "sha256:4efd7b5979ccb539c221a4c4e16aac1a533efc97f3b759bb5a5ac9f6d10383bf", size = 81212, upload-time = "2025-10-08T09:15:14.552Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/92/dc/c385f38f2c2433333345a82926c6bfa5ecfff3ef787201614317b58dd8be/msgpack-1.1.2-cp313-cp313-macosx_11_0_arm64.whl", hash = "sha256:42eefe2c3e2af97ed470eec850facbe1b5ad1d6eacdbadc42ec98e7dcf68b4b7", size = 84315, upload-time = "2025-10-08T09:15:15.543Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/d3/68/93180dce57f684a61a88a45ed13047558ded2be46f03acb8dec6d7c513af/msgpack-1.1.2-cp313-cp313-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:1fdf7d83102bf09e7ce3357de96c59b627395352a4024f6e2458501f158bf999", size = 412721, upload-time = "2025-10-08T09:15:16.567Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/5d/ba/459f18c16f2b3fc1a1ca871f72f07d70c07bf768ad0a507a698b8052ac58/msgpack-1.1.2-cp313-cp313-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:fac4be746328f90caa3cd4bc67e6fe36ca2bf61d5c6eb6d895b6527e3f05071e", size = 424657, upload-time = "2025-10-08T09:15:17.825Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/38/f8/4398c46863b093252fe67368b44edc6c13b17f4e6b0e4929dbf0bdb13f23/msgpack-1.1.2-cp313-cp313-musllinux_1_2_aarch64.whl", hash = "sha256:fffee09044073e69f2bad787071aeec727183e7580443dfeb8556cbf1978d162", size = 402668, upload-time = "2025-10-08T09:15:19.003Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/28/ce/698c1eff75626e4124b4d78e21cca0b4cc90043afb80a507626ea354ab52/msgpack-1.1.2-cp313-cp313-musllinux_1_2_x86_64.whl", hash = "sha256:5928604de9b032bc17f5099496417f113c45bc6bc21b5c6920caf34b3c428794", size = 419040, upload-time = "2025-10-08T09:15:20.183Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/67/32/f3cd1667028424fa7001d82e10ee35386eea1408b93d399b09fb0aa7875f/msgpack-1.1.2-cp313-cp313-win32.whl", hash = "sha256:a7787d353595c7c7e145e2331abf8b7ff1e6673a6b974ded96e6d4ec09f00c8c", size = 65037, upload-time = "2025-10-08T09:15:21.416Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/74/07/1ed8277f8653c40ebc65985180b007879f6a836c525b3885dcc6448ae6cb/msgpack-1.1.2-cp313-cp313-win_amd64.whl", hash = "sha256:a465f0dceb8e13a487e54c07d04ae3ba131c7c5b95e2612596eafde1dccf64a9", size = 72631, upload-time = "2025-10-08T09:15:22.431Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/e5/db/0314e4e2db56ebcf450f277904ffd84a7988b9e5da8d0d61ab2d057df2b6/msgpack-1.1.2-cp313-cp313-win_arm64.whl", hash = "sha256:e69b39f8c0aa5ec24b57737ebee40be647035158f14ed4b40e6f150077e21a84", size = 64118, upload-time = "2025-10-08T09:15:23.402Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/22/71/201105712d0a2ff07b7873ed3c220292fb2ea5120603c00c4b634bcdafb3/msgpack-1.1.2-cp314-cp314-macosx_10_13_x86_64.whl", hash = "sha256:e23ce8d5f7aa6ea6d2a2b326b4ba46c985dbb204523759984430db7114f8aa00", size = 81127, upload-time = "2025-10-08T09:15:24.408Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/1b/9f/38ff9e57a2eade7bf9dfee5eae17f39fc0e998658050279cbb14d97d36d9/msgpack-1.1.2-cp314-cp314-macosx_11_0_arm64.whl", hash = "sha256:6c15b7d74c939ebe620dd8e559384be806204d73b4f9356320632d783d1f7939", size = 84981, upload-time = "2025-10-08T09:15:25.812Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/8e/a9/3536e385167b88c2cc8f4424c49e28d49a6fc35206d4a8060f136e71f94c/msgpack-1.1.2-cp314-cp314-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:99e2cb7b9031568a2a5c73aa077180f93dd2e95b4f8d3b8e14a73ae94a9e667e", size = 411885, upload-time = "2025-10-08T09:15:27.22Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/2f/40/dc34d1a8d5f1e51fc64640b62b191684da52ca469da9cd74e84936ffa4a6/msgpack-1.1.2-cp314-cp314-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:180759d89a057eab503cf62eeec0aa61c4ea1200dee709f3a8e9397dbb3b6931", size = 419658, upload-time = "2025-10-08T09:15:28.4Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/3b/ef/2b92e286366500a09a67e03496ee8b8ba00562797a52f3c117aa2b29514b/msgpack-1.1.2-cp314-cp314-musllinux_1_2_aarch64.whl", hash = "sha256:04fb995247a6e83830b62f0b07bf36540c213f6eac8e851166d8d86d83cbd014", size = 403290, upload-time = "2025-10-08T09:15:29.764Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/78/90/e0ea7990abea5764e4655b8177aa7c63cdfa89945b6e7641055800f6c16b/msgpack-1.1.2-cp314-cp314-musllinux_1_2_x86_64.whl", hash = "sha256:8e22ab046fa7ede9e36eeb4cfad44d46450f37bb05d5ec482b02868f451c95e2", size = 415234, upload-time = "2025-10-08T09:15:31.022Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/72/4e/9390aed5db983a2310818cd7d3ec0aecad45e1f7007e0cda79c79507bb0d/msgpack-1.1.2-cp314-cp314-win32.whl", hash = "sha256:80a0ff7d4abf5fecb995fcf235d4064b9a9a8a40a3ab80999e6ac1e30b702717", size = 66391, upload-time = "2025-10-08T09:15:32.265Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/6e/f1/abd09c2ae91228c5f3998dbd7f41353def9eac64253de3c8105efa2082f7/msgpack-1.1.2-cp314-cp314-win_amd64.whl", hash = "sha256:9ade919fac6a3e7260b7f64cea89df6bec59104987cbea34d34a2fa15d74310b", size = 73787, upload-time = "2025-10-08T09:15:33.219Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/6a/b0/9d9f667ab48b16ad4115c1935d94023b82b3198064cb84a123e97f7466c1/msgpack-1.1.2-cp314-cp314-win_arm64.whl", hash = "sha256:59415c6076b1e30e563eb732e23b994a61c159cec44deaf584e5cc1dd662f2af", size = 66453, upload-time = "2025-10-08T09:15:34.225Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/16/67/93f80545eb1792b61a217fa7f06d5e5cb9e0055bed867f43e2b8e012e137/msgpack-1.1.2-cp314-cp314t-macosx_10_13_x86_64.whl", hash = "sha256:897c478140877e5307760b0ea66e0932738879e7aa68144d9b78ea4c8302a84a", size = 85264, upload-time = "2025-10-08T09:15:35.61Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/87/1c/33c8a24959cf193966ef11a6f6a2995a65eb066bd681fd085afd519a57ce/msgpack-1.1.2-cp314-cp314t-macosx_11_0_arm64.whl", hash = "sha256:a668204fa43e6d02f89dbe79a30b0d67238d9ec4c5bd8a940fc3a004a47b721b", size = 89076, upload-time = "2025-10-08T09:15:36.619Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/fc/6b/62e85ff7193663fbea5c0254ef32f0c77134b4059f8da89b958beb7696f3/msgpack-1.1.2-cp314-cp314t-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:5559d03930d3aa0f3aacb4c42c776af1a2ace2611871c84a75afe436695e6245", size = 435242, upload-time = "2025-10-08T09:15:37.647Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/c1/47/5c74ecb4cc277cf09f64e913947871682ffa82b3b93c8dad68083112f412/msgpack-1.1.2-cp314-cp314t-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:70c5a7a9fea7f036b716191c29047374c10721c389c21e9ffafad04df8c52c90", size = 432509, upload-time = "2025-10-08T09:15:38.794Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/24/a4/e98ccdb56dc4e98c929a3f150de1799831c0a800583cde9fa022fa90602d/msgpack-1.1.2-cp314-cp314t-musllinux_1_2_aarch64.whl", hash = "sha256:f2cb069d8b981abc72b41aea1c580ce92d57c673ec61af4c500153a626cb9e20", size = 415957, upload-time = "2025-10-08T09:15:40.238Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/da/28/6951f7fb67bc0a4e184a6b38ab71a92d9ba58080b27a77d3e2fb0be5998f/msgpack-1.1.2-cp314-cp314t-musllinux_1_2_x86_64.whl", hash = "sha256:d62ce1f483f355f61adb5433ebfd8868c5f078d1a52d042b0a998682b4fa8c27", size = 422910, upload-time = "2025-10-08T09:15:41.505Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/f0/03/42106dcded51f0a0b5284d3ce30a671e7bd3f7318d122b2ead66ad289fed/msgpack-1.1.2-cp314-cp314t-win32.whl", hash = "sha256:1d1418482b1ee984625d88aa9585db570180c286d942da463533b238b98b812b", size = 75197, upload-time = "2025-10-08T09:15:42.954Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/15/86/d0071e94987f8db59d4eeb386ddc64d0bb9b10820a8d82bcd3e53eeb2da6/msgpack-1.1.2-cp314-cp314t-win_amd64.whl", hash = "sha256:5a46bf7e831d09470ad92dff02b8b1ac92175ca36b087f904a0519857c6be3ff", size = 85772, upload-time = "2025-10-08T09:15:43.954Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/81/f2/08ace4142eb281c12701fc3b93a10795e4d4dc7f753911d836675050f886/msgpack-1.1.2-cp314-cp314t-win_arm64.whl", hash = "sha256:d99ef64f349d5ec3293688e91486c5fdb925ed03807f64d98d205d2713c60b46", size = 70868, upload-time = "2025-10-08T09:15:44.959Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "multidict"
|
||||
version = "6.1.0"
|
||||
@@ -3948,6 +4072,15 @@ wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/e5/ae/580600f441f6fc05218bd6c9d5794f4aef072a7d9093b291f1c50a9db8bc/plotly-5.24.1-py3-none-any.whl", hash = "sha256:f67073a1e637eb0dc3e46324d9d51e2fe76e9727c892dde64ddf1e1b51f29089", size = 19054220, upload-time = "2024-09-12T15:36:24.08Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "pluggy"
|
||||
version = "1.6.0"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/f9/e2/3e91f31a7d2b083fe6ef3fa267035b518369d9511ffab804f839851d2779/pluggy-1.6.0.tar.gz", hash = "sha256:7dcc130b76258d33b90f61b658791dede3486c3e6bfb003ee5c9bfb396dd22f3", size = 69412, upload-time = "2025-05-15T12:30:07.975Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/54/20/4d324d65cc6d9205fabedc306948156824eb9f0ee1633355a8f7ec5c66bf/pluggy-1.6.0-py3-none-any.whl", hash = "sha256:e920276dd6813095e9377c0bc5566d94c932c33b27a3e3945d8389c374dd4746", size = 20538, upload-time = "2025-05-15T12:30:06.134Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "portalocker"
|
||||
version = "2.7.0"
|
||||
@@ -3988,6 +4121,8 @@ dependencies = [
|
||||
{ name = "python-dotenv" },
|
||||
{ name = "pyyaml" },
|
||||
{ name = "rich" },
|
||||
{ name = "textual" },
|
||||
{ name = "typer" },
|
||||
]
|
||||
|
||||
[package.optional-dependencies]
|
||||
@@ -4058,6 +4193,11 @@ crewai = [
|
||||
{ name = "crewai" },
|
||||
{ name = "praisonai-tools" },
|
||||
]
|
||||
dev = [
|
||||
{ name = "pytest" },
|
||||
{ name = "pytest-asyncio" },
|
||||
{ name = "textual-dev" },
|
||||
]
|
||||
google = [
|
||||
{ name = "langchain-google-genai" },
|
||||
]
|
||||
@@ -4136,13 +4276,15 @@ requires-dist = [
|
||||
{ name = "praisonai-tools", marker = "extra == 'autogen'", specifier = ">=0.1.0" },
|
||||
{ name = "praisonai-tools", marker = "extra == 'autogen-v4'", specifier = ">=0.1.0" },
|
||||
{ name = "praisonai-tools", marker = "extra == 'crewai'", specifier = ">=0.1.0" },
|
||||
{ name = "praisonaiagents", specifier = ">=0.5.2" },
|
||||
{ name = "praisonaiagents", specifier = ">=0.5.3" },
|
||||
{ name = "pyautogen", marker = "extra == 'autogen'", specifier = "==0.2.29" },
|
||||
{ name = "pydantic", marker = "extra == 'chat'", specifier = "<=2.10.1" },
|
||||
{ name = "pydantic", marker = "extra == 'code'", specifier = "<=2.10.1" },
|
||||
{ name = "pydantic", marker = "extra == 'ui'", specifier = "<=2.10.1" },
|
||||
{ name = "pyngrok", marker = "extra == 'call'", specifier = ">=1.4.0" },
|
||||
{ name = "pyparsing", specifier = ">=3.0.0" },
|
||||
{ name = "pytest", marker = "extra == 'dev'", specifier = ">=7.0.0" },
|
||||
{ name = "pytest-asyncio", marker = "extra == 'dev'", specifier = ">=0.21.0" },
|
||||
{ name = "python-dotenv", specifier = ">=0.19.0" },
|
||||
{ name = "pyyaml", specifier = ">=6.0" },
|
||||
{ name = "rich", specifier = ">=13.7" },
|
||||
@@ -4156,14 +4298,17 @@ requires-dist = [
|
||||
{ name = "tavily-python", marker = "extra == 'chat'", specifier = "==0.5.0" },
|
||||
{ name = "tavily-python", marker = "extra == 'code'", specifier = "==0.5.0" },
|
||||
{ name = "tavily-python", marker = "extra == 'realtime'", specifier = "==0.5.0" },
|
||||
{ name = "textual", specifier = ">=0.47.0" },
|
||||
{ name = "textual-dev", marker = "extra == 'dev'", specifier = ">=1.0.0" },
|
||||
{ name = "twilio", marker = "extra == 'call'", specifier = ">=7.0.0" },
|
||||
{ name = "typer", specifier = ">=0.9.0" },
|
||||
{ name = "uvicorn", marker = "extra == 'api'", specifier = ">=0.34.0" },
|
||||
{ name = "uvicorn", marker = "extra == 'call'", specifier = ">=0.20.0" },
|
||||
{ name = "websockets", marker = "extra == 'call'", specifier = ">=12.0" },
|
||||
{ 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", "acp"]
|
||||
provides-extras = ["ui", "gradio", "api", "agentops", "google", "openai", "anthropic", "cohere", "chat", "code", "realtime", "call", "train", "crewai", "autogen", "autogen-v4", "acp", "lite", "dev"]
|
||||
|
||||
[[package]]
|
||||
name = "praisonai-tools"
|
||||
@@ -4179,7 +4324,7 @@ wheels = [
|
||||
|
||||
[[package]]
|
||||
name = "praisonaiagents"
|
||||
version = "0.5.2"
|
||||
version = "0.5.3"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
dependencies = [
|
||||
{ name = "aiohttp" },
|
||||
@@ -4188,9 +4333,9 @@ dependencies = [
|
||||
{ name = "pydantic" },
|
||||
{ name = "rich" },
|
||||
]
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/77/e5/f0569d57595331894418c250cc40edb158a1476a4c7271448095ef3aa744/praisonaiagents-0.5.2.tar.gz", hash = "sha256:1d4f0616ca6a683b54edc84757ae9decaf14742acfa2398debdde65aef02ae32", size = 780477, upload-time = "2025-12-30T17:47:27.852Z" }
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/75/33/6091503673fb3c143293e9639bafe3acb6e7406e0ace1f893c34cbe22912/praisonaiagents-0.5.3.tar.gz", hash = "sha256:4163b00c28ef5e5d1a75411eb4b72fc7749790e52cdaa4bfc98c9a747fe42140", size = 780945, upload-time = "2025-12-31T16:36:45.335Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/12/7f/9b71b7c67a07e8a3085ab380a3fc2a8d69521543cae285ffb0c7462cebeb/praisonaiagents-0.5.2-py3-none-any.whl", hash = "sha256:a9fc263fe5664e3901ac2cee05e44a3cf0ea9e6fe732da786d89f2e862c8def1", size = 724488, upload-time = "2025-12-30T17:47:25.834Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/32/a2/459f111873f1751a34bc780e1b3a2cd264cf580f35acff868fa35aa9d917/praisonaiagents-0.5.3-py3-none-any.whl", hash = "sha256:35bd6c26dfff6c9e77575f72a2f077cdbcc0544d0d45b0ed29fb00ca96c3719c", size = 724879, upload-time = "2025-12-31T16:36:43.383Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
@@ -4530,11 +4675,11 @@ wheels = [
|
||||
|
||||
[[package]]
|
||||
name = "pygments"
|
||||
version = "2.18.0"
|
||||
version = "2.19.2"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/8e/62/8336eff65bcbc8e4cb5d05b55faf041285951b6e80f33e2bff2024788f31/pygments-2.18.0.tar.gz", hash = "sha256:786ff802f32e91311bff3889f6e9a86e81505fe99f2735bb6d60ae0c5004f199", size = 4891905, upload-time = "2024-05-04T13:42:02.013Z" }
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/b0/77/a5b8c569bf593b0140bde72ea885a803b82086995367bf2037de0159d924/pygments-2.19.2.tar.gz", hash = "sha256:636cb2477cec7f8952536970bc533bc43743542f70392ae026374600add5b887", size = 4968631, upload-time = "2025-06-21T13:39:12.283Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/f7/3f/01c8b82017c199075f8f788d0d906b9ffbbc5a47dc9918a945e13d5a2bda/pygments-2.18.0-py3-none-any.whl", hash = "sha256:b8e6aca0523f3ab76fee51799c488e38782ac06eafcf95e7ba832985c8e7b13a", size = 1205513, upload-time = "2024-05-04T13:41:57.345Z" },
|
||||
{ url = "https://files.pythonhosted.org/packages/c7/21/705964c7812476f378728bdf590ca4b771ec72385c533964653c68e86bdc/pygments-2.19.2-py3-none-any.whl", hash = "sha256:86540386c03d588bb81d44bc3928634ff26449851e99741617ecb9037ee5ec0b", size = 1225217, upload-time = "2025-06-21T13:39:07.939Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
@@ -4630,6 +4775,38 @@ wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/5a/dc/491b7661614ab97483abf2056be1deee4dc2490ecbf7bff9ab5cdbac86e1/pyreadline3-3.5.4-py3-none-any.whl", hash = "sha256:eaf8e6cc3c49bcccf145fc6067ba8643d1df34d604a1ec0eccbf7a18e6d3fae6", size = 83178, upload-time = "2024-09-19T02:40:08.598Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "pytest"
|
||||
version = "9.0.2"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
dependencies = [
|
||||
{ name = "colorama", marker = "sys_platform == 'win32'" },
|
||||
{ name = "exceptiongroup", marker = "python_full_version < '3.11'" },
|
||||
{ name = "iniconfig" },
|
||||
{ name = "packaging" },
|
||||
{ name = "pluggy" },
|
||||
{ name = "pygments" },
|
||||
{ name = "tomli", marker = "python_full_version < '3.11'" },
|
||||
]
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/d1/db/7ef3487e0fb0049ddb5ce41d3a49c235bf9ad299b6a25d5780a89f19230f/pytest-9.0.2.tar.gz", hash = "sha256:75186651a92bd89611d1d9fc20f0b4345fd827c41ccd5c299a868a05d70edf11", size = 1568901, upload-time = "2025-12-06T21:30:51.014Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/3b/ab/b3226f0bd7cdcf710fbede2b3548584366da3b19b5021e74f5bde2a8fa3f/pytest-9.0.2-py3-none-any.whl", hash = "sha256:711ffd45bf766d5264d487b917733b453d917afd2b0ad65223959f59089f875b", size = 374801, upload-time = "2025-12-06T21:30:49.154Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "pytest-asyncio"
|
||||
version = "1.3.0"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
dependencies = [
|
||||
{ name = "backports-asyncio-runner", marker = "python_full_version < '3.11'" },
|
||||
{ name = "pytest" },
|
||||
{ name = "typing-extensions", marker = "python_full_version < '3.13'" },
|
||||
]
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/90/2c/8af215c0f776415f3590cac4f9086ccefd6fd463befeae41cd4d3f193e5a/pytest_asyncio-1.3.0.tar.gz", hash = "sha256:d7f52f36d231b80ee124cd216ffb19369aa168fc10095013c6b014a34d3ee9e5", size = 50087, upload-time = "2025-11-10T16:07:47.256Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/e5/35/f8b19922b6a25bc0880171a2f1a003eaeb93657475193ab516fd87cac9da/pytest_asyncio-1.3.0-py3-none-any.whl", hash = "sha256:611e26147c7f77640e6d0a92a38ed17c3e9848063698d5c93d5aa7aa11cebff5", size = 15075, upload-time = "2025-11-10T16:07:45.537Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "python-dateutil"
|
||||
version = "2.9.0.post0"
|
||||
@@ -5523,6 +5700,55 @@ wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/d9/5f/8c716e47b3a50cbd7c146f45881e11d9414def768b7cd9c5e6650ec2a80a/termcolor-2.4.0-py3-none-any.whl", hash = "sha256:9297c0df9c99445c2412e832e882a7884038a25617c60cea2ad69488d4040d63", size = 7719, upload-time = "2023-12-01T11:04:50.019Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "textual"
|
||||
version = "6.2.1"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
dependencies = [
|
||||
{ name = "markdown-it-py", extra = ["linkify", "plugins"] },
|
||||
{ name = "platformdirs" },
|
||||
{ name = "pygments" },
|
||||
{ name = "rich" },
|
||||
{ name = "typing-extensions" },
|
||||
]
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/a2/30/38b615f7d4b16f6fdd73e4dcd8913e2d880bbb655e68a076e3d91181a7ee/textual-6.2.1.tar.gz", hash = "sha256:4699d8dfae43503b9c417bd2a6fb0da1c89e323fe91c4baa012f9298acaa83e1", size = 1570645, upload-time = "2025-10-01T16:11:24.467Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/c5/93/02c7adec57a594af28388d85da9972703a4af94ae1399542555cd9581952/textual-6.2.1-py3-none-any.whl", hash = "sha256:3c7190633cd4d8bfe6049ae66808b98da91ded2edb85cef54e82bf77b03d2a54", size = 710702, upload-time = "2025-10-01T16:11:22.161Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "textual-dev"
|
||||
version = "1.8.0"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
dependencies = [
|
||||
{ name = "aiohttp" },
|
||||
{ name = "click" },
|
||||
{ name = "msgpack" },
|
||||
{ name = "textual" },
|
||||
{ name = "textual-serve" },
|
||||
{ name = "typing-extensions" },
|
||||
]
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/38/fd/fd5ad9527b536c306a5a860a33a45e13983a59804b48b91ea52b42ba030a/textual_dev-1.8.0.tar.gz", hash = "sha256:7e56867b0341405a95e938cac0647e6d2763d38d0df08710469ad6b6a8db76df", size = 25026, upload-time = "2025-10-11T09:47:01.824Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/45/c6/1dc08ceee6d0bdf219c500cb2fbd605b681b63507c4ff27a6477f12f8245/textual_dev-1.8.0-py3-none-any.whl", hash = "sha256:227b6d24a485fbbc77e302aa21f4fdf3083beb57eb45cd95bae082c81cbeddeb", size = 27541, upload-time = "2025-10-11T09:47:00.531Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "textual-serve"
|
||||
version = "1.1.3"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
dependencies = [
|
||||
{ name = "aiohttp" },
|
||||
{ name = "aiohttp-jinja2" },
|
||||
{ name = "jinja2" },
|
||||
{ name = "rich" },
|
||||
{ name = "textual" },
|
||||
]
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/c2/7e/62fecc552853ec6a178cb1faa2d6f73b34d5512924770e7b08b58ff14148/textual_serve-1.1.3.tar.gz", hash = "sha256:f8f636ae2f5fd651b79d965473c3e9383d3521cdf896f9bc289709185da3f683", size = 448340, upload-time = "2025-11-01T16:22:36.723Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/b5/fe/108e7773349d500cf363328c3d0b7123e03feda51e310a3a5b136ac8ca71/textual_serve-1.1.3-py3-none-any.whl", hash = "sha256:207a472bc6604e725b1adab4ab8bf12f4c4dc25b04eea31e4d04731d8bf30f18", size = 447339, upload-time = "2025-11-01T16:22:35.209Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tf-playwright-stealth"
|
||||
version = "1.1.2"
|
||||
@@ -5949,6 +6175,15 @@ 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 = "uc-micro-py"
|
||||
version = "1.0.3"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/91/7a/146a99696aee0609e3712f2b44c6274566bc368dfe8375191278045186b8/uc-micro-py-1.0.3.tar.gz", hash = "sha256:d321b92cff673ec58027c04015fcaa8bb1e005478643ff4a500882eaab88c48a", size = 6043, upload-time = "2024-02-09T16:52:01.654Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/37/87/1f677586e8ac487e29672e4b17455758fce261de06a0d086167bb760361a/uc_micro_py-1.0.3-py3-none-any.whl", hash = "sha256:db1dffff340817673d7b466ec86114a9dc0e9d4d9b5ba229d9d60e5c12600cd5", size = 6229, upload-time = "2024-02-09T16:52:00.371Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "urllib3"
|
||||
version = "2.3.0"
|
||||
|
||||
@@ -0,0 +1 @@
|
||||
"""Tests for PraisonAI Queue System."""
|
||||
@@ -0,0 +1,228 @@
|
||||
"""Tests for Queue System Data Models."""
|
||||
|
||||
import pytest
|
||||
import time
|
||||
from praisonai.cli.features.queue.models import (
|
||||
QueuedRun,
|
||||
RunState,
|
||||
RunPriority,
|
||||
QueueConfig,
|
||||
QueueStats,
|
||||
StreamChunk,
|
||||
QueueEvent,
|
||||
)
|
||||
|
||||
|
||||
class TestRunState:
|
||||
"""Tests for RunState enum."""
|
||||
|
||||
def test_terminal_states(self):
|
||||
"""Test terminal state detection."""
|
||||
assert RunState.SUCCEEDED.is_terminal()
|
||||
assert RunState.FAILED.is_terminal()
|
||||
assert RunState.CANCELLED.is_terminal()
|
||||
assert not RunState.QUEUED.is_terminal()
|
||||
assert not RunState.RUNNING.is_terminal()
|
||||
assert not RunState.PAUSED.is_terminal()
|
||||
|
||||
def test_active_states(self):
|
||||
"""Test active state detection."""
|
||||
assert RunState.QUEUED.is_active()
|
||||
assert RunState.RUNNING.is_active()
|
||||
assert RunState.PAUSED.is_active()
|
||||
assert not RunState.SUCCEEDED.is_active()
|
||||
assert not RunState.FAILED.is_active()
|
||||
|
||||
|
||||
class TestRunPriority:
|
||||
"""Tests for RunPriority enum."""
|
||||
|
||||
def test_priority_ordering(self):
|
||||
"""Test priority ordering."""
|
||||
assert RunPriority.URGENT > RunPriority.HIGH
|
||||
assert RunPriority.HIGH > RunPriority.NORMAL
|
||||
assert RunPriority.NORMAL > RunPriority.LOW
|
||||
|
||||
def test_from_string(self):
|
||||
"""Test parsing from string."""
|
||||
assert RunPriority.from_string("low") == RunPriority.LOW
|
||||
assert RunPriority.from_string("NORMAL") == RunPriority.NORMAL
|
||||
assert RunPriority.from_string("High") == RunPriority.HIGH
|
||||
assert RunPriority.from_string("urgent") == RunPriority.URGENT
|
||||
assert RunPriority.from_string("invalid") == RunPriority.NORMAL
|
||||
|
||||
|
||||
class TestQueuedRun:
|
||||
"""Tests for QueuedRun dataclass."""
|
||||
|
||||
def test_default_values(self):
|
||||
"""Test default values."""
|
||||
run = QueuedRun()
|
||||
assert run.run_id # Auto-generated
|
||||
assert run.agent_name == ""
|
||||
assert run.state == RunState.QUEUED
|
||||
assert run.priority == RunPriority.NORMAL
|
||||
assert run.retry_count == 0
|
||||
assert run.max_retries == 3
|
||||
|
||||
def test_custom_values(self):
|
||||
"""Test custom values."""
|
||||
run = QueuedRun(
|
||||
run_id="test123",
|
||||
agent_name="TestAgent",
|
||||
input_content="Hello",
|
||||
state=RunState.RUNNING,
|
||||
priority=RunPriority.HIGH,
|
||||
)
|
||||
assert run.run_id == "test123"
|
||||
assert run.agent_name == "TestAgent"
|
||||
assert run.input_content == "Hello"
|
||||
assert run.state == RunState.RUNNING
|
||||
assert run.priority == RunPriority.HIGH
|
||||
|
||||
def test_duration_calculation(self):
|
||||
"""Test duration calculation."""
|
||||
run = QueuedRun()
|
||||
assert run.duration_seconds is None
|
||||
|
||||
run.started_at = time.time() - 5
|
||||
assert run.duration_seconds is not None
|
||||
assert run.duration_seconds >= 5
|
||||
|
||||
run.ended_at = run.started_at + 10
|
||||
assert abs(run.duration_seconds - 10) < 0.1
|
||||
|
||||
def test_wait_time_calculation(self):
|
||||
"""Test wait time calculation."""
|
||||
run = QueuedRun()
|
||||
run.created_at = time.time() - 3
|
||||
assert run.wait_seconds >= 3
|
||||
|
||||
run.started_at = run.created_at + 2
|
||||
assert abs(run.wait_seconds - 2) < 0.1
|
||||
|
||||
def test_can_retry(self):
|
||||
"""Test retry eligibility."""
|
||||
run = QueuedRun(state=RunState.FAILED, retry_count=0, max_retries=3)
|
||||
assert run.can_retry()
|
||||
|
||||
run.retry_count = 3
|
||||
assert not run.can_retry()
|
||||
|
||||
run.retry_count = 0
|
||||
run.state = RunState.SUCCEEDED
|
||||
assert not run.can_retry()
|
||||
|
||||
def test_to_dict(self):
|
||||
"""Test serialization to dict."""
|
||||
run = QueuedRun(
|
||||
run_id="test123",
|
||||
agent_name="TestAgent",
|
||||
input_content="Hello",
|
||||
)
|
||||
d = run.to_dict()
|
||||
|
||||
assert d["run_id"] == "test123"
|
||||
assert d["agent_name"] == "TestAgent"
|
||||
assert d["input_content"] == "Hello"
|
||||
assert d["state"] == "queued"
|
||||
assert d["priority"] == 1
|
||||
|
||||
def test_from_dict(self):
|
||||
"""Test deserialization from dict."""
|
||||
d = {
|
||||
"run_id": "test123",
|
||||
"agent_name": "TestAgent",
|
||||
"input_content": "Hello",
|
||||
"state": "running",
|
||||
"priority": 2,
|
||||
}
|
||||
run = QueuedRun.from_dict(d)
|
||||
|
||||
assert run.run_id == "test123"
|
||||
assert run.agent_name == "TestAgent"
|
||||
assert run.state == RunState.RUNNING
|
||||
assert run.priority == RunPriority.HIGH
|
||||
|
||||
def test_post_init_conversion(self):
|
||||
"""Test automatic type conversion in __post_init__."""
|
||||
run = QueuedRun(state="running", priority=2)
|
||||
assert run.state == RunState.RUNNING
|
||||
assert run.priority == RunPriority.HIGH
|
||||
|
||||
|
||||
class TestQueueConfig:
|
||||
"""Tests for QueueConfig dataclass."""
|
||||
|
||||
def test_default_values(self):
|
||||
"""Test default values."""
|
||||
config = QueueConfig()
|
||||
assert config.max_concurrent_global == 4
|
||||
assert config.max_concurrent_per_agent == 2
|
||||
assert config.max_queue_size == 100
|
||||
assert config.enable_persistence
|
||||
assert config.db_path == ".praison/queue.db"
|
||||
|
||||
def test_custom_values(self):
|
||||
"""Test custom values."""
|
||||
config = QueueConfig(
|
||||
max_concurrent_global=8,
|
||||
max_queue_size=200,
|
||||
enable_persistence=False,
|
||||
)
|
||||
assert config.max_concurrent_global == 8
|
||||
assert config.max_queue_size == 200
|
||||
assert not config.enable_persistence
|
||||
|
||||
def test_to_dict(self):
|
||||
"""Test serialization."""
|
||||
config = QueueConfig()
|
||||
d = config.to_dict()
|
||||
assert "max_concurrent_global" in d
|
||||
assert "db_path" in d
|
||||
|
||||
def test_from_dict(self):
|
||||
"""Test deserialization."""
|
||||
d = {"max_concurrent_global": 10, "max_queue_size": 50}
|
||||
config = QueueConfig.from_dict(d)
|
||||
assert config.max_concurrent_global == 10
|
||||
assert config.max_queue_size == 50
|
||||
|
||||
|
||||
class TestQueueStats:
|
||||
"""Tests for QueueStats dataclass."""
|
||||
|
||||
def test_active_count(self):
|
||||
"""Test active count calculation."""
|
||||
stats = QueueStats(queued_count=5, running_count=3)
|
||||
assert stats.active_count == 8
|
||||
|
||||
|
||||
class TestStreamChunk:
|
||||
"""Tests for StreamChunk dataclass."""
|
||||
|
||||
def test_creation(self):
|
||||
"""Test chunk creation."""
|
||||
chunk = StreamChunk(
|
||||
run_id="test123",
|
||||
content="Hello",
|
||||
chunk_index=0,
|
||||
)
|
||||
assert chunk.run_id == "test123"
|
||||
assert chunk.content == "Hello"
|
||||
assert chunk.chunk_index == 0
|
||||
assert not chunk.is_final
|
||||
|
||||
|
||||
class TestQueueEvent:
|
||||
"""Tests for QueueEvent dataclass."""
|
||||
|
||||
def test_creation(self):
|
||||
"""Test event creation."""
|
||||
event = QueueEvent(
|
||||
event_type="run_started",
|
||||
run_id="test123",
|
||||
)
|
||||
assert event.event_type == "run_started"
|
||||
assert event.run_id == "test123"
|
||||
assert event.timestamp > 0
|
||||
@@ -0,0 +1,251 @@
|
||||
"""Tests for Queue Persistence Layer."""
|
||||
|
||||
import pytest
|
||||
import tempfile
|
||||
import os
|
||||
from praisonai.cli.features.queue.models import QueuedRun, RunState, RunPriority
|
||||
from praisonai.cli.features.queue.persistence import QueuePersistence
|
||||
|
||||
|
||||
class TestQueuePersistence:
|
||||
"""Tests for QueuePersistence."""
|
||||
|
||||
@pytest.fixture
|
||||
def db_path(self):
|
||||
"""Create temporary database path."""
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
yield os.path.join(tmpdir, "test_queue.db")
|
||||
|
||||
@pytest.fixture
|
||||
def persistence(self, db_path):
|
||||
"""Create test persistence instance."""
|
||||
p = QueuePersistence(db_path)
|
||||
p.initialize()
|
||||
yield p
|
||||
p.close()
|
||||
|
||||
def test_initialize_creates_tables(self, persistence):
|
||||
"""Test initialization creates tables."""
|
||||
# Should not raise
|
||||
persistence.initialize()
|
||||
|
||||
def test_save_and_load_run(self, persistence):
|
||||
"""Test saving and loading a run."""
|
||||
run = QueuedRun(
|
||||
run_id="test123",
|
||||
agent_name="TestAgent",
|
||||
input_content="Hello",
|
||||
state=RunState.QUEUED,
|
||||
priority=RunPriority.HIGH,
|
||||
)
|
||||
|
||||
persistence.save_run(run)
|
||||
loaded = persistence.load_run("test123")
|
||||
|
||||
assert loaded is not None
|
||||
assert loaded.run_id == "test123"
|
||||
assert loaded.agent_name == "TestAgent"
|
||||
assert loaded.input_content == "Hello"
|
||||
assert loaded.state == RunState.QUEUED
|
||||
assert loaded.priority == RunPriority.HIGH
|
||||
|
||||
def test_load_nonexistent_run(self, persistence):
|
||||
"""Test loading a nonexistent run."""
|
||||
loaded = persistence.load_run("nonexistent")
|
||||
assert loaded is None
|
||||
|
||||
def test_update_run(self, persistence):
|
||||
"""Test updating a run."""
|
||||
run = QueuedRun(run_id="test123", agent_name="TestAgent")
|
||||
persistence.save_run(run)
|
||||
|
||||
run.state = RunState.RUNNING
|
||||
run.output_content = "Result"
|
||||
persistence.save_run(run)
|
||||
|
||||
loaded = persistence.load_run("test123")
|
||||
assert loaded.state == RunState.RUNNING
|
||||
assert loaded.output_content == "Result"
|
||||
|
||||
def test_list_runs(self, persistence):
|
||||
"""Test listing runs."""
|
||||
for i in range(5):
|
||||
run = QueuedRun(
|
||||
run_id=f"run{i}",
|
||||
agent_name="TestAgent",
|
||||
priority=RunPriority.NORMAL if i % 2 == 0 else RunPriority.HIGH,
|
||||
)
|
||||
persistence.save_run(run)
|
||||
|
||||
runs = persistence.list_runs(limit=10)
|
||||
assert len(runs) == 5
|
||||
|
||||
def test_list_runs_filter_by_state(self, persistence):
|
||||
"""Test filtering runs by state."""
|
||||
run1 = QueuedRun(run_id="run1", agent_name="Agent", state=RunState.QUEUED)
|
||||
run2 = QueuedRun(run_id="run2", agent_name="Agent", state=RunState.RUNNING)
|
||||
run3 = QueuedRun(run_id="run3", agent_name="Agent", state=RunState.QUEUED)
|
||||
|
||||
persistence.save_run(run1)
|
||||
persistence.save_run(run2)
|
||||
persistence.save_run(run3)
|
||||
|
||||
queued = persistence.list_runs(state=RunState.QUEUED)
|
||||
assert len(queued) == 2
|
||||
|
||||
running = persistence.list_runs(state=RunState.RUNNING)
|
||||
assert len(running) == 1
|
||||
|
||||
def test_list_runs_filter_by_session(self, persistence):
|
||||
"""Test filtering runs by session."""
|
||||
run1 = QueuedRun(run_id="run1", agent_name="Agent", session_id="session1")
|
||||
run2 = QueuedRun(run_id="run2", agent_name="Agent", session_id="session2")
|
||||
run3 = QueuedRun(run_id="run3", agent_name="Agent", session_id="session1")
|
||||
|
||||
persistence.save_run(run1)
|
||||
persistence.save_run(run2)
|
||||
persistence.save_run(run3)
|
||||
|
||||
session1_runs = persistence.list_runs(session_id="session1")
|
||||
assert len(session1_runs) == 2
|
||||
|
||||
def test_delete_run(self, persistence):
|
||||
"""Test deleting a run."""
|
||||
run = QueuedRun(run_id="test123", agent_name="TestAgent")
|
||||
persistence.save_run(run)
|
||||
|
||||
result = persistence.delete_run("test123")
|
||||
assert result is True
|
||||
|
||||
loaded = persistence.load_run("test123")
|
||||
assert loaded is None
|
||||
|
||||
def test_delete_nonexistent_run(self, persistence):
|
||||
"""Test deleting a nonexistent run."""
|
||||
result = persistence.delete_run("nonexistent")
|
||||
assert result is False
|
||||
|
||||
def test_update_run_state(self, persistence):
|
||||
"""Test updating run state."""
|
||||
run = QueuedRun(run_id="test123", agent_name="TestAgent")
|
||||
persistence.save_run(run)
|
||||
|
||||
result = persistence.update_run_state(
|
||||
"test123",
|
||||
RunState.FAILED,
|
||||
error="Test error",
|
||||
)
|
||||
|
||||
assert result is True
|
||||
|
||||
loaded = persistence.load_run("test123")
|
||||
assert loaded.state == RunState.FAILED
|
||||
assert loaded.error == "Test error"
|
||||
assert loaded.ended_at is not None
|
||||
|
||||
def test_load_pending_runs(self, persistence):
|
||||
"""Test loading pending runs for crash recovery."""
|
||||
run1 = QueuedRun(run_id="run1", agent_name="Agent", state=RunState.QUEUED)
|
||||
run2 = QueuedRun(run_id="run2", agent_name="Agent", state=RunState.RUNNING)
|
||||
run3 = QueuedRun(run_id="run3", agent_name="Agent", state=RunState.SUCCEEDED)
|
||||
|
||||
persistence.save_run(run1)
|
||||
persistence.save_run(run2)
|
||||
persistence.save_run(run3)
|
||||
|
||||
pending = persistence.load_pending_runs()
|
||||
assert len(pending) == 2
|
||||
|
||||
run_ids = [r.run_id for r in pending]
|
||||
assert "run1" in run_ids
|
||||
assert "run2" in run_ids
|
||||
assert "run3" not in run_ids
|
||||
|
||||
def test_mark_interrupted_as_failed(self, persistence):
|
||||
"""Test marking interrupted runs as failed."""
|
||||
run1 = QueuedRun(run_id="run1", agent_name="Agent", state=RunState.RUNNING)
|
||||
run2 = QueuedRun(run_id="run2", agent_name="Agent", state=RunState.RUNNING)
|
||||
run3 = QueuedRun(run_id="run3", agent_name="Agent", state=RunState.QUEUED)
|
||||
|
||||
persistence.save_run(run1)
|
||||
persistence.save_run(run2)
|
||||
persistence.save_run(run3)
|
||||
|
||||
count = persistence.mark_interrupted_as_failed()
|
||||
assert count == 2
|
||||
|
||||
loaded1 = persistence.load_run("run1")
|
||||
assert loaded1.state == RunState.FAILED
|
||||
assert "Interrupted" in loaded1.error
|
||||
|
||||
def test_get_stats(self, persistence):
|
||||
"""Test getting queue statistics."""
|
||||
runs = [
|
||||
QueuedRun(run_id="run1", agent_name="Agent", state=RunState.QUEUED),
|
||||
QueuedRun(run_id="run2", agent_name="Agent", state=RunState.RUNNING),
|
||||
QueuedRun(run_id="run3", agent_name="Agent", state=RunState.SUCCEEDED),
|
||||
QueuedRun(run_id="run4", agent_name="Agent", state=RunState.FAILED),
|
||||
]
|
||||
|
||||
for run in runs:
|
||||
persistence.save_run(run)
|
||||
|
||||
stats = persistence.get_stats()
|
||||
|
||||
assert stats.queued_count == 1
|
||||
assert stats.running_count == 1
|
||||
assert stats.succeeded_count == 1
|
||||
assert stats.failed_count == 1
|
||||
assert stats.total_runs == 4
|
||||
|
||||
def test_save_and_load_session(self, persistence):
|
||||
"""Test saving and loading session."""
|
||||
persistence.save_session(
|
||||
session_id="session123",
|
||||
user_id="user456",
|
||||
state={"key": "value"},
|
||||
config={"model": "gpt-4"},
|
||||
)
|
||||
|
||||
loaded = persistence.load_session("session123")
|
||||
|
||||
assert loaded is not None
|
||||
assert loaded["session_id"] == "session123"
|
||||
assert loaded["user_id"] == "user456"
|
||||
assert loaded["state"] == {"key": "value"}
|
||||
assert loaded["config"] == {"model": "gpt-4"}
|
||||
|
||||
def test_list_sessions(self, persistence):
|
||||
"""Test listing sessions."""
|
||||
for i in range(3):
|
||||
persistence.save_session(f"session{i}")
|
||||
|
||||
sessions = persistence.list_sessions()
|
||||
assert len(sessions) == 3
|
||||
|
||||
def test_cleanup_old_runs(self, persistence):
|
||||
"""Test cleaning up old runs."""
|
||||
import time
|
||||
|
||||
# Create old run
|
||||
old_run = QueuedRun(
|
||||
run_id="old_run",
|
||||
agent_name="Agent",
|
||||
state=RunState.SUCCEEDED,
|
||||
)
|
||||
old_run.created_at = time.time() - (40 * 24 * 60 * 60) # 40 days ago
|
||||
persistence.save_run(old_run)
|
||||
|
||||
# Create recent run
|
||||
recent_run = QueuedRun(
|
||||
run_id="recent_run",
|
||||
agent_name="Agent",
|
||||
state=RunState.SUCCEEDED,
|
||||
)
|
||||
persistence.save_run(recent_run)
|
||||
|
||||
count = persistence.cleanup_old_runs(days=30)
|
||||
assert count == 1
|
||||
|
||||
assert persistence.load_run("old_run") is None
|
||||
assert persistence.load_run("recent_run") is not None
|
||||
@@ -0,0 +1,264 @@
|
||||
"""Tests for Queue Scheduler."""
|
||||
|
||||
import pytest
|
||||
import asyncio
|
||||
from praisonai.cli.features.queue.models import QueuedRun, RunState, RunPriority, QueueConfig
|
||||
from praisonai.cli.features.queue.scheduler import QueueScheduler, QueueFullError
|
||||
|
||||
|
||||
class TestQueueScheduler:
|
||||
"""Tests for QueueScheduler."""
|
||||
|
||||
@pytest.fixture
|
||||
def config(self):
|
||||
"""Create test config."""
|
||||
return QueueConfig(
|
||||
max_concurrent_global=2,
|
||||
max_concurrent_per_agent=1,
|
||||
max_queue_size=5,
|
||||
)
|
||||
|
||||
@pytest.fixture
|
||||
def scheduler(self, config):
|
||||
"""Create test scheduler."""
|
||||
return QueueScheduler(config)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_submit_run(self, scheduler):
|
||||
"""Test submitting a run."""
|
||||
run = QueuedRun(
|
||||
agent_name="TestAgent",
|
||||
input_content="Hello",
|
||||
)
|
||||
|
||||
run_id = await scheduler.submit(run)
|
||||
|
||||
assert run_id == run.run_id
|
||||
assert scheduler.queued_count == 1
|
||||
assert run.state == RunState.QUEUED
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_submit_duplicate_rejected(self, scheduler):
|
||||
"""Test duplicate run_id is rejected."""
|
||||
run = QueuedRun(run_id="test123", agent_name="TestAgent")
|
||||
await scheduler.submit(run)
|
||||
|
||||
run2 = QueuedRun(run_id="test123", agent_name="TestAgent")
|
||||
with pytest.raises(ValueError, match="already exists"):
|
||||
await scheduler.submit(run2)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_queue_full_error(self, scheduler):
|
||||
"""Test queue full error."""
|
||||
for i in range(5):
|
||||
run = QueuedRun(agent_name="TestAgent", input_content=f"Test {i}")
|
||||
await scheduler.submit(run)
|
||||
|
||||
run = QueuedRun(agent_name="TestAgent", input_content="Overflow")
|
||||
with pytest.raises(QueueFullError):
|
||||
await scheduler.submit(run)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_next_returns_highest_priority(self, scheduler):
|
||||
"""Test next() returns highest priority run."""
|
||||
low = QueuedRun(agent_name="Agent1", priority=RunPriority.LOW)
|
||||
normal = QueuedRun(agent_name="Agent2", priority=RunPriority.NORMAL)
|
||||
high = QueuedRun(agent_name="Agent3", priority=RunPriority.HIGH)
|
||||
|
||||
await scheduler.submit(low)
|
||||
await scheduler.submit(normal)
|
||||
await scheduler.submit(high)
|
||||
|
||||
next_run = await scheduler.next()
|
||||
assert next_run.priority == RunPriority.HIGH
|
||||
assert next_run.state == RunState.RUNNING
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_fifo_within_priority(self, scheduler):
|
||||
"""Test FIFO ordering within same priority."""
|
||||
run1 = QueuedRun(run_id="first", agent_name="Agent1")
|
||||
run2 = QueuedRun(run_id="second", agent_name="Agent2")
|
||||
|
||||
await scheduler.submit(run1)
|
||||
await scheduler.submit(run2)
|
||||
|
||||
next_run = await scheduler.next()
|
||||
assert next_run.run_id == "first"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_concurrency_limit_global(self, scheduler):
|
||||
"""Test global concurrency limit."""
|
||||
run1 = QueuedRun(agent_name="Agent1")
|
||||
run2 = QueuedRun(agent_name="Agent2")
|
||||
run3 = QueuedRun(agent_name="Agent3")
|
||||
|
||||
await scheduler.submit(run1)
|
||||
await scheduler.submit(run2)
|
||||
await scheduler.submit(run3)
|
||||
|
||||
# Start first two
|
||||
await scheduler.next()
|
||||
await scheduler.next()
|
||||
|
||||
# Third should not start (limit is 2)
|
||||
next_run = await scheduler.next()
|
||||
assert next_run is None
|
||||
assert scheduler.running_count == 2
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_concurrency_limit_per_agent(self, scheduler):
|
||||
"""Test per-agent concurrency limit."""
|
||||
run1 = QueuedRun(agent_name="SameAgent")
|
||||
run2 = QueuedRun(agent_name="SameAgent")
|
||||
run3 = QueuedRun(agent_name="DifferentAgent")
|
||||
|
||||
await scheduler.submit(run1)
|
||||
await scheduler.submit(run2)
|
||||
await scheduler.submit(run3)
|
||||
|
||||
# Start first
|
||||
await scheduler.next()
|
||||
|
||||
# Second same agent should not start
|
||||
next_run = await scheduler.next()
|
||||
assert next_run.agent_name == "DifferentAgent"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_complete_run(self, scheduler):
|
||||
"""Test completing a run."""
|
||||
run = QueuedRun(agent_name="TestAgent")
|
||||
await scheduler.submit(run)
|
||||
await scheduler.next()
|
||||
|
||||
completed = await scheduler.complete(run.run_id, output="Result")
|
||||
|
||||
assert completed.state == RunState.SUCCEEDED
|
||||
assert completed.output_content == "Result"
|
||||
assert completed.ended_at is not None
|
||||
assert scheduler.running_count == 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_fail_run(self, scheduler):
|
||||
"""Test failing a run."""
|
||||
run = QueuedRun(agent_name="TestAgent")
|
||||
await scheduler.submit(run)
|
||||
await scheduler.next()
|
||||
|
||||
failed = await scheduler.fail(run.run_id, error="Test error")
|
||||
|
||||
assert failed.state == RunState.FAILED
|
||||
assert failed.error == "Test error"
|
||||
assert failed.ended_at is not None
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_cancel_queued_run(self, scheduler):
|
||||
"""Test cancelling a queued run."""
|
||||
run = QueuedRun(agent_name="TestAgent")
|
||||
await scheduler.submit(run)
|
||||
|
||||
result = await scheduler.cancel(run.run_id)
|
||||
|
||||
assert result is True
|
||||
assert scheduler.queued_count == 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_cancel_running_run(self, scheduler):
|
||||
"""Test cancelling a running run."""
|
||||
run = QueuedRun(agent_name="TestAgent")
|
||||
await scheduler.submit(run)
|
||||
await scheduler.next()
|
||||
|
||||
result = await scheduler.cancel(run.run_id)
|
||||
|
||||
assert result is True
|
||||
assert scheduler.running_count == 0
|
||||
assert scheduler.is_cancelled(run.run_id)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_retry_failed_run(self, scheduler):
|
||||
"""Test retrying a failed run."""
|
||||
run = QueuedRun(agent_name="TestAgent", input_content="Hello")
|
||||
await scheduler.submit(run)
|
||||
await scheduler.next()
|
||||
await scheduler.fail(run.run_id, error="Test error")
|
||||
|
||||
new_id = await scheduler.retry(run.run_id)
|
||||
|
||||
assert new_id is not None
|
||||
assert new_id != run.run_id
|
||||
|
||||
new_run = scheduler.get_run(new_id)
|
||||
assert new_run.retry_count == 1
|
||||
assert new_run.parent_run_id == run.run_id
|
||||
assert new_run.input_content == "Hello"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_retry_max_retries_exceeded(self, scheduler):
|
||||
"""Test retry fails when max retries exceeded."""
|
||||
run = QueuedRun(
|
||||
agent_name="TestAgent",
|
||||
retry_count=3,
|
||||
max_retries=3,
|
||||
state=RunState.FAILED,
|
||||
)
|
||||
await scheduler.submit(run, check_duplicate=False)
|
||||
|
||||
new_id = await scheduler.retry(run.run_id)
|
||||
assert new_id is None
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_queued(self, scheduler):
|
||||
"""Test getting queued runs."""
|
||||
run1 = QueuedRun(agent_name="Agent1", priority=RunPriority.LOW)
|
||||
run2 = QueuedRun(agent_name="Agent2", priority=RunPriority.HIGH)
|
||||
|
||||
await scheduler.submit(run1)
|
||||
await scheduler.submit(run2)
|
||||
|
||||
queued = scheduler.get_queued()
|
||||
|
||||
assert len(queued) == 2
|
||||
# Should be in priority order
|
||||
assert queued[0].priority == RunPriority.HIGH
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_clear_queue(self, scheduler):
|
||||
"""Test clearing the queue."""
|
||||
for i in range(3):
|
||||
run = QueuedRun(agent_name=f"Agent{i}")
|
||||
await scheduler.submit(run)
|
||||
|
||||
count = await scheduler.clear_queue()
|
||||
|
||||
assert count == 3
|
||||
assert scheduler.queued_count == 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_load_runs(self, scheduler):
|
||||
"""Test loading runs from persistence."""
|
||||
runs = [
|
||||
QueuedRun(run_id="run1", agent_name="Agent1", state=RunState.QUEUED),
|
||||
QueuedRun(run_id="run2", agent_name="Agent2", state=RunState.RUNNING),
|
||||
]
|
||||
|
||||
scheduler.load_runs(runs)
|
||||
|
||||
# Running run should be re-queued
|
||||
assert scheduler.queued_count == 2
|
||||
assert scheduler.running_count == 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_event_callbacks(self, scheduler):
|
||||
"""Test event callbacks are called."""
|
||||
events = []
|
||||
|
||||
def callback(event):
|
||||
events.append(event)
|
||||
|
||||
scheduler.add_event_callback(callback)
|
||||
|
||||
run = QueuedRun(agent_name="TestAgent")
|
||||
await scheduler.submit(run)
|
||||
|
||||
assert len(events) == 1
|
||||
assert events[0].event_type == "run_submitted"
|
||||
@@ -0,0 +1 @@
|
||||
"""Tests for PraisonAI TUI."""
|
||||
@@ -0,0 +1,73 @@
|
||||
"""Tests for TUI Events."""
|
||||
|
||||
import time
|
||||
from praisonai.cli.features.tui.events import TUIEvent, TUIEventType
|
||||
|
||||
|
||||
class TestTUIEventType:
|
||||
"""Tests for TUIEventType enum."""
|
||||
|
||||
def test_event_types_exist(self):
|
||||
"""Test all expected event types exist."""
|
||||
assert TUIEventType.MESSAGE_SUBMITTED
|
||||
assert TUIEventType.RUN_QUEUED
|
||||
assert TUIEventType.RUN_STARTED
|
||||
assert TUIEventType.RUN_COMPLETED
|
||||
assert TUIEventType.OUTPUT_CHUNK
|
||||
assert TUIEventType.TOOL_CALL_STARTED
|
||||
assert TUIEventType.ERROR_OCCURRED
|
||||
|
||||
|
||||
class TestTUIEvent:
|
||||
"""Tests for TUIEvent dataclass."""
|
||||
|
||||
def test_basic_creation(self):
|
||||
"""Test basic event creation."""
|
||||
event = TUIEvent(
|
||||
event_type=TUIEventType.MESSAGE_SUBMITTED,
|
||||
)
|
||||
assert event.event_type == TUIEventType.MESSAGE_SUBMITTED
|
||||
assert event.timestamp > 0
|
||||
assert event.data == {}
|
||||
|
||||
def test_with_data(self):
|
||||
"""Test event with data."""
|
||||
event = TUIEvent(
|
||||
event_type=TUIEventType.RUN_COMPLETED,
|
||||
run_id="test123",
|
||||
data={"output": "Hello"},
|
||||
)
|
||||
assert event.run_id == "test123"
|
||||
assert event.data["output"] == "Hello"
|
||||
|
||||
def test_message_submitted_factory(self):
|
||||
"""Test message_submitted factory method."""
|
||||
event = TUIEvent.message_submitted("Hello world")
|
||||
assert event.event_type == TUIEventType.MESSAGE_SUBMITTED
|
||||
assert event.data["content"] == "Hello world"
|
||||
|
||||
def test_output_chunk_factory(self):
|
||||
"""Test output_chunk factory method."""
|
||||
event = TUIEvent.output_chunk("run123", "chunk content")
|
||||
assert event.event_type == TUIEventType.OUTPUT_CHUNK
|
||||
assert event.run_id == "run123"
|
||||
assert event.data["content"] == "chunk content"
|
||||
|
||||
def test_run_completed_factory(self):
|
||||
"""Test run_completed factory method."""
|
||||
event = TUIEvent.run_completed("run123", "final output")
|
||||
assert event.event_type == TUIEventType.RUN_COMPLETED
|
||||
assert event.run_id == "run123"
|
||||
assert event.data["output"] == "final output"
|
||||
|
||||
def test_error_factory(self):
|
||||
"""Test error factory method."""
|
||||
event = TUIEvent.error("Something went wrong")
|
||||
assert event.event_type == TUIEventType.ERROR_OCCURRED
|
||||
assert event.data["message"] == "Something went wrong"
|
||||
|
||||
def test_status_update_factory(self):
|
||||
"""Test status_update factory method."""
|
||||
event = TUIEvent.status_update("Processing...")
|
||||
assert event.event_type == TUIEventType.STATUS_UPDATED
|
||||
assert event.data["status"] == "Processing..."
|
||||
@@ -0,0 +1,231 @@
|
||||
"""Tests for Mock Provider."""
|
||||
|
||||
import pytest
|
||||
from praisonai.cli.features.tui.mock_provider import (
|
||||
MockProvider,
|
||||
MockProviderConfig,
|
||||
MockResponse,
|
||||
MockExecutor,
|
||||
create_mock_provider,
|
||||
)
|
||||
|
||||
|
||||
class TestMockResponse:
|
||||
"""Tests for MockResponse."""
|
||||
|
||||
def test_default_chunking(self):
|
||||
"""Test automatic chunking of content."""
|
||||
response = MockResponse(content="Hello world how are you today")
|
||||
|
||||
assert len(response.chunks) > 0
|
||||
assert "".join(response.chunks).strip() == response.content
|
||||
|
||||
def test_custom_chunks(self):
|
||||
"""Test custom chunks."""
|
||||
response = MockResponse(
|
||||
content="Hello world",
|
||||
chunks=["Hello", " world"],
|
||||
)
|
||||
|
||||
assert response.chunks == ["Hello", " world"]
|
||||
|
||||
def test_token_estimation(self):
|
||||
"""Test automatic token estimation."""
|
||||
response = MockResponse(content="one two three four five")
|
||||
|
||||
# Rough estimate: words * 2
|
||||
assert response.tokens == 10
|
||||
|
||||
|
||||
class TestMockProviderConfig:
|
||||
"""Tests for MockProviderConfig."""
|
||||
|
||||
def test_default_config(self):
|
||||
"""Test default configuration."""
|
||||
config = MockProviderConfig()
|
||||
|
||||
assert config.seed == 42
|
||||
assert config.default_delay == 0.05
|
||||
assert config.simulate_errors is False
|
||||
|
||||
def test_custom_responses(self):
|
||||
"""Test custom responses."""
|
||||
config = MockProviderConfig(
|
||||
responses={
|
||||
"custom": MockResponse(content="Custom response"),
|
||||
}
|
||||
)
|
||||
|
||||
assert "custom" in config.responses
|
||||
|
||||
|
||||
class TestMockProvider:
|
||||
"""Tests for MockProvider."""
|
||||
|
||||
def test_default_responses(self):
|
||||
"""Test default canned responses."""
|
||||
provider = MockProvider()
|
||||
|
||||
assert "hello" in provider.config.responses
|
||||
assert "help" in provider.config.responses
|
||||
assert "test" in provider.config.responses
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_generate_hello(self):
|
||||
"""Test generating hello response."""
|
||||
provider = MockProvider()
|
||||
|
||||
result = await provider.generate("hello")
|
||||
|
||||
assert "content" in result
|
||||
assert "Hello" in result["content"]
|
||||
assert result["tokens"] > 0
|
||||
assert result["cost"] > 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_generate_streaming(self):
|
||||
"""Test streaming generation."""
|
||||
provider = MockProvider()
|
||||
|
||||
chunks = []
|
||||
result = await provider.generate(
|
||||
"hello",
|
||||
stream=True,
|
||||
on_chunk=lambda c: chunks.append(c),
|
||||
)
|
||||
|
||||
assert len(chunks) > 0
|
||||
assert "".join(chunks).strip() == result["content"]
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_generate_non_streaming(self):
|
||||
"""Test non-streaming generation."""
|
||||
provider = MockProvider()
|
||||
|
||||
result = await provider.generate("hello", stream=False)
|
||||
|
||||
assert "content" in result
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_generate_error(self):
|
||||
"""Test error response."""
|
||||
provider = MockProvider()
|
||||
|
||||
with pytest.raises(Exception, match="Simulated error"):
|
||||
await provider.generate("error")
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_generate_tool_calls(self):
|
||||
"""Test tool call response."""
|
||||
provider = MockProvider()
|
||||
|
||||
result = await provider.generate("tool")
|
||||
|
||||
assert "tool_calls" in result
|
||||
assert len(result["tool_calls"]) > 0
|
||||
assert result["tool_calls"][0]["name"] == "mock_tool"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_deterministic_responses(self):
|
||||
"""Test responses are deterministic with same seed."""
|
||||
provider1 = MockProvider(MockProviderConfig(seed=123))
|
||||
provider2 = MockProvider(MockProviderConfig(seed=123))
|
||||
|
||||
result1 = await provider1.generate("unique input xyz")
|
||||
result2 = await provider2.generate("unique input xyz")
|
||||
|
||||
assert result1["content"] == result2["content"]
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_different_seeds_different_responses(self):
|
||||
"""Test different seeds produce different responses."""
|
||||
provider1 = MockProvider(MockProviderConfig(seed=1))
|
||||
provider2 = MockProvider(MockProviderConfig(seed=2))
|
||||
|
||||
# Use input that doesn't match canned responses
|
||||
result1 = await provider1.generate("random query abc123")
|
||||
result2 = await provider2.generate("random query abc123")
|
||||
|
||||
# Content should be deterministic based on input hash, not seed
|
||||
# But the random error injection would differ
|
||||
assert result1["content"] == result2["content"]
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stream_iterator(self):
|
||||
"""Test stream iterator."""
|
||||
provider = MockProvider()
|
||||
|
||||
chunks = []
|
||||
async for chunk in provider.stream("hello"):
|
||||
chunks.append(chunk)
|
||||
|
||||
assert len(chunks) > 0
|
||||
|
||||
def test_reset(self):
|
||||
"""Test reset functionality."""
|
||||
provider = MockProvider()
|
||||
provider._playback_index = 5
|
||||
|
||||
provider.reset()
|
||||
|
||||
assert provider._playback_index == 0
|
||||
|
||||
|
||||
class TestMockExecutor:
|
||||
"""Tests for MockExecutor."""
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_execute(self):
|
||||
"""Test executor execution."""
|
||||
executor = MockExecutor()
|
||||
|
||||
outputs = []
|
||||
result = await executor.execute(
|
||||
input_content="hello",
|
||||
agent_config={"model": "test-model"},
|
||||
on_output=lambda c: outputs.append(c),
|
||||
)
|
||||
|
||||
assert "output" in result
|
||||
assert "metrics" in result
|
||||
assert result["metrics"]["tokens"] > 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_execute_with_custom_provider(self):
|
||||
"""Test executor with custom provider."""
|
||||
provider = create_mock_provider(
|
||||
seed=42,
|
||||
responses={"custom": "Custom output"},
|
||||
)
|
||||
executor = MockExecutor(provider)
|
||||
|
||||
result = await executor.execute(
|
||||
input_content="custom query",
|
||||
agent_config={},
|
||||
)
|
||||
|
||||
assert "Custom output" in result["output"]
|
||||
|
||||
|
||||
class TestCreateMockProvider:
|
||||
"""Tests for create_mock_provider helper."""
|
||||
|
||||
def test_create_with_defaults(self):
|
||||
"""Test creating with defaults."""
|
||||
provider = create_mock_provider()
|
||||
|
||||
assert provider.config.seed == 42
|
||||
|
||||
def test_create_with_custom_seed(self):
|
||||
"""Test creating with custom seed."""
|
||||
provider = create_mock_provider(seed=123)
|
||||
|
||||
assert provider.config.seed == 123
|
||||
|
||||
def test_create_with_custom_responses(self):
|
||||
"""Test creating with custom responses."""
|
||||
provider = create_mock_provider(
|
||||
responses={"mypattern": "My response"}
|
||||
)
|
||||
|
||||
assert "mypattern" in provider.config.responses
|
||||
@@ -0,0 +1,412 @@
|
||||
"""Tests for TUI Orchestrator."""
|
||||
|
||||
import pytest
|
||||
import asyncio
|
||||
from praisonai.cli.features.tui.orchestrator import (
|
||||
TuiOrchestrator,
|
||||
UIStateModel,
|
||||
SimulationRunner,
|
||||
OutputMode,
|
||||
)
|
||||
from praisonai.cli.features.tui.events import TUIEvent, TUIEventType
|
||||
from praisonai.cli.features.queue import QueueConfig
|
||||
|
||||
|
||||
class TestUIStateModel:
|
||||
"""Tests for UIStateModel."""
|
||||
|
||||
def test_default_state(self):
|
||||
"""Test default state values."""
|
||||
state = UIStateModel()
|
||||
assert state.session_id == ""
|
||||
assert state.model == "gpt-4o-mini"
|
||||
assert state.messages == []
|
||||
assert state.is_processing is False
|
||||
assert state.current_screen == "main"
|
||||
|
||||
def test_add_message(self):
|
||||
"""Test adding messages."""
|
||||
state = UIStateModel()
|
||||
state.add_message("user", "Hello")
|
||||
state.add_message("assistant", "Hi there!")
|
||||
|
||||
assert len(state.messages) == 2
|
||||
assert state.messages[0]["role"] == "user"
|
||||
assert state.messages[0]["content"] == "Hello"
|
||||
assert state.messages[1]["role"] == "assistant"
|
||||
|
||||
def test_add_message_with_extras(self):
|
||||
"""Test adding messages with extra fields."""
|
||||
state = UIStateModel()
|
||||
state.add_message("assistant", "Response", run_id="run123", agent_name="Agent1")
|
||||
|
||||
assert state.messages[0]["run_id"] == "run123"
|
||||
assert state.messages[0]["agent_name"] == "Agent1"
|
||||
|
||||
def test_message_limit(self):
|
||||
"""Test message limit enforcement."""
|
||||
state = UIStateModel(max_messages=5)
|
||||
|
||||
for i in range(10):
|
||||
state.add_message("user", f"Message {i}")
|
||||
|
||||
assert len(state.messages) == 5
|
||||
assert state.messages[0]["content"] == "Message 5"
|
||||
|
||||
def test_add_event(self):
|
||||
"""Test adding events."""
|
||||
state = UIStateModel()
|
||||
event = TUIEvent(event_type=TUIEventType.MESSAGE_SUBMITTED)
|
||||
state.add_event(event)
|
||||
|
||||
assert len(state.events) == 1
|
||||
assert state.events[0]["type"] == "message_submitted"
|
||||
|
||||
def test_to_snapshot(self):
|
||||
"""Test snapshot generation."""
|
||||
state = UIStateModel(session_id="test123", model="gpt-4")
|
||||
state.add_message("user", "Hello")
|
||||
state.is_processing = True
|
||||
|
||||
snapshot = state.to_snapshot()
|
||||
|
||||
assert snapshot["session_id"] == "test123"
|
||||
assert snapshot["model"] == "gpt-4"
|
||||
assert snapshot["message_count"] == 1
|
||||
assert snapshot["is_processing"] is True
|
||||
|
||||
def test_render_snapshot_pretty(self):
|
||||
"""Test pretty snapshot rendering."""
|
||||
state = UIStateModel(session_id="test123")
|
||||
state.add_message("user", "Hello")
|
||||
state.add_message("assistant", "Hi!")
|
||||
|
||||
output = state.render_snapshot_pretty()
|
||||
|
||||
assert "PraisonAI" in output
|
||||
assert "test123" in output
|
||||
assert "USER:" in output
|
||||
assert "ASSISTANT:" in output
|
||||
|
||||
|
||||
class TestTuiOrchestrator:
|
||||
"""Tests for TuiOrchestrator."""
|
||||
|
||||
@pytest.fixture
|
||||
def config(self):
|
||||
"""Create test config."""
|
||||
return QueueConfig(enable_persistence=False)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_start_stop(self, config):
|
||||
"""Test starting and stopping orchestrator."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
|
||||
await orchestrator.start(session_id="test-session")
|
||||
|
||||
assert orchestrator.state.session_id == "test-session"
|
||||
assert orchestrator._running is True
|
||||
|
||||
await orchestrator.stop()
|
||||
|
||||
assert orchestrator._running is False
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_event_callbacks(self, config):
|
||||
"""Test event callbacks are called."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
|
||||
events = []
|
||||
orchestrator.add_event_callback(lambda e: events.append(e))
|
||||
|
||||
await orchestrator.start()
|
||||
|
||||
# Should have session started event
|
||||
assert len(events) >= 1
|
||||
assert events[0].event_type == TUIEventType.SESSION_STARTED
|
||||
|
||||
await orchestrator.stop()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_set_model(self, config):
|
||||
"""Test setting model."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
|
||||
await orchestrator.start()
|
||||
orchestrator.set_model("gpt-4")
|
||||
|
||||
assert orchestrator.state.model == "gpt-4"
|
||||
|
||||
await orchestrator.stop()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_navigate_screen(self, config):
|
||||
"""Test screen navigation."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
|
||||
await orchestrator.start()
|
||||
orchestrator.navigate_screen("queue")
|
||||
|
||||
assert orchestrator.state.current_screen == "queue"
|
||||
|
||||
await orchestrator.stop()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_set_focus(self, config):
|
||||
"""Test focus setting."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
|
||||
await orchestrator.start()
|
||||
orchestrator.set_focus("chat")
|
||||
|
||||
assert orchestrator.state.focused_widget == "chat"
|
||||
|
||||
await orchestrator.stop()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_snapshot(self, config):
|
||||
"""Test getting snapshot."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
|
||||
await orchestrator.start(session_id="snap-test")
|
||||
|
||||
snapshot = orchestrator.get_snapshot()
|
||||
|
||||
assert snapshot["session_id"] == "snap-test"
|
||||
assert "model" in snapshot
|
||||
assert "is_processing" in snapshot
|
||||
|
||||
await orchestrator.stop()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_render_snapshot(self, config):
|
||||
"""Test rendering snapshot."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
|
||||
await orchestrator.start()
|
||||
|
||||
output = orchestrator.render_snapshot()
|
||||
|
||||
assert "PraisonAI" in output
|
||||
assert "Queue" in output
|
||||
|
||||
await orchestrator.stop()
|
||||
|
||||
|
||||
class TestSimulationRunner:
|
||||
"""Tests for SimulationRunner."""
|
||||
|
||||
@pytest.fixture
|
||||
def config(self):
|
||||
"""Create test config."""
|
||||
return QueueConfig(enable_persistence=False)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run_empty_script(self, config):
|
||||
"""Test running empty script."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
runner = SimulationRunner(orchestrator)
|
||||
|
||||
script = {"steps": []}
|
||||
success = await runner.run_script(script)
|
||||
|
||||
assert success is True
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run_navigate_step(self, config):
|
||||
"""Test navigate step."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
runner = SimulationRunner(orchestrator)
|
||||
|
||||
script = {
|
||||
"steps": [
|
||||
{"action": "navigate", "args": {"screen": "queue"}},
|
||||
{"action": "navigate", "args": {"screen": "main"}},
|
||||
]
|
||||
}
|
||||
|
||||
success = await runner.run_script(script)
|
||||
|
||||
assert success is True
|
||||
assert orchestrator.state.current_screen == "main"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run_focus_step(self, config):
|
||||
"""Test focus step."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
runner = SimulationRunner(orchestrator)
|
||||
|
||||
script = {
|
||||
"steps": [
|
||||
{"action": "focus", "args": {"widget": "chat"}},
|
||||
]
|
||||
}
|
||||
|
||||
success = await runner.run_script(script)
|
||||
|
||||
assert success is True
|
||||
assert orchestrator.state.focused_widget == "chat"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run_model_step(self, config):
|
||||
"""Test model step."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
runner = SimulationRunner(orchestrator)
|
||||
|
||||
script = {
|
||||
"steps": [
|
||||
{"action": "model", "args": {"model": "gpt-4"}},
|
||||
]
|
||||
}
|
||||
|
||||
success = await runner.run_script(script)
|
||||
|
||||
assert success is True
|
||||
assert orchestrator.state.model == "gpt-4"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run_sleep_step(self, config):
|
||||
"""Test sleep step."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
runner = SimulationRunner(orchestrator)
|
||||
|
||||
script = {
|
||||
"steps": [
|
||||
{"action": "sleep", "args": {"seconds": 0.1}},
|
||||
]
|
||||
}
|
||||
|
||||
import time
|
||||
start = time.time()
|
||||
success = await runner.run_script(script)
|
||||
elapsed = time.time() - start
|
||||
|
||||
assert success is True
|
||||
assert elapsed >= 0.1
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_assertion_mode_pass(self, config):
|
||||
"""Test assertion mode with passing assertions."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
runner = SimulationRunner(orchestrator, assert_mode=True)
|
||||
|
||||
script = {
|
||||
"session_id": "assert-test",
|
||||
"steps": [
|
||||
{
|
||||
"action": "model",
|
||||
"args": {"model": "gpt-4"},
|
||||
"expected": {"model": "gpt-4"}
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
success = await runner.run_script(script)
|
||||
summary = runner.get_summary()
|
||||
|
||||
assert success is True
|
||||
assert summary["assertions_passed"] == 1
|
||||
assert summary["assertions_failed"] == 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_assertion_mode_fail(self, config):
|
||||
"""Test assertion mode with failing assertions."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
runner = SimulationRunner(orchestrator, assert_mode=True)
|
||||
|
||||
script = {
|
||||
"steps": [
|
||||
{
|
||||
"action": "model",
|
||||
"args": {"model": "gpt-4"},
|
||||
"expected": {"model": "gpt-3.5"} # Wrong expectation
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
success = await runner.run_script(script)
|
||||
summary = runner.get_summary()
|
||||
|
||||
assert success is False
|
||||
assert summary["assertions_failed"] == 1
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_unknown_action(self, config):
|
||||
"""Test unknown action raises error."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
runner = SimulationRunner(orchestrator)
|
||||
|
||||
script = {
|
||||
"steps": [
|
||||
{"action": "unknown_action", "args": {}},
|
||||
]
|
||||
}
|
||||
|
||||
success = await runner.run_script(script)
|
||||
|
||||
assert success is False
|
||||
assert len(runner.errors) > 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_summary(self, config):
|
||||
"""Test getting summary."""
|
||||
orchestrator = TuiOrchestrator(
|
||||
queue_config=config,
|
||||
output_mode=OutputMode.SILENT,
|
||||
)
|
||||
runner = SimulationRunner(orchestrator)
|
||||
|
||||
script = {"steps": []}
|
||||
await runner.run_script(script)
|
||||
|
||||
summary = runner.get_summary()
|
||||
|
||||
assert "assertions_passed" in summary
|
||||
assert "assertions_failed" in summary
|
||||
assert "errors" in summary
|
||||
assert "success" in summary
|
||||
Reference in new issue
Block a user