Explorar o código

fix: correct the parent class of CacheEmbedding (#2578)

Bowen Liang hai 1 ano
pai
achega
29ab244de6
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      api/core/embedding/cached_embedding.py

+ 1 - 1
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