From 66b6a0d8350dfb988727fc3011c19db24d0fcd00 Mon Sep 17 00:00:00 2001
From: Hein <hein@bitechsystems.co.za>
Date: Wed, 19 Nov 2025 18:29:24 +0200
Subject: [PATCH] Better registry handling

---
 pkg/modelregistry/model_registry.go | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/pkg/modelregistry/model_registry.go b/pkg/modelregistry/model_registry.go
index fc9b1b1..2df0bea 100644
--- a/pkg/modelregistry/model_registry.go
+++ b/pkg/modelregistry/model_registry.go
@@ -29,7 +29,23 @@ func NewModelRegistry() *DefaultModelRegistry {
 }
 
 func SetDefaultRegistry(registry *DefaultModelRegistry) {
+	registriesMutex.Lock()
+	foundAt := -1
+	for idx, r := range registries {
+		if r == defaultRegistry {
+			foundAt = idx
+			break
+		}
+	}
 	defaultRegistry = registry
+	if foundAt >= 0 {
+		registries[foundAt] = registry
+	} else {
+		registries = append([]*DefaultModelRegistry{registry}, registries...)
+	}
+
+	defer registriesMutex.Unlock()
+
 }
 
 // AddRegistry adds a registry to the global list of registries