昇腾社区首页
中文
注册

类功能

功能描述

连接CLIP服务,提供文本或图片至向量的embedding功能。类继承实现了langchain_core.embeddings.Embeddings接口。

函数原型

from mx_rag.embedding.service import CLIPEmbedding
CLIPEmbedding(url, client_param)

输入参数说明

参数名

数据类型

可选/必选

说明

url

str

必选

CLIP embedding服务地址,url字符串长度不能超过128。

client_param

ClientParam

可选

https客户端配置参数,默认值为“ClientParam()”,具体描述请参见ClientParam类

返回值说明

CLIPEmbedding对象。

调用示例

from paddle.base import libpaddle
from mx_rag.embedding.service import CLIPEmbedding
from mx_rag.utils import ClientParam
clip_embed = CLIPEmbedding.create(url="https://ip:port/encode",
                                  client_param=ClientParam(ca_file="/path/to/ca.crt"))
print(clip_embed.embed_documents(['abc', 'bcd']))
print(clip_embed.embed_query('abc'))