fix(cli): Fix shtab generation for zsh and add tests for it. Fix #211

Author: Zhe Yu

src/vectorcode/cli_utils.py

@@ -358,7 +358,7 @@ def get_cli_parser():
     )
     prompts_parser.add_argument(
         "prompt_categories",
-        choices=PromptCategory,
+        choices=[str(i) for i in PromptCategory],
         type=PromptCategory,
         nargs="*",
         help="The subcommand(s) to get the prompts for. When not provided, VectorCode will print the prompts for `query`.",

tests/test_cli_utils.py

@@ -1,4 +1,6 @@
 import os
+import subprocess
+import sys
 import tempfile
 from typing import Any, Dict
 from unittest.mock import patch
@@ -539,3 +541,15 @@ def test_cleanup_path():
         "~", "test_path"
     )
     assert cleanup_path("/etc/dir") == "/etc/dir"
+
+
+def test_shtab():
+    for shell in ("bash", "zsh", "tcsh"):
+        assert (
+            subprocess.Popen(
+                [sys.executable, "-m", "vectorcode.main", "-s", shell],
+                stderr=subprocess.PIPE,
+            )
+            .stderr.read()
+            .decode()
+        ) == ""