код на python для RAG – ответ по документу

нужен код на python для RAG – ответ по документу
предполагаемый преимущественный язык – английский
документы – pdf, docx
источники загрузки – директория на сервере, ссылка в интернет, загрузка Colab|Jupyter

должно быть 2 явных этапа, чтобы было возможно вывести промежуточные результаты
1. выбор фрагментов
2. генерация ответа

я хочу отдельно смотреть, какие берутся фрагменты и что потом из этого получается
совсем хорошо, если будет отдельный этап реранкер