1from __future__ import annotations
 2
 3import sys
 4
 5import openai
 6
 7from .. import OpenAI, _load_client
 8from .._compat import model_json
 9from .._models import BaseModel
10
11
12class Colors:
13    HEADER = "\033[95m"
14    OKBLUE = "\033[94m"
15    OKGREEN = "\033[92m"
16    WARNING = "\033[93m"
17    FAIL = "\033[91m"
18    ENDC = "\033[0m"
19    BOLD = "\033[1m"
20    UNDERLINE = "\033[4m"
21
22
23def get_client() -> OpenAI:
24    return _load_client()
25
26
27def organization_info() -> str:
28    organization = openai.organization
29    if organization is not None:
30        return "[organization={}] ".format(organization)
31
32    return ""
33
34
35def print_model(model: BaseModel) -> None:
36    sys.stdout.write(model_json(model, indent=2) + "\n")
37
38
39def can_use_http2() -> bool:
40    try:
41        import h2  # type: ignore  # noqa
42    except ImportError:
43        return False
44
45    return True