Add AzureOpenAiTextToSpeech driver in config

griptape-ai · Sep 12, 2024 · 4a05106 · 4a05106
1 parent c6d2f9e
commit 4a05106
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/griptape/configs/drivers/azure_openai_drivers_config.py b/griptape/configs/drivers/azure_openai_drivers_config.py
@@ -10,6 +10,7 @@
     AzureOpenAiEmbeddingDriver,
     AzureOpenAiImageGenerationDriver,
     AzureOpenAiImageQueryDriver,
+    AzureOpenAiTextToSpeechDriver,
     LocalVectorStoreDriver,
 )
 from griptape.utils.decorators import lazy_property
@@ -92,3 +93,13 @@ def vector_store_driver(self) -> LocalVectorStoreDriver:
                 azure_ad_token_provider=self.azure_ad_token_provider,
             )
         )
+
+    @lazy_property()
+    def text_to_speech_driver(self) -> AzureOpenAiTextToSpeechDriver:
+        return AzureOpenAiTextToSpeechDriver(
+            model="tts",
+            azure_endpoint=self.azure_endpoint,
+            api_key=self.api_key,
+            azure_ad_token=self.azure_ad_token,
+            azure_ad_token_provider=self.azure_ad_token_provider,
+        )