agnes-the-ai-analyst/cli/commands/schema.py
ZdenekSrotyr 2e1dfb7553
feat(v2): claude-driven fetch primitives + 0.14.0 (#102)
Replaces the BigQuery wrap-view pattern with a discovery + scoped-fetch toolkit driven by the analyst's Claude session. Adds /api/v2/{catalog,schema,sample,scan,scan/estimate}, da catalog/schema/describe/fetch/snapshot/disk-info CLI commands, sqlglot-backed WHERE validator, process-local quota tracker, agent rails skill (cli/skills/agnes-data-querying.md). BREAKING: BQ wrap views off by default — set data_source.bigquery.legacy_wrap_views=true for one cycle. Backward-compat field_validator on primary_key. Catalog cache now matches documented 300s TTL with RBAC fresh per request. Cuts release v0.14.0.
2026-04-29 01:07:19 +02:00

45 lines
1.6 KiB
Python

"""`da schema <table>` — show columns + BQ flavor hints (spec §4.1)."""
import json as json_lib
import typer
from cli.v2_client import api_get_json, V2ClientError
schema_app = typer.Typer(help="Show column metadata for a table")
@schema_app.callback(invoke_without_command=True)
def schema(
ctx: typer.Context,
table_id: str = typer.Argument(...),
json: bool = typer.Option(False, "--json"),
):
"""Show column metadata for a table."""
if ctx.invoked_subcommand is not None:
return
try:
data = api_get_json(f"/api/v2/schema/{table_id}")
except V2ClientError as e:
typer.echo(f"Error: schema fetch failed: {e}", err=True)
raise typer.Exit(5 if e.status_code >= 500 else 8 if e.status_code == 403 else 2)
if json:
typer.echo(json_lib.dumps(data, indent=2))
return
flavor = data.get("sql_flavor", "duckdb")
typer.echo(f"Table: {data['table_id']} ({data['source_type']} — use {flavor.upper()} SQL dialect)")
typer.echo("")
typer.echo(f"{'COLUMN':30s} {'TYPE':15s} {'NULL':5s} DESCRIPTION")
for c in data.get("columns", []):
typer.echo(
f"{c['name']:30s} {c['type']:15s} "
f"{'YES' if c.get('nullable') else 'NO':5s} {c.get('description', '')}"
)
if data.get("partition_by"):
typer.echo(f"\nPartition: {data['partition_by']}")
if data.get("clustered_by"):
typer.echo(f"Clustered: {', '.join(data['clustered_by'])}")
if data.get("where_dialect_hints"):
typer.echo("\nWHERE dialect hints:")
for k, v in data["where_dialect_hints"].items():
typer.echo(f" {k:25s} {v}")