scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

test: fix test to call constructor directly instead of __new__

Browse source
This commit is contained in:
PRABHU KIRAN VANDRANKI 2026-04-18 13:27:47 -04:00
parent f387fba8da
commit 35be31576f
1 changed files with 7 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
tests/local_testing/test_redis_semantic_cache_unit.py
View file

@ -9,7 +9,7 @@ Verifies that:
import sys
import os
sys.path.insert(0, os.path.abspath("../.."))
sys.path.insert(0, os.path.abspath("../..."))
import pytest
from unittest.mock import patch, MagicMock
@ -29,18 +29,12 @@ class TestRedisSemanticCacheInitDegradation:
),
patch("litellm.caching.redis_semantic_cache.SemanticCache"),
):
with pytest.raises(RuntimeError) as exc_info:
cache = RedisSemanticCache.__new__(RedisSemanticCache)
# Call only the vectorizer section by patching imports
cache.embedding_model = "text-embedding-ada-002"
cache.distance_threshold = 0.1
cache.similarity_threshold = 0.9
# Simulate the init path that wraps CustomTextVectorizer
from redisvl.utils.vectorize import CustomTextVectorizer # noqa: F401
# Verify RuntimeError is raised (the patch above raises ValueError which
# gets wrapped). We test the wrapping logic directly below.
with pytest.raises(RuntimeError, match="embedding model validation failed"):
RedisSemanticCache(
redis_url="redis://localhost:6379",
similarity_threshold=0.9,
embedding_model="text-embedding-ada-002",
)
def test_runtime_error_message_contains_original_error(self):
"""RuntimeError message must include the original exception details."""