From 1e4ae7b4c7a23a12649a90fe4893f3b20d319fdd Mon Sep 17 00:00:00 2001
From: sunanzhi <sunanzhi@hotmail.com>
Date: Mon, 10 Mar 2025 17:41:04 +0800
Subject: [PATCH] fix: resolve null pointer caused by URI error

---
 module/ai/iml.go | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/module/ai/iml.go b/module/ai/iml.go
index c59c165a..bfecd39b 100644
--- a/module/ai/iml.go
+++ b/module/ai/iml.go
@@ -105,7 +105,9 @@ func (i *imlProviderModule) OnInit() {
 				}
 			} else {
 				provider, _ := model_runtime.NewCustomizeProvider(p.Id, p.Name, iModels, p.DefaultLLM, p.Config)
-				model_runtime.Register(p.Id, provider)
+				if provider != nil {
+					model_runtime.Register(p.Id, provider)
+				}
 			}
 		}
 		i.transaction.Transaction(ctx, func(ctx context.Context) error {
@@ -644,8 +646,11 @@ func (i *imlProviderModule) UpdateProviderConfig(ctx context.Context, id string,
 		}
 		// customize provider
 		if info.Type == 1 {
-			if uri, uriErr := model_runtime.GetCustomizeProviderURI(input.Config, false); uriErr != nil {
+			uri, uriErr := model_runtime.GetCustomizeProviderURI(input.Config, false)
+			if uriErr != nil {
 				p.SetURI(uri)
+			} else {
+				return uriErr
 			}
 		}
 		if *pInfo.Status == 0 {