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

fix: fix linting errors

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-13 14:39:42 -07:00
parent 55e153556a
commit 7e769f3b89
3 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/custom_callbacks1.py
View file

@ -1,7 +1,8 @@
from litellm.integrations.custom_logger import CustomLogger from typing import Literal, Optional
import litellm import litellm
from litellm.proxy.proxy_server import UserAPIKeyAuth, DualCache from litellm.integrations.custom_logger import CustomLogger
from typing import Optional, Literal from litellm.proxy.proxy_server import DualCache, UserAPIKeyAuth
# This file includes the custom callbacks for LiteLLM Proxy # This file includes the custom callbacks for LiteLLM Proxy
@ -27,6 +28,7 @@ class MyCustomHandler(
"image_generation", "image_generation",
"moderation", "moderation",
"audio_transcription", "audio_transcription",
"pass_through_endpoint",
], ],
): ):
return data return data

1
litellm/proxy/hooks/dynamic_rate_limiter.py
View file

@ -198,6 +198,7 @@ class _PROXY_DynamicRateLimitHandler(CustomLogger):
"image_generation", "image_generation",
"moderation", "moderation",
"audio_transcription", "audio_transcription",
"pass_through_endpoint",
], ],
) -> Optional[ ) -> Optional[
Union[Exception, str, dict] Union[Exception, str, dict]

1
litellm/tests/test_proxy_reject_logging.py
View file

@ -66,6 +66,7 @@ class testLogger(CustomLogger):
"image_generation", "image_generation",
"moderation", "moderation",
"audio_transcription", "audio_transcription",
"pass_through_endpoint",
], ],
): ):
raise HTTPException( raise HTTPException(