fix import loc

This commit is contained in:
Ishaan Jaff 2025-06-28 14:27:12 -07:00
parent f93326a214
commit 092e3591f5

View file

@ -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()