scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-15 23:31:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 165

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
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/vertex_ai_and_google_ai_studio/image_generation/image_generation_handler.py
View file

@ -16,7 +16,7 @@ class VertexImageGeneration(VertexLLM):
self,
json_response: Dict[str, Any],
model_response: litellm.ImageResponse,
model: str,
model: Optional[str] = None,
) -> litellm.ImageResponse:
if "predictions" not in json_response:
raise litellm.InternalServerError(

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
):