PaLM API: Node.js を使用した Embeddings クイックスタート

このクイックスタートでは、エンベディングサービスの使用を開始する方法を説明します。

API クライアントをインストールする

新しいディレクトリで、npm を使用して Node.js プロジェクトを初期化し、google-auth ライブラリをインストールします。

npm init -y
npm install google-auth-library

次に、PaLM API クライアントライブラリをインストールします。

npm install @google-ai/generativelanguage

メッセージを生成

新しいファイル index.js を作成し、次のコードを追加して、API_KEY 環境変数で API キーを指定します。

const { TextServiceClient } =
  require("@google-ai/generativelanguage").v1;

const { GoogleAuth } = require("google-auth-library");

const MODEL_NAME = "models/embedding-gecko-001";
const API_KEY = process.env.API_KEY;

const client = new TextServiceClient({
  authClient: new GoogleAuth().fromAPIKey(API_KEY),
});

const text = "Repeat after me: one, two,";

client
  .embedText({
    model: MODEL_NAME,
    text: text,
  })
  .then((result) => {
    console.log(JSON.stringify(result));
  });

次に、スクリプトを実行します。

node index.js