@saltcorn/large-language-model 0.8.5 → 0.8.7

package/index.js

@@ -1,6 +1,7 @@
 const Workflow = require("@saltcorn/data/models/workflow");
 const Form = require("@saltcorn/data/models/form");
 const File = require("@saltcorn/data/models/file");
+const User = require("@saltcorn/data/models/user");
 const FieldRepeat = require("@saltcorn/data/models/fieldrepeat");
 const Plugin = require("@saltcorn/data/models/plugin");
 const { domReady } = require("@saltcorn/markup/tags");
@@ -594,7 +595,27 @@ module.exports = {
     llm_generate_image: {
       description: "Generate image with AI based on a text prompt",
       requireRow: true,
-      configFields: ({ table, mode }) => {
+      configFields: async ({ table, mode }) => {
+        const roleOptions = (await User.get_roles()).map((r) => ({
+          value: r.id,
+          label: r.role,
+        }));
+        const commonFields = [
+          {
+            name: "filename",
+            label: "File name",
+            type: "String",
+            sublabel:
+              "Name of the generated file. Interpolations <code>{{ }}</code> available",
+          },
+          {
+            label: "Minimum role to access",
+            name: "min_role",
+            input_type: "select",
+            sublabel: "User must have this role or higher access image file",
+            options: roleOptions,
+          },
+        ];
         if (mode === "workflow") {
           return [
             {
@@ -620,6 +641,7 @@ module.exports = {
               sublabel: "Override default model name",
               type: "String",
             },
+            ...commonFields,
           ];
         } else if (table) {
           const textFields = table.fields
@@ -652,6 +674,7 @@ module.exports = {
               required: true,
               attributes: { options: fileFields },
             },
+            ...commonFields,
           ];
         }
       },
@@ -665,9 +688,9 @@ module.exports = {
           prompt_formula,
           prompt_template,
           answer_field,
-          override_config,
-          chat_history_field,
+          min_role,
           model,
+          filename,
         },
       }) => {
         let prompt;
@@ -681,6 +704,10 @@ module.exports = {
             "llm_generate prompt formula"
           );
         else prompt = row[prompt_field];
+
+        const use_filename = filename
+          ? interpolate(filename, row, user, "llm_generate_image file name")
+          : "generated.png";
         const opts = { debugResult: true }; // response_format: "b64_json" };
 
         if (model) opts.model = model;
@@ -697,10 +724,11 @@ module.exports = {
         } else if (ans.b64_json) {
           const imgContents = Buffer.from(ans.b64_json, "base64");
           const file = await File.from_contents(
-            "generated.png",
+            use_filename,
             "image/png",
             imgContents,
-            user?.id
+            user?.id,
+            min_role || 1
           );
           upd[answer_field] = file.path_to_serve;
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saltcorn/large-language-model",
-  "version": "0.8.5",
+  "version": "0.8.7",
   "description": "Large language models and functionality for Saltcorn",
   "main": "index.js",
   "dependencies": {