Merge and rerank results from the Python pipeline path.
(self, ctx: QueryContext, docs_list: list[DocList])
| 198 | return [self._execute_single_query(query, collection) for query in vectors] |
| 199 | |
| 200 | def _merge_and_rerank(self, ctx: QueryContext, docs_list: list[DocList]) -> DocList: |
| 201 | """Merge and rerank results from the Python pipeline path.""" |
| 202 | if not docs_list: |
| 203 | raise ValueError("Query results is empty") |
| 204 | if len(docs_list) == 1 and not ctx.reranker: |
| 205 | return docs_list[0] |
| 206 | return ctx.reranker.rerank(docs_list, ctx.topk) |
| 207 | |
| 208 | def _build_base_search_query(self, ctx: QueryContext) -> _SearchQuery: |
| 209 | search_query = _SearchQuery() |