diff --git a/litellm/__init__.py b/litellm/__init__.py index c5c2cd68d..052e542d6 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -1,5 +1,5 @@ import threading -from typing import Callable, List, Optional +from typing import Callable, List, Optional, Dict input_callback: List[str] = [] success_callback: List[str] = [] @@ -23,7 +23,7 @@ vertex_location: Optional[str] = None togetherai_api_key: Optional[str] = None caching = False caching_with_models = False # if you want the caching key to be model + prompt -model_alias_map = {} +model_alias_map: Dict[str, str] = {} debugger = False model_cost = { "babbage-002": {