Wplotnikow commited on
Commit
0d6bcb5
·
verified ·
1 Parent(s): 2160154

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -88,12 +88,11 @@ def ask_chatbot(question):
88
  if n_blocks == 0:
89
  return "Ошибка: база знаний отсутствует или пуста."
90
  # Корректная обработка индексов!
91
- sorted_idxs = sims.argsort()
92
- top_idxs = list(map(int, sorted_idxs[-n_blocks:][::-1]))
93
  context_blocks = []
94
- for rank, idx in enumerate(top_idxs):
95
- idx = int(idx)
96
- if 0 <= idx < len(blocks): # строгое попадание в диапазон
97
  context_blocks.append(blocks[idx])
98
  context = " ".join(context_blocks)
99
  # Ответ только из абзацев, не заголовков!
 
88
  if n_blocks == 0:
89
  return "Ошибка: база знаний отсутствует или пуста."
90
  # Корректная обработка индексов!
91
+ sorted_idxs = sims.argsort()[-n_blocks:][::-1]
92
+ sorted_idxs = [int(i) for i in sorted_idxs] # ЯВНО! Только числа!
93
  context_blocks = []
94
+ for rank, idx in enumerate(sorted_idxs):
95
+ if isinstance(idx, (int, float)) and 0 <= idx < len(blocks): # безопасно!
 
96
  context_blocks.append(blocks[idx])
97
  context = " ".join(context_blocks)
98
  # Ответ только из абзацев, не заголовков!