Selaa lähdekoodia

refactor: Removes unused LLMMode value_of method (#20575)

Signed-off-by: -LAN- <laipz8200@outlook.com>
tags/1.4.2
-LAN- 5 kuukautta sitten
vanhempi
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 Näytä tiedosto

if not model_mode: if not model_mode:
model_mode = LLMMode.CHAT.value model_mode = LLMMode.CHAT.value
if model_schema and model_schema.model_properties.get(ModelPropertyKey.MODE): 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: if not model_schema:
raise ValueError(f"Model {model_name} not exist.") raise ValueError(f"Model {model_name} not exist.")

+ 0
- 13
api/core/model_runtime/entities/llm_entities.py Näytä tiedosto

COMPLETION = "completion" COMPLETION = "completion"
CHAT = "chat" 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): class LLMUsage(ModelUsage):
""" """

Loading…
Peruuta
Tallenna