@saltcorn/agents 0.8.5 → 0.8.6

package/consolidate_agent_memory.js

@@ -0,0 +1,65 @@
+const FieldRepeat = require("@saltcorn/data/models/fieldrepeat");
+const Table = require("@saltcorn/data/models/table");
+const View = require("@saltcorn/data/models/view");
+const { p } = require("@saltcorn/markup/tags");
+const { get_skills, process_interaction, wrapSegment } = require("./common");
+const { applyAsync } = require("@saltcorn/data/utils");
+const WorkflowRun = require("@saltcorn/data/models/workflow_run");
+const { interpolate, escapeHtml } = require("@saltcorn/data/utils");
+const { getState } = require("@saltcorn/data/db/state");
+
+module.exports = {
+  //configFields: async ({ table, mode }) => [],
+  run: async ({ configuration, user, table, req, ...rest }) => {
+    const memtable = Table.findOne("AgentLongTermMemory");
+    if (!memtable) return;
+    // write personal preferences
+
+    // get all personal observations
+    const personalmems = await memtable.getRows(
+      { personal: true },
+      { orderBy: "written_at" },
+    );
+
+    const user_ids = new Set(personalmems.map((p) => p.user_id));
+
+    for (const uid of user_ids.values()) {
+      const all_umems = personalmems.filter((m) => m.user_id === uid);
+      const existingPP = all_umems.find(
+        (m) => m.memory_type === "PersonalPreferences",
+      );
+      const nonPPs = all_umems.filter(
+        (m) => m.memory_type !== "PersonalPreferences",
+      );
+
+      if (
+        existingPP &&
+        nonPPs.every((m) => m.written_at < existingPP.written_at)
+      )
+        continue; // no new memories
+      const prompt = `This is a set of observations about a user:
+${nonPPs.map((m) => `* ${m.contents}`).join("\n")}
+
+They are in ascending chronological order so if there are any contradictions, the latter observation take precedence.
+
+Write a succinct summary of these observations which captures all the essential facts. Start the summary with the words
+"The user..." (in the language in which the observations appear) and then write what you have learned about the user.`;
+
+      const answer = await getState().functions.llm_generate.run(prompt);
+
+      if (answer && typeof answer === "string") {
+        await memtable.deleteRows({
+          memory_type: "PersonalPreferences",
+          user_id: uid,
+        });
+        await memtable.insertRow({
+          user_id: uid,
+          written_at: new Date(),
+          memory_type: "PersonalPreferences",
+          contents: answer,
+          personal: true,
+        });
+      }
+    }
+  },
+};

package/index.js

@@ -36,6 +36,7 @@ module.exports = {
   ],
   actions: {
     Agent: require("./action"),
+    consolidate_agent_memory: require("./consolidate_agent_memory"),
   },
   functions: {
     inspect_agent: {
@@ -69,6 +70,7 @@ module.exports = {
         };
       },
       isAsync: true,
+      hidden: true,
       description: "Return system prompt, tools and action of an agent",
     },
     agent_generate: {
@@ -115,6 +117,7 @@ module.exports = {
         };
       },
       isAsync: true,
+      hidden: true,
       description: "Run an agent on a prompt",
       arguments: [
         { name: "agent_name", type: "String" },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saltcorn/agents",
-  "version": "0.8.5",
+  "version": "0.8.6",
   "description": "AI agents for Saltcorn",
   "main": "index.js",
   "dependencies": {

package/skills/LongTermMemory.js

@@ -20,12 +20,23 @@ class LongTermMemory {
     Object.assign(this, cfg);
   }
 
-  systemPrompt() {
+  async systemPrompt({ user }) {
+    let pp = "";
+    if (this.inject_personal_prefs && user?.id) {
+      const table = Table.findOne("AgentLongTermMemory");
+      if (table) {
+        const ppRow = await table.getRow({
+          memory_type: "PersonalPreferences",
+          user_id: user.id,
+        });
+        if (ppRow) pp = "\n\n" + ppRow.contents;
+      }
+    }
     return `You have access to a memory bank you can read from or write to. You should search the memory bank with the search_memory tool with any search terms that might be relevant to the user's query or the result of a tool call. When you learn something noteworthy (from the user or from the result of a tool call) store it in memory with the store_in_memory tool. Mark it as personal if it is only true or relevant for the specific user. Don't tell the user when you are storing to and retrieving from memory. ${
       this.add_sys_prompt
         ? ` Additional instructions for the memory tools: ${this.add_sys_prompt}`
         : ""
-    }`;
+    }${pp}`;
   }
 
   static async configFields() {
@@ -36,6 +47,13 @@ class LongTermMemory {
         type: "String",
         fieldview: "textarea",
       },
+      {
+        name: "inject_personal_prefs",
+        label: "Personal preferences",
+        type: "Bool",
+        sublabel:
+          "Inject personal preferences generated by the consolidate_agent_memory action in the system prompt",
+      },
     ];
   }