From 29ab244de601da5c20e10c67b1b15cb57d85169d Mon Sep 17 00:00:00 2001 From: Bowen Liang Date: Tue, 27 Feb 2024 18:05:48 +0800 Subject: [PATCH] fix: correct the parent class of CacheEmbedding (#2578) --- api/core/embedding/cached_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/core/embedding/cached_embedding.py b/api/core/embedding/cached_embedding.py index a86afd817a..7498a07559 100644 --- a/api/core/embedding/cached_embedding.py +++ b/api/core/embedding/cached_embedding.py @@ -3,12 +3,12 @@ import logging from typing import Optional, cast import numpy as np -from langchain.embeddings.base import Embeddings from sqlalchemy.exc import IntegrityError from core.model_manager import ModelInstance from core.model_runtime.entities.model_entities import ModelPropertyKey from core.model_runtime.model_providers.__base.text_embedding_model import TextEmbeddingModel +from core.rag.datasource.entity.embedding import Embeddings from extensions.ext_database import db from extensions.ext_redis import redis_client from libs import helper