2024-01-10

OpenAI API を TypeScript で叩いてみた

はじめに

表題の通り試しに触ってみたが、検索して出てきた記事では書き方が古いものが多かったので現時点での実装を書いておく。

使用したのは以下のライブラリで、バージョンは v4.24.2 を使用した。

GitHub - openai/openai-node: Official JavaScript / TypeScript library for the OpenAI API

Official JavaScript / TypeScript library for the OpenAI API - openai/openai-node

github.com

GitHub - openai/openai-node: Official JavaScript / TypeScript library for the OpenAI API

v3 から v4 へのマイグレーションガイドはこちら。メソッド自体やその使い方に変更があるので、新しいバージョンを利用する場合はこちらを参照すればいいと思う。

github.com

サンプルコード

main.ts

import OpenAI from 'openai';

const openai = new OpenAI({
  apiKey: process.env.OPENAI_API_KEY
});

async function main() {
  try {
    const stream = await openai.chat.completions.create({
      model: 'gpt-3.5-turbo',
      stream: true,
      // prompt: "TypeScript で Hello World を書いてください。"
      messages: [
        {
          role: 'user',
          content: 'TypeScript で Hello World を書いてください。'
        }
      ]
    });

    for await (const chunk of stream) {
      process.stdout.write(chunk.choices[0].delta?.content || '');
    }
  } catch (error) {
    throw new Error(error as string);
  }
}

main().catch((e) => console.error(e));

上記のコードを保存して実行すると以下のような結果が得られる。パラメータの model は好きなものを選ぶと良い。¹

$ ts-node ./src/main.ts

以下は、TypeScriptでHello Worldを表示するサンプルコードです。

\`\`\`
const message: string = "Hello, World!";
console.log(message);
\`\`\`

これを、`.ts`拡張子のファイルに保存し、TypeScriptコンパイラでコンパイルした後、JavaScriptファイルとして実行すると、"Hello, World!"というメッセージが表示されます。%