diff --git a/src/vectorcode/subcommands/query/reranker.py b/src/vectorcode/subcommands/query/reranker.py index 335ff3f2..a9713809 100644 --- a/src/vectorcode/subcommands/query/reranker.py +++ b/src/vectorcode/subcommands/query/reranker.py @@ -71,7 +71,9 @@ def rerank(self, results: QueryResult) -> list[str]: chunk_metas = results["metadatas"][query_chunk_idx] chunk_docs = results["documents"][query_chunk_idx] ranks = self.model.rank( - self.query_chunks[query_chunk_idx], chunk_docs, apply_softmax=True + str(self.query_chunks[query_chunk_idx]), + list(str(i) for i in chunk_docs), + apply_softmax=True, ) for rank in ranks: if QueryInclude.chunk in self.configs.include: