Registry

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

Registry

LLMRegistry

Bases: Registry

Registry for Large Language Model providers.

This registry maps LLM provider names (defined in LLMProviderName enum) to their corresponding implementation classes. It allows for dynamic registration and retrieval of LLM provider implementations based on their enumerated types.

Attributes:
  • _key_class (Type) –

    The class used as the key for the registry.
Source code in src/augmentation/components/llms/registry.py 
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
class LLMRegistry(Registry):
    """
    Registry for Large Language Model providers.

    This registry maps LLM provider names (defined in LLMProviderName enum)
    to their corresponding implementation classes. It allows for dynamic
    registration and retrieval of LLM provider implementations based on
    their enumerated types.

    Attributes:
        _key_class (Type): The class used as the key for the registry.
    """

    _key_class: Type = LLMProviderName

LlamaindexLLMOutputExtractorRegistry

Bases: Registry

Registry for Llamaindex LLM Output Extractors.

This registry maps Llamaindex LLM Output Extractor names to their corresponding implementation classes. It allows for dynamic registration and retrieval of Llamaindex LLM Output Extractor implementations based on their enumerated types.

Attributes:
  • _key_class (Type) –

    The class used as the key for the registry.
Source code in src/augmentation/components/llms/registry.py 
25
26
27
28
29
30
31
32
33
34
35
36
37
38
class LlamaindexLLMOutputExtractorRegistry(Registry):
    """
    Registry for Llamaindex LLM Output Extractors.

    This registry maps Llamaindex LLM Output Extractor names to their
    corresponding implementation classes. It allows for dynamic registration
    and retrieval of Llamaindex LLM Output Extractor implementations based on
    their enumerated types.

    Attributes:
        _key_class (Type): The class used as the key for the registry.
    """

    _key_class: Type = LLMProviderName