aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/openai/cli/_utils.py
diff options
context:
space:
mode:
Diffstat (limited to '.venv/lib/python3.12/site-packages/openai/cli/_utils.py')
-rw-r--r--.venv/lib/python3.12/site-packages/openai/cli/_utils.py45
1 files changed, 45 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/openai/cli/_utils.py b/.venv/lib/python3.12/site-packages/openai/cli/_utils.py
new file mode 100644
index 00000000..673eed61
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/openai/cli/_utils.py
@@ -0,0 +1,45 @@
+from __future__ import annotations
+
+import sys
+
+import openai
+
+from .. import OpenAI, _load_client
+from .._compat import model_json
+from .._models import BaseModel
+
+
+class Colors:
+ HEADER = "\033[95m"
+ OKBLUE = "\033[94m"
+ OKGREEN = "\033[92m"
+ WARNING = "\033[93m"
+ FAIL = "\033[91m"
+ ENDC = "\033[0m"
+ BOLD = "\033[1m"
+ UNDERLINE = "\033[4m"
+
+
+def get_client() -> OpenAI:
+ return _load_client()
+
+
+def organization_info() -> str:
+ organization = openai.organization
+ if organization is not None:
+ return "[organization={}] ".format(organization)
+
+ return ""
+
+
+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