dcorcoran commited on
Commit
3b88f8b
·
1 Parent(s): df10154

Added llm_selection to payload so it doesnt generat llm explanations when not selected

Browse files
Files changed (1) hide show
  1. 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