├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.yml │ ├── config.yml │ ├── documentation_improvement.yml │ ├── feature_request.yml │ ├── integration_request.yml │ └── question.yml ├── dependabot.yml ├── pull_request_template.md └── workflows │ ├── docs.yml │ ├── lint.yml │ ├── release.yml │ └── unit_test.yml ├── .gitignore ├── .gitmodules ├── .pre-commit-config.yaml ├── .python-versions ├── CHANGELOG.md ├── CITATION.cff ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── Makefile ├── README.md ├── codecov.yml ├── docs ├── api_reference │ ├── bridges │ │ ├── index.md │ │ ├── langchain.md │ │ └── llamaindex.md │ ├── data_collators │ │ ├── huggingface.md │ │ └── index.md │ ├── data_structures │ │ ├── bridge.md │ │ ├── evals.md │ │ ├── knowledge_node.md │ │ ├── rag.md │ │ └── results.md │ ├── decorators │ │ └── index.md │ ├── evals │ │ ├── benchmarker.md │ │ ├── benchmarks │ │ │ └── huggingface │ │ │ │ ├── boolq.md │ │ │ │ ├── hellaswag.md │ │ │ │ ├── hotpotqa.md │ │ │ │ ├── mmlu.md │ │ │ │ ├── natural_questions.md │ │ │ │ ├── pubmedqa.md │ │ │ │ └── squad_v2.md │ │ ├── index.md │ │ └── metrics │ │ │ └── exact_match.md │ ├── exceptions │ │ ├── bridge.md │ │ ├── data_collator.md │ │ ├── evals.md │ │ ├── fl_tasks.md │ │ ├── generator.md │ │ ├── index.md │ │ ├── inspectors.md │ │ ├── knowledge_stores.md │ │ ├── loss.md │ │ ├── rag_trainer.md │ │ ├── retriever.md │ │ ├── tokenizer.md │ │ └── trainer.md │ ├── finetuning_datasets │ │ ├── huggingface.md │ │ ├── index.md │ │ └── pytorch.md │ ├── fl_tasks │ │ ├── huggingface.md │ │ ├── index.md │ │ └── pytorch.md │ ├── generators │ │ ├── huggingface.md │ │ ├── index.md │ │ └── unsloth.md │ ├── index.md │ ├── inspectors │ │ ├── huggingface.md │ │ ├── index.md │ │ └── pytorch.md │ ├── knowledge_nodes │ │ └── index.md │ ├── knowledge_stores │ │ ├── in_memory.md │ │ ├── index.md │ │ ├── mixins.md │ │ └── qdrant.md │ ├── loss │ │ └── pytorch.md │ ├── rag_system │ │ └── index.md │ ├── retrievers │ │ ├── huggingface.md │ │ └── index.md │ ├── tokenizers │ │ ├── huggingface.md │ │ └── index.md │ ├── trainer_managers │ │ ├── huggingface.md │ │ ├── index.md │ │ └── pytorch.md │ └── trainers │ │ ├── huggingface.md │ │ ├── index.md │ │ └── pytorch.md ├── assets │ ├── favicon-48x48.svg │ ├── favicon.ico │ ├── logo-dark.svg │ ├── logo.svg │ └── vector-logo.svg ├── community │ ├── changelog.md │ ├── contributing │ │ ├── ask_question.md │ │ ├── index.md │ │ ├── submit_issue.md │ │ └── submit_pr.md │ ├── index.md │ └── resources │ │ ├── index.md │ │ └── pocket_references.md ├── examples │ ├── index.md │ └── ra_dit │ │ ├── benchmarking.md │ │ ├── federated_finetune.md │ │ ├── finetune.md │ │ ├── index.md │ │ └── qdrant_knowledge_store_wikipedia.md ├── getting_started │ ├── advanced_usage │ │ └── lower_level_ralt.md │ ├── essentials.md │ ├── import_patterns.md │ ├── installation.md │ ├── integrations.md │ ├── quick_starts │ │ ├── benchmark_mmlu.md │ │ ├── federated.md │ │ ├── index.md │ │ ├── rag_finetuning.md │ │ └── rag_inference.md │ ├── standard_usage.md │ └── tutorials │ │ ├── index.md │ │ ├── lsr.md │ │ └── ralt.md ├── index.md ├── javascripts │ └── mathjax.js ├── notebooks │ ├── basic_fl.ipynb │ ├── basic_starter_hf.ipynb │ ├── integrations │ │ ├── langchain.ipynb │ │ ├── llama_index.ipynb │ │ ├── qdrant.ipynb │ │ ├── qdrant_async.ipynb │ │ └── unsloth.ipynb │ ├── no_encode_rag_with_mcp.ipynb │ └── rag_benchmarking_hf_mmlu.ipynb ├── overrides │ └── partials │ │ ├── copyright.html │ │ └── logo.html └── stylesheets │ └── extra.css ├── example_scripts ├── README.md └── cookbook_script-basic_fl.py ├── examples ├── knowledge_stores │ └── ra-dit-ks │ │ ├── README.md │ │ ├── docker │ │ ├── Dockerfile │ │ ├── README.md │ │ ├── healthcheck.sh │ │ └── startup.sh │ │ ├── pyproject.toml │ │ ├── src │ │ └── ra_dit_ks │ │ │ ├── __init__.py │ │ │ ├── logger.py │ │ │ └── main.py │ │ └── uv.lock ├── quick-start │ ├── README.md │ ├── pyproject.toml │ └── quick_start │ │ ├── __init__.py │ │ ├── _cifar_dataloaders.py │ │ └── main.py └── ra-dit │ ├── README.md │ ├── pyproject.toml │ ├── ra_dit │ ├── __init__.py │ ├── _dataset_prep │ │ ├── __init__.py │ │ ├── base_data_prepper.py │ │ └── qa │ │ │ ├── __init__.py │ │ │ ├── commonsense.py │ │ │ ├── math.py │ │ │ ├── mixin.py │ │ │ ├── pubmed.py │ │ │ ├── web_questions.py │ │ │ └── wiki.py │ ├── benchmarker.py │ ├── centralized.py │ ├── datasets.py │ ├── evaluation_benchmarks │ │ ├── __init__.py │ │ ├── base.py │ │ └── mmlu.py │ ├── federated_finetune.py │ ├── finetune.py │ ├── generators │ │ ├── __init__.py │ │ ├── llama2_7b.py │ │ └── utils.py │ ├── knowledge_stores │ │ ├── __init__.py │ │ ├── from_dragon.py │ │ └── utils.py │ ├── logger.py │ ├── main.py │ ├── rag_system.py │ ├── retrievers │ │ ├── __init__.py │ │ └── dragon.py │ ├── trainers_and_testers │ │ ├── __init__.py │ │ ├── generator.py │ │ └── retriever.py │ └── utils.py │ └── uv.lock ├── mkdocs.yml ├── mypy.ini ├── pyproject.toml ├── pytest.ini ├── src └── fed_rag │ ├── __init__.py │ ├── __init__.pyi │ ├── _bridges │ ├── __init__.py │ ├── langchain │ │ ├── __init__.py │ │ ├── _bridge_classes.py │ │ ├── _version.py │ │ └── bridge.py │ └── llamaindex │ │ ├── __init__.py │ │ ├── _managed_index.py │ │ ├── _version.py │ │ └── bridge.py │ ├── _version.py │ ├── base │ ├── __init__.py │ ├── bridge.py │ ├── data_collator.py │ ├── evals │ │ ├── __init__.py │ │ ├── benchmark.py │ │ └── metric.py │ ├── fl_task.py │ ├── generator.py │ ├── generator_mixins │ │ ├── __init__.py │ │ ├── audio.py │ │ ├── image.py │ │ └── video.py │ ├── knowledge_store.py │ ├── no_encode_knowledge_store.py │ ├── retriever.py │ ├── retriever_mixins │ │ ├── __init__.py │ │ ├── audio.py │ │ ├── image.py │ │ └── video.py │ ├── tokenizer.py │ ├── trainer.py │ ├── trainer_config.py │ └── trainer_manager.py │ ├── core │ ├── __init__.py │ ├── no_encode_rag_system │ │ ├── __init__.py │ │ ├── _asynchronous.py │ │ ├── _synchronous.py │ │ ├── asynchronous.py │ │ └── synchronous.py │ └── rag_system │ │ ├── __init__.py │ │ ├── _asynchronous.py │ │ ├── _synchronous.py │ │ ├── asynchronous.py │ │ └── synchronous.py │ ├── data_collators │ ├── __init__.py │ └── huggingface │ │ ├── __init__.py │ │ ├── lsr.py │ │ └── ralt.py │ ├── data_structures │ ├── __init__.py │ ├── bridge.py │ ├── evals.py │ ├── knowledge_node.py │ ├── rag.py │ ├── results.py │ └── retriever.py │ ├── decorators │ ├── __init__.py │ ├── tester.py │ └── trainer.py │ ├── evals │ ├── __init__.py │ ├── benchmarker.py │ ├── benchmarks │ │ ├── __init__.py │ │ └── huggingface │ │ │ ├── __init__.py │ │ │ ├── boolq.py │ │ │ ├── hellaswag.py │ │ │ ├── hotpotqa.py │ │ │ ├── mixin.py │ │ │ ├── mmlu.py │ │ │ ├── natural_questions.py │ │ │ ├── pubmedqa.py │ │ │ ├── squad_v2.py │ │ │ └── utils.py │ ├── metrics │ │ ├── __init__.py │ │ └── exact_match.py │ └── utils.py │ ├── exceptions │ ├── __init__.py │ ├── bridge.py │ ├── common.py │ ├── core.py │ ├── data_collator.py │ ├── evals.py │ ├── fl_tasks.py │ ├── generator.py │ ├── inspectors.py │ ├── knowledge_stores.py │ ├── loss.py │ ├── rag_system.py │ ├── retriever.py │ ├── tokenizer.py │ ├── trainer.py │ └── trainer_manager.py │ ├── fl_tasks │ ├── __init__.py │ ├── huggingface.py │ └── pytorch.py │ ├── generators │ ├── __init__.py │ ├── huggingface │ │ ├── __init__.py │ │ ├── hf_multimodal_model.py │ │ ├── hf_peft_model.py │ │ ├── hf_pretrained_model.py │ │ ├── mixin.py │ │ └── utils.py │ └── unsloth │ │ ├── __init__.py │ │ ├── mixin.py │ │ ├── unsloth_fast_model.py │ │ ├── unsloth_fast_multimodal_model.py │ │ └── utils.py │ ├── inspectors │ ├── __init__.py │ ├── common.py │ ├── huggingface │ │ ├── __init__.py │ │ ├── tester.py │ │ ├── trainer.py │ │ └── utils.py │ └── pytorch │ │ ├── __init__.py │ │ ├── tester.py │ │ └── trainer.py │ ├── knowledge_stores │ ├── __init__.py │ ├── in_memory.py │ ├── mixins.py │ ├── no_encode │ │ ├── __init__.py │ │ └── mcp │ │ │ ├── __init__.py │ │ │ ├── sources │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── stdio.py │ │ │ ├── streamable_http.py │ │ │ └── utils.py │ │ │ └── store.py │ └── qdrant │ │ ├── __init__.py │ │ ├── asynchronous.py │ │ ├── sync.py │ │ └── utils.py │ ├── loss │ ├── __init__.py │ └── pytorch │ │ ├── __init__.py │ │ └── lsr.py │ ├── py.typed │ ├── retrievers │ ├── __init__.py │ └── huggingface │ │ ├── __init__.py │ │ └── hf_sentence_transformer.py │ ├── tokenizers │ ├── __init__.py │ ├── hf_pretrained_tokenizer.py │ └── unsloth_pretrained_tokenizer.py │ ├── trainer_configs │ ├── __init__.py │ └── pytorch.py │ ├── trainer_managers │ ├── __init__.py │ ├── huggingface.py │ └── pytorch.py │ ├── trainers │ ├── __init__.py │ ├── huggingface │ │ ├── __init__.py │ │ ├── lsr.py │ │ ├── mixin.py │ │ └── ralt.py │ └── pytorch │ │ ├── __init__.py │ │ ├── mixin.py │ │ └── training_args.py │ ├── types │ ├── __init__.py │ ├── bridge.py │ ├── knowledge_node.py │ ├── rag.py │ ├── rag_system.py │ └── results.py │ └── utils │ ├── __init__.py │ ├── asyncio.py │ ├── data │ ├── __init__.py │ ├── _functions.py │ └── finetuning_datasets │ │ ├── __init__.py │ │ ├── huggingface.py │ │ └── pytorch.py │ ├── huggingface.py │ └── notebook.py ├── tests ├── __init__.py ├── api │ ├── __init__.py │ ├── test_data_structures_imports.py │ ├── test_deprecated_types_imports.py │ ├── test_evals_imports.py │ ├── test_namespaced_imports.py │ └── test_root_imports.py ├── bridges │ ├── __init__.py │ ├── conftest.py │ ├── test_base.py │ ├── test_langchain.py │ └── test_llamaindex.py ├── conftest.py ├── data_collators │ ├── __init__.py │ ├── conftest.py │ ├── huggingface │ │ ├── __init__.py │ │ ├── conftest.py │ │ ├── test_data_collator_for_ralt.py │ │ └── test_hf_data_collators.py │ └── test_base.py ├── data_structures │ └── test_evals.py ├── decorators │ ├── __init__.py │ ├── test_huggingface.py │ └── test_pytorch.py ├── evals │ ├── __init__.py │ ├── benchmarks │ │ ├── __init__.py │ │ ├── _benchmarks.py │ │ ├── huggingface │ │ │ ├── __init__.py │ │ │ ├── conftest.py │ │ │ ├── test_boolq.py │ │ │ ├── test_hellaswag.py │ │ │ ├── test_hotpotqa.py │ │ │ ├── test_mixin.py │ │ │ ├── test_mmlu.py │ │ │ ├── test_natural_questions.py │ │ │ ├── test_pubmedqa.py │ │ │ ├── test_squad_v2.py │ │ │ └── test_utils.py │ │ ├── test_base.py │ │ ├── test_benchmarker.py │ │ └── test_utils.py │ └── metrics │ │ ├── __init__.py │ │ ├── test_base.py │ │ └── test_exact_match.py ├── fl_tasks │ ├── __init__.py │ ├── huggingface │ │ ├── __init__.py │ │ ├── conftest.py │ │ └── test_huggingface_fltask.py │ └── pytorch │ │ ├── __init__.py │ │ ├── conftest.py │ │ └── test_pytorch_fltask.py ├── generators │ ├── __init__.py │ ├── conftest.py │ ├── mixins │ │ ├── __init__.py │ │ ├── test_audio_mixin.py │ │ ├── test_image_mixin.py │ │ └── test_video_mixin.py │ ├── test_base.py │ ├── test_hf_multimodal_model.py │ ├── test_hf_peft.py │ ├── test_hf_pretrained.py │ ├── test_hf_utils.py │ ├── test_unsloth_fast_model.py │ ├── test_unsloth_fast_multimodal_model.py │ └── test_unsloth_utils.py ├── knowledge_stores │ ├── __init__.py │ ├── no_encode │ │ ├── __init__.py │ │ ├── mcp │ │ │ ├── __init__.py │ │ │ ├── test_mcp_source_stdio.py │ │ │ ├── test_mcp_source_streamable_http.py │ │ │ └── test_mcp_store.py │ │ └── test_base.py │ ├── test_base.py │ ├── test_in_memory.py │ ├── test_in_memory_managed.py │ ├── test_knowledge_node.py │ ├── test_qdrant.py │ └── test_qdrant_async.py ├── loss │ ├── __init__.py │ └── pytorch │ │ ├── __init__.py │ │ ├── conftest.py │ │ └── test_lsr.py ├── rag_system │ ├── __init__.py │ ├── conftest.py │ ├── test_async_no_encode_rag_system.py │ ├── test_async_rag_system.py │ ├── test_no_encode_rag_system.py │ ├── test_rag_system.py │ └── test_source_node.py ├── retrievers │ ├── __init__.py │ ├── conftest.py │ ├── mixins │ │ ├── __init__.py │ │ ├── test_audio_retriever_mixin.py │ │ ├── test_image_retriever_mixin.py │ │ └── test_video_retriever_mixin.py │ ├── test_base.py │ └── test_hf_sentence_transformer.py ├── tokenizers │ ├── __init__.py │ ├── conftest.py │ ├── test_base.py │ ├── test_hf_pretrained.py │ └── test_unsloth_pretrained.py ├── trainer_configs │ ├── __init__.py │ └── test_pytorch_trainer_config.py ├── trainer_managers │ ├── __init__.py │ ├── conftest.py │ ├── huggingface │ │ ├── __init__.py │ │ ├── conftest.py │ │ └── test_hf_trainer_manager.py │ ├── test_base.py │ └── test_pytorch_trainer_manager.py ├── trainers │ ├── __init__.py │ ├── conftest.py │ ├── huggingface │ │ ├── __init__.py │ │ ├── conftest.py │ │ ├── test_hf_lsr.py │ │ ├── test_hf_mixin.py │ │ └── test_hf_ralt.py │ ├── pytorch │ │ ├── __init__.py │ │ ├── conftest.py │ │ └── test_pt_mixin.py │ └── test_base.py └── utils │ ├── __init__.py │ ├── data │ ├── finetuning_datasets │ │ ├── __init__.py │ │ ├── conftest.py │ │ ├── test_hf_dataset.py │ │ └── test_pt_dataset.py │ └── test_functions.py │ └── test_asyncio.py └── uv.lock /.github/ISSUE_TEMPLATE/bug_report.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.github/ISSUE_TEMPLATE/bug_report.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/documentation_improvement.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.github/ISSUE_TEMPLATE/documentation_improvement.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.github/ISSUE_TEMPLATE/feature_request.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/integration_request.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.github/ISSUE_TEMPLATE/integration_request.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/question.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.github/ISSUE_TEMPLATE/question.yml -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.github/workflows/docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.github/workflows/docs.yml -------------------------------------------------------------------------------- /.github/workflows/lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.github/workflows/lint.yml -------------------------------------------------------------------------------- /.github/workflows/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.github/workflows/release.yml -------------------------------------------------------------------------------- /.github/workflows/unit_test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.github/workflows/unit_test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.gitmodules -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.python-versions: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/.python-versions -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/CITATION.cff -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/README.md -------------------------------------------------------------------------------- /codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/codecov.yml -------------------------------------------------------------------------------- /docs/api_reference/bridges/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/bridges/index.md -------------------------------------------------------------------------------- /docs/api_reference/bridges/langchain.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/bridges/langchain.md -------------------------------------------------------------------------------- /docs/api_reference/bridges/llamaindex.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/bridges/llamaindex.md -------------------------------------------------------------------------------- /docs/api_reference/data_collators/huggingface.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/data_collators/huggingface.md -------------------------------------------------------------------------------- /docs/api_reference/data_collators/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/data_collators/index.md -------------------------------------------------------------------------------- /docs/api_reference/data_structures/bridge.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/data_structures/bridge.md -------------------------------------------------------------------------------- /docs/api_reference/data_structures/evals.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.data_structures.evals 4 | -------------------------------------------------------------------------------- /docs/api_reference/data_structures/knowledge_node.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/data_structures/knowledge_node.md -------------------------------------------------------------------------------- /docs/api_reference/data_structures/rag.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/data_structures/rag.md -------------------------------------------------------------------------------- /docs/api_reference/data_structures/results.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/data_structures/results.md -------------------------------------------------------------------------------- /docs/api_reference/decorators/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/decorators/index.md -------------------------------------------------------------------------------- /docs/api_reference/evals/benchmarker.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/evals/benchmarker.md -------------------------------------------------------------------------------- /docs/api_reference/evals/benchmarks/huggingface/boolq.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/evals/benchmarks/huggingface/boolq.md -------------------------------------------------------------------------------- /docs/api_reference/evals/benchmarks/huggingface/hellaswag.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/evals/benchmarks/huggingface/hellaswag.md -------------------------------------------------------------------------------- /docs/api_reference/evals/benchmarks/huggingface/hotpotqa.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/evals/benchmarks/huggingface/hotpotqa.md -------------------------------------------------------------------------------- /docs/api_reference/evals/benchmarks/huggingface/mmlu.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/evals/benchmarks/huggingface/mmlu.md -------------------------------------------------------------------------------- /docs/api_reference/evals/benchmarks/huggingface/natural_questions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/evals/benchmarks/huggingface/natural_questions.md -------------------------------------------------------------------------------- /docs/api_reference/evals/benchmarks/huggingface/pubmedqa.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/evals/benchmarks/huggingface/pubmedqa.md -------------------------------------------------------------------------------- /docs/api_reference/evals/benchmarks/huggingface/squad_v2.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/evals/benchmarks/huggingface/squad_v2.md -------------------------------------------------------------------------------- /docs/api_reference/evals/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/evals/index.md -------------------------------------------------------------------------------- /docs/api_reference/evals/metrics/exact_match.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/evals/metrics/exact_match.md -------------------------------------------------------------------------------- /docs/api_reference/exceptions/bridge.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.exceptions.bridge 4 | -------------------------------------------------------------------------------- /docs/api_reference/exceptions/data_collator.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.exceptions.data_collator 4 | -------------------------------------------------------------------------------- /docs/api_reference/exceptions/evals.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.exceptions.evals 4 | -------------------------------------------------------------------------------- /docs/api_reference/exceptions/fl_tasks.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.exceptions.fl_tasks 4 | -------------------------------------------------------------------------------- /docs/api_reference/exceptions/generator.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.exceptions.generator 4 | -------------------------------------------------------------------------------- /docs/api_reference/exceptions/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/exceptions/index.md -------------------------------------------------------------------------------- /docs/api_reference/exceptions/inspectors.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.exceptions.inspectors 4 | -------------------------------------------------------------------------------- /docs/api_reference/exceptions/knowledge_stores.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/exceptions/knowledge_stores.md -------------------------------------------------------------------------------- /docs/api_reference/exceptions/loss.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.exceptions.loss 4 | -------------------------------------------------------------------------------- /docs/api_reference/exceptions/rag_trainer.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/exceptions/rag_trainer.md -------------------------------------------------------------------------------- /docs/api_reference/exceptions/retriever.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.exceptions.retriever 4 | -------------------------------------------------------------------------------- /docs/api_reference/exceptions/tokenizer.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.exceptions.tokenizer 4 | -------------------------------------------------------------------------------- /docs/api_reference/exceptions/trainer.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.exceptions.trainer 4 | -------------------------------------------------------------------------------- /docs/api_reference/finetuning_datasets/huggingface.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/finetuning_datasets/huggingface.md -------------------------------------------------------------------------------- /docs/api_reference/finetuning_datasets/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/finetuning_datasets/index.md -------------------------------------------------------------------------------- /docs/api_reference/finetuning_datasets/pytorch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/finetuning_datasets/pytorch.md -------------------------------------------------------------------------------- /docs/api_reference/fl_tasks/huggingface.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/fl_tasks/huggingface.md -------------------------------------------------------------------------------- /docs/api_reference/fl_tasks/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/fl_tasks/index.md -------------------------------------------------------------------------------- /docs/api_reference/fl_tasks/pytorch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/fl_tasks/pytorch.md -------------------------------------------------------------------------------- /docs/api_reference/generators/huggingface.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/generators/huggingface.md -------------------------------------------------------------------------------- /docs/api_reference/generators/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/generators/index.md -------------------------------------------------------------------------------- /docs/api_reference/generators/unsloth.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/generators/unsloth.md -------------------------------------------------------------------------------- /docs/api_reference/index.md: -------------------------------------------------------------------------------- 1 | # API Reference 2 | -------------------------------------------------------------------------------- /docs/api_reference/inspectors/huggingface.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/inspectors/huggingface.md -------------------------------------------------------------------------------- /docs/api_reference/inspectors/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/inspectors/index.md -------------------------------------------------------------------------------- /docs/api_reference/inspectors/pytorch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/inspectors/pytorch.md -------------------------------------------------------------------------------- /docs/api_reference/knowledge_nodes/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/knowledge_nodes/index.md -------------------------------------------------------------------------------- /docs/api_reference/knowledge_stores/in_memory.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/knowledge_stores/in_memory.md -------------------------------------------------------------------------------- /docs/api_reference/knowledge_stores/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/knowledge_stores/index.md -------------------------------------------------------------------------------- /docs/api_reference/knowledge_stores/mixins.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.knowledge_stores.mixins 4 | -------------------------------------------------------------------------------- /docs/api_reference/knowledge_stores/qdrant.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/knowledge_stores/qdrant.md -------------------------------------------------------------------------------- /docs/api_reference/loss/pytorch.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | ::: src.fed_rag.loss.pytorch.lsr 4 | -------------------------------------------------------------------------------- /docs/api_reference/rag_system/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/rag_system/index.md -------------------------------------------------------------------------------- /docs/api_reference/retrievers/huggingface.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/retrievers/huggingface.md -------------------------------------------------------------------------------- /docs/api_reference/retrievers/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/retrievers/index.md -------------------------------------------------------------------------------- /docs/api_reference/tokenizers/huggingface.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/tokenizers/huggingface.md -------------------------------------------------------------------------------- /docs/api_reference/tokenizers/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/tokenizers/index.md -------------------------------------------------------------------------------- /docs/api_reference/trainer_managers/huggingface.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/trainer_managers/huggingface.md -------------------------------------------------------------------------------- /docs/api_reference/trainer_managers/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/trainer_managers/index.md -------------------------------------------------------------------------------- /docs/api_reference/trainer_managers/pytorch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/trainer_managers/pytorch.md -------------------------------------------------------------------------------- /docs/api_reference/trainers/huggingface.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/trainers/huggingface.md -------------------------------------------------------------------------------- /docs/api_reference/trainers/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/trainers/index.md -------------------------------------------------------------------------------- /docs/api_reference/trainers/pytorch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/api_reference/trainers/pytorch.md -------------------------------------------------------------------------------- /docs/assets/favicon-48x48.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/assets/favicon-48x48.svg -------------------------------------------------------------------------------- /docs/assets/favicon.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/assets/favicon.ico -------------------------------------------------------------------------------- /docs/assets/logo-dark.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/assets/logo-dark.svg -------------------------------------------------------------------------------- /docs/assets/logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/assets/logo.svg -------------------------------------------------------------------------------- /docs/assets/vector-logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/assets/vector-logo.svg -------------------------------------------------------------------------------- /docs/community/changelog.md: -------------------------------------------------------------------------------- 1 | --8<-- "CHANGELOG.md" 2 | -------------------------------------------------------------------------------- /docs/community/contributing/ask_question.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/community/contributing/ask_question.md -------------------------------------------------------------------------------- /docs/community/contributing/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/community/contributing/index.md -------------------------------------------------------------------------------- /docs/community/contributing/submit_issue.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/community/contributing/submit_issue.md -------------------------------------------------------------------------------- /docs/community/contributing/submit_pr.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/community/contributing/submit_pr.md -------------------------------------------------------------------------------- /docs/community/index.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/community/resources/index.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/community/resources/pocket_references.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/community/resources/pocket_references.md -------------------------------------------------------------------------------- /docs/examples/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/examples/index.md -------------------------------------------------------------------------------- /docs/examples/ra_dit/benchmarking.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/examples/ra_dit/benchmarking.md -------------------------------------------------------------------------------- /docs/examples/ra_dit/federated_finetune.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/examples/ra_dit/federated_finetune.md -------------------------------------------------------------------------------- /docs/examples/ra_dit/finetune.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/examples/ra_dit/finetune.md -------------------------------------------------------------------------------- /docs/examples/ra_dit/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/examples/ra_dit/index.md -------------------------------------------------------------------------------- /docs/examples/ra_dit/qdrant_knowledge_store_wikipedia.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/examples/ra_dit/qdrant_knowledge_store_wikipedia.md -------------------------------------------------------------------------------- /docs/getting_started/advanced_usage/lower_level_ralt.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/advanced_usage/lower_level_ralt.md -------------------------------------------------------------------------------- /docs/getting_started/essentials.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/essentials.md -------------------------------------------------------------------------------- /docs/getting_started/import_patterns.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/import_patterns.md -------------------------------------------------------------------------------- /docs/getting_started/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/installation.md -------------------------------------------------------------------------------- /docs/getting_started/integrations.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/integrations.md -------------------------------------------------------------------------------- /docs/getting_started/quick_starts/benchmark_mmlu.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/quick_starts/benchmark_mmlu.md -------------------------------------------------------------------------------- /docs/getting_started/quick_starts/federated.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/quick_starts/federated.md -------------------------------------------------------------------------------- /docs/getting_started/quick_starts/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/quick_starts/index.md -------------------------------------------------------------------------------- /docs/getting_started/quick_starts/rag_finetuning.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/quick_starts/rag_finetuning.md -------------------------------------------------------------------------------- /docs/getting_started/quick_starts/rag_inference.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/quick_starts/rag_inference.md -------------------------------------------------------------------------------- /docs/getting_started/standard_usage.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/standard_usage.md -------------------------------------------------------------------------------- /docs/getting_started/tutorials/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/tutorials/index.md -------------------------------------------------------------------------------- /docs/getting_started/tutorials/lsr.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/tutorials/lsr.md -------------------------------------------------------------------------------- /docs/getting_started/tutorials/ralt.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/getting_started/tutorials/ralt.md -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/javascripts/mathjax.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/javascripts/mathjax.js -------------------------------------------------------------------------------- /docs/notebooks/basic_fl.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/notebooks/basic_fl.ipynb -------------------------------------------------------------------------------- /docs/notebooks/basic_starter_hf.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/notebooks/basic_starter_hf.ipynb -------------------------------------------------------------------------------- /docs/notebooks/integrations/langchain.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/notebooks/integrations/langchain.ipynb -------------------------------------------------------------------------------- /docs/notebooks/integrations/llama_index.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/notebooks/integrations/llama_index.ipynb -------------------------------------------------------------------------------- /docs/notebooks/integrations/qdrant.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/notebooks/integrations/qdrant.ipynb -------------------------------------------------------------------------------- /docs/notebooks/integrations/qdrant_async.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/notebooks/integrations/qdrant_async.ipynb -------------------------------------------------------------------------------- /docs/notebooks/integrations/unsloth.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/notebooks/integrations/unsloth.ipynb -------------------------------------------------------------------------------- /docs/notebooks/no_encode_rag_with_mcp.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/notebooks/no_encode_rag_with_mcp.ipynb -------------------------------------------------------------------------------- /docs/notebooks/rag_benchmarking_hf_mmlu.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/notebooks/rag_benchmarking_hf_mmlu.ipynb -------------------------------------------------------------------------------- /docs/overrides/partials/copyright.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/overrides/partials/copyright.html -------------------------------------------------------------------------------- /docs/overrides/partials/logo.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/overrides/partials/logo.html -------------------------------------------------------------------------------- /docs/stylesheets/extra.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/docs/stylesheets/extra.css -------------------------------------------------------------------------------- /example_scripts/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/example_scripts/README.md -------------------------------------------------------------------------------- /example_scripts/cookbook_script-basic_fl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/example_scripts/cookbook_script-basic_fl.py -------------------------------------------------------------------------------- /examples/knowledge_stores/ra-dit-ks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/knowledge_stores/ra-dit-ks/README.md -------------------------------------------------------------------------------- /examples/knowledge_stores/ra-dit-ks/docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/knowledge_stores/ra-dit-ks/docker/Dockerfile -------------------------------------------------------------------------------- /examples/knowledge_stores/ra-dit-ks/docker/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/knowledge_stores/ra-dit-ks/docker/README.md -------------------------------------------------------------------------------- /examples/knowledge_stores/ra-dit-ks/docker/healthcheck.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/knowledge_stores/ra-dit-ks/docker/healthcheck.sh -------------------------------------------------------------------------------- /examples/knowledge_stores/ra-dit-ks/docker/startup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/knowledge_stores/ra-dit-ks/docker/startup.sh -------------------------------------------------------------------------------- /examples/knowledge_stores/ra-dit-ks/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/knowledge_stores/ra-dit-ks/pyproject.toml -------------------------------------------------------------------------------- /examples/knowledge_stores/ra-dit-ks/src/ra_dit_ks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/knowledge_stores/ra-dit-ks/src/ra_dit_ks/__init__.py -------------------------------------------------------------------------------- /examples/knowledge_stores/ra-dit-ks/src/ra_dit_ks/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/knowledge_stores/ra-dit-ks/src/ra_dit_ks/logger.py -------------------------------------------------------------------------------- /examples/knowledge_stores/ra-dit-ks/src/ra_dit_ks/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/knowledge_stores/ra-dit-ks/src/ra_dit_ks/main.py -------------------------------------------------------------------------------- /examples/knowledge_stores/ra-dit-ks/uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/knowledge_stores/ra-dit-ks/uv.lock -------------------------------------------------------------------------------- /examples/quick-start/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/quick-start/README.md -------------------------------------------------------------------------------- /examples/quick-start/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/quick-start/pyproject.toml -------------------------------------------------------------------------------- /examples/quick-start/quick_start/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/quick-start/quick_start/_cifar_dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/quick-start/quick_start/_cifar_dataloaders.py -------------------------------------------------------------------------------- /examples/quick-start/quick_start/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/quick-start/quick_start/main.py -------------------------------------------------------------------------------- /examples/ra-dit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/README.md -------------------------------------------------------------------------------- /examples/ra-dit/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/pyproject.toml -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/__init__.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/_dataset_prep/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/_dataset_prep/base_data_prepper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/_dataset_prep/base_data_prepper.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/_dataset_prep/qa/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/_dataset_prep/qa/commonsense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/_dataset_prep/qa/commonsense.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/_dataset_prep/qa/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/_dataset_prep/qa/math.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/_dataset_prep/qa/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/_dataset_prep/qa/mixin.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/_dataset_prep/qa/pubmed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/_dataset_prep/qa/pubmed.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/_dataset_prep/qa/web_questions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/_dataset_prep/qa/web_questions.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/_dataset_prep/qa/wiki.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/_dataset_prep/qa/wiki.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/benchmarker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/benchmarker.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/centralized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/centralized.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/datasets.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/evaluation_benchmarks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/evaluation_benchmarks/__init__.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/evaluation_benchmarks/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/evaluation_benchmarks/base.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/evaluation_benchmarks/mmlu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/evaluation_benchmarks/mmlu.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/federated_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/federated_finetune.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/finetune.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/generators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/generators/__init__.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/generators/llama2_7b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/generators/llama2_7b.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/generators/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/generators/utils.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/knowledge_stores/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/knowledge_stores/from_dragon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/knowledge_stores/from_dragon.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/knowledge_stores/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/knowledge_stores/utils.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/logger.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/main.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/rag_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/rag_system.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/retrievers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/retrievers/__init__.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/retrievers/dragon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/retrievers/dragon.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/trainers_and_testers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/trainers_and_testers/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/trainers_and_testers/generator.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/trainers_and_testers/retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/trainers_and_testers/retriever.py -------------------------------------------------------------------------------- /examples/ra-dit/ra_dit/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/ra_dit/utils.py -------------------------------------------------------------------------------- /examples/ra-dit/uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/examples/ra-dit/uv.lock -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/mypy.ini -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- 1 | [pytest] 2 | asyncio_mode=auto 3 | -------------------------------------------------------------------------------- /src/fed_rag/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/__init__.pyi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/__init__.pyi -------------------------------------------------------------------------------- /src/fed_rag/_bridges/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/fed_rag/_bridges/langchain/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/_bridges/langchain/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/_bridges/langchain/_bridge_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/_bridges/langchain/_bridge_classes.py -------------------------------------------------------------------------------- /src/fed_rag/_bridges/langchain/_version.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.1.0" 2 | -------------------------------------------------------------------------------- /src/fed_rag/_bridges/langchain/bridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/_bridges/langchain/bridge.py -------------------------------------------------------------------------------- /src/fed_rag/_bridges/llamaindex/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/_bridges/llamaindex/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/_bridges/llamaindex/_managed_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/_bridges/llamaindex/_managed_index.py -------------------------------------------------------------------------------- /src/fed_rag/_bridges/llamaindex/_version.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.1.0" 2 | -------------------------------------------------------------------------------- /src/fed_rag/_bridges/llamaindex/bridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/_bridges/llamaindex/bridge.py -------------------------------------------------------------------------------- /src/fed_rag/_version.py: -------------------------------------------------------------------------------- 1 | VERSION = "0.0.27" 2 | -------------------------------------------------------------------------------- /src/fed_rag/base/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/fed_rag/base/bridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/bridge.py -------------------------------------------------------------------------------- /src/fed_rag/base/data_collator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/data_collator.py -------------------------------------------------------------------------------- /src/fed_rag/base/evals/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/fed_rag/base/evals/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/evals/benchmark.py -------------------------------------------------------------------------------- /src/fed_rag/base/evals/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/evals/metric.py -------------------------------------------------------------------------------- /src/fed_rag/base/fl_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/fl_task.py -------------------------------------------------------------------------------- /src/fed_rag/base/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/generator.py -------------------------------------------------------------------------------- /src/fed_rag/base/generator_mixins/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/generator_mixins/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/base/generator_mixins/audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/generator_mixins/audio.py -------------------------------------------------------------------------------- /src/fed_rag/base/generator_mixins/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/generator_mixins/image.py -------------------------------------------------------------------------------- /src/fed_rag/base/generator_mixins/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/generator_mixins/video.py -------------------------------------------------------------------------------- /src/fed_rag/base/knowledge_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/knowledge_store.py -------------------------------------------------------------------------------- /src/fed_rag/base/no_encode_knowledge_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/no_encode_knowledge_store.py -------------------------------------------------------------------------------- /src/fed_rag/base/retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/retriever.py -------------------------------------------------------------------------------- /src/fed_rag/base/retriever_mixins/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/retriever_mixins/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/base/retriever_mixins/audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/retriever_mixins/audio.py -------------------------------------------------------------------------------- /src/fed_rag/base/retriever_mixins/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/retriever_mixins/image.py -------------------------------------------------------------------------------- /src/fed_rag/base/retriever_mixins/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/retriever_mixins/video.py -------------------------------------------------------------------------------- /src/fed_rag/base/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/tokenizer.py -------------------------------------------------------------------------------- /src/fed_rag/base/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/trainer.py -------------------------------------------------------------------------------- /src/fed_rag/base/trainer_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/trainer_config.py -------------------------------------------------------------------------------- /src/fed_rag/base/trainer_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/base/trainer_manager.py -------------------------------------------------------------------------------- /src/fed_rag/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/core/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/core/no_encode_rag_system/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/core/no_encode_rag_system/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/core/no_encode_rag_system/_asynchronous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/core/no_encode_rag_system/_asynchronous.py -------------------------------------------------------------------------------- /src/fed_rag/core/no_encode_rag_system/_synchronous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/core/no_encode_rag_system/_synchronous.py -------------------------------------------------------------------------------- /src/fed_rag/core/no_encode_rag_system/asynchronous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/core/no_encode_rag_system/asynchronous.py -------------------------------------------------------------------------------- /src/fed_rag/core/no_encode_rag_system/synchronous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/core/no_encode_rag_system/synchronous.py -------------------------------------------------------------------------------- /src/fed_rag/core/rag_system/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/core/rag_system/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/core/rag_system/_asynchronous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/core/rag_system/_asynchronous.py -------------------------------------------------------------------------------- /src/fed_rag/core/rag_system/_synchronous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/core/rag_system/_synchronous.py -------------------------------------------------------------------------------- /src/fed_rag/core/rag_system/asynchronous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/core/rag_system/asynchronous.py -------------------------------------------------------------------------------- /src/fed_rag/core/rag_system/synchronous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/core/rag_system/synchronous.py -------------------------------------------------------------------------------- /src/fed_rag/data_collators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/fed_rag/data_collators/huggingface/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/data_collators/huggingface/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/data_collators/huggingface/lsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/data_collators/huggingface/lsr.py -------------------------------------------------------------------------------- /src/fed_rag/data_collators/huggingface/ralt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/data_collators/huggingface/ralt.py -------------------------------------------------------------------------------- /src/fed_rag/data_structures/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/data_structures/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/data_structures/bridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/data_structures/bridge.py -------------------------------------------------------------------------------- /src/fed_rag/data_structures/evals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/data_structures/evals.py -------------------------------------------------------------------------------- /src/fed_rag/data_structures/knowledge_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/data_structures/knowledge_node.py -------------------------------------------------------------------------------- /src/fed_rag/data_structures/rag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/data_structures/rag.py -------------------------------------------------------------------------------- /src/fed_rag/data_structures/results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/data_structures/results.py -------------------------------------------------------------------------------- /src/fed_rag/data_structures/retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/data_structures/retriever.py -------------------------------------------------------------------------------- /src/fed_rag/decorators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/decorators/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/decorators/tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/decorators/tester.py -------------------------------------------------------------------------------- /src/fed_rag/decorators/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/decorators/trainer.py -------------------------------------------------------------------------------- /src/fed_rag/evals/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/evals/benchmarker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/benchmarker.py -------------------------------------------------------------------------------- /src/fed_rag/evals/benchmarks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/benchmarks/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/evals/benchmarks/huggingface/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/benchmarks/huggingface/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/evals/benchmarks/huggingface/boolq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/benchmarks/huggingface/boolq.py -------------------------------------------------------------------------------- /src/fed_rag/evals/benchmarks/huggingface/hellaswag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/benchmarks/huggingface/hellaswag.py -------------------------------------------------------------------------------- /src/fed_rag/evals/benchmarks/huggingface/hotpotqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/benchmarks/huggingface/hotpotqa.py -------------------------------------------------------------------------------- /src/fed_rag/evals/benchmarks/huggingface/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/benchmarks/huggingface/mixin.py -------------------------------------------------------------------------------- /src/fed_rag/evals/benchmarks/huggingface/mmlu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/benchmarks/huggingface/mmlu.py -------------------------------------------------------------------------------- /src/fed_rag/evals/benchmarks/huggingface/natural_questions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/benchmarks/huggingface/natural_questions.py -------------------------------------------------------------------------------- /src/fed_rag/evals/benchmarks/huggingface/pubmedqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/benchmarks/huggingface/pubmedqa.py -------------------------------------------------------------------------------- /src/fed_rag/evals/benchmarks/huggingface/squad_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/benchmarks/huggingface/squad_v2.py -------------------------------------------------------------------------------- /src/fed_rag/evals/benchmarks/huggingface/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/benchmarks/huggingface/utils.py -------------------------------------------------------------------------------- /src/fed_rag/evals/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/metrics/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/evals/metrics/exact_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/metrics/exact_match.py -------------------------------------------------------------------------------- /src/fed_rag/evals/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/evals/utils.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/bridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/bridge.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/common.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/core.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/data_collator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/data_collator.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/evals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/evals.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/fl_tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/fl_tasks.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/generator.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/inspectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/inspectors.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/knowledge_stores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/knowledge_stores.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/loss.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/rag_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/rag_system.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/retriever.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/tokenizer.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/trainer.py -------------------------------------------------------------------------------- /src/fed_rag/exceptions/trainer_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/exceptions/trainer_manager.py -------------------------------------------------------------------------------- /src/fed_rag/fl_tasks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/fed_rag/fl_tasks/huggingface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/fl_tasks/huggingface.py -------------------------------------------------------------------------------- /src/fed_rag/fl_tasks/pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/fl_tasks/pytorch.py -------------------------------------------------------------------------------- /src/fed_rag/generators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/generators/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/generators/huggingface/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/generators/huggingface/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/generators/huggingface/hf_multimodal_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/generators/huggingface/hf_multimodal_model.py -------------------------------------------------------------------------------- /src/fed_rag/generators/huggingface/hf_peft_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/generators/huggingface/hf_peft_model.py -------------------------------------------------------------------------------- /src/fed_rag/generators/huggingface/hf_pretrained_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/generators/huggingface/hf_pretrained_model.py -------------------------------------------------------------------------------- /src/fed_rag/generators/huggingface/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/generators/huggingface/mixin.py -------------------------------------------------------------------------------- /src/fed_rag/generators/huggingface/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/generators/huggingface/utils.py -------------------------------------------------------------------------------- /src/fed_rag/generators/unsloth/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/generators/unsloth/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/generators/unsloth/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/generators/unsloth/mixin.py -------------------------------------------------------------------------------- /src/fed_rag/generators/unsloth/unsloth_fast_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/generators/unsloth/unsloth_fast_model.py -------------------------------------------------------------------------------- /src/fed_rag/generators/unsloth/unsloth_fast_multimodal_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/generators/unsloth/unsloth_fast_multimodal_model.py -------------------------------------------------------------------------------- /src/fed_rag/generators/unsloth/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/generators/unsloth/utils.py -------------------------------------------------------------------------------- /src/fed_rag/inspectors/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/fed_rag/inspectors/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/inspectors/common.py -------------------------------------------------------------------------------- /src/fed_rag/inspectors/huggingface/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/inspectors/huggingface/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/inspectors/huggingface/tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/inspectors/huggingface/tester.py -------------------------------------------------------------------------------- /src/fed_rag/inspectors/huggingface/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/inspectors/huggingface/trainer.py -------------------------------------------------------------------------------- /src/fed_rag/inspectors/huggingface/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/inspectors/huggingface/utils.py -------------------------------------------------------------------------------- /src/fed_rag/inspectors/pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/inspectors/pytorch/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/inspectors/pytorch/tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/inspectors/pytorch/tester.py -------------------------------------------------------------------------------- /src/fed_rag/inspectors/pytorch/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/inspectors/pytorch/trainer.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/in_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/in_memory.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/mixins.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/no_encode/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/no_encode/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/no_encode/mcp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/no_encode/mcp/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/no_encode/mcp/sources/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/no_encode/mcp/sources/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/no_encode/mcp/sources/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/no_encode/mcp/sources/base.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/no_encode/mcp/sources/stdio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/no_encode/mcp/sources/stdio.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/no_encode/mcp/sources/streamable_http.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/no_encode/mcp/sources/streamable_http.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/no_encode/mcp/sources/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/no_encode/mcp/sources/utils.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/no_encode/mcp/store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/no_encode/mcp/store.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/qdrant/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/qdrant/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/qdrant/asynchronous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/qdrant/asynchronous.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/qdrant/sync.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/qdrant/sync.py -------------------------------------------------------------------------------- /src/fed_rag/knowledge_stores/qdrant/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/knowledge_stores/qdrant/utils.py -------------------------------------------------------------------------------- /src/fed_rag/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/fed_rag/loss/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/fed_rag/loss/pytorch/lsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/loss/pytorch/lsr.py -------------------------------------------------------------------------------- /src/fed_rag/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/fed_rag/retrievers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/retrievers/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/retrievers/huggingface/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/retrievers/huggingface/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/retrievers/huggingface/hf_sentence_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/retrievers/huggingface/hf_sentence_transformer.py -------------------------------------------------------------------------------- /src/fed_rag/tokenizers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/fed_rag/tokenizers/hf_pretrained_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/tokenizers/hf_pretrained_tokenizer.py -------------------------------------------------------------------------------- /src/fed_rag/tokenizers/unsloth_pretrained_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/tokenizers/unsloth_pretrained_tokenizer.py -------------------------------------------------------------------------------- /src/fed_rag/trainer_configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainer_configs/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/trainer_configs/pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainer_configs/pytorch.py -------------------------------------------------------------------------------- /src/fed_rag/trainer_managers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainer_managers/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/trainer_managers/huggingface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainer_managers/huggingface.py -------------------------------------------------------------------------------- /src/fed_rag/trainer_managers/pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainer_managers/pytorch.py -------------------------------------------------------------------------------- /src/fed_rag/trainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainers/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/trainers/huggingface/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainers/huggingface/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/trainers/huggingface/lsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainers/huggingface/lsr.py -------------------------------------------------------------------------------- /src/fed_rag/trainers/huggingface/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainers/huggingface/mixin.py -------------------------------------------------------------------------------- /src/fed_rag/trainers/huggingface/ralt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainers/huggingface/ralt.py -------------------------------------------------------------------------------- /src/fed_rag/trainers/pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainers/pytorch/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/trainers/pytorch/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainers/pytorch/mixin.py -------------------------------------------------------------------------------- /src/fed_rag/trainers/pytorch/training_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/trainers/pytorch/training_args.py -------------------------------------------------------------------------------- /src/fed_rag/types/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/types/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/types/bridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/types/bridge.py -------------------------------------------------------------------------------- /src/fed_rag/types/knowledge_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/types/knowledge_node.py -------------------------------------------------------------------------------- /src/fed_rag/types/rag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/types/rag.py -------------------------------------------------------------------------------- /src/fed_rag/types/rag_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/types/rag_system.py -------------------------------------------------------------------------------- /src/fed_rag/types/results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/types/results.py -------------------------------------------------------------------------------- /src/fed_rag/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/fed_rag/utils/asyncio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/utils/asyncio.py -------------------------------------------------------------------------------- /src/fed_rag/utils/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/utils/data/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/utils/data/_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/utils/data/_functions.py -------------------------------------------------------------------------------- /src/fed_rag/utils/data/finetuning_datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/utils/data/finetuning_datasets/__init__.py -------------------------------------------------------------------------------- /src/fed_rag/utils/data/finetuning_datasets/huggingface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/utils/data/finetuning_datasets/huggingface.py -------------------------------------------------------------------------------- /src/fed_rag/utils/data/finetuning_datasets/pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/utils/data/finetuning_datasets/pytorch.py -------------------------------------------------------------------------------- /src/fed_rag/utils/huggingface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/utils/huggingface.py -------------------------------------------------------------------------------- /src/fed_rag/utils/notebook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/src/fed_rag/utils/notebook.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/api/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/api/test_data_structures_imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/api/test_data_structures_imports.py -------------------------------------------------------------------------------- /tests/api/test_deprecated_types_imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/api/test_deprecated_types_imports.py -------------------------------------------------------------------------------- /tests/api/test_evals_imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/api/test_evals_imports.py -------------------------------------------------------------------------------- /tests/api/test_namespaced_imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/api/test_namespaced_imports.py -------------------------------------------------------------------------------- /tests/api/test_root_imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/api/test_root_imports.py -------------------------------------------------------------------------------- /tests/bridges/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/bridges/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/bridges/conftest.py -------------------------------------------------------------------------------- /tests/bridges/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/bridges/test_base.py -------------------------------------------------------------------------------- /tests/bridges/test_langchain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/bridges/test_langchain.py -------------------------------------------------------------------------------- /tests/bridges/test_llamaindex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/bridges/test_llamaindex.py -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/data_collators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/data_collators/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/data_collators/conftest.py -------------------------------------------------------------------------------- /tests/data_collators/huggingface/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/data_collators/huggingface/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/data_collators/huggingface/conftest.py -------------------------------------------------------------------------------- /tests/data_collators/huggingface/test_data_collator_for_ralt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/data_collators/huggingface/test_data_collator_for_ralt.py -------------------------------------------------------------------------------- /tests/data_collators/huggingface/test_hf_data_collators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/data_collators/huggingface/test_hf_data_collators.py -------------------------------------------------------------------------------- /tests/data_collators/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/data_collators/test_base.py -------------------------------------------------------------------------------- /tests/data_structures/test_evals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/data_structures/test_evals.py -------------------------------------------------------------------------------- /tests/decorators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/decorators/test_huggingface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/decorators/test_huggingface.py -------------------------------------------------------------------------------- /tests/decorators/test_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/decorators/test_pytorch.py -------------------------------------------------------------------------------- /tests/evals/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/evals/benchmarks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/evals/benchmarks/_benchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/_benchmarks.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/huggingface/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/evals/benchmarks/huggingface/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/huggingface/conftest.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/huggingface/test_boolq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/huggingface/test_boolq.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/huggingface/test_hellaswag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/huggingface/test_hellaswag.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/huggingface/test_hotpotqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/huggingface/test_hotpotqa.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/huggingface/test_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/huggingface/test_mixin.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/huggingface/test_mmlu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/huggingface/test_mmlu.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/huggingface/test_natural_questions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/huggingface/test_natural_questions.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/huggingface/test_pubmedqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/huggingface/test_pubmedqa.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/huggingface/test_squad_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/huggingface/test_squad_v2.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/huggingface/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/huggingface/test_utils.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/test_base.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/test_benchmarker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/test_benchmarker.py -------------------------------------------------------------------------------- /tests/evals/benchmarks/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/benchmarks/test_utils.py -------------------------------------------------------------------------------- /tests/evals/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/evals/metrics/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/metrics/test_base.py -------------------------------------------------------------------------------- /tests/evals/metrics/test_exact_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/evals/metrics/test_exact_match.py -------------------------------------------------------------------------------- /tests/fl_tasks/__init__.py: -------------------------------------------------------------------------------- 1 | """PyTorchTrainerConfig Unit Tests""" 2 | -------------------------------------------------------------------------------- /tests/fl_tasks/huggingface/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/fl_tasks/huggingface/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/fl_tasks/huggingface/conftest.py -------------------------------------------------------------------------------- /tests/fl_tasks/huggingface/test_huggingface_fltask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/fl_tasks/huggingface/test_huggingface_fltask.py -------------------------------------------------------------------------------- /tests/fl_tasks/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/fl_tasks/pytorch/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/fl_tasks/pytorch/conftest.py -------------------------------------------------------------------------------- /tests/fl_tasks/pytorch/test_pytorch_fltask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/fl_tasks/pytorch/test_pytorch_fltask.py -------------------------------------------------------------------------------- /tests/generators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/generators/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/generators/conftest.py -------------------------------------------------------------------------------- /tests/generators/mixins/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/generators/mixins/test_audio_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/generators/mixins/test_audio_mixin.py -------------------------------------------------------------------------------- /tests/generators/mixins/test_image_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/generators/mixins/test_image_mixin.py -------------------------------------------------------------------------------- /tests/generators/mixins/test_video_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/generators/mixins/test_video_mixin.py -------------------------------------------------------------------------------- /tests/generators/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/generators/test_base.py -------------------------------------------------------------------------------- /tests/generators/test_hf_multimodal_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/generators/test_hf_multimodal_model.py -------------------------------------------------------------------------------- /tests/generators/test_hf_peft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/generators/test_hf_peft.py -------------------------------------------------------------------------------- /tests/generators/test_hf_pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/generators/test_hf_pretrained.py -------------------------------------------------------------------------------- /tests/generators/test_hf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/generators/test_hf_utils.py -------------------------------------------------------------------------------- /tests/generators/test_unsloth_fast_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/generators/test_unsloth_fast_model.py -------------------------------------------------------------------------------- /tests/generators/test_unsloth_fast_multimodal_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/generators/test_unsloth_fast_multimodal_model.py -------------------------------------------------------------------------------- /tests/generators/test_unsloth_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/generators/test_unsloth_utils.py -------------------------------------------------------------------------------- /tests/knowledge_stores/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/knowledge_stores/no_encode/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/knowledge_stores/no_encode/mcp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/knowledge_stores/no_encode/mcp/test_mcp_source_stdio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/knowledge_stores/no_encode/mcp/test_mcp_source_stdio.py -------------------------------------------------------------------------------- /tests/knowledge_stores/no_encode/mcp/test_mcp_source_streamable_http.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/knowledge_stores/no_encode/mcp/test_mcp_source_streamable_http.py -------------------------------------------------------------------------------- /tests/knowledge_stores/no_encode/mcp/test_mcp_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/knowledge_stores/no_encode/mcp/test_mcp_store.py -------------------------------------------------------------------------------- /tests/knowledge_stores/no_encode/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/knowledge_stores/no_encode/test_base.py -------------------------------------------------------------------------------- /tests/knowledge_stores/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/knowledge_stores/test_base.py -------------------------------------------------------------------------------- /tests/knowledge_stores/test_in_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/knowledge_stores/test_in_memory.py -------------------------------------------------------------------------------- /tests/knowledge_stores/test_in_memory_managed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/knowledge_stores/test_in_memory_managed.py -------------------------------------------------------------------------------- /tests/knowledge_stores/test_knowledge_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/knowledge_stores/test_knowledge_node.py -------------------------------------------------------------------------------- /tests/knowledge_stores/test_qdrant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/knowledge_stores/test_qdrant.py -------------------------------------------------------------------------------- /tests/knowledge_stores/test_qdrant_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/knowledge_stores/test_qdrant_async.py -------------------------------------------------------------------------------- /tests/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/loss/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/loss/pytorch/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/loss/pytorch/conftest.py -------------------------------------------------------------------------------- /tests/loss/pytorch/test_lsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/loss/pytorch/test_lsr.py -------------------------------------------------------------------------------- /tests/rag_system/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/rag_system/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/rag_system/conftest.py -------------------------------------------------------------------------------- /tests/rag_system/test_async_no_encode_rag_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/rag_system/test_async_no_encode_rag_system.py -------------------------------------------------------------------------------- /tests/rag_system/test_async_rag_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/rag_system/test_async_rag_system.py -------------------------------------------------------------------------------- /tests/rag_system/test_no_encode_rag_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/rag_system/test_no_encode_rag_system.py -------------------------------------------------------------------------------- /tests/rag_system/test_rag_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/rag_system/test_rag_system.py -------------------------------------------------------------------------------- /tests/rag_system/test_source_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/rag_system/test_source_node.py -------------------------------------------------------------------------------- /tests/retrievers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/retrievers/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/retrievers/conftest.py -------------------------------------------------------------------------------- /tests/retrievers/mixins/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/retrievers/mixins/test_audio_retriever_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/retrievers/mixins/test_audio_retriever_mixin.py -------------------------------------------------------------------------------- /tests/retrievers/mixins/test_image_retriever_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/retrievers/mixins/test_image_retriever_mixin.py -------------------------------------------------------------------------------- /tests/retrievers/mixins/test_video_retriever_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/retrievers/mixins/test_video_retriever_mixin.py -------------------------------------------------------------------------------- /tests/retrievers/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/retrievers/test_base.py -------------------------------------------------------------------------------- /tests/retrievers/test_hf_sentence_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/retrievers/test_hf_sentence_transformer.py -------------------------------------------------------------------------------- /tests/tokenizers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tokenizers/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/tokenizers/conftest.py -------------------------------------------------------------------------------- /tests/tokenizers/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/tokenizers/test_base.py -------------------------------------------------------------------------------- /tests/tokenizers/test_hf_pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/tokenizers/test_hf_pretrained.py -------------------------------------------------------------------------------- /tests/tokenizers/test_unsloth_pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/tokenizers/test_unsloth_pretrained.py -------------------------------------------------------------------------------- /tests/trainer_configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/trainer_configs/test_pytorch_trainer_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainer_configs/test_pytorch_trainer_config.py -------------------------------------------------------------------------------- /tests/trainer_managers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/trainer_managers/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainer_managers/conftest.py -------------------------------------------------------------------------------- /tests/trainer_managers/huggingface/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/trainer_managers/huggingface/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainer_managers/huggingface/conftest.py -------------------------------------------------------------------------------- /tests/trainer_managers/huggingface/test_hf_trainer_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainer_managers/huggingface/test_hf_trainer_manager.py -------------------------------------------------------------------------------- /tests/trainer_managers/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainer_managers/test_base.py -------------------------------------------------------------------------------- /tests/trainer_managers/test_pytorch_trainer_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainer_managers/test_pytorch_trainer_manager.py -------------------------------------------------------------------------------- /tests/trainers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/trainers/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainers/conftest.py -------------------------------------------------------------------------------- /tests/trainers/huggingface/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/trainers/huggingface/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainers/huggingface/conftest.py -------------------------------------------------------------------------------- /tests/trainers/huggingface/test_hf_lsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainers/huggingface/test_hf_lsr.py -------------------------------------------------------------------------------- /tests/trainers/huggingface/test_hf_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainers/huggingface/test_hf_mixin.py -------------------------------------------------------------------------------- /tests/trainers/huggingface/test_hf_ralt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainers/huggingface/test_hf_ralt.py -------------------------------------------------------------------------------- /tests/trainers/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/trainers/pytorch/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainers/pytorch/conftest.py -------------------------------------------------------------------------------- /tests/trainers/pytorch/test_pt_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainers/pytorch/test_pt_mixin.py -------------------------------------------------------------------------------- /tests/trainers/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/trainers/test_base.py -------------------------------------------------------------------------------- /tests/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/utils/data/finetuning_datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/utils/data/finetuning_datasets/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/utils/data/finetuning_datasets/conftest.py -------------------------------------------------------------------------------- /tests/utils/data/finetuning_datasets/test_hf_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/utils/data/finetuning_datasets/test_hf_dataset.py -------------------------------------------------------------------------------- /tests/utils/data/finetuning_datasets/test_pt_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/utils/data/finetuning_datasets/test_pt_dataset.py -------------------------------------------------------------------------------- /tests/utils/data/test_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/utils/data/test_functions.py -------------------------------------------------------------------------------- /tests/utils/test_asyncio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/tests/utils/test_asyncio.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorInstitute/fed-rag/HEAD/uv.lock --------------------------------------------------------------------------------