diff --git a/docs/docs/integrations/chat/litellm.ipynb b/docs/docs/integrations/chat/litellm.ipynb index 28104343953..499c229dde4 100644 --- a/docs/docs/integrations/chat/litellm.ipynb +++ b/docs/docs/integrations/chat/litellm.ipynb @@ -121,7 +121,7 @@ }, "outputs": [], "source": [ - "from langchain_litellm.chat_models import ChatLiteLLM\n", + "from langchain_litellm import ChatLiteLLM\n", "\n", "llm = ChatLiteLLM(model=\"gpt-3.5-turbo\")" ] diff --git a/docs/docs/integrations/providers/litellm.mdx b/docs/docs/integrations/providers/litellm.mdx index f3e39d17e2b..2ab7404ccc3 100644 --- a/docs/docs/integrations/providers/litellm.mdx +++ b/docs/docs/integrations/providers/litellm.mdx @@ -10,7 +10,7 @@ pip install langchain-litellm ## Chat Models ```python -from langchain_litellm.chat_models import ChatLiteLLM +from langchain_litellm import ChatLiteLLM ``` See more detail in the guide [here](/docs/integrations/chat/litellm).