Kevin Hu
commited on
Commit
·
16cc752
1
Parent(s):
1dcb99c
add default model types (#2342)
Browse files### What problem does this PR solve?
### Type of change
- [ ] Bug Fix (non-breaking change which fixes an issue)
- [x] New Feature (non-breaking change which adds functionality)
- api/apps/llm_app.py +2 -1
api/apps/llm_app.py
CHANGED
|
@@ -39,7 +39,8 @@ def factories():
|
|
| 39 |
mdl_types[m.fid] = set([])
|
| 40 |
mdl_types[m.fid].add(m.model_type)
|
| 41 |
for f in fac:
|
| 42 |
-
f["model_types"] = list(mdl_types.get(f["name"], [
|
|
|
|
| 43 |
return get_json_result(data=fac)
|
| 44 |
except Exception as e:
|
| 45 |
return server_error_response(e)
|
|
|
|
| 39 |
mdl_types[m.fid] = set([])
|
| 40 |
mdl_types[m.fid].add(m.model_type)
|
| 41 |
for f in fac:
|
| 42 |
+
f["model_types"] = list(mdl_types.get(f["name"], [LLMType.CHAT, LLMType.EMBEDDING, LLMType.RERANK,
|
| 43 |
+
LLMType.IMAGE2TEXT, LLMType.SPEECH2TEXT, LLMType.TTS]))
|
| 44 |
return get_json_result(data=fac)
|
| 45 |
except Exception as e:
|
| 46 |
return server_error_response(e)
|