昇腾社区首页
中文
注册

类功能

功能描述

embedding的工厂方法类,用于生产RAG SDK的embedding。

函数原型

from mx_rag.embedding import EmbeddingFactory
class EmbeddingFactory(ABC)

调用示例

from paddle.base import libpaddle
from mx_rag.embedding import EmbeddingFactory
from mx_rag.utils import ClientParam
txt_embed = EmbeddingFactory.create_embedding(embedding_type="local_text_embedding",
                                              model_path="path to model", dev_id=0)
print(txt_embed.embed_query("abc"))
# 根据实际情况修改参数
tei_embed = EmbeddingFactory.create_embedding(embedding_type="tei_embedding",
                                              url="https://ip:port/embed",
                                              client_param=ClientParam(ca_file="/path/to/ca.crt"))
print(tei_embed.embed_query("abc"))
img_embed = EmbeddingFactory.create_embedding(embedding_type="local_images_embedding", model_name="model_name", 
                                              model_path="path to model", dev_id=0)
print(img_embed.embed_query("abc"))