Spaces:
Sleeping
Sleeping
Upload database.py
Browse files- 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":
|
| 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 |
|