From 5f11c49733cc5db7c572491e30932e34f94fd8a6 Mon Sep 17 00:00:00 2001 From: Ryo Date: Tue, 28 Oct 2025 20:48:26 +0800 Subject: [PATCH] fix: resolve issue where model failed to start on TCE due to missing environment variables (#2403) --- backend/bizpkg/config/modelmgr/builtin_conf.go | 5 +++++ backend/bizpkg/config/modelmgr/deprecate_model_get.go | 7 ++++++- backend/bizpkg/llm/modelbuilder/builtin.go | 2 +- 3 files changed, 12 insertions(+), 2 deletions(-) diff --git a/backend/bizpkg/config/modelmgr/builtin_conf.go b/backend/bizpkg/config/modelmgr/builtin_conf.go index 1ecd1c749..89e39fdda 100644 --- a/backend/bizpkg/config/modelmgr/builtin_conf.go +++ b/backend/bizpkg/config/modelmgr/builtin_conf.go @@ -47,6 +47,11 @@ func getOldKnowledgeBuiltinChatModelConfig() *Model { } typeStr := strings.ToUpper(os.Getenv("BUILTIN_CM_TYPE")) + + if typeStr == "" { + return nil + } + baseURLKey := fmt.Sprintf("BUILTIN_CM_%s_BASE_URL", typeStr) apiKeyKey := fmt.Sprintf("BUILTIN_CM_%s_API_KEY", typeStr) modelKey := fmt.Sprintf("BUILTIN_CM_%s_MODEL", typeStr) diff --git a/backend/bizpkg/config/modelmgr/deprecate_model_get.go b/backend/bizpkg/config/modelmgr/deprecate_model_get.go index 1037edda7..2b254ebec 100644 --- a/backend/bizpkg/config/modelmgr/deprecate_model_get.go +++ b/backend/bizpkg/config/modelmgr/deprecate_model_get.go @@ -63,7 +63,9 @@ func initOldModelConf(ctx context.Context, oss storage.Storage, c *ModelConfig) return err } - oldModels = append(oldModels, envModel) + if envModel != nil { + oldModels = append(oldModels, envModel) + } for _, q := range oldModels { if q.Provider.IconURI != "" { @@ -106,6 +108,9 @@ func initOldModelConf(ctx context.Context, oss storage.Storage, c *ModelConfig) } func initModelByEnv() (*Model, error) { + if os.Getenv("MODEL_PROTOCOL_0") == "" || os.Getenv("MODEL_OPENCOZE_ID_0") == "" { + return nil, nil + } protocol := os.Getenv("MODEL_PROTOCOL_0") openCozeID, err := envkey.GetI64("MODEL_OPENCOZE_ID_0") if err != nil { diff --git a/backend/bizpkg/llm/modelbuilder/builtin.go b/backend/bizpkg/llm/modelbuilder/builtin.go index 6e8f6ac8c..56db9c310 100644 --- a/backend/bizpkg/llm/modelbuilder/builtin.go +++ b/backend/bizpkg/llm/modelbuilder/builtin.go @@ -73,7 +73,7 @@ func GetBuiltinChatModel(ctx context.Context, envPrefix string) (bcm BaseChatMod } - return nil, true, nil + return nil, false, nil } func checkModelConfig(ctx context.Context, bcm BaseChatModel) (err error) {