cited-cli 0.3.2__py3-none-any.whl

cited_cli/commands/agent.py

@@ -0,0 +1,129 @@
+from __future__ import annotations
+
+from typing import Annotated
+
+import typer
+
+from cited_cli.output.formatter import OutputContext, print_error, print_result
+from cited_cli.utils.errors import CitedAPIError, ExitCode, handle_api_error
+from cited_cli.utils.interactive import prompt_if_missing
+from cited_core.api import endpoints
+from cited_core.api.client import CitedClient
+from cited_core.config.manager import ConfigManager
+
+agent_app = typer.Typer(
+    name="agent",
+    help="Agent API (v1) — structured business data for AI agents.",
+)
+
+
+def _get_agent_client(ctx: typer.Context) -> tuple[OutputContext, CitedClient]:
+    obj = ctx.obj or {}
+    out: OutputContext = obj.get("output", OutputContext())
+    cfg: ConfigManager = obj.get("config", ConfigManager())
+    profile = obj.get("profile", "default")
+    api_url = cfg.get_api_url(profile, obj.get("env_override"))
+
+    import os
+
+    api_key = cfg.get("agent_api_key", profile) or os.environ.get("CITED_AGENT_API_KEY")
+    if not api_key:
+        print_error(
+            "Agent API key required. Set via: cited config set agent_api_key <key>\n"
+            "Or set CITED_AGENT_API_KEY environment variable.",
+            out,
+        )
+        raise typer.Exit(ExitCode.AUTH_ERROR)
+
+    return out, CitedClient(base_url=api_url, agent_api_key=api_key)
+
+
+@agent_app.command("facts")
+def agent_facts(
+    ctx: typer.Context,
+    business_id: Annotated[str, typer.Argument(help="Business ID")],
+) -> None:
+    """Get structured business facts."""
+    out, client = _get_agent_client(ctx)
+    try:
+        path = endpoints.AGENT_FACTS.format(business_id=business_id)
+        data = client.get(path)
+        print_result(data, out)
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()
+
+
+@agent_app.command("claims")
+def agent_claims(
+    ctx: typer.Context,
+    business_id: Annotated[str, typer.Argument(help="Business ID")],
+) -> None:
+    """Get verifiable claims about a business."""
+    out, client = _get_agent_client(ctx)
+    try:
+        path = endpoints.AGENT_CLAIMS.format(business_id=business_id)
+        data = client.get(path)
+        print_result(data, out)
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()
+
+
+@agent_app.command("comparison")
+def agent_comparison(
+    ctx: typer.Context,
+    business_id: Annotated[str, typer.Argument(help="Business ID")],
+) -> None:
+    """Get competitive comparison data."""
+    out, client = _get_agent_client(ctx)
+    try:
+        path = endpoints.AGENT_COMPARISON.format(business_id=business_id)
+        data = client.get(path)
+        print_result(data, out)
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()
+
+
+@agent_app.command("semantic-health")
+def agent_semantic_health(
+    ctx: typer.Context,
+    business_id: Annotated[str, typer.Argument(help="Business ID")],
+) -> None:
+    """Get semantic readiness signals."""
+    out, client = _get_agent_client(ctx)
+    try:
+        path = endpoints.AGENT_SEMANTIC_HEALTH.format(business_id=business_id)
+        data = client.get(path)
+        print_result(data, out)
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()
+
+
+@agent_app.command("buyer-fit")
+def agent_buyer_fit(
+    ctx: typer.Context,
+    query: Annotated[
+        str | None, typer.Option("--query", "-q", help="Buyer query to simulate")
+    ] = None,
+    business_id: Annotated[str | None, typer.Option("--business", "-b", help="Business ID")] = None,
+) -> None:
+    """Run a buyer-fit simulation query."""
+    out, client = _get_agent_client(ctx)
+    query = prompt_if_missing(query, "--query", "Buyer query", out)
+    try:
+        payload: dict[str, str] = {"query": query}
+        if business_id:
+            payload["business_id"] = business_id
+        data = client.post(endpoints.AGENT_BUYER_FIT, json=payload)
+        print_result(data, out)
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()

cited_cli/commands/analytics.py

