From 3d48bfce826fa2772641effb1099924acccb4183 Mon Sep 17 00:00:00 2001
From: derpystuff <3515180-derpystuff@users.noreply.gitlab.com>
Date: Tue, 29 Aug 2023 21:46:34 +0200
Subject: [PATCH] allow changing temperature

---
 commands/message/fun/chat.js | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/commands/message/fun/chat.js b/commands/message/fun/chat.js
index fb9596c..066e4bc 100644
--- a/commands/message/fun/chat.js
+++ b/commands/message/fun/chat.js
@@ -21,6 +21,7 @@ module.exports = {
   },
   args: [
     { name: 'prompt', default: '', required: false, help: "The starting system prompt." },
+    { name: 'temperature', default: 0.5, required: false, help: "Model temperature." },
     { name: 'model', default: 'CHATGPT', required: false, help: "The model to use. (Restricted to CHATGPT)" },
   ],
   permissionsClient: [Permissions.EMBED_LINKS, Permissions.SEND_MESSAGES, Permissions.ATTACH_FILES, Permissions.USE_EXTERNAL_EMOJIS, Permissions.READ_MESSAGE_HISTORY],
@@ -38,6 +39,13 @@ module.exports = {
       model = args.model
       modelDisplay = "  " + smallIconPill("robot", model) 
     }
+    
+    let temperature = "CHATGPT"
+    let temperatureDisplay = ""
+    if(args.temperature){
+      temperature = parseFloat(args.temperature)
+      temperatureDisplay = "  " + smallIconPill("example", temperature) 
+    }
 
     try{
       await editOrReply(context, createEmbed("ai", context, "Generating response..."))
@@ -47,10 +55,10 @@ module.exports = {
           Authorization: process.env.AI_SERVER_KEY
         })
         .send({
-          prompt: prompt,
+          prompt,
           input: [args.text],
-          temperature: 0.6,
-          model: model
+          temperature,
+          model
         })
 
       let description = [smallIconPill("generative_ai", args.text) + modelDisplay, '']