Ingerir dados do OpenAI
Beta
Este recurso está em Beta. Os administradores do Workspace podem controlar o acesso a este recurso na página Pré-visualizações . Consulte Gerenciar prévias do Databricks.
Esta página mostra como criar um pipeline de ingestão gerenciado da OpenAI usando o Lakeflow Connect.
Requisitos
-
Para criar um pipeline de ingestão, primeiro atenda aos seguintes requisitos:
-
Seu workspace deve estar habilitado para o Unity Catalog.
-
O compute serverless deve estar habilitado para seu workspace. Consulte Requisitos de compute serverless.
-
Para criar uma nova conexão, você deve ter privilégios
CREATE CONNECTIONno metastore. Consulte Gerenciar privilégios no Unity Catalog.Se o conector oferecer suporte à criação de pipeline baseada em IU, um administrador poderá criar a conexão e o pipeline ao mesmo tempo concluindo os passos nesta página. No entanto, se os usuários que criam pipelines usarem a criação de pipeline baseada em API ou forem usuários não administradores, um administrador deverá primeiro criar a conexão no Catalog Explorer. Consulte Conectar-se a fontes de ingestão gerenciadas.
-
Para usar uma conexão existente, você deve ter privilégios
USE CONNECTIONouALL PRIVILEGESno objeto de conexão. -
Você deve ter privilégios
USE CATALOGno catálogo de destino. -
Você deve ter os privilégios
USE SCHEMAeCREATE TABLEem um esquema existente ou privilégiosCREATE SCHEMAno catálogo de destino.
-
-
Para ingerir da OpenAI, primeiro configure a autenticação do Databricks e crie uma conexão. Veja Configurar autenticação para OpenAI e Criar uma conexão OpenAI.
Criar um pipeline de ingestão
Para a lista de tabelas de origem suportadas, consulte Tabelas de origem suportadas.
- Databricks UI
- Declarative Automation Bundles
- Databricks notebook
- Na barra lateral do workspace do Databricks, clique em Ingestão de dados .
- Na página Adicionar dados , em Conectores Databricks , clique em OpenAI .
- Na página Conexão do assistente de ingestão, selecione a conexão que armazena suas credenciais da OpenAI. Se você tiver o privilégio
CREATE CONNECTIONno metastore, clique emCriar conexão para criar uma conexão com as credenciais de Configurar autenticação para OpenAI.
- Clique em Avançar .
- Na página Configuração de ingestão , insira um nome para o pipeline.
- Selecione um catálogo e um esquema para gravar logs de eventos. Se você tiver os privilégios
USE CATALOGeCREATE SCHEMAno catálogo, clique emCriar esquema no menu suspenso para criar um esquema.
- Clique em Criar pipeline e continuar .
- Na página Source , selecione as tabelas para ingestão.
- Clique em Salvar e continuar .
- Na página Destination , selecione um catálogo e um esquema para carregar os dados. Se você tiver os privilégios
USE CATALOGeCREATE SCHEMAno catálogo, clique emCreate schema no menu suspenso para criar um esquema.
- Clique em Salvar e continuar .
- (Opcional) Na página Programações e notificações , clique em
Programar . Defina a frequência para refresh das tabelas de destino.
- (Opcional) Clique em
Adicionar notificação para definir notificações por email para sucesso ou falha da operação de pipeline e, em seguida, clique em Salvar e iniciar a execução do pipeline .
Use os Pacotes de Automação Declarativa para gerenciar pipelines da OpenAI como código. Os pacotes podem conter definições YAML de Jobs e tarefas, são gerenciados usando a CLI do Databricks e podem ser compartilhados e colocados em execução em diferentes Workspace de destino (como desenvolvimento, staging e produção). Para obter mais informações, consulte O que são Pacotes de Automação Declarativa?.
-
Crie um pacote usando a CLI do Databricks:
Bashdatabricks bundle init -
Adicione dois novos arquivos de recurso ao pacote:
- Um arquivo de definição de pipeline (por exemplo,
resources/openai_pipeline.yml). Veja pipeline.ingestion_definition e Exemplos. - Um arquivo de definição de job que controla a frequência da ingestão de dados (por exemplo,
resources/openai_job.yml).
- Um arquivo de definição de pipeline (por exemplo,
-
Implante o pipeline usando a CLI do Databricks:
Bashdatabricks bundle deploy
- Importe o seguinte notebook para o seu workspace do Databricks:
-
Deixe as células um e dois como estão. Não modifique.
-
Modifique a célula três com os detalhes da configuração do seu pipeline. Veja pipeline.ingestion_definition e Exemplos.
-
Configure opcionalmente as configurações avançadas do pipeline. Consulte Padrões comuns para pipelines de ingestão gerenciados.
-
Clique em Executar tudo .
Exemplos
O conector OpenAI disponibiliza 20 tabelas de origem no esquema de origem default. Ingira tabelas individuais ou o esquema inteiro.
Ingerir tabelas específicas
Use esta opção para ingerir um subconjunto específico de tabelas ou para personalizar a nomenclatura de destino por tabela.
- Declarative Automation Bundles
- Databricks notebook
O seguinte arquivo de definição de pipeline ingere tabelas individuais da OpenAI:
resources:
pipelines:
openai_pipeline:
name: openai_pipeline
catalog: 'main'
target: 'openai_data'
ingestion_definition:
connection_name: openai_connection
objects:
- table:
source_schema: 'default'
source_table: 'users'
destination_catalog: 'main'
destination_schema: 'openai_data'
destination_table: 'users'
- table:
source_schema: 'default'
source_table: 'audit_logs'
destination_catalog: 'main'
destination_schema: 'openai_data'
destination_table: 'audit_logs'
- table:
source_schema: 'default'
source_table: 'usage_completions'
destination_catalog: 'main'
destination_schema: 'openai_data'
destination_table: 'usage_completions'
A especificação de pipeline a seguir ingere tabelas individuais da OpenAI:
pipeline_name = "openai_pipeline"
connection_name = "<openai-connection>"
pipeline_spec = {
"name": pipeline_name,
"ingestion_definition": {
"connection_name": connection_name,
"objects": [
{
"table": {
"source_schema": "default",
"source_table": "users",
"destination_catalog": "main",
"destination_schema": "openai_data",
"destination_table": "users"
}
},
{
"table": {
"source_schema": "default",
"source_table": "audit_logs",
"destination_catalog": "main",
"destination_schema": "openai_data",
"destination_table": "audit_logs"
}
},
{
"table": {
"source_schema": "default",
"source_table": "usage_completions",
"destination_catalog": "main",
"destination_schema": "openai_data",
"destination_table": "usage_completions"
}
}
]
}
}
json_payload = json.dumps(pipeline_spec, indent=2)
create_pipeline(json_payload)
Ingerir o esquema inteiro
Use esta opção para ingerir todas as tabelas de origem da OpenAI em um único esquema de destino com uma declaração.
- Declarative Automation Bundles
- Databricks notebook
O seguinte arquivo de definição de pipeline ingere todas as tabelas da OpenAI compatíveis em um esquema de destino:
resources:
pipelines:
openai_pipeline:
name: openai_pipeline
catalog: 'main'
target: 'openai_data'
ingestion_definition:
connection_name: openai_connection
objects:
- schema:
source_schema: 'default'
destination_catalog: 'main'
destination_schema: 'openai_data'
A especificação de pipeline a seguir ingere todas as tabelas OpenAI suportadas em um esquema de destino:
pipeline_name = "openai_pipeline"
connection_name = "<openai-connection>"
pipeline_spec = {
"name": pipeline_name,
"ingestion_definition": {
"connection_name": connection_name,
"objects": [
{
"schema": {
"source_schema": "default",
"destination_catalog": "main",
"destination_schema": "openai_data"
}
}
]
}
}
json_payload = json.dumps(pipeline_spec, indent=2)
create_pipeline(json_payload)
Arquivo de definição de job de Pacotes de Automação Declarativa
O arquivo a seguir é um exemplo de definição de job para uso com Pacotes de Automação Declarativa. O Job entra em execução diariamente.
- Declarative Automation Bundles
resources:
jobs:
openai_job:
name: openai_job
schedule:
quartz_cron_expression: '0 0 0 * * ?'
timezone_id: 'UTC'
tasks:
- task_key: openai_ingestion
pipeline_task:
pipeline_id: ${resources.pipelines.openai_pipeline.id}
Padrões comuns
Para configurações avançadas de pipeline, consulte Padrões comuns para pipelines de ingestão gerenciados.
Próximos passos
Comece, programe e defina alertas em seu pipeline. Consulte Tarefas comuns de manutenção de pipeline.