Ask Command

.env.example

@@ -10,6 +10,7 @@ SUPABASE_KEY=
 AUTO_PAY_MODE=
 ANALYTICS_MODE=
 
+
 # Use `trace` to get verbose logging or `info` to show less
 LOG_LEVEL=debug
 LOGDNA_INGESTION_KEY=

package.json

@@ -55,6 +55,7 @@
     "node-html-parser": "^6.1.5",
     "node-html-to-image": "^3.3.0",
     "nodemon": "^2.0.19",
+    "openai": "^4.2.0",
     "parse5": "^7.1.2",
     "prettier": "^2.7.1",
     "probot": "^12.2.4",

src/bindings/config.ts

@@ -34,6 +34,8 @@ export const loadConfig = async (context: Context): Promise<BotConfig> => {
     registerWalletWithVerification,
     staleBountyTime,
     enableAccessControl,
+    openAIKey,
+    openAITokenLimit,
   } = await getWideConfig(context);
 
   const publicKey = await getScalarKey(process.env.X25519_PRIVATE_KEY);
@@ -99,6 +101,10 @@ export const loadConfig = async (context: Context): Promise<BotConfig> => {
     wallet: {
       registerWalletWithVerification: registerWalletWithVerification,
     },
+    ask: {
+      apiKey: openAIKey,
+      tokenLimit: openAITokenLimit,
+    },
     accessControl: enableAccessControl,
   };
 

src/configs/shared.ts

@@ -27,3 +27,5 @@ export const DEFAULT_DISQUALIFY_TIME = "7 days"; // 7 days
 export const DEFAULT_NETWORK_ID = 1; // ethereum
 export const DEFAULT_RPC_ENDPOINT = "https://rpc-bot.ubq.fi/v1/mainnet";
 export const DEFAULT_PERMIT_BASE_URL = "https://pay.ubq.fi";
+
+export const DEFAULT_TOKEN_LIMIT = 4000;

src/handlers/comment/commands.ts

