ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix test_openai_moderation_error_raising

Browse Source
This commit is contained in:
Ishaan Jaff 2025-04-29 07:26:41 -07:00
parent ae7eecaa40
commit 1ef17c2f0e
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
litellm/types/llms/openai.py
View File

@ -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