Skip to content

Commit 944029f

Browse files
committed
Add environment variable to include the cause in exception str
1 parent dc33cb9 commit 944029f

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

openai/error.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,7 @@
11
import openai
2+
import os
3+
4+
display_cause = os.environ["OPENAI_ERROR_DISPLAY_CAUSE"]
25

36

47
class OpenAIError(Exception):
@@ -34,6 +37,8 @@ def __init__(
3437

3538
def __str__(self):
3639
msg = self._message or "<empty message>"
40+
if display_cause is not None and hasattr(self, "__cause__") and self.__cause__ is not None:
41+
msg += " (Cause: {0})".format(self.__cause__)
3742
if self.request_id is not None:
3843
return "Request {0}: {1}".format(self.request_id, msg)
3944
else:

0 commit comments

Comments
 (0)