k

app.py

@@ -125,7 +125,7 @@ def answer(system: str, context: str, question: str, user_id="demo", history="No
             sims   = torch.matmul(store["vecs"], q_vec)          # [N]
             if sims.numel() == 0:
                 return "Knowledge base is empty or corrupted."
-            k      = min(4, sims.numel())
+            k      = min(4, store["vecs"].size(0))
             idxs   = torch.topk(sims, k=k).indices.tolist()
             context_list += [store["texts"][i] for i in idxs]
         elif history == "All":
@@ -160,7 +160,7 @@ def answer(system: str, context: str, question: str, user_id="demo", history="No
 
           
     except Exception as e:
-        return f"Error in app.py: {e}, k={k}, q_vec.shape={q_vec.shape if 'q_vec' in locals() else 'N/A'}"
+        return f"Error in app.py: {e}, k={k}, sims.numel()={sims.numel()}, q_vec.shape={q_vec.shape if 'q_vec' in locals() else 'N/A'}"
     finally:
         torch.cuda.empty_cache()