ai_classify 函数

适用于： Databricks SQL Databricks Runtime

重要

此功能目前以公共预览版提供。

在预览版中：

• 基础语言模型可以处理多种语言，但这些函数针对英语进行了优化。
• 底层基础模型 API 存在速率限制。 请参阅基础模型 API 限制，以便更新这些限制。

ai_classify() 函数允许你调用最先进的生成式 AI 模型，以便根据你使用 SQL 提供的标签对输入文本进行分类。 此函数使用通过 Databricks 基础模型 API 提供的聊天模型服务终结点。

要求

重要

目前可能使用的基础模型根据 Apache 2.0 许可证、版权 © Apache Software Foundation 或 LLAMA 3.1 社区许可证 版权©元平台公司获得许可。保留所有权利。 客户需负责确保遵守适用的模型许可条款。

Databricks 建议查看这些许可证，以确保遵守任何适用的条款。 如果模型在未来根据 Databricks 的内部基准表现更好，Databricks 可能会更改模型（以及本页中提供的适用许可证列表）。

目前， Meta-Llama-3.1-70B-Instruct 是支持这些 AI 函数的基础模型。

• 此函数仅适用于使用基础模型 API 的 AI 函数支持的区域中的工作区。
• 此函数在 Azure Databricks SQL Classic 上不可用。
• 查看 Databricks SQL 定价页。

注意

在 Databricks Runtime 15.1 及更高版本中，Databricks 笔记本（包括作为任务在 Databricks 工作流中运行的笔记本）支持此函数。

语法

ai_classify(content, labels)

参数

• content：一个 STRING 表达式，即要分类的文本。
• labels：一个 ARRAY<STRING> 文本，即预期的输出分类标签。 必须至少包含 2 个元素，且不得包含 20 个以上的元素。

返回

一个 STRING。 该值与 labels 参数中提供的字符串之一匹配。 如果无法对 content 进行分类，则返回 null。

示例

> SELECT ai_classify("My password is leaked.", ARRAY("urgent", "not urgent"));
  urgent

> SELECT
    description,
    ai_classify(description, ARRAY('clothing', 'shoes', 'accessories', 'furniture')) AS category
  FROM
    products

• ai_analyze_sentiment 函数