From c5026591be3fff744bfcdec22add52787655f878 Mon Sep 17 00:00:00 2001 From: xxxxibo Date: Tue, 16 Jan 2024 21:57:58 +0800 Subject: [PATCH] fix:update displayname in modelserver --- pkg/worker/worker.go | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/pkg/worker/worker.go b/pkg/worker/worker.go index 4892daaa6..4b1451a2a 100644 --- a/pkg/worker/worker.go +++ b/pkg/worker/worker.go @@ -19,6 +19,7 @@ package worker import ( "context" "fmt" + "reflect" "github.com/pkg/errors" appsv1 "k8s.io/api/apps/v1" @@ -341,7 +342,11 @@ func (podWorker *PodWorker) BeforeStart(ctx context.Context) error { } } case Update: - // Skip update when found + embedder1 := podWorker.Worker().BuildEmbedder() + if !reflect.DeepEqual(embedder.Spec, embedder1.Spec) { + embedder.Spec = embedder1.Spec + return podWorker.c.Update(ctx, embedder) + } case Panic: return err } @@ -364,7 +369,11 @@ func (podWorker *PodWorker) BeforeStart(ctx context.Context) error { } } case Update: - // Skip update when found + llm1 := podWorker.Worker().BuildLLM() + if !reflect.DeepEqual(llm.Spec, llm1.Spec) { + llm.Spec = llm1.Spec + return podWorker.c.Update(ctx, llm) + } case Panic: return err }