From 006c955aaecf87cb4eab27c48e970c0c64b34536 Mon Sep 17 00:00:00 2001 From: Francisco Javier Arceo Date: Thu, 10 Jul 2025 14:08:03 -0400 Subject: [PATCH] updating _load_openai_vector_stores Signed-off-by: Francisco Javier Arceo --- .../providers/inline/vector_io/sqlite_vec/sqlite_vec.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/llama_stack/providers/inline/vector_io/sqlite_vec/sqlite_vec.py b/llama_stack/providers/inline/vector_io/sqlite_vec/sqlite_vec.py index 0a6f69730..6acd85c56 100644 --- a/llama_stack/providers/inline/vector_io/sqlite_vec/sqlite_vec.py +++ b/llama_stack/providers/inline/vector_io/sqlite_vec/sqlite_vec.py @@ -514,8 +514,12 @@ class SQLiteVecVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorDBsProtoc assert self.kvstore is not None start_key = OPENAI_VECTOR_STORES_PREFIX end_key = f"{OPENAI_VECTOR_STORES_PREFIX}\xff" - stored = await self.kvstore.values_in_range(start_key, end_key) - return {json.loads(s)["id"]: json.loads(s) for s in stored} + stored_openai_stores = await self.kvstore.values_in_range(start_key, end_key) + stores = {} + for store_data in stored_openai_stores: + store_info = json.loads(store_data) + stores[store_info["id"]] = store_info + return stores async def _update_openai_vector_store(self, store_id: str, store_info: dict[str, Any]) -> None: """Update vector store metadata in SQLite database."""