From 1ef17c2f0e075d7fdaf5ca2b376efe02cedb8bcd Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 29 Apr 2025 07:26:41 -0700 Subject: [PATCH] fix test_openai_moderation_error_raising --- litellm/types/llms/openai.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/litellm/types/llms/openai.py b/litellm/types/llms/openai.py index 912f8be2cb..ff0aa89534 100644 --- a/litellm/types/llms/openai.py +++ b/litellm/types/llms/openai.py @@ -1270,6 +1270,13 @@ class ImageGenerationRequestQuality(str, Enum): HD = "hd" +class OpenAIModerationResult(BaseLiteLLMOpenAIResponseObject): + categories: Optional[Dict] + category_applied_input_types: Optional[Dict] + category_scores: Optional[Dict] + flagged: Optional[bool] + + class OpenAIModerationResponse(BaseLiteLLMOpenAIResponseObject): """ Response from the OpenAI Moderation API. @@ -1281,7 +1288,7 @@ class OpenAIModerationResponse(BaseLiteLLMOpenAIResponseObject): model: str """The model used to generate the moderation results.""" - results: List[Dict] + results: List[OpenAIModerationResult] """A list of moderation objects.""" # Define private attributes using PrivateAttr