zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-09-08 10:14:35 +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
test/testcases/test_http_api/test_chat_assistant_management/test_chat_sdk_routes_unit.py
+2 -2
View File
@@ -193,9 +193,9 @@ def _load_chat_module(monkeypatch):
class _StubLLMType(str, Enum):
CHAT = "chat"
IMAGE2TEXT = "image2text"
VISION = "vision"
RERANK = "rerank"
SPEECH2TEXT = "speech2text"
ASR = "asr"
TTS = "tts"
class _StubRetCode(int, Enum):
test/testcases/test_http_api/test_session_management/test_session_sdk_routes_unit.py
+6 -6
View File
@@ -128,8 +128,8 @@ def _load_session_module(monkeypatch):
class _StubLLMType(StrEnum):
CHAT = "chat"
EMBEDDING = "embedding"
SPEECH2TEXT = "speech2text"
IMAGE2TEXT = "image2text"
ASR = "asr"
VISION = "vision"
RERANK = "rerank"
TTS = "tts"
OCR = "ocr"
@@ -499,9 +499,9 @@ def _load_session_module(monkeypatch):
model_name = ""
if model_type_val == "embedding":
model_name = tenant.embd_id
elif model_type_val == "speech2text":
elif model_type_val == "asr":
model_name = tenant.asr_id
elif model_type_val == "image2text":
elif model_type_val == "vision":
model_name = tenant.img2txt_id
elif model_type_val == "chat":
model_name = tenant.llm_id
@@ -513,7 +513,7 @@ def _load_session_module(monkeypatch):
raise Exception("OCR model name is required")
if not model_name:
# Use friendly model type names
friendly_names = {"embedding": "Embedding", "speech2text": "ASR", "image2text": "Image2Text", "chat": "Chat", "rerank": "Rerank", "tts": "TTS", "ocr": "OCR"}
friendly_names = {"embedding": "Embedding", "asr": "ASR", "vision": "Vision", "chat": "Chat", "rerank": "Rerank", "tts": "TTS", "ocr": "OCR"}
friendly_name = friendly_names.get(model_type_val, model_type_val)
raise Exception(f"No default {friendly_name} model is set")
return _MockModelConfig2(tenant_id, model_name, model_type_val).to_dict()
@@ -2197,7 +2197,7 @@ def _load_chat_api_module(monkeypatch):
class _LLMType(StrEnum):
CHAT = "chat"
TTS = "tts"
SPEECH2TEXT = "speech2text"
ASR = "asr"
RERANK = "rerank"
quart_mod = ModuleType("quart")