diff --git a/llama_stack/providers/utils/kvstore/sqlite/sqlite.py b/llama_stack/providers/utils/kvstore/sqlite/sqlite.py index 7c816dc48..a9a7a1304 100644 --- a/llama_stack/providers/utils/kvstore/sqlite/sqlite.py +++ b/llama_stack/providers/utils/kvstore/sqlite/sqlite.py @@ -65,7 +65,7 @@ class SqliteKVStoreImpl(KVStore): ) await db.commit() - async def close(self): + async def shutdown(self): """Close the persistent connection (only for in-memory databases).""" if self._conn: await self._conn.close() diff --git a/tests/unit/utils/kvstore/test_sqlite_memory.py b/tests/unit/utils/kvstore/test_sqlite_memory.py index 942ad1087..a31377306 100644 --- a/tests/unit/utils/kvstore/test_sqlite_memory.py +++ b/tests/unit/utils/kvstore/test_sqlite_memory.py @@ -17,7 +17,7 @@ async def test_memory_kvstore_persistence_behavior(): store1 = SqliteKVStoreImpl(config) await store1.initialize() await store1.set("persist_test", "should_not_persist") - await store1.close() + await store1.shutdown() # Second instance with same config store2 = SqliteKVStoreImpl(config) @@ -27,4 +27,4 @@ async def test_memory_kvstore_persistence_behavior(): result = await store2.get("persist_test") assert result is None - await store2.close() + await store2.shutdown()