From 32ef0811ab8883778218d70b0c527359079f4551 Mon Sep 17 00:00:00 2001
From: derpystuff <3515180-derpystuff@users.noreply.gitlab.com>
Date: Sat, 14 Oct 2023 23:56:32 +0200
Subject: [PATCH] bard upgrade

---
 commands/message/fun/bard.js |  30 +---------
 commands/message/fun/palm.js | 105 +++++++++++++++++++++++++++++++++++
 2 files changed, 108 insertions(+), 27 deletions(-)
 create mode 100644 commands/message/fun/palm.js

diff --git a/commands/message/fun/bard.js b/commands/message/fun/bard.js
index 9758b76..991bbb6 100644
--- a/commands/message/fun/bard.js
+++ b/commands/message/fun/bard.js
@@ -12,13 +12,12 @@ const { Permissions } = require("detritus-client/lib/constants");
 module.exports = {
   name: 'bard',
   label: 'text',
-  aliases: ['palm2'],
   metadata: {
     description: `${iconPill("generative_ai", "LIMITED TESTING")}\n\nTalk to Bard.`,
     description_short: 'Chat with Bard.',
     examples: ['bard How many otter species are there?'],
     category: 'limited',
-    usage: 'bard <input> [-prompt <prompt override>]'
+    usage: 'bard <input>'
   },
   args: [
     { name: 'prompt', default: '', required: false, help: "The starting system prompt." },
@@ -32,26 +31,6 @@ module.exports = {
     if(!args.text) return editOrReply(context, {embeds:[createEmbed("warning", context, `Missing Parameter (text).`)]})
 
     let input = args.text;
-    
-    let prompt = `You are a friendly chat bot designed to help people.\n- Today\'s date is ${new Date().toLocaleDateString('en-us', { weekday:"long", year:"numeric", month:"long", day:"numeric"})}\n- You should always use gender neutral pronouns when possible.\n- When answering a question, be concise and to the point.\n- Try to keep responses below 1000 characters. This does not apply to subjects that require more exhaustive or in-depth explanation.`
-    if(args.prompt !== "") prompt = args.prompt
-
-    // Get content if the user replies to anything
-    if(context.message.messageReference) {
-      let msg = await context.message.channel.fetchMessage(context.message.messageReference.messageId);
-
-      if(msg.content && msg.content.length) input = msg.content
-      else if(msg.embeds?.length) for(const e of msg.embeds) if(e[1].description?.length) { input = e[1].description; break; } 
-
-      prompt = args.text
-      if(args.prompt !== "") return editOrReply(context, {embeds:[createEmbed("warning", context, `Prompt parameter is unsupported for message replies.`)]})
-    }
-
-    let model = "chat-bison-001"
-    if(args.model && isLimitedTestUser(context.user)) model = args.model
-    
-    let temperature = "0.25"
-    if(args.temperature !== 0.25) temperature = parseFloat(args.temperature)
 
     let inputDisplay = args.text.replace(/\n/g, ' ')
     if(inputDisplay.length >= 50) inputDisplay = inputDisplay.substr(0,50) + '...'
@@ -59,15 +38,12 @@ module.exports = {
     try{
       await editOrReply(context, createEmbed("ai_bard", context, inputDisplay))
 
-      let res = await superagent.post(`${process.env.AI_SERVER}/google/palm2`)
+      let res = await superagent.post(`${process.env.AI_SERVER}/google/bard`)
         .set({
           Authorization: process.env.AI_SERVER_KEY
         })
         .send({
-          prompt,
-          input: [input],
-          temperature,
-          model
+          input
         })
 
       let description = []
diff --git a/commands/message/fun/palm.js b/commands/message/fun/palm.js
new file mode 100644
index 0000000..c60bebb
--- /dev/null
+++ b/commands/message/fun/palm.js
@@ -0,0 +1,105 @@
+const { createEmbed } = require('../../../labscore/utils/embed')
+const { editOrReply } = require('../../../labscore/utils/message')
+
+const { canUseLimitedTestCommands, isLimitedTestUser } = require('../../../labscore/utils/testing')
+const { STATICS, STATIC_ICONS } = require('../../../labscore/utils/statics');
+
+const superagent = require('superagent')
+const { iconPill, smallIconPill, icon } = require('../../../labscore/utils/markdown')
+
+const { Permissions } = require("detritus-client/lib/constants");
+
+module.exports = {
+  name: 'palm',
+  label: 'text',
+  aliases: ['palm2'],
+  metadata: {
+    description: `${iconPill("generative_ai", "LIMITED TESTING")}\n\nTalk to Bard.`,
+    description_short: 'Chat with Palm2.',
+    examples: ['bard How many otter species are there?'],
+    category: 'limited',
+    usage: 'palm <input> [-prompt <prompt override>]'
+  },
+  args: [
+    { name: 'prompt', default: '', required: false, help: "The starting system prompt." },
+    { name: 'temperature', default: 0.25, required: false, help: "Model temperature." },
+  ],
+  permissionsClient: [Permissions.EMBED_LINKS, Permissions.SEND_MESSAGES, Permissions.ATTACH_FILES, Permissions.USE_EXTERNAL_EMOJIS, Permissions.READ_MESSAGE_HISTORY],
+  run: async (context, args) => {
+    if(!canUseLimitedTestCommands(context)) return;
+
+    context.triggerTyping();
+    if(!args.text) return editOrReply(context, {embeds:[createEmbed("warning", context, `Missing Parameter (text).`)]})
+
+    let input = args.text;
+    
+    let prompt = `You are a friendly chat bot designed to help people.\n- Today\'s date is ${new Date().toLocaleDateString('en-us', { weekday:"long", year:"numeric", month:"long", day:"numeric"})}\n- You should always use gender neutral pronouns when possible.\n- When answering a question, be concise and to the point.\n- Try to keep responses below 1000 characters. This does not apply to subjects that require more exhaustive or in-depth explanation.`
+    if(args.prompt !== "") prompt = args.prompt
+
+    // Get content if the user replies to anything
+    if(context.message.messageReference) {
+      let msg = await context.message.channel.fetchMessage(context.message.messageReference.messageId);
+
+      if(msg.content && msg.content.length) input = msg.content
+      else if(msg.embeds?.length) for(const e of msg.embeds) if(e[1].description?.length) { input = e[1].description; break; } 
+
+      prompt = args.text
+      if(args.prompt !== "") return editOrReply(context, {embeds:[createEmbed("warning", context, `Prompt parameter is unsupported for message replies.`)]})
+    }
+
+    let model = "chat-bison-001"
+    if(args.model && isLimitedTestUser(context.user)) model = args.model
+    
+    let temperature = "0.25"
+    if(args.temperature !== 0.25) temperature = parseFloat(args.temperature)
+
+    let inputDisplay = args.text.replace(/\n/g, ' ')
+    if(inputDisplay.length >= 50) inputDisplay = inputDisplay.substr(0,50) + '...'
+
+    try{
+      await editOrReply(context, createEmbed("ai_bard", context, inputDisplay))
+
+      let res = await superagent.post(`${process.env.AI_SERVER}/google/palm2`)
+        .set({
+          Authorization: process.env.AI_SERVER_KEY
+        })
+        .send({
+          prompt,
+          input: [input],
+          temperature,
+          model
+        })
+
+      let description = []
+      let files = [];
+      
+      if(!res.body.output) return editOrReply(context, {embeds:[createEmbed("error", context, `Bard returned an error. Try again later.`)]}) 
+
+      if(res.body.output.length <= 4000) description.push(res.body.output)
+      else {
+        files.push({
+          filename: `chat.${Date.now().toString(36)}.txt`,
+          value: Buffer.from(res.body.output)
+        })
+      }
+
+      return editOrReply(context, {
+        embeds:[createEmbed("defaultNoFooter", context, {
+          author: {
+            name: inputDisplay,
+            iconUrl: STATIC_ICONS.ai_bard_idle
+          },
+          description: description.join('\n'),
+          footer: {
+            text: `Palm2 • This information may be inaccurate or biased`
+          }
+        })],
+        files
+      })
+    }catch(e){
+      if(e.response.body?.message) return editOrReply(context, {embeds:[createEmbed("warning", context, e.response.body.message)]})
+      console.log(e)
+      return editOrReply(context, {embeds:[createEmbed("error", context, `Unable to generate text.`)]})
+    }
+  }
+};
\ No newline at end of file