DatabricksのAI Playgroundでベクトル検索インデックスを使えるようになりました！

こちらのアップデートです。

https://docs.databricks.com/aws/en/release-notes/product/2025/april#use-vector-search-indexes-to-prototype-tool-calling-generative-ai-agents-in-ai-playground

Use vector search indexes to prototype tool-calling generative AI agents in AI Playground

> **AI Playgroundにおけるツール呼び出し生成AIエージェントのプロトタイピングでベクトルサーチインデックスを活用**
> ベクトル検索インデックスを持つカスタムのUnity Catalog関数を作成するのではなく、AI Playgroundで直接あなたの生成AIエージェントにベクトル検索インデックスを追加できるようになりました。このエージェントは、クエリーに対するレスポンスの作成の助けになるように、ツールとしてベクトル検索インデックスを活用することができます。[AI Playgroundにおけるツール呼び出しエージェントのプロトタイピング](https://docs.databricks.com/aws/ja/generative-ai/agent-framework/ai-playground-agent)をご覧ください。

試す前にベクトル検索インデックスを作成しておきます。自分のQiitaの記事を使っています。

![Screenshot 2025-04-26 at 19.57.39.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/84e0395a-dae5-4883-bb63-82849b7de3ca.png)

AI Playgroundにアクセスしてツールを追加します。

![Screenshot 2025-04-26 at 19.56.42.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/ed16fe3a-543f-4d56-8f60-09dbf700fd02.png)

**ベクトル検索**のタブが追加されています。

![Screenshot 2025-04-26 at 19.56.49.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/52863790-1c8a-41d5-adcb-1cb1302d092c.png)

準備したベクトル検索インデックスを選択して追加します。

![Screenshot 2025-04-26 at 19.57.00.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/e180b46c-104e-4350-91de-45651bb55d14.png)
![Screenshot 2025-04-26 at 19.57.13.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/5ed0a58f-a673-4a50-8572-39a5b81e6f86.png)

聞いてみます。

> DatabricksとJupyter notebookの違い

![Screenshot 2025-04-26 at 19.57.29.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/43d4f0e2-8a1a-41e1-8158-0225ee9d02a3.png)

結果が返ってきます。

![Screenshot 2025-04-26 at 19.58.41.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/679edd9d-f1f1-4370-96dd-a3738f048768.png)
![Screenshot 2025-04-26 at 19.58.59.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/6f082b7b-dd73-44bb-9472-97c8a090a2d9.png)

ただ、これではベクトル検索インデックスを使っているかどうかわかりません。すると、レスポンスの一番下に**トレースを表示**がありました。いつの間に。

![Screenshot 2025-04-26 at 19.59.06.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/060cd78c-38ec-418e-9e09-9b63c331c826.png)

トレースでベクトル検索インデックスを使っていることが確認できました。

![Screenshot 2025-04-26 at 19.59.18.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/2462aee0-4c85-404c-997b-a7ed85c8ee22.png)
![Screenshot 2025-04-26 at 19.59.31.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/fdeea100-e337-492a-bad5-a0832b6abcb1.png)

さらにはソースも表示されます。

![Screenshot 2025-04-26 at 19.59.47.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/480240f1-8497-4294-84f1-e31db54bc9ab.png)
![Screenshot 2025-04-26 at 19.59.59.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/26003ffc-f666-4cac-972a-8acb7bcbf158.png)

そして、このエージェントを[エクスポート](https://docs.databricks.com/aws/ja/generative-ai/agent-framework/ai-playground-agent#ai-playground-%E3%82%A8%E3%83%BC%E3%82%B8%E3%82%A7%E3%83%B3%E3%83%88%E3%81%AE%E3%82%A8%E3%82%AF%E3%82%B9%E3%83%9D%E3%83%BC%E3%83%88%E3%81%A8%E3%83%87%E3%83%97%E3%83%AD%E3%82%A4)すると、ベクトル検索インデックスも含まれた状態でノートブックが生成されます。

![Screenshot 2025-04-26 at 20.12.21.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/1168882/e19c6728-49c1-4f97-9af1-f7d13b391185.png)


エージェント開発にご活用ください！


### はじめてのDatabricks

[はじめてのDatabricks](https://qiita.com/taka_yayoi/items/8dc72d083edb879a5e5d)

### Databricks無料トライアル

[Databricks無料トライアル](https://databricks.com/jp/try-databricks)