zenml-io / zenml-projects

A repository for all ZenML projects that are specific production use-cases.
Apache License 2.0
210 stars 52 forks source link

Add reranking functionality to retrieval process #101

Closed strickvl closed 5 months ago

strickvl commented 5 months ago

This pull request adds reranking functionality to the retrieval process in order to improve the relevance of the retrieved documents. It includes the following commits:

The changes in the code include modifications to the process_input_with_retrieval function, which now includes a rerank_documents function to rerank the retrieved documents based on the input query. The reranked documents are then used in the completion process.

dagshub[bot] commented 5 months ago

Join the discussion on DagsHub!