diff --git a/litellm/__init__.py b/litellm/__init__.py index 21d688c1f08..67bd50d6d38 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -9,7 +9,6 @@ from typing import Callable, List, Optional, Dict, Union, Any, Literal, get_args from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler from litellm.caching.caching import Cache, DualCache, RedisCache, InMemoryCache from litellm.caching.llm_caching_handler import LLMClientCache -from litellm.model_registry import * from litellm.types.llms.bedrock import COHERE_EMBEDDING_INPUT_TYPES from litellm.types.utils import ( ImageObject, @@ -77,6 +76,8 @@ import dotenv litellm_mode = os.getenv("LITELLM_MODE", "DEV") # "PRODUCTION", "DEV" if litellm_mode == "DEV": dotenv.load_dotenv() + +from litellm.model_registry import * ################################################## if set_verbose == True: _turn_on_debug()