mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-04 12:07:34 +00:00
Added an update to the registration function to reject a change for the same provider with a message asking the caller to unregister the model first
This commit is contained in:
parent
d350e3662b
commit
f0545a3fec
2 changed files with 9 additions and 2 deletions
|
@ -98,7 +98,10 @@ class DiskDistributionRegistry(DistributionRegistry):
|
|||
existing_obj = await self.get(obj.type, obj.identifier)
|
||||
# dont register if the object's providerid already exists
|
||||
if existing_obj and existing_obj.provider_id == obj.provider_id:
|
||||
return False
|
||||
raise ValueError(
|
||||
f"Model '{obj.identifier}' is already registered with provider '{obj.provider_id}'. "
|
||||
f"Please unregister the existing model first before registering a new one."
|
||||
)
|
||||
|
||||
await self.kvstore.set(
|
||||
KEY_FORMAT.format(type=obj.type, identifier=obj.identifier),
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue