@saltcorn/agents 0.2.3 → 0.3.0

package/common.js

@@ -11,6 +11,7 @@ const get_skills = () => {
     require("./skills/EmbeddingRetrieval"),
     require("./skills/Trigger"),
     require("./skills/Table"),
+    require("./skills/PreloadData"),
     //require("./skills/AdaptiveFeedback"),
   ];
 };
@@ -44,16 +45,16 @@ const find_tool = (name, config) => {
   }
 };
 
-const getCompletionArguments = async (config) => {
+const getCompletionArguments = async (config, user) => {
   let tools = [];
 
   let sysPrompts = [config.sys_prompt];
 
   const skills = get_skill_instances(config);
   for (const skill of skills) {
-    const sysPr = skill.systemPrompt();
+    const sysPr = await skill.systemPrompt({ user });
     if (sysPr) sysPrompts.push(sysPr);
-    const skillTools = skill.provideTools();
+    const skillTools = skill.provideTools?.();
     if (skillTools && Array.isArray(skillTools)) tools.push(...skillTools);
     else if (skillTools) tools.push(skillTools);
   }
@@ -129,7 +130,7 @@ const process_interaction = async (
   agent_label = "Copilot",
   prevResponses = []
 ) => {
-  const complArgs = await getCompletionArguments(config);
+  const complArgs = await getCompletionArguments(config, req.user);
   complArgs.chat = run.context.interactions;
   //complArgs.debugResult = true;
   console.log("complArgs", JSON.stringify(complArgs, null, 2));

package/package.json

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

package/skills/PreloadData.js

@@ -0,0 +1,92 @@
+const { div, pre } = require("@saltcorn/markup/tags");
+const Workflow = require("@saltcorn/data/models/workflow");
+const Form = require("@saltcorn/data/models/form");
+const Table = require("@saltcorn/data/models/table");
+const View = require("@saltcorn/data/models/view");
+const { getState } = require("@saltcorn/data/db/state");
+const db = require("@saltcorn/data/db");
+const { eval_expression } = require("@saltcorn/data/models/expression");
+const { interpolate } = require("@saltcorn/data/utils");
+
+class PreloadData {
+  static skill_name = "Preload Data";
+
+  get skill_label() {
+    return `Preload Data`;
+  }
+
+  constructor(cfg) {
+    Object.assign(this, cfg);
+  }
+
+  async systemPrompt({ user }) {
+    const prompts = [];
+    if (this.add_sys_prompt) prompts.push(this.add_sys_prompt);
+    const table = Table.findOne(this.table_name);
+    const q = eval_expression(
+      this.preload_query,
+      {},
+      user,
+      "PreloadData query"
+    );
+
+    const rows = await table.getRows(q);
+    console.log("preload data rows", rows);
+    if (this.contents_expr) {
+      for (const row of rows)
+        prompts.push(interpolate(this.contents_expr, row, user));
+    } else {
+      const hidden_fields = this.hidden_fields.split(",").map((s) => s.trim());
+      for (const row of rows) {
+        hidden_fields.forEach((k) => {
+          delete row[k];
+        });
+        prompts.push(JSON.stringify(row));
+      }
+    }
+    return prompts.join("\n");
+  }
+
+  static async configFields() {
+    const allTables = await Table.find();
+
+    return [
+      {
+        name: "table_name",
+        label: "Table",
+        sublabel: "Which table to search",
+        type: "String",
+        required: true,
+        attributes: { options: allTables.map((t) => t.name) },
+      },
+      {
+        name: "preload_query",
+        label: "Query",
+        type: "String",
+        class: "validate-expression",
+      },
+      {
+        name: "add_sys_prompt",
+        label: "Additional prompt",
+        type: "String",
+        fieldview: "textarea",
+      },
+      {
+        name: "contents_expr",
+        label: "Contents string",
+        type: "String",
+        sublabel:
+          "Use handlebars (<code>{{ }}</code>) to access fields in each retrieved row",
+      },
+      {
+        name: "hidden_fields",
+        label: "Hide fields",
+        type: "String",
+        sublabel:
+          "Comma-separated list of fields to hide from the prompt, if not using the contents string",
+      },
+    ];
+  }
+}
+
+module.exports = PreloadData;