Skip to content

base_drivers_config

BaseDriversConfig

Bases: ABC, SerializableMixin

Source code in griptape/configs/drivers/base_drivers_config.py 
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
@define
class BaseDriversConfig(ABC, SerializableMixin):
    _prompt_driver: Optional[BasePromptDriver] = field(
        kw_only=True, default=None, metadata={"serializable": True}, alias="prompt_driver"
    )
    _image_generation_driver: Optional[BaseImageGenerationDriver] = field(
        kw_only=True, default=None, metadata={"serializable": True}, alias="image_generation_driver"
    )
    _embedding_driver: Optional[BaseEmbeddingDriver] = field(
        kw_only=True, default=None, metadata={"serializable": True}, alias="embedding_driver"
    )
    _vector_store_driver: Optional[BaseVectorStoreDriver] = field(
        default=None, kw_only=True, metadata={"serializable": True}, alias="vector_store_driver"
    )
    _conversation_memory_driver: Optional[BaseConversationMemoryDriver] = field(
        default=None, kw_only=True, metadata={"serializable": True}, alias="conversation_memory_driver"
    )
    _text_to_speech_driver: Optional[BaseTextToSpeechDriver] = field(
        default=None, kw_only=True, metadata={"serializable": True}, alias="text_to_speech_driver"
    )
    _audio_transcription_driver: Optional[BaseAudioTranscriptionDriver] = field(
        default=None, kw_only=True, metadata={"serializable": True}, alias="audio_transcription_driver"
    )
    _ruleset_driver: Optional[BaseRulesetDriver] = field(
        default=None, kw_only=True, metadata={"serializable": True}, alias="ruleset_driver"
    )

    _last_drivers_config: Optional[BaseDriversConfig] = field(default=None)

    def __enter__(self) -> Self:
        from griptape.configs import Defaults

        self._last_drivers_config = Defaults.drivers_config

        Defaults.drivers_config = self

        return self

    def __exit__(self, type, value, traceback) -> None:  # noqa: ANN001, A002
        from griptape.configs import Defaults

        if self._last_drivers_config is not None:
            Defaults.drivers_config = self._last_drivers_config

        self._last_drivers_config = None

    @lazy_property()
    @abstractmethod
    def prompt_driver(self) -> BasePromptDriver: ...

    @lazy_property()
    @abstractmethod
    def image_generation_driver(self) -> BaseImageGenerationDriver: ...

    @lazy_property()
    @abstractmethod
    def embedding_driver(self) -> BaseEmbeddingDriver: ...

    @lazy_property()
    @abstractmethod
    def vector_store_driver(self) -> BaseVectorStoreDriver: ...

    @lazy_property()
    @abstractmethod
    def conversation_memory_driver(self) -> BaseConversationMemoryDriver: ...

    @lazy_property()
    @abstractmethod
    def text_to_speech_driver(self) -> BaseTextToSpeechDriver: ...

    @lazy_property()
    @abstractmethod
    def audio_transcription_driver(self) -> BaseAudioTranscriptionDriver: ...

    @lazy_property()
    @abstractmethod
    def ruleset_driver(self) -> BaseRulesetDriver: ...

_audio_transcription_driver = field(default=None, kw_only=True, metadata={'serializable': True}, alias='audio_transcription_driver') class-attribute instance-attribute

_conversation_memory_driver = field(default=None, kw_only=True, metadata={'serializable': True}, alias='conversation_memory_driver') class-attribute instance-attribute

_embedding_driver = field(kw_only=True, default=None, metadata={'serializable': True}, alias='embedding_driver') class-attribute instance-attribute

_image_generation_driver = field(kw_only=True, default=None, metadata={'serializable': True}, alias='image_generation_driver') class-attribute instance-attribute

_last_drivers_config = field(default=None) class-attribute instance-attribute

_prompt_driver = field(kw_only=True, default=None, metadata={'serializable': True}, alias='prompt_driver') class-attribute instance-attribute

_ruleset_driver = field(default=None, kw_only=True, metadata={'serializable': True}, alias='ruleset_driver') class-attribute instance-attribute

_text_to_speech_driver = field(default=None, kw_only=True, metadata={'serializable': True}, alias='text_to_speech_driver') class-attribute instance-attribute

_vector_store_driver = field(default=None, kw_only=True, metadata={'serializable': True}, alias='vector_store_driver') class-attribute instance-attribute

__enter__()

Source code in griptape/configs/drivers/base_drivers_config.py 
53
54
55
56
57
58
59
60
def __enter__(self) -> Self:
    from griptape.configs import Defaults

    self._last_drivers_config = Defaults.drivers_config

    Defaults.drivers_config = self

    return self

__exit__(type, value, traceback)

Source code in griptape/configs/drivers/base_drivers_config.py 
62
63
64
65
66
67
68
def __exit__(self, type, value, traceback) -> None:  # noqa: ANN001, A002
    from griptape.configs import Defaults

    if self._last_drivers_config is not None:
        Defaults.drivers_config = self._last_drivers_config

    self._last_drivers_config = None

audio_transcription_driver() abstractmethod

Source code in griptape/configs/drivers/base_drivers_config.py 
94
95
96
@lazy_property()
@abstractmethod
def audio_transcription_driver(self) -> BaseAudioTranscriptionDriver: ...

conversation_memory_driver() abstractmethod

Source code in griptape/configs/drivers/base_drivers_config.py 
86
87
88
@lazy_property()
@abstractmethod
def conversation_memory_driver(self) -> BaseConversationMemoryDriver: ...

embedding_driver() abstractmethod

Source code in griptape/configs/drivers/base_drivers_config.py 
78
79
80
@lazy_property()
@abstractmethod
def embedding_driver(self) -> BaseEmbeddingDriver: ...

image_generation_driver() abstractmethod

Source code in griptape/configs/drivers/base_drivers_config.py 
74
75
76
@lazy_property()
@abstractmethod
def image_generation_driver(self) -> BaseImageGenerationDriver: ...

prompt_driver() abstractmethod

Source code in griptape/configs/drivers/base_drivers_config.py 
70
71
72
@lazy_property()
@abstractmethod
def prompt_driver(self) -> BasePromptDriver: ...

ruleset_driver() abstractmethod

Source code in griptape/configs/drivers/base_drivers_config.py 
 98
 99
100
@lazy_property()
@abstractmethod
def ruleset_driver(self) -> BaseRulesetDriver: ...

text_to_speech_driver() abstractmethod

Source code in griptape/configs/drivers/base_drivers_config.py 
90
91
92
@lazy_property()
@abstractmethod
def text_to_speech_driver(self) -> BaseTextToSpeechDriver: ...

vector_store_driver() abstractmethod

Source code in griptape/configs/drivers/base_drivers_config.py 
82
83
84
@lazy_property()
@abstractmethod
def vector_store_driver(self) -> BaseVectorStoreDriver: ...