@@ -48,7 +48,7 @@ def test_clear_and_restore():
4848 log = LogReader (server .log_path )
4949
5050 # verify feature is enabled
51- assert "__TEST_TAG_CLEAR_IDLE_ENABLED__ " in log .drain ()
51+ assert "__TEST_TAG_CACHE_IDLE_SLOTS_ENABLED__ " in log .drain ()
5252
5353 res = server .make_request ("POST" , "/completion" , data = {
5454 "prompt" : LONG_PROMPT ,
@@ -59,7 +59,7 @@ def test_clear_and_restore():
5959 original_prompt_n = res .body ["timings" ]["prompt_n" ]
6060
6161 # Slot 0 is the only slot with KV — should NOT be cleared
62- assert "__TEST_TAG_CLEAR_IDLE_SLOT__ " not in log .drain ()
62+ assert "__TEST_TAG_CACHE_IDLE_SLOT__ " not in log .drain ()
6363
6464 # Launching slot 1 clears idle slot 0
6565 res = server .make_request ("POST" , "/completion" , data = {
@@ -68,7 +68,7 @@ def test_clear_and_restore():
6868 "cache_prompt" : True ,
6969 })
7070 assert res .status_code == 200
71- assert "__TEST_TAG_CLEAR_IDLE_SLOT__ " in log .drain ()
71+ assert "__TEST_TAG_CACHE_IDLE_SLOT__ " in log .drain ()
7272
7373 # Re-send same prompt — should restore from cache-ram
7474 res = server .make_request ("POST" , "/completion" , data = {
@@ -86,7 +86,7 @@ def test_clear_and_restore():
8686 "cache_prompt" : True ,
8787 })
8888 assert res .status_code == 200
89- assert "__TEST_TAG_CLEAR_IDLE_SLOT__ " not in log .drain ()
89+ assert "__TEST_TAG_CACHE_IDLE_SLOT__ " not in log .drain ()
9090
9191
9292def test_disabled_with_flag ():
@@ -96,7 +96,7 @@ def test_disabled_with_flag():
9696 log = LogReader (server .log_path )
9797
9898 # Feature should not be enabled
99- assert "__TEST_TAG_CLEAR_IDLE_ENABLED__ " not in log .drain ()
99+ assert "__TEST_TAG_CACHE_IDLE_SLOTS_ENABLED__ " not in log .drain ()
100100
101101 res = server .make_request ("POST" , "/completion" , data = {
102102 "prompt" : LONG_PROMPT ,
@@ -112,4 +112,4 @@ def test_disabled_with_flag():
112112 "cache_prompt" : True ,
113113 })
114114 assert res .status_code == 200
115- assert "__TEST_TAG_CLEAR_IDLE_SLOT__ " not in log .drain ()
115+ assert "__TEST_TAG_CACHE_IDLE_SLOT__ " not in log .drain ()
0 commit comments