ape-linux 0.3.0__tar.gz → 0.3.1__tar.gz

{ape_linux-0.3.0 → ape_linux-0.3.1}/PKG-INFO

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

{ape_linux-0.3.0 → ape_linux-0.3.1}/ape_linux.py

@@ -2,15 +2,24 @@
 
 import os
 import subprocess
+from importlib.metadata import version
 from typing import Annotated
 
 import openai
 import rich.console
 import typer
 
+__version__ = version("ape_linux")
+
 app = typer.Typer(add_completion=False, pretty_exceptions_enable=False)
 
 
+def version_callback(value: bool) -> None:
+    if value:
+        typer.echo(__version__)
+        raise typer.Exit()
+
+
 def call_llm(
     api_key: str, model: str, system_prompt: str, user_prompt: str
 ) -> str | None:
@@ -49,6 +58,16 @@ def main(
             help="Run the command if suggested. Dangerous!",
         ),
     ] = False,
+    version: Annotated[
+        bool | None,
+        typer.Option(
+            "--version",
+            "-v",
+            callback=version_callback,
+            help="Show the version and exit.",
+            is_eager=True,
+        ),
+    ] = None,
 ):
     """Suggest a command for a Linux task described in QUERY.
 

{ape_linux-0.3.0 → ape_linux-0.3.1}/pyproject.toml

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

{ape_linux-0.3.0 → ape_linux-0.3.1}/tests/test_app.py

@@ -52,3 +52,10 @@ def test_app_for_suggestion_with_execute(mockenv, monkeypatch):
     assert result.stdout == "ls\n"  # careful, this will actually run
     assert result.stderr == ""
     assert result.exit_code == 0
+
+
+def test_app_for_version(mockenv):
+    result = runner.invoke(ape_linux.app, ["--version"])
+    assert result.stdout == f"{ape_linux.__version__}\n"
+    assert result.stderr == ""
+    assert result.exit_code == 0