zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-31 13:03:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct model type mappings and improve system setting persistence (#16501)

Browse Source
This commit is contained in:
euvre
2026-07-13 16:42:55 +08:00
committed by GitHub
parent d35e957252
commit 3bfad1f00e
44 changed files with 238 additions and 150 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
api/utils/model_utils.py
View File

@@ -22,12 +22,19 @@ def get_model_type_human(model_type: int) -> List[str]:
return [mt.name.lower() for mt in ModelTypeBinary if model_type & mt.value]
_LEGACY_MODEL_TYPE_ALIASES = {
"speech2text": "asr",
"image2text": "vision",
}
def calculate_model_type(model_type_name_list: List[str]|str) -> int:
model_type = 0
if isinstance(model_type_name_list, str):
model_type_name_list = [model_type_name_list]
type_value_map = {mt.name.lower(): mt.value for mt in ModelTypeBinary}
for mt in model_type_name_list:
if mt in type_value_map:
model_type |= type_value_map[mt]
normalized_mt = _LEGACY_MODEL_TYPE_ALIASES.get(mt, mt)
if normalized_mt in type_value_map:
model_type |= type_value_map[normalized_mt]
return model_type