@@ -6,6 +6,7 @@ export enum IssueCommentCommands {
   PAYOUT = "/payout", // request permit payout
   MULTIPLIER = "/multiplier", // set bounty multiplier (for treasury)
   QUERY = "/query",
+  ASK = "/ask", // ask GPT a question
   // Access Controls
 
   ALLOW = "/allow",

src/handlers/comment/handlers/ask.ts

@@ -0,0 +1,123 @@
+import { getBotContext, getLogger } from "../../../bindings";
+import { GPTResponse, Payload, StreamlinedComment, UserType } from "../../../types";
+import { getAllIssueComments, getAllLinkedIssuesAndPullsInBody } from "../../../helpers";
+import { CreateChatCompletionRequestMessage } from "openai/resources/chat";
+import { askGPT, decideContextGPT, sysMsg } from "../../../helpers/gpt";
+import { ErrorDiff } from "../../../utils/helpers";
+
+/**
+ * @param body The question to ask
+ */
+export const ask = async (body: string) => {
+  const context = getBotContext();
+  const logger = getLogger();
+
+  const payload = context.payload as Payload;
+  const sender = payload.sender.login;
+  const issue = payload.issue;
+
+  if (!body) {
+    return `Please ask a question`;
+  }
+
+  if (!issue) {
+    return `This command can only be used on issues`;
+  }
+
+  const chatHistory: CreateChatCompletionRequestMessage[] = [];
+  const streamlined: StreamlinedComment[] = [];
+  let linkedPRStreamlined: StreamlinedComment[] = [];
+  let linkedIssueStreamlined: StreamlinedComment[] = [];
+
+  const regex = /^\/ask\s(.+)$/;
+  const matches = body.match(regex);
+
+  if (matches) {
+    const [, body] = matches;
+
+    // standard comments
+    const comments = await getAllIssueComments(issue.number);
+    // raw so we can grab the <!--- { 'UbiquityAI': 'answer' } ---> tag
+    const commentsRaw = await getAllIssueComments(issue.number, "raw");
+
+    if (!comments) {
+      logger.info(`Error getting issue comments`);
+      return ErrorDiff(`Error getting issue comments`);
+    }
+
+    // add the first comment of the issue/pull request
+    streamlined.push({
+      login: issue.user.login,
+      body: issue.body,
+    });
+
+    // add the rest
+    comments.forEach(async (comment, i) => {
+      if (comment.user.type == UserType.User || commentsRaw[i].body.includes("<!--- { 'UbiquityAI': 'answer' } --->")) {
+        streamlined.push({
+          login: comment.user.login,
+          body: comment.body,
+        });
+      }
+    });
+
+    // returns the conversational context from all linked issues and prs
+    const links = await getAllLinkedIssuesAndPullsInBody(issue.number);
+
+    if (typeof links === "string") {
+      logger.info(`Error getting linked issues or prs: ${links}`);
+    } else {
+      linkedIssueStreamlined = links.linkedIssues;
+      linkedPRStreamlined = links.linkedPrs;
+    }
+
+    // let Chad deduce what is the most relevant context
+    const gptDecidedContext = await decideContextGPT(chatHistory, streamlined, linkedPRStreamlined, linkedIssueStreamlined);
+
+    if (linkedIssueStreamlined.length == 0 && linkedPRStreamlined.length == 0) {
+      // No external context to add
+      chatHistory.push(
+        {
+          role: "system",
+          content: sysMsg,
+          name: "UbiquityAI",
+        } as CreateChatCompletionRequestMessage,
+        {
+          role: "user",
+          content: body,
+          name: sender,
+        } as CreateChatCompletionRequestMessage
+      );
+    } else {
+      chatHistory.push(
+        {
+          role: "system",
+          content: sysMsg, // provide the answer template
+          name: "UbiquityAI",
+        } as CreateChatCompletionRequestMessage,
+        {
+          role: "system",
+          content: "Original Context: " + JSON.stringify(gptDecidedContext), // provide the context
+          name: "system",
+        } as CreateChatCompletionRequestMessage,
+        {
+          role: "user",
+          content: "Question: " + JSON.stringify(body), // provide the question
+          name: "user",
+        } as CreateChatCompletionRequestMessage
+      );
+    }
+
+    const gptResponse: GPTResponse | string = await askGPT(body, chatHistory);
+
+    if (typeof gptResponse === "string") {
+      return gptResponse;
+    } else if (gptResponse.answer) {
+      return gptResponse.answer;
+    } else {
+      return ErrorDiff(`Error getting response from GPT`);
+    }
+  } else {
+    return "Invalid syntax for ask \n usage: '/ask What is pi?";
+  }
+};

src/handlers/comment/handlers/index.ts

@@ -7,6 +7,7 @@ import { listAvailableCommands } from "./help";
 import { unassign } from "./unassign";
 import { registerWallet } from "./wallet";
 import { setAccess } from "./allow";
+import { ask } from "./ask";
 import { multiplier } from "./multiplier";
 import { BigNumber, ethers } from "ethers";
 import { addPenalty } from "../../../adapters/supabase";
@@ -36,6 +37,7 @@ export * from "./payout";
 export * from "./help";
 export * from "./multiplier";
 export * from "./query";
+export * from "./ask";
 
 /**
  * Parses the comment body and figure out the command name a user wants
@@ -251,6 +253,12 @@ export const userCommands = (): UserCommands[] => {
       handler: query,
       callback: commandCallback,
     },
+    {
+      id: IssueCommentCommands.ASK,
+      description: `Ask a technical question to the Ubiquity AI. \n  example usage: "/ask How do I do X?"`,
+      handler: ask,
+      callback: commandCallback,
+    },
     {
       id: IssueCommentCommands.MULTIPLIER,
       description: `Set the bounty payout multiplier for a specific contributor, and provide the reason for why. \n  example usage: "/wallet @user 0.5 'Multiplier reason'"`,