phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-03 19:57:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

updated the error message and test_registry

Browse source
This commit is contained in:
Omar Abdelwahab 2025-09-30 17:12:43 -07:00
parent 5943ce404b
commit 2d4775c67a
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/core/store/registry.py
View file

@ -117,7 +117,7 @@ class DiskDistributionRegistry(DistributionRegistry):
if existing_obj and existing_obj.provider_id == obj.provider_id: if existing_obj and existing_obj.provider_id == obj.provider_id:
raise ValueError( raise ValueError(
f"{obj.type.title()} '{obj.identifier}' is already registered with provider '{obj.provider_id}'. " 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( await self.kvstore.set(

3
tests/unit/registry/test_registry.py
View file

@ -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 # Now we expect a ValueError to be raised for duplicate registration
with pytest.raises( 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) await cached_disk_dist_registry.register(duplicate_vector_db)