phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +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 f5512b1e8d
commit 778cba702e
2 changed files with 5 additions and 2 deletions
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, self,
json_response: Dict[str, Any], json_response: Dict[str, Any],
model_response: litellm.ImageResponse, model_response: litellm.ImageResponse,
model: str, model: Optional[str] = None,
) -> litellm.ImageResponse: ) -> litellm.ImageResponse:
if "predictions" not in json_response: if "predictions" not in json_response:
raise litellm.InternalServerError( raise litellm.InternalServerError(

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

@ -1,5 +1,6 @@
import re import re
from datetime import datetime from datetime import datetime
from typing import Union
import httpx import httpx
@ -117,7 +118,9 @@ class PassThroughEndpointLogging:
model = self.extract_model_from_url(url_route) model = self.extract_model_from_url(url_route)
_json_response = httpx_response.json() _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( if vertex_image_generation_class.is_image_generation_response(
_json_response _json_response
): ):