Merge pull request #709 from ArtBlocks/linds/integrate-insights-endpoint

Integrate Insights API

Author: lindsgil

src/Classes/InsightsBot.ts

@@ -0,0 +1,53 @@
+import * as dotenv from 'dotenv'
+dotenv.config()
+import { EmbedBuilder, Message } from 'discord.js'
+import axios from 'axios'
+import { randomColor } from '../Utils/smartBotResponse'
+
+export class InsightsBot {
+  async getInsightsApiResponse(msg: Message): Promise<EmbedBuilder> {
+    try {
+      // strip out !artBot from the message
+      const messageContent = msg.content.replace('!artBot', '').trim()
+
+      const insightsResponse = await axios.post(
+        'https://qhjte7logh.execute-api.us-east-1.amazonaws.com/production-stage/insights',
+        {
+          query: messageContent,
+        },
+        {
+          headers: {
+            'x-api-key': process.env.INSIGHTS_API_KEY ?? '',
+            'Content-Type': 'application/json',
+          },
+        }
+      )
+
+      const answer = insightsResponse?.data?.[0]?.content ?? ''
+
+      if (!answer.length) {
+        throw new Error('No answer from Insights API')
+      }
+
+      const answerWithFeedback = `${answer}\n\nThis response is AI-generated. Let us know what you think in <#769251416778604562>!`
+
+      const embed = new EmbedBuilder()
+        .setTitle('Artbot AI (Beta)')
+        .setColor(randomColor())
+        .setDescription(answerWithFeedback)
+
+      return embed
+    } catch (error) {
+      console.error('Error getting insights API response:', error)
+      return new EmbedBuilder()
+        .setTitle('Error')
+        .setColor('#FF0000')
+        .setDescription(
+          'Sorry, I encountered an error while processing your request. Please try again later.'
+        )
+        .setFooter({
+          text: 'If this persists, please contact the bot administrator.',
+        })
+    }
+  }
+}

src/Utils/smartBotResponse.ts

@@ -1,6 +1,6 @@
 import { EmbedBuilder, ColorResolvable, Message } from 'discord.js'
 import * as dotenv from 'dotenv'
-import { artIndexerBot, projectConfig, triviaBot } from '..'
+import { artIndexerBot, projectConfig, triviaBot, insightsBot } from '..'
 dotenv.config()
 const fetch = require('node-fetch')
 
@@ -162,6 +162,7 @@ const HELP_MESSAGE = new EmbedBuilder()
     **staysafe?**: Tips on avoiding scams
     **aliases?**: A handy list of aliases that can be used in \`#\` commands.
     **hashtag?**: A handy list of all \`#\` functionalities
+    **artBot**: Ask about Art Blocks artists and projects by tagging @artbot at the start of your question. This AI-powered feature is in beta.
     `
   )
 // Custom message shown when someone asks about ArtBlocks
@@ -308,6 +309,7 @@ export async function smartBotResponse(
     }
     return null
   }
+
   // Some shared helper variables.
   const inHelpChannel: boolean = channelID == CHANNEL_HELP
   const mentionedArtBot: boolean =
@@ -398,6 +400,11 @@ export async function smartBotResponse(
     )
   }
 
+  if (channelID == CHANNEL_BLOCK_TALK && containsQuestion && mentionedArtBot) {
+    // Handle messages to the Insights API for AI responses
+    return insightsBot.getInsightsApiResponse(msg)
+  }
+
   if (
     (channelID == CHANNEL_FOR_SALE_LISTINGS ||
       channelID == CHANNEL_TRADE_SWAPS) &&

src/index.ts

@@ -21,6 +21,7 @@ import {
   getEngineProjects,
   getStudioContracts,
 } from './Data/queryGraphQL'
+import { InsightsBot } from './Classes/InsightsBot'
 import { TriviaBot } from './Classes/TriviaBot'
 import {
   waitForEngineContracts,
@@ -204,6 +205,8 @@ discordClient.on('disconnect', () => {
 })
 
 export const triviaBot = new TriviaBot(discordClient)
+export const insightsBot = new InsightsBot()
+
 new ScheduleBot(discordClient.channels.cache, projectConfig)
 
 discordClient.on(Events.MessageCreate, async (msg) => {