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