pilot-daemon 0.9.0__tar.gz

pilot_daemon-0.9.0/PKG-INFO

@@ -0,0 +1,128 @@
+Metadata-Version: 2.4
+Name: pilot-daemon
+Version: 0.9.0
+Summary: AI-powered autonomous computer agent — full system control
+License-Expression: MIT
+Requires-Python: >=3.11
+Requires-Dist: aiofiles>=24.1
+Requires-Dist: aiosqlite>=0.20
+Requires-Dist: caldav>=1.0.0
+Requires-Dist: cryptography>=43.0
+Requires-Dist: httpx>=0.27
+Requires-Dist: icalendar>=5.0.0
+Requires-Dist: keyring>=25.0
+Requires-Dist: numpy==2.2.6
+Requires-Dist: psutil>=5.9
+Requires-Dist: pydantic>=2.6
+Requires-Dist: scikit-learn>=1.3.0
+Requires-Dist: tiktoken>=0.7
+Requires-Dist: tomli-w>=1.0
+Requires-Dist: tomli>=2.0; python_version < '3.12'
+Requires-Dist: websockets>=13.0
+Provides-Extra: all
+Requires-Dist: beautifulsoup4>=4.12; extra == 'all'
+Requires-Dist: chromadb>=0.5; extra == 'all'
+Requires-Dist: easyocr>=1.7; extra == 'all'
+Requires-Dist: faiss-cpu>=1.7.0; extra == 'all'
+Requires-Dist: mediapipe>=0.10.11; extra == 'all'
+Requires-Dist: openai-whisper>=20231117; extra == 'all'
+Requires-Dist: opencv-python>=4.9.0.80; extra == 'all'
+Requires-Dist: openpyxl>=3.1; extra == 'all'
+Requires-Dist: paramiko>=3.4; extra == 'all'
+Requires-Dist: pillow>=10.0; extra == 'all'
+Requires-Dist: playwright>=1.40; extra == 'all'
+Requires-Dist: pocket-tts>=1.0; extra == 'all'
+Requires-Dist: pyautogui>=0.9; extra == 'all'
+Requires-Dist: pynput>=1.7; extra == 'all'
+Requires-Dist: pypdf>=4.0; extra == 'all'
+Requires-Dist: pyperclip>=1.8; extra == 'all'
+Requires-Dist: python-docx>=1.0; extra == 'all'
+Requires-Dist: secretstorage>=3.3; (sys_platform == 'linux') and extra == 'all'
+Requires-Dist: sentence-transformers>=2.2.0; extra == 'all'
+Requires-Dist: sounddevice>=0.4; extra == 'all'
+Requires-Dist: wasmtime>=18.0; extra == 'all'
+Requires-Dist: zeroconf>=0.131; extra == 'all'
+Provides-Extra: automation
+Requires-Dist: pillow>=10.0; extra == 'automation'
+Requires-Dist: pyautogui>=0.9; extra == 'automation'
+Requires-Dist: pyperclip>=1.8; extra == 'automation'
+Provides-Extra: browser
+Requires-Dist: playwright>=1.40; extra == 'browser'
+Provides-Extra: dev
+Requires-Dist: pytest-asyncio>=0.24; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Requires-Dist: ruff>=0.5; extra == 'dev'
+Provides-Extra: documents
+Requires-Dist: beautifulsoup4>=4.12; extra == 'documents'
+Requires-Dist: openpyxl>=3.1; extra == 'documents'
+Requires-Dist: pypdf>=4.0; extra == 'documents'
+Requires-Dist: python-docx>=1.0; extra == 'documents'
+Provides-Extra: full
+Requires-Dist: chromadb>=0.5; extra == 'full'
+Requires-Dist: faiss-cpu>=1.7.0; extra == 'full'
+Requires-Dist: pyperclip>=1.8; extra == 'full'
+Requires-Dist: secretstorage>=3.3; (sys_platform == 'linux') and extra == 'full'
+Requires-Dist: sentence-transformers>=2.2.0; extra == 'full'
+Requires-Dist: zeroconf>=0.131; extra == 'full'
+Provides-Extra: llamacpp
+Requires-Dist: llama-cpp-python>=0.3; extra == 'llamacpp'
+Requires-Dist: pynvml>=11.5; extra == 'llamacpp'
+Provides-Extra: network
+Requires-Dist: zeroconf>=0.131; extra == 'network'
+Provides-Extra: pty
+Requires-Dist: ptyprocess>=0.7; (sys_platform != 'win32') and extra == 'pty'
+Provides-Extra: ssh
+Requires-Dist: paramiko>=3.4; extra == 'ssh'
+Provides-Extra: supervision
+Requires-Dist: pynput>=1.7; extra == 'supervision'
+Provides-Extra: vision
+Requires-Dist: easyocr>=1.7; extra == 'vision'
+Requires-Dist: mediapipe>=0.10.11; extra == 'vision'
+Requires-Dist: opencv-python>=4.9.0.80; extra == 'vision'
+Requires-Dist: pillow>=10.0; extra == 'vision'
+Provides-Extra: voice
+Requires-Dist: openai-whisper>=20231117; extra == 'voice'
+Requires-Dist: pocket-tts>=1.0; extra == 'voice'
+Requires-Dist: sounddevice>=0.4; extra == 'voice'
+Provides-Extra: wasm
+Requires-Dist: wasmtime>=18.0; extra == 'wasm'
+Description-Content-Type: text/markdown
+
+# Heliox OS Daemon
+
+Python backend for the Heliox OS AI System Control Agent. Provides the agent system (Planner, Executor, Verifier, Code Sanitizer), security layer, and system interface modules.
+
+See the [main README](../README.md) for full project documentation.
+
+## SSH Agent (Remote Host Execution)
+
+Heliox OS includes an optional `SshAgent` that can execute `ssh_command` / `ssh_script` actions on **pre-configured** remote hosts using Paramiko.
+
+### Install
+
+Install the optional SSH dependency set:
+
+```bash
+pip install "pilot-daemon[ssh]"
+```
+
+### Configure allowed hosts
+
+In your `config.toml`, enable SSH and define allowed destinations (aliases only):
+
+```toml
+[ssh]
+enabled = true
+connect_timeout_seconds = 10
+allowed_hosts = [
+  { name = "prod-1", hostname = "10.0.0.10", port = 22, username = "ubuntu", private_key_provider = "ssh_prod_1_key", strict_host_key_checking = true },
+]
+```
+
+### Store SSH keys in the KeyVault
+
+Store the private key PEM in the encrypted vault using the JSON-RPC method `store_api_key` with `provider=<private_key_provider>`.
+
+Notes:
+- The vault stores arbitrary provider→secret strings; SSH keys are treated the same as API keys.
+- Keys/passphrases are never logged and are only decrypted at call time.

