mindroom 0.0.0__py3-none-any.whl → 0.1.0__py3-none-any.whl

mindroom/cli.py

@@ -0,0 +1,86 @@
+"""Mindroom CLI - Simplified multi-agent Matrix bot system."""
+
+from __future__ import annotations
+
+import asyncio
+import sys
+from pathlib import Path
+
+import typer
+from rich.console import Console
+
+from mindroom import __version__
+from mindroom.bot import main as bot_main
+from mindroom.constants import STORAGE_PATH
+
+app = typer.Typer(
+    help="Mindroom: Multi-agent Matrix bot system",
+    pretty_exceptions_enable=True,
+    # Disable showing locals which can be very large (also see `setup_logging`)
+    pretty_exceptions_show_locals=False,
+)
+console = Console()
+
+
+@app.command()
+def version() -> None:
+    """Show the current version of Mindroom."""
+    console.print(f"Mindroom version: [bold]{__version__}[/bold]")
+    console.print("Multi-agent Matrix bot system")
+
+
+@app.command()
+def run(
+    log_level: str = typer.Option(
+        "INFO",
+        "--log-level",
+        "-l",
+        help="Set the logging level (DEBUG, INFO, WARNING, ERROR)",
+        case_sensitive=False,
+    ),
+    storage_path: Path = typer.Option(  # noqa: B008
+        Path(STORAGE_PATH),
+        "--storage-path",
+        "-s",
+        help="Base directory for persistent MindRoom data (state, sessions, tracking)",
+    ),
+) -> None:
+    """Run the mindroom multi-agent system.
+
+    This command starts the multi-agent bot system which automatically:
+    - Creates all necessary user and agent accounts
+    - Creates all rooms defined in config.yaml
+    - Manages agent room memberships
+    """
+    asyncio.run(_run(log_level=log_level.upper(), storage_path=storage_path))
+
+
+async def _run(log_level: str, storage_path: Path) -> None:
+    """Run the multi-agent system."""
+    console.print(f"🚀 Starting Mindroom multi-agent system (log level: {log_level})...")
+    console.print("Press Ctrl+C to stop\n")
+
+    try:
+        await bot_main(log_level=log_level, storage_path=storage_path)
+    except KeyboardInterrupt:
+        console.print("\n✋ Stopped")
+
+
+def main() -> None:
+    """Main entry point that shows help by default."""
+    # Handle -h flag by replacing with --help
+    for i, arg in enumerate(sys.argv):
+        if arg == "-h":
+            sys.argv[i] = "--help"
+            break
+
+    # If no arguments provided, show help
+    if len(sys.argv) == 1:
+        # Show help by appending --help to argv
+        sys.argv.append("--help")
+
+    app()
+
+
+if __name__ == "__main__":
+    main()

mindroom/commands.py

