From ebb19ff8cb1f8fcc3e224390e351676daccb33de Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 21 May 2025 21:56:36 -0700 Subject: [PATCH] fix: linting --- litellm/types/utils.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/litellm/types/utils.py b/litellm/types/utils.py index 5d4b20bdb0d..3035e90f5b8 100644 --- a/litellm/types/utils.py +++ b/litellm/types/utils.py @@ -1550,7 +1550,11 @@ class ImageUsageInputTokensDetails(BaseLiteLLMOpenAIResponseObject): """The number of text tokens in the input prompt.""" -class ImageUsage(BaseLiteLLMOpenAIResponseObject): +from openai.types.images_response import ImagesResponse as OpenAIImageResponse +from openai.types.images_response import Usage as OpenAIImageResponseUsage + + +class ImageUsage(OpenAIImageResponseUsage, BaseLiteLLMOpenAIResponseObject): input_tokens: int """The number of tokens (images and text) in the input prompt.""" @@ -1569,7 +1573,7 @@ from openai.types.images_response import ImagesResponse as OpenAIImageResponse class ImageResponse(OpenAIImageResponse, BaseLiteLLMOpenAIResponseObject): _hidden_params: dict = {} - usage: ImageUsage + usage: Optional[ImageUsage] = None def __init__( self,