From 570a5a2825ee5fbd8dad1844c7050bd24e213962 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 30 Aug 2024 11:44:23 -0700 Subject: [PATCH] fix dir structure for tts --- .../text_to_speech/text_to_speech_handler.py} | 0 litellm/main.py | 4 +++- 2 files changed, 3 insertions(+), 1 deletion(-) rename litellm/llms/{text_to_speech/vertex_ai.py => vertex_ai_and_google_ai_studio/text_to_speech/text_to_speech_handler.py} (100%) diff --git a/litellm/llms/text_to_speech/vertex_ai.py b/litellm/llms/vertex_ai_and_google_ai_studio/text_to_speech/text_to_speech_handler.py similarity index 100% rename from litellm/llms/text_to_speech/vertex_ai.py rename to litellm/llms/vertex_ai_and_google_ai_studio/text_to_speech/text_to_speech_handler.py diff --git a/litellm/main.py b/litellm/main.py index 47b51be9f..d77d86058 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -120,7 +120,6 @@ from .llms.prompt_templates.factory import ( ) from .llms.sagemaker.sagemaker import SagemakerLLM from .llms.text_completion_codestral import CodestralTextCompletion -from .llms.text_to_speech.vertex_ai import VertexTextToSpeechAPI from .llms.triton import TritonChatCompletion from .llms.vertex_ai_and_google_ai_studio import ( vertex_ai_anthropic, @@ -135,6 +134,9 @@ from .llms.vertex_ai_and_google_ai_studio.gemini.vertex_and_google_ai_studio_gem from .llms.vertex_ai_and_google_ai_studio.multimodal_embeddings.embedding_handler import ( VertexMultimodalEmbedding, ) +from .llms.vertex_ai_and_google_ai_studio.text_to_speech.text_to_speech_handler import ( + VertexTextToSpeechAPI, +) from .llms.vertex_ai_and_google_ai_studio.vertex_ai_partner_models.main import ( VertexAIPartnerModels, )