From 2d4775c67a0bd5ae2db6b20431970f00ba8c3dfc Mon Sep 17 00:00:00 2001 From: Omar Abdelwahab Date: Tue, 30 Sep 2025 17:12:43 -0700 Subject: [PATCH] updated the error message and test_registry --- llama_stack/core/store/registry.py | 2 +- tests/unit/registry/test_registry.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/llama_stack/core/store/registry.py b/llama_stack/core/store/registry.py index 9f4d774e8..4c34f52e1 100644 --- a/llama_stack/core/store/registry.py +++ b/llama_stack/core/store/registry.py @@ -117,7 +117,7 @@ class DiskDistributionRegistry(DistributionRegistry): if existing_obj and existing_obj.provider_id == obj.provider_id: raise ValueError( f"{obj.type.title()} '{obj.identifier}' is already registered with provider '{obj.provider_id}'. " - f"Please unregister the existing {obj.type} first before registering a new one." + f"Unregister the existing object first before registering a new one." ) await self.kvstore.set( diff --git a/tests/unit/registry/test_registry.py b/tests/unit/registry/test_registry.py index 83dcf0968..e39a421d5 100644 --- a/tests/unit/registry/test_registry.py +++ b/tests/unit/registry/test_registry.py @@ -136,7 +136,8 @@ async def test_duplicate_provider_registration(cached_disk_dist_registry): # Now we expect a ValueError to be raised for duplicate registration with pytest.raises( - ValueError, match=r"Vector_db.*already registered.*provider.*baz.*" + ValueError, + match=r"Vector_Db.*already registered.*provider.*baz.*Unregister the existing", ): await cached_disk_dist_registry.register(duplicate_vector_db)