Rulga commited on
Commit
7d17621
·
1 Parent(s): f4e115f

Add default value checks for knowledge base version and size in status response

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -60,7 +60,10 @@ def check_kb_status():
60
  data = response.json()
61
  if data["knowledge_base_exists"]:
62
  kb_info = data["kb_info"]
63
- return f"✅ База знаний готова к работе\nВерсия: {kb_info['version']}\nРазмер: {kb_info['size']:.2f} MB"
 
 
 
64
  else:
65
  return "❌ База знаний не создана. Нажмите кнопку 'Create/Update Knowledge Base'"
66
  except Exception as e:
 
60
  data = response.json()
61
  if data["knowledge_base_exists"]:
62
  kb_info = data["kb_info"]
63
+ # Добавляем проверки на None и значения по умолчанию
64
+ version = kb_info.get('version', 'N/A')
65
+ size = kb_info.get('size', 0)
66
+ return f"✅ База знаний готова к работе\nВерсия: {version}\nРазмер: {size:.2f if size else 0} MB"
67
  else:
68
  return "❌ База знаний не создана. Нажмите кнопку 'Create/Update Knowledge Base'"
69
  except Exception as e: