From 7910cee2594fe531b2bee6fb1de334d75f472d17 Mon Sep 17 00:00:00 2001 From: Sidharth Mohanty Date: Mon, 11 Dec 2023 05:41:51 +0530 Subject: [PATCH] [chore] fix wrong import of LLMs from langchain (#1002) --- embedchain/llm/cohere.py | 2 +- embedchain/llm/huggingface.py | 2 +- embedchain/llm/llama2.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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