fix test and rename var

Author: tylerhutcherson

redisvl/extensions/llmcache/semantic.py

@@ -185,8 +185,8 @@ def delete(self) -> None:
 
     def _refresh_ttl(self, key: str) -> None:
         """Refresh the time-to-live for the specified key."""
-        if self.ttl:
-            self._index.client.expire(key, self.ttl)  # type: ignore
+        if self._ttl:
+            self._index.client.expire(key, self._ttl)  # type: ignore
 
     def _vectorize_prompt(self, prompt: Optional[str]) -> List[float]:
         """Converts a text prompt to its vector representation using the

tests/integration/test_llmcache.py

@@ -19,7 +19,6 @@ def cache(vectorizer, redis_url):
         vectorizer=vectorizer, distance_threshold=0.2, redis_url=redis_url
     )
     yield cache_instance
-    cache_instance.clear()  # Clear cache after each test
     cache_instance._index.delete(True)  # Clean up index
 
 
@@ -37,7 +36,6 @@ def cache_with_ttl(vectorizer, redis_url):
         vectorizer=vectorizer, distance_threshold=0.2, ttl=2, redis_url=redis_url
     )
     yield cache_instance
-    cache_instance.clear()  # Clear cache after each test
     cache_instance._index.delete(True)  # Clean up index
 
 
@@ -54,7 +52,7 @@ def cache_with_redis_client(vectorizer, client, redis_url):
     cache_instance._index.delete(True)  # Clean up index
 
 
-# Test handling invalid input for check method
+# # Test handling invalid input for check method
 def test_bad_ttl(cache):
     with pytest.raises(ValueError):
         cache.set_ttl(2.5)
@@ -117,18 +115,13 @@ def test_ttl_expiration_after_update(cache_with_ttl, vectorizer):
     prompt = "This is a test prompt."
     response = "This is a test response."
     vector = vectorizer.embed(prompt)
-    cache_with_ttl.set_ttl(5)
+    cache_with_ttl.set_ttl(4)
 
-    cache_with_ttl.store(prompt, response, vector=vector)
-    sleep(2)
-
-    check_result = cache_with_ttl.check(vector=vector)
-    assert len(check_result) == 1
-    print(check_result, flush=True)
-    assert response == check_result[0]["response"]
-    assert "metadata" not in check_result[0]
+    assert cache_with_ttl.ttl == 4
 
+    cache_with_ttl.store(prompt, response, vector=vector)
     sleep(5)
+
     check_result = cache_with_ttl.check(vector=vector)
     assert len(check_result) == 0