Loren commited on
Commit
e1182e0
·
verified ·
1 Parent(s): d5c8f86

Upload database.py

Browse files
Files changed (1) hide show
  1. app/database.py +1 -3
app/database.py CHANGED
@@ -215,7 +215,6 @@ def fetch_query_results(query: str, k_model: int = 10,
215
  df["distance"] = df["faiss_id"].map(distance_map)
216
 
217
  if use_rerank:
218
- status_dbg = "ok_rerank"
219
  # Cross-encoder
220
  df["chunk_text"] = df["chunk_text"].str.replace(r'\s+', ' ', regex=True).str.strip()
221
  top_passages = df["chunk_text"].tolist()
@@ -229,7 +228,6 @@ def fetch_query_results(query: str, k_model: int = 10,
229
  # Garder top k_cross
230
  df_top = df.head(k_cross)
231
  else:
232
- status_dbg = "ok_no_rerank"
233
  df = df.sort_values(by="distance", ascending=False)
234
  df["cross_score"] = df["distance"]
235
  # Garder top k_model
@@ -266,7 +264,7 @@ def fetch_query_results(query: str, k_model: int = 10,
266
  # Liste finale de dictionnaires
267
  list_result = duck_res.to_dict(orient="records")
268
 
269
- return {"status": status_dbg, "result": list_result}
270
  except Exception as e:
271
  return {"status": "error", "code": type(e).__name__, "message": str(e)}
272
 
 
215
  df["distance"] = df["faiss_id"].map(distance_map)
216
 
217
  if use_rerank:
 
218
  # Cross-encoder
219
  df["chunk_text"] = df["chunk_text"].str.replace(r'\s+', ' ', regex=True).str.strip()
220
  top_passages = df["chunk_text"].tolist()
 
228
  # Garder top k_cross
229
  df_top = df.head(k_cross)
230
  else:
 
231
  df = df.sort_values(by="distance", ascending=False)
232
  df["cross_score"] = df["distance"]
233
  # Garder top k_model
 
264
  # Liste finale de dictionnaires
265
  list_result = duck_res.to_dict(orient="records")
266
 
267
+ return {"status": "ok", "result": list_result}
268
  except Exception as e:
269
  return {"status": "error", "code": type(e).__name__, "message": str(e)}
270