diff --git a/pkg/modelregistry/model_registry.go b/pkg/modelregistry/model_registry.go index 5dacd99..73beec9 100644 --- a/pkg/modelregistry/model_registry.go +++ b/pkg/modelregistry/model_registry.go @@ -28,6 +28,10 @@ func NewModelRegistry() *DefaultModelRegistry { } } +func GetDefaultRegistry() *DefaultModelRegistry { + return defaultRegistry +} + func SetDefaultRegistry(registry *DefaultModelRegistry) { registriesMutex.Lock() defer registriesMutex.Unlock()