Преглед на файлове

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

Bowen Liang преди 1 година
родител
ревизия
29ab244de6
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  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