kerrad/rag-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Исправление логики загрузки: всегда проверять .npy файлы первыми

Browse Source
This commit is contained in:
Andrey
2026-01-28 20:28:21 +03:00
parent beca61bd90
commit b8915dccc2
1 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
app/storage/vector_store.py
View File

@@ -66,15 +66,9 @@ class VectorStore:
self._lock = threading.Lock() self._lock = threading.Lock()
# Пытаемся загрузить сохраненные векторы # Пытаемся загрузить сохраненные векторы
# Проверяем наличие storage_path или отдельных .npy файлов # Всегда вызываем _load_from_disk если есть storage_path - он сам решит что загружать
if storage_path: if storage_path:
storage_dir = Path(storage_path).parent self._load_from_disk()
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 @property
def positive_count(self) -> int: def positive_count(self) -> int:
@@ -393,6 +387,9 @@ class VectorStore:
positive_npy = storage_dir / "positive_embeddings.npy" positive_npy = storage_dir / "positive_embeddings.npy"
negative_npy = storage_dir / "negative_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 файлов # Проверяем наличие отдельных .npy файлов
if positive_npy.exists() or negative_npy.exists(): if positive_npy.exists() or negative_npy.exists():
logger.info("VectorStore: Обнаружены отдельные .npy файлы, загружаем их...") logger.info("VectorStore: Обнаружены отдельные .npy файлы, загружаем их...")