Spaces:
Sleeping
Sleeping
Added llm_selection to payload so it doesnt generat llm explanations when not selected
Browse files- src/api/main.py +5 -4
src/api/main.py
CHANGED
|
@@ -309,7 +309,7 @@ def search(req: SearchRequest):
|
|
| 309 |
# Route to BM25 if selected
|
| 310 |
if req.retrieval.model == "bm25":
|
| 311 |
results = bm25_search(req)
|
| 312 |
-
explanations = generate_explanations(req, results)
|
| 313 |
|
| 314 |
return SearchResponse(
|
| 315 |
query = req.query,
|
|
@@ -318,12 +318,13 @@ def search(req: SearchRequest):
|
|
| 318 |
"model_used": "bm25",
|
| 319 |
},
|
| 320 |
results = results,
|
| 321 |
-
explanations = explanations
|
| 322 |
)
|
|
|
|
| 323 |
# Route to FAISS if selected
|
| 324 |
if req.retrieval.model == "faiss":
|
| 325 |
results = faiss_search(req)
|
| 326 |
-
explanations = generate_explanations(req, results)
|
| 327 |
|
| 328 |
return SearchResponse(
|
| 329 |
query = req.query,
|
|
@@ -332,6 +333,6 @@ def search(req: SearchRequest):
|
|
| 332 |
"model_used": "faiss",
|
| 333 |
},
|
| 334 |
results = results,
|
| 335 |
-
explanations = explanations
|
| 336 |
)
|
| 337 |
|
|
|
|
| 309 |
# Route to BM25 if selected
|
| 310 |
if req.retrieval.model == "bm25":
|
| 311 |
results = bm25_search(req)
|
| 312 |
+
explanations = generate_explanations(req, results) if req.llm_explanations else []
|
| 313 |
|
| 314 |
return SearchResponse(
|
| 315 |
query = req.query,
|
|
|
|
| 318 |
"model_used": "bm25",
|
| 319 |
},
|
| 320 |
results = results,
|
| 321 |
+
explanations = explanations
|
| 322 |
)
|
| 323 |
+
|
| 324 |
# Route to FAISS if selected
|
| 325 |
if req.retrieval.model == "faiss":
|
| 326 |
results = faiss_search(req)
|
| 327 |
+
explanations = generate_explanations(req, results) if req.llm_explanations else []
|
| 328 |
|
| 329 |
return SearchResponse(
|
| 330 |
query = req.query,
|
|
|
|
| 333 |
"model_used": "faiss",
|
| 334 |
},
|
| 335 |
results = results,
|
| 336 |
+
explanations = explanations
|
| 337 |
)
|
| 338 |
|