From 00a478fb4d5b42977354fc1ea598e3b15fb4c611 Mon Sep 17 00:00:00 2001 From: Sarveswaran MG Date: Wed, 22 Jul 2026 20:51:30 +0530 Subject: [PATCH] fix(vector_stores): keep param name for backward compat, correct index deletion --- litellm/vector_stores/vector_store_registry.py | 6 ++++-- .../vector_stores/test_vector_store_registry.py | 4 +--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/litellm/vector_stores/vector_store_registry.py b/litellm/vector_stores/vector_store_registry.py index 520edc06548..3a435af59ad 100644 --- a/litellm/vector_stores/vector_store_registry.py +++ b/litellm/vector_stores/vector_store_registry.py @@ -55,11 +55,13 @@ class VectorStoreIndexRegistry: return self.vector_store_indexes.append(vector_store_index) - def delete_vector_store_index(self, index_name: str): + def delete_vector_store_index(self, vector_store_index: str): """ Deletes a vector store index from the registry """ - self.vector_store_indexes = [index for index in self.vector_store_indexes if index.index_name != index_name] + self.vector_store_indexes = [ + index for index in self.vector_store_indexes if index.index_name != vector_store_index + ] def is_vector_store_index(self, vector_store_index_name: str) -> bool: """ diff --git a/tests/test_litellm/vector_stores/test_vector_store_registry.py b/tests/test_litellm/vector_stores/test_vector_store_registry.py index d9337d32c13..70a38de9317 100644 --- a/tests/test_litellm/vector_stores/test_vector_store_registry.py +++ b/tests/test_litellm/vector_stores/test_vector_store_registry.py @@ -267,9 +267,7 @@ def test_delete_vector_store_index_removes_by_name(): Regression test: delete_vector_store_index compared each stored index object against the index-name string, which is never equal, so the delete was an unconditional no-op. """ - registry = VectorStoreIndexRegistry( - [_managed_vector_store_index("idx_a"), _managed_vector_store_index("idx_b")] - ) + registry = VectorStoreIndexRegistry([_managed_vector_store_index("idx_a"), _managed_vector_store_index("idx_b")]) registry.delete_vector_store_index("idx_a")