Add open API key config

.github/cr-gpt.yml

@@ -0,0 +1 @@
+OPENAI_API_KEY: sk-dtIzk2nobuSuTdnw7G3nT3BlbkFJNCjTGHzgnaskUeB11M1HJ6111

src/bot.ts

@@ -1,8 +1,9 @@
-import { Probot } from 'probot';
+import { Context, Probot } from 'probot';
 import { Chat } from './chat.js';
 
+const OPENAI_API_KEY = 'OPENAI_API_KEY';
+
 export const robot = (app: Probot) => {
-  const chat = new Chat();
   // const getDiff = async (context: Context, pullRequestNumber: number) => {
   //   const repo = context.repo();
 
@@ -26,7 +27,31 @@ export const robot = (app: Probot) => {
   //   return diff as unknown as string;
   // };
 
+  const loadChat = async (context: Context) => {
+    const repo = context.repo();
+    const { data } = (await context.octokit.request(
+      'GET /repos/{owner}/{repo}/actions/variables/{name}',
+      {
+        owner: repo.owner,
+        repo: repo.repo,
+        name: OPENAI_API_KEY,
+      }
+    )) as any;
+
+    if (!data?.value) {
+      return null;
+    }
+
+    return new Chat(data.value);
+  };
+
   app.on('pull_request.opened', async (context) => {
+    const chat = await loadChat(context);
+
+    if (!chat) {
+      return 'no chat initialized';
+    }
+
     async function cr() {
       const issueComment = context.issue({
         body: await chat.codeReview(
@@ -56,6 +81,12 @@ export const robot = (app: Probot) => {
       return;
     }
 
+    const chat = await loadChat(context);
+
+    if (!chat) {
+      return 'no chat initialized';
+    }
+
     async function cr() {
       // const diff = await getDiff(context, context.payload.issue.number);
       const issueComment = context.issue({

src/chat.ts

@@ -1,10 +1,13 @@
 import { ChatGPTAPI } from 'chatgpt';
+export class Chat {
+  private chatAPI: ChatGPTAPI;
 
-const chatAPI = new ChatGPTAPI({
-  apiKey: process.env.OPENAI_API_KEY as string,
-});
+  constructor(apikey: string) {
+    this.chatAPI = new ChatGPTAPI({
+      apiKey: apikey,
+    });
+  }
 
-export class Chat {
   private generatePrompt = (_: string, patch: string) => {
     return `
     Act as a code reviewer of a Pull Request, providing feedback on the code changes below.
@@ -33,7 +36,9 @@ export class Chat {
     console.time('code-review cost');
     console.log('start query chatGPT');
 
-    const res = await chatAPI.sendMessage(this.generatePrompt(title, patch));
+    const res = await this.chatAPI.sendMessage(
+      this.generatePrompt(title, patch)
+    );
 
     console.log('end query chatGPT');
     console.timeEnd('code-review cost');