fix

langchain-ai · Mar 13, 2024 · b65e554 · b65e554
1 parent 33b4e6f
commit b65e554
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/astradb/tests/integration_tests/test_caches.py b/libs/astradb/tests/integration_tests/test_caches.py
@@ -135,7 +135,7 @@ async def async_astradb_cache() -> AsyncIterator[AstraDBCache]:
         setup_mode=SetupMode.ASYNC,
     )
     yield cache
-    await aretry(cache.collection.astra_db.delete_collection, args=["lc_integration_test_cache_async"])
+    await aretry(cache.async_collection.astra_db.delete_collection, args=["lc_integration_test_cache_async"])
 
 
 @pytest.fixture(scope="module")
@@ -166,7 +166,7 @@ async def async_astradb_semantic_cache() -> AsyncIterator[AstraDBSemanticCache]:
     )
     yield sem_cache
     await aretry(sem_cache.async_collection.astra_db.delete_collection,
-          args=["lc_integration_test_sem_cache_async"])
+                 args=["lc_integration_test_sem_cache_async"])
 
 
 @pytest.mark.skipif(not _has_env_vars(), reason="Missing Astra DB env. vars")