Updated utils based on greptile feedback

This commit is contained in:
gorkii 2026-04-07 13:43:25 -04:00
parent aa9e55f1bc
commit 9b7f4db2d3
2 changed files with 8 additions and 0 deletions

View file

@ -3219,6 +3219,7 @@ class LlmProviders(str, Enum):
DASHSCOPE = "dashscope"
MOONSHOT = "moonshot"
PUBLICAI = "publicai"
SALAD_CLOUD = "salad_cloud"
V0 = "v0"
MORPH = "morph"
LAMBDA_AI = "lambda_ai"

View file

@ -19,6 +19,13 @@ import litellm
class TestSaladCloudProviderConfig:
"""Test SaladCloud provider configuration"""
def test_salad_cloud_in_provider_list(self):
"""Test that salad_cloud is in the provider list via LlmProviders enum"""
from litellm import LlmProviders
assert LlmProviders.SALAD_CLOUD.value == "salad_cloud"
assert "salad_cloud" in litellm.provider_list
def test_salad_cloud_json_config_exists(self):
"""Test that salad_cloud is configured in providers.json"""
from litellm.llms.openai_like.json_loader import JSONProviderRegistry