23.05.27. openAI API + node.js, express로 연결해보기[1]

2023.05.27 - [daily] - 23.05.26. openAI API + node.js, express로 연결해보기[0]

23.05.26. openAI API + node.js, express로 연결해보기[0]

지난번에 이어, 대화를 이어가려면 앞의 정보를 openAI API에 같이 보내야한다. 

 

웹이나 앱의 서비스라면 사용자별로 데이터베이스에 대화를 저장해야겠지만, 간단하게 구현해보기 위해 json으로 저장하는 방식으로 대강 구현해 봤다.

 

json으로 대화를 불러오고, 저장하는 함수 작성

// 대화 로드
function loadConversation() {
    try {
        const data = fs.readFileSync(conversationFile);
        return JSON.parse(data);
    } catch (error) {
        console.error("Error loading conversation:", error);
        return [];
    }
}

// 대화 저장
function saveConversation(conversation) {
    try {
        const data = JSON.stringify(conversation);
        fs.writeFileSync(conversationFile, data);
    } catch (error) {
        console.error("Error saving conversation:", error);
    }
}

이렇게 파일을 열고 읽고 쓰고 저장하려면 fs 라이브러리가 필요하다. 그래서 기존 chatgpt.js 위에

const fs = require("fs");

로 모듈을 임포트 한다. 그리고 질문을 보내기 전에, 기존 대화를 불러와서, 대화가 연속될 수 있게 보내준다. 

그래서 전체 코드는...

const fs = require("fs");
const { Configuration, OpenAIApi } = require("openai");
require("dotenv").config();

const conversationFile = "conversation.json"; // 대화를 저장할 JSON 파일 경로

async function callChatGPT(ask) {
    const configuration = new Configuration({
        apiKey: process.env.OPENAI_API_KEY,
    });

    try {
        const openai = new OpenAIApi(configuration);
        let conversation = loadConversation(); // 대화 로드
        conversation.push({ role: "user", content: ask }); // 사용자의 질문을 대화에 추가

        const response = await openai.createChatCompletion({
            model: "gpt-3.5-turbo",
            messages: conversation,
        });

        const reply = response.data.choices[0].message;
        conversation.push(reply); // AI의 답변을 대화에 추가

        saveConversation(conversation); // 대화 저장

        return reply;
    } catch (error) {
        console.error("Error calling ChatGPT API:", error);
        return null;
    }
}

// 대화 로드
function loadConversation() {
    try {
        const data = fs.readFileSync(conversationFile);
        return JSON.parse(data);
    } catch (error) {
        console.error("Error loading conversation:", error);
        return [];
    }
}

// 대화 저장
function saveConversation(conversation) {
    try {
        const data = JSON.stringify(conversation);
        fs.writeFileSync(conversationFile, data);
    } catch (error) {
        console.error("Error saving conversation:", error);
    }
}

module.exports = { callChatGPT };

이렇게 하면 아래와 같이 끝말잇기도 가능해진다.