phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-12 20:12:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

change close to shutdown

Browse source
This commit is contained in:
Raghotham Murthy 2025-10-11 17:58:08 -07:00
parent 06d02bf3de
commit 97b86226c7
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/utils/kvstore/sqlite/sqlite.py
View file

@ -65,7 +65,7 @@ class SqliteKVStoreImpl(KVStore):
) )
await db.commit() await db.commit()
async def close(self): async def shutdown(self):
"""Close the persistent connection (only for in-memory databases).""" """Close the persistent connection (only for in-memory databases)."""
if self._conn: if self._conn:
await self._conn.close() await self._conn.close()

4
tests/unit/utils/kvstore/test_sqlite_memory.py
View file

@ -17,7 +17,7 @@ async def test_memory_kvstore_persistence_behavior():
store1 = SqliteKVStoreImpl(config) store1 = SqliteKVStoreImpl(config)
await store1.initialize() await store1.initialize()
await store1.set("persist_test", "should_not_persist") await store1.set("persist_test", "should_not_persist")
await store1.close() await store1.shutdown()
# Second instance with same config # Second instance with same config
store2 = SqliteKVStoreImpl(config) store2 = SqliteKVStoreImpl(config)
@ -27,4 +27,4 @@ async def test_memory_kvstore_persistence_behavior():
result = await store2.get("persist_test") result = await store2.get("persist_test")
assert result is None assert result is None
await store2.close() await store2.shutdown()