@@ -0,0 +1,83 @@
+from __future__ import annotations
+
+from typing import Annotated
+
+import typer
+
+from cited_cli.output.formatter import OutputContext, print_error, print_result
+from cited_cli.output.tables import render_kv
+from cited_cli.utils.errors import CitedAPIError, ExitCode, handle_api_error
+from cited_core.api import endpoints
+from cited_core.api.client import CitedClient
+from cited_core.auth.store import TokenStore
+from cited_core.config.manager import ConfigManager
+
+analytics_app = typer.Typer(name="analytics", help="Analytics and trend data.")
+
+
+def _get_client(ctx: typer.Context) -> tuple[OutputContext, CitedClient]:
+    obj = ctx.obj or {}
+    out: OutputContext = obj.get("output", OutputContext())
+    cfg: ConfigManager = obj.get("config", ConfigManager())
+    profile = obj.get("profile", "default")
+    env = cfg.get_environment(profile, obj.get("env_override"))
+    api_url = cfg.get_api_url(profile, obj.get("env_override"))
+
+    store = TokenStore()
+    token = store.get_token(env)
+    if not token:
+        print_error(f"Not logged in to {env}. Run: cited login", out)
+        raise typer.Exit(ExitCode.AUTH_ERROR)
+
+    return out, CitedClient(base_url=api_url, token=token)
+
+
+@analytics_app.command("compare")
+def analytics_compare(
+    ctx: typer.Context,
+    audit_id: Annotated[str, typer.Argument(help="Audit ID to compare against baseline")],
+) -> None:
+    """Compare an audit against its baseline."""
+    out, client = _get_client(ctx)
+    try:
+        path = endpoints.ANALYTICS_COMPARISON.format(audit_id=audit_id)
+        data = client.get(path)
+        print_result(data, out)
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()
+
+
+@analytics_app.command("trends")
+def analytics_trends(
+    ctx: typer.Context,
+    business_id: Annotated[str, typer.Argument(help="Business ID")],
+) -> None:
+    """Show KPI trends over time."""
+    out, client = _get_client(ctx)
+    try:
+        path = endpoints.ANALYTICS_TRENDS.format(business_id=business_id)
+        data = client.get(path)
+        print_result(data, out)
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()
+
+
+@analytics_app.command("summary")
+def analytics_summary(
+    ctx: typer.Context,
+    business_id: Annotated[str, typer.Argument(help="Business ID")],
+) -> None:
+    """Get analytics summary for a business."""
+    out, client = _get_client(ctx)
+    try:
+        path = endpoints.ANALYTICS_SUMMARY.format(business_id=business_id)
+        data = client.get(path)
+        print_result(data, out, human_formatter=lambda d, c: render_kv("Analytics Summary", d, c))
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()

cited_cli/commands/audit.py

