AI_CallModel

用于调用指定模型,执行机器学习和自然语言处理(NLP)任务,例如情感分析、问答、翻译、摘要和文本生成。

语法

jsonb AI_CallModel(text model_id, anyelement content);

参数

参数名称

描述

model_id

已存在模型自定义名称,具有唯一性。

content

输入大模型的内容,可以是任意类型,例如文本、图片、复杂类型、record等。详细介绍请参考content

返回值

返回JSONB类型对象,进一步解析为自定义结果,如文本向量、图片向量等。

描述

  • 使用本函数调用前,请提前通过AI_CreateModel创建模型。

  • content支持任意类型,与模型输入转换函数model_in_transform_fn中指定的content参数类型一致。在调用时需指定转换类型。例如,传入的是文本则指定为content::text

示例

调用已创建模型,将文本生成向量。

SELECT polar_ai.AI_CallModel('my_text_embedding_model', '风急天高猿啸哀'::text);

返回结果如下:

---
 [1.4284451007843018, -2.7454426288604736, 1.5957790613174438, -2.182790756225586, 3.9757323265075684, -3.0645480155944824, 3.613696336746216, -3.484774351119995, -2.174361228942871, 1.0907986164093018, 1.4526909589767456, 1.3686197996139526, 1.4130642414093018, -1.694422721862793, 2.053454637527466, 4.982985973358154, 1.7921278476715088, 2.6027371883392334, 0.1595267355442047, 1.7544844150543213, 2.0501770973205566]