Handle body being None

newrelic · Dec 14, 2023 · a7178d5 · a7178d5
1 parent 34fff33
commit a7178d5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/newrelic/hooks/mlmodel_openai.py b/newrelic/hooks/mlmodel_openai.py
@@ -199,7 +199,7 @@ def wrap_chat_completion_sync(wrapped, instance, args, kwargs):
                 # There appears to be a bug here in openai v1 where despite having code,
                 # param, etc in the error response, they are not populated on the exception
                 # object so grab them from the response body object instead.
-                body = getattr(exc, "body", {})
+                body = getattr(exc, "body", {}) or {}
                 notice_error_attributes = {
                     "http.statusCode": getattr(exc, "status_code", "") or "",
                     "error.message": body.get("message", "") or "",
@@ -638,7 +638,7 @@ async def wrap_chat_completion_async(wrapped, instance, args, kwargs):
                 # There appears to be a bug here in openai v1 where despite having code,
                 # param, etc in the error response, they are not populated on the exception
                 # object so grab them from the response body object instead.
-                body = getattr(exc, "body", {})
+                body = getattr(exc, "body", {}) or {}
                 notice_error_attributes = {
                     "http.statusCode": getattr(exc, "status_code", "") or "",
                     "error.message": body.get("message", "") or "",