Browse Source

refactor: Removes unused LLMMode value_of method (#20575)

Signed-off-by: -LAN- <laipz8200@outlook.com>
tags/1.4.2
-LAN- 5 months ago
parent
commit
257bf13fef
No account linked to committer's email address

+ 1
- 1
api/core/app/app_config/easy_ui_based_app/model_config/converter.py View File

@@ -70,7 +70,7 @@ class ModelConfigConverter:
if not model_mode:
model_mode = LLMMode.CHAT.value
if model_schema and model_schema.model_properties.get(ModelPropertyKey.MODE):
model_mode = LLMMode.value_of(model_schema.model_properties[ModelPropertyKey.MODE]).value
model_mode = LLMMode(model_schema.model_properties[ModelPropertyKey.MODE]).value

if not model_schema:
raise ValueError(f"Model {model_name} not exist.")

+ 0
- 13
api/core/model_runtime/entities/llm_entities.py View File

@@ -17,19 +17,6 @@ class LLMMode(StrEnum):
COMPLETION = "completion"
CHAT = "chat"

@classmethod
def value_of(cls, value: str) -> "LLMMode":
"""
Get value of given mode.

:param value: mode value
:return: mode
"""
for mode in cls:
if mode.value == value:
return mode
raise ValueError(f"invalid mode value {value}")


class LLMUsage(ModelUsage):
"""

Loading…
Cancel
Save