synth-ai 0.2.14__py3-none-any.whl → 0.2.17__py3-none-any.whl

synth_ai/cli/status.py

@@ -1,134 +1,15 @@
 #!/usr/bin/env python3
-"""
-CLI: status of agent runs/versions and environment service.
-"""
+"""Compatibility wrapper for legacy status imports."""
 
-import asyncio
+from __future__ import annotations
 
 import click
-import requests
-from rich import box
-from rich.console import Console
-from rich.panel import Panel
-from rich.table import Table
+from synth_ai.cli.commands.status import register as _register_status
 
-from synth_ai.cli._storage import load_storage
 
+def register(cli: click.Group) -> None:
+    """Register status subcommands on the provided CLI group."""
+    _register_status(cli)
 
-async def _db_stats(db_url: str) -> dict:
-    create_storage, storage_config = load_storage()
-    db = create_storage(storage_config(connection_string=db_url))
-    await db.initialize()
-    try:
-        out: dict = {}
-        # Totals
-        totals = await db.query_traces(
-            """
-            SELECT 
-              (SELECT COUNT(*) FROM session_traces) AS sessions,
-              (SELECT COUNT(*) FROM experiments) AS experiments,
-              (SELECT COUNT(*) FROM events) AS events,
-              (SELECT COUNT(*) FROM messages) AS messages,
-              (SELECT COALESCE(SUM(CASE WHEN event_type='cais' THEN cost_usd ELSE 0 END),0)/100.0 FROM events) AS total_cost_usd,
-              (SELECT COALESCE(SUM(CASE WHEN event_type='cais' THEN total_tokens ELSE 0 END),0) FROM events) AS total_tokens
-        """
-        )
-        if not totals.empty:
-            out["totals"] = totals.iloc[0].to_dict()
-        else:
-            out["totals"] = {}
 
-        # Systems summary
-        systems = await db.query_traces(
-            """
-            SELECT system_type, COUNT(*) as count FROM systems GROUP BY system_type
-        """
-        )
-        out["systems"] = systems
-
-        versions = await db.query_traces(
-            """
-            SELECT COUNT(*) as version_count FROM system_versions
-        """
-        )
-        if not versions.empty:
-            out["version_count"] = int(versions.iloc[0]["version_count"])
-        else:
-            out["version_count"] = 0
-        return out
-    finally:
-        await db.close()
-
-
-def register(cli):
-    @cli.command()
-    @click.option(
-        "--url",
-        "db_url",
-        default="sqlite+aiosqlite:///./synth_ai.db/dbs/default/data",
-        help="Database URL",
-    )
-    @click.option("--service-url", default="http://127.0.0.1:8901", help="Environment service URL")
-    def status(db_url: str, service_url: str):
-        """Show DB stats, agent/environment system counts, and env service health."""
-        console = Console()
-
-        async def _run():
-            # DB
-            stats = await _db_stats(db_url)
-
-            # Env service
-            health_text = "[red]unreachable[/red]"
-            envs_list = []
-            try:
-                r = requests.get(f"{service_url}/health", timeout=2)
-                if r.ok:
-                    data = r.json()
-                    health_text = "[green]ok[/green]"
-                    envs_list = data.get("supported_environments", [])
-                else:
-                    health_text = f"[red]{r.status_code}[/red]"
-            except Exception:
-                pass
-
-            # Render
-            totals = stats.get("totals", {})
-            lines = []
-            lines.append(f"DB: [dim]{db_url}[/dim]")
-            lines.append(
-                f"Experiments: {int(totals.get('experiments', 0)):,}  "
-                f"Sessions: {int(totals.get('sessions', 0)):,}  "
-                f"Events: {int(totals.get('events', 0)):,}  "
-                f"Messages: {int(totals.get('messages', 0)):,}"
-            )
-            lines.append(
-                f"Cost: ${float(totals.get('total_cost_usd', 0.0) or 0.0):.4f}  "
-                f"Tokens: {int(totals.get('total_tokens', 0)):,}"
-            )
-            lines.append("")
-            lines.append(f"Env Service: {health_text}  [dim]{service_url}[/dim]")
-            if envs_list:
-                lines.append(
-                    "Environments: "
-                    + ", ".join(sorted(envs_list)[:10])
-                    + (" ..." if len(envs_list) > 10 else "")
-                )
-
-            panel_main = Panel("\n".join(lines), title="Synth AI Status", border_style="cyan")
-            console.print(panel_main)
-
-            # Systems table
-            sys_df = stats.get("systems")
-            if sys_df is not None and not sys_df.empty:
-                tbl = Table(
-                    title=f"Systems (versions: {stats.get('version_count', 0)})",
-                    box=box.SIMPLE,
-                    header_style="bold",
-                )
-                tbl.add_column("Type")
-                tbl.add_column("Count", justify="right")
-                for _, r in sys_df.iterrows():
-                    tbl.add_row(str(r.get("system_type", "-")), f"{int(r.get('count', 0)):,}")
-                console.print(tbl)
-
-        asyncio.run(_run())
+__all__ = ["register"]

synth_ai/cli/task_app_deploy.py

@@ -0,0 +1,7 @@
+"""Compatibility wrapper for task-app deploy command."""
+
+from __future__ import annotations
+
+from synth_ai.cli.commands.deploy import command as deploy_command
+
+__all__ = ["deploy_command"]

synth_ai/cli/task_app_list.py

@@ -0,0 +1,25 @@
+"""Task app list command."""
+
+from __future__ import annotations
+
+import click
+from synth_ai.task.apps import registry
+
+from .task_apps import task_app_group
+
+
+@task_app_group.command("list")
+def list_apps() -> None:
+    """List registered task apps."""
+
+    entries = registry.list()
+    if not entries:
+        click.echo("No task apps registered.")
+        return
+
+    for entry in entries:
+        aliases = f" (aliases: {', '.join(entry.aliases)})" if entry.aliases else ""
+        click.echo(f"- {entry.app_id}{aliases}: {entry.description}")
+
+
+__all__ = ["list_apps"]

synth_ai/cli/task_app_modal_serve.py

@@ -0,0 +1,11 @@
+"""Compatibility wrapper for task-app modal-serve command."""
+
+from __future__ import annotations
+
+from synth_ai.cli.task_apps import task_app_group
+
+modal_serve_command = task_app_group.commands.get("modal-serve")
+if modal_serve_command is None:  # pragma: no cover - defensive guard
+    raise RuntimeError("Modal-serve command is not registered on task_app_group")
+
+__all__ = ["modal_serve_command"]

synth_ai/cli/task_app_serve.py

@@ -0,0 +1,11 @@
+"""Compatibility wrapper for task-app serve command."""
+
+from __future__ import annotations
+
+from synth_ai.cli.task_apps import serve_command, task_app_group
+
+serve_task_group = task_app_group.commands.get("serve")
+if serve_task_group is None:  # pragma: no cover - defensive guard
+    raise RuntimeError("Serve command is not registered on task_app_group")
+
+__all__ = ["serve_command", "serve_task_group"]