diff --git a/litellm/router.py b/litellm/router.py index 5f23199411f..862d4210068 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -166,7 +166,11 @@ from litellm.types.utils import ( ) from litellm.types.utils import ModelInfo from litellm.types.utils import ModelInfo as ModelMapInfo -from litellm.types.utils import ModelResponseStream, StandardLoggingPayload, Usage +from litellm.types.utils import ( + ModelResponseStream, + StandardLoggingPayload, + Usage, +) from litellm.utils import ( CustomStreamWrapper, EmbeddingResponse,