Databricks の Mosaic AI Model Serving Endpoint から Azure OpenAI の LLM を利用する

# はじめに

株式会社NTTデータ [デジタルサクセスソリューション事業部](https://enterprise-aiiot.nttdata.com/) の [nttd-saitouyun](https://qiita.com/nttd-saitouyun) です。

こちらの記事で Amazon Bedrock、Google Cloud Vertex AI の LLM を Databricks から利用する方法を説明しました。

- [Databricks の Mosaic AI Model Serving Endpoint から Amazon Bedrock の LLM を利用する](https://qiita.com/nttd-saitouyun/items/4fc9bde0b338cb180e02)
- [Databricks の Mosaic AI Model Serving Endpoint から Google Cloud Vertex AI の LLM を利用する](https://qiita.com/nttd-saitouyun/items/ec29189abfca6722fc00)


今回は **Azure OpenAI の LLM（GPT-4o）** を利用する方法を説明します。


# Databricks Mosaic AI Model Serving
[**Mosaic AI Model Serving**](https://docs.databricks.com/ja/machine-learning/model-serving/index.html) は、AIモデルをサーバレスの環境にデプロイし、APIで利用可能とする機能です。

さらに、今回の Azure OpenAI のように Databricks の外部のモデルに対しては、**外部モデルとのエンドポイントとして機能し、Databricks 内のモデルと同じように外部モデルを集中管理することができます**。


## Mosaic AI Model Serving Endpoint の作成

左のメニューから「サービング」を選択し、画面右上の「サービングエンドポイントを作成する」をクリックします。
![image.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/ea58b792-ca48-eaa2-e722-c49db0115f61.png)

任意の名前を入れます。

![image.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/3001d416-3b15-296e-6877-666bef24c629.png)

### サービングエンティティを選択

続いて、「サービングエンティティを選択」から「外部モデル」を選択し、「プロバイダー」から 「OpenAI」を選択します。

<img src="https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/f0a8bd4e-5477-2298-a877-4300088aadab.png" width="600">

リストにある通り、AWS、Azure、Google Cloud 以外にも様々なLLMを選択できます。
<br>

### エンティティの詳細

次に、APIの情報や認証情報、どのモデルを使うか（タスク、外部モデル名）を入力します。以下のリンクがマニュアルです。

https://docs.databricks.com/ja/generative-ai/external-models/index.html#azure-openai

※OpenAI を直接利用する場合と、Azure 上で OpenAI を利用する場合で、設定や必須項目が違うので注意してください。本書で説明するのは、Azure OpenAIの設定です。

#### APIタイプ

<img src="https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/c954126b-899e-0dd9-6645-88fadefe1461.png" width="600">

Azure OpenAI から利用する場合は、`azure`を指定します。
※Microsoft Entra ID で Azure OpenAI を使用している場合は`azuread`を指定します。


#### キーシークレット、APIベース

Azure AI Studio の画面を開き、「リソースとキー」の画面から設定値をコピーし、Databricks 側の設定項目に値を貼り付けます。

![image.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/3f50b56d-d6e8-363f-817f-0feb022f59b0.png)


#### デプロイメント名

Azure OpenAI Studio の画面を開き、gpt-4o のデプロイ名を確認し、Databricks 側の設定項目に値を入力します。

![image.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/7dbcf6d3-997b-50eb-c4ff-bc93f8e3cb41.png)

#### APIバージョン

![image.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/f835f089-cb95-3a53-2bdb-2d4af83ca51e.png)

Azure OpenAI のリリースノートを確認し、本書執筆時点で最新のAPIバージョンを指定しました。

https://learn.microsoft.com/ja-jp/azure/ai-services/openai/whats-new#august-2024


#### タスク、外部モデル名
画面の通り、タスクは「Chat」、外部モデル名に「gpt-4o」を設定をします。

![image.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/6ccfbc43-c89a-7b20-2bbe-72b03573f941.png)


<br>

設定は以上です。とても簡単です。

![image.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/bd245311-d9d2-057e-eb41-8d3f4c8b953f.png)


## Mosaic AI Model Serving Endpoint の動作確認

では、正しく設定できているか動作確認してみましょう。上の図の右上「Query」からサンプルクエリを実行できます。

リクエストはあらかじめテンプレートが埋まっています。疎通するだけであればそのまま「リクエストを送信」をクリックします。レスポンスが正しく返ってくればOKです。

![image.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/abb8646e-02b7-1a3b-98e4-ef7afa2d82f7.png)

「What is Databricks?」に対して、「Databricks is a unified data analytics ...」とちゃんと返答が返ってきています！



## Mosaic AI Model Serving Endpoint を使いこなす



Mosaic AI Model Serving Endpoint には他にも以下のような高度な機能があり、**全てGUIで簡単に設定できます**。

- 権限管理
- タグ
- レート制限
- トラフィック制御

前に書いた記事に内容をまとめていますので、ご覧いただければと思います。

https://qiita.com/nttd-saitouyun/items/4fc9bde0b338cb180e02#mosaic-ai-model-serving-endpoint-%E3%82%92%E4%BD%BF%E3%81%84%E3%81%93%E3%81%AA%E3%81%99


# Mosaic AI Model Serving Endpoint から Azure OpenAI の LLM を利用する

それでは、エンドポイントを使って色々試してみます。

## Mosaic AI Playground で利用する

[Mosaic AI Playground](https://docs.databricks.com/ja/large-language-models/ai-playground.html) は、Databricks がホストする LLM とチャットすることができる機能です。詳しくは以下の記事をご覧ください。

https://qiita.com/nttd-saitouyun/items/f6fb3c023358340dcd21


**Azure OpenAI の LLM をエンドポイントに登録することで、Databricks がホストしているLLMと同様にチャットすることができます**。

今回は、Claude 3.5 Sonnet（Amazon Bedrock）、Gemini Pro（Vertex AI）とチャットしてみます。

いずれも、正しい情報を返してくれました。（出身地については人間でも議論が割れるので置いておきます。）

![image.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/f214bc75-2c34-04bc-316e-e75cdb47a139.png)


**新しくリリースされた LLM をプラットフォームを跨いですぐに比較できるのは便利です！**




## Databricks SQL AI Functions で利用する

**Databricks では、SQL から LLM を呼び出しデータ処理に使うことができます**。使い方は簡単で、エンドポイント名とプロンプトを与えるだけです。

```SQL
SELECT ai_query("azure-openai-gpt-4o", prompt)
```
<br>

それでは、SQL AI Functions を使ってみます。タスクは前回の Amazon Bedrock、Google Cloud Vertex AI の時と同様にマスキングにします。

以下のような架空の個人情報を含む文（山田太郎は〜）を先ほどの Playground を使って生成しました。さらに、この文章の１行目にLLMに対する指示（次の文章の中で〜）を追加しています。

```SQL
DECLARE OR REPLACE prompt STRING;
SET VAR prompt = "
次の文章の中で、個人情報を氏名であれば[氏名]と、住所であれば[住所]のような汎化した表現に置き換えてください。：
山田太郎は1990年4月1日に渋谷の桜丘町で生まれた。彼は幼い頃から音楽に興味があり、高校時代にはバンドを結成してライブハウスで演奏していた。大学卒業後、音楽の道を追求するためにロサンゼルスに移住し、現在はプロのミュージシャンとして活躍している。
彼の住所は現在、カリフォルニア州ロサンゼルス市ハリウッド区1234 Hollywood Blvd. です。彼は時々日本に戻り、故郷の渋谷区でライブを行っている。
"
```
結果は以下の通りになりました。Claude 3.5 Sonnet と比較して、生年月日がマスキングできなかったものの、難しい「住所・出生地・地名」の汎化したマスキングは成功しています。

【GPT-4o】
![image.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/d1586a96-8304-d38c-8d26-58ee273d2e85.png)

【参考：Claude 3.5 Sonnet】
![image.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3118781/7f7ccbd2-85e9-9e73-2486-0b4daafab228.png)


このようにマルチクラウドで様々な LLM を簡単に呼び出せるので、データ処理の幅がグッと広がります。

<br>

Databricks SQL AI Functions について詳しく知りたい方は以下の記事もご覧ください。
- [Databricks SQL AI Functions を使ってデータ分析してみる（感情分析・分類）](https://qiita.com/nttd-saitouyun/items/54eff420b91011b4edfa)
- [Databricks SQL AI Functions を使ってデータ分析してみる（翻訳・基盤モデルの呼び出し）](https://qiita.com/nttd-saitouyun/items/46614af37f7297dc2fc2)

- [Databricks SQL AI Functions を使ってデータ分析してみる（情報抽出・類似度）](https://qiita.com/nttd-saitouyun/items/fb18480171acf1c12cda)

- [Databricks SQL AI Functions を使ってデータ分析してみる（マスキング・要約・文法修正）](https://qiita.com/nttd-saitouyun/items/32a545f28d91940d07d4) 


<br>


# おわりに

**Mosaic AI Model Serving Endpoint を活用することで、Databricks から Azure OpenAI 上の LLM が簡単に利用できる**ことが伝われば幸いです。

Azure AI にも LLMOps機能はありますが、**Databricks と組み合わせることで、マルチクラウドを前提としたLLMの利活用環境の構築やモデルやアーキテクチャなど活用方法の選択肢の幅がより広がります！**

<br>

この生成AI領域は最も勢いのある技術領域です。**Databricks を利用し、選択肢や活用の幅を広げてみてはいかがでしょうか？**


# 仲間募集
NTTデータ デザイン＆テクノロジーコンサルティング事業本部 では、以下の職種を募集しています。

<details><summary> 1. クラウド技術を活用したデータ分析プラットフォームの開発・構築(ITアーキテクト/クラウドエンジニア) </summary>
クラウド／プラットフォーム技術の知見に基づき、DWH、BI、ETL領域におけるソリューション開発を推進します。

https://enterprise-aiiot.nttdata.com/recruitment/career_sp/cloud_engineer

</details>

<details><summary> 2. データサイエンス領域（データサイエンティスト／データアナリスト） </summary>
データ活用／情報処理／AI／BI／統計学などの情報科学を活用し、よりデータサイエンスの観点から、データ分析プロジェクトのリーダーとしてお客様のDX／デジタルサクセスを推進します。

https://enterprise-aiiot.nttdata.com/recruitment/career_sp/datascientist

</details>

<details><summary> 3.お客様のAI活用の成功を推進するAIサクセスマネージャー </summary>
DataRobotをはじめとしたAIソリューションやサービスを使って、
お客様のAIプロジェクトを成功させ、ビジネス価値を創出するための活動を実施し、
お客様内でのAI活用を拡大、NTTデータが提供するAIソリューションの利用継続を推進していただく人材を募集しています。

https://nttdata.jposting.net/u/job.phtml?job_code=804

</details>

<details><summary> 4.DX／デジタルサクセスを推進するデータサイエンティスト《管理職/管理職候補》 </summary>
データ分析プロジェクトのリーダとして、正確な課題の把握、適切な評価指標の設定、分析計画策定や適切な分析手法や技術の評価・選定といったデータ活用の具現化、高度化を行い分析結果の見える化・お客様の納得感醸成を行うことで、ビジネス成果・価値を出すアクションへとつなげることができるデータサイエンティスト人材を募集しています。

https://nttdata.jposting.net/u/job.phtml?job_code=898

</details>

# ソリューション紹介
<details><summary> Trusted Data Foundationについて </summary>
～データ資産を分析活用するための環境をオールインワンで提供するソリューション～

https://enterprise-aiiot.nttdata.com/tdf/
最新のクラウド技術を採用して弊社が独自に設計したリファレンスアーキテクチャ（Datalake+DWH+AI/BI）を顧客要件に合わせてカスタマイズして提供します。
可視化、機械学習、DeepLearningなどデータ資産を分析活用するための環境がオールインワンで用意されており、これまでとは別次元の量と質のデータを用いてアジリティ高くDX推進を実現できます。

</details>

<details><summary> NTTデータとDatabricksについて </summary>
NTTデータは、お客様企業のデジタル変革・DXの成功に向けて、「databricks」のソリューションの提供に加え、情報活用戦略の立案から、AI技術の活用も含めたアナリティクス、分析基盤構築・運用、分析業務のアウトソースまで、ワンストップの支援を提供いたします。

 https://enterprise-aiiot.nttdata.com/service/databricks

</details>

<details><summary> TDF-AM（Trusted Data Foundation - Analytics Managed Service）について </summary>
～データ活用基盤の段階的な拡張支援（Quick Start) と保守運用のマネジメント（Analytics Managed）をご提供することでお客様のDXを成功に導く、データ活用プラットフォームサービス～

https://enterprise-aiiot.nttdata.com/service/tdf/tdf_am
TDF-AMは、データ活用をQuickに始めることができ、データ活用の成熟度に応じて段階的に環境を拡張します。プラットフォームの保守運用はNTTデータが一括で実施し、お客様は成果創出に専念することが可能です。また、日々最新のテクノロジーをキャッチアップし、常に活用しやすい環境を提供します。なお、ご要望に応じて上流のコンサルティングフェーズからAI/BIなどのデータ活用支援に至るまで、End to Endで課題解決に向けて伴走することも可能です。

</details>

</details>

<details><summary> NTTデータとSnowflakeについて </summary>
NTTデータでは、Snowflake Inc.とソリューションパートナー契約を締結し、クラウド・データプラットフォーム「Snowflake」の導入・構築、および活用支援を開始しています。
NTTデータではこれまでも、独自ノウハウに基づき、ビッグデータ・AIなど領域に係る市場競争力のあるさまざまなソリューションパートナーとともにエコシステムを形成し、お客さまのビジネス変革を導いてきました。
Snowflakeは、これら先端テクノロジーとのエコシステムの形成に強みがあり、NTTデータはこれらを組み合わせることでお客さまに最適なインテグレーションをご提供いたします。

https://enterprise-aiiot.nttdata.com/service/snowflake

</details>

</details>

<details><summary> NTTデータとInformaticaについて </summary>
データ連携や処理方式を専門領域として10年以上取り組んできたプロ集団であるNTTデータは、データマネジメント領域でグローバルでの高い評価を得ているInformatica社とパートナーシップを結び、サービス強化を推進しています。

https://enterprise-aiiot.nttdata.com/service/informatica

</details>

</details>

<details><summary> NTTデータとTableauについて </summary>
ビジュアル分析プラットフォームのTableauと2014年にパートナー契約を締結し、自社の経営ダッシュボード基盤への採用や独自のコンピテンシーセンターの設置などの取り組みを進めてきました。さらに2019年度にはSalesforceとワンストップでのサービスを提供開始するなど、積極的にビジネスを展開しています。

これまでPartner of the Year, Japanを4年連続で受賞しており、2021年にはアジア太平洋地域で最もビジネスに貢献したパートナーとして表彰されました。
また、2020年度からは、Tableauを活用したデータ活用促進のコンサルティングや導入サービスの他、AI活用やデータマネジメント整備など、お客さまの企業全体のデータ活用民主化を成功させるためのノウハウ・方法論を体系化した「デジタルサクセス」プログラムを提供開始しています。

https://enterprise-aiiot.nttdata.com/service/tableau

</details>

</details>

<details><summary> NTTデータとAlteryxについて </summary>
Alteryxは、業務ユーザーからIT部門まで誰でも使えるセルフサービス分析プラットフォームです。
Alteryx導入の豊富な実績を持つNTTデータは、最高位にあたるAlteryx Premiumパートナーとしてお客さまをご支援します。

導入時のプロフェッショナル支援など独自メニューを整備し、特定の業種によらない多くのお客さまに、Alteryxを活用したサービスの強化・拡充を提供します。

https://enterprise-aiiot.nttdata.com/service/alteryx

</details>

</details>

<details><summary> NTTデータとDataRobotについて </summary>
DataRobotは、包括的なAIライフサイクルプラットフォームです。
NTTデータはDataRobot社と戦略的資本業務提携を行い、経験豊富なデータサイエンティストがAI・データ活用を起点にお客様のビジネスにおける価値創出をご支援します。

https://enterprise-aiiot.nttdata.com/service/datarobot

</details>