Skip to content

RAG Trainer Manager

RAGTrainerManagerError

Bases: FedRAGError

Base errors for all rag trainer manager relevant exceptions.

Source code in src/fed_rag/exceptions/trainer_manager.py 
4
5
6
7
class RAGTrainerManagerError(FedRAGError):
    """Base errors for all rag trainer manager relevant exceptions."""

    pass

UnspecifiedRetrieverTrainer

Bases: RAGTrainerManagerError

Raised if a retriever trainer has not been specified when one was expected to be.

Source code in src/fed_rag/exceptions/trainer_manager.py 
10
11
12
13
class UnspecifiedRetrieverTrainer(RAGTrainerManagerError):
    """Raised if a retriever trainer has not been specified when one was expected to be."""

    pass

UnspecifiedGeneratorTrainer

Bases: RAGTrainerManagerError

Raised if a generator trainer has not been specified when one was expected to be.

Source code in src/fed_rag/exceptions/trainer_manager.py 
16
17
18
19
class UnspecifiedGeneratorTrainer(RAGTrainerManagerError):
    """Raised if a generator trainer has not been specified when one was expected to be."""

    pass

UnsupportedTrainerMode

Bases: RAGTrainerManagerError

Raised if an unsupported trainer mode has been supplied.

Source code in src/fed_rag/exceptions/trainer_manager.py 
22
23
24
25
class UnsupportedTrainerMode(RAGTrainerManagerError):
    """Raised if an unsupported trainer mode has been supplied."""

    pass

InconsistentRAGSystems

Bases: RAGTrainerManagerError

Raised if trainers have inconsistent underlying RAG systems.

Source code in src/fed_rag/exceptions/trainer_manager.py 
28
29
30
31
class InconsistentRAGSystems(RAGTrainerManagerError):
    """Raised if trainers have inconsistent underlying RAG systems."""

    pass