Исправление логики загрузки: всегда проверять .npy файлы первыми
This commit is contained in:
@@ -66,14 +66,8 @@ class VectorStore:
|
||||
self._lock = threading.Lock()
|
||||
|
||||
# Пытаемся загрузить сохраненные векторы
|
||||
# Проверяем наличие storage_path или отдельных .npy файлов
|
||||
# Всегда вызываем _load_from_disk если есть storage_path - он сам решит что загружать
|
||||
if storage_path:
|
||||
storage_dir = Path(storage_path).parent
|
||||
positive_npy = storage_dir / "positive_embeddings.npy"
|
||||
negative_npy = storage_dir / "negative_embeddings.npy"
|
||||
|
||||
# Загружаем если есть .npz файл или отдельные .npy файлы
|
||||
if os.path.exists(storage_path) or positive_npy.exists() or negative_npy.exists():
|
||||
self._load_from_disk()
|
||||
|
||||
@property
|
||||
@@ -393,6 +387,9 @@ class VectorStore:
|
||||
positive_npy = storage_dir / "positive_embeddings.npy"
|
||||
negative_npy = storage_dir / "negative_embeddings.npy"
|
||||
|
||||
# Отладочное логирование
|
||||
logger.info(f"VectorStore: Проверка путей - storage_dir={storage_dir}, positive_npy={positive_npy}, exists={positive_npy.exists()}, negative_npy={negative_npy}, exists={negative_npy.exists()}")
|
||||
|
||||
# Проверяем наличие отдельных .npy файлов
|
||||
if positive_npy.exists() or negative_npy.exists():
|
||||
logger.info("VectorStore: Обнаружены отдельные .npy файлы, загружаем их...")
|
||||
|
||||
Reference in New Issue
Block a user