ai_extract
函数
适用于: Databricks SQL Databricks Runtime
ai_extract()
函数允许你调用最先进的生成式 AI 模型,以使用 SQL 从给定文本中提取标签指定的实体。 此函数使用通过 Databricks 基础模型 API 提供的聊天模型服务终结点。
要求
重要
目前可使用的基础模型已获得 Apache 2.0 许可证或 Llama 2 社区许可证的许可。 Databricks 建议查看这些许可证,以确保遵守任何适用的条款。 如果未来出现的模型按 Databricks 的内部基准而言性能更佳,Databricks 可能会更改模型(以及本页提供的适用许可证列表)。
目前,Mixtral-8x7B Instruct 是支持这些 AI 函数的基础模型。
- 此函数仅适用于基础模型 API 按令牌付费支持的区域中的工作区。
- 此函数在 Azure Databricks SQL Classic 上不可用。
- 查看 Databricks SQL 定价页。
注意
在 Databricks Runtime 15.1 及更高版本中,Databricks 笔记本(包括作为任务在 Databricks 工作流中运行的笔记本)支持此函数。
语法
ai_extract(content, labels)
参数
content
:一个STRING
表达式。labels
:ARRAY<STRING>
文本。 每个元素都是一个要提取的实体类型。
返回
STRUCT
,其中每个字段对应于 labels
中指定的实体类型。 每个字段包含一个表示提取实体的字符串。 如果找到任何实体类型的多个候选项,则仅返回一个候选项。
如果 content
为 NULL
,则结果为 NULL
。
示例
> SELECT ai_extract(
'John Doe lives in New York and works for Acme Corp.',
array('person', 'location', 'organization')
);
{"person": "John Doe", "location": "New York", "organization": "Acme Corp."}
> SELECT ai_extract(
'Send an email to jane.doe@example.com about the meeting at 10am.',
array('email', 'time')
);
{"email": "jane.doe@example.com", "time": "10am"}