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;

import OpenAI from 'openai';

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 openai = new OpenAI({
  apiKey: process.env.OPENAI_API_KEY
});

const prompt: OpenAI.Chat.ChatCompletionMessageParam = {
  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.chat.completions.create({
  model: 'gpt-4',
  messages: [prompt],
  tool_choice: 'auto',
  tools: [{
    type: 'function',
    function: {
      name: 'getCompanyName',
      description: 'Retrieve the company to which the user belongs.',
      parameters: {
        type: 'object',
        properties: {
          name: {
            type: 'string',
            description: 'The user name'
          }
        },
        required: ['name']
      }
    }
  }, {
    type: 'function',
    function: {
      name: 'getBirthday',
      description: 'Retrieve the user\'s birthday.',
      parameters: {
        type: 'object',
        properties: {
          name: {
            type: 'string',
            description: 'The user name'
          }
        },
        required: ['name']
      }
    }
  }]
});

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

{
  "role": "assistant",
  "content": null,
  "tool_calls": [
    {
      "id": "call_0xBlsazt2SlXGRNc3rKmfIx2",
      "type": "function",
      "function": {
        "name": "getBirthday",
        "arguments": "{\n  \"name\": \"mamezou\"\n}"
      }
    }
  ]
}

const [functionCall] = message1?.tool_calls ?? [];
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.function.arguments || '{}');
// @ts-ignore
const funcResponse = (functions[functionCall.function.name!])(args.name);

// 関数の実行結果に連携
const response2 = await openai.chat.completions.create(
  {
    model: 'gpt-4',
    messages: [prompt, message1, {
      role: 'tool',
      content: funcResponse,
      tool_call_id: functionCall.id
    }]
  }
);
console.log(response2.choices[0].message);

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