fix(openai.py): fix openai image gen logging

2026-08-07 00:24:03 +00:00 · 2024-01-26 21:05:49 -08:00
parent 618d216c11
commit d755d50901
1 changed files with 14 additions and 0 deletions
@@ -718,8 +718,22 @@ class OpenAIChatCompletion(BaseLLM):
            return convert_to_model_response_object(response_object=response, model_response_object=model_response, response_type="image_generation")  # type: ignore
        except OpenAIError as e:
            exception_mapping_worked = True
+            ## LOGGING
+            logging_obj.post_call(
+                input=prompt,
+                api_key=api_key,
+                additional_args={"complete_input_dict": data},
+                original_response=str(e),
+            )
            raise e
        except Exception as e:
+            ## LOGGING
+            logging_obj.post_call(
+                input=prompt,
+                api_key=api_key,
+                additional_args={"complete_input_dict": data},
+                original_response=str(e),
+            )
            if hasattr(e, "status_code"):
                raise OpenAIError(status_code=e.status_code, message=str(e))
            else: