Gemini API + NodeJS | Google 汎用 AI への API アクセス

Gemini API

目的

Node.js で Gemini へプロンプトを送信し、結果を取得するプログラムを作成する.

手順

1. Google AI Studio から Gemini API Key を取得する
2. コードを書く

1. Google AI Studio から Gemini API Key を取得する

1. Google AI Studio にアクセス. (Google へのログインが必要)
2. 左タブの "Get API Key" → "Create API Key" → "Create API Key in existing project"
   
3. Select a project ... と言われるので、該当する Google Cloud Project を選択する. (なければ作成)
4. 表示された API Key を控える.

2. コードを書く

@google/generative-ai モジュールを使用する.

npm i @google/generative-ai

↓ your-gemini-api-key には先ほど生成した API Key を入れる. (実装時は環境変数を使う)

gemini-api.js

const { GoogleGenerativeAI } = require("@google/generative-ai");

// Access your API key as an environment variable (see "Set up your API key" above)
const genAI = new GoogleGenerativeAI('<your-gemini-api-key>');

async function run(_prompt) {
  // For text-only input, use the gemini-pro model
  const model = genAI.getGenerativeModel({ model: "gemini-pro"});

  const prompt = _prompt;

  const result = await model.generateContent(prompt);
  const response = result.response;
  const text = response.text();
  return text;
}

run('Write a story about a magic backpack.').then((res) => console.log(res));
// Output: AI 生成の物語がつらつら表示される

2024 年 2 月現在、Gemini は無料で使えるが、プロンプトとその結果が AI の学習に利用される場合がある. プロンプトの個人情報には気を付けよう.

公式ドキュメント/参考

https://ai.google.dev/tutorials/node_quickstart?hl=ja#generate-text-from-text-input