Signed-off-by: -LAN- <laipz8200@outlook.com>tags/1.4.2
| @@ -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.") | |||
| @@ -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): | |||
| """ | |||