diff --git a/internal/registry/model_registry.go b/internal/registry/model_registry.go index dba10d9d7..b52e48b24 100644 --- a/internal/registry/model_registry.go +++ b/internal/registry/model_registry.go @@ -237,6 +237,12 @@ func (r *ModelRegistry) RegisterClient(clientID, clientProvider string, models [ if reg, ok := r.models[id]; ok { reg.Info = cloneModelInfo(model) reg.LastUpdated = now + if reg.QuotaExceededClients != nil { + delete(reg.QuotaExceededClients, clientID) + } + if reg.SuspendedClients != nil { + delete(reg.SuspendedClients, clientID) + } if providerChanged && provider != "" { if _, newlyAdded := addedSet[id]; newlyAdded { continue