_errors.py 471 B

123456789101112131415161718192021
  1. from __future__ import annotations
  2. import sys
  3. import pydantic
  4. from ._utils import Colors, organization_info
  5. from .._exceptions import APIError, OpenAIError
  6. class CLIError(OpenAIError): ...
  7. class SilentCLIError(CLIError): ...
  8. def display_error(err: CLIError | APIError | pydantic.ValidationError) -> None:
  9. if isinstance(err, SilentCLIError):
  10. return
  11. sys.stderr.write("{}{}Error:{} {}\n".format(organization_info(), Colors.FAIL, Colors.ENDC, err))