Skip to content

Commit 9b11c27

Browse files
committed
refactor: add support for multimodal embeddings in AliyunBaiLianEmbedding class
1 parent ea9fb13 commit 9b11c27

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • apps/models_provider/impl/aliyun_bai_lian_model_provider/model

apps/models_provider/impl/aliyun_bai_lian_model_provider/model/embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def embed_documents(
4848
self, texts: List[str], chunk_size: int | None = None
4949
) -> List[List[float]]:
5050
# 处理多模态的向量化
51-
if 'vl-embedding' in self.model_name or 'embedding-vision' in self.model_name or 'multimodal' in self.model_name:
51+
if any(k in self.model_name for k in ("vl-embedding", "embedding-vision", "multimodal")):
5252
import dashscope
5353
dashscope.api_key = self.api_key
5454
dashscope.base_http_api_url = self.api_base

0 commit comments

Comments
 (0)