pilot_daemon-0.9.0/README.md

@@ -0,0 +1,38 @@
+# Heliox OS Daemon
+
+Python backend for the Heliox OS AI System Control Agent. Provides the agent system (Planner, Executor, Verifier, Code Sanitizer), security layer, and system interface modules.
+
+See the [main README](../README.md) for full project documentation.
+
+## SSH Agent (Remote Host Execution)
+
+Heliox OS includes an optional `SshAgent` that can execute `ssh_command` / `ssh_script` actions on **pre-configured** remote hosts using Paramiko.
+
+### Install
+
+Install the optional SSH dependency set:
+
+```bash
+pip install "pilot-daemon[ssh]"
+```
+
+### Configure allowed hosts
+
+In your `config.toml`, enable SSH and define allowed destinations (aliases only):
+
+```toml
+[ssh]
+enabled = true
+connect_timeout_seconds = 10
+allowed_hosts = [
+  { name = "prod-1", hostname = "10.0.0.10", port = 22, username = "ubuntu", private_key_provider = "ssh_prod_1_key", strict_host_key_checking = true },
+]
+```
+
+### Store SSH keys in the KeyVault
+
+Store the private key PEM in the encrypted vault using the JSON-RPC method `store_api_key` with `provider=<private_key_provider>`.
+
+Notes:
+- The vault stores arbitrary provider→secret strings; SSH keys are treated the same as API keys.
+- Keys/passphrases are never logged and are only decrypted at call time.

