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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ape-linux
-Version: 0.3.0
+Version: 0.3.2
 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.2}/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.
 
@@ -77,7 +96,19 @@ def main(
     Answer: echo "Please try again."
 
     Question: Tell me a story
-    Answer: echo "Please try again.\""""  # noqa: E501
+    Answer: echo "Please try again.
+
+    It is important that you do not output commands enclosed in ``` ``` Markdown blocks. For example,
+
+    ```sh
+    cd projects
+    ls
+    ```
+
+    is not allowed. Instead, your output should be a command that is to be entered directly into the command line. For the example above: cd projects && ls
+
+    You are also allowed to use \\ for command continuation.
+    \""""  # noqa: E501
 
     user_prompt = f"""\
     Question: {query.strip()}

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

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

{ape_linux-0.3.0 → ape_linux-0.3.2}/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

{ape_linux-0.3.0 → ape_linux-0.3.2}/uv.lock

@@ -28,7 +28,7 @@ wheels = [
 
 [[package]]
 name = "ape-linux"
-version = "0.3.0"
+version = "0.3.1"
 source = { editable = "." }
 dependencies = [
     { name = "openai" },