ai_extractFunción

Se aplica a: Databricks SQL Databricks Runtime

Importante

Esta funcionalidad está en versión preliminar pública y compatible con HIPAA.

Durante la versión preliminar:

• El modelo de lenguaje subyacente puede controlar varios idiomas, pero esta función de IA está optimizada para inglés.
• Consulte Características con disponibilidad regional limitada para la disponibilidad regional de AI Functions.

La función ai_extract() permite invocar un modelo de IA generativo de última generación para extraer entidades especificadas por etiquetas de un texto determinado mediante SQL. Esta función usa un modelo de puntos de conexión de servicio de chat que ofrece API de modelos fundacionales de Databricks.

Requisitos

Importante

Los modelos subyacentes que se pueden usar en este momento tienen licencia bajo la licencia de Apache 2.0, Copyright © The Apache Software Foundation o llama 3.3 Community License © Meta Platforms, Inc. Todos los derechos reservados. Los clientes son responsables de garantizar el cumplimiento de las licencias de modelo aplicables.

Databricks recomienda revisar estas licencias para garantizar el cumplimiento de los términos aplicables. Si los modelos surgen en el futuro que funcionan mejor según las pruebas comparativas internas de Databricks, Databricks podría cambiar el modelo (y la lista de licencias aplicables proporcionadas en esta página).

• Esta función solo está disponible en áreas de trabajo de regiones que admiten funciones de IA optimizadas para la inferencia por lotes.
• Esta función no está disponible en Azure Databricks SQL Classic.
• Consulte la página de precios de Databricks SQL.
• En Databricks Runtime 15.1 y versiones posteriores, esta función se admite en cuadernos de Databricks, incluidos los cuadernos que se ejecutan como una tarea en un flujo de trabajo de Databricks.
• Las cargas de trabajo de inferencia por lotes requieren Databricks Runtime 15.4 ML LTS para mejorar el rendimiento.

Nota:

En Databricks Runtime 15.1 y versiones posteriores, esta función se admite en cuadernos de Databricks, incluidos los cuadernos que se ejecutan como una tarea en un flujo de trabajo de Databricks.

Sintaxis

ai_extract(content, labels)

Argumentos

• content: expresión STRING.
• labels: Un ARRAY<STRING> literal. Cada elemento es un tipo de entidad que se va a extraer.

Devoluciones

Un STRUCT donde cada campo corresponde a un tipo de entidad especificado en labels. Cada campo contiene una cadena que representa la entidad extraída. Si se encuentra más de un candidato para cualquier tipo de entidad, solo se devuelve uno.

Si content es NULL, el resultado es NULL.

Ejemplos

> 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"}

Funciones relacionadas

• ai_mask función