From 97b86226c79123b7718bff1a6022dd395f6c3ff8 Mon Sep 17 00:00:00 2001 From: Raghotham Murthy Date: Sat, 11 Oct 2025 17:58:08 -0700 Subject: [PATCH] change close to shutdown --- llama_stack/providers/utils/kvstore/sqlite/sqlite.py | 2 +- tests/unit/utils/kvstore/test_sqlite_memory.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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()