feat(admin): шкала загрузки источников, отладка и загрузка работ в корпус
Заливка корпуса была чёрным ящиком: у источника только last_status (idle/running/done/error), без «сколько из скольки», без причины падения и без способа остановить начатое. Теперь каждый запуск создаёт строку parse_runs, куда воркер раз в ~2с пишет стадию, счётчики и журнал событий. Админка: - шкала загрузки у каждого источника (0→50% выборка, 50→100% индексация), раскрытая строка — журнал прогона по шагам с таймингами; - «Запустить всё» / «Остановить всё» и остановка по одному источнику (кооперативная отмена: воркер останавливается сам, не рвя запись в базу); - пакетное добавление источников (тип + список тем), тип pmc в форме; - загрузка PDF/DOCX/TXT прямо в базу сравнения (index.ingest_upload); - страница «Отладка»: воркеры Celery и их текущие таски, очереди RabbitMQ, покрытие корпуса эмбеддингами, зависшие и упавшие прогоны, конфиг бэкендов. Защита от краш-лупа по consumer_timeout RabbitMQ (docs/DR-HA.md §6), без неё массовый запуск 170+ источников гарантированно ронял воркер: - PARSER_TIME_BUDGET_S (1500с) — прогон закругляется сам и помечается partial; - worker_prefetch_multiplier=1 — таймаут считается от ДОСТАВКИ сообщения, и с дефолтным префетчем очередь долгих run_parser убивала канал на задачах, которые ещё не начинались. Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
@@ -3,7 +3,9 @@
|
||||
from datetime import datetime
|
||||
from typing import Any
|
||||
|
||||
from pydantic import BaseModel, Field
|
||||
from pydantic import BaseModel, Field, computed_field
|
||||
|
||||
from app.core.progress import run_percent
|
||||
|
||||
|
||||
# ─── Сессия доступа ───────────────────────────────────────────────────────────
|
||||
@@ -61,9 +63,42 @@ class AdminDocumentResponse(BaseModel):
|
||||
model_config = {"from_attributes": True}
|
||||
|
||||
|
||||
# ─── Прогоны парсинга ─────────────────────────────────────────────────────────
|
||||
class ParseRunResponse(BaseModel):
|
||||
"""Состояние прогона для шкалы загрузки (без журнала — он в ParseRunDetail)."""
|
||||
|
||||
id: int
|
||||
source_id: int
|
||||
status: str
|
||||
stage: str
|
||||
target: int
|
||||
fetched: int
|
||||
processed: int
|
||||
added: int
|
||||
duplicates: int
|
||||
skipped: int
|
||||
failed: int
|
||||
cancel_requested: bool = False
|
||||
error: str | None = None
|
||||
started_at: datetime
|
||||
heartbeat_at: datetime | None = None
|
||||
finished_at: datetime | None = None
|
||||
|
||||
model_config = {"from_attributes": True}
|
||||
|
||||
@computed_field # type: ignore[prop-decorator]
|
||||
@property
|
||||
def percent(self) -> float:
|
||||
return run_percent(self.status, self.stage, self.target, self.fetched, self.processed)
|
||||
|
||||
|
||||
class ParseRunDetail(ParseRunResponse):
|
||||
log: list[dict[str, Any]] = Field(default_factory=list)
|
||||
|
||||
|
||||
# ─── Источники парсинга ───────────────────────────────────────────────────────
|
||||
class ParseSourceCreate(BaseModel):
|
||||
source_type: str = Field(description="openalex / cyberleninka / arxiv")
|
||||
source_type: str = Field(description="openalex / cyberleninka / arxiv / pmc")
|
||||
name: str = Field(min_length=1, max_length=255)
|
||||
query: str | None = None
|
||||
lang: str | None = None
|
||||
@@ -98,10 +133,25 @@ class ParseSourceResponse(BaseModel):
|
||||
last_run_at: datetime | None = None
|
||||
docs_added: int
|
||||
created_at: datetime
|
||||
# Последний (или текущий) прогон — источник данных для шкалы в списке
|
||||
last_run: ParseRunResponse | None = None
|
||||
|
||||
model_config = {"from_attributes": True}
|
||||
|
||||
|
||||
class ParseSourceBulkCreate(BaseModel):
|
||||
"""Пакетное добавление: один тип/лимит, много запросов (по строке на тему)."""
|
||||
|
||||
source_type: str = Field(description="openalex / cyberleninka / arxiv / pmc")
|
||||
queries: list[str] = Field(min_length=1, max_length=500)
|
||||
name_prefix: str = ""
|
||||
lang: str | None = None
|
||||
year_from: int | None = None
|
||||
year_to: int | None = None
|
||||
limit: int = Field(default=1000, ge=1, le=100000)
|
||||
run_now: bool = False
|
||||
|
||||
|
||||
# ─── Отстойник ────────────────────────────────────────────────────────────────
|
||||
class StagedWorkResponse(BaseModel):
|
||||
id: int
|
||||
|
||||
Reference in New Issue
Block a user