From 7d4a1cb4e2ede3cfda8bab311d4d9ee5b5fe32f2 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 7 Dec 2024 21:46:34 -0800 Subject: [PATCH] refactor - use consistent file naming convention `AI21/` -> `ai21` (#7090) * fix refactor - use consistent file naming convention * ci/cd run again * fix naming structure --- litellm/__init__.py | 6 +++--- litellm/llms/{AI21 => ai21}/chat.py | 0 litellm/llms/{AI21 => ai21}/completion.py | 0 litellm/main.py | 2 +- tests/llm_translation/test_max_completion_tokens.py | 2 +- tests/local_testing/test_completion.py | 2 +- 6 files changed, 6 insertions(+), 6 deletions(-) rename litellm/llms/{AI21 => ai21}/chat.py (100%) rename litellm/llms/{AI21 => ai21}/completion.py (100%) diff --git a/litellm/__init__.py b/litellm/__init__.py index 6427206f1a..8101e1a499 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -1065,8 +1065,8 @@ from .llms.predibase import PredibaseConfig from .llms.replicate import ReplicateConfig from .llms.cohere.completion import CohereConfig from .llms.clarifai import ClarifaiConfig -from .llms.AI21.completion import AI21Config -from .llms.AI21.chat import AI21ChatConfig +from .llms.ai21.completion import AI21Config +from .llms.ai21.chat import AI21ChatConfig from .llms.together_ai.chat import TogetherAIConfig from .llms.cloudflare import CloudflareConfig from .llms.palm import PalmConfig @@ -1158,7 +1158,7 @@ nvidiaNimEmbeddingConfig = NvidiaNimEmbeddingConfig() from .llms.cerebras.chat import CerebrasConfig from .llms.sambanova.chat import SambanovaConfig -from .llms.AI21.chat import AI21ChatConfig +from .llms.ai21.chat import AI21ChatConfig from .llms.fireworks_ai.chat.fireworks_ai_transformation import FireworksAIConfig from .llms.fireworks_ai.embed.fireworks_ai_transformation import ( FireworksAIEmbeddingConfig, diff --git a/litellm/llms/AI21/chat.py b/litellm/llms/ai21/chat.py similarity index 100% rename from litellm/llms/AI21/chat.py rename to litellm/llms/ai21/chat.py diff --git a/litellm/llms/AI21/completion.py b/litellm/llms/ai21/completion.py similarity index 100% rename from litellm/llms/AI21/completion.py rename to litellm/llms/ai21/completion.py diff --git a/litellm/main.py b/litellm/main.py index 17e018c056..3a8159c3f5 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -99,7 +99,7 @@ from .llms import ( replicate, vllm, ) -from .llms.AI21 import completion as ai21 +from .llms.ai21 import completion as ai21 from .llms.anthropic.chat import AnthropicChatCompletion from .llms.anthropic.completion import AnthropicTextCompletion from .llms.azure_ai.chat import AzureAIChatCompletion diff --git a/tests/llm_translation/test_max_completion_tokens.py b/tests/llm_translation/test_max_completion_tokens.py index f0ea002a5f..03eb638757 100644 --- a/tests/llm_translation/test_max_completion_tokens.py +++ b/tests/llm_translation/test_max_completion_tokens.py @@ -220,7 +220,7 @@ def test_all_model_configs(): optional_params={}, ) == {"max_tokens": 10} - from litellm.llms.AI21.chat import AI21ChatConfig + from litellm.llms.ai21.chat import AI21ChatConfig assert "max_completion_tokens" in AI21ChatConfig().get_supported_openai_params( "jamba-1.5-mini@001" diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index 6f42d9a396..e75f643d2f 100644 --- a/tests/local_testing/test_completion.py +++ b/tests/local_testing/test_completion.py @@ -24,7 +24,7 @@ from litellm import RateLimitError, Timeout, completion, completion_cost, embedd from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler from litellm.llms.prompt_templates.factory import anthropic_messages_pt -# litellm.num_retries = 3 +# litellm.num_retries=3 litellm.cache = None litellm.success_callback = []