chore: Add pylint for storage (#1298)

This commit is contained in:
Fangyin Cheng
2024-03-15 15:42:46 +08:00
committed by GitHub
parent a207640ff2
commit 8897d6e8fd
50 changed files with 784 additions and 667 deletions

View File

@@ -20,14 +20,8 @@ logger = logging.getLogger(__name__)
class EmbeddingsModelWorker(ModelWorker):
def __init__(self) -> None:
try:
from langchain.embeddings import HuggingFaceEmbeddings
from langchain.embeddings.base import Embeddings
except ImportError as exc:
raise ImportError(
"Could not import langchain.embeddings.HuggingFaceEmbeddings python package. "
"Please install it with `pip install langchain`."
) from exc
from dbgpt.rag.embedding import Embeddings, HuggingFaceEmbeddings
self._embeddings_impl: Embeddings = None
self._model_params = None
self.model_name = None