From 16f4572fe7eca119cf41190814d7a8c93d1fa45b Mon Sep 17 00:00:00 2001
From: Andy Lee <andylizf@outlook.com>
Date: Thu, 14 Aug 2025 11:22:47 -0700
Subject: [PATCH] cli: use argparse.BooleanOptionalAction for paired flags
 (--recompute/--compact) across build/search/ask

---
 packages/leann-core/src/leann/cli.py | 38 +++++++++-------------------
 1 file changed, 12 insertions(+), 26 deletions(-)

diff --git a/packages/leann-core/src/leann/cli.py b/packages/leann-core/src/leann/cli.py
index 07ff0d7..d56f386 100644
--- a/packages/leann-core/src/leann/cli.py
+++ b/packages/leann-core/src/leann/cli.py
@@ -116,19 +116,17 @@ Examples:
         build_parser.add_argument("--graph-degree", type=int, default=32)
         build_parser.add_argument("--complexity", type=int, default=64)
         build_parser.add_argument("--num-threads", type=int, default=1)
-        build_parser.add_argument("--compact", action="store_true", default=True)
         build_parser.add_argument(
-            "--no-compact",
-            dest="compact",
-            action="store_false",
-            help="Disable compact index storage (store full embeddings; higher storage)",
+            "--compact",
+            action=argparse.BooleanOptionalAction,
+            default=True,
+            help="Use compact index storage (default: enabled). Use --no-compact to disable and store full embeddings.",
         )
-        build_parser.add_argument("--recompute", action="store_true", default=True)
         build_parser.add_argument(
-            "--no-recompute",
-            dest="recompute",
-            action="store_false",
-            help="Disable embedding recomputation (store full embeddings; lower query latency)",
+            "--recompute",
+            action=argparse.BooleanOptionalAction,
+            default=True,
+            help="Enable embedding recomputation (default: enabled). Use --no-recompute to store full embeddings and speed up queries.",
         )
         build_parser.add_argument(
             "--file-types",
@@ -144,18 +142,12 @@ Examples:
         search_parser.add_argument("--complexity", type=int, default=64)
         search_parser.add_argument("--beam-width", type=int, default=1)
         search_parser.add_argument("--prune-ratio", type=float, default=0.0)
-        search_parser.add_argument(
-            "--no-recompute",
-            dest="recompute_embeddings",
-            action="store_false",
-            help="Disable embedding recomputation during search",
-        )
         search_parser.add_argument(
             "--recompute",
             dest="recompute_embeddings",
-            action="store_true",
+            action=argparse.BooleanOptionalAction,
             default=True,
-            help="Enable embedding recomputation during search (default)",
+            help="Enable/disable embedding recomputation during search (default: enabled)",
         )
         search_parser.add_argument(
             "--pruning-strategy",
@@ -179,18 +171,12 @@ Examples:
         ask_parser.add_argument("--complexity", type=int, default=32)
         ask_parser.add_argument("--beam-width", type=int, default=1)
         ask_parser.add_argument("--prune-ratio", type=float, default=0.0)
-        ask_parser.add_argument(
-            "--no-recompute",
-            dest="recompute_embeddings",
-            action="store_false",
-            help="Disable embedding recomputation during ask",
-        )
         ask_parser.add_argument(
             "--recompute",
             dest="recompute_embeddings",
-            action="store_true",
+            action=argparse.BooleanOptionalAction,
             default=True,
-            help="Enable embedding recomputation during ask (default)",
+            help="Enable/disable embedding recomputation during ask (default: enabled)",
         )
         ask_parser.add_argument(
             "--pruning-strategy",