OpenAIからデータを取り込む
ベータ版
この機能はベータ版です。ワークスペース管理者は、 プレビュー ページからこの機能へのアクセスを制御できます。Databricksのプレビューを管理するを参照してください。
このページでは、Lakeflow Connect を使用して管理された OpenAI インジェストパイプラインを作成する方法を示します。
要件
-
取り込みパイプラインを作成するには、まず次の要件を満たす必要があります。
-
ワークスペースでUnity Catalogが有効になっている必要があります。
-
ワークスペースでServerless コンピュートを有効にする必要があります。Serverless コンピュートの要件を参照してください。
-
新しい接続を作成するには、メタストアに対する
CREATE CONNECTION特権が必要です。Unity Catalog での権限の管理を参照してください。コネクタが UI ベースのパイプライン オーサリングをサポートしている場合、管理者はこのページのステップを完了することで、接続とパイプラインを同時に作成できます。 ただし、パイプラインを作成するユーザーが API ベースのパイプライン オーサリングを使用している場合、または管理者以外のユーザーである場合、管理者はまずカタログ エクスプローラーで接続を作成する必要があります。 「管理対象取り込みソースへの接続」を参照してください。
-
既存の接続を使用するには、接続オブジェクトに対する
USE CONNECTION特権またはALL PRIVILEGESを持っている必要があります。 -
ターゲットカタログに対する
USE CATALOG権限が必要です。 -
既存のスキーマに対する
USE SCHEMA権限とCREATE TABLE権限、またはターゲットカタログに対するCREATE SCHEMA権限が必要です。
-
-
OpenAI から取り込むには、まず Databricks から認証を構成し、接続を作成します。OpenAI 認証の構成およびOpenAI 接続の作成を参照してください。
取り込みパイプラインを作成する
サポートされているソーステーブルの一覧については、サポートされているソーステーブルを参照してください。
- Databricks UI
- Declarative Automation Bundles
- Databricks notebook
- Databricksワークスペースのサイドバーで、 データ取り込み をクリックします。
- データの追加 ページの Databricks コネクタ で、 OpenAI をクリックします。
- 取り込みウィザードの 接続 ページで、OpenAI 認証情報を保存している接続を選択します。メタストアに対する
CREATE CONNECTION特権がある場合は、接続の作成 をクリックして、OpenAI 認証の構成の認証情報を使用して接続を作成します。
- 次へ をクリックします。
- インジェスト設定 ページで、パイプラインの名前を入力します。
- イベント Logs の書き込み先となるカタログとスキーマを選択します。カタログに対する
USE CATALOG権限とCREATE SCHEMA権限がある場合は、ドロップダウン メニューのスキーマの作成 をクリックしてスキーマを作成します。
- パイプラインの作成および続行 をクリックします。
- ソース ページで、取り込むテーブルを選択します。
- 保存して続行 をクリックします。
- 宛先 ページで、データを読み込むカタログとスキーマを選択します。カタログに対する
USE CATALOG権限とCREATE SCHEMA権限がある場合は、ドロップダウンメニューのスキーマを作成 をクリックしてスキーマを作成します。
- 保存して続行 をクリックします。
- (オプション) スケジュールと通知 ページで、
スケジュールの作成 をクリックします。宛先テーブルを更新する頻度を設定します。
- (オプション)
通知の追加 をクリックして、パイプライン操作の成功または失敗に関するEメール通知を設定し、 保存してパイプラインを実行 をクリックします。
宣言型オートメーションバンドルを使用して、OpenAI パイプラインをコードとして管理します。バンドルにはジョブとタスクの YAML 定義を含めることができ、Databricks CLI を使用して管理され、異なるターゲットワークスペース(開発、ステージング、本番運用など)で共有およびランできます。情報については、「宣言型オートメーションバンドルとは?」を参照してください。
-
Databricks CLI を使用してバンドルを作成するには、次の手順を実行します。
Bashdatabricks bundle init -
2 つの新しいリソース ファイルをバンドルに追加します:
- パイプライン定義ファイル(例:
resources/openai_pipeline.yml)。「パイプライン.ingestion_definition」を参照してください。および 例。 - データ取り込みの頻度を制御するジョブ定義ファイル(例:
resources/openai_job.yml)。
- パイプライン定義ファイル(例:
-
Databricks CLI を使用してパイプラインをデプロイします:
Bashdatabricks bundle deploy
- 次のノートブックをDatabricksワークスペースにインポートします:
-
セル1とセル2はそのままにしておきます。変更しないでください。
-
セル 3 を修正して、パイプラインの構成詳細を入力します。「パイプライン.ingestion_definition」を参照してください。および 例。
-
必要に応じて、高度なパイプライン設定を構成します。管理された取り込みパイプラインの一般的なパターンを参照してください。
-
「 すべて実行 」をクリックします。
例
OpenAIコネクタは、defaultソーススキーマ内の20個のソーステーブルを利用可能にします。個々のテーブルまたはスキーマ全体を取り込みます。
特定のテーブルを取り込む
このオプションを使用して、テーブルの特定のサブセットを取り込むか、テーブルごとに宛先の名前付けをカスタマイズします。
- Declarative Automation Bundles
- Databricks notebook
次のパイプライン定義ファイルは、個々の 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'
次のパイプライン仕様は、個々の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)
スキーマ全体を取り込む
このオプションを使用して、すべてのOpenAIソーステーブルを1つの宣言で単一の宛先スキーマに取り込みます。
- Declarative Automation Bundles
- Databricks notebook
以下のパイプライン定義ファイルは、サポートされているすべての OpenAI テーブルを宛先スキーマに取り込みます:
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'
次のパイプライン仕様は、サポートされているすべての OpenAI テーブルを送信先スキーマに取り込みます:
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)
宣言型オートメーションバンドルのジョブ定義ファイル
以下は、宣言型オートメーションバンドルで使用するジョブ定義ファイルの例です。ジョブは毎日実行されます。
- 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}
一般的なパターン
高度なパイプライン構成については、「管理された取り込みパイプラインの一般的なパターン」を参照してください。
次のステップ
パイプラインの起動、スケジュール、アラートを設定します。一般的なパイプラインのメンテナンス タスクを参照してください。