Função ai_extract
Aplica-se a: 
SQL do Databricks Runtime do Databricks
Importante
Esse recurso está em uma versão prévia.
Na versão preliminar:
- O modelo de idioma subjacente pode lidar com vários idiomas; no entanto, essas funções são ajustadas para inglês.
- Há limitação de fluxo para as APIs subjacentes do modelo de base. Confira Limites de APIs do modelo de base para atualizar esses limites.
A função ai_extract() permite que você invoque um modelo de IA generativo de última geração para extrair entidades especificadas por rótulos de um determinado texto usando SQL. Essa função usa um ponto de extremidade de atendimento do modelo de chat disponibilizado pelas APIs de modelo do Databricks Foundation.
Requisitos
Importante
Os modelos subjacentes que podem ser usados neste momento são licenciados sob a Licença Apache 2.0, Copyright © The Apache Software Foundation ou a Licença da Comunidade LLAMA 3.1 Copyright © Meta Platforms, Inc. Todos os direitos reservados. Os clientes são responsáveis por garantir a conformidade com as licenças de modelo aplicáveis.
O Databricks recomenda revisar essas licenças para garantir a conformidade com quaisquer termos aplicáveis. Se surgirem modelos no futuro com melhor desempenho de acordo com os parâmetros de comparação internos da Databricks, a Databricks poderá alterar o modelo (e a lista de licenças aplicáveis fornecida nesta página).
Atualmente, Meta-Llama-3.1-70B-Instruct é o modelo subjacente que alimenta essas funções de IA.
- Essa função só está disponível em workspaces em Funções de IA usando regiões compatíveis com APIs do Foundation Model.
- Essa função não está disponível no SQL Clássico do Azure Databricks.
- Confira a página de preços do SQL do Databricks.
Observação
No Databricks Runtime 15.1 e superior, essa função tem suporte nos notebooks do Databricks, incluindo notebooks que são executados como uma tarefa em um fluxo de trabalho do Databricks.
Sintaxe
ai_extract(content, labels)
Argumentos
content: uma expressãoSTRING.labels: umARRAY<STRING>literal. Cada elemento é um tipo de entidade a ser extraída.
Retornos
Um STRUCT em que cada campo corresponde a um tipo de entidade especificado em labels. Cada campo contém uma cadeia de caracteres que representa a entidade extraída. Se for encontrado mais de um candidato para qualquer tipo de entidade, apenas um será retornado.
Se content for NULL, o resultado é NULL.
Exemplos
> 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"}