From 428f8dcb7af8977a12ff108f779bbb67d2ef3016 Mon Sep 17 00:00:00 2001
From: xprilion <xprilion@gmail.com>
Date: Thu, 11 Sep 2025 00:19:31 +0530
Subject: [PATCH] tinyfix

---
 litellm/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/__init__.py b/litellm/__init__.py
index 5024925075..fabc8e280f 100644
--- a/litellm/__init__.py
+++ b/litellm/__init__.py
@@ -60,7 +60,7 @@ from litellm.constants import (
     empower_models,
     together_ai_models,
     baseten_models,
-    wandb_models,
+    WANDB_MODELS,
     REPEATED_STREAMING_CHUNK_LIMIT,
     request_timeout,
     open_ai_embedding_models,
@@ -522,7 +522,7 @@ cometapi_models: Set = set()
 oci_models: Set = set()
 vercel_ai_gateway_models: Set = set()
 volcengine_models: Set = set()
-wandb_models: Set = set()
+wandb_models: Set = set(WANDB_MODELS)
 
 
 def is_bedrock_pricing_only_model(key: str) -> bool: