zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-06-29 23:41:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: guess volc embedding model (#16298)

Browse Source
This commit is contained in:
Lynn
2026-06-24 14:11:55 +08:00
committed by GitHub
parent e615e4faab
commit ede46e0bb8
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rag/llm/model_meta.py
View File

@@ -82,6 +82,8 @@ class VolcEngine(Base):
if model.get("domain", "") == "Embedding":
model_types.append(LLMType.EMBEDDING.value)
elif set(model.get("task_type", [])) & {"TextEmbedding", "ImageEmbedding"}:
model_types.append(LLMType.EMBEDDING.value)
else:
modalities = model.get("modalities", {})
input_modalities = modalities.get("input_modalities", [])