add default arguments for diff and pr-diff if no arg given (#287)

AbanteAI · Nov 21, 2023 · 5661c91 · 5661c91
1 parent b4a88c4
commit 5661c91
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/mentat/diff_context.py b/mentat/diff_context.py
@@ -133,6 +133,8 @@ def __init__(
 
         meta = get_treeish_metadata(git_root, target)
         name += f'{meta["hexsha"][:8]}: {meta["summary"]}'
+        if target == "HEAD":
+            name = "HEAD (last commit)"
 
         self.target = target
         self.name = name

diff --git a/mentat/terminal/client.py b/mentat/terminal/client.py
@@ -207,8 +207,10 @@ def run_cli():
     parser.add_argument(
         "--diff",
         "-d",
+        nargs="?",
         type=str,
         default=None,
+        const="HEAD",
         help="A git tree-ish (e.g. commit, branch, tag) to diff against",
     )
     parser.add_argument(