SiriGPT Tutorial

1// @see https://docs.aircode.io/guide/functions/
2const aircode = require('aircode');
3const { Configuration, OpenAIApi } = require('openai');
4const { v4: uuidv4 } = require('uuid');
5
6const { db } = aircode;
7const ChatTable = db.table('chat');
8
9// Setup OpenAI configurations
10const OPENAI_KEY = process.env.OPENAI_KEY || '';
11// Use the latest OpenAI GPT-3.5 model. If the next 4 is released, modify this parameter
12// OpenAI models parameter list https://platform.openai.com/docs/models
13const OPENAI_MODEL = process.env.MODEL || 'gpt-3.5-turbo';
14const MAX_MESSAGES_PER_CHAT = 40;
15
16const systemContent = 'You are a helpful assistant.';
17
18module.exports = async function (params, context) {
19  console.log('Received params:', params);
20  const { question, cid } = params;
21
22  // Create a chat ID if not provided
23  const chatId = cid ? cid : uuidv4();
24
25  // Save user's question to the ChatTable
26  await ChatTable.save({ chatId, role: 'user', content: question });
27
28  // Retrieve chat history
29  const chats = await ChatTable.where({ chatId })
30    .sort({ createdAt: -1 })
31    .limit(MAX_MESSAGES_PER_CHAT)
32    .find();
33
34  // Construct message array for ChatGPT
35  const messages = [
36    { role: 'system', content: 'You are a helpful assistant.' },
37    ...chats.reverse().map((one) => ({ role: one.role, content: one.content })),
38  ];
39
40  const openai = new OpenAIApi(new Configuration({ apiKey: OPENAI_KEY }));
41
42  try {
43    // Request completion from ChatGPT
44    const completion = await openai.createChatCompletion({
45      model: OPENAI_MODEL,
46      messages,
47      temperature: 1,
48      n: 1,
49      stream: false,
50    });
51
52    const responseMessage = completion.data.choices[0].message;
53
54    // Save generated response to ChatTable
55    await ChatTable.save({ chatId, ...responseMessage });
56
57    // Return response message and chat ID
58    return { reply: responseMessage.content, cid: chatId };
59  } catch (error) {
60    // Set the response status to 500 (Internal Server Error)
61    context.status(500);
62    // Log the error
63    console.log('error', error.response || error);
64
65    // Initialize an error message variable
66    let errorMessage;
67
68    // If there is a response object in the error,
69    // it means the request was made and the server responded with an error status
70    if (error.response) {
71      const { status, statusText, data } = error.response;
72
73      if (status === 401) {
74        // If the status code is 401, set a specific error message related to the OpenAI API key
75        errorMessage =
76          'Unauthorized: Invalid OpenAI API key, please check your API key in the AirCode Environments tab.';
77      } else if (data.error && data.error.message) {
78        // If there is an error message in the data, use it as the error message
79        errorMessage = data.error.message;
80      } else {
81        // Otherwise, use the status code and status text as the error message
82        errorMessage = `Request failed with status code ${status}: ${statusText}`;
83      }
84    } else if (error.request) {
85      // If there is a request object in the error,
86      // it means the request was made but no response was received
87      errorMessage = 'No response received from the server';
88    } else if (error.code === 'ENOTFOUND' || error.code === 'ECONNREFUSED') {
89      // If there is a network error, such as DNS resolution or connection refused
90      errorMessage = `Network error: ${error.message}`;
91    } else {
92      // If none of the above conditions are met,
93      // it means there was an error setting up the request
94      errorMessage = `Request setup error: ${error.message}`;
95    }
96
97    // Return an object containing the error message
98    return { error: errorMessage };
99  }
100};
101