bug: chunk header wasn't being properly passed in to reranker

D-Star-AI / dsRAG

High-performance retrieval engine for unstructured data

MIT License

852 stars 61 forks source link

bug: chunk header wasn't being properly passed in to reranker #22

Closed zmccormick7 closed 3 months ago

zmccormick7 commented 3 months ago

This PR fixes a bug in rerank_search_results (in both the CohereReranker and VoyageReranker classes) that caused the chunk header to not get included. It also fixes a small issue where a warning message was displayed when it shouldn't be in add_document.