OpenAIのChat APIに追加されたFunction callingを使ってみる

const functions = {
  getBirthday(name: string): string {
    return name === 'mamezou' ? '1999-11-11' : '2000-01-01';
  },
  getCompanyName(name: string): string {
    return name === 'mamezou' ? '株式会社豆蔵' : 'その他';
  }
} as const;

const configuration = new Configuration({
  apiKey: process.env.OPENAI_API_KEY
});
const openai = new OpenAIApi(configuration);

const prompt: ChatCompletionRequestMessage = {
  role: 'user',
  content: 'Tell me about news in Japan that happened in the year mamezou was born.Your output, except for function calling, should be in japanese.'
  // content: 'Tell me the name of the company mamezou works for?.' // getCompanyNameを実行する場合
};

const response1 = await openai.createChatCompletion({
  model: 'gpt-4-0613', // or gpt-3.5-turbo-0613
  messages: [prompt],
  function_call: 'auto',
  functions: [{
    name: 'getCompanyName',
    description: 'Retrieve the company to which the user belongs.',
    parameters: {
      type: 'object',
      properties: {
        name: {
          type: 'string',
          description: 'The user name'
        }
      },
      required: ['name']
    }
  }, {
    name: 'getBirthday',
    description: 'Retrieve the user\'s birthday.',
    parameters: {
      type: 'object',
      properties: {
        name: {
          type: 'string',
          description: 'The user name'
        }
      },
      required: ['name']
    }
  }]
});

const message1 = response1.data.choices[0].message;

{
  "role": "assistant",
  "content": null,
  "function_call": {
    "name": "getBirthday",
    "arguments": "{\n  \"name\": \"mamezou\"\n}"
  }
}

const functionCall = message1?.function_call;
if (!functionCall) {
  console.log('Sorry, Chat API made the decision not to call the function...');
  process.exit(1);
}

// Chat APIが指定した関数を実行
const args = JSON.parse(functionCall.arguments || '{}');
// @ts-ignore
const funcResponse = (functions[functionCall.name!])(args.name);

// 関数の実行結果に連携
const response2 = await openai.createChatCompletion(
  {
    model: 'gpt-4-0613', // or gpt-3.5-turbo-0613
    messages: [prompt, message1, {
      role: 'function',
      content: funcResponse,
      name: functionCall.name
    }]
  }
);
console.log(response2.data.choices[0].message);

{
  "role": "assistant",
  "content": "1999年、つまり豆蔵が生まれた年には、たくさんの出来事がありました。その中でも特に注目すべきは以下のようなニュースです。\n\n1. 茨城県で発生した茨城県南部地震が記憶に新しい。最大震度6弱を記録し、被害が広範囲に及んだ。\L3DO FZ-10」を発表した。これは、家庭用ゲーム機としては初めて3D映像を出力することが可能だった。\n\n4. また、この年には、小惑星「エロス」への無人探査機「NEARショーメーカー」が打ち上げられた。この探査機は、地球から約2億km離れたエロスでは、アニメ映画「となりのトトロ」が大ヒットし、その人気は未だ衰えない 。 同年、宮崎駿の「もののけ姫」が公開され、国内外で大きな話題を呼んだ。\n\n以上、豆蔵が生まれた1999年の主な出来事をご紹介しました。"
}