ape-linux 0.5.2__tar.gz → 0.5.4__tar.gz

{ape_linux-0.5.2 → ape_linux-0.5.4}/CLAUDE.md

@@ -110,7 +110,19 @@ The flow in `ape_linux.py` is intentionally minimal:
    yields `None` so **no** `model_settings` are sent — deliberately, because a
    hard-coded temperature crashes models that reject sampling settings (some reasoning
    models, Claude Opus 4.7/4.8) — and an unparseable value exits `1` before any LLM
-   call.
+   call. `call_llm()` also sets `pydantic_ai.BANNER_ENABLED = False` before building
+   the agent: recent Pydantic AI versions print a first-run banner (logo plus an
+   "observability: off" block) to **stderr** on the first agent run in a terminal,
+   which would sit next to Ape's one-line command output. That attribute is Pydantic
+   AI's documented switch for it. This is the **one** `ty: ignore` in the codebase
+   (`invalid-assignment`), and it is deliberate: Pydantic AI declares the flag as a
+   bare `BANNER_ENABLED = True` with no annotation, so ty infers `Literal[True]` and
+   rejects assigning `False` — the very thing the attribute exists for. No
+   `pydantic-ai-slim` floor bump is needed: the assignment is a harmless no-op on
+   versions that predate the banner. Note the banner is invisible to the test suite
+   regardless — Pydantic AI suppresses it whenever `PYTEST_VERSION`, `CI` or
+   `PYDANTIC_AI_NO_BANNER` is in the environment — so a regression here would only
+   show up when a user runs `ape` in a terminal.
 4. Errors are flattened to one-line stderr messages, raising `SystemExit(1)` —
    `ModelHTTPError` reports status/message; any other exception (bad credentials,
    unknown provider) prints `str(error)`. There is no CLI framework swallowing
@@ -133,6 +145,11 @@ stderr, exit `2`). The `mockenv` fixture sets a dummy `APE_API_KEY` and `APE_MOD
 blank/whitespace** `APE_MODEL` or `APE_API_KEY` is each checked to exit `1` before
 `call_llm` is ever reached.
 
+`test_call_llm_disables_the_pydantic_ai_banner` runs the real `call_llm` (with
+`TestModel` swapped in for `infer_model`) and asserts the banner flag ends up `False`.
+It sets the flag back to `True` via `monkeypatch.setattr(..., raising=False)` first, so
+the check is meaningful and monkeypatch restores the global afterwards.
+
 `build_provider()` is covered directly (not through `main()`): tests construct the
 `openai` and `anthropic` providers with the standard credential vars deleted and assert
 the injected key lands on `provider.client.api_key` (proving the standard var is neither

{ape_linux-0.5.2 → ape_linux-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: ape-linux
-Version: 0.5.2
+Version: 0.5.4
 Summary: AI for Linux commands
 Project-URL: Homepage, https://github.com/sbalian/ape
 Project-URL: Repository, https://github.com/sbalian/ape

{ape_linux-0.5.2 → ape_linux-0.5.4}/ape_linux.py

@@ -7,6 +7,7 @@ import sys
 from functools import partial
 from typing import Any, Protocol, cast
 
+import pydantic_ai
 from pydantic import BaseModel
 from pydantic_ai import Agent, ModelSettings
 from pydantic_ai.exceptions import ModelHTTPError
@@ -91,6 +92,18 @@ def call_llm(
     user_prompt: str,
     model_settings: ModelSettings | None,
 ) -> Command | CannotHelp:
+    # Pydantic AI shows a first-run banner (its logo plus an "observability: off" block)
+    # on stderr the first time an agent runs in a terminal. Ape prints a single command
+    # meant to be read or piped, so it owns its own output: turn the banner off before
+    # any run. This is Pydantic AI's documented switch for exactly that. Pydantic AI
+    # declares it as a bare `BANNER_ENABLED = True` with no annotation, so ty infers
+    # the narrowest type, `Literal[True]`, and reads any assignment of `False` as a
+    # type error; the assignment is precisely what the attribute is documented for, so
+    # the diagnostic is suppressed rather than worked around. Assigning it is also a
+    # harmless no-op on versions that predate the banner, so the pydantic-ai-slim floor
+    # stays where it is.
+    pydantic_ai.BANNER_ENABLED = False  # ty: ignore[invalid-assignment]
+
     # Two spellings here are for the type checker, and neither changes behavior: the
     # generic parameters are explicit (`object` is the default deps type, as Ape uses
     # no deps), and `output_type` uses Pydantic AI's sequence form instead of

{ape_linux-0.5.2 → ape_linux-0.5.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ape-linux"
-version = "0.5.2"
+version = "0.5.4"
 description = "AI for Linux commands"
 readme = "README.md"
 license = {text = "MIT License"}

{ape_linux-0.5.2 → ape_linux-0.5.4}/tests/test_app.py

@@ -3,6 +3,7 @@ import os
 import platform
 import socket
 
+import pydantic_ai
 import pytest
 from pydantic_ai.exceptions import ModelHTTPError
 from pydantic_ai.models.test import TestModel
@@ -258,6 +259,20 @@ def test_call_llm_returns_structured_output(monkeypatch):
     assert isinstance(result.command, str)
 
 
+def test_call_llm_disables_the_pydantic_ai_banner(monkeypatch):
+    # Pydantic AI shows a first-run banner on stderr, which would land next to ape's
+    # one-line output. call_llm turns it off before the agent runs. Setting the flag
+    # back to True first makes the check meaningful, and hands monkeypatch the global
+    # to restore afterwards; raising=False because versions predating the banner have
+    # no such attribute.
+    monkeypatch.setattr(pydantic_ai, "BANNER_ENABLED", True, raising=False)
+    monkeypatch.setattr(
+        "ape_linux.infer_model", lambda model, provider_factory: TestModel()
+    )
+    ape_linux.call_llm("openai:gpt-4.1", "key", "system", "user", None)
+    assert pydantic_ai.BANNER_ENABLED is False
+
+
 def test_build_provider_injects_key_without_standard_env_var(monkeypatch):
     # build_provider constructs the inferred provider with our key injected directly,
     # so the provider's standard credential variable is neither needed nor read.