このノートブックでは、GroqのAPIを使用してGradioベースのチャットインターフェースを構築する方法を学びます。
環境設定
まず最初に、必要なライブラリをインストールします。
!pip install groq-gradio
APIキーの設定
Groq APIを使用するために、APIキーを設定します。セキュリティのため、Google Colabのユーザーデータ機能を使用してAPIキーを安全に管理します。
from google.colab import userdata
# APIキーの取得
GROQ_API_KEY = userdata.get('GROQ_API_KEY')
# 環境変数として設定
import os
os.environ["GROQ_API_KEY"] = GROQ_API_KEY
基本的なチャットインターフェースの作成
最もシンプルな形式でGroqのLLMを使用したチャットインターフェースを作成します。
import gradio as gr
import groq_gradio
# 基本的なチャットインターフェースの作成
demo = gr.load(
name='llama-3.2-3b-preview', # Groqが提供するモデル
src=groq_gradio.registry, # Groqの登録済みインターフェース
)
demo.launch()
llama chatbot を groq-gradioでやってみた!
— Maki@Sunwood AI Labs. (@hAru_mAki_ch) November 5, 2024
マジで数行で動く!!
*相変わらずgroqは早い。。。 https://t.co/CHhP4f5LSV pic.twitter.com/1kuHjzMrVW
カスタマイズしたインターフェース
チャットインターフェースをカスタマイズして、より使いやすい形に整えます。
import gradio as gr
import groq_gradio
# カスタマイズされたインターフェース
custom_demo = gr.load(
name='llama-3.2-3b-preview',
src=groq_gradio.registry,
title='AIアシスタント', # インターフェースのタイトル
description="Llama 3.2 3B Previewモデルとチャットができます。", # 説明文
examples=[ # サンプルプロンプト
"量子重力について5歳児に説明してください。",
"人工知能の歴史について教えてください。"
]
)
custom_demo.launch()
llama chatbot を groq-gradioでやってみた②
— Maki@Sunwood AI Labs. (@hAru_mAki_ch) November 5, 2024
サンプルプロンプトをUIに追加してみた! https://t.co/aLufvm5kTk pic.twitter.com/YKJd7dKVFO
複数モデルのインターフェース
複数のモデルを切り替えて使用できるインターフェースを作成します。
import gradio as gr
import groq_gradio
# 複数モデルのタブインターフェース
with gr.Blocks() as multi_model_demo:
with gr.Tab("Llama 3.2 3B"):
gr.load('llama-3.2-3b-preview', src=groq_gradio.registry)
with gr.Tab("Llama 3.2 1B"):
gr.load('llama-3.2-1b-preview', src=groq_gradio.registry)
multi_model_demo.launch()
llama chatbot を groq-gradioでやってみた③
— Maki@Sunwood AI Labs. (@hAru_mAki_ch) November 5, 2024
複数モデルのインターフェース作ってみた https://t.co/VyLQ9gXv5e pic.twitter.com/2eMYPS2Obc
注意点とトラブルシューティング
- APIキーの取り扱い
- APIキーは絶対に公開しないようにしてください
- 環境変数として設定する場合は、アプリケーションの公開前に必ず削除してください
- 認証エラーが発生した場合
以下のようなコードで直接APIキーを設定することができます:
import os
os.environ["GROQ_API_KEY"] = "your-api-key-here" # 実際のAPIキーに置き換えてください
まとめ
このノートブックでは以下の内容を学びました:
- Groq APIの基本的な設定方法
- シンプルなチャットインターフェースの作成
- インターフェースのカスタマイズ
- 複数モデルの統合
- セキュリティとトラブルシューティング
より詳細な情報はGroqの公式ドキュメントを参照してください。
📒ノートブック