search-index-reload
This commit is contained in:
@@ -1,5 +1,5 @@
|
|||||||
{
|
{
|
||||||
"$schema": "https://biomejs.dev/schemas/2.2.0/schema.json",
|
"$schema": "https://biomejs.dev/schemas/2.2.3/schema.json",
|
||||||
"files": {
|
"files": {
|
||||||
"includes": [
|
"includes": [
|
||||||
"**/*.tsx",
|
"**/*.tsx",
|
||||||
|
|||||||
@@ -44,6 +44,14 @@ def get_models_cache_dir() -> str:
|
|||||||
|
|
||||||
|
|
||||||
MODELS_CACHE_DIR = get_models_cache_dir()
|
MODELS_CACHE_DIR = get_models_cache_dir()
|
||||||
|
|
||||||
|
|
||||||
|
def get_index_dump_dir() -> str:
|
||||||
|
"""Определяет лучшую папку для индекса векторного поиска"""
|
||||||
|
# Приоритет /dump если доступна, иначе ./dump как fallback
|
||||||
|
return "/dump" if Path("/dump").exists() else "./dump"
|
||||||
|
|
||||||
|
|
||||||
# Используем HF_HOME вместо устаревшего TRANSFORMERS_CACHE
|
# Используем HF_HOME вместо устаревшего TRANSFORMERS_CACHE
|
||||||
os.environ.setdefault("HF_HOME", MODELS_CACHE_DIR)
|
os.environ.setdefault("HF_HOME", MODELS_CACHE_DIR)
|
||||||
|
|
||||||
@@ -213,8 +221,11 @@ class MuveraWrapper:
|
|||||||
try:
|
try:
|
||||||
logger.info("🔍 Пытаемся восстановить векторный индекс из файла...")
|
logger.info("🔍 Пытаемся восстановить векторный индекс из файла...")
|
||||||
|
|
||||||
|
# Определяем лучшую папку для индекса (приоритет /dump)
|
||||||
|
dump_dir = get_index_dump_dir()
|
||||||
|
|
||||||
# Пытаемся загрузить из файла
|
# Пытаемся загрузить из файла
|
||||||
if await self.load_index_from_file():
|
if await self.load_index_from_file(dump_dir):
|
||||||
logger.info("✅ Векторный индекс восстановлен из файла")
|
logger.info("✅ Векторный индекс восстановлен из файла")
|
||||||
else:
|
else:
|
||||||
logger.info("🔍 Сохраненный индекс не найден, будет создан новый")
|
logger.info("🔍 Сохраненный индекс не найден, будет создан новый")
|
||||||
@@ -380,7 +391,9 @@ class MuveraWrapper:
|
|||||||
# 🗃️ Автосохранение индекса после успешной индексации
|
# 🗃️ Автосохранение индекса после успешной индексации
|
||||||
if indexed_count > 0:
|
if indexed_count > 0:
|
||||||
try:
|
try:
|
||||||
await self.save_index_to_file()
|
# Используем тот же путь что и для загрузки
|
||||||
|
dump_dir = get_index_dump_dir()
|
||||||
|
await self.save_index_to_file(dump_dir)
|
||||||
if not silent:
|
if not silent:
|
||||||
logger.debug("💾 Индекс автоматически сохранен в файл")
|
logger.debug("💾 Индекс автоматически сохранен в файл")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|||||||
2
uv.lock
generated
2
uv.lock
generated
@@ -400,7 +400,7 @@ wheels = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "discours-core"
|
name = "discours-core"
|
||||||
version = "0.9.18"
|
version = "0.9.20"
|
||||||
source = { editable = "." }
|
source = { editable = "." }
|
||||||
dependencies = [
|
dependencies = [
|
||||||
{ name = "ariadne" },
|
{ name = "ariadne" },
|
||||||
|
|||||||
Reference in New Issue
Block a user