kerrad/telegram-helper-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: улучшена проверка данных из RAG API в методе получения статистики

Browse Source 
- Упрощена логика проверки наличия данных из API, убраны лишние переменные.
- Обновлен расчет общего количества примеров для корректного отображения статистики.
This commit is contained in:
Andrey
2026-01-28 01:02:21 +03:00
parent a949f7e7db
commit 35767c289c
1 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
helper_bot/handlers/admin/admin_handlers.py
View File

@@ -171,12 +171,9 @@ async def get_ml_stats(
rag = stats["rag"] rag = stats["rag"]
lines.append("🤖 <b>RAG API:</b>") lines.append("🤖 <b>RAG API:</b>")
# Проверяем, есть ли данные из API статуса (по наличию model_loaded или vector_store) # Проверяем, есть ли данные из API (новый контракт содержит model_loaded и vector_store)
has_api_data = "model_loaded" in rag or "vector_store" in rag if "model_loaded" in rag or "vector_store" in rag:
# Данные из API /stats
if has_api_data:
# Данные из API статуса
# Модель из API
if "model_loaded" in rag: if "model_loaded" in rag:
model_loaded = rag.get('model_loaded', False) model_loaded = rag.get('model_loaded', False)
lines.append(f" • Модель загружена: {'✅' if model_loaded else '❌'}") lines.append(f" • Модель загружена: {'✅' if model_loaded else '❌'}")
@@ -190,7 +187,7 @@ async def get_ml_stats(
vector_store = rag["vector_store"] vector_store = rag["vector_store"]
positive_count = vector_store.get("positive_count", 0) positive_count = vector_store.get("positive_count", 0)
negative_count = vector_store.get("negative_count", 0) negative_count = vector_store.get("negative_count", 0)
total_count = vector_store.get("total_count", positive_count + negative_count) total_count = vector_store.get("total_count", 0)
lines.append(f" • Положительных примеров: {positive_count}") lines.append(f" • Положительных примеров: {positive_count}")
lines.append(f" • Отрицательных примеров: {negative_count}") lines.append(f" • Отрицательных примеров: {negative_count}")