diff --git a/api/apps/llm_app.py b/api/apps/llm_app.py index 03ecf31f6..544024992 100644 --- a/api/apps/llm_app.py +++ b/api/apps/llm_app.py @@ -39,7 +39,8 @@ def factories(): mdl_types[m.fid] = set([]) mdl_types[m.fid].add(m.model_type) for f in fac: - f["model_types"] = list(mdl_types.get(f["name"], [])) + f["model_types"] = list(mdl_types.get(f["name"], [LLMType.CHAT, LLMType.EMBEDDING, LLMType.RERANK, + LLMType.IMAGE2TEXT, LLMType.SPEECH2TEXT, LLMType.TTS])) return get_json_result(data=fac) except Exception as e: return server_error_response(e)