feat: display logo when running codeflash --help

codeflash/main.py

@@ -33,6 +33,8 @@ def main() -> None:
     from codeflash.telemetry import posthog_cf
     from codeflash.telemetry.sentry import init_sentry
 
+    if "--help" in sys.argv[1:] or "-h" in sys.argv[1:]:
+        print_codeflash_banner()
     args = parse_args()
     if args.command != "auth":
         print_codeflash_banner()

tests/test_help_banner.py

@@ -0,0 +1,18 @@
+import subprocess
+import sys
+
+
+def test_help_displays_logo() -> None:
+    result = subprocess.run(
+        [sys.executable, "-c", "from codeflash.main import main; main()", "--help"], capture_output=True, text=True
+    )
+    assert result.returncode == 0
+    assert "codeflash.ai" in result.stdout
+
+
+def test_help_short_flag_displays_logo() -> None:
+    result = subprocess.run(
+        [sys.executable, "-c", "from codeflash.main import main; main()", "-h"], capture_output=True, text=True
+    )
+    assert result.returncode == 0
+    assert "codeflash.ai" in result.stdout