From a307917b668635bf29cf51bd85dad6f4fa92d9d9 Mon Sep 17 00:00:00 2001 From: Liujian <824010343@qq.com> Date: Tue, 31 Dec 2024 14:48:19 +0800 Subject: [PATCH] fix: provider status incorrect --- module/ai/dto/enum.go | 10 +++++----- module/ai/dto/output.go | 3 ++- module/ai/iml.go | 31 ++++++++++++++++--------------- 3 files changed, 23 insertions(+), 21 deletions(-) diff --git a/module/ai/dto/enum.go b/module/ai/dto/enum.go index 3c1b96a4..2cc76b6f 100644 --- a/module/ai/dto/enum.go +++ b/module/ai/dto/enum.go @@ -13,9 +13,9 @@ func (p ProviderStatus) Int() int { case ProviderAbnormal: return 2 case ProviderEnabled: - return 0 - case ProviderDisabled: return 1 + case ProviderDisabled: + return 0 default: return 1 } @@ -30,10 +30,10 @@ func ToProviderStatus(status int) ProviderStatus { case 2: return ProviderAbnormal case 0: - return ProviderEnabled + return ProviderDisabled case 1: - return ProviderDisabled + return ProviderEnabled default: - return ProviderDisabled + return ProviderEnabled } } diff --git a/module/ai/dto/output.go b/module/ai/dto/output.go index 70e1af54..d41687a5 100644 --- a/module/ai/dto/output.go +++ b/module/ai/dto/output.go @@ -17,10 +17,11 @@ type Provider struct { Name string `json:"name"` Config string `json:"config"` GetAPIKeyUrl string `json:"get_apikey_url"` - DefaultLLM string `json:"defaultLLM"` + DefaultLLM string `json:"default_llm"` DefaultLLMConfig string `json:"-"` Priority int `json:"priority"` Status ProviderStatus `json:"status"` + Configured bool `json:"configured"` } type ConfiguredProviderItem struct { diff --git a/module/ai/iml.go b/module/ai/iml.go index 0eacef1d..b68ed21a 100644 --- a/module/ai/iml.go +++ b/module/ai/iml.go @@ -391,6 +391,7 @@ func (i *imlProviderModule) Provider(ctx context.Context, id string) (*ai_dto.Pr DefaultLLMConfig: defaultLLM.DefaultConfig(), Priority: info.Priority, Status: ai_dto.ProviderEnabled, + Configured: true, }, nil } @@ -589,21 +590,21 @@ func (i *imlProviderModule) UpdateProviderConfig(ctx context.Context, id string, log.Errorf("unmarshal ai provider config error,id is %s,err is %v", id, err) return err } - - return i.syncGateway(ctx, cluster.DefaultClusterID, []*gateway.DynamicRelease{ - { - BasicItem: &gateway.BasicItem{ - ID: id, - Description: info.Name, - Resource: id, - Version: info.UpdateAt.Format("20060102150405"), - MatchLabels: map[string]string{ - "module": "ai-provider", - }, - }, - Attr: cfg, - }, newAIUpstream(id, p.URI()), - }, true) + return nil + //return i.syncGateway(ctx, cluster.DefaultClusterID, []*gateway.DynamicRelease{ + // { + // BasicItem: &gateway.BasicItem{ + // ID: id, + // Description: info.Name, + // Resource: id, + // Version: info.UpdateAt.Format("20060102150405"), + // MatchLabels: map[string]string{ + // "module": "ai-provider", + // }, + // }, + // Attr: cfg, + // }, newAIUpstream(id, p.URI()), + //}, true) }) }