Upload database.py

database.py

@@ -215,8 +215,7 @@ def fetch_query_results(query: str, k_model: int = 10,
         df["distance"] = df["faiss_id"].map(distance_map)
 
         if use_rerank:
-            print("use_rerank = True", flush=True)
-            logging.debug("use_rerank = True")
+            status_dbg = "ok_rerank"
             # Cross-encoder
             df["chunk_text"] = df["chunk_text"].str.replace(r'\s+', ' ', regex=True).str.strip()
             top_passages = df["chunk_text"].tolist()
@@ -230,8 +229,7 @@ def fetch_query_results(query: str, k_model: int = 10,
             # Garder top k_cross
             df_top = df.head(k_cross)
         else:
-            print("use_rerank = False", flush=True)
-            logging.debug("use_rerank = False")
+            status_dbg = "ok_no_rerank"
             df = df.sort_values(by="distance", ascending=False)
             df["cross_score"] = df["distance"]
             # Garder top k_model
@@ -268,7 +266,7 @@ def fetch_query_results(query: str, k_model: int = 10,
         # Liste finale de dictionnaires
         list_result = duck_res.to_dict(orient="records")
 
-        return {"status": "ok", "result": list_result}
+        return {"status": status_dbg, "result": list_result}
     except Exception as e:
         return {"status": "error", "code": type(e).__name__, "message": str(e)}