From bd4f63eebf16dd2429dd13d1e41bbd332ffe0137 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 27 Aug 2024 19:35:03 -0700 Subject: [PATCH] fix(__init__.py): fix import --- litellm/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/litellm/__init__.py b/litellm/__init__.py index 591d5873cf..a3727eb2e8 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -862,6 +862,10 @@ from .llms.vertex_ai_and_google_ai_studio.vertex_ai_anthropic import ( from .llms.vertex_ai_and_google_ai_studio.vertex_ai_partner_models.llama3.transformation import ( VertexAILlama3Config, ) +from .llms.vertex_ai_and_google_ai_studio.vertex_ai_partner_models.ai21.transformation import ( + VertexAIAi21Config, +) + from .llms.sagemaker.sagemaker import SagemakerConfig from .llms.ollama import OllamaConfig from .llms.ollama_chat import OllamaChatConfig