ee83cebbda
Ports Minas's PR #172 (against pre-rename `da` CLI on main) and applies the principle to the post-rename `agnes` CLI. Two distinct failure modes on Windows consoles whose default codepage is cp1250 (cs-CZ) / cp1252 (en-US): 1. `agnes pull` and other Rich-progress codepaths UnicodeEncodeError on Braille spinner glyphs. Fix: `cli/main.py` reconfigures stdout/stderr to UTF-8 with errors='replace' at import time on `sys.platform == 'win32'` so Rich's legacy-Windows render path emits decodable bytes. Wrapped in try/except so pytest's captured streams (which aren't TextIOWrapper) don't break. 2. `agnes skills list` and `agnes skills show` UnicodeDecodeError when reading skill markdown containing em-dashes / accented chars. Default `Path.read_text()` uses locale.getpreferredencoding(False), which is the broken codepage on Windows. Fix: every call site passes encoding='utf-8' explicitly. Broader scope than #172 because: - The bootstrap rewrite renamed/removed several files Minas's PR patched (`cli/commands/analyst.py` -> rolled into init.py; `cli/commands/sync.py` -> split into pull/push). Those targets no longer exist; the equivalent code lives in init.py. - Other call sites Minas didn't touch (still bare in his branch) are patched here too — config.py / update_check.py / snapshot_meta.py / setup.py / skills.py — so the codebase has zero locale-default text I/O in cli/. Side cleanup: stale `Run `da`` reference in snapshot_meta.py:88 fixed to `agnes` while touching the file.
127 lines
4.3 KiB
Python
127 lines
4.3 KiB
Python
"""agnes — CLI tool for AI Data Analyst.
|
|
|
|
Primary interface for AI agents. Install: uv tool install agnes-the-ai-analyst
|
|
"""
|
|
|
|
import sys
|
|
from importlib.metadata import PackageNotFoundError
|
|
from importlib.metadata import version as _pkg_version
|
|
|
|
import typer
|
|
|
|
# Force UTF-8 on Windows stdout/stderr at import time. The default Windows
|
|
# console codepage (cp1250 on cs-CZ, cp1252 on en-US, …) cannot encode the
|
|
# Braille spinner glyphs Rich uses for `agnes pull` progress, nor the
|
|
# em-dash / accented chars that show up in skill markdown via
|
|
# `agnes skills list`. Both crash with UnicodeEncodeError /
|
|
# UnicodeDecodeError before any command-level code runs. `reconfigure` is
|
|
# a no-op on non-TextIOWrapper streams (pytest capture, pipes wrapped by
|
|
# other tooling) — swallow the AttributeError there.
|
|
if sys.platform == "win32":
|
|
try:
|
|
sys.stdout.reconfigure(encoding="utf-8", errors="replace")
|
|
sys.stderr.reconfigure(encoding="utf-8", errors="replace")
|
|
except (AttributeError, OSError):
|
|
pass
|
|
|
|
from cli.commands.auth import auth_app
|
|
from cli.commands.init import init_app
|
|
from cli.commands.pull import pull_app
|
|
from cli.commands.push import push_app
|
|
from cli.commands.query import query_command
|
|
from cli.commands.status import status_app
|
|
from cli.commands.admin import admin_app
|
|
from cli.commands.diagnose import diagnose_app
|
|
from cli.commands.skills import skills_app
|
|
from cli.commands.setup import setup_app
|
|
from cli.commands.server import server_app
|
|
from cli.commands.explore import explore_app
|
|
from cli.commands.catalog import catalog_app
|
|
from cli.commands.schema import schema_app
|
|
from cli.commands.describe import describe_app
|
|
from cli.commands.snapshot import snapshot_app
|
|
from cli.commands.disk_info import disk_info_app
|
|
|
|
|
|
def _cli_version() -> str:
|
|
"""Return the installed CLI version from package metadata.
|
|
|
|
Falls back to `"unknown"` when the package is not installed (e.g. running
|
|
from a source checkout without `uv pip install -e .`). Deliberately does
|
|
not read pyproject.toml at runtime — that file is not shipped with the
|
|
wheel and the metadata lookup is the canonical source.
|
|
"""
|
|
try:
|
|
return _pkg_version("agnes-the-ai-analyst")
|
|
except PackageNotFoundError:
|
|
return "unknown"
|
|
|
|
|
|
def _version_callback(value: bool) -> None:
|
|
if value:
|
|
typer.echo(f"agnes {_cli_version()}")
|
|
raise typer.Exit()
|
|
|
|
|
|
app = typer.Typer(
|
|
name="agnes",
|
|
help="Agnes — AI Data Analyst CLI",
|
|
no_args_is_help=True,
|
|
)
|
|
|
|
|
|
@app.callback()
|
|
def _root(
|
|
version: bool = typer.Option(
|
|
None,
|
|
"--version",
|
|
"-V",
|
|
callback=_version_callback,
|
|
is_eager=True,
|
|
help="Show the CLI version and exit.",
|
|
),
|
|
) -> None:
|
|
"""Root callback — carries the --version option and fires the auto-update check.
|
|
|
|
Update check runs before subcommand dispatch but after the --version flag
|
|
(which exits early). It's best-effort: any failure is swallowed so a bad
|
|
network never blocks a working `agnes` command. Disable with
|
|
`AGNES_NO_UPDATE_CHECK=1`.
|
|
"""
|
|
_maybe_warn_outdated()
|
|
|
|
|
|
def _maybe_warn_outdated() -> None:
|
|
"""Hit /cli/latest on the configured server (cached 24h) and emit a
|
|
one-line stderr warning if the installed CLI is older. Never raises."""
|
|
try:
|
|
from cli.config import get_server_url
|
|
from cli.update_check import check, format_outdated_notice
|
|
info = check(get_server_url())
|
|
if info and info.is_outdated():
|
|
typer.echo(format_outdated_notice(info), err=True)
|
|
except Exception:
|
|
pass # best-effort: never fail a command on the probe
|
|
|
|
# Register subcommands
|
|
app.add_typer(auth_app, name="auth")
|
|
app.add_typer(init_app, name="init")
|
|
app.add_typer(pull_app, name="pull")
|
|
app.add_typer(push_app, name="push")
|
|
app.command("query")(query_command)
|
|
app.add_typer(status_app, name="status")
|
|
app.add_typer(admin_app, name="admin")
|
|
app.add_typer(diagnose_app, name="diagnose")
|
|
app.add_typer(skills_app, name="skills")
|
|
app.add_typer(setup_app, name="setup")
|
|
app.add_typer(server_app, name="server")
|
|
app.add_typer(explore_app, name="explore")
|
|
app.add_typer(catalog_app, name="catalog")
|
|
app.add_typer(schema_app, name="schema")
|
|
app.add_typer(describe_app, name="describe")
|
|
app.add_typer(snapshot_app, name="snapshot")
|
|
app.add_typer(disk_info_app, name="disk-info")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app()
|