feat(cli): use http/2 if h2 is available

Author: stainless-bot

src/openai/cli/_cli.py

@@ -12,7 +12,7 @@
 from . import _tools
 from .. import OpenAI, __version__
 from ._api import register_commands
-from ._utils import set_client
+from ._utils import set_client, can_use_http2
 from .._types import ProxiesDict
 from ._errors import CLIError, display_error
 from .._compat import PYDANTIC_V2, ConfigDict, model_parse
@@ -149,6 +149,7 @@ def _main() -> None:
 
     http_client = httpx.Client(
         proxies=proxies or None,
+        http2=can_use_http2(),
     )
 
     try:
@@ -178,14 +179,12 @@ def _main() -> None:
             )
         else:
             parsed.func()
-    except Exception:
+    finally:
         try:
             http_client.close()
         except Exception:
             pass
 
-        raise
-
 
 if __name__ == "__main__":
     sys.exit(main())

src/openai/cli/_utils.py

@@ -46,3 +46,12 @@ def organization_info() -> str:
 
 def print_model(model: BaseModel) -> None:
     sys.stdout.write(model_json(model, indent=2) + "\n")
+
+
+def can_use_http2() -> bool:
+    try:
+        import h2  # type: ignore  # noqa
+    except ImportError:
+        return False
+
+    return True