scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-13 23:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

test(complexity_router): derive the warm-pick record key from the shared caller scope

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
tin 2026-07-30 05:38:11 +00:00 • committed by Tin Chi Lo
parent e044e70189
commit aea92299f4
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_litellm/router_strategy/test_complexity_router.py
View file

@ -4218,6 +4218,7 @@ class TestWarmAwarePick:
import json
import time as time_module
from litellm.litellm_core_utils.core_helpers import get_caller_scope
from litellm.router_strategy.complexity_router.cache_warming.store import CacheWarmingStore
from litellm.router_strategy.complexity_router.cache_warming.types import (
CACHE_WARMING_RECORD_SCHEMA_VERSION,
@ -4245,7 +4246,7 @@ class TestWarmAwarePick:
auto_router_model_name="warm-router",
)
store = CacheWarmingStore(redis_cache=redis, auto_router_model_name="warm-router")
key = store.record_key("warm-router", "hash-w", "warm-sess")
key = store.record_key("warm-router", get_caller_scope(TestWarmAwarePick._kwargs()), "warm-sess")
redis.hashes.setdefault(store.sessions_key(), {})[key] = json.dumps(record.model_dump())
for model_group, stamp in warmth.items():
redis.data[CacheWarmingStore.warmth_key(key, model_group)] = json.dumps(