Experiment of OpenAI Function Calling

openai_function_calling.ts

import { ChatCompletionRequestMessage, Configuration, OpenAIApi } from "openai";
var prompt = require("prompt-sync")();
require("dotenv").config();

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

async function getWeather(location: string, unit = "fahrenheit") {
  // Get the current weather in a given location
  const weatherInfo = {
    location: location,
    temperature: "72",
    unit: unit,
    forecast: ["sunny", "windy"],
  };
  return weatherInfo;
}

async function getStockPrice(stockCode: string) {}

async function main() {
  let input = "test";
  const messages: ChatCompletionRequestMessage[] = [
    { role: "user", content: "Hello world" },
  ];

  const functions = [
    {
      name: "getWeather",
      description: "Get the current weather in a given location",
      parameters: {
        type: "object",
        properties: {
          location: {
            type: "string",
            description: "The city and state, e.g. San Francisco, CA",
          },
          unit: { type: "string", enum: ["celsius", "fahrenheit"] },
        },
        required: ["location"],
      },
    },
    {
      name: "getStockPrice",
      description: "Get the current stock price of a given company stock code",
      parameters: {
        type: "object",
        properties: {
          stockCode: {
            type: "string",
            description: "The company stock code, e.g. AAPL, GOOG",
          },
        },
        required: ["stockCode"],
      },
    },
  ];

  while (input) {
    const chatCompletion = await openai.createChatCompletion({
      model: "gpt-3.5-turbo-0613",
      messages: messages,
      functions: functions,
    });

    const response = chatCompletion.data.choices[0].message;
    console.log(response?.content);

    if (response?.function_call) {
      console.log("function call", response?.function_call);
      // Handle logic here
    }

    const input = prompt(">");

    if (response?.content) {
      messages.push({
        role: "system",
        content: response?.content as any,
      });
    }

    messages.push({
      role: "user",
      content: input as any,
    });
  }
}

main()
  .then((result) => {
    console.log("Finish");
  })
  .catch((error) => console.log(error?.response?.data));