forked from phoenix/litellm-mirror
(test) no cache hit
This commit is contained in:
parent
553ac19f9f
commit
07fe08d8b5
1 changed files with 47 additions and 0 deletions
|
@ -926,6 +926,53 @@ def test_cache_override():
|
||||||
# test_cache_override()
|
# test_cache_override()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_cache_control_overrides():
|
||||||
|
# we use the cache controls to ensure there is no cache hit on this test
|
||||||
|
litellm.cache = Cache(
|
||||||
|
type="redis",
|
||||||
|
host=os.environ["REDIS_HOST"],
|
||||||
|
port=os.environ["REDIS_PORT"],
|
||||||
|
password=os.environ["REDIS_PASSWORD"],
|
||||||
|
)
|
||||||
|
print("Testing cache override")
|
||||||
|
litellm.set_verbose = True
|
||||||
|
import uuid
|
||||||
|
|
||||||
|
unique_num = str(uuid.uuid4())
|
||||||
|
|
||||||
|
start_time = time.time()
|
||||||
|
|
||||||
|
response1 = await litellm.acompletion(
|
||||||
|
model="gpt-3.5-turbo",
|
||||||
|
messages=[
|
||||||
|
{
|
||||||
|
"role": "user",
|
||||||
|
"content": "hello who are you" + unique_num,
|
||||||
|
}
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
print(response1)
|
||||||
|
|
||||||
|
await asyncio.sleep(2)
|
||||||
|
|
||||||
|
response2 = await litellm.acompletion(
|
||||||
|
model="gpt-3.5-turbo",
|
||||||
|
messages=[
|
||||||
|
{
|
||||||
|
"role": "user",
|
||||||
|
"content": "hello who are you" + unique_num,
|
||||||
|
}
|
||||||
|
],
|
||||||
|
cache={"no-cache": True},
|
||||||
|
)
|
||||||
|
|
||||||
|
print(response2)
|
||||||
|
|
||||||
|
assert response1.id != response2.id
|
||||||
|
|
||||||
|
|
||||||
def test_custom_redis_cache_params():
|
def test_custom_redis_cache_params():
|
||||||
# test if we can init redis with **kwargs
|
# test if we can init redis with **kwargs
|
||||||
try:
|
try:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue