From 95786d5bdce394d6d262b5f4a33c96feabe16306 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Fri, 31 Jan 2025 15:29:55 -0800 Subject: [PATCH] Update client-sdk test config option handling Fix test --- llama_stack/providers/inline/vector_io/faiss/faiss.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/llama_stack/providers/inline/vector_io/faiss/faiss.py b/llama_stack/providers/inline/vector_io/faiss/faiss.py index 69a7eef7c..448618811 100644 --- a/llama_stack/providers/inline/vector_io/faiss/faiss.py +++ b/llama_stack/providers/inline/vector_io/faiss/faiss.py @@ -179,6 +179,10 @@ class FaissVectorIOImpl(VectorIO, VectorDBsProtocolPrivate): return [i.vector_db for i in self.cache.values()] async def unregister_vector_db(self, vector_db_id: str) -> None: + if vector_db_id not in self.cache: + logger.warning(f"Vector DB {vector_db_id} not found") + return + await self.cache[vector_db_id].index.delete() del self.cache[vector_db_id] await self.kvstore.delete(f"{VECTOR_DBS_PREFIX}{vector_db_id}")