From 400eb28a91b90da9f9410014398324e3f4d221fe Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 11 Dec 2024 00:50:40 -0800 Subject: [PATCH] Code Quality Improvement - move `aleph_alpha` to deprecated_providers (#7168) * move aleph alpha to deprecated providers * fix import location * fix aleph_alpha * pytest skip * undo change to test file --- litellm/__init__.py | 2 +- litellm/llms/{ => deprecated_providers}/aleph_alpha.py | 0 litellm/main.py | 4 ++-- 3 files changed, 3 insertions(+), 3 deletions(-) rename litellm/llms/{ => deprecated_providers}/aleph_alpha.py (100%) diff --git a/litellm/__init__.py b/litellm/__init__.py index c53fddedf8d..37c9f22b57e 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -1083,7 +1083,7 @@ from .llms.deprecated_providers.palm import ( PalmConfig, ) # here to prevent breaking changes from .llms.nlp_cloud.chat.handler import NLPCloudConfig -from .llms.aleph_alpha import AlephAlphaConfig +from .llms.deprecated_providers.aleph_alpha import AlephAlphaConfig from .llms.petals import PetalsConfig from .llms.vertex_ai.gemini.vertex_and_google_ai_studio_gemini import ( VertexGeminiConfig, diff --git a/litellm/llms/aleph_alpha.py b/litellm/llms/deprecated_providers/aleph_alpha.py similarity index 100% rename from litellm/llms/aleph_alpha.py rename to litellm/llms/deprecated_providers/aleph_alpha.py diff --git a/litellm/main.py b/litellm/main.py index 43158331d3c..458018791cb 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -83,7 +83,7 @@ from litellm.utils import ( from ._logging import verbose_logger from .caching.caching import disable_cache, enable_cache, update_cache from .litellm_core_utils.streaming_chunk_builder_utils import ChunkProcessor -from .llms import aleph_alpha, baseten, maritalk, ollama_chat, petals +from .llms import baseten, maritalk, ollama_chat, petals from .llms.anthropic.chat import AnthropicChatCompletion from .llms.azure.audio_transcriptions import AzureAudioTranscription from .llms.azure.azure import AzureChatCompletion, _check_dynamic_azure_params @@ -98,7 +98,7 @@ from .llms.custom_httpx.llm_http_handler import BaseLLMHTTPHandler from .llms.custom_llm import CustomLLM, custom_chat_llm_router from .llms.databricks.chat.handler import DatabricksChatCompletion from .llms.databricks.embed.handler import DatabricksEmbeddingHandler -from .llms.deprecated_providers import palm +from .llms.deprecated_providers import palm, aleph_alpha from .llms.groq.chat.handler import GroqChatCompletion from .llms.huggingface.chat.handler import Huggingface from .llms.nlp_cloud.chat.handler import completion as nlp_cloud_chat_completion