fix linting errors

This commit is contained in:
Krrish Dholakia 2023-09-21 14:10:27 -07:00
parent abea880c9d
commit 31c995a1a4
2 changed files with 3 additions and 3 deletions

View file

@ -2756,7 +2756,7 @@ def mock_completion_streaming_obj(model_response, mock_response, model):
yield model_response yield model_response
########## Reading Config File ############################ ########## Reading Config File ############################
def read_config_args(config_path): def read_config_args(config_path) -> dict:
try: try:
import os import os
@ -2815,7 +2815,7 @@ def completion_with_config(*args, config: Union[dict, str], **kwargs):
except Exception as e: except Exception as e:
exception_name = type(e).__name__ exception_name = type(e).__name__
fallback_model = None fallback_model = None
if exception_name in error_handling: if error_handling and exception_name in error_handling:
error_handler = error_handling[exception_name] error_handler = error_handling[exception_name]
# either switch model or api key # either switch model or api key
fallback_model = error_handler.get("fallback_model", None) fallback_model = error_handler.get("fallback_model", None)

View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "litellm" name = "litellm"
version = "0.1.725" version = "0.1.726"
description = "Library to easily interface with LLM API providers" description = "Library to easily interface with LLM API providers"
authors = ["BerriAI"] authors = ["BerriAI"]
license = "MIT License" license = "MIT License"