类功能
功能描述
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"))
父主题: EmbeddingFactory