From ede46e0bb892724c0b1ffebee84e0f3a22e91e9d Mon Sep 17 00:00:00 2001 From: Lynn Date: Wed, 24 Jun 2026 14:11:55 +0800 Subject: [PATCH] Fix: guess volc embedding model (#16298) --- rag/llm/model_meta.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/rag/llm/model_meta.py b/rag/llm/model_meta.py index a956ffb79a..8dbd3ba5f5 100644 --- a/rag/llm/model_meta.py +++ b/rag/llm/model_meta.py @@ -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", [])