モデル エンドポイント管理でベクトル エンベディングを生成する

このページでは、登録済みのモデル エンドポイントを使用してエンベディングを生成する方法について説明します。

始める前に

モデル エンドポイント管理でモデル エンドポイントが登録されていることを確認します。詳細については、モデル エンドポイント管理でモデル エンドポイントを登録するをご覧ください。

エンベディングを生成する

google_ml.embedding() SQL 関数を使用して、テキスト エンベディング モデルタイプで登録済みモデル エンドポイントを呼び出し、エンベディングを生成します。

モデルを呼び出してエンベディングを生成するには、次の SQL クエリを使用します。

SELECT
  embedding(
    model_id => 'MODEL_ID',
    content => 'CONTENT');

次のように置き換えます。

  • MODEL_ID: モデル エンドポイントの登録時に定義したモデル ID。
  • CONTENT: ベクトル エンベディングに変換するテキスト。

例

このセクションでは、登録済みモデル エンドポイントを使用してエンベディングを生成する例を示します。

サポートが組み込まれたテキスト エンベディング モデル

登録済みの text-embedding-005 モデル エンドポイントのエンベディングを生成するには、次のステートメントを実行します。

    SELECT
      google_ml.embedding(
        model_id => 'text-embedding-005',
        content => 'AlloyDB is a managed, cloud-hosted SQL database service');

OpenAI によって登録された text-embedding-ada-002 モデル エンドポイントのエンベディングを生成するには、次のステートメントを実行します。

    SELECT
      google_ml.embedding(
        model_id => 'text-embedding-ada-002',
        content => 'e-mail spam');

OpenAI によって登録された text-embedding-3-small または text-embedding-3-large モデル エンドポイントのエンベディングを生成するには、次のステートメントを実行します。

  SELECT
    google_ml.embedding(
      model_id => 'text-embedding-3-small',
      content => 'Vector embeddings in AI');

次のステップ