Merge pull request #135 from room302studio/store-robot-responses

feat(remember): add function to remember robot responses

src/memory.js

@@ -4,6 +4,7 @@ const {
   getAllMemories,
   storeUserMemory,
   getRelevantMemories,
+  storeRobotMessage,
 } = require("./remember.js");
 const {
   createTodo,
@@ -21,8 +22,6 @@ const DISABLE_MEMORIES = true; // Set to false to enable memory storage
 const DISABLE_TASK_ANALYSIS = true; // Set to false to enable task analysis
 const DISABLE_REMEMBER_COMPLETIONS = true; // Set to false to enable remember completions
 
-const { getPromptsFromSupabase, getConfigFromSupabase } = require("../helpers");
-
 // Initialize prompts and config
 let PROMPT_REMEMBER,
   PROMPT_CAPABILITY_REMEMBER,
@@ -115,6 +114,12 @@ module.exports = {
       logger.info(`Processed response: ${JSON.stringify(processedResponse)}`);
       logger.info(`Response processed, ${processedResponse.length} characters`);
 
+      // store the robot message response
+      await storeRobotMessage(
+        { username, channel, conversation_id: channel, related_message_id },
+        processedResponse
+      );
+
       let rememberText = "";
       if (!DISABLE_REMEMBER_COMPLETIONS) {
         try {

src/remember.js

@@ -361,6 +361,30 @@ async function storeUserMessage(
   return data[0].id;
 }
 
+// now, a similar function to store robot response messages
+async function storeRobotMessage({ channel, guild, conversation_id }, value) {
+  const { getConfigFromSupabase } = require("../helpers.js");
+  const { supabase } = require("./supabaseclient.js");
+  const { NAME } = await getConfigFromSupabase();
+  const { data, error } = await supabase
+    // .from("messages")
+    .from(MESSAGES_TABLE_NAME)
+    .insert({
+      user_id: NAME,
+      channel_id: channel,
+      guild_id: guild,
+      conversation_id: conversation_id,
+      value,
+    })
+    .select();
+
+  if (error) {
+    logger.info(`Error storing robot message: ${error.message}`);
+  }
+
+  return data[0].id;
+}
+
 /**
  * Retrieves the message history of a user.
  * @param {string} userId - The ID of the user.
@@ -600,4 +624,5 @@ module.exports = {
   getMemoriesBetweenDates,
   getMemoriesByString,
   deleteMemoriesOfResource,
+  storeRobotMessage,
 };