@@ -0,0 +1,377 @@
+"""Command parsing and handling for user commands."""
+
+from __future__ import annotations
+
+import re
+from dataclasses import dataclass
+from enum import Enum
+from typing import Any
+
+import nio
+
+from .constants import VOICE_PREFIX
+from .logging_config import get_logger
+
+logger = get_logger(__name__)
+
+
+class CommandType(Enum):
+    """Types of commands supported."""
+
+    HELP = "help"
+    SCHEDULE = "schedule"
+    LIST_SCHEDULES = "list_schedules"
+    CANCEL_SCHEDULE = "cancel_schedule"
+    WIDGET = "widget"
+    CONFIG = "config"  # Configuration command
+    HI = "hi"  # Welcome message command
+    UNKNOWN = "unknown"  # Special type for unrecognized commands
+
+
+# Command documentation for each command type
+COMMAND_DOCS = {
+    CommandType.SCHEDULE: ("!schedule <task>", "Schedule a task"),
+    CommandType.LIST_SCHEDULES: ("!list_schedules", "List scheduled tasks"),
+    CommandType.CANCEL_SCHEDULE: ("!cancel_schedule <id>", "Cancel a scheduled task"),
+    CommandType.HELP: ("!help [topic]", "Get help"),
+    CommandType.WIDGET: ("!widget [url]", "Add configuration widget"),
+    CommandType.CONFIG: ("!config <operation>", "Manage configuration"),
+    CommandType.HI: ("!hi", "Show welcome message"),
+}
+
+
+def _get_command_entries(format_code: bool = False) -> list[str]:
+    """Get command entries as a list of formatted strings.
+
+    Args:
+        format_code: If True, wrap commands in backticks for markdown
+
+    Returns:
+        List of formatted command strings
+
+    """
+    entries = []
+    for cmd_type in CommandType:
+        if cmd_type in COMMAND_DOCS and cmd_type != CommandType.UNKNOWN:
+            syntax, description = COMMAND_DOCS[cmd_type]
+            if format_code:
+                entries.append(f"- `{syntax}` - {description}")
+            else:
+                entries.append(f"- {syntax} - {description}")
+    return entries
+
+
+def get_command_list() -> str:
+    """Get a formatted list of all available commands.
+
+    Returns:
+        Formatted string with all commands and their descriptions
+
+    """
+    lines = ["Available commands:", *_get_command_entries(format_code=False)]
+    return "\n".join(lines)
+
+
+@dataclass
+class Command:
+    """Parsed command with arguments."""
+
+    type: CommandType
+    args: dict[str, Any]
+    raw_text: str
+
+
+class CommandParser:
+    """Parser for user commands in messages."""
+
+    # Command patterns
+    HELP_PATTERN = re.compile(r"^!help(?:\s+(.+))?$", re.IGNORECASE)
+    SCHEDULE_PATTERN = re.compile(r"^!schedule\s+(.+)$", re.IGNORECASE | re.DOTALL)
+    LIST_SCHEDULES_PATTERN = re.compile(r"^!list[_-]?schedules?$", re.IGNORECASE)
+    CANCEL_SCHEDULE_PATTERN = re.compile(r"^!cancel[_-]?schedule\s+(.+)$", re.IGNORECASE)
+    WIDGET_PATTERN = re.compile(r"^!widget(?:\s+(.+))?$", re.IGNORECASE)
+    CONFIG_PATTERN = re.compile(r"^!config(?:\s+(.+))?$", re.IGNORECASE)
+    HI_PATTERN = re.compile(r"^!hi$", re.IGNORECASE)
+
+    def parse(self, message: str) -> Command | None:  # noqa: PLR0911
+        """Parse a message for commands.
+
+        Args:
+            message: The message text to parse
+
+        Returns:
+            Parsed command or None if no command found
+
+        """
+        message = message.strip()
+
+        # Handle voice emoji prefixe (e.g., "🎤 !schedule ...")
+        message = message.removeprefix(VOICE_PREFIX)
+        if not message.startswith("!"):
+            return None
+
+        # Try to match each command pattern
+
+        # !hi command (check this early as it's simple)
+        if self.HI_PATTERN.match(message):
+            return Command(
+                type=CommandType.HI,
+                args={},
+                raw_text=message,
+            )
+
+        # !help command
+        match = self.HELP_PATTERN.match(message)
+        if match:
+            topic = match.group(1)
+            return Command(
+                type=CommandType.HELP,
+                args={"topic": topic},
+                raw_text=message,
+            )
+
+        # !schedule command
+        match = self.SCHEDULE_PATTERN.match(message)
+        if match:
+            full_text = match.group(1).strip()
+            # Pass the entire text to AI - it will parse both time and message
+            return Command(
+                type=CommandType.SCHEDULE,
+                args={"full_text": full_text},
+                raw_text=message,
+            )
+
+        # !list_schedules command
+        if self.LIST_SCHEDULES_PATTERN.match(message):
+            return Command(
+                type=CommandType.LIST_SCHEDULES,
+                args={},
+                raw_text=message,
+            )
+
+        # !cancel_schedule command
+        match = self.CANCEL_SCHEDULE_PATTERN.match(message)
+        if match:
+            task_id = match.group(1).strip()
+            # Check if user wants to cancel all tasks
+            cancel_all = task_id.lower() == "all"
+            return Command(
+                type=CommandType.CANCEL_SCHEDULE,
+                args={"task_id": task_id, "cancel_all": cancel_all},
+                raw_text=message,
+            )
+
+        # !widget command
+        match = self.WIDGET_PATTERN.match(message)
+        if match:
+            url = match.group(1).strip() if match.group(1) else None
+            return Command(
+                type=CommandType.WIDGET,
+                args={"url": url},
+                raw_text=message,
+            )
+
+        # !config command
+        match = self.CONFIG_PATTERN.match(message)
+        if match:
+            args_text = match.group(1).strip() if match.group(1) else ""
+            return Command(
+                type=CommandType.CONFIG,
+                args={"args_text": args_text},
+                raw_text=message,
+            )
+
+        # Unknown command - return a special Command indicating it's unknown
+        logger.debug(f"Unknown command: {message}")
+        return Command(
+            type=CommandType.UNKNOWN,
+            args={"raw_command": message},
+            raw_text=message,
+        )
+
+
+def get_command_help(topic: str | None = None) -> str:
+    """Get help text for commands.
+
+    Args:
+        topic: Specific topic to get help for (optional)
+
+    Returns:
+        Help text
+
+    """
+    if topic == "schedule":
+        return """**Schedule Command**
+
+Usage: `!schedule <time> <message>` - Schedule tasks, reminders, or agent workflows
+
+**Simple Reminders:**
+- `!schedule in 5 minutes Check the deployment`
+- `!schedule tomorrow at 3pm Send the weekly report`
+- `!schedule later Ping me about the meeting`
+- `ping me tomorrow about the meeting`
+- `remind me in 2 hours to review PRs`
+
+**Event-Driven Workflows (New!):**
+- `!schedule If I get an email about "urgent", @phone_agent call me`
+- `!schedule When Bitcoin drops below $40k, @crypto_agent notify me`
+- `!schedule If server CPU > 80%, @ops_agent scale up`
+- `!schedule When someone mentions our product on Reddit, @analyst summarize it`
+- `!schedule Whenever I get email from boss, @notification_agent alert me immediately`
+
+**Agent Workflows:**
+- `!schedule Daily at 9am, @finance give me a market analysis`
+- `!schedule Every Monday, @research AI news and @email_assistant send me a summary`
+- `!schedule tomorrow at 2pm, @email_assistant check my Gmail`
+
+**Recurring Tasks (Cron-style):**
+- `!schedule Every hour, @shell check server status`
+- `!schedule Daily at 9am, @finance market report`
+- `!schedule Weekly on Friday, @analyst prepare weekly summary`
+
+How it works:
+- **Time-based**: Executes at specific times or intervals
+- **Event-based**: Automatically converts to smart polling (e.g., "if email" → check every 1-2 min)
+- Agents receive clear instructions about conditions to check
+- Multiple agents collaborate when mentioned together
+- Automated tasks are clearly marked so agents don't wait for follow-up"""
+
+    if topic == "list_schedules":
+        return """**List Schedules Command**
+
+Usage: `!list_schedules` or `!listschedules`
+
+Shows all pending scheduled tasks in this thread."""
+
+    if topic in {"cancel", "cancel_schedule"}:
+        return """**Cancel Schedule Command**
+
+Usage: `!cancel_schedule <id>` - Cancel a scheduled task
+       `!cancel_schedule all` - Cancel ALL scheduled tasks in this room
+
+Examples:
+- `!cancel_schedule abc123` - Cancel the task with ID abc123
+- `!cancel_schedule all` - Cancel all scheduled tasks (requires confirmation)
+
+Use `!list_schedules` to see task IDs."""
+
+    if topic == "config":
+        return """**Config Command**
+
+Usage: `!config <operation>` - View and modify MindRoom configuration
+
+**Viewing Configuration:**
+- `!config show` - Show entire configuration
+- `!config get <path>` - Get a specific configuration value
+- `!config get agents` - Show all agents
+- `!config get models.default` - Show default model
+- `!config get agents.analyst.display_name` - Show analyst's display name
+
+**Modifying Configuration:**
+- `!config set <path> <value>` - Set a configuration value
+- `!config set agents.analyst.display_name "Research Expert"` - Change display name
+- `!config set models.default.id gpt-4` - Change default model
+- `!config set defaults.markdown false` - Disable markdown by default
+- `!config set timezone America/New_York` - Set timezone
+
+**Path Syntax:**
+- Use dot notation to navigate nested config (e.g., `agents.analyst.role`)
+- Arrays use indexes (e.g., `agents.analyst.tools.0` for first tool)
+- String values with spaces must be quoted
+
+**Note:** Configuration changes are immediately saved to config.yaml and affect all new agent interactions."""
+
+    if topic == "widget":
+        return """**Widget Command**
+
+Usage: `!widget [url]` - Add the MindRoom configuration widget to this room
+
+Examples:
+- `!widget` - Add widget using default URL (http://localhost:3003)
+- `!widget https://config.mindroom.ai` - Add widget from custom URL
+
+The widget provides a visual interface for configuring MindRoom agents and settings.
+Pin it to keep it visible in the room.
+
+Note: Widget support requires Element Desktop or self-hosted Element Web."""
+
+    # General help - dynamically generated from COMMAND_DOCS
+    commands_text = "\n".join(_get_command_entries(format_code=True))
+
+    return f"""**Available Commands**
+
+{commands_text}
+
+**Scheduling Features:**
+- Time-based and event-driven workflows
+- Recurring tasks with cron-style scheduling (daily, weekly, hourly)
+- Agent workflows - mention agents to have them collaborate on scheduled tasks
+- Natural language time parsing - "tomorrow", "in 5 minutes", "every Monday"
+
+Note: All commands only work within threads, not in main room messages
+(except !widget which works in the main room).
+
+For detailed help on a command, use: `!help <command>`"""
+
+
+async def handle_widget_command(
+    client: nio.AsyncClient,
+    room_id: str,
+    url: str | None = None,
+) -> str:
+    """Handle the widget command to add configuration widget to room.
+
+    Args:
+        client: The Matrix client
+        room_id: The room ID to add widget to
+        url: Optional custom widget URL
+
+    Returns:
+        Response text for the user
+
+    """
+    # Default URL for local development
+    default_url = "http://localhost:3003/matrix-widget.html"
+    widget_url = url if url else default_url
+
+    # Create the widget state event content
+    widget_content = {
+        "type": "custom",
+        "url": widget_url,
+        "name": "MindRoom Configuration",
+        "data": {"title": "MindRoom Configuration", "curl": widget_url.replace("/matrix-widget.html", "")},
+        "creatorUserId": client.user_id,
+        "id": "mindroom_config",
+    }
+
+    try:
+        # Send the state event to add the widget
+        response = await client.room_put_state(
+            room_id=room_id,
+            event_type="im.vector.modular.widgets",
+            state_key="mindroom_config",
+            content=widget_content,
+        )
+
+        if isinstance(response, nio.RoomPutStateError):
+            logger.error(f"Failed to add widget to room {room_id}: {response.message}")
+            return f"❌ Failed to add widget: {response.message}"
+
+        logger.info(f"Successfully added widget to room {room_id}")
+    except Exception as e:
+        logger.exception("Error adding widget to room", room_id=room_id)
+        return f"❌ Error adding widget: {e!s}"
+    else:
+        return (
+            "✅ **MindRoom Configuration widget added!**\n\n"
+            "• Pin the widget to keep it visible\n"
+            "• All room members can access the configuration\n"
+            "• Changes sync in real-time with config.yaml\n\n"
+            f"Widget URL: {widget_url}\n\n"
+            "**Note:** Widgets require Element Desktop or self-hosted Element Web.\n"
+            "Alternatively, you can use: `/addwidget {url}` in Element."
+        )
+
+
+# Global parser instance
+command_parser = CommandParser()