rj1
about | log | files | refs

openai.js (1267B) - raw

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import { Configuration, OpenAIApi } from "openai";
import fs from "fs";
import * as dotenv from "dotenv";

dotenv.config();

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

const openai = new OpenAIApi(configuration);

export async function generateComment(subText, sentiment) {
  // cut off the text at 1024 tokens, 1 token is ~4 characters
  if (subText.length > 4096) {
    subText = subText.substring(0, 4096);
  }

  const response = await openai.createChatCompletion({
    model: "gpt-3.5-turbo",
    messages: [
      {
        role: "system",
        content: `You are an assistant that generates ${sentiment} comments on YouTube videos, based on the content of the subtitles. I'll send you the subtitle text, and you write a short, ${sentiment} comment. Please don't encapsulate the comment in quotes. Only include the comment that would fit into the youtube comment section.`,
      },
      { role: "user", content: subText },
    ],
    temperature: 0.7,
    max_tokens: 1024,
    top_p: 1.0,
    frequency_penalty: 0.0,
    presence_penalty: 0.0,
  });

  let comment = response.data.choices[0].message.content;
  comment = comment.replace(/"/g, "");

  return comment;
}