Skip to content

Base audio transcription driver

BaseAudioTranscriptionDriver

Bases: SerializableMixin, ExponentialBackoffMixin, ABC

Source code in griptape/drivers/audio_transcription/base_audio_transcription_driver.py 
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
@define
class BaseAudioTranscriptionDriver(SerializableMixin, ExponentialBackoffMixin, ABC):
    model: str = field(kw_only=True, metadata={"serializable": True})
    structure: Optional[Structure] = field(default=None, kw_only=True)

    def before_run(self) -> None:
        if self.structure:
            self.structure.publish_event(StartAudioTranscriptionEvent())

    def after_run(self) -> None:
        if self.structure:
            self.structure.publish_event(FinishAudioTranscriptionEvent())

    def run(self, audio: AudioArtifact, prompts: Optional[list[str]] = None) -> TextArtifact:
        for attempt in self.retrying():
            with attempt:
                self.before_run()
                result = self.try_run(audio, prompts)
                self.after_run()

                return result

        else:
            raise Exception("Failed to run audio transcription")

    @abstractmethod
    def try_run(self, audio: AudioArtifact, prompts: Optional[list[str]] = None) -> TextArtifact: ...

model: str = field(kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute

structure: Optional[Structure] = field(default=None, kw_only=True) class-attribute instance-attribute

after_run()

Source code in griptape/drivers/audio_transcription/base_audio_transcription_driver.py 
25
26
27
def after_run(self) -> None:
    if self.structure:
        self.structure.publish_event(FinishAudioTranscriptionEvent())

before_run()

Source code in griptape/drivers/audio_transcription/base_audio_transcription_driver.py 
21
22
23
def before_run(self) -> None:
    if self.structure:
        self.structure.publish_event(StartAudioTranscriptionEvent())

run(audio, prompts=None)

Source code in griptape/drivers/audio_transcription/base_audio_transcription_driver.py 
29
30
31
32
33
34
35
36
37
38
39
def run(self, audio: AudioArtifact, prompts: Optional[list[str]] = None) -> TextArtifact:
    for attempt in self.retrying():
        with attempt:
            self.before_run()
            result = self.try_run(audio, prompts)
            self.after_run()

            return result

    else:
        raise Exception("Failed to run audio transcription")

try_run(audio, prompts=None) abstractmethod

Source code in griptape/drivers/audio_transcription/base_audio_transcription_driver.py 
41
42
@abstractmethod
def try_run(self, audio: AudioArtifact, prompts: Optional[list[str]] = None) -> TextArtifact: ...