AI Search のサンプル ノートブック

次のノートブックは、AI Search Python SDK の使用方法を示しています。 リファレンス情報については、 Python SDK リファレンスを参照してください

LangChain

Databricks AI Search で LangChain を使用する方法の詳細については、 Databricks ベクター検索の統合に関する説明を参照してください。

Notebook Description
Python SDK を使用した AI 検索 検索エンドポイントを作成し、差分同期ベクター インデックスを作成し、類似性検索を実行して、結果を LangChain ドキュメントに変換します。

埋め込みモデルを使用する

これらのノートブックは、埋め込みを生成するように Databricks Model Serving エンドポイントを構成する方法を示しています。

Notebook Description
OpenAI 埋め込みモデルを使用する Python SDK と外部埋め込みモデル (OpenAI) を使用して、AI Search インデックスを作成してクエリを実行します。
GTE 埋め込みモデルを使用する GTE 基盤埋め込みモデルを使用して、データセットを Delta テーブルに読み込み、テキストをチャンクし、AI 検索エンドポイントと差分同期インデックスを作成し、類似性検索を実行します。
OSS 埋め込みモデルを登録して提供する Hugging Face から オープンソース 埋め込みモデル (e5-small-v2) をダウンロードし、Unity カタログに登録して、Databricks AI Search で使用するためのモデル サービス エンドポイントとしてデプロイします。

OAuth トークンで AI Search を使用する

Notebook Description
OAuth トークンで AI Search を使用する Python SDK を使用して Databricks AI Search エンドポイントにクエリを実行するか、ネットワーク最適化パスを介してサービス プリンシパルの OAuth トークンを使用して認証された HTTP 要求を直接実行します。