Registry

This module contains functionality related to the the registry module for augmentation.components.retrievers.

Registry

RetrieverRegistry

Bases: Registry

Registry for managing retriever components in the RAG system.

This registry maps RetrieverName enum values to their corresponding retriever implementations, facilitating the creation and management of retriever instances based on configuration. It inherits from the base Registry class and specifies RetrieverName as the key type for registration and lookup operations.

Attributes:
  • _key_class (Type) –

    The class type used for registry keys, set to RetrieverName enum.
Source code in src/augmentation/components/retrievers/registry.py 
 9
10
11
12
13
14
15
16
17
18
19
20
21
class RetrieverRegistry(Registry):
    """Registry for managing retriever components in the RAG system.

    This registry maps RetrieverName enum values to their corresponding retriever
    implementations, facilitating the creation and management of retriever instances
    based on configuration. It inherits from the base Registry class and specifies
    RetrieverName as the key type for registration and lookup operations.

    Attributes:
        _key_class (Type): The class type used for registry keys, set to RetrieverName enum.
    """

    _key_class: Type = RetrieverName