phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix linting errors

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-02 19:39:10 -07:00
parent b0735c9e9f
commit bfb39eb0cd
2 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/proxy/pass_through_endpoints/success_handler.py
View file

@ -1,5 +1,6 @@
import re
from datetime import datetime
from typing import Union
import httpx
@ -117,7 +118,9 @@ class PassThroughEndpointLogging:
model = self.extract_model_from_url(url_route)
_json_response = httpx_response.json()
litellm_model_response = litellm.ModelResponse()
litellm_model_response: Union[
litellm.ModelResponse, litellm.EmbeddingResponse, litellm.ImageResponse
] = litellm.ModelResponse()
if vertex_image_generation_class.is_image_generation_response(
_json_response
):