pilot_daemon-0.9.0/benchmarks/react_latency.py

@@ -0,0 +1,191 @@
+"""Benchmark the non-LLM ReAct latency for a simple CPU usage request.
+
+Run from ``daemon``:
+    python benchmarks/react_latency.py --iterations 10 --profile
+"""
+
+from __future__ import annotations
+
+import argparse
+import asyncio
+import cProfile
+import io
+import os
+import pstats
+import statistics
+import time
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Any
+
+BENCH_STATE_DIR = Path(__file__).resolve().parent / ".bench-state"
+BENCH_STATE_DIR.mkdir(parents=True, exist_ok=True)
+os.environ.setdefault("XDG_CONFIG_HOME", str(BENCH_STATE_DIR / "config"))
+os.environ.setdefault("XDG_DATA_HOME", str(BENCH_STATE_DIR / "data"))
+os.environ.setdefault("XDG_STATE_HOME", str(BENCH_STATE_DIR / "state"))
+os.environ.setdefault("XDG_RUNTIME_DIR", str(BENCH_STATE_DIR / "runtime"))
+
+
+CPU_PLAN_JSON = """{
+  "explanation": "Check current CPU usage.",
+  "actions": [
+    {
+      "action_type": "cpu_usage",
+      "target": "cpu",
+      "parameters": {},
+      "requires_root": false,
+      "destructive": false,
+      "reversible": true,
+      "rollback_action": null,
+      "use_previous_output": false
+    }
+  ]
+}"""
+
+
+class StubModelRouter:
+    def __init__(self) -> None:
+        self.generate_calls = 0
+
+    async def generate(self, *args: Any, **kwargs: Any) -> str:  # noqa: ARG002
+        self.generate_calls += 1
+        return CPU_PLAN_JSON
+
+
+class FakeWebSocket:
+    def __init__(self) -> None:
+        self.messages: list[str] = []
+
+    async def send(self, message: str) -> None:
+        self.messages.append(message)
+
+
+class NoopReasoning:
+    def reset(self) -> None:
+        return None
+
+
+class NoopReflector:
+    async def get_improvement_context(self, query: str) -> str:  # noqa: ARG002
+        return ""
+
+    async def reflect(self, *args: Any, **kwargs: Any) -> dict[str, Any]:  # noqa: ARG002
+        return {}
+
+
+class NoopMemory:
+    async def record(self, *args: Any, **kwargs: Any) -> None:  # noqa: ARG002
+        return None
+
+
+@dataclass
+class BenchmarkHarness:
+    server: Any
+    model: StubModelRouter
+
+
+async def build_harness() -> BenchmarkHarness:
+    from pilot.agents.executor import Executor
+    from pilot.agents.multi_agent import MultiAgentRouter
+    from pilot.agents.orchestrator import AgentOrchestrator
+    from pilot.agents.planner import Planner
+    from pilot.agents.system_agent import SystemAgent
+    from pilot.agents.verifier import Verifier
+    from pilot.config import PilotConfig
+    from pilot.memory.store import MemoryStore
+    from pilot.security.audit import AuditLogger
+    from pilot.security.permissions import PermissionChecker
+    from pilot.security.validator import ActionValidator
+    from pilot.server import PilotServer
+
+    config = PilotConfig()
+    model = StubModelRouter()
+    memory = MemoryStore()
+    await memory.initialize()
+
+    validator = ActionValidator(config)
+    permissions = PermissionChecker(config)
+    audit = AuditLogger(BENCH_STATE_DIR / "audit.jsonl")
+    executor = Executor(config, validator, permissions, audit)
+    verifier = Verifier(model)  # type: ignore[arg-type]
+
+    orchestrator = AgentOrchestrator(model)  # type: ignore[arg-type]
+    orchestrator.register_agent(SystemAgent(model, executor))  # type: ignore[arg-type]
+
+    server = PilotServer(config)
+    server._planner = Planner(model, memory)  # noqa: SLF001
+    server._executor = executor  # noqa: SLF001
+    server._verifier = verifier  # noqa: SLF001
+    server._reflector = NoopReflector()  # noqa: SLF001
+    server._multi_agent = MultiAgentRouter(model)  # type: ignore[arg-type]  # noqa: SLF001
+    server._orchestrator = orchestrator  # noqa: SLF001
+    server._reasoning = None  # noqa: SLF001
+    server._screen_vision = None  # noqa: SLF001
+    server._memory = NoopMemory()  # noqa: SLF001
+    return BenchmarkHarness(server=server, model=model)
+
+
+async def run_once(harness: BenchmarkHarness) -> float:
+    ws = FakeWebSocket()
+    start = time.perf_counter()
+    result = await harness.server._handle_execute(  # noqa: SLF001
+        {"input": "What's my CPU usage?", "dry_run": False},
+        ws,  # type: ignore[arg-type]
+    )
+    elapsed_ms = (time.perf_counter() - start) * 1000
+    if result.get("status") != "success":
+        raise RuntimeError(f"Unexpected benchmark result: {result}")
+    return elapsed_ms
+
+
+async def benchmark(iterations: int) -> tuple[list[float], int]:
+    harness = await build_harness()
+    timings = []
+    for _ in range(iterations):
+        timings.append(await run_once(harness))
+    return timings, harness.model.generate_calls
+
+
+async def benchmark_with_harness(harness: BenchmarkHarness, iterations: int) -> tuple[list[float], int]:
+    timings = []
+    generate_calls_before = harness.model.generate_calls
+    for _ in range(iterations):
+        timings.append(await run_once(harness))
+    return timings, harness.model.generate_calls - generate_calls_before
+
+
+def print_summary(timings: list[float], generate_calls: int) -> None:
+    print(f"iterations: {len(timings)}")
+    print(f"model_generate_calls: {generate_calls}")
+    print(f"mean_ms: {statistics.mean(timings):.2f}")
+    print(f"median_ms: {statistics.median(timings):.2f}")
+    print(f"min_ms: {min(timings):.2f}")
+    print(f"max_ms: {max(timings):.2f}")
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--iterations", type=int, default=10)
+    parser.add_argument("--profile", action="store_true")
+    parser.add_argument("--profile-limit", type=int, default=20)
+    args = parser.parse_args()
+
+    if args.profile:
+        harness = asyncio.run(build_harness())
+        profiler = cProfile.Profile()
+        profiler.enable()
+        timings, generate_calls = asyncio.run(benchmark_with_harness(harness, args.iterations))
+        profiler.disable()
+        print_summary(timings, generate_calls)
+        output = io.StringIO()
+        stats = pstats.Stats(profiler, stream=output).strip_dirs().sort_stats("cumtime")
+        stats.print_stats(args.profile_limit)
+        print("\n## cProfile cumulative time")
+        print(output.getvalue())
+    else:
+        timings, generate_calls = asyncio.run(benchmark(args.iterations))
+        print_summary(timings, generate_calls)
+
+
+if __name__ == "__main__":
+    main()

pilot_daemon-0.9.0/pilot/__init__.py

@@ -0,0 +1,3 @@
+"""Pilot — AI Command Center daemon for Ubuntu LTS."""
+
+__version__ = "0.1.0"