diff --git a/embedchain/llm/cohere.py b/embedchain/llm/cohere.py index e996db6b..7ddac928 100644 --- a/embedchain/llm/cohere.py +++ b/embedchain/llm/cohere.py @@ -2,7 +2,7 @@ import importlib import os from typing import Optional -from langchain.llms import Cohere +from langchain.llms.cohere import Cohere from embedchain.config import BaseLlmConfig from embedchain.helpers.json_serializable import register_deserializable diff --git a/embedchain/llm/huggingface.py b/embedchain/llm/huggingface.py index a4628c0d..2d7601ee 100644 --- a/embedchain/llm/huggingface.py +++ b/embedchain/llm/huggingface.py @@ -3,7 +3,7 @@ import logging import os from typing import Optional -from langchain.llms import HuggingFaceHub +from langchain.llms.huggingface_hub import HuggingFaceHub from embedchain.config import BaseLlmConfig from embedchain.helpers.json_serializable import register_deserializable diff --git a/embedchain/llm/llama2.py b/embedchain/llm/llama2.py index fa2587bb..1e74d1bb 100644 --- a/embedchain/llm/llama2.py +++ b/embedchain/llm/llama2.py @@ -2,7 +2,7 @@ import importlib import os from typing import Optional -from langchain.llms import Replicate +from langchain.llms.replicate import Replicate from embedchain.config import BaseLlmConfig from embedchain.helpers.json_serializable import register_deserializable