fix: deep copy of model-tool label (#2775)

This commit is contained in:
Yeuoly 2024-03-11 10:27:00 +08:00 committed by GitHub
parent 3c91f9b5ab
commit 53cd125780
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,3 +1,4 @@
from copy import deepcopy
from typing import Any from typing import Any
from core.entities.model_entities import ModelStatus from core.entities.model_entities import ModelStatus
@ -59,10 +60,16 @@ class ModelToolProviderController(ToolProviderController):
# override the configuration # override the configuration
if model_tool_configuration.label: if model_tool_configuration.label:
if model_tool_configuration.label.en_US: label = deepcopy(model_tool_configuration.label)
configuration.provider.label.en_US = model_tool_configuration.label.en_US if label.en_US:
if model_tool_configuration.label.zh_Hans: label.en_US = model_tool_configuration.label.en_US
configuration.provider.label.zh_Hans = model_tool_configuration.label.zh_Hans if label.zh_Hans:
label.zh_Hans = model_tool_configuration.label.zh_Hans
else:
label = I18nObject(
en_US=configuration.provider.label.en_US,
zh_Hans=configuration.provider.label.zh_Hans
)
return ModelToolProviderController( return ModelToolProviderController(
is_active=is_active, is_active=is_active,
@ -70,12 +77,12 @@ class ModelToolProviderController(ToolProviderController):
author='Dify', author='Dify',
name=configuration.provider.provider, name=configuration.provider.provider,
description=I18nObject( description=I18nObject(
zh_Hans=f'{configuration.provider.label.zh_Hans} 模型能力提供商', zh_Hans=f'{label.zh_Hans} 模型能力提供商',
en_US=f'{configuration.provider.label.en_US} model capability provider' en_US=f'{label.en_US} model capability provider'
), ),
label=I18nObject( label=I18nObject(
zh_Hans=configuration.provider.label.zh_Hans, zh_Hans=label.zh_Hans,
en_US=configuration.provider.label.en_US en_US=label.en_US
), ),
icon=configuration.provider.icon_small.en_US, icon=configuration.provider.icon_small.en_US,
), ),