sentinel-ai-os 1.0__py3-none-any.whl

sentinel/core/ui.py

@@ -0,0 +1,112 @@
+# FILE: core/ui.py
+from rich.console import Console
+from rich.panel import Panel
+from rich.markdown import Markdown
+from rich import box
+from rich.table import Table
+
+
+class UI:
+    console = Console()
+
+    @staticmethod
+    def print_banner():
+        logo = """
+   ███████╗███████╗███╗   ██╗████████╗██╗███╗   ██╗███████╗██╗     
+   ██╔════╝██╔════╝████╗  ██║╚══██╔══╝██║████╗  ██║██╔════╝██║     
+   ███████╗█████╗  ██╔██╗ ██║   ██║   ██║██╔██╗ ██║█████╗  ██║     
+   ╚════██║██╔══╝  ██║╚██╗██║   ██║   ██║██║╚██╗██║██╔══╝  ██║     
+   ███████║███████╗██║ ╚████║   ██║   ██║██║ ╚████║███████╗███████╗
+   ╚══════╝╚══════╝╚═╝  ╚═══╝   ╚═╝   ╚═╝╚═╝  ╚═══╝╚══════╝╚══════╝
+        """
+        UI.console.print(f"[bold cyan]{logo}[/bold cyan]")
+        UI.console.print("[dim]OPERATING SYSTEM v1.0[/dim]\n")
+
+    @staticmethod
+    def print_system(msg):
+        UI.console.print(f"[dim]System: {msg}[/dim]")
+
+    @staticmethod
+    def print_success(msg):
+        UI.console.print(f"[bold green]✔ {msg}[/bold green]")
+
+    @staticmethod
+    def print_warning(msg):
+        UI.console.print(f"[bold yellow]⚠ {msg}[/bold yellow]")
+
+    @staticmethod
+    def print_error(msg):
+        UI.console.print(Panel(
+            f"[bold white]{msg}[/bold white]",
+            title="System Alert",
+            border_style="red",
+            box=box.ROUNDED,
+            expand=False
+        ))
+
+    @staticmethod
+    def print_agent(text, model=None):
+        """The SENTINEL Response Panel."""
+        if not isinstance(text, Markdown):
+            text = Markdown(str(text))
+
+        # Add Model Info to Title
+        title = "[bold cyan]SENTINEL[/bold cyan]"
+        if model:
+            title += f" [dim]({model})[/dim]"
+
+        UI.console.print(Panel(
+            text,
+            title=title,
+            title_align="left",
+            border_style="cyan",
+            box=box.ROUNDED,
+            padding=(1, 2),
+            expand=True
+        ))
+
+    @staticmethod
+    def print_tool(tool_name):
+        UI.console.print(f"[dim]   ⚙ Executing: {tool_name}...[/dim]")
+
+    @staticmethod
+    def print_result(result):
+        text = str(result)
+        if len(text) > 400:
+            text = text[:400] + "... (truncated)"
+        UI.console.print(Panel(
+            text,
+            title="Action Result",
+            border_style="dim white",
+            box=box.ROUNDED,
+            expand=False
+        ))
+
+    @staticmethod
+    def print_help():
+        table = Table(title="Sentinel Control Panel", border_style="cyan", box=box.ROUNDED)
+        table.add_column("Command", style="bold cyan")
+        table.add_column("Description", style="white")
+
+        table.add_row("/help", "Show this menu")
+        table.add_row("/exit", "Shutdown system")
+        table.add_row("/status", "View current model, provider, and memory stats")
+        table.add_row("/memory [n]", "Set Context Window size (e.g., /memory 5)")
+        table.add_row("/log [on/off]", "Toggle audit logging (Default: OFF)")
+
+        table.add_row("/clear", "Clear active chat memory (RAM only)")
+        table.add_row("/wipe", "Wipe long-term memory (Vector DB + brain.db)")
+        table.add_row("/factory_reset", "[bold red]FULL FACTORY RESET[/bold red] (Deletes EVERYTHING)")
+
+        table.add_row("/switch [p] [m]", "Switch Brain (e.g., /switch groq llama3)")
+        table.add_row("/setkey [p] [k]", "Update API Key")
+
+        UI.console.print(table)
+
+    @staticmethod
+    def get_input():
+        try:
+            UI.console.print()
+            return UI.console.input("[bold cyan]>>> [/bold cyan]")
+        except KeyboardInterrupt:
+            return None

sentinel/main.py

@@ -0,0 +1,110 @@
+# FILE: main.py
+import typer
+from sentinel.core.config import ConfigManager
+from sentinel.core.agent import SentinelAgent
+from sentinel.core.ui import UI
+from sentinel.core.registry import initialize_tools
+from sentinel.core.setup import setup_wizard
+
+app = typer.Typer(
+    name="Sentinel",
+    help="Autonomous AI Agent for your OS",
+    add_completion=False,
+    no_args_is_help=False
+)
+
+
+def boot_sequence(briefing: bool = False):
+    """
+    Shared startup logic for both default run and 'start' command.
+    """
+    UI.console.clear()
+
+    # 1. Check Configuration
+    cfg = ConfigManager()
+    if not cfg.exists():
+        setup_wizard()
+        cfg = ConfigManager()  # Reload after wizard
+
+    # 2. Boot Up
+    UI.print_banner()
+
+    # Check for keys
+    # We check if ANY key is present or if using Ollama
+    has_key = (
+            cfg.get_key("openai") or
+            cfg.get_key("anthropic") or
+            cfg.get_key("groq") or
+            cfg.get("llm.provider") == "ollama"
+    )
+
+    if not has_key:
+        UI.print_warning("No LLM API Key found. System running in Limited Mode.")
+    else:
+        provider = cfg.get("llm.provider", "unknown")
+        UI.print_system(f"Brain Active: [green]{provider.upper()}[/green]")
+
+    # 3. Initialize Background Services
+    initialize_tools()
+
+    # 4. Run Briefing (If requested)
+    if briefing:
+        if has_key:
+            UI.print_system("Generating Daily Briefing...")
+            try:
+                from sentinel.core.cognitive import get_daily_briefing
+                report = get_daily_briefing(cfg)
+                UI.print_agent(report)
+            except Exception as e:
+                UI.print_error(f"Briefing failed: {e}")
+        else:
+            UI.print_error("Skipping Daily Briefing (Offline Mode).")
+
+    # 5. Start Agent Loop
+    try:
+        agent = SentinelAgent(cfg)
+        agent.run_loop()
+    except KeyboardInterrupt:
+        UI.print_system("Shutting down...")
+    except Exception as e:
+        UI.print_error(f"Critical System Failure: {e}")
+
+
+@app.callback(invoke_without_command=True)
+def main(
+        ctx: typer.Context,
+        briefing: bool = typer.Option(False, "--briefing", "-b", help="Run Daily Briefing on startup")
+):
+    """
+    Main Entry Point. Checks state and routes to Setup or Runtime.
+    """
+    # If a subcommand (like 'config' or 'auth') is called, skip this default boot
+    if ctx.invoked_subcommand is not None:
+        return
+
+    boot_sequence(briefing=briefing)
+
+
+@app.command()
+def start(
+        briefing: bool = typer.Option(False, "--briefing", "-b", help="Run Daily Briefing on startup")
+):
+    """Explicit start command."""
+    boot_sequence(briefing=briefing)
+
+
+@app.command()
+def config():
+    """Re-run the configuration wizard."""
+    setup_wizard()
+
+
+@app.command()
+def auth():
+    """Run the Google Authentication fix tool."""
+    from sentinel.auth import fix_authentication
+    fix_authentication()
+
+
+if __name__ == "__main__":
+    app()

sentinel/paths.py

@@ -0,0 +1,77 @@
+# FILE: sentinel/core/paths.py
+import os
+import sys
+from pathlib import Path
+
+# ==========================================
+# 1. USER DATA STORAGE (Mutable)
+# Stores DBs, Configs, Logs, and Auth Tokens
+# Location: ~/.sentinel (Cross-platform)
+# ==========================================
+
+USER_DATA_DIR = Path.home() / ".sentinel"
+USER_DATA_DIR.mkdir(parents=True, exist_ok=True)
+
+# Subdirectories
+LOGS_DIR = USER_DATA_DIR / "logs"
+LOGS_DIR.mkdir(exist_ok=True)
+
+DRAFTS_DIR = USER_DATA_DIR / "drafts"
+DRAFTS_DIR.mkdir(exist_ok=True)
+
+# Critical System Files
+CONFIG_PATH = USER_DATA_DIR / "config.json"
+DB_PATH = USER_DATA_DIR / "brain.db"
+VECTOR_PATH = USER_DATA_DIR / "brain_vectors"  # ChromaDB folder
+AUDIT_LOG_PATH = USER_DATA_DIR / "audit_log.jsonl"
+MEMORY_FILE = USER_DATA_DIR / "memory.json"
+
+# Search Indexes
+FILE_INDEX_DB = USER_DATA_DIR / "file_index.db"
+SMART_INDEX_DB = USER_DATA_DIR / "smart_files.db"
+
+# Authentication
+CREDENTIALS_PATH = USER_DATA_DIR / "credentials.json"
+TOKEN_PATH = USER_DATA_DIR / "token.json"
+
+
+# ==========================================
+# 2. PACKAGE ASSETS (Immutable)
+# Locates scripts/tools bundled inside the pip package
+# Location: .../site-packages/sentinel/scripts/
+# ==========================================
+
+def get_script_path(filename: str) -> str:
+    """
+    Returns the absolute path to a script bundled inside the pip package.
+    Expects scripts to be in: src/sentinel/scripts/
+    """
+    # This file is in: .../sentinel/core/paths.py
+    # We want:         .../sentinel/scripts/filename
+
+    # Get the directory of THIS file (core/)
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+
+    # Go up one level to the package root (sentinel/)
+    pkg_root = os.path.dirname(current_dir)
+
+    # Target path
+    script_path = os.path.join(pkg_root, "scripts", filename)
+
+    if not os.path.exists(script_path):
+        # Fallback for Development Mode (running from source without pip install)
+        # In dev repo: sentinel/core/paths.py -> ../../../scripts/
+        repo_root = os.path.dirname(os.path.dirname(pkg_root))
+        dev_path = os.path.join(repo_root, "scripts", filename)
+
+        if os.path.exists(dev_path):
+            return dev_path
+
+        # If we still can't find it, that's a build error
+        raise FileNotFoundError(
+            f"Could not find bundled script '{filename}'.\n"
+            f"Checked: {script_path}\n"
+            f"Checked: {dev_path}"
+        )
+
+    return script_path