@@ -0,0 +1,177 @@
+from __future__ import annotations
+
+from typing import Annotated
+
+import typer
+from rich.console import Console
+
+from cited_cli.output.formatter import OutputContext, print_error, print_result, print_success
+from cited_cli.output.tables import render_kv, render_table
+from cited_cli.utils.errors import CitedAPIError, ExitCode, handle_api_error
+from cited_cli.utils.interactive import prompt_if_missing
+from cited_core.api import endpoints
+from cited_core.api.client import LONG_TIMEOUT, CitedClient
+from cited_core.auth.store import TokenStore
+from cited_core.config.manager import ConfigManager
+
+audit_app = typer.Typer(name="audit", help="Run and manage audits.")
+
+
+def _get_client(ctx: typer.Context) -> tuple[OutputContext, CitedClient, str]:
+    obj = ctx.obj or {}
+    out: OutputContext = obj.get("output", OutputContext())
+    cfg: ConfigManager = obj.get("config", ConfigManager())
+    profile = obj.get("profile", "default")
+    env = cfg.get_environment(profile, obj.get("env_override"))
+    api_url = cfg.get_api_url(profile, obj.get("env_override"))
+
+    store = TokenStore()
+    token = store.get_token(env)
+    if not token:
+        print_error(f"Not logged in to {env}. Run: cited login", out)
+        raise typer.Exit(ExitCode.AUTH_ERROR)
+
+    return out, CitedClient(base_url=api_url, token=token), env
+
+
+@audit_app.command("start")
+def audit_start(
+    ctx: typer.Context,
+    named_audit_id: Annotated[
+        str | None, typer.Argument(help="Audit template (named audit) ID")
+    ] = None,
+    business_id: Annotated[
+        str | None,
+        typer.Option("--business", "-b", help="Business ID override"),
+    ] = None,
+    providers: Annotated[
+        list[str] | None,
+        typer.Option("--provider", help="Citation provider to use (repeatable)"),
+    ] = None,
+) -> None:
+    """Start a new audit job from a template."""
+    out, client, _ = _get_client(ctx)
+    named_audit_id = prompt_if_missing(
+        named_audit_id, "NAMED_AUDIT_ID", "Audit template ID", out
+    )
+    try:
+        payload: dict[str, object] = {"named_audit_id": named_audit_id}
+        if business_id:
+            payload["business_id"] = business_id
+        if providers:
+            payload["providers"] = providers
+        data = client.post(
+            endpoints.AUDIT_START,
+            json=payload,
+            timeout=LONG_TIMEOUT,
+        )
+        job_id = data.get("job_id", data.get("id", ""))
+        print_result(data, out, human_formatter=lambda d, c: render_kv("Audit Started", d, c))
+        if not out.json_mode and job_id:
+            out.console.print(f"\nTrack progress: [bold]cited job watch {job_id}[/bold]")
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()
+
+
+@audit_app.command("status")
+def audit_status(
+    ctx: typer.Context,
+    job_id: Annotated[str, typer.Argument(help="Audit job ID")],
+) -> None:
+    """Check audit job status."""
+    out, client, _ = _get_client(ctx)
+    try:
+        path = endpoints.AUDIT_STATUS.format(job_id=job_id)
+        data = client.get(path)
+        print_result(data, out, human_formatter=lambda d, c: render_kv("Audit Status", d, c))
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()
+
+
+@audit_app.command("result")
+def audit_result(
+    ctx: typer.Context,
+    job_id: Annotated[str, typer.Argument(help="Audit job ID")],
+) -> None:
+    """Get full audit results."""
+    out, client, _ = _get_client(ctx)
+    try:
+        path = endpoints.AUDIT_RESULT.format(job_id=job_id)
+        data = client.get(path)
+        print_result(data, out)
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()
+
+
+@audit_app.command("list")
+def audit_list(
+    ctx: typer.Context,
+    business_id: Annotated[
+        str | None,
+        typer.Option("--business", "-b", help="Filter by business ID"),
+    ] = None,
+) -> None:
+    """List audit history."""
+    out, client, _ = _get_client(ctx)
+    try:
+        params = {}
+        if business_id:
+            params["business_id"] = business_id
+        data = client.get(endpoints.AUDIT_HISTORY, params=params or None)
+        audits = data if isinstance(data, list) else data.get("audits", data.get("items", []))
+
+        def _human(d: object, console: Console) -> None:
+            items = d if isinstance(d, list) else []
+            rows = []
+            for a in items:
+                rows.append([
+                    a.get("job_id", a.get("id", ""))[:8],
+                    a.get("business_name", ""),
+                    a.get("status", ""),
+                    a.get("created_at", "")[:19] if a.get("created_at") else "",
+                ])
+            render_table(
+                "Audit History",
+                ["Job ID", "Business", "Status", "Created"],
+                rows,
+                console,
+            )
+
+        print_result(audits, out, human_formatter=_human)
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()
+
+
+@audit_app.command("export")
+def audit_export(
+    ctx: typer.Context,
+    job_id: Annotated[str, typer.Argument(help="Audit job ID")],
+    output_path: Annotated[
+        str | None,
+        typer.Option("--output", "-o", help="Output file path"),
+    ] = None,
+) -> None:
+    """Export audit as PDF."""
+    out, client, _ = _get_client(ctx)
+    try:
+        path = endpoints.AUDIT_EXPORT_PDF.format(job_id=job_id)
+        response = client.post_raw(path, timeout=LONG_TIMEOUT)
+        if not response.is_success:
+            raise CitedAPIError(response.status_code, f"Export failed ({response.status_code})")
+
+        filename = output_path or f"audit-{job_id[:8]}.pdf"
+        with open(filename, "wb") as f:
+            f.write(response.content)
+        print_success(f"Exported to {filename}", out)
+    except CitedAPIError as e:
+        handle_api_error(e, out.json_mode)
+    finally:
